Changeset 39106 in vbox for trunk/src/VBox/Additions/common/VBoxService/VBoxServiceVMInfo.cpp

Timestamp:

Oct 25, 2011 9:45:40 AM (13 years ago)

Author:

vboxsync

Message:

VBoxService/VMInfo: Hook CTRL_LOGOFF_EVENT to re-trigger user enumeration on Windows guests to get immediate disconnects from VRDP when a user has been logged out.

File:

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

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

@@ -83 +83 @@
 
 
+#ifdef RT_OS_WINDOWS
+static BOOL WINAPI VBoxServiceVMInfoConsoleControlHandler(DWORD dwCtrlType)
+{
+    int rc = VINF_SUCCESS;
+    bool fEventHandled = FALSE;
+    switch (dwCtrlType)
+    {
+        case CTRL_LOGOFF_EVENT:
+            VBoxServiceVerbose(2, "VMInfo: Received logged-off event\n");
+            /* Trigger a re-enumeration of all logged-in users by unblocking
+             * the multi event semaphore of the VMInfo thread. */
+            if (g_hVMInfoEvent)
+                rc = RTSemEventMultiSignal(g_hVMInfoEvent);
+            fEventHandled = TRUE;
+            break;
+        default:
+            break;
+        /** @todo Add other events here. */
+    }
+
+    if (RT_FAILURE(rc))
+        VBoxServiceError("VMInfo: Event %ld handled with error rc=%Rrc\n",
+                         dwCtrlType, rc);
+    return fEventHandled;
+}
+#endif /* RT_OS_WINDOWS */
+
+
 /** @copydoc VBOXSERVICE::pfnPreInit */
 static DECLCALLBACK(int) VBoxServiceVMInfoPreInit(void)
@@ -154 +182 @@
         VBoxServicePropCacheUpdateEntry(&g_VMInfoPropCache, "/VirtualBox/GuestInfo/Net/Count",
                                         VBOXSERVICEPROPCACHEFLAG_TEMPORARY | VBOXSERVICEPROPCACHEFLAG_ALWAYS_UPDATE, NULL /* Delete on exit */);
+
+#ifdef RT_OS_WINDOWS
+# ifndef RT_OS_NT4
+    /* Install console control handler. */
+    if (!SetConsoleCtrlHandler((PHANDLER_ROUTINE)VBoxServiceVMInfoConsoleControlHandler, TRUE /* Add handler */))
+    {
+        VBoxServiceError("VMInfo: Unable to add console control handler, error=%ld\n", GetLastError());
+        /* Just skip this error, not critical. */
+    }
+# endif /* !RT_OS_NT4 */
+#endif /* RT_OS_WINDOWS */
     }
     return rc;
@@ -812 +851 @@
             break;
         }
+        else if (RT_LIKELY(RT_SUCCESS(rc2)))
+        {
+            /* Reset event semaphore if it got triggered. */
+            rc2 = RTSemEventMultiReset(g_hVMInfoEvent);
+            if (RT_FAILURE(rc2))
+                rc2 = VBoxServiceError("VMInfo: RTSemEventMultiReset failed; rc2=%Rrc\n", rc2);
+        }
     }
 
@@ -833 +879 @@
 {
     int rc;
+
+#ifdef RT_OS_WINDOWS
+# ifndef RT_OS_NT4
+    /* Install console control handler. */
+    if (!SetConsoleCtrlHandler((PHANDLER_ROUTINE)NULL, FALSE /* Remove handler */))
+    {
+        VBoxServiceError("VMInfo: Unable to remove console control handler, error=%ld\n", GetLastError());
+        /* Just skip this error, not critical. */
+    }
+# endif /* !RT_OS_NT4 */
+#endif
 
     if (g_hVMInfoEvent != NIL_RTSEMEVENTMULTI)