BOOL WINAPI DllMain(HINSTANCE hDLLInst, DWORD fdwReason, LPVOID lpvReserved)
{
BOOL bStatus = TRUE;
switch (fdwReason) {
case DLL_PROCESS_ATTACH:
RTR3InitDll(0);
VbglR3Init();
LogRel(("VBOXNP: DLL loaded.\n"));
break;
case DLL_PROCESS_DETACH:
LogRel(("VBOXNP: DLL unloaded.\n"));
VbglR3Term();
/// @todo RTR3Term();
break;
case DLL_THREAD_ATTACH:
break;
case DLL_THREAD_DETACH:
break;
default:
break;
}
return(bStatus);
}
VBoxGuestDeskbarView::~VBoxGuestDeskbarView()
{
delete fIcon;
if (fClipboardService)
{
fClipboardService->Disconnect();
delete fClipboardService;
}
VbglR3Term();
}
/** Clean up if we get a signal or something. This is extern so that we
* can call it from other compilation units. */
void VBoxClient::CleanUp()
{
/* We never release this, as we end up with a call to exit(3) which is not
* async-safe. Until we fix this application properly, we should be sure
* never to exit from anywhere except from this method. */
int rc = RTCritSectEnter(&g_critSect);
if (RT_FAILURE(rc))
{
RTPrintf("VBoxClient: Failure while acquiring the global critical section, rc=%Rrc\n", rc);
abort();
}
if (g_pService)
g_pService->cleanup();
if (g_szPidFile[0] && g_hPidFile)
VbglR3ClosePidFile(g_szPidFile, g_hPidFile);
VbglR3Term();
exit(0);
}
BOOL WINAPI DllMain(HINSTANCE hDLLInst, DWORD fdwReason, LPVOID lpvReserved)
{
(void) lpvReserved; (void) hDLLInst;
switch (fdwReason)
{
case DLL_PROCESS_ATTACH:
{
int rc;
rc = RTR3InitDll(RTR3INIT_FLAGS_UNOBTRUSIVE); CRASSERT(rc==0);
# ifdef IN_GUEST
rc = VbglR3Init();
# endif
LogRel(("crUtil DLL loaded.\n"));
# if defined(DEBUG_misha)
char aName[MAX_PATH];
GetModuleFileNameA(hDLLInst, aName, RT_ELEMENTS(aName));
crDbgCmdSymLoadPrint(aName, hDLLInst);
# endif
break;
}
case DLL_PROCESS_DETACH:
{
LogRel(("crUtil DLL unloaded."));
# ifdef IN_GUEST
VbglR3Term();
# endif
}
default:
break;
}
return TRUE;
}
/**
* Main function
*/
int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
/* Do not use a global namespace ("Global\\") for mutex name here, will blow up NT4 compatibility! */
HANDLE hMutexAppRunning = CreateMutex(NULL, FALSE, "VBoxTray");
if ( hMutexAppRunning != NULL
&& GetLastError() == ERROR_ALREADY_EXISTS)
{
/* Close the mutex for this application instance. */
CloseHandle (hMutexAppRunning);
hMutexAppRunning = NULL;
return 0;
}
LogRel(("VBoxTray: %s r%s\n", RTBldCfgVersion(), RTBldCfgRevisionStr()));
int rc = RTR3Init();
if (RT_SUCCESS(rc))
{
rc = VbglR3Init();
if (RT_SUCCESS(rc))
rc = vboxTrayOpenBaseDriver();
}
if (RT_SUCCESS(rc))
{
/* Save instance handle. */
ghInstance = hInstance;
hlpReportStatus(VBoxGuestFacilityStatus_Init);
rc = vboxTrayCreateToolWindow();
if (RT_SUCCESS(rc))
{
rc = vboxTraySetupSeamless();
if (RT_SUCCESS(rc))
{
Log(("VBoxTray: Init successful\n"));
rc = vboxTrayServiceMain();
if (RT_SUCCESS(rc))
hlpReportStatus(VBoxGuestFacilityStatus_Terminating);
vboxTrayShutdownSeamless();
}
vboxTrayDestroyToolWindow();
}
if (RT_SUCCESS(rc))
hlpReportStatus(VBoxGuestFacilityStatus_Terminated);
}
if (RT_FAILURE(rc))
{
LogRel(("VBoxTray: Error while starting, rc=%Rrc\n", rc));
hlpReportStatus(VBoxGuestFacilityStatus_Failed);
}
LogRel(("VBoxTray: Ended\n"));
vboxTrayCloseBaseDriver();
/* Release instance mutex. */
if (hMutexAppRunning != NULL)
{
CloseHandle(hMutexAppRunning);
hMutexAppRunning = NULL;
}
VbglR3Term();
return RT_SUCCESS(rc) ? 0 : 1;
}
status_t VBoxMouse::SystemShuttingDown()
{
VbglR3Term();
return B_OK;
}
/** @todo Move this part in VbglR3 and just provide a callback for the platform-specific
notification stuff, since this is very similar to the VBoxTray code. */
static int run(struct VBCLSERVICE **ppInterface, bool fDaemonised)
{
int rc;
LogFlowFunc(("\n"));
NOREF(ppInterface);
/* Initialise the guest library. */
rc = VbglR3InitUser();
if (RT_FAILURE(rc))
VBClFatalError(("Failed to connect to the VirtualBox kernel service, rc=%Rrc\n", rc));
/* Because we need desktop notifications to be displayed, wait
* some time to make the desktop environment load (as a work around). */
if (fDaemonised)
RTThreadSleep(30 * 1000 /* Wait 30 seconds */);
# ifdef VBOX_WITH_DBUS
rc = RTDBusLoadLib();
if (RT_FAILURE(rc))
LogRel(("VBoxClient: D-Bus seems not to be installed; no host version check/notification done.\n"));
# else
rc = VERR_NOT_IMPLEMENTED;
# endif /* VBOX_WITH_DBUS */
# ifdef VBOX_WITH_GUEST_PROPS
uint32_t uGuestPropSvcClientID;
if (RT_SUCCESS(rc))
{
rc = VbglR3GuestPropConnect(&uGuestPropSvcClientID);
if (RT_FAILURE(rc))
LogRel(("VBoxClient: Cannot connect to guest property service while chcking for host version! rc = %Rrc\n", rc));
}
if (RT_SUCCESS(rc))
{
char *pszHostVersion;
char *pszGuestVersion;
bool bUpdate;
rc = VbglR3HostVersionCheckForUpdate(uGuestPropSvcClientID, &bUpdate, &pszHostVersion, &pszGuestVersion);
if (RT_SUCCESS(rc))
{
if (bUpdate)
{
char szMsg[1024];
char szTitle[64];
/** @todo add some translation macros here */
RTStrPrintf(szTitle, sizeof(szTitle), "VirtualBox Guest Additions update available!");
#ifndef VBOX_OSE
RTStrPrintf(szMsg, sizeof(szMsg), "Your guest is currently running the Guest Additions version %s. "
"We recommend updating to the latest version (%s) by choosing the "
"install option from the Devices menu.", pszGuestVersion, pszHostVersion);
#else
/* This is the message which appears for non-Oracle builds of the
* Guest Additions. Distributors are encouraged to customise this. */
RTStrPrintf(szMsg, sizeof(szMsg), "Your virtual machine is currently running the Guest Additions version %s. Since you are running a version of the Guest Additions provided by the operating system you installed in the virtual machine we recommend that you update it to at least version %s using that system's update features, or alternatively that you remove this version and then install the " VBOX_VENDOR_SHORT " Guest Additions package using the install option from the Devices menu. Please consult the documentation for the operating system you are running to find out how to update or remove the current Guest Additions package.", pszGuestVersion, pszHostVersion);
#endif
rc = showNotify(szTitle, szMsg);
LogRel(("VBoxClient: VirtualBox Guest Additions update available!"));
if (RT_FAILURE(rc))
LogRel(("VBoxClient: Could not show version notifier tooltip! rc = %d\n", rc));
}
/* Store host version to not notify again */
rc = VbglR3HostVersionLastCheckedStore(uGuestPropSvcClientID, pszHostVersion);
VbglR3GuestPropReadValueFree(pszHostVersion);
VbglR3GuestPropReadValueFree(pszGuestVersion);
}
VbglR3GuestPropDisconnect(uGuestPropSvcClientID);
}
# endif /* VBOX_WITH_GUEST_PROPS */
VbglR3Term();
LogFlowFunc(("returning %Rrc\n", rc));
return rc;
}
/**
* The main loop for the VBoxClient daemon.
*/
int main(int argc, char *argv[])
{
/* Initialise our runtime before all else. */
int rc = RTR3InitExe(argc, &argv, 0);
if (RT_FAILURE(rc))
return RTMsgInitFailure(rc);
int rcClipboard;
const char *pszFileName = RTPathFilename(argv[0]);
bool fDaemonise = true;
/* Have any fatal errors occurred yet? */
bool fSuccess = true;
/* Do we know which service we wish to run? */
bool fHaveService = false;
if (NULL == pszFileName)
pszFileName = "VBoxClient";
/* Initialise our global clean-up critical section */
rc = RTCritSectInit(&g_critSect);
if (RT_FAILURE(rc))
{
/* Of course, this should never happen. */
RTPrintf("%s: Failed to initialise the global critical section, rc=%Rrc\n", pszFileName, rc);
return 1;
}
/* Parse our option(s) */
/** @todo Use RTGetOpt() if the arguments become more complex. */
for (int i = 1; i < argc; ++i)
{
if (!strcmp(argv[i], "-d") || !strcmp(argv[i], "--nodaemon"))
fDaemonise = false;
else if (!strcmp(argv[i], "--clipboard"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetClipboardService();
else
fSuccess = false;
}
else if (!strcmp(argv[i], "--display"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetDisplayService();
else
fSuccess = false;
}
else if (!strcmp(argv[i], "--seamless"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetSeamlessService();
else
fSuccess = false;
}
else if (!strcmp(argv[i], "--checkhostversion"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetHostVersionService();
else
fSuccess = false;
}
#ifdef VBOX_WITH_DRAG_AND_DROP
else if (!strcmp(argv[i], "--draganddrop"))
{
if (g_pService == NULL)
g_pService = VBoxClient::GetDragAndDropService();
else
fSuccess = false;
}
#endif /* VBOX_WITH_DRAG_AND_DROP */
else if (!strcmp(argv[i], "-h") || !strcmp(argv[i], "--help"))
{
vboxClientUsage(pszFileName);
return 0;
}
else
{
RTPrintf("%s: unrecognized option `%s'\n", pszFileName, argv[i]);
RTPrintf("Try `%s --help' for more information\n", pszFileName);
return 1;
}
}
if (!fSuccess || !g_pService)
{
vboxClientUsage(pszFileName);
return 1;
}
/* Get the path for the pidfiles */
rc = RTPathUserHome(g_szPidFile, sizeof(g_szPidFile));
if (RT_FAILURE(rc))
{
RTPrintf("VBoxClient: failed to get home directory, rc=%Rrc. Exiting.\n", rc);
LogRel(("VBoxClient: failed to get home directory, rc=%Rrc. Exiting.\n", rc));
return 1;
}
rc = RTPathAppend(g_szPidFile, sizeof(g_szPidFile), g_pService->getPidFilePath());
if (RT_FAILURE(rc))
{
RTPrintf("VBoxClient: RTPathAppend failed with rc=%Rrc. Exiting.\n", rc);
LogRel(("VBoxClient: RTPathAppend failed with rc=%Rrc. Exiting.\n", rc));
return 1;
}
/* Initialise the guest library. */
if (RT_FAILURE(VbglR3InitUser()))
{
RTPrintf("Failed to connect to the VirtualBox kernel service\n");
LogRel(("Failed to connect to the VirtualBox kernel service\n"));
return 1;
}
if (fDaemonise)
{
rc = VbglR3Daemonize(false /* fNoChDir */, false /* fNoClose */);
if (RT_FAILURE(rc))
{
RTPrintf("VBoxClient: failed to daemonize. Exiting.\n");
LogRel(("VBoxClient: failed to daemonize. Exiting.\n"));
# ifdef DEBUG
RTPrintf("Error %Rrc\n", rc);
# endif
return 1;
}
}
if (g_szPidFile[0] && RT_FAILURE(VbglR3PidFile(g_szPidFile, &g_hPidFile)))
{
RTPrintf("Failed to create a pidfile. Exiting.\n");
LogRel(("Failed to create a pidfile. Exiting.\n"));
VbglR3Term();
return 1;
}
/* Set signal handlers to clean up on exit. */
vboxClientSetSignalHandlers();
/* Set an X11 error handler, so that we don't die when we get unavoidable errors. */
XSetErrorHandler(vboxClientXLibErrorHandler);
/* Set an X11 I/O error handler, so that we can shutdown properly on fatal errors. */
XSetIOErrorHandler(vboxClientXLibIOErrorHandler);
g_pService->run(fDaemonise);
VBoxClient::CleanUp();
return 1; /* We should never get here. */
}
static void cleanup(struct VBCLSERVICE **ppInterface)
{
NOREF(ppInterface);
VbglR3Term();
}
static void cleanup(struct VBCLSERVICE **ppInterface)
{
NOREF(ppInterface);
VbglR3SeamlessSetCap(false);
VbglR3Term();
}
/**
* Shuts down pam_vbox.
*
* @return IPRT status code.
* @param hPAM PAM handle.
*/
static void pam_vbox_shutdown(pam_handle_t *hPAM)
{
RT_NOREF1(hPAM);
VbglR3Term();
}
int main(int argc, char *argv[])
{
int rc;
int c;
bool fDemonize = true;
static char *szLogFileName = NULL;
/* Parse command line */
while((c = getopt(argc, argv, "fvl:")) != -1)
{
switch(c)
{
case 'f':
fDemonize = false;
break;
case 'v':
g_cVerbosity++;
break;
case 'l':
szLogFileName = RTStrDup(optarg);
break;
default : usage(argv[0]);
}
}
/* No more arguments allowed */
if ((argc - optind) != 0)
usage(argv[0]);
if (fDemonize)
{
rc = RTProcDaemonizeUsingFork(true /* fNoChDir */, false /* fNoClose */, NULL);
if (RT_FAILURE(rc))
{
RTPrintf("failed to run into background\n");
return 1;
}
}
rc = RTR3InitExe(argc, &argv, 0);
if (RT_FAILURE(rc))
{
RTPrintf("RTR3InitExe() failed: (%Rrc)\n", rc);
return RTMsgInitFailure(rc);
}
rc = VbglR3Init();
if (RT_SUCCESS(rc))
{
rc = vbclInitLogger(szLogFileName);
if (RT_SUCCESS(rc))
{
rc = vbclStartServices();
if (RT_SUCCESS(rc))
{
vbclWait();
vbclStopServices();
}
else
{
RTPrintf("failed to start services: (%Rrc)\n", rc);
}
vbclTermLogger(szLogFileName);
}
else
{
RTPrintf("failed to start logger: (%Rrc)\n", rc);
}
VbglR3Term();
}
else
{
RTPrintf("failed to initialize guest library: (%Rrc)\n", rc);
}
return 0;
}
