From 10b2a55df1d98fc9a777f540231a50b6dfe6aed9 Mon Sep 17 00:00:00 2001 From: "Frank A. Krueger" Date: Thu, 15 Jun 2017 21:29:57 -0700 Subject: [PATCH] Cancel session if a bad message is received --- Ooui/UI.cs | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Ooui/UI.cs b/Ooui/UI.cs index 70fd545..32e9766 100644 --- a/Ooui/UI.cs +++ b/Ooui/UI.cs @@ -366,9 +366,11 @@ namespace Ooui if (receiveResult.MessageType == WebSocketMessageType.Close) { await webSocket.CloseAsync (WebSocketCloseStatus.NormalClosure, "", token).ConfigureAwait (false); + sessionCts.Cancel (); } else if (receiveResult.MessageType == WebSocketMessageType.Binary) { await webSocket.CloseAsync (WebSocketCloseStatus.InvalidMessageType, "Cannot accept binary frame", token).ConfigureAwait (false); + sessionCts.Cancel (); } else { var size = receiveResult.Count;