Changeset 39202 in vbox for trunk/src/VBox

Timestamp:

Nov 4, 2011 11:07:23 AM (13 years ago)

Author:

vboxsync

Message:

VBoxService/Windows: Added gentle shutdown when user does CTRL+C/CTRL+BREAK/close the console window.

File:

• trunk/src/VBox/Additions/common/VBoxService/VBoxService.cpp (modified) (3 diffs)

trunk/src/VBox/Additions/common/VBoxService/VBoxService.cpp

@@ -374 +374 @@
 
 
+#ifdef RT_OS_WINDOWS
+static BOOL WINAPI VBoxServiceConsoleControlHandler(DWORD dwCtrlType)
+{
+    int rc = VINF_SUCCESS;
+    bool fEventHandled = FALSE;
+    switch (dwCtrlType)
+    {
+        /* User pressed CTRL+C or CTRL+BREAK or an external event was sent
+         * via GenerateConsoleCtrlEvent(). */
+        case CTRL_BREAK_EVENT:
+        case CTRL_CLOSE_EVENT:
+        case CTRL_C_EVENT:
+            VBoxServiceVerbose(2, "ControlHandler: Received break/close event\n");
+            rc = VBoxServiceStopServices();
+            fEventHandled = TRUE;
+            break;
+        default:
+            break;
+        /** @todo Add other events here. */
+    }
+
+    if (RT_FAILURE(rc))
+        VBoxServiceError("ControlHandler: Event %ld handled with error rc=%Rrc\n",
+                         dwCtrlType, rc);
+    return fEventHandled;
+}
+#endif /* RT_OS_WINDOWS */
+
+
 /**
  * Starts the service.
@@ -555 +584 @@
     RTSemEventDestroy(g_hEvtWindowsService);
     g_hEvtWindowsService = NIL_RTSEMEVENT;
-
 #else
     /*
@@ -864 +892 @@
          *          and return immediately.
          */
+#ifdef RT_OS_WINDOWS
+# ifndef RT_OS_NT4
+        /* Install console control handler. */
+        if (!SetConsoleCtrlHandler((PHANDLER_ROUTINE)VBoxServiceConsoleControlHandler, TRUE /* Add handler */))
+        {
+            VBoxServiceError("Unable to add console control handler, error=%ld\n", GetLastError());
+            /* Just skip this error, not critical. */
+        }
+# endif /* !RT_OS_NT4 */
+#endif /* RT_OS_WINDOWS */
         rc = VBoxServiceStartServices();
         rcExit = RT_SUCCESS(rc) ? RTEXITCODE_SUCCESS : RTEXITCODE_FAILURE;
         if (RT_SUCCESS(rc))
             VBoxServiceMainWait();
+#ifdef RT_OS_WINDOWS
+# ifndef RT_OS_NT4
+        /* Uninstall console control handler. */
+        if (!SetConsoleCtrlHandler((PHANDLER_ROUTINE)NULL, FALSE /* Remove handler */))
+        {
+            VBoxServiceError("Unable to remove console control handler, error=%ld\n", GetLastError());
+            /* Just skip this error, not critical. */
+        }
+# endif /* !RT_OS_NT4 */
+#else /* !RT_OS_WINDOWS */
+        /* On Windows - since we're running as a console application - we already stopped all services
+         * through the console control handler. So only do the stopping of services here on other platforms
+         * where the break/shutdown/whatever signal was just received. */
         VBoxServiceStopServices();
+#endif /* RT_OS_WINDOWS */
     }
     VBoxServiceReportStatus(VBoxGuestFacilityStatus_Terminated);