fix: 连接tcp失败时有概率崩溃软件

chenxuuu · Oct 16, 2024 · b1d1c2f · b1d1c2f
1 parent b5a7670
commit b1d1c2f
Showing 1 changed file with 33 additions and 5 deletions.
diff --git a/llcom/Pages/SocketClientPage.xaml.cs b/llcom/Pages/SocketClientPage.xaml.cs
@@ -142,7 +142,7 @@ private void ConnectButton_Click(object sender, RoutedEventArgs e)
                         return;
                     }
 
-                    if(so.isSSL)
+                    if (so.isSSL)
                     {
                         var networkStream = new NetworkStream(s);
                         var ssl = new SslStream(
@@ -155,7 +155,7 @@ private void ConnectButton_Click(object sender, RoutedEventArgs e)
                         {
                             ssl.AuthenticateAsClient("llcom tcp ssl client");
                         }
-                        catch(Exception ssle)
+                        catch (Exception ssle)
                         {
                             ShowData($"❗ SSL error {ssle.Message}");
                             socketNow = null;
@@ -166,13 +166,41 @@ private void ConnectButton_Click(object sender, RoutedEventArgs e)
                             ShowData("❌ Server disconnected");
                             return;
                         }
-                        socketNow = new SocketObj(ssl);
-                        ssl.BeginRead(so.buffer, 0, StateObject.BUFFER_SIZE, new AsyncCallback(Read_Callback), so);
+                        try
+                        {
+                            socketNow = new SocketObj(ssl);
+                            ssl.BeginRead(so.buffer, 0, StateObject.BUFFER_SIZE, new AsyncCallback(Read_Callback), so);
+                        }
+                        catch (Exception ex)
+                        {
+                            ShowData($"❗ Server connect error {ex.Message}");
+                            socketNow = null;
+                            IsConnected = false;
+                            Changeable = true;
+                            s.Close();
+                            s.Dispose();
+                            ShowData("❌ Server disconnected");
+                            return;
+                        }
                     }
                     else
                     {
                         so.workSocket = s;
-                        s.BeginReceive(so.buffer, 0, StateObject.BUFFER_SIZE, 0, new AsyncCallback(Read_Callback), so);
+                        try
+                        {
+                            s.BeginReceive(so.buffer, 0, StateObject.BUFFER_SIZE, 0, new AsyncCallback(Read_Callback), so);
+                        }
+                        catch(Exception ex)
+                        {
+                            ShowData($"❗ Server connect error {ex.Message}");
+                            socketNow = null;
+                            IsConnected = false;
+                            Changeable = true;
+                            s.Close();
+                            s.Dispose();
+                            ShowData("❌ Server disconnected");
+                            return;
+                        }
                     }
                 }), s);
             }