VBoxService.cpp@ 103251

Last change on this file since 103251 was 98103, checked in by vboxsync, 2 years ago
Copyright year updates by scm.
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 42.6 KB

Line
1	/* $Id: VBoxService.cpp 98103 2023-01-17 14:15:46Z vboxsync $ */
2	/** @file
3	* VBoxService - Guest Additions Service Skeleton.
4	*/
5
6	/*
7	* Copyright (C) 2007-2023 Oracle and/or its affiliates.
8	*
9	* This file is part of VirtualBox base platform packages, as
10	* available from https://www.virtualbox.org.
11	*
12	* This program is free software; you can redistribute it and/or
13	* modify it under the terms of the GNU General Public License
14	* as published by the Free Software Foundation, in version 3 of the
15	* License.
16	*
17	* This program is distributed in the hope that it will be useful, but
18	* WITHOUT ANY WARRANTY; without even the implied warranty of
19	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20	* General Public License for more details.
21	*
22	* You should have received a copy of the GNU General Public License
23	* along with this program; if not, see <https://www.gnu.org/licenses>.
24	*
25	* SPDX-License-Identifier: GPL-3.0-only
26	*/
27
28
29	/** @page pg_vgsvc VBoxService
30	*
31	* VBoxService is a root daemon for implementing guest additions features.
32	*
33	* It is structured as one binary that contains many sub-services. The reason
34	* for this is partially historical and partially practical. The practical
35	* reason is that the VBoxService binary is typically statically linked, at
36	* least with IPRT and the guest library, so we save quite a lot of space having
37	* on single binary instead individual binaries for each sub-service and their
38	* helpers (currently up to 9 subservices and 8 helpers). The historical is
39	* simply that it started its life on OS/2 dreaming of conquring Windows next,
40	* so it kind of felt natural to have it all in one binary.
41	*
42	* Even if it's structured as a single binary, it is possible, by using command
43	* line options, to start each subservice as an individual process.
44	*
45	* Subservices:
46	* - @subpage pg_vgsvc_timesync "Time Synchronization"
47	* - @subpage pg_vgsvc_vminfo "VM Information"
48	* - @subpage pg_vgsvc_vmstats "VM Statistics"
49	* - @subpage pg_vgsvc_gstctrl "Guest Control"
50	* - @subpage pg_vgsvc_pagesharing "Page Sharing"
51	* - @subpage pg_vgsvc_memballoon "Memory Balooning"
52	* - @subpage pg_vgsvc_cpuhotplug "CPU Hot-Plugging"
53	* - @subpage pg_vgsvc_automount "Shared Folder Automounting"
54	* - @subpage pg_vgsvc_clipboard "Clipboard (OS/2 only)"
55	*
56	* Now, since the service predates a lot of stuff, including RTGetOpt, we're
57	* currently doing our own version of argument parsing here, which is kind of
58	* stupid. That will hopefully be cleaned up eventually.
59	*/
60
61
62	/*********************************************************************************************************************************
63	* Header Files *
64	*********************************************************************************************************************************/
65	/** @todo LOG_GROUP*/
66	#ifndef _MSC_VER
67	# include <unistd.h>
68	#endif
69	#ifndef RT_OS_WINDOWS
70	# include <errno.h>
71	# include <signal.h>
72	# ifdef RT_OS_OS2
73	# define pthread_sigmask sigprocmask
74	# endif
75	#endif
76	#ifdef RT_OS_FREEBSD
77	# include <pthread.h>
78	#endif
79
80	#include <package-generated.h>
81	#include "product-generated.h"
82
83	#include <iprt/asm.h>
84	#include <iprt/buildconfig.h>
85	#include <iprt/initterm.h>
86	#include <iprt/file.h>
87	#ifdef DEBUG
88	# include <iprt/memtracker.h>
89	#endif
90	#include <iprt/env.h>
91	#include <iprt/message.h>
92	#include <iprt/path.h>
93	#include <iprt/process.h>
94	#include <iprt/semaphore.h>
95	#include <iprt/string.h>
96	#include <iprt/stream.h>
97	#include <iprt/system.h>
98	#include <iprt/thread.h>
99
100	#include <VBox/err.h>
101	#include <VBox/log.h>
102
103	#include "VBoxServiceInternal.h"
104	#include "VBoxServiceUtils.h"
105	#ifdef VBOX_WITH_VBOXSERVICE_CONTROL
106	# include "VBoxServiceControl.h"
107	#endif
108	#ifdef VBOX_WITH_VBOXSERVICE_TOOLBOX
109	# include "VBoxServiceToolBox.h"
110	#endif
111
112
113	/*********************************************************************************************************************************
114	* Global Variables *
115	*********************************************************************************************************************************/
116	/** The program name (derived from argv[0]). */
117	char g_pszProgName = (char )"";
118	/** The current verbosity level. */
119	unsigned g_cVerbosity = 0;
120	char g_szLogFile[RTPATH_MAX + 128] = "";
121	char g_szPidFile[RTPATH_MAX] = "";
122	/** Logging parameters. */
123	/** @todo Make this configurable later. */
124	static PRTLOGGER g_pLoggerRelease = NULL;
125	static uint32_t g_cHistory = 10; /* Enable log rotation, 10 files. */
126	static uint32_t g_uHistoryFileTime = RT_SEC_1DAY; /* Max 1 day per file. */
127	static uint64_t g_uHistoryFileSize = 100 * _1M; /* Max 100MB per file. */
128	/** Critical section for (debug) logging. */
129	#ifdef DEBUG
130	RTCRITSECT g_csLog;
131	#endif
132	/** The default service interval (the -i \| --interval) option). */
133	uint32_t g_DefaultInterval = 0;
134	#ifdef RT_OS_WINDOWS
135	/** Signal shutdown to the Windows service thread. */
136	static bool volatile g_fWindowsServiceShutdown;
137	/** Event the Windows service thread waits for shutdown. */
138	static RTSEMEVENT g_hEvtWindowsService;
139	#endif
140
141	/**
142	* The details of the services that has been compiled in.
143	*/
144	static struct
145	{
146	/** Pointer to the service descriptor. */
147	PCVBOXSERVICE pDesc;
148	/** The worker thread. NIL_RTTHREAD if it's the main thread. */
149	RTTHREAD Thread;
150	/** Whether Pre-init was called. */
151	bool fPreInited;
152	/** Shutdown indicator. */
153	bool volatile fShutdown;
154	/** Indicator set by the service thread exiting. */
155	bool volatile fStopped;
156	/** Whether the service was started or not. */
157	bool fStarted;
158	/** Whether the service is enabled or not. */
159	bool fEnabled;
160	} g_aServices[] =
161	{
162	#ifdef VBOX_WITH_VBOXSERVICE_CONTROL
163	{ &g_Control, NIL_RTTHREAD, false, false, false, false, true },
164	#endif
165	#ifdef VBOX_WITH_VBOXSERVICE_TIMESYNC
166	{ &g_TimeSync, NIL_RTTHREAD, false, false, false, false, true },
167	#endif
168	#ifdef VBOX_WITH_VBOXSERVICE_CLIPBOARD
169	{ &g_Clipboard, NIL_RTTHREAD, false, false, false, false, true },
170	#endif
171	#ifdef VBOX_WITH_VBOXSERVICE_VMINFO
172	{ &g_VMInfo, NIL_RTTHREAD, false, false, false, false, true },
173	#endif
174	#ifdef VBOX_WITH_VBOXSERVICE_CPUHOTPLUG
175	{ &g_CpuHotPlug, NIL_RTTHREAD, false, false, false, false, true },
176	#endif
177	#ifdef VBOX_WITH_VBOXSERVICE_MANAGEMENT
178	# ifdef VBOX_WITH_MEMBALLOON
179	{ &g_MemBalloon, NIL_RTTHREAD, false, false, false, false, true },
180	# endif
181	{ &g_VMStatistics, NIL_RTTHREAD, false, false, false, false, true },
182	#endif
183	#if defined(VBOX_WITH_VBOXSERVICE_PAGE_SHARING)
184	{ &g_PageSharing, NIL_RTTHREAD, false, false, false, false, true },
185	#endif
186	#ifdef VBOX_WITH_SHARED_FOLDERS
187	{ &g_AutoMount, NIL_RTTHREAD, false, false, false, false, true },
188	#endif
189	};
190
191
192	/*
193	* Default call-backs for services which do not need special behaviour.
194	*/
195
196	/**
197	* @interface_method_impl{VBOXSERVICE,pfnPreInit, Default Implementation}
198	*/
199	DECLCALLBACK(int) VGSvcDefaultPreInit(void)
200	{
201	return VINF_SUCCESS;
202	}
203
204
205	/**
206	* @interface_method_impl{VBOXSERVICE,pfnOption, Default Implementation}
207	*/
208	DECLCALLBACK(int) VGSvcDefaultOption(const char **ppszShort, int argc,
209	char *argv, int pi)
210	{
211	NOREF(ppszShort);
212	NOREF(argc);
213	NOREF(argv);
214	NOREF(pi);
215
216	return -1;
217	}
218
219
220	/**
221	* @interface_method_impl{VBOXSERVICE,pfnInit, Default Implementation}
222	*/
223	DECLCALLBACK(int) VGSvcDefaultInit(void)
224	{
225	return VINF_SUCCESS;
226	}
227
228
229	/**
230	* @interface_method_impl{VBOXSERVICE,pfnTerm, Default Implementation}
231	*/
232	DECLCALLBACK(void) VGSvcDefaultTerm(void)
233	{
234	return;
235	}
236
237
238	/**
239	* @callback_method_impl{FNRTLOGPHASE, Release logger callback}
240	*/
241	static DECLCALLBACK(void) vgsvcLogHeaderFooter(PRTLOGGER pLoggerRelease, RTLOGPHASE enmPhase, PFNRTLOGPHASEMSG pfnLog)
242	{
243	/* Some introductory information. */
244	static RTTIMESPEC s_TimeSpec;
245	char szTmp[256];
246	if (enmPhase == RTLOGPHASE_BEGIN)
247	RTTimeNow(&s_TimeSpec);
248	RTTimeSpecToString(&s_TimeSpec, szTmp, sizeof(szTmp));
249
250	switch (enmPhase)
251	{
252	case RTLOGPHASE_BEGIN:
253	{
254	pfnLog(pLoggerRelease,
255	"VBoxService %s r%s (verbosity: %u) %s (%s %s) release log\n"
256	"Log opened %s\n",
257	RTBldCfgVersion(), RTBldCfgRevisionStr(), g_cVerbosity, VBOX_BUILD_TARGET,
258	__DATE__, __TIME__, szTmp);
259
260	int vrc = RTSystemQueryOSInfo(RTSYSOSINFO_PRODUCT, szTmp, sizeof(szTmp));
261	if (RT_SUCCESS(vrc) \|\| vrc == VERR_BUFFER_OVERFLOW)
262	pfnLog(pLoggerRelease, "OS Product: %s\n", szTmp);
263	vrc = RTSystemQueryOSInfo(RTSYSOSINFO_RELEASE, szTmp, sizeof(szTmp));
264	if (RT_SUCCESS(vrc) \|\| vrc == VERR_BUFFER_OVERFLOW)
265	pfnLog(pLoggerRelease, "OS Release: %s\n", szTmp);
266	vrc = RTSystemQueryOSInfo(RTSYSOSINFO_VERSION, szTmp, sizeof(szTmp));
267	if (RT_SUCCESS(vrc) \|\| vrc == VERR_BUFFER_OVERFLOW)
268	pfnLog(pLoggerRelease, "OS Version: %s\n", szTmp);
269	vrc = RTSystemQueryOSInfo(RTSYSOSINFO_SERVICE_PACK, szTmp, sizeof(szTmp));
270	if (RT_SUCCESS(vrc) \|\| vrc == VERR_BUFFER_OVERFLOW)
271	pfnLog(pLoggerRelease, "OS Service Pack: %s\n", szTmp);
272
273	/* the package type is interesting for Linux distributions */
274	char szExecName[RTPATH_MAX];
275	char *pszExecName = RTProcGetExecutablePath(szExecName, sizeof(szExecName));
276	pfnLog(pLoggerRelease,
277	"Executable: %s\n"
278	"Process ID: %u\n"
279	"Package type: %s"
280	#ifdef VBOX_OSE
281	" (OSE)"
282	#endif
283	"\n",
284	pszExecName ? pszExecName : "unknown",
285	RTProcSelf(),
286	VBOX_PACKAGE_STRING);
287	break;
288	}
289
290	case RTLOGPHASE_PREROTATE:
291	pfnLog(pLoggerRelease, "Log rotated - Log started %s\n", szTmp);
292	break;
293
294	case RTLOGPHASE_POSTROTATE:
295	pfnLog(pLoggerRelease, "Log continuation - Log started %s\n", szTmp);
296	break;
297
298	case RTLOGPHASE_END:
299	pfnLog(pLoggerRelease, "End of log file - Log started %s\n", szTmp);
300	break;
301
302	default:
303	/* nothing */
304	break;
305	}
306	}
307
308
309	/**
310	* Creates the default release logger outputting to the specified file.
311	*
312	* Pass NULL to disabled logging.
313	*
314	* @return IPRT status code.
315	* @param pszLogFile Filename for log output. NULL disables logging
316	* (r=bird: No, it doesn't!).
317	*/
318	int VGSvcLogCreate(const char *pszLogFile)
319	{
320	/* Create release logger (stdout + file). */
321	static const char * const s_apszGroups[] = VBOX_LOGGROUP_NAMES;
322	RTUINT fFlags = RTLOGFLAGS_PREFIX_THREAD \| RTLOGFLAGS_PREFIX_TIME;
323	#if defined(RT_OS_WINDOWS) \|\| defined(RT_OS_OS2)
324	fFlags \|= RTLOGFLAGS_USECRLF;
325	#endif
326	int rc = RTLogCreateEx(&g_pLoggerRelease, "VBOXSERVICE_RELEASE_LOG", fFlags, "all",
327	RT_ELEMENTS(s_apszGroups), s_apszGroups, UINT32_MAX /cMaxEntriesPerGroup/,
328	0 /cBufDescs/, NULL /paBufDescs/, RTLOGDEST_STDOUT \| RTLOGDEST_USER,
329	vgsvcLogHeaderFooter, g_cHistory, g_uHistoryFileSize, g_uHistoryFileTime,
330	NULL /pOutputIf/, NULL /pvOutputIfUser/,
331	NULL /pErrInfo/, "%s", pszLogFile ? pszLogFile : "");
332	if (RT_SUCCESS(rc))
333	{
334	/* register this logger as the release logger */
335	RTLogRelSetDefaultInstance(g_pLoggerRelease);
336
337	/* Explicitly flush the log in case of VBOXSERVICE_RELEASE_LOG=buffered. */
338	RTLogFlush(g_pLoggerRelease);
339	}
340
341	return rc;
342	}
343
344
345	/**
346	* Logs a verbose message.
347	*
348	* @param pszFormat The message text.
349	* @param va Format arguments.
350	*/
351	void VGSvcLogV(const char *pszFormat, va_list va)
352	{
353	#ifdef DEBUG
354	int rc = RTCritSectEnter(&g_csLog);
355	if (RT_SUCCESS(rc))
356	{
357	#endif
358	char *psz = NULL;
359	RTStrAPrintfV(&psz, pszFormat, va);
360
361	AssertPtr(psz);
362	LogRel(("%s", psz));
363
364	RTStrFree(psz);
365	#ifdef DEBUG
366	RTCritSectLeave(&g_csLog);
367	}
368	#endif
369	}
370
371
372	/**
373	* Destroys the currently active logging instance.
374	*/
375	void VGSvcLogDestroy(void)
376	{
377	RTLogDestroy(RTLogRelSetDefaultInstance(NULL));
378	}
379
380
381	/**
382	* Displays the program usage message.
383	*
384	* @returns 1.
385	*/
386	static int vgsvcUsage(void)
387	{
388	RTPrintf("Usage: %s [-f\|--foreground] [-v\|--verbose] [-l\|--logfile <file>]\n"
389	" [-p\|--pidfile <file>] [-i\|--interval <seconds>]\n"
390	" [--disable-<service>] [--enable-<service>]\n"
391	" [--only-<service>] [-h\|-?\|--help]\n", g_pszProgName);
392	#ifdef RT_OS_WINDOWS
393	RTPrintf(" [-r\|--register] [-u\|--unregister]\n");
394	#endif
395	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
396	if (g_aServices[j].pDesc->pszUsage)
397	RTPrintf("%s\n", g_aServices[j].pDesc->pszUsage);
398	RTPrintf("\n"
399	"Options:\n"
400	" -i \| --interval The default interval.\n"
401	" -f \| --foreground Don't daemonize the program. For debugging.\n"
402	" -l \| --logfile <file> Enables logging to a file.\n"
403	" -p \| --pidfile <file> Write the process ID to a file.\n"
404	" -v \| --verbose Increment the verbosity level. For debugging.\n"
405	" -V \| --version Show version information.\n"
406	" -h \| -? \| --help Show this message and exit with status 1.\n"
407	);
408	#ifdef RT_OS_WINDOWS
409	RTPrintf(" -r \| --register Installs the service.\n"
410	" -u \| --unregister Uninstall service.\n");
411	#endif
412
413	RTPrintf("\n"
414	"Service-specific options:\n");
415	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
416	{
417	RTPrintf(" --enable-%-14s Enables the %s service. (default)\n", g_aServices[j].pDesc->pszName, g_aServices[j].pDesc->pszName);
418	RTPrintf(" --disable-%-13s Disables the %s service.\n", g_aServices[j].pDesc->pszName, g_aServices[j].pDesc->pszName);
419	RTPrintf(" --only-%-16s Only enables the %s service.\n", g_aServices[j].pDesc->pszName, g_aServices[j].pDesc->pszName);
420	if (g_aServices[j].pDesc->pszOptions)
421	RTPrintf("%s", g_aServices[j].pDesc->pszOptions);
422	}
423	RTPrintf("\n"
424	" Copyright (C) 2009-" VBOX_C_YEAR " " VBOX_VENDOR "\n");
425
426	return 1;
427	}
428
429
430	/**
431	* Displays an error message.
432	*
433	* @returns RTEXITCODE_FAILURE.
434	* @param pszFormat The message text.
435	* @param ... Format arguments.
436	*/
437	RTEXITCODE VGSvcError(const char *pszFormat, ...)
438	{
439	va_list args;
440	va_start(args, pszFormat);
441	char *psz = NULL;
442	RTStrAPrintfV(&psz, pszFormat, args);
443	va_end(args);
444
445	AssertPtr(psz);
446	LogRel(("Error: %s", psz));
447
448	RTStrFree(psz);
449
450	return RTEXITCODE_FAILURE;
451	}
452
453
454	/**
455	* Displays a verbose message based on the currently
456	* set global verbosity level.
457	*
458	* @param iLevel Minimum log level required to display this message.
459	* @param pszFormat The message text.
460	* @param ... Format arguments.
461	*/
462	void VGSvcVerbose(unsigned iLevel, const char *pszFormat, ...)
463	{
464	if (iLevel <= g_cVerbosity)
465	{
466	va_list va;
467	va_start(va, pszFormat);
468	VGSvcLogV(pszFormat, va);
469	va_end(va);
470	}
471	}
472
473
474	/**
475	* Reports the current VBoxService status to the host.
476	*
477	* This makes sure that the Failed state is sticky.
478	*
479	* @return IPRT status code.
480	* @param enmStatus Status to report to the host.
481	*/
482	int VGSvcReportStatus(VBoxGuestFacilityStatus enmStatus)
483	{
484	/*
485	* VBoxGuestFacilityStatus_Failed is sticky.
486	*/
487	static VBoxGuestFacilityStatus s_enmLastStatus = VBoxGuestFacilityStatus_Inactive;
488	VGSvcVerbose(4, "Setting VBoxService status to %u\n", enmStatus);
489	if (s_enmLastStatus != VBoxGuestFacilityStatus_Failed)
490	{
491	int rc = VbglR3ReportAdditionsStatus(VBoxGuestFacilityType_VBoxService, enmStatus, 0 /* Flags */);
492	if (RT_FAILURE(rc))
493	{
494	VGSvcError("Could not report VBoxService status (%u), rc=%Rrc\n", enmStatus, rc);
495	return rc;
496	}
497	s_enmLastStatus = enmStatus;
498	}
499	return VINF_SUCCESS;
500	}
501
502
503	/**
504	* Gets a 32-bit value argument.
505	* @todo Get rid of this and VGSvcArgString() as soon as we have RTOpt handling.
506	*
507	* @returns 0 on success, non-zero exit code on error.
508	* @param argc The argument count.
509	* @param argv The argument vector
510	* @param psz Where in *pi to start looking for the value argument.
511	* @param pi Where to find and perhaps update the argument index.
512	* @param pu32 Where to store the 32-bit value.
513	* @param u32Min The minimum value.
514	* @param u32Max The maximum value.
515	*/
516	int VGSvcArgUInt32(int argc, char *argv, const char psz, int pi, uint32_t pu32, uint32_t u32Min, uint32_t u32Max)
517	{
518	if (psz == ':' \|\| psz == '=')
519	psz++;
520	if (!*psz)
521	{
522	if (*pi + 1 >= argc)
523	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing value for the '%s' argument\n", argv[*pi]);
524	psz = argv[++*pi];
525	}
526
527	char *pszNext;
528	int rc = RTStrToUInt32Ex(psz, &pszNext, 0, pu32);
529	if (RT_FAILURE(rc) \|\| *pszNext)
530	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Failed to convert interval '%s' to a number\n", psz);
531	if (pu32 < u32Min \|\| pu32 > u32Max)
532	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "The timesync interval of %RU32 seconds is out of range [%RU32..%RU32]\n",
533	*pu32, u32Min, u32Max);
534	return 0;
535	}
536
537
538	/** @todo Get rid of this and VGSvcArgUInt32() as soon as we have RTOpt handling. */
539	static int vgsvcArgString(int argc, char *argv, const char psz, int pi, char pszBuf, size_t cbBuf)
540	{
541	AssertPtrReturn(pszBuf, VERR_INVALID_POINTER);
542	AssertReturn(cbBuf, VERR_INVALID_PARAMETER);
543
544	if (psz == ':' \|\| psz == '=')
545	psz++;
546	if (!*psz)
547	{
548	if (*pi + 1 >= argc)
549	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Missing string for the '%s' argument\n", argv[*pi]);
550	psz = argv[++*pi];
551	}
552
553	if (!RTStrPrintf(pszBuf, cbBuf, "%s", psz))
554	return RTMsgErrorExit(RTEXITCODE_FAILURE, "String for '%s' argument too big\n", argv[*pi]);
555	return 0;
556	}
557
558
559	/**
560	* The service thread.
561	*
562	* @returns Whatever the worker function returns.
563	* @param ThreadSelf My thread handle.
564	* @param pvUser The service index.
565	*/
566	static DECLCALLBACK(int) vgsvcThread(RTTHREAD ThreadSelf, void *pvUser)
567	{
568	const unsigned i = (uintptr_t)pvUser;
569
570	#ifndef RT_OS_WINDOWS
571	/*
572	* Block all signals for this thread. Only the main thread will handle signals.
573	*/
574	sigset_t signalMask;
575	sigfillset(&signalMask);
576	pthread_sigmask(SIG_BLOCK, &signalMask, NULL);
577	#endif
578
579	int rc = g_aServices[i].pDesc->pfnWorker(&g_aServices[i].fShutdown);
580	ASMAtomicXchgBool(&g_aServices[i].fShutdown, true);
581	RTThreadUserSignal(ThreadSelf);
582	return rc;
583	}
584
585
586	/**
587	* Lazily calls the pfnPreInit method on each service.
588	*
589	* @returns VBox status code, error message displayed.
590	*/
591	static RTEXITCODE vgsvcLazyPreInit(void)
592	{
593	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
594	if (!g_aServices[j].fPreInited)
595	{
596	int rc = g_aServices[j].pDesc->pfnPreInit();
597	if (RT_FAILURE(rc))
598	return VGSvcError("Service '%s' failed pre-init: %Rrc\n", g_aServices[j].pDesc->pszName, rc);
599	g_aServices[j].fPreInited = true;
600	}
601	return RTEXITCODE_SUCCESS;
602	}
603
604
605	/**
606	* Count the number of enabled services.
607	*/
608	static unsigned vgsvcCountEnabledServices(void)
609	{
610	unsigned cEnabled = 0;
611	for (unsigned i = 0; i < RT_ELEMENTS(g_aServices); i++)
612	cEnabled += g_aServices[i].fEnabled;
613	return cEnabled;
614	}
615
616
617	#ifdef RT_OS_WINDOWS
618	/**
619	* Console control event callback.
620	*
621	* @returns TRUE if handled, FALSE if not.
622	* @param dwCtrlType The control event type.
623	*
624	* @remarks This is generally called on a new thread, so we're racing every
625	* other thread in the process.
626	*/
627	static BOOL WINAPI vgsvcWinConsoleControlHandler(DWORD dwCtrlType) RT_NOTHROW_DEF
628	{
629	int rc = VINF_SUCCESS;
630	bool fEventHandled = FALSE;
631	switch (dwCtrlType)
632	{
633	/* User pressed CTRL+C or CTRL+BREAK or an external event was sent
634	* via GenerateConsoleCtrlEvent(). */
635	case CTRL_BREAK_EVENT:
636	case CTRL_CLOSE_EVENT:
637	case CTRL_C_EVENT:
638	VGSvcVerbose(2, "ControlHandler: Received break/close event\n");
639	rc = VGSvcStopServices();
640	fEventHandled = TRUE;
641	break;
642	default:
643	break;
644	/** @todo Add other events here. */
645	}
646
647	if (RT_FAILURE(rc))
648	VGSvcError("ControlHandler: Event %ld handled with error rc=%Rrc\n",
649	dwCtrlType, rc);
650	return fEventHandled;
651	}
652	#endif /* RT_OS_WINDOWS */
653
654
655	/**
656	* Starts the service.
657	*
658	* @returns VBox status code, errors are fully bitched.
659	*
660	* @remarks Also called from VBoxService-win.cpp, thus not static.
661	*/
662	int VGSvcStartServices(void)
663	{
664	int rc;
665
666	VGSvcReportStatus(VBoxGuestFacilityStatus_Init);
667
668	/*
669	* Initialize the services.
670	*/
671	VGSvcVerbose(2, "Initializing services ...\n");
672	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
673	if (g_aServices[j].fEnabled)
674	{
675	rc = g_aServices[j].pDesc->pfnInit();
676	if (RT_FAILURE(rc))
677	{
678	if (rc != VERR_SERVICE_DISABLED)
679	{
680	VGSvcError("Service '%s' failed to initialize: %Rrc\n", g_aServices[j].pDesc->pszName, rc);
681	VGSvcReportStatus(VBoxGuestFacilityStatus_Failed);
682	return rc;
683	}
684
685	g_aServices[j].fEnabled = false;
686	VGSvcVerbose(0, "Service '%s' was disabled because of missing functionality\n", g_aServices[j].pDesc->pszName);
687	}
688	}
689
690	/*
691	* Start the service(s).
692	*/
693	VGSvcVerbose(2, "Starting services ...\n");
694	rc = VINF_SUCCESS;
695	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
696	{
697	if (!g_aServices[j].fEnabled)
698	continue;
699
700	VGSvcVerbose(2, "Starting service '%s' ...\n", g_aServices[j].pDesc->pszName);
701	rc = RTThreadCreate(&g_aServices[j].Thread, vgsvcThread, (void *)(uintptr_t)j, 0,
702	RTTHREADTYPE_DEFAULT, RTTHREADFLAGS_WAITABLE, g_aServices[j].pDesc->pszName);
703	if (RT_FAILURE(rc))
704	{
705	VGSvcError("RTThreadCreate failed, rc=%Rrc\n", rc);
706	break;
707	}
708	g_aServices[j].fStarted = true;
709
710	/* Wait for the thread to initialize. */
711	/** @todo There is a race between waiting and checking
712	* the fShutdown flag of a thread here and processing
713	* the thread's actual worker loop. If the thread decides
714	* to exit the loop before we skipped the fShutdown check
715	* below the service will fail to start! */
716	/** @todo This presumably means either a one-shot service or that
717	* something has gone wrong. In the second case treating it as failure
718	* to start is probably right, so we need a way to signal the first
719	* rather than leaving the idle thread hanging around. A flag in the
720	* service description? */
721	RTThreadUserWait(g_aServices[j].Thread, 60 * 1000);
722	if (g_aServices[j].fShutdown)
723	{
724	VGSvcError("Service '%s' failed to start!\n", g_aServices[j].pDesc->pszName);
725	rc = VERR_GENERAL_FAILURE;
726	}
727	}
728
729	if (RT_SUCCESS(rc))
730	VGSvcVerbose(1, "All services started.\n");
731	else
732	{
733	VGSvcError("An error occcurred while the services!\n");
734	VGSvcReportStatus(VBoxGuestFacilityStatus_Failed);
735	}
736	return rc;
737	}
738
739
740	/**
741	* Stops and terminates the services.
742	*
743	* This should be called even when VBoxServiceStartServices fails so it can
744	* clean up anything that we succeeded in starting.
745	*
746	* @remarks Also called from VBoxService-win.cpp, thus not static.
747	*/
748	int VGSvcStopServices(void)
749	{
750	VGSvcReportStatus(VBoxGuestFacilityStatus_Terminating);
751
752	/*
753	* Signal all the services.
754	*/
755	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
756	ASMAtomicWriteBool(&g_aServices[j].fShutdown, true);
757
758	/*
759	* Do the pfnStop callback on all running services.
760	*/
761	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
762	if (g_aServices[j].fStarted)
763	{
764	VGSvcVerbose(3, "Calling stop function for service '%s' ...\n", g_aServices[j].pDesc->pszName);
765	g_aServices[j].pDesc->pfnStop();
766	}
767
768	VGSvcVerbose(3, "All stop functions for services called\n");
769
770	/*
771	* Wait for all the service threads to complete.
772	*/
773	int rc = VINF_SUCCESS;
774	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
775	{
776	if (!g_aServices[j].fEnabled) /* Only stop services which were started before. */
777	continue;
778	if (g_aServices[j].Thread != NIL_RTTHREAD)
779	{
780	VGSvcVerbose(2, "Waiting for service '%s' to stop ...\n", g_aServices[j].pDesc->pszName);
781	int rc2 = VINF_SUCCESS;
782	for (int i = 0; i < 30; i++) /* Wait 30 seconds in total */
783	{
784	rc2 = RTThreadWait(g_aServices[j].Thread, 1000 /* Wait 1 second */, NULL);
785	if (RT_SUCCESS(rc2))
786	break;
787	#ifdef RT_OS_WINDOWS
788	/* Notify SCM that it takes a bit longer ... */
789	VGSvcWinSetStopPendingStatus(i + j*32);
790	#endif
791	}
792	if (RT_FAILURE(rc2))
793	{
794	VGSvcError("Service '%s' failed to stop. (%Rrc)\n", g_aServices[j].pDesc->pszName, rc2);
795	rc = rc2;
796	}
797	}
798	VGSvcVerbose(3, "Terminating service '%s' (%d) ...\n", g_aServices[j].pDesc->pszName, j);
799	g_aServices[j].pDesc->pfnTerm();
800	}
801
802	#ifdef RT_OS_WINDOWS
803	/*
804	* Wake up and tell the main() thread that we're shutting down (it's
805	* sleeping in VBoxServiceMainWait).
806	*/
807	ASMAtomicWriteBool(&g_fWindowsServiceShutdown, true);
808	if (g_hEvtWindowsService != NIL_RTSEMEVENT)
809	{
810	VGSvcVerbose(3, "Stopping the main thread...\n");
811	int rc2 = RTSemEventSignal(g_hEvtWindowsService);
812	AssertRC(rc2);
813	}
814	#endif
815
816	VGSvcVerbose(2, "Stopping services returning: %Rrc\n", rc);
817	VGSvcReportStatus(RT_SUCCESS(rc) ? VBoxGuestFacilityStatus_Paused : VBoxGuestFacilityStatus_Failed);
818	return rc;
819	}
820
821
822	/**
823	* Block the main thread until the service shuts down.
824	*
825	* @remarks Also called from VBoxService-win.cpp, thus not static.
826	*/
827	void VGSvcMainWait(void)
828	{
829	int rc;
830
831	VGSvcReportStatus(VBoxGuestFacilityStatus_Active);
832
833	#ifdef RT_OS_WINDOWS
834	/*
835	* Wait for the semaphore to be signalled.
836	*/
837	VGSvcVerbose(1, "Waiting in main thread\n");
838	rc = RTSemEventCreate(&g_hEvtWindowsService);
839	AssertRC(rc);
840	while (!ASMAtomicReadBool(&g_fWindowsServiceShutdown))
841	{
842	rc = RTSemEventWait(g_hEvtWindowsService, RT_INDEFINITE_WAIT);
843	AssertRC(rc);
844	}
845	RTSemEventDestroy(g_hEvtWindowsService);
846	g_hEvtWindowsService = NIL_RTSEMEVENT;
847	#else
848	/*
849	* Wait explicitly for a HUP, INT, QUIT, ABRT or TERM signal, blocking
850	* all important signals.
851	*
852	* The annoying EINTR/ERESTART loop is for the benefit of Solaris where
853	* sigwait returns when we receive a SIGCHLD. Kind of makes sense since
854	* the signal has to be delivered... Anyway, darwin (10.9.5) has a much
855	* worse way of dealing with SIGCHLD, apparently it'll just return any
856	* of the signals we're waiting on when SIGCHLD becomes pending on this
857	* thread. So, we wait for SIGCHLD here and ignores it.
858	*/
859	sigset_t signalMask;
860	sigemptyset(&signalMask);
861	sigaddset(&signalMask, SIGHUP);
862	sigaddset(&signalMask, SIGINT);
863	sigaddset(&signalMask, SIGQUIT);
864	sigaddset(&signalMask, SIGABRT);
865	sigaddset(&signalMask, SIGTERM);
866	sigaddset(&signalMask, SIGCHLD);
867	pthread_sigmask(SIG_BLOCK, &signalMask, NULL);
868
869	int iSignal;
870	do
871	{
872	iSignal = -1;
873	rc = sigwait(&signalMask, &iSignal);
874	}
875	while ( rc == EINTR
876	# ifdef ERESTART
877	\|\| rc == ERESTART
878	# endif
879	\|\| iSignal == SIGCHLD
880	);
881
882	VGSvcVerbose(3, "VGSvcMainWait: Received signal %d (rc=%d)\n", iSignal, rc);
883	#endif /* !RT_OS_WINDOWS */
884	}
885
886
887	/**
888	* Report VbglR3InitUser / VbglR3Init failure.
889	*
890	* @returns RTEXITCODE_FAILURE
891	* @param rcVbgl The failing status code.
892	*/
893	static RTEXITCODE vbglInitFailure(int rcVbgl)
894	{
895	if (rcVbgl == VERR_ACCESS_DENIED)
896	return RTMsgErrorExit(RTEXITCODE_FAILURE,
897	"Insufficient privileges to start %s! Please start with Administrator/root privileges!\n",
898	g_pszProgName);
899	return RTMsgErrorExit(RTEXITCODE_FAILURE, "VbglR3Init failed with rc=%Rrc\n", rcVbgl);
900	}
901
902
903	int main(int argc, char **argv)
904	{
905	RTEXITCODE rcExit;
906
907	/*
908	* Init globals and such.
909	*
910	* Note! The --utf8-argv stuff is an internal hack to avoid locale configuration
911	* issues preventing us from passing non-ASCII string to child processes.
912	*/
913	uint32_t fIprtFlags = 0;
914	#ifdef VBOXSERVICE_ARG1_UTF8_ARGV
915	if (argc > 1 && strcmp(argv[1], VBOXSERVICE_ARG1_UTF8_ARGV) == 0)
916	{
917	argv[1] = argv[0];
918	argv++;
919	argc--;
920	fIprtFlags \|= RTR3INIT_FLAGS_UTF8_ARGV;
921	}
922	#endif
923	int rc = RTR3InitExe(argc, &argv, fIprtFlags);
924	if (RT_FAILURE(rc))
925	return RTMsgInitFailure(rc);
926
927	g_pszProgName = RTPathFilename(argv[0]);
928	#ifdef RT_OS_WINDOWS
929	VGSvcWinResolveApis();
930	#endif
931	#ifdef DEBUG
932	rc = RTCritSectInit(&g_csLog);
933	AssertRC(rc);
934	#endif
935
936	#ifdef VBOX_WITH_VBOXSERVICE_TOOLBOX
937	/*
938	* Run toolbox code before all other stuff since these things are simpler
939	* shell/file/text utility like programs that just happens to be inside
940	* VBoxService and shouldn't be subject to /dev/vboxguest, pid-files and
941	* global mutex restrictions.
942	*/
943	if (VGSvcToolboxMain(argc, argv, &rcExit))
944	return rcExit;
945	#endif
946
947	bool fUserSession = false;
948	#ifdef VBOX_WITH_VBOXSERVICE_CONTROL
949	/*
950	* Check if we're the specially spawned VBoxService.exe process that
951	* handles a guest control session.
952	*/
953	if ( argc >= 2
954	&& !RTStrICmp(argv[1], VBOXSERVICECTRLSESSION_GETOPT_PREFIX))
955	fUserSession = true;
956	#endif
957
958	/*
959	* Connect to the kernel part before daemonizing and before we do the sub-service
960	* pre-init just in case one of services needs do to some initial stuff with it.
961	*
962	* However, we do not fail till after we've parsed arguments, because that will
963	* prevent useful stuff like --help, --register, --unregister and --version from
964	* working when the driver hasn't been installed/loaded yet.
965	*/
966	int const rcVbgl = fUserSession ? VbglR3InitUser() : VbglR3Init();
967
968	#ifdef RT_OS_WINDOWS
969	/*
970	* Check if we're the specially spawned VBoxService.exe process that
971	* handles page fusion. This saves an extra statically linked executable.
972	*/
973	if ( argc == 2
974	&& !RTStrICmp(argv[1], "pagefusion"))
975	{
976	if (RT_SUCCESS(rcVbgl))
977	return VGSvcPageSharingWorkerChild();
978	return vbglInitFailure(rcVbgl);
979	}
980	#endif
981
982	#ifdef VBOX_WITH_VBOXSERVICE_CONTROL
983	/*
984	* Check if we're the specially spawned VBoxService.exe process that
985	* handles a guest control session.
986	*/
987	if (fUserSession)
988	{
989	if (RT_SUCCESS(rcVbgl))
990	return VGSvcGstCtrlSessionSpawnInit(argc, argv);
991	return vbglInitFailure(rcVbgl);
992	}
993	#endif
994
995	/*
996	* Parse the arguments.
997	*
998	* Note! This code predates RTGetOpt, thus the manual parsing.
999	*/
1000	bool fDaemonize = true;
1001	bool fDaemonized = false;
1002	for (int i = 1; i < argc; i++)
1003	{
1004	const char *psz = argv[i];
1005	if (*psz != '-')
1006	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Unknown argument '%s'\n", psz);
1007	psz++;
1008
1009	/* translate long argument to short */
1010	if (*psz == '-')
1011	{
1012	psz++;
1013	size_t cch = strlen(psz);
1014	#define MATCHES(strconst) ( cch == sizeof(strconst) - 1 \
1015	&& !memcmp(psz, strconst, sizeof(strconst) - 1) )
1016	if (MATCHES("foreground"))
1017	psz = "f";
1018	else if (MATCHES("verbose"))
1019	psz = "v";
1020	else if (MATCHES("version"))
1021	psz = "V";
1022	else if (MATCHES("help"))
1023	psz = "h";
1024	else if (MATCHES("interval"))
1025	psz = "i";
1026	#ifdef RT_OS_WINDOWS
1027	else if (MATCHES("register"))
1028	psz = "r";
1029	else if (MATCHES("unregister"))
1030	psz = "u";
1031	#endif
1032	else if (MATCHES("logfile"))
1033	psz = "l";
1034	else if (MATCHES("pidfile"))
1035	psz = "p";
1036	else if (MATCHES("daemonized"))
1037	{
1038	fDaemonized = true;
1039	continue;
1040	}
1041	else
1042	{
1043	bool fFound = false;
1044
1045	if (cch > sizeof("enable-") && !memcmp(psz, RT_STR_TUPLE("enable-")))
1046	for (unsigned j = 0; !fFound && j < RT_ELEMENTS(g_aServices); j++)
1047	if ((fFound = !RTStrICmp(psz + sizeof("enable-") - 1, g_aServices[j].pDesc->pszName)))
1048	g_aServices[j].fEnabled = true;
1049
1050	if (cch > sizeof("disable-") && !memcmp(psz, RT_STR_TUPLE("disable-")))
1051	for (unsigned j = 0; !fFound && j < RT_ELEMENTS(g_aServices); j++)
1052	if ((fFound = !RTStrICmp(psz + sizeof("disable-") - 1, g_aServices[j].pDesc->pszName)))
1053	g_aServices[j].fEnabled = false;
1054
1055	if (cch > sizeof("only-") && !memcmp(psz, RT_STR_TUPLE("only-")))
1056	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
1057	{
1058	g_aServices[j].fEnabled = !RTStrICmp(psz + sizeof("only-") - 1, g_aServices[j].pDesc->pszName);
1059	if (g_aServices[j].fEnabled)
1060	fFound = true;
1061	}
1062
1063	if (!fFound)
1064	{
1065	rcExit = vgsvcLazyPreInit();
1066	if (rcExit != RTEXITCODE_SUCCESS)
1067	return rcExit;
1068	for (unsigned j = 0; !fFound && j < RT_ELEMENTS(g_aServices); j++)
1069	{
1070	rc = g_aServices[j].pDesc->pfnOption(NULL, argc, argv, &i);
1071	fFound = rc == VINF_SUCCESS;
1072	if (fFound)
1073	break;
1074	if (rc != -1)
1075	return rc;
1076	}
1077	}
1078	if (!fFound)
1079	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Unknown option '%s'\n", argv[i]);
1080	continue;
1081	}
1082	#undef MATCHES
1083	}
1084
1085	/* handle the string of short options. */
1086	do
1087	{
1088	switch (*psz)
1089	{
1090	case 'i':
1091	rc = VGSvcArgUInt32(argc, argv, psz + 1, &i, &g_DefaultInterval, 1, (UINT32_MAX / 1000) - 1);
1092	if (rc)
1093	return rc;
1094	psz = NULL;
1095	break;
1096
1097	case 'f':
1098	fDaemonize = false;
1099	break;
1100
1101	case 'v':
1102	g_cVerbosity++;
1103	break;
1104
1105	case 'V':
1106	RTPrintf("%sr%s\n", RTBldCfgVersion(), RTBldCfgRevisionStr());
1107	return RTEXITCODE_SUCCESS;
1108
1109	case 'h':
1110	case '?':
1111	return vgsvcUsage();
1112
1113	#ifdef RT_OS_WINDOWS
1114	case 'r':
1115	return VGSvcWinInstall();
1116
1117	case 'u':
1118	return VGSvcWinUninstall();
1119	#endif
1120
1121	case 'l':
1122	{
1123	rc = vgsvcArgString(argc, argv, psz + 1, &i, g_szLogFile, sizeof(g_szLogFile));
1124	if (rc)
1125	return rc;
1126	psz = NULL;
1127	break;
1128	}
1129
1130	case 'p':
1131	{
1132	rc = vgsvcArgString(argc, argv, psz + 1, &i, g_szPidFile, sizeof(g_szPidFile));
1133	if (rc)
1134	return rc;
1135	psz = NULL;
1136	break;
1137	}
1138
1139	default:
1140	{
1141	rcExit = vgsvcLazyPreInit();
1142	if (rcExit != RTEXITCODE_SUCCESS)
1143	return rcExit;
1144
1145	bool fFound = false;
1146	for (unsigned j = 0; j < RT_ELEMENTS(g_aServices); j++)
1147	{
1148	rc = g_aServices[j].pDesc->pfnOption(&psz, argc, argv, &i);
1149	fFound = rc == VINF_SUCCESS;
1150	if (fFound)
1151	break;
1152	if (rc != -1)
1153	return rc;
1154	}
1155	if (!fFound)
1156	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Unknown option '%c' (%s)\n", *psz, argv[i]);
1157	break;
1158	}
1159	}
1160	} while (psz && *++psz);
1161	}
1162
1163	/* Now we can report the VBGL failure. */
1164	if (RT_FAILURE(rcVbgl))
1165	return vbglInitFailure(rcVbgl);
1166
1167	/* Check that at least one service is enabled. */
1168	if (vgsvcCountEnabledServices() == 0)
1169	return RTMsgErrorExit(RTEXITCODE_SYNTAX, "At least one service must be enabled\n");
1170
1171	rc = VGSvcLogCreate(g_szLogFile[0] ? g_szLogFile : NULL);
1172	if (RT_FAILURE(rc))
1173	return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to create release log '%s', rc=%Rrc\n",
1174	g_szLogFile[0] ? g_szLogFile : "<None>", rc);
1175
1176	/* Call pre-init if we didn't do it already. */
1177	rcExit = vgsvcLazyPreInit();
1178	if (rcExit != RTEXITCODE_SUCCESS)
1179	return rcExit;
1180
1181	#ifdef VBOX_WITH_VBOXSERVICE_DRMRESIZE
1182	# ifdef RT_OS_LINUX
1183	rc = VbglR3DrmClientStart();
1184	if (RT_FAILURE(rc))
1185	VGSvcVerbose(0, "VMSVGA DRM resizing client not started, rc=%Rrc\n", rc);
1186	# endif /* RT_OS_LINUX */
1187	#endif /* VBOX_WITH_VBOXSERVICE_DRMRESIZE */
1188
1189	#ifdef RT_OS_WINDOWS
1190	/*
1191	* Make sure only one instance of VBoxService runs at a time. Create a
1192	* global mutex for that.
1193	*
1194	* Note! The \\Global\ namespace was introduced with Win2K, thus the
1195	* version check.
1196	* Note! If the mutex exists CreateMutex will open it and set last error to
1197	* ERROR_ALREADY_EXISTS.
1198	*/
1199	OSVERSIONINFOEX OSInfoEx;
1200	RT_ZERO(OSInfoEx);
1201	OSInfoEx.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
1202
1203
1204	SetLastError(NO_ERROR);
1205	HANDLE hMutexAppRunning;
1206	if (RTSystemGetNtVersion() >= RTSYSTEM_MAKE_NT_VERSION(5,0,0)) /* Windows 2000 */
1207	hMutexAppRunning = CreateMutexW(NULL, FALSE, L"Global\\" RT_CONCAT(L,VBOXSERVICE_NAME));
1208	else
1209	hMutexAppRunning = CreateMutexW(NULL, FALSE, RT_CONCAT(L,VBOXSERVICE_NAME));
1210	if (hMutexAppRunning == NULL)
1211	{
1212	DWORD dwErr = GetLastError();
1213	if ( dwErr == ERROR_ALREADY_EXISTS
1214	\|\| dwErr == ERROR_ACCESS_DENIED)
1215	{
1216	VGSvcError("%s is already running! Terminating.\n", g_pszProgName);
1217	return RTEXITCODE_FAILURE;
1218	}
1219
1220	VGSvcError("CreateMutex failed with last error %u! Terminating.\n", GetLastError());
1221	return RTEXITCODE_FAILURE;
1222	}
1223
1224	#else /* !RT_OS_WINDOWS */
1225	/* On other OSes we have PID file support provided by the actual service definitions / service wrapper scripts,
1226	* like vboxadd-service.sh on Linux or vboxservice.xml on Solaris. */
1227	#endif /* !RT_OS_WINDOWS */
1228
1229	VGSvcVerbose(0, "%s r%s started. Verbose level = %d\n", RTBldCfgVersion(), RTBldCfgRevisionStr(), g_cVerbosity);
1230
1231	/*
1232	* Daemonize if requested.
1233	*/
1234	if (fDaemonize && !fDaemonized)
1235	{
1236	#ifdef RT_OS_WINDOWS
1237	VGSvcVerbose(2, "Starting service dispatcher ...\n");
1238	rcExit = VGSvcWinEnterCtrlDispatcher();
1239	#else
1240	VGSvcVerbose(1, "Daemonizing...\n");
1241	rc = VbglR3Daemonize(false /* fNoChDir /, false / fNoClose */,
1242	false /* fRespawn /, NULL / pcRespawn */);
1243	if (RT_FAILURE(rc))
1244	return VGSvcError("Daemon failed: %Rrc\n", rc);
1245	/* in-child */
1246	#endif
1247	}
1248	#ifdef RT_OS_WINDOWS
1249	else
1250	#endif
1251	{
1252	/*
1253	* Windows: We're running the service as a console application now. Start the
1254	* services, enter the main thread's run loop and stop them again
1255	* when it returns.
1256	*
1257	* POSIX: This is used for both daemons and console runs. Start all services
1258	* and return immediately.
1259	*/
1260	#ifdef RT_OS_WINDOWS
1261	/* Install console control handler. */
1262	if (!SetConsoleCtrlHandler(vgsvcWinConsoleControlHandler, TRUE /* Add handler */))
1263	{
1264	VGSvcError("Unable to add console control handler, error=%ld\n", GetLastError());
1265	/* Just skip this error, not critical. */
1266	}
1267	#endif /* RT_OS_WINDOWS */
1268	rc = VGSvcStartServices();
1269	RTFILE hPidFile = NIL_RTFILE;
1270	if (RT_SUCCESS(rc))
1271	if (g_szPidFile[0])
1272	rc = VbglR3PidFile(g_szPidFile, &hPidFile);
1273	rcExit = RT_SUCCESS(rc) ? RTEXITCODE_SUCCESS : RTEXITCODE_FAILURE;
1274	if (RT_SUCCESS(rc))
1275	VGSvcMainWait();
1276	if (g_szPidFile[0] && hPidFile != NIL_RTFILE)
1277	VbglR3ClosePidFile(g_szPidFile, hPidFile);
1278	#ifdef RT_OS_WINDOWS
1279	/* Uninstall console control handler. */
1280	if (!SetConsoleCtrlHandler((PHANDLER_ROUTINE)NULL, FALSE /* Remove handler */))
1281	{
1282	VGSvcError("Unable to remove console control handler, error=%ld\n", GetLastError());
1283	/* Just skip this error, not critical. */
1284	}
1285	#else /* !RT_OS_WINDOWS */
1286	/* On Windows - since we're running as a console application - we already stopped all services
1287	* through the console control handler. So only do the stopping of services here on other platforms
1288	* where the break/shutdown/whatever signal was just received. */
1289	VGSvcStopServices();
1290	#endif /* RT_OS_WINDOWS */
1291	}
1292	VGSvcReportStatus(VBoxGuestFacilityStatus_Terminated);
1293
1294	#ifdef RT_OS_WINDOWS
1295	/*
1296	* Cleanup mutex.
1297	*/
1298	CloseHandle(hMutexAppRunning);
1299	#endif
1300
1301	VGSvcVerbose(0, "Ended.\n");
1302
1303	#ifdef DEBUG
1304	RTCritSectDelete(&g_csLog);
1305	//RTMemTrackerDumpAllToStdOut();
1306	#endif
1307
1308	VGSvcLogDestroy();
1309
1310	return rcExit;
1311	}

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Additions/common/VBoxService/VBoxService.cpp@ 103251

Download in other formats: