Added check in channel loop to see if client had disconnected

nickproud · Jul 15, 2023 · a9d391f · a9d391f
1 parent cbaa346
commit a9d391f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 11 deletions.
diff --git a/Channel.cs b/Channel.cs
@@ -38,20 +38,26 @@ public void Open(TcpClient client)
 
                 while(isOpen)
                 {
-                    while ((position = stream.Read(buffer, 0, buffer.Length)) != 0 && isOpen)
+                    if (clientDisconnected())
                     {
-                        data = Encoding.UTF8.GetString(buffer, 0, position);
-                        var args = new DataReceivedArgs()
+                        Close();
+                    }
+                    else
+                    {
+                        while ((position = stream.Read(buffer, 0, buffer.Length)) != 0 && isOpen)
                         {
-                            Message = data,
-                            ConnectionId = Id,
-                            ThisChannel = this
-                        };
+                            data = Encoding.UTF8.GetString(buffer, 0, position);
+                            var args = new DataReceivedArgs()
+                            {
+                                Message = data,
+                                ConnectionId = Id,
+                                ThisChannel = this
+                            };
 
-                        thisServer.OnDataIn(args);
-                        if(!isOpen) { break; }
+                            thisServer.OnDataIn(args);
+                            if(!isOpen) { break; }
+                        }
                     }
-
                 }
             }
         }
@@ -62,6 +68,7 @@ public void Send(string message)
             stream.Write(data, 0, data.Length);
         }
 
+
         public void Close()
         {
             Dispose(false);
@@ -88,5 +95,10 @@ public void Dispose()
             Dispose(disposing: true);
             GC.SuppressFinalize(this);
         }
+
+        private bool clientDisconnected()
+        {
+            return (thisClient.Client.Available == 0 && thisClient.Client.Poll(1, SelectMode.SelectRead));
+        }
     }
 }
diff --git a/EasyTCP.csproj b/EasyTCP.csproj
@@ -1,7 +1,7 @@
 <Project Sdk="Microsoft.NET.Sdk">
 
   <PropertyGroup>
-    <TargetFramework>netcoreapp3.1</TargetFramework>
+    <TargetFramework>net6.0</TargetFramework>
   </PropertyGroup>
 
 </Project>