Add errors for strange situations. Fix #555

RevenantX · Jul 18, 2024 · 132a807 · 132a807
1 parent e99f35f
commit 132a807
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/LiteNetLib/NetManager.HashSet.cs b/LiteNetLib/NetManager.HashSet.cs
@@ -85,6 +85,11 @@ private void ClearPeerSet()
 
         private bool ContainsPeer(NetPeer item)
         {
+            if (item == null)
+            {
+                NetDebug.WriteError($"Contains peer null: {item}");
+                return false;
+            }
             if (_buckets != null)
             {
                 int hashCode = item.GetHashCode() & Lower31BitMask;
@@ -121,6 +126,11 @@ public bool TryGetPeerById(int id, out NetPeer peer)
 
         private void AddPeer(NetPeer peer)
         {
+            if (peer == null)
+            {
+                NetDebug.WriteError($"Add peer null: {peer}");
+                return;
+            }
             _peersLock.EnterWriteLock();
             if (_headPeer != null)
             {
@@ -149,6 +159,11 @@ private void RemovePeer(NetPeer peer)
 
         private void RemovePeerInternal(NetPeer peer)
         {
+            if (peer == null)
+            {
+                NetDebug.WriteError($"Remove peer null: {peer}");
+                return;
+            }
             if (!RemovePeerFromSet(peer))
                 return;
             if (peer == _headPeer)

diff --git a/LiteNetLibSampleUnity/Assets/LiteNetLib.dll b/LiteNetLibSampleUnity/Assets/LiteNetLib.dll