Add update/reboot and update/shutdown menu options

Author: dmex

ProcessHacker/actions.c

@@ -911,6 +911,33 @@ BOOLEAN PhUiRestartComputer(
             }
         }
         break;
+    case PH_POWERACTION_TYPE_UPDATE:
+        {
+            if (!PhGetIntegerSetting(L"EnableWarnings") || PhShowConfirmMessage(
+                WindowHandle,
+                L"update and restart",
+                L"the computer",
+                NULL,
+                FALSE
+                ))
+            {
+                ULONG status;
+
+                status = InitiateShutdown(
+                    NULL,
+                    NULL,
+                    0,
+                    SHUTDOWN_RESTART | SHUTDOWN_INSTALL_UPDATES,
+                    SHTDN_REASON_FLAG_PLANNED
+                    );
+
+                if (status == ERROR_SUCCESS)
+                    return TRUE;
+
+                PhShowStatus(WindowHandle, L"Unable to restart the computer.", 0, status);
+            }
+        }
+        break;
     }
 
     return FALSE;
@@ -1026,6 +1053,33 @@ BOOLEAN PhUiShutdownComputer(
             }
         }
         break;
+    case PH_POWERACTION_TYPE_UPDATE:
+        {
+            if (!PhGetIntegerSetting(L"EnableWarnings") || PhShowConfirmMessage(
+                WindowHandle,
+                L"update and shutdown",
+                L"the computer",
+                NULL,
+                FALSE
+                ))
+            {
+                ULONG status;
+
+                status = InitiateShutdown(
+                    NULL,
+                    NULL,
+                    0,
+                    SHUTDOWN_POWEROFF | SHUTDOWN_INSTALL_UPDATES,
+                    SHTDN_REASON_FLAG_PLANNED
+                    );
+
+                if (status == ERROR_SUCCESS)
+                    return TRUE;
+
+                PhShowStatus(WindowHandle, L"Unable to shut down the computer.", 0, status);
+            }
+        }
+        break;
     }
 
     return FALSE;

ProcessHacker/include/actions.h

@@ -80,6 +80,7 @@ typedef enum _PH_POWERACTION_TYPE
     PH_POWERACTION_TYPE_CRITICAL,
     PH_POWERACTION_TYPE_ADVANCEDBOOT,
     PH_POWERACTION_TYPE_FIRMWAREBOOT,
+    PH_POWERACTION_TYPE_UPDATE,
     PH_POWERACTION_TYPE_MAXIMUM
 } PH_POWERACTION_TYPE;
 

ProcessHacker/mainwnd.c

@@ -2191,6 +2191,12 @@ BOOLEAN PhMwpExecuteComputerCommand(
     case ID_COMPUTER_SHUTDOWN_CRITICAL:
         PhUiShutdownComputer(WindowHandle, PH_POWERACTION_TYPE_CRITICAL, 0);
         return TRUE;
+    case ID_COMPUTER_RESTART_UPDATE:
+        PhUiRestartComputer(WindowHandle, PH_POWERACTION_TYPE_UPDATE, 0);
+        return TRUE;
+    case ID_COMPUTER_SHUTDOWN_UPDATE:
+        PhUiShutdownComputer(WindowHandle, PH_POWERACTION_TYPE_UPDATE, 0);
+        return TRUE;
     }
 
     return FALSE;
@@ -2263,6 +2269,9 @@ PPH_EMENU PhpCreateComputerMenu(
     PhInsertEMenuItem(menuItem, PhCreateEMenuItem(0, ID_COMPUTER_SLEEP, L"&Sleep", NULL, NULL), ULONG_MAX);
     PhInsertEMenuItem(menuItem, PhCreateEMenuItem(0, ID_COMPUTER_HIBERNATE, L"&Hibernate", NULL, NULL), ULONG_MAX);
     PhInsertEMenuItem(menuItem, PhCreateEMenuSeparator(), ULONG_MAX);
+    PhInsertEMenuItem(menuItem, PhCreateEMenuItem(0, ID_COMPUTER_RESTART_UPDATE, L"Update and restart", NULL, NULL), ULONG_MAX);
+    PhInsertEMenuItem(menuItem, PhCreateEMenuItem(0, ID_COMPUTER_SHUTDOWN_UPDATE, L"Update and shut down", NULL, NULL), ULONG_MAX);
+    PhInsertEMenuItem(menuItem, PhCreateEMenuSeparator(), ULONG_MAX);
     PhInsertEMenuItem(menuItem, PhCreateEMenuItem(0, ID_COMPUTER_RESTART, L"R&estart", NULL, NULL), ULONG_MAX);
     PhInsertEMenuItem(menuItem, PhCreateEMenuItem(0, ID_COMPUTER_RESTARTADVOPTIONS, L"Restart to advanced options", NULL, NULL), ULONG_MAX);
     PhInsertEMenuItem(menuItem, PhCreateEMenuItem(0, ID_COMPUTER_RESTARTBOOTOPTIONS, L"Restart to boot options", NULL, NULL), ULONG_MAX);