VBoxService-win.cpp@ 99828

Last change on this file since 99828 was 99828, checked in by vboxsync, 18 months ago
*: A bunch of adjustments that allows using /permissive- with Visual C++ (qt 6.x necessity).
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 24.0 KB

Line
1	/* $Id: VBoxService-win.cpp 99828 2023-05-17 13:48:57Z vboxsync $ */
2	/** @file
3	* VBoxService - Guest Additions Service Skeleton, Windows Specific Parts.
4	*/
5
6	/*
7	* Copyright (C) 2009-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	/*********************************************************************************************************************************
30	* Header Files *
31	*********************************************************************************************************************************/
32	#include <iprt/assert.h>
33	#include <iprt/err.h>
34	#include <iprt/ldr.h>
35	#include <iprt/system.h> /* For querying OS version. */
36
37	#define WIN32_NO_STATUS
38	#include <iprt/win/ws2tcpip.h>
39	#include <iprt/win/winsock2.h>
40	#undef WIN32_NO_STATUS
41	#include <iprt/nt/nt-and-windows.h>
42	#include <iprt/win/iphlpapi.h>
43	#include <aclapi.h>
44	#include <tlhelp32.h>
45	#define _NTDEF_
46	#include <Ntsecapi.h>
47
48	#include "VBoxServiceInternal.h"
49
50
51	/*********************************************************************************************************************************
52	* Internal Functions *
53	*********************************************************************************************************************************/
54	static void WINAPI vgsvcWinMain(DWORD argc, LPTSTR *argv);
55
56
57	/*********************************************************************************************************************************
58	* Global Variables *
59	*********************************************************************************************************************************/
60	static DWORD g_dwWinServiceLastStatus = 0;
61	SERVICE_STATUS_HANDLE g_hWinServiceStatus = NULL;
62	/** The semaphore for the dummy Windows service. */
63	static RTSEMEVENT g_WindowsEvent = NIL_RTSEMEVENT;
64
65	static SERVICE_TABLE_ENTRYA const g_aServiceTable[] =
66	{
67	{ (PSTR)VBOXSERVICE_NAME, vgsvcWinMain },
68	{ NULL, NULL}
69	};
70
71	/** @name APIs from ADVAPI32.DLL.
72	* @{ */
73	decltype(RegisterServiceCtrlHandlerExA) g_pfnRegisterServiceCtrlHandlerExA; /< W2K+ /
74	decltype(ChangeServiceConfig2A) g_pfnChangeServiceConfig2A; /< W2K+ /
75	decltype(GetNamedSecurityInfoA) g_pfnGetNamedSecurityInfoA; /< NT4+ /
76	decltype(SetEntriesInAclA) g_pfnSetEntriesInAclA; /< NT4+ /
77	decltype(SetNamedSecurityInfoA) g_pfnSetNamedSecurityInfoA; /< NT4+ /
78	decltype(LsaNtStatusToWinError) g_pfnLsaNtStatusToWinError; /< NT3.51+ /
79	/** @} */
80
81	/** @name API from KERNEL32.DLL
82	* @{ */
83	decltype(CreateToolhelp32Snapshot) g_pfnCreateToolhelp32Snapshot; /< W2K+, but Geoff says NT4. Hmm. /
84	decltype(Process32First) g_pfnProcess32First; /< W2K+, but Geoff says NT4. Hmm. /
85	decltype(Process32Next) g_pfnProcess32Next; /< W2K+, but Geoff says NT4. Hmm. /
86	decltype(Module32First) g_pfnModule32First; /< W2K+, but Geoff says NT4. Hmm. /
87	decltype(Module32Next) g_pfnModule32Next; /< W2K+, but Geoff says NT4. Hmm. /
88	decltype(GetSystemTimeAdjustment) g_pfnGetSystemTimeAdjustment; /< NT 3.50+ /
89	decltype(SetSystemTimeAdjustment) g_pfnSetSystemTimeAdjustment; /< NT 3.50+ /
90	/** @} */
91
92	/** @name API from NTDLL.DLL
93	* @{ */
94	decltype(ZwQuerySystemInformation) g_pfnZwQuerySystemInformation; /< NT4 (where as NtQuerySystemInformation is W2K). /
95	/** @} */
96
97	/** @name API from IPHLPAPI.DLL
98	* @{ */
99	decltype(GetAdaptersInfo) *g_pfnGetAdaptersInfo;
100	/** @} */
101
102	/** @name APIs from WS2_32.DLL
103	* @note WSAIoctl is not present in wsock32.dll, so no point in trying the
104	* fallback here.
105	* @{ */
106	decltype(WSAStartup) *g_pfnWSAStartup;
107	decltype(WSACleanup) *g_pfnWSACleanup;
108	decltype(WSASocketA) *g_pfnWSASocketA;
109	decltype(WSAIoctl) *g_pfnWSAIoctl;
110	decltype(WSAGetLastError) *g_pfnWSAGetLastError;
111	decltype(closesocket) *g_pfnclosesocket;
112	decltype(inet_ntoa) *g_pfninet_ntoa;
113
114	/** @} */
115
116	/**
117	* Resolve APIs not present on older windows versions.
118	*/
119	void VGSvcWinResolveApis(void)
120	{
121	RTLDRMOD hLdrMod;
122	#define RESOLVE_SYMBOL(a_fn) do { RT_CONCAT(g_pfn, a_fn) = (decltype(a_fn) *)RTLdrGetFunction(hLdrMod, #a_fn); } while (0)
123
124	/* From ADVAPI32.DLL: */
125	int rc = RTLdrLoadSystem("advapi32.dll", true /fNoUnload/, &hLdrMod);
126	AssertRC(rc);
127	if (RT_SUCCESS(rc))
128	{
129	RESOLVE_SYMBOL(RegisterServiceCtrlHandlerExA);
130	RESOLVE_SYMBOL(ChangeServiceConfig2A);
131	RESOLVE_SYMBOL(GetNamedSecurityInfoA);
132	RESOLVE_SYMBOL(SetEntriesInAclA);
133	RESOLVE_SYMBOL(SetNamedSecurityInfoA);
134	RESOLVE_SYMBOL(LsaNtStatusToWinError);
135	RTLdrClose(hLdrMod);
136	}
137
138	/* From KERNEL32.DLL: */
139	rc = RTLdrLoadSystem("kernel32.dll", true /fNoUnload/, &hLdrMod);
140	AssertRC(rc);
141	if (RT_SUCCESS(rc))
142	{
143	RESOLVE_SYMBOL(CreateToolhelp32Snapshot);
144	RESOLVE_SYMBOL(Process32First);
145	RESOLVE_SYMBOL(Process32Next);
146	RESOLVE_SYMBOL(Module32First);
147	RESOLVE_SYMBOL(Module32Next);
148	RESOLVE_SYMBOL(GetSystemTimeAdjustment);
149	RESOLVE_SYMBOL(SetSystemTimeAdjustment);
150	RTLdrClose(hLdrMod);
151	}
152
153	/* From NTDLL.DLL: */
154	rc = RTLdrLoadSystem("ntdll.dll", true /fNoUnload/, &hLdrMod);
155	AssertRC(rc);
156	if (RT_SUCCESS(rc))
157	{
158	RESOLVE_SYMBOL(ZwQuerySystemInformation);
159	RTLdrClose(hLdrMod);
160	}
161
162	/* From IPHLPAPI.DLL: */
163	rc = RTLdrLoadSystem("iphlpapi.dll", true /fNoUnload/, &hLdrMod);
164	if (RT_SUCCESS(rc))
165	{
166	RESOLVE_SYMBOL(GetAdaptersInfo);
167	RTLdrClose(hLdrMod);
168	}
169
170	/* From WS2_32.DLL: */
171	rc = RTLdrLoadSystem("ws2_32.dll", true /fNoUnload/, &hLdrMod);
172	if (RT_SUCCESS(rc))
173	{
174	RESOLVE_SYMBOL(WSAStartup);
175	RESOLVE_SYMBOL(WSACleanup);
176	RESOLVE_SYMBOL(WSASocketA);
177	RESOLVE_SYMBOL(WSAIoctl);
178	RESOLVE_SYMBOL(WSAGetLastError);
179	RESOLVE_SYMBOL(closesocket);
180	RESOLVE_SYMBOL(inet_ntoa);
181	RTLdrClose(hLdrMod);
182	}
183	}
184
185
186	/**
187	* @todo Add full unicode support.
188	* @todo Add event log capabilities / check return values.
189	*/
190	static int vgsvcWinAddAceToObjectsSecurityDescriptor(LPCSTR pszObjName, SE_OBJECT_TYPE enmObjectType, const char *pszTrustee,
191	TRUSTEE_FORM enmTrusteeForm, DWORD dwAccessRights, ACCESS_MODE fAccessMode,
192	DWORD dwInheritance)
193	{
194	int rc;
195	if ( g_pfnGetNamedSecurityInfoA
196	&& g_pfnSetEntriesInAclA
197	&& g_pfnSetNamedSecurityInfoA)
198	{
199	/* Get a pointer to the existing DACL. */
200	PSECURITY_DESCRIPTOR pSD = NULL;
201	PACL pOldDACL = NULL;
202	DWORD rcWin = g_pfnGetNamedSecurityInfoA(pszObjName, enmObjectType, DACL_SECURITY_INFORMATION,
203	NULL, NULL, &pOldDACL, NULL, &pSD);
204	if (rcWin == ERROR_SUCCESS)
205	{
206	/* Initialize an EXPLICIT_ACCESS structure for the new ACE. */
207	EXPLICIT_ACCESSA ExplicitAccess;
208	RT_ZERO(ExplicitAccess);
209	ExplicitAccess.grfAccessPermissions = dwAccessRights;
210	ExplicitAccess.grfAccessMode = fAccessMode;
211	ExplicitAccess.grfInheritance = dwInheritance;
212	ExplicitAccess.Trustee.TrusteeForm = enmTrusteeForm;
213	ExplicitAccess.Trustee.ptstrName = (char *)pszTrustee;
214
215	/* Create a new ACL that merges the new ACE into the existing DACL. */
216	PACL pNewDACL = NULL;
217	rcWin = g_pfnSetEntriesInAclA(1, &ExplicitAccess, pOldDACL, &pNewDACL);
218	if (rcWin == ERROR_SUCCESS)
219	{
220	/* Attach the new ACL as the object's DACL. */
221	rcWin = g_pfnSetNamedSecurityInfoA((PSTR)pszObjName, enmObjectType, DACL_SECURITY_INFORMATION,
222	NULL, NULL, pNewDACL, NULL);
223	if (rcWin == ERROR_SUCCESS)
224	rc = VINF_SUCCESS;
225	else
226	{
227	VGSvcError("AddAceToObjectsSecurityDescriptor: SetNamedSecurityInfo: Error %u\n", rcWin);
228	rc = RTErrConvertFromWin32(rcWin);
229	}
230	if (pNewDACL)
231	LocalFree(pNewDACL);
232	}
233	else
234	{
235	VGSvcError("AddAceToObjectsSecurityDescriptor: SetEntriesInAcl: Error %u\n", rcWin);
236	rc = RTErrConvertFromWin32(rcWin);
237	}
238	if (pSD)
239	LocalFree(pSD);
240	}
241	else
242	{
243	if (rcWin == ERROR_FILE_NOT_FOUND)
244	VGSvcError("AddAceToObjectsSecurityDescriptor: Object not found/installed: %s\n", pszObjName);
245	else
246	VGSvcError("AddAceToObjectsSecurityDescriptor: GetNamedSecurityInfo: Error %u\n", rcWin);
247	rc = RTErrConvertFromWin32(rcWin);
248	}
249	}
250	else
251	rc = VINF_SUCCESS; /* fake it */
252	return rc;
253	}
254
255
256	/** Reports our current status to the SCM. */
257	static BOOL vgsvcWinSetStatus(DWORD dwStatus, DWORD dwCheckPoint)
258	{
259	if (g_hWinServiceStatus == NULL) /* Program could be in testing mode, so no service environment available. */
260	return FALSE;
261
262	VGSvcVerbose(2, "Setting service status to: %ld\n", dwStatus);
263	g_dwWinServiceLastStatus = dwStatus;
264
265	SERVICE_STATUS ss;
266	RT_ZERO(ss);
267
268	ss.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
269	ss.dwCurrentState = dwStatus;
270	/* Don't accept controls when in start pending state. */
271	if (ss.dwCurrentState != SERVICE_START_PENDING)
272	{
273	ss.dwControlsAccepted = SERVICE_ACCEPT_STOP \| SERVICE_ACCEPT_SHUTDOWN;
274
275	/* Don't use SERVICE_ACCEPT_SESSIONCHANGE on Windows 2000 or earlier. This makes SCM angry. */
276	char szOSVersion[32];
277	int rc = RTSystemQueryOSInfo(RTSYSOSINFO_RELEASE, szOSVersion, sizeof(szOSVersion));
278	if (RT_SUCCESS(rc))
279	{
280	if (RTStrVersionCompare(szOSVersion, "5.1") >= 0)
281	ss.dwControlsAccepted \|= SERVICE_ACCEPT_SESSIONCHANGE;
282	}
283	else
284	VGSvcError("Error determining OS version, rc=%Rrc\n", rc);
285	}
286
287	ss.dwWin32ExitCode = NO_ERROR;
288	ss.dwServiceSpecificExitCode = 0; /* Not used */
289	ss.dwCheckPoint = dwCheckPoint;
290	ss.dwWaitHint = 3000;
291
292	BOOL fStatusSet = SetServiceStatus(g_hWinServiceStatus, &ss);
293	if (!fStatusSet)
294	VGSvcError("Error reporting service status=%ld (controls=%x, checkpoint=%ld) to SCM: %ld\n",
295	dwStatus, ss.dwControlsAccepted, dwCheckPoint, GetLastError());
296	return fStatusSet;
297	}
298
299
300	/**
301	* Reports SERVICE_STOP_PENDING to SCM.
302	*
303	* @param uCheckPoint Some number.
304	*/
305	void VGSvcWinSetStopPendingStatus(uint32_t uCheckPoint)
306	{
307	vgsvcWinSetStatus(SERVICE_STOP_PENDING, uCheckPoint);
308	}
309
310
311	static RTEXITCODE vgsvcWinSetDesc(SC_HANDLE hService)
312	{
313	/* On W2K+ there's ChangeServiceConfig2() which lets us set some fields
314	like a longer service description. */
315	if (g_pfnChangeServiceConfig2A)
316	{
317	/** @todo On Vista+ SERVICE_DESCRIPTION also supports localized strings! */
318	SERVICE_DESCRIPTION desc;
319	desc.lpDescription = (LPSTR)VBOXSERVICE_DESCRIPTION;
320	if (!g_pfnChangeServiceConfig2A(hService, SERVICE_CONFIG_DESCRIPTION, &desc))
321	{
322	VGSvcError("Cannot set the service description! Error: %ld\n", GetLastError());
323	return RTEXITCODE_FAILURE;
324	}
325	}
326	return RTEXITCODE_SUCCESS;
327	}
328
329
330	/**
331	* Installs the service.
332	*/
333	RTEXITCODE VGSvcWinInstall(void)
334	{
335	VGSvcVerbose(1, "Installing service ...\n");
336
337	TCHAR imagePath[MAX_PATH] = { 0 };
338	GetModuleFileName(NULL, imagePath, sizeof(imagePath));
339
340	SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
341	if (hSCManager == NULL)
342	{
343	VGSvcError("Could not open SCM! Error: %ld\n", GetLastError());
344	return RTEXITCODE_FAILURE;
345	}
346
347	RTEXITCODE rc = RTEXITCODE_SUCCESS;
348	SC_HANDLE hService = CreateService(hSCManager,
349	VBOXSERVICE_NAME, VBOXSERVICE_FRIENDLY_NAME,
350	SERVICE_ALL_ACCESS,
351	SERVICE_WIN32_OWN_PROCESS \| SERVICE_INTERACTIVE_PROCESS,
352	SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL,
353	imagePath, NULL, NULL, NULL, NULL, NULL);
354	if (hService != NULL)
355	VGSvcVerbose(0, "Service successfully installed!\n");
356	else
357	{
358	DWORD dwErr = GetLastError();
359	switch (dwErr)
360	{
361	case ERROR_SERVICE_EXISTS:
362	VGSvcVerbose(1, "Service already exists, just updating the service config.\n");
363	hService = OpenService(hSCManager, VBOXSERVICE_NAME, SERVICE_ALL_ACCESS);
364	if (hService)
365	{
366	if (ChangeServiceConfig(hService,
367	SERVICE_WIN32_OWN_PROCESS \| SERVICE_INTERACTIVE_PROCESS,
368	SERVICE_DEMAND_START,
369	SERVICE_ERROR_NORMAL,
370	imagePath,
371	NULL,
372	NULL,
373	NULL,
374	NULL,
375	NULL,
376	VBOXSERVICE_FRIENDLY_NAME))
377	VGSvcVerbose(1, "The service config has been successfully updated.\n");
378	else
379	rc = VGSvcError("Could not change service config! Error: %ld\n", GetLastError());
380	}
381	else
382	rc = VGSvcError("Could not open service! Error: %ld\n", GetLastError());
383	break;
384
385	default:
386	rc = VGSvcError("Could not create service! Error: %ld\n", dwErr);
387	break;
388	}
389	}
390
391	if (rc == RTEXITCODE_SUCCESS)
392	rc = vgsvcWinSetDesc(hService);
393
394	CloseServiceHandle(hService);
395	CloseServiceHandle(hSCManager);
396	return rc;
397	}
398
399	/**
400	* Uninstalls the service.
401	*/
402	RTEXITCODE VGSvcWinUninstall(void)
403	{
404	VGSvcVerbose(1, "Uninstalling service ...\n");
405
406	SC_HANDLE hSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);
407	if (hSCManager == NULL)
408	{
409	VGSvcError("Could not open SCM! Error: %d\n", GetLastError());
410	return RTEXITCODE_FAILURE;
411	}
412
413	RTEXITCODE rcExit;
414	SC_HANDLE hService = OpenService(hSCManager, VBOXSERVICE_NAME, SERVICE_ALL_ACCESS );
415	if (hService != NULL)
416	{
417	if (DeleteService(hService))
418	{
419	/*
420	* ???
421	*/
422	HKEY hKey = NULL;
423	if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
424	"SYSTEM\\CurrentControlSet\\Services\\EventLog\\System",
425	0,
426	KEY_ALL_ACCESS,
427	&hKey)
428	== ERROR_SUCCESS)
429	{
430	RegDeleteKey(hKey, VBOXSERVICE_NAME);
431	RegCloseKey(hKey);
432	}
433
434	VGSvcVerbose(0, "Service successfully uninstalled!\n");
435	rcExit = RTEXITCODE_SUCCESS;
436	}
437	else
438	rcExit = VGSvcError("Could not remove service! Error: %d\n", GetLastError());
439	CloseServiceHandle(hService);
440	}
441	else
442	rcExit = VGSvcError("Could not open service! Error: %d\n", GetLastError());
443	CloseServiceHandle(hSCManager);
444
445	return rcExit;
446	}
447
448
449	static int vgsvcWinStart(void)
450	{
451	int rc = VINF_SUCCESS;
452
453	/*
454	* Create a well-known SID for the "Builtin Users" group and modify the ACE
455	* for the shared folders miniport redirector DN (whatever DN means).
456	*/
457	PSID pBuiltinUsersSID = NULL;
458	SID_IDENTIFIER_AUTHORITY SIDAuthWorld = SECURITY_LOCAL_SID_AUTHORITY;
459	if (AllocateAndInitializeSid(&SIDAuthWorld, 1, SECURITY_LOCAL_RID, 0, 0, 0, 0, 0, 0, 0, &pBuiltinUsersSID))
460	{
461	rc = vgsvcWinAddAceToObjectsSecurityDescriptor("\\\\.\\VBoxMiniRdrDN", SE_FILE_OBJECT,
462	(LPTSTR)pBuiltinUsersSID, TRUSTEE_IS_SID,
463	FILE_GENERIC_READ \| FILE_GENERIC_WRITE, SET_ACCESS, NO_INHERITANCE);
464	/* If we don't find our "VBoxMiniRdrDN" (for Shared Folders) object above,
465	don't report an error; it just might be not installed. Otherwise this
466	would cause the SCM to hang on starting up the service. */
467	if (rc == VERR_FILE_NOT_FOUND \|\| rc == VERR_PATH_NOT_FOUND)
468	rc = VINF_SUCCESS;
469
470	FreeSid(pBuiltinUsersSID);
471	}
472	else
473	rc = RTErrConvertFromWin32(GetLastError());
474	if (RT_SUCCESS(rc))
475	{
476	/*
477	* Start the service.
478	*/
479	vgsvcWinSetStatus(SERVICE_START_PENDING, 0);
480
481	rc = VGSvcStartServices();
482	if (RT_SUCCESS(rc))
483	{
484	vgsvcWinSetStatus(SERVICE_RUNNING, 0);
485	VGSvcMainWait();
486	}
487	else
488	{
489	vgsvcWinSetStatus(SERVICE_STOPPED, 0);
490	#if 0 /** @todo r=bird: Enable this if SERVICE_CONTROL_STOP isn't triggered automatically */
491	VGSvcStopServices();
492	#endif
493	}
494	}
495	else
496	vgsvcWinSetStatus(SERVICE_STOPPED, 0);
497
498	if (RT_FAILURE(rc))
499	VGSvcError("Service failed to start with rc=%Rrc!\n", rc);
500
501	return rc;
502	}
503
504
505	/**
506	* Call StartServiceCtrlDispatcher.
507	*
508	* The main() thread invokes this when not started in foreground mode. It
509	* won't return till the service is being shutdown (unless start up fails).
510	*
511	* @returns RTEXITCODE_SUCCESS on normal return after service shutdown.
512	* Something else on failure, error will have been reported.
513	*/
514	RTEXITCODE VGSvcWinEnterCtrlDispatcher(void)
515	{
516	if (!StartServiceCtrlDispatcher(&g_aServiceTable[0]))
517	return VGSvcError("StartServiceCtrlDispatcher: %u. Please start %s with option -f (foreground)!\n",
518	GetLastError(), g_pszProgName);
519	return RTEXITCODE_SUCCESS;
520	}
521
522
523	/**
524	* Event code to description.
525	*
526	* @returns String.
527	* @param dwEvent The event code.
528	*/
529	static const char *vgsvcWTSStateToString(DWORD dwEvent)
530	{
531	switch (dwEvent)
532	{
533	case WTS_CONSOLE_CONNECT: return "A session was connected to the console terminal";
534	case WTS_CONSOLE_DISCONNECT: return "A session was disconnected from the console terminal";
535	case WTS_REMOTE_CONNECT: return "A session connected to the remote terminal";
536	case WTS_REMOTE_DISCONNECT: return "A session was disconnected from the remote terminal";
537	case WTS_SESSION_LOGON: return "A user has logged on to a session";
538	case WTS_SESSION_LOGOFF: return "A user has logged off the session";
539	case WTS_SESSION_LOCK: return "A session has been locked";
540	case WTS_SESSION_UNLOCK: return "A session has been unlocked";
541	case WTS_SESSION_REMOTE_CONTROL: return "A session has changed its remote controlled status";
542	#ifdef WTS_SESSION_CREATE
543	case WTS_SESSION_CREATE: return "A session has been created";
544	#endif
545	#ifdef WTS_SESSION_TERMINATE
546	case WTS_SESSION_TERMINATE: return "The session has been terminated";
547	#endif
548	default: return "Uknonwn state";
549	}
550	}
551
552
553	/**
554	* Common control handler.
555	*
556	* @returns Return code for NT5+.
557	* @param dwControl The control code.
558	*/
559	static DWORD vgsvcWinCtrlHandlerCommon(DWORD dwControl)
560	{
561	DWORD rcRet = NO_ERROR;
562	switch (dwControl)
563	{
564	case SERVICE_CONTROL_INTERROGATE:
565	vgsvcWinSetStatus(g_dwWinServiceLastStatus, 0);
566	break;
567
568	case SERVICE_CONTROL_STOP:
569	case SERVICE_CONTROL_SHUTDOWN:
570	{
571	vgsvcWinSetStatus(SERVICE_STOP_PENDING, 0);
572
573	int rc2 = VGSvcStopServices();
574	if (RT_FAILURE(rc2))
575	rcRet = ERROR_GEN_FAILURE;
576	else
577	{
578	rc2 = VGSvcReportStatus(VBoxGuestFacilityStatus_Terminated);
579	AssertRC(rc2);
580	}
581
582	vgsvcWinSetStatus(SERVICE_STOPPED, 0);
583	break;
584	}
585
586	default:
587	VGSvcVerbose(1, "Control handler: Function not implemented: %#x\n", dwControl);
588	rcRet = ERROR_CALL_NOT_IMPLEMENTED;
589	break;
590	}
591
592	return rcRet;
593	}
594
595
596	/**
597	* Callback registered by RegisterServiceCtrlHandler on NT4 and earlier.
598	*/
599	static VOID WINAPI vgsvcWinCtrlHandlerNt4(DWORD dwControl) RT_NOTHROW_DEF
600	{
601	VGSvcVerbose(2, "Control handler (NT4): dwControl=%#x\n", dwControl);
602	vgsvcWinCtrlHandlerCommon(dwControl);
603	}
604
605
606	/**
607	* Callback registered by RegisterServiceCtrlHandler on NT5 and later.
608	*/
609	static DWORD WINAPI
610	vgsvcWinCtrlHandlerNt5Plus(DWORD dwControl, DWORD dwEventType, LPVOID lpEventData, LPVOID lpContext) RT_NOTHROW_DEF
611	{
612	VGSvcVerbose(2, "Control handler: dwControl=%#x, dwEventType=%#x\n", dwControl, dwEventType);
613	RT_NOREF1(lpContext);
614
615	switch (dwControl)
616	{
617	default:
618	return vgsvcWinCtrlHandlerCommon(dwControl);
619
620	case SERVICE_CONTROL_SESSIONCHANGE: /* Only Windows 2000 and up. */
621	{
622	AssertPtr(lpEventData);
623	PWTSSESSION_NOTIFICATION pNotify = (PWTSSESSION_NOTIFICATION)lpEventData;
624	Assert(pNotify->cbSize == sizeof(WTSSESSION_NOTIFICATION));
625
626	VGSvcVerbose(1, "Control handler: %s (Session=%ld, Event=%#x)\n",
627	vgsvcWTSStateToString(dwEventType), pNotify->dwSessionId, dwEventType);
628
629	/* Handle all events, regardless of dwEventType. */
630	int rc2 = VGSvcVMInfoSignal();
631	AssertRC(rc2);
632
633	return NO_ERROR;
634	}
635	}
636	}
637
638
639	static void WINAPI vgsvcWinMain(DWORD argc, LPTSTR *argv)
640	{
641	RT_NOREF2(argc, argv);
642	VGSvcVerbose(2, "Registering service control handler ...\n");
643	if (g_pfnRegisterServiceCtrlHandlerExA)
644	g_hWinServiceStatus = g_pfnRegisterServiceCtrlHandlerExA(VBOXSERVICE_NAME, vgsvcWinCtrlHandlerNt5Plus, NULL);
645	else
646	g_hWinServiceStatus = RegisterServiceCtrlHandlerA(VBOXSERVICE_NAME, vgsvcWinCtrlHandlerNt4);
647	if (g_hWinServiceStatus != NULL)
648	{
649	VGSvcVerbose(2, "Service control handler registered.\n");
650	vgsvcWinStart();
651	}
652	else
653	{
654	DWORD dwErr = GetLastError();
655	switch (dwErr)
656	{
657	case ERROR_INVALID_NAME:
658	VGSvcError("Invalid service name!\n");
659	break;
660	case ERROR_SERVICE_DOES_NOT_EXIST:
661	VGSvcError("Service does not exist!\n");
662	break;
663	default:
664	VGSvcError("Could not register service control handle! Error: %ld\n", dwErr);
665	break;
666	}
667	}
668	}
669

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

source: vbox/trunk/src/VBox/Additions/common/VBoxService/VBoxService-win.cpp@ 99828

Download in other formats: