Accounting for transport enabled

Chykary · Aug 19, 2020 · ec84968 · ec84968
1 parent 8b2e135
commit ec84968
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/Common.cs b/Common.cs
@@ -91,7 +91,7 @@ public void ReceiveData()
         {
             try
             {
-                while (Receive(out SteamId clientSteamID, out byte[] internalMessage, internal_ch))
+                while (transport.enabled && Receive(out SteamId clientSteamID, out byte[] internalMessage, internal_ch))
                 {
                     if (internalMessage.Length == 1)
                     {
@@ -106,7 +106,7 @@ public void ReceiveData()
 
                 for (int chNum = 0; chNum < channels.Length; chNum++)
                 {
-                    while (Receive(out SteamId clientSteamID, out byte[] receiveBuffer, chNum))
+                    while (transport.enabled && Receive(out SteamId clientSteamID, out byte[] receiveBuffer, chNum))
                     {
                         OnReceiveData(receiveBuffer, clientSteamID, chNum);
                     }

diff --git a/FizzyFacepunch.cs b/FizzyFacepunch.cs
@@ -46,8 +46,11 @@ private void Awake()
 
         private void LateUpdate()
         {
-            SteamClient.RunCallbacks();
-            activeNode?.ReceiveData();
+            if (enabled)
+            {
+                SteamClient.RunCallbacks();
+                activeNode?.ReceiveData();
+            }
         }
 
         public override bool ClientConnected() => ClientActive() && client.Connected;