Addendum #3 to dd0d929 (Added security features to the loader)

multitheftauto · Aug 1, 2016 · 9fd56d8 · 9fd56d8
1 parent c4ef69a
commit 9fd56d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/Client/loader/MainFunctions.cpp b/Client/loader/MainFunctions.cpp
@@ -1272,6 +1272,7 @@ void HandleOnQuitCommand( void )
         {
             strOperation = "open";
             strFile = PathJoin( strMTASAPath, MTA_EXE_NAME );
+            CheckService ( CHECK_SERVICE_RESTART_GAME );
         }
         else
             CheckService ( CHECK_SERVICE_POST_GAME );     // Stop service here if quit command is not 'restart'

diff --git a/Client/loader/Utils.h b/Client/loader/Utils.h
@@ -28,6 +28,7 @@ enum
     CHECK_SERVICE_POST_GAME = 5,
     CHECK_SERVICE_PRE_CREATE = 6,
     CHECK_SERVICE_POST_CREATE = 7,
+    CHECK_SERVICE_RESTART_GAME = 8,
 };
 
 #ifndef PROCESS_CREATION_MITIGATION_POLICY_IMAGE_LOAD_PREFER_SYSTEM32_ALWAYS_ON