fixed a bug about receiving package

kerryjiang · May 25, 2024 · 696e7b5 · 696e7b5
1 parent f097935
commit 696e7b5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/WebSocket4Net/WebSocket.cs b/src/WebSocket4Net/WebSocket.cs
@@ -204,6 +204,9 @@ public async ValueTask<WebSocketPackage> ReceiveAsync(bool returnControlPackage)
         {
             var package = await base.ReceiveAsync();
 
+            if (package == null)
+                return null;
+
             if (package.OpCode != OpCode.Binary && package.OpCode != OpCode.Text && package.OpCode != OpCode.Handshake)
             {
                 await HandleControlPackage(package);