void VBoxSeamlessInstallHook()
{
if (gCtx.pfnVBoxHookInstallWindowTracker)
{
/* Check current visible region state */
VBoxSeamlessCheckWindows();
gCtx.pfnVBoxHookInstallWindowTracker(gCtx.hModule);
}
}
static void VBoxSeamlessInstallHook()
{
if (gCtx.pfnVBoxHookInstallWindowTracker)
{
/* Check current visible region state */
VBoxSeamlessCheckWindows(true);
HMODULE hMod = (HMODULE)RTLdrGetNativeHandle(gCtx.hModHook);
Assert(hMod != (HMODULE)~(uintptr_t)0);
gCtx.pfnVBoxHookInstallWindowTracker(hMod);
}
}
static void VBoxSeamlessInstallHook(void)
{
PVBOXSEAMLESSCONTEXT pCtx = &g_Ctx; /** @todo r=andy Use instance data via service lookup (add void *pInstance). */
AssertPtr(pCtx);
if (pCtx->pfnVBoxHookInstallWindowTracker)
{
/* Check current visible region state */
VBoxSeamlessCheckWindows(true);
HMODULE hMod = (HMODULE)RTLdrGetNativeHandle(pCtx->hModHook);
Assert(hMod != (HMODULE)~(uintptr_t)0);
pCtx->pfnVBoxHookInstallWindowTracker(hMod);
}
}
/**
* Window procedure for our tool window
*/
static LRESULT CALLBACK vboxToolWndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
case WM_CREATE:
{
Log(("VBoxTray: Tool window created\n"));
int rc = vboxTrayRegisterGlobalMessages(&s_vboxGlobalMessageTable[0]);
if (RT_FAILURE(rc))
Log(("VBoxTray: Error registering global window messages, rc=%Rrc\n", rc));
return 0;
}
case WM_CLOSE:
return 0;
case WM_DESTROY:
Log(("VBoxTray: Tool window destroyed\n"));
KillTimer(ghwndToolWindow, TIMERID_VBOXTRAY_CHECK_HOSTVERSION);
return 0;
case WM_TIMER:
switch (wParam)
{
case TIMERID_VBOXTRAY_CHECK_HOSTVERSION:
if (RT_SUCCESS(VBoxCheckHostVersion()))
{
/* After successful run we don't need to check again. */
KillTimer(ghwndToolWindow, TIMERID_VBOXTRAY_CHECK_HOSTVERSION);
}
return 0;
default:
break;
}
break; /* Make sure other timers get processed the usual way! */
case WM_VBOXTRAY_TRAY_ICON:
switch (lParam)
{
case WM_LBUTTONDBLCLK:
break;
case WM_RBUTTONDOWN:
break;
}
return 0;
case WM_VBOX_INSTALL_SEAMLESS_HOOK:
VBoxSeamlessInstallHook();
return 0;
case WM_VBOX_REMOVE_SEAMLESS_HOOK:
VBoxSeamlessRemoveHook();
return 0;
case WM_VBOX_SEAMLESS_UPDATE:
VBoxSeamlessCheckWindows();
return 0;
case WM_VBOXTRAY_VM_RESTORED:
VBoxRestoreSession();
return 0;
case WM_VBOXTRAY_VRDP_CHECK:
VBoxRestoreCheckVRDP();
return 0;
default:
/* Handle all globally registered window messages. */
if (vboxTrayHandleGlobalMessages(&s_vboxGlobalMessageTable[0], uMsg,
wParam, lParam))
{
return 0; /* We handled the message. @todo Add return value!*/
}
break; /* We did not handle the message, dispatch to DefWndProc. */
}
/* Only if message was *not* handled by our switch above, dispatch
* to DefWindowProc. */
return DefWindowProc(hWnd, uMsg, wParam, lParam);
}
static int vboxTrayServiceMain(void)
{
int rc = VINF_SUCCESS;
Log(("VBoxTray: Entering vboxTrayServiceMain\n"));
ghStopSem = CreateEvent(NULL, TRUE, FALSE, NULL);
if (ghStopSem == NULL)
{
rc = RTErrConvertFromWin32(GetLastError());
Log(("VBoxTray: CreateEvent for stopping VBoxTray failed, rc=%Rrc\n", rc));
}
else
{
/*
* Start services listed in the vboxServiceTable.
*/
VBOXSERVICEENV svcEnv;
svcEnv.hInstance = ghInstance;
svcEnv.hDriver = ghVBoxDriver;
/* Initializes disp-if to default (XPDM) mode. */
VBoxDispIfInit(&svcEnv.dispIf); /* Cannot fail atm. */
#ifdef VBOX_WITH_WDDM
/*
* For now the display mode will be adjusted to WDDM mode if needed
* on display service initialization when it detects the display driver type.
*/
#endif
/* Finally start all the built-in services! */
rc = vboxTrayStartServices(&svcEnv, vboxServiceTable);
if (RT_FAILURE(rc))
{
/* Terminate service if something went wrong. */
vboxTrayStopServices(&svcEnv, vboxServiceTable);
}
else
{
rc = vboxTrayCreateTrayIcon();
if ( RT_SUCCESS(rc)
&& gMajorVersion >= 5) /* Only for W2K and up ... */
{
/* We're ready to create the tooltip balloon.
Check in 10 seconds (@todo make seconds configurable) ... */
SetTimer(ghwndToolWindow,
TIMERID_VBOXTRAY_CHECK_HOSTVERSION,
10 * 1000, /* 10 seconds */
NULL /* No timerproc */);
}
if (RT_SUCCESS(rc))
{
/* Do the Shared Folders auto-mounting stuff. */
rc = VBoxSharedFoldersAutoMount();
if (RT_SUCCESS(rc))
{
/* Report the host that we're up and running! */
hlpReportStatus(VBoxGuestFacilityStatus_Active);
}
}
if (RT_SUCCESS(rc))
{
/* Boost thread priority to make sure we wake up early for seamless window notifications
* (not sure if it actually makes any difference though). */
SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_HIGHEST);
/*
* Main execution loop
* Wait for the stop semaphore to be posted or a window event to arrive
*/
DWORD dwEventCount = 2;
HANDLE hWaitEvent[2] = { ghStopSem, ghSeamlessNotifyEvent };
if (0 == ghSeamlessNotifyEvent) /* If seamless mode is not active / supported, reduce event array count. */
dwEventCount = 1;
Log(("VBoxTray: Number of events to wait in main loop: %ld\n", dwEventCount));
while (true)
{
DWORD waitResult = MsgWaitForMultipleObjectsEx(dwEventCount, hWaitEvent, 500, QS_ALLINPUT, 0);
waitResult = waitResult - WAIT_OBJECT_0;
/* Only enable for message debugging, lots of traffic! */
//Log(("VBoxTray: Wait result = %ld\n", waitResult));
if (waitResult == 0)
{
Log(("VBoxTray: Event 'Exit' triggered\n"));
/* exit */
break;
}
else if ( waitResult == 1
&& ghSeamlessNotifyEvent != 0) /* Only jump in, if seamless is active! */
{
Log(("VBoxTray: Event 'Seamless' triggered\n"));
/* seamless window notification */
VBoxSeamlessCheckWindows();
}
else
{
/* timeout or a window message, handle it */
MSG msg;
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
{
Log(("VBoxTray: msg %p\n", msg.message));
if (msg.message == WM_QUIT)
{
Log(("VBoxTray: WM_QUIT!\n"));
SetEvent(ghStopSem);
continue;
}
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}
}
Log(("VBoxTray: Returned from main loop, exiting ...\n"));
}
Log(("VBoxTray: Waiting for services to stop ...\n"));
vboxTrayStopServices(&svcEnv, vboxServiceTable);
} /* Services started */
CloseHandle(ghStopSem);
} /* Stop event created */
vboxTrayRemoveTrayIcon();
Log(("VBoxTray: Leaving vboxTrayServiceMain with rc=%Rrc\n", rc));
return rc;
}
