fix(lib): network exception

GameProgressive · Nov 3, 2023 · d7eb831 · d7eb831
1 parent 8887cf1
commit d7eb831
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/src/Libraries/Core/src/Network/Http/Server/HttpConnectionManager.cs b/src/Libraries/Core/src/Network/Http/Server/HttpConnectionManager.cs
@@ -15,7 +15,21 @@ public class HttpConnectionManager : IConnectionManager
     public HttpConnectionManager(IPEndPoint endPoint)
     {
         Listener = new HttpListener();
-        Listener.Prefixes.Add($"http://localhost:{endPoint.Port}/");
+        string host;
+        if (endPoint.Address.ToString() == "0.0.0.0")
+        {
+            host = "+";
+        }
+        else if (endPoint.Address.ToString() == "127.0.0.1")
+        {
+            host = "localhost";
+        }
+        else
+        {
+            host = endPoint.Address.ToString();
+        }
+
+        Listener.Prefixes.Add($"http://{host}:{endPoint.Port}/");
     }
     public void Start()
     {

diff --git a/src/Libraries/Core/src/Network/Tcp/Server/TcpConnection.cs b/src/Libraries/Core/src/Network/Tcp/Server/TcpConnection.cs
@@ -1,4 +1,3 @@
-using System;
 using System.Linq;
 using System.Net;
 using System.Net.Sockets;
@@ -47,13 +46,14 @@ public void OnConnected()
 
         private void StartReceiving()
         {
-            var stream = Client.GetStream();
-            byte[] buffer = new byte[2048];
-            int bytesRead;
+
             while (true)
             {
                 try
                 {
+                    var stream = Client.GetStream();
+                    byte[] buffer = new byte[2048];
+                    int bytesRead;
                     bytesRead = stream.Read(buffer, 0, buffer.Length);
                     if (bytesRead == 0)
                     {
@@ -64,7 +64,7 @@ private void StartReceiving()
                     var receivedData = buffer.Take(bytesRead).ToArray();
                     OnReceive(receivedData);
                 }
-                catch (Exception ex)
+                catch (SocketException ex)
                 {
                     LogWriter.LogError(ex);
                     OnDisconnected();