Changeset 70615 in vbox for trunk/src/VBox/Runtime/r3/win

Timestamp:

Jan 17, 2018 8:28:16 PM (7 years ago)

Author:

vboxsync

Message:

RTSystemShutdown/win: More reliable power off, I hope.

File:

• trunk/src/VBox/Runtime/r3/win/RTSystemShutdown-win.cpp (modified) (2 diffs)

trunk/src/VBox/Runtime/r3/win/RTSystemShutdown-win.cpp

@@ -44 +44 @@
     AssertReturn(!(fFlags & ~RTSYSTEM_SHUTDOWN_VALID_MASK), VERR_INVALID_PARAMETER);
 
+    /*
+     * Before we start, try grant the necessary privileges.
+     */
+    DWORD  dwErr;
+    HANDLE hToken = NULL;
+    if (OpenThreadToken(GetCurrentThread(), TOKEN_ADJUST_PRIVILEGES, TRUE /*OpenAsSelf*/, &hToken))
+        dwErr = NO_ERROR;
+    else
+    {
+        dwErr  = GetLastError();
+        if (dwErr == ERROR_NO_TOKEN)
+        {
+            if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
+                dwErr = NO_ERROR;
+            else
+                dwErr = GetLastError();
+        }
+    }
+    if (dwErr == NO_ERROR)
+    {
+        union
+        {
+            TOKEN_PRIVILEGES TokenPriv;
+            char ab[sizeof(TOKEN_PRIVILEGES) + sizeof(LUID_AND_ATTRIBUTES)];
+        } u;
+        u.TokenPriv.PrivilegeCount = 1;
+        u.TokenPriv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
+        if (LookupPrivilegeValue(NULL /*localhost*/, SE_SHUTDOWN_NAME, &u.TokenPriv.Privileges[0].Luid))
+        {
+            if (!AdjustTokenPrivileges(hToken,
+                                       FALSE /*DisableAllPrivileges*/,
+                                       &u.TokenPriv,
+                                       RT_OFFSETOF(TOKEN_PRIVILEGES, Privileges[1]),
+                                       NULL,
+                                       NULL) )
+                dwErr = GetLastError();
+        }
+        else
+            dwErr = GetLastError();
+        CloseHandle(hToken);
+    }
+
+    /*
+     * Do some parameter conversion.
+     */
     PRTUTF16 pwszLogMsg;
     int rc = RTStrToUtf16(pszLogMsg, &pwszLogMsg);
@@ -49 +94 @@
         return rc;
     DWORD cSecsTimeout = (cMsDelay + 499) / 1000;
-    BOOL  fRebootAfterShutdown = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_REBOOT
-                               ? TRUE : FALSE;
-    BOOL  fForceAppsClosed = fFlags & RTSYSTEM_SHUTDOWN_FORCE ? TRUE : FALSE;
 
     /*
-     * Do the
+     * If we're told to power off the system, we should try use InitiateShutdownW (6.0+)
+     * or ExitWindowsEx (3.50) rather than InitiateSystemShutdownW, because these other
+     * APIs allows us to explicitly specify that we want to power off.
+     *
+     * Note! For NT version 4, 3.51, and 3.50 the system may instaed reboot since the
+     *       x86 HALs typically didn't know how to perform a power off.
      */
-    if (InitiateSystemShutdownW(NULL /*pwszMachineName = NULL = localhost*/,
-                                pwszLogMsg,
-                                cSecsTimeout,
-                                fForceAppsClosed,
-                                fRebootAfterShutdown))
-        rc = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_HALT ? VINF_SYS_MAY_POWER_OFF : VINF_SUCCESS;
-    else
+    bool fDone = false;
+    if (   (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_POWER_OFF
+        || (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_POWER_OFF_HALT)
     {
-        /* If we failed because of missing privileges, try get the right to
-           shut down the system and call the api again. */
-        DWORD dwErr = GetLastError();
-        rc = RTErrConvertFromWin32(dwErr);
-        if (dwErr == ERROR_ACCESS_DENIED)
+        /* This API has the grace period thing. */
+        decltype(InitiateShutdownW) *pfnInitiateShutdownW;
+        pfnInitiateShutdownW = (decltype(InitiateShutdownW) *)GetProcAddress(GetModuleHandleW(L"ADVAPI32.DLL"), "InitiateShutdownW");
+        if (pfnInitiateShutdownW)
         {
-            HANDLE hToken = NULL;
-            if (OpenThreadToken(GetCurrentThread(),
-                                TOKEN_ADJUST_PRIVILEGES,
-                                TRUE /*OpenAsSelf*/,
-                                &hToken))
-                dwErr = NO_ERROR;
-            else
+            DWORD fShutdownFlags = SHUTDOWN_POWEROFF;
+            if (fFlags & RTSYSTEM_SHUTDOWN_FORCE)
+                fShutdownFlags |= SHUTDOWN_FORCE_OTHERS | SHUTDOWN_FORCE_SELF;
+            DWORD fReason = SHTDN_REASON_MAJOR_OTHER | (fFlags & RTSYSTEM_SHUTDOWN_PLANNED ? SHTDN_REASON_FLAG_PLANNED : 0);
+            dwErr = pfnInitiateShutdownW(NULL /*pwszMachineName*/, pwszLogMsg, cSecsTimeout, fShutdownFlags, fReason);
+            if (dwErr == ERROR_INVALID_PARAMETER)
             {
-                dwErr  = GetLastError();
-                if (dwErr == ERROR_NO_TOKEN)
-                {
-                    if (OpenProcessToken(GetCurrentProcess(),
-                                         TOKEN_ADJUST_PRIVILEGES,
-                                         &hToken))
-                        dwErr = NO_ERROR;
-                    else
-                        dwErr = GetLastError();
-                }
+                fReason &= ~SHTDN_REASON_FLAG_PLANNED; /* just in case... */
+                dwErr = pfnInitiateShutdownW(NULL /*pwszMachineName*/, pwszLogMsg, cSecsTimeout, fShutdownFlags, fReason);
             }
-
-            if (dwErr == NO_ERROR)
+            if (dwErr == ERROR_SUCCESS)
             {
-                union
-                {
-                    TOKEN_PRIVILEGES TokenPriv;
-                    char ab[sizeof(TOKEN_PRIVILEGES) + sizeof(LUID_AND_ATTRIBUTES)];
-                } u;
-                u.TokenPriv.PrivilegeCount = 1;
-                u.TokenPriv.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
-                if (LookupPrivilegeValue(NULL /*localhost*/, SE_SHUTDOWN_NAME, &u.TokenPriv.Privileges[0].Luid))
-                {
-                    if (AdjustTokenPrivileges(hToken,
-                                              FALSE /*DisableAllPrivileges*/,
-                                              &u.TokenPriv,
-                                              RT_OFFSETOF(TOKEN_PRIVILEGES, Privileges[1]),
-                                              NULL,
-                                              NULL) )
-                    {
-                        if (InitiateSystemShutdownW(NULL /*pwszMachineName = NULL = localhost*/,
-                                                    pwszLogMsg,
-                                                    cSecsTimeout,
-                                                    fForceAppsClosed,
-                                                    fRebootAfterShutdown))
-                            rc = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_HALT ? VINF_SYS_MAY_POWER_OFF : VINF_SUCCESS;
-                        else
-                        {
-                            dwErr = GetLastError();
-                            rc = RTErrConvertFromWin32(dwErr);
-                        }
-                    }
-                    CloseHandle(hToken);
-                }
+                rc = VINF_SUCCESS;
+                fDone = true;
             }
         }
+
+        if (!fDone)
+        {
+            /* No grace period here, too bad. */
+            decltype(ExitWindowsEx) *pfnExitWindowsEx;
+            pfnExitWindowsEx = (decltype(ExitWindowsEx) *)GetProcAddress(GetModuleHandleW(L"USER32.DLL"), "ExitWindowsEx");
+            if (pfnExitWindowsEx)
+            {
+                DWORD fExitWindows = EWX_POWEROFF | EWX_SHUTDOWN;
+                if (fFlags & RTSYSTEM_SHUTDOWN_FORCE)
+                    fExitWindows |= EWX_FORCE | EWX_FORCEIFHUNG;
+
+                if (pfnExitWindowsEx(fExitWindows, SHTDN_REASON_MAJOR_OTHER))
+                    fDone = true;
+                else if (pfnExitWindowsEx(fExitWindows & ~EWX_FORCEIFHUNG, SHTDN_REASON_MAJOR_OTHER))
+                    fDone = true;
+            }
+        }
+    }
+
+    /*
+     * Fall back on the oldest API.
+     */
+    if (!fDone)
+    {
+        BOOL fRebootAfterShutdown = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_REBOOT
+                                   ? TRUE : FALSE;
+        BOOL fForceAppsClosed     = fFlags & RTSYSTEM_SHUTDOWN_FORCE ? TRUE : FALSE;
+        if (InitiateSystemShutdownW(NULL /*pwszMachineName = NULL = localhost*/,
+                                    pwszLogMsg,
+                                    cSecsTimeout,
+                                    fForceAppsClosed,
+                                    fRebootAfterShutdown))
+            rc = (fFlags & RTSYSTEM_SHUTDOWN_ACTION_MASK) == RTSYSTEM_SHUTDOWN_HALT ? VINF_SYS_MAY_POWER_OFF : VINF_SUCCESS;
+        else
+            rc = RTErrConvertFromWin32(dwErr);
     }