Changeset 45703 in vbox for trunk/src/VBox/Additions/WINNT/VBoxHook

Timestamp:

Apr 24, 2013 2:31:56 PM (12 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

85249

Message:

VBoxTray/Seamless: disable seamless on desktop change (UAC, Lock)

Location:

trunk/src/VBox/Additions/WINNT/VBoxHook

Files:

• VBoxHook.cpp (modified) (4 diffs)
• VBoxHook.def (modified) (1 diff)
• testcase/tstHook.cpp (modified) (1 diff)

trunk/src/VBox/Additions/WINNT/VBoxHook/VBoxHook.cpp

@@ -19 +19 @@
 
 #pragma data_seg("SHARED")
-static HWINEVENTHOOK    hEventHook[2]    = {0};
+static HWINEVENTHOOK    hWinEventHook[2]    = {0};
+static HWINEVENTHOOK    hDesktopEventHook   = NULL;
 #pragma data_seg()
 #pragma comment(linker, "/section:SHARED,RWS")
 
-static HANDLE   hNotifyEvent = 0;
+static HANDLE   hWinNotifyEvent = 0;
+static HANDLE   hDesktopNotifyEvent = 0;
 
 #ifdef DEBUG
@@ -33 +35 @@
 
 
-void CALLBACK VBoxHandleWinEvent(HWINEVENTHOOK hook, DWORD event, HWND hwnd,
+static void CALLBACK VBoxHandleWinEvent(HWINEVENTHOOK hook, DWORD event, HWND hwnd,
                                  LONG idObject, LONG idChild,
                                  DWORD dwEventThread, DWORD dwmsEventTime)
@@ -76 +78 @@
         }
 #endif
-        if (!hNotifyEvent)
+        if (!hWinNotifyEvent)
         {
-            hNotifyEvent = OpenEvent(EVENT_MODIFY_STATE, FALSE, VBOXHOOK_GLOBAL_EVENT_NAME);
-            dprintf(("OpenEvent returned %x (last err=%x)\n", hNotifyEvent, GetLastError()));
+            hWinNotifyEvent = OpenEvent(EVENT_MODIFY_STATE, FALSE, VBOXHOOK_GLOBAL_WT_EVENT_NAME);
+            dprintf(("OpenEvent returned %x (last err=%x)\n", hWinNotifyEvent, GetLastError()));
         }
-        BOOL ret = SetEvent(hNotifyEvent);
-        dprintf(("SetEvent %x returned %d (last error %x)\n", hNotifyEvent, ret, GetLastError()));
+        BOOL ret = SetEvent(hWinNotifyEvent);
+        dprintf(("SetEvent %x returned %d (last error %x)\n", hWinNotifyEvent, ret, GetLastError()));
         break;
     }
 }
 
+static void CALLBACK VBoxHandleDesktopEvent(HWINEVENTHOOK hook, DWORD event, HWND hwnd,
+                                 LONG idObject, LONG idChild,
+                                 DWORD dwEventThread, DWORD dwmsEventTime)
+{
+    if (!hDesktopNotifyEvent)
+    {
+        hDesktopNotifyEvent = OpenEvent(EVENT_MODIFY_STATE, FALSE, VBOXHOOK_GLOBAL_DT_EVENT_NAME);
+        dprintf(("OpenEvent returned %x (last err=%x)\n", hDesktopNotifyEvent, GetLastError()));
+    }
+    BOOL ret = SetEvent(hDesktopNotifyEvent);
+    dprintf(("SetEvent %x returned %d (last error %x)\n", hDesktopNotifyEvent, ret, GetLastError()));
+}
+
+BOOL VBoxHookInstallActiveDesktopTracker(HMODULE hDll)
+{
+    if (hDesktopEventHook)
+        return TRUE;
+
+    CoInitialize(NULL);
+    hDesktopEventHook = SetWinEventHook(EVENT_SYSTEM_DESKTOPSWITCH, EVENT_SYSTEM_DESKTOPSWITCH,
+                                    hDll,
+                                    VBoxHandleDesktopEvent,
+                                    0, 0,
+                                    0);
+
+    return !!hDesktopEventHook;
+
+}
+
+BOOL VBoxHookRemoveActiveDesktopTracker()
+{
+    if (hDesktopEventHook)
+    {
+        UnhookWinEvent(hDesktopEventHook);
+        CoUninitialize();
+    }
+    hDesktopEventHook = 0;
+    return TRUE;
+}
 
 /* Install the global message hook */
-BOOL VBoxInstallHook(HMODULE hDll)
-{
-    if (hEventHook[0] || hEventHook[1])
+BOOL VBoxHookInstallWindowTracker(HMODULE hDll)
+{
+    if (hWinEventHook[0] || hWinEventHook[1])
         return TRUE;
 
     CoInitialize(NULL);
-    hEventHook[0] = SetWinEventHook(EVENT_OBJECT_LOCATIONCHANGE, EVENT_OBJECT_LOCATIONCHANGE,
+    hWinEventHook[0] = SetWinEventHook(EVENT_OBJECT_LOCATIONCHANGE, EVENT_OBJECT_LOCATIONCHANGE,
                                     hDll,
                                     VBoxHandleWinEvent,
@@ -101 +142 @@
                                     WINEVENT_INCONTEXT | WINEVENT_SKIPOWNPROCESS);
 
-    hEventHook[1] = SetWinEventHook(EVENT_OBJECT_CREATE, EVENT_OBJECT_HIDE,
+    hWinEventHook[1] = SetWinEventHook(EVENT_OBJECT_CREATE, EVENT_OBJECT_HIDE,
                                     hDll,
                                     VBoxHandleWinEvent,
                                     0, 0,
                                     WINEVENT_INCONTEXT | WINEVENT_SKIPOWNPROCESS);
-    return !!hEventHook[0];
+    return !!hWinEventHook[0];
 }
 
 /* Remove the global message hook */
-BOOL VBoxRemoveHook()
-{
-    if (hEventHook[0] && hEventHook[1])
-    {
-        UnhookWinEvent(hEventHook[0]);
-        UnhookWinEvent(hEventHook[1]);
+BOOL VBoxHookRemoveWindowTracker()
+{
+    if (hWinEventHook[0] && hWinEventHook[1])
+    {
+        UnhookWinEvent(hWinEventHook[0]);
+        UnhookWinEvent(hWinEventHook[1]);
         CoUninitialize();
     }
-    hEventHook[0]  = hEventHook[1] = 0;
-    return true;
+    hWinEventHook[0]  = hWinEventHook[1] = 0;
+    return TRUE;
 }
 

trunk/src/VBox/Additions/WINNT/VBoxHook/VBoxHook.def

@@ -19 +19 @@
 
 EXPORTS
-   VBoxInstallHook
-   VBoxRemoveHook
+    VBoxHookInstallActiveDesktopTracker
+    VBoxHookRemoveActiveDesktopTracker
+    VBoxHookInstallWindowTracker
+    VBoxHookRemoveWindowTracker

trunk/src/VBox/Additions/WINNT/VBoxHook/testcase/tstHook.cpp

@@ -22 +22 @@
     printf("Enabling global hook\n");
 
-    HANDLE hEvent = CreateEvent(NULL, FALSE, FALSE, VBOXHOOK_GLOBAL_EVENT_NAME);
+    HANDLE hEvent = CreateEvent(NULL, FALSE, FALSE, VBOXHOOK_GLOBAL_WT_EVENT_NAME);
 
-    VBoxInstallHook(GetModuleHandle("VBoxHook.dll"));
+    VBoxHookInstallWindowTracker(GetModuleHandle("VBoxHook.dll"));
     getchar();
 
     printf("Disabling global hook\n");
-    VBoxRemoveHook();
+    VBoxHookRemoveWindowTracker();
     return 0;
 }