fix: setting reason too None if packet length is not 3

MirageNet · Apr 10, 2023 · dcc4fa9 · dcc4fa9
1 parent 3c369b8
commit dcc4fa9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Assets/Mirage/Runtime/SocketLayer/Peer.cs b/Assets/Mirage/Runtime/SocketLayer/Peer.cs
@@ -542,7 +542,12 @@ internal void RemoveConnection(Connection connection)
 
         private void HandleConnectionDisconnect(Connection connection, Packet packet)
         {
-            var reason = (DisconnectReason)packet.Buffer.array[2];
+            DisconnectReason reason;
+            if (packet.Length == 3)
+                reason = (DisconnectReason)packet.Buffer.array[2];
+            else
+                reason = DisconnectReason.None;
+
             connection.Disconnect(reason, false);
         }