diff --git a/CollabVMSharp/CollabVMClient.cs b/CollabVMSharp/CollabVMClient.cs index 2e9f84e..2229c28 100644 --- a/CollabVMSharp/CollabVMClient.cs +++ b/CollabVMSharp/CollabVMClient.cs @@ -173,7 +173,16 @@ public class CollabVMClient { MemoryStream ms = new(); WebSocketReceiveResult res; do { - res = await socket.ReceiveAsync(receivebuffer, CancellationToken.None); + try { + res = await socket.ReceiveAsync(receivebuffer, CancellationToken.None); + } catch (WebSocketException e) { +#if DEBUG + Console.Error.WriteLine($"Got {e.Message} while reading from WebSocket, closing connection"); +#endif + Cleanup(true); + return; + } + if (res.MessageType == WebSocketMessageType.Close) { this.Disconnect(); return;