From df29ed5a6fb290ed9e3dfdf64e0953f8f39df93c Mon Sep 17 00:00:00 2001
From: Luke Pulverenti <luke.pulverenti@gmail.com>
Date: Fri, 27 Sep 2013 17:08:51 -0400
Subject: [PATCH] cancel web socket operations on dispose

---
 .../HttpServer/NativeWebSocket.cs                     | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/MediaBrowser.Server.Implementations/HttpServer/NativeWebSocket.cs b/MediaBrowser.Server.Implementations/HttpServer/NativeWebSocket.cs
index 6d6aa63250..a91dbbe4cf 100644
--- a/MediaBrowser.Server.Implementations/HttpServer/NativeWebSocket.cs
+++ b/MediaBrowser.Server.Implementations/HttpServer/NativeWebSocket.cs
@@ -25,6 +25,8 @@ namespace MediaBrowser.Server.Implementations.HttpServer
         /// <value>The web socket.</value>
         private System.Net.WebSockets.WebSocket WebSocket { get; set; }
 
+        private CancellationTokenSource _cancellationTokenSource = new CancellationTokenSource();
+
         /// <summary>
         /// Initializes a new instance of the <see cref="NativeWebSocket" /> class.
         /// </summary>
@@ -79,7 +81,7 @@ namespace MediaBrowser.Server.Implementations.HttpServer
 
                 try
                 {
-                    bytes = await ReceiveBytesAsync(CancellationToken.None).ConfigureAwait(false);
+                    bytes = await ReceiveBytesAsync(_cancellationTokenSource.Token).ConfigureAwait(false);
                 }
                 catch (OperationCanceledException)
                 {
@@ -144,7 +146,9 @@ namespace MediaBrowser.Server.Implementations.HttpServer
                 _logger.Warn("Unrecognized WebSocketMessageType: {0}", type.ToString());
             }
 
-            return WebSocket.SendAsync(new ArraySegment<byte>(bytes), nativeType, true, cancellationToken);
+            var linkedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, _cancellationTokenSource.Token);
+
+            return WebSocket.SendAsync(new ArraySegment<byte>(bytes), nativeType, true, linkedTokenSource.Token);
         }
 
         /// <summary>
@@ -163,6 +167,9 @@ namespace MediaBrowser.Server.Implementations.HttpServer
         {
             if (dispose)
             {
+                _cancellationTokenSource.Cancel();
+                _cancellationTokenSource.Dispose();
+
                 WebSocket.Dispose();
             }
         }