[UMPNPMGR] PNP_GetGlobalState returns CM_GLOBAL_STATE_SHUTTING_DOWN o…

…n service shutdown
DarkFire01 · Nov 1, 2024 · 91bc5b5 · 91bc5b5
1 parent 978503c
commit 91bc5b5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/base/services/umpnpmgr/precomp.h b/base/services/umpnpmgr/precomp.h
@@ -88,6 +88,7 @@ RpcServerThread(
 extern HKEY hEnumKey;
 extern HKEY hClassKey;
 extern BOOL g_IsUISuppressed;
+extern BOOL g_ShuttingDown;
 
 BOOL
 GetSuppressNewUIValue(VOID);

diff --git a/base/services/umpnpmgr/rpcserver.c b/base/services/umpnpmgr/rpcserver.c
@@ -687,6 +687,9 @@ PNP_GetGlobalState(
 
     *pulState = CM_GLOBAL_STATE_CAN_DO_UI | CM_GLOBAL_STATE_SERVICES_AVAILABLE;
 
+    if (g_ShuttingDown)
+        *pulState |= CM_GLOBAL_STATE_SHUTTING_DOWN;
+
     return CR_SUCCESS;
 }
 

diff --git a/base/services/umpnpmgr/umpnpmgr.c b/base/services/umpnpmgr/umpnpmgr.c
@@ -44,6 +44,7 @@ static SERVICE_STATUS ServiceStatus;
 HKEY hEnumKey = NULL;
 HKEY hClassKey = NULL;
 BOOL g_IsUISuppressed = FALSE;
+BOOL g_ShuttingDown = FALSE;
 
 /* FUNCTIONS *****************************************************************/
 
@@ -52,6 +53,9 @@ UpdateServiceStatus(
     _In_ DWORD dwState,
     _In_ DWORD dwCheckPoint)
 {
+    if ((dwState == SERVICE_STOPPED) || (dwState == SERVICE_STOP_PENDING))
+        g_ShuttingDown = TRUE;
+
     ServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
     ServiceStatus.dwCurrentState = dwState;
     ServiceStatus.dwWin32ExitCode = 0;