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 /// The web socket. private System.Net.WebSockets.WebSocket WebSocket { get; set; } + private CancellationTokenSource _cancellationTokenSource = new CancellationTokenSource(); + /// /// Initializes a new instance of the class. /// @@ -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(bytes), nativeType, true, cancellationToken); + var linkedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, _cancellationTokenSource.Token); + + return WebSocket.SendAsync(new ArraySegment(bytes), nativeType, true, linkedTokenSource.Token); } /// @@ -163,6 +167,9 @@ namespace MediaBrowser.Server.Implementations.HttpServer { if (dispose) { + _cancellationTokenSource.Cancel(); + _cancellationTokenSource.Dispose(); + WebSocket.Dispose(); } }