process-win.cpp@ 32657

Last change on this file since 32657 was 32657, checked in by vboxsync, 14 years ago
IPRT/process-win: Added fallback for environment block creation (Userenv-API / Profile environment block creation).
Property svn:eol-style set to `native` Property svn:keywords set to `Id`
File size: 47.9 KB

Line
1	/* $Id: process-win.cpp 32657 2010-09-21 08:53:52Z vboxsync $ */
2	/** @file
3	* IPRT - Process, Windows.
4	*/
5
6	/*
7	* Copyright (C) 2006-2010 Oracle Corporation
8	*
9	* This file is part of VirtualBox Open Source Edition (OSE), as
10	* available from http://www.virtualbox.org. This file is free software;
11	* you can redistribute it and/or modify it under the terms of the GNU
12	* General Public License (GPL) as published by the Free Software
13	* Foundation, in version 2 as it comes in the "COPYING" file of the
14	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16	*
17	* The contents of this file may alternatively be used under the terms
18	* of the Common Development and Distribution License Version 1.0
19	* (CDDL) only, as it comes in the "COPYING.CDDL" file of the
20	* VirtualBox OSE distribution, in which case the provisions of the
21	* CDDL are applicable instead of those of the GPL.
22	*
23	* You may elect to license modified versions of this file under the
24	* terms and conditions of either the GPL or the CDDL or both.
25	*/
26
27
28	/*******************************************************************************
29	* Header Files *
30	*******************************************************************************/
31	#define LOG_GROUP RTLOGGROUP_PROCESS
32
33	#include <Userenv.h>
34	#include <Windows.h>
35	#include <tlhelp32.h>
36	#include <process.h>
37	#include <errno.h>
38	#include <Strsafe.h>
39
40	#include <iprt/process.h>
41	#include "internal/iprt.h"
42
43	#include <iprt/assert.h>
44	#include <iprt/critsect.h>
45	#include <iprt/file.h>
46	#include <iprt/err.h>
47	#include <iprt/env.h>
48	#include <iprt/getopt.h>
49	#include <iprt/initterm.h>
50	#include <iprt/ldr.h>
51	#include <iprt/mem.h>
52	#include <iprt/once.h>
53	#include <iprt/pipe.h>
54	#include <iprt/string.h>
55	#include <iprt/socket.h>
56
57
58	/*******************************************************************************
59	* Structures and Typedefs *
60	*******************************************************************************/
61	typedef WINADVAPI BOOL WINAPI FNCREATEPROCESSWITHLOGON(LPCWSTR,
62	LPCWSTR,
63	LPCWSTR,
64	DWORD,
65	LPCWSTR,
66	LPWSTR,
67	DWORD,
68	LPVOID,
69	LPCWSTR,
70	LPSTARTUPINFOW,
71	LPPROCESS_INFORMATION);
72	typedef FNCREATEPROCESSWITHLOGON *PFNCREATEPROCESSWITHLOGON;
73
74	typedef DWORD WINAPI FNWTSGETACTIVECONSOLESESSIONID();
75	typedef FNWTSGETACTIVECONSOLESESSIONID *PFNWTSGETACTIVECONSOLESESSIONID;
76
77	typedef HANDLE WINAPI FNCREATETOOLHELP32SNAPSHOT(DWORD, DWORD);
78	typedef FNCREATETOOLHELP32SNAPSHOT *PFNCREATETOOLHELP32SNAPSHOT;
79
80	typedef BOOL WINAPI FNPROCESS32FIRST(HANDLE, LPPROCESSENTRY32);
81	typedef FNPROCESS32FIRST *PFNPROCESS32FIRST;
82
83	typedef BOOL WINAPI FNPROCESS32NEXT(HANDLE, LPPROCESSENTRY32);
84	typedef FNPROCESS32NEXT *PFNPROCESS32NEXT;
85
86	typedef BOOL WINAPI FNENUMPROCESSES(DWORD, DWORD, DWORD);
87	typedef FNENUMPROCESSES *PFNENUMPROCESSES;
88
89	typedef DWORD FNGETMODULEBASENAME(HANDLE, HMODULE, LPTSTR, DWORD);
90	typedef FNGETMODULEBASENAME *PFNGETMODULEBASENAME;
91
92	typedef BOOL WINAPI FNCREATEENVIRONMENTBLOCK(LPVOID *, HANDLE, BOOL);
93	typedef FNCREATEENVIRONMENTBLOCK *PFNCREATEENVIRONMENTBLOCK;
94
95	typedef BOOL WINAPI FNPFNDESTROYENVIRONMENTBLOCK(LPVOID);
96	typedef FNPFNDESTROYENVIRONMENTBLOCK *PFNPFNDESTROYENVIRONMENTBLOCK;
97
98	typedef BOOL WINAPI FNLOADUSERPROFILEW(HANDLE, LPPROFILEINFOW);
99	typedef FNLOADUSERPROFILEW *PFNLOADUSERPROFILEW;
100
101	typedef BOOL WINAPI FNUNLOADUSERPROFILE(HANDLE, HANDLE);
102	typedef FNUNLOADUSERPROFILE *PFNUNLOADUSERPROFILE;
103
104
105	/*******************************************************************************
106	* Global Variables *
107	*******************************************************************************/
108	/** Init once structure. */
109	static RTONCE g_rtProcWinInitOnce = RTONCE_INITIALIZER;
110	/** Critical section protecting the process array. */
111	static RTCRITSECT g_CritSect;
112	/** The number of processes in the array. */
113	static uint32_t g_cProcesses;
114	/** The current allocation size. */
115	static uint32_t g_cProcessesAlloc;
116	/** Array containing the live or non-reaped child processes. */
117	static struct RTPROCWINENTRY
118	{
119	/** The process ID. */
120	ULONG_PTR pid;
121	/** The process handle. */
122	HANDLE hProcess;
123	} *g_paProcesses;
124
125
126	/**
127	* Clean up the globals.
128	*
129	* @param enmReason Ignored.
130	* @param iStatus Ignored.
131	* @param pvUser Ignored.
132	*/
133	static DECLCALLBACK(void) rtProcWinTerm(RTTERMREASON enmReason, int32_t iStatus, void *pvUser)
134	{
135	NOREF(pvUser); NOREF(iStatus); NOREF(enmReason);
136
137	RTCritSectDelete(&g_CritSect);
138
139	size_t i = g_cProcesses;
140	while (i-- > 0)
141	{
142	CloseHandle(g_paProcesses[i].hProcess);
143	g_paProcesses[i].hProcess = NULL;
144	}
145	RTMemFree(g_paProcesses);
146
147	g_paProcesses = NULL;
148	g_cProcesses = 0;
149	g_cProcessesAlloc = 0;
150	}
151
152
153	/**
154	* Initialize the globals.
155	*
156	* @returns IPRT status code.
157	* @param pvUser1 Ignored.
158	* @param pvUser2 Ignored.
159	*/
160	static DECLCALLBACK(int32_t) rtProcWinInitOnce(void pvUser1, void pvUser2)
161	{
162	NOREF(pvUser1); NOREF(pvUser2);
163
164	g_cProcesses = 0;
165	g_cProcessesAlloc = 0;
166	g_paProcesses = NULL;
167	int rc = RTCritSectInit(&g_CritSect);
168	if (RT_SUCCESS(rc))
169	{
170	/** @todo init once, terminate once - this is a generic thing which should
171	* have some kind of static and simpler setup! */
172	rc = RTTermRegisterCallback(rtProcWinTerm, NULL);
173	if (RT_SUCCESS(rc))
174	return rc;
175	RTCritSectDelete(&g_CritSect);
176	}
177	return rc;
178	}
179
180
181	/**
182	* Gets the process handle for a process from g_paProcesses.
183	*
184	* @returns Process handle if found, NULL if not.
185	* @param pid The process to remove (pid).
186	*/
187	static HANDLE rtProcWinFindPid(RTPROCESS pid)
188	{
189	HANDLE hProcess = NULL;
190
191	RTCritSectEnter(&g_CritSect);
192	uint32_t i = g_cProcesses;
193	while (i-- > 0)
194	if (g_paProcesses[i].pid == pid)
195	{
196	hProcess = g_paProcesses[i].hProcess;
197	break;
198	}
199	RTCritSectLeave(&g_CritSect);
200
201	return hProcess;
202	}
203
204
205	/**
206	* Removes a process from g_paProcesses.
207	*
208	* @param pid The process to remove (pid).
209	*/
210	static void rtProcWinRemovePid(RTPROCESS pid)
211	{
212	RTCritSectEnter(&g_CritSect);
213	uint32_t i = g_cProcesses;
214	while (i-- > 0)
215	if (g_paProcesses[i].pid == pid)
216	{
217	g_cProcesses--;
218	uint32_t cToMove = g_cProcesses - i;
219	if (cToMove)
220	memmove(&g_paProcesses[i], &g_paProcesses[i + 1], cToMove * sizeof(g_paProcesses[0]));
221	break;
222	}
223	RTCritSectLeave(&g_CritSect);
224	}
225
226
227	/**
228	* Adds a process to g_paProcesses.
229	*
230	* @returns IPRT status code.
231	* @param pid The process id.
232	* @param hProcess The process handle.
233	*/
234	static int rtProcWinAddPid(RTPROCESS pid, HANDLE hProcess)
235	{
236	RTCritSectEnter(&g_CritSect);
237
238	uint32_t i = g_cProcesses;
239	if (i >= g_cProcessesAlloc)
240	{
241	void pvNew = RTMemRealloc(g_paProcesses, (i + 16) sizeof(g_paProcesses[0]));
242	if (RT_UNLIKELY(!pvNew))
243	{
244	RTCritSectLeave(&g_CritSect);
245	return VERR_NO_MEMORY;
246	}
247	g_paProcesses = (struct RTPROCWINENTRY *)pvNew;
248	g_cProcessesAlloc = i + 16;
249	}
250
251	g_paProcesses[i].pid = pid;
252	g_paProcesses[i].hProcess = hProcess;
253	g_cProcesses = i + 1;
254
255	RTCritSectLeave(&g_CritSect);
256	return VINF_SUCCESS;
257	}
258
259
260	RTR3DECL(int) RTProcCreate(const char pszExec, const char const *papszArgs, RTENV Env, unsigned fFlags, PRTPROCESS pProcess)
261	{
262	return RTProcCreateEx(pszExec, papszArgs, Env, fFlags,
263	NULL, NULL, NULL, /* standard handles */
264	NULL /pszAsUser/, NULL /* pszPassword*/,
265	pProcess);
266	}
267
268
269	/**
270	* Get the process token (not the process handle like the name might indicate)
271	* of the process indicated by @a dwPID if the @a pSID matches.
272	*
273	* @returns IPRT status code.
274	*
275	* @param dwPID The process identifier.
276	* @param pSID The secure identifier of the user.
277	* @param phToken Where to return the token handle - duplicate,
278	* caller closes it on success.
279	*/
280	static int rtProcGetProcessHandle(DWORD dwPID, PSID pSID, PHANDLE phToken)
281	{
282	AssertPtr(pSID);
283	AssertPtr(phToken);
284
285	DWORD dwErr;
286	BOOL fRc;
287	bool fFound = false;
288	HANDLE hProc = OpenProcess(MAXIMUM_ALLOWED, TRUE, dwPID);
289	if (hProc != NULL)
290	{
291	HANDLE hTokenProc;
292	fRc = OpenProcessToken(hProc,
293	TOKEN_ADJUST_PRIVILEGES \| TOKEN_QUERY \| TOKEN_DUPLICATE
294	\| TOKEN_ASSIGN_PRIMARY \| TOKEN_ADJUST_SESSIONID \| TOKEN_READ \| TOKEN_WRITE,
295	&hTokenProc);
296	if (fRc)
297	{
298	DWORD dwSize = 0;
299	fRc = GetTokenInformation(hTokenProc, TokenUser, NULL, 0, &dwSize);
300	if (!fRc)
301	dwErr = GetLastError();
302	if ( !fRc
303	&& dwErr == ERROR_INSUFFICIENT_BUFFER
304	&& dwSize > 0)
305	{
306	PTOKEN_USER pTokenUser = (PTOKEN_USER)RTMemAlloc(dwSize);
307	AssertPtrReturn(pTokenUser, VERR_NO_MEMORY);
308	RT_ZERO(*pTokenUser);
309	if ( GetTokenInformation(hTokenProc,
310	TokenUser,
311	(LPVOID)pTokenUser,
312	dwSize,
313	&dwSize))
314	{
315	if ( IsValidSid(pTokenUser->User.Sid)
316	&& EqualSid(pTokenUser->User.Sid, pSID))
317	{
318	if (DuplicateTokenEx(hTokenProc, MAXIMUM_ALLOWED,
319	NULL, SecurityIdentification, TokenPrimary, phToken))
320	{
321	/*
322	* So we found the process instance which belongs to the user we want to
323	* to run our new process under. This duplicated token will be used for
324	* the actual CreateProcessAsUserW() call then.
325	*/
326	fFound = true;
327	}
328	else
329	dwErr = GetLastError();
330	}
331	}
332	else
333	dwErr = GetLastError();
334	RTMemFree(pTokenUser);
335	}
336	else
337	dwErr = GetLastError();
338	CloseHandle(hTokenProc);
339	}
340	else
341	dwErr = GetLastError();
342	CloseHandle(hProc);
343	}
344	else
345	dwErr = GetLastError();
346	if (fFound)
347	return VINF_SUCCESS;
348	if (dwErr != NO_ERROR)
349	return RTErrConvertFromWin32(dwErr);
350	return VERR_NOT_FOUND; /* No error occured, but we didn't find the right process. */
351	}
352
353
354	/**
355	* Finds a one of the processes in @a papszNames running with user @a pSID and
356	* returns a duplicate handle to its token.
357	*
358	* @returns Success indicator.
359	* @param papszNames The process candidates, in prioritized order.
360	* @param pSID The secure identifier of the user.
361	* @param phToken Where to return the token handle - duplicate,
362	* caller closes it on success.
363	*/
364	static bool rtProcFindProcessByName(const char * const *papszNames, PSID pSID, PHANDLE phToken)
365	{
366	AssertPtr(papszNames);
367	AssertPtr(pSID);
368	AssertPtr(phToken);
369
370	DWORD dwErr = NO_ERROR;
371	bool fFound = false;
372
373	/*
374	* On modern systems (W2K+) try the Toolhelp32 API first; this is more stable
375	* and reliable. Fallback to EnumProcess on NT4.
376	*/
377	RTLDRMOD hKernel32;
378	int rc = RTLdrLoad("Kernel32.dll", &hKernel32);
379	if (RT_SUCCESS(rc))
380	{
381	PFNCREATETOOLHELP32SNAPSHOT pfnCreateToolhelp32Snapshot;
382	rc = RTLdrGetSymbol(hKernel32, "CreateToolhelp32Snapshot", (void**)&pfnCreateToolhelp32Snapshot);
383	if (RT_SUCCESS(rc))
384	{
385	PFNPROCESS32FIRST pfnProcess32First;
386	rc = RTLdrGetSymbol(hKernel32, "Process32First", (void**)&pfnProcess32First);
387	if (RT_SUCCESS(rc))
388	{
389	PFNPROCESS32NEXT pfnProcess32Next;
390	rc = RTLdrGetSymbol(hKernel32, "Process32Next", (void**)&pfnProcess32Next);
391	if (RT_SUCCESS(rc))
392	{
393	HANDLE hSnap = pfnCreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
394	if (hSnap != INVALID_HANDLE_VALUE)
395	{
396	for (size_t i = 0; papszNames[i] && !fFound; i++)
397	{
398	PROCESSENTRY32 procEntry;
399	procEntry.dwSize = sizeof(PROCESSENTRY32);
400	if (pfnProcess32First(hSnap, &procEntry))
401	{
402	do
403	{
404	if ( _stricmp(procEntry.szExeFile, papszNames[i]) == 0
405	&& RT_SUCCESS(rtProcGetProcessHandle(procEntry.th32ProcessID, pSID, phToken)))
406	{
407	fFound = true;
408	break;
409	}
410	} while (pfnProcess32Next(hSnap, &procEntry));
411	}
412	else /* Process32First */
413	dwErr = GetLastError();
414	if (FAILED(dwErr))
415	break;
416	}
417	CloseHandle(hSnap);
418	}
419	else /* hSnap == INVALID_HANDLE_VALUE */
420	dwErr = GetLastError();
421	}
422	}
423	}
424	else /* CreateToolhelp32Snapshot / Toolhelp32 API not available. */
425	{
426	/*
427	* NT4 needs a copy of "PSAPI.dll" (redistributed by Microsoft and not
428	* part of the OS) in order to get a lookup. If we don't have this DLL
429	* we are not able to get a token and therefore no UI will be visible.
430	*/
431	RTLDRMOD hPSAPI;
432	int rc = RTLdrLoad("PSAPI.dll", &hPSAPI);
433	if (RT_SUCCESS(rc))
434	{
435	PFNENUMPROCESSES pfnEnumProcesses;
436	rc = RTLdrGetSymbol(hPSAPI, "EnumProcesses", (void**)&pfnEnumProcesses);
437	if (RT_SUCCESS(rc))
438	{
439	PFNGETMODULEBASENAME pfnGetModuleBaseName;
440	rc = RTLdrGetSymbol(hPSAPI, "GetModuleBaseName", (void**)&pfnGetModuleBaseName);
441	if (RT_SUCCESS(rc))
442	{
443	/** @todo Retry if pBytesReturned equals cbBytes! */
444	DWORD adwPIDs[4096]; /* Should be sufficient for now. */
445	DWORD cbBytes = 0;
446	if (pfnEnumProcesses(adwPIDs, sizeof(adwPIDs), &cbBytes))
447	{
448	for (size_t i = 0; papszNames[i] && !fFound; i++)
449	{
450	for (DWORD dwIdx = 0; dwIdx < cbBytes/sizeof(DWORD) && !fFound; dwIdx++)
451	{
452	HANDLE hProc = OpenProcess(PROCESS_QUERY_INFORMATION \| PROCESS_VM_READ,
453	FALSE, adwPIDs[dwIdx]);
454	if (hProc)
455	{
456	char *pszProcName = NULL;
457	DWORD dwSize = 128;
458	do
459	{
460	RTMemRealloc(pszProcName, dwSize);
461	if (pfnGetModuleBaseName(hProc, 0, pszProcName, dwSize) == dwSize)
462	dwSize += 128;
463	if (dwSize > _4K) /* Play safe. */
464	break;
465	} while (GetLastError() == ERROR_INSUFFICIENT_BUFFER);
466
467	if (pszProcName)
468	{
469	if ( _stricmp(pszProcName, papszNames[i]) == 0
470	&& RT_SUCCESS(rtProcGetProcessHandle(adwPIDs[dwIdx], pSID, phToken)))
471	{
472	fFound = true;
473	}
474	}
475	if (pszProcName)
476	RTStrFree(pszProcName);
477	CloseHandle(hProc);
478	}
479	}
480	}
481	}
482	else
483	dwErr = GetLastError();
484	}
485	}
486	}
487	}
488	RTLdrClose(hKernel32);
489	}
490	Assert(dwErr == NO_ERROR);
491	return fFound;
492	}
493
494
495	/**
496	* Logs on a specified user and returns its primary token.
497	*
498	* @return int
499	*
500	* @param pwszUser User name.
501	* @param pwszPassword Password.
502	* @param pwszDomain Domain (not used at the moment).
503	* @param phToken Pointer to store the logon token.
504	*/
505	static int rtProcUserLogon(PRTUTF16 pwszUser, PRTUTF16 pwszPassword, PRTUTF16 pwszDomain, HANDLE *phToken)
506	{
507	int rc = VINF_SUCCESS;
508	/** @todo Add domain support! */
509	BOOL fRc = LogonUserW(pwszUser,
510	/*
511	* Because we have to deal with http://support.microsoft.com/kb/245683
512	* for NULL domain names when running on NT4 here, pass an empty string if so.
513	* However, passing FQDNs should work!
514	*/
515	((DWORD)(LOBYTE(LOWORD(GetVersion()))) < 5) /* < Windows 2000. */
516	? L"" /* NT4 and older. */
517	: NULL, /* Windows 2000 and up. */
518	pwszPassword,
519	LOGON32_LOGON_INTERACTIVE,
520	LOGON32_PROVIDER_DEFAULT,
521	phToken);
522	if (!fRc)
523	rc = RTErrConvertFromWin32(GetLastError());
524	return rc;
525	}
526
527
528	/**
529	* Logs off a user, specified by the given tokken.
530	*
531	* @param hToken The token (=user) to log off.
532	*/
533	static void rtProcUserLogoff(HANDLE hToken)
534	{
535	CloseHandle(hToken);
536	}
537
538
539	/**
540	* Creates an environment block out of a handed in Unicode and RTENV block.
541	* The RTENV block can overwrite entries already present in the Unicode block.
542	*
543	* @return IPRT status code.
544	*
545	* @param pvBlock Unicode block (array) of environment entries; name=value
546	* @param hEnv Handle of an existing RTENV block to use.
547	* @param ppwszBlock Pointer to the final output.
548	*/
549	static int rtProcEnvironmentCreateInternal(VOID pvBlock, RTENV hEnv, PRTUTF16 ppwszBlock)
550	{
551	int rc = VINF_SUCCESS;
552	RTENV hEnvTemp;
553	rc = RTEnvClone(&hEnvTemp, hEnv);
554	if (RT_SUCCESS(rc))
555	{
556	PRTUTF16 pBlock = (PRTUTF16)pvBlock;
557	while ( pBlock
558	&& pBlock != '\0'
559	&& RT_SUCCESS(rc))
560	{
561	char *pszEntry;
562	rc = RTUtf16ToUtf8(pBlock, &pszEntry);
563	if (RT_SUCCESS(rc))
564	{
565	/* Don't overwrite values which we already have set to a custom value
566	* specified in hEnv ... */
567	if (!RTEnvExistEx(hEnv, pszEntry))
568	rc = RTEnvPutEx(hEnvTemp, pszEntry);
569	RTStrFree(pszEntry);
570	}
571
572	size_t l;
573	/* 32k should be the maximum the environment block can have on Windows. */
574	if (FAILED(StringCbLengthW((LPCWSTR)pBlock, _32K * sizeof(RTUTF16), &l)))
575	break;
576	pBlock += l / sizeof(RTUTF16);
577	if (pBlock[1] == '\0') /* Did we reach the double zero termination (\0\0)? */
578	break;
579	pBlock++; /* Skip zero termination of current string and advance to next string ... */
580	}
581
582	if (RT_SUCCESS(rc))
583	rc = RTEnvQueryUtf16Block(hEnvTemp, ppwszBlock);
584	RTEnvDestroy(hEnvTemp);
585	}
586	return rc;
587	}
588
589
590	/**
591	* Builds up the environment block for a specified user (identified by a token),
592	* whereas hEnv is an additional set of environment variables which overwrite existing
593	* values of the user profile. ppwszBlock needs to be destroyed after usage
594	* calling rtProcEnvironmentDestroy().
595	*
596	* @return IPRT status code.
597	*
598	* @param hToken Token of the user to use.
599	* @param hEnv Own environment block to extend/overwrite the profile's data with.
600	* @param ppwszBlock Pointer to a pointer of the final UTF16 environment block.
601	*/
602	static int rtProcEnvironmentCreateFromToken(HANDLE hToken, RTENV hEnv, PRTUTF16 *ppwszBlock)
603	{
604	RTLDRMOD hUserenv;
605	int rc = RTLdrLoad("Userenv.dll", &hUserenv);
606	if (RT_SUCCESS(rc))
607	{
608	PFNCREATEENVIRONMENTBLOCK pfnCreateEnvironmentBlock;
609	rc = RTLdrGetSymbol(hUserenv, "CreateEnvironmentBlock", (void**)&pfnCreateEnvironmentBlock);
610	if (RT_SUCCESS(rc))
611	{
612	PFNPFNDESTROYENVIRONMENTBLOCK pfnDestroyEnvironmentBlock;
613	rc = RTLdrGetSymbol(hUserenv, "DestroyEnvironmentBlock", (void**)&pfnDestroyEnvironmentBlock);
614	if (RT_SUCCESS(rc))
615	{
616	LPVOID pEnvBlockProfile = NULL;
617	if (pfnCreateEnvironmentBlock(&pEnvBlockProfile, hToken, FALSE /* Don't inherit from parent. */))
618	{
619	rc = rtProcEnvironmentCreateInternal(pEnvBlockProfile, hEnv, ppwszBlock);
620	pfnDestroyEnvironmentBlock(pEnvBlockProfile);
621	}
622	else
623	rc = RTErrConvertFromWin32(GetLastError());
624	}
625	}
626	RTLdrClose(hUserenv);
627	}
628	/* If we don't have the Userenv-API for whatever reason or something with the
629	* native environment block failed, try to return at least our own environment block. */
630	if (RT_FAILURE(rc))
631	rc = RTEnvQueryUtf16Block(hEnv, ppwszBlock);
632	return rc;
633	}
634
635
636	/**
637	* Builds up the environment block for a specified user (identified by user name, password
638	* and domain), whereas hEnv is an additional set of environment variables which overwrite
639	* existing values of the user profile. ppwszBlock needs to be destroyed after usage
640	* calling rtProcEnvironmentDestroy().
641	*
642	* @return IPRT status code.
643	*
644	* @param pwszUser User name.
645	* @param pwszPassword Password.
646	* @param pwszDomain Domain.
647	* @param hEnv Own environment block to extend/overwrite the profile's data with.
648	* @param ppwszBlock Pointer to a pointer of the final UTF16 environment block.
649	*/
650	static int rtProcEnvironmentCreateFromAccount(PRTUTF16 pwszUser, PRTUTF16 pwszPassword, PRTUTF16 pwszDomain,
651	RTENV hEnv, PRTUTF16 *ppwszBlock)
652	{
653	HANDLE hToken;
654	int rc = rtProcUserLogon(pwszUser, pwszPassword, pwszDomain, &hToken);
655	if (RT_SUCCESS(rc))
656	{
657	rc = rtProcEnvironmentCreateFromToken(hToken, hEnv, ppwszBlock);
658	rtProcUserLogoff(hToken);
659	}
660	return rc;
661	}
662
663
664	/**
665	* Destroys an environment block formerly created by rtProcEnvironmentCreateInternal(),
666	* rtProcEnvironmentCreateFromToken() or rtProcEnvironmentCreateFromAccount().
667	*
668	* @param ppwszBlock Environment block to destroy.
669	*/
670	static void rtProcEnvironmentDestroy(PRTUTF16 ppwszBlock)
671	{
672	RTEnvFreeUtf16Block(ppwszBlock);
673	}
674
675
676	static int rtProcCreateAsUserHlp(PRTUTF16 pwszUser, PRTUTF16 pwszPassword, PRTUTF16 pwszExec, PRTUTF16 pwszCmdLine,
677	RTENV hEnv, DWORD dwCreationFlags,
678	STARTUPINFOW pStartupInfo, PROCESS_INFORMATION pProcInfo, uint32_t fFlags)
679	{
680	int rc = VINF_SUCCESS;
681	BOOL fRc = FALSE;
682	DWORD dwErr = NO_ERROR;
683
684	/*
685	* If we run as a service CreateProcessWithLogon will fail,
686	* so don't even try it (because of Local System context).
687	*/
688	if (!(fFlags & RTPROC_FLAGS_SERVICE))
689	{
690	RTLDRMOD hAdvAPI32;
691	rc = RTLdrLoad("Advapi32.dll", &hAdvAPI32);
692	if (RT_SUCCESS(rc))
693	{
694	/*
695	* This may fail on too old (NT4) platforms or if the calling process
696	* is running on a SYSTEM account (like a service, ERROR_ACCESS_DENIED) on newer
697	* platforms (however, this works on W2K!).
698	*/
699	PFNCREATEPROCESSWITHLOGON pfnCreateProcessWithLogonW;
700	rc = RTLdrGetSymbol(hAdvAPI32, "CreateProcessWithLogonW", (void**)&pfnCreateProcessWithLogonW);
701	if (RT_SUCCESS(rc))
702	{
703	PRTUTF16 pwszzBlock;
704	rc = rtProcEnvironmentCreateFromAccount(pwszUser, pwszPassword, NULL /* Domain */,
705	hEnv, &pwszzBlock);
706	if (RT_SUCCESS(rc))
707	{
708	fRc = pfnCreateProcessWithLogonW(pwszUser,
709	NULL, /* lpDomain*/
710	pwszPassword,
711	1 /LOGON_WITH_PROFILE/, /* dwLogonFlags */
712	pwszExec,
713	pwszCmdLine,
714	dwCreationFlags,
715	pwszzBlock,
716	NULL, /* pCurrentDirectory */
717	pStartupInfo,
718	pProcInfo);
719	if (!fRc)
720	dwErr = GetLastError();
721	rtProcEnvironmentDestroy(pwszzBlock);
722	}
723	}
724	RTLdrClose(hAdvAPI32);
725	}
726	}
727
728	/*
729	* Did the API call above fail because we're running on a too old OS (NT4) or
730	* we're running as a Windows service?
731	*/
732	if ( RT_FAILURE(rc)
733	\|\| (fFlags & RTPROC_FLAGS_SERVICE))
734	{
735	/*
736	* So if we want to start a process from a service (RTPROC_FLAGS_SERVICE),
737	* we have to do the following:
738	* - Check the credentials supplied and get the user SID.
739	* - If valid get the correct Explorer/VBoxTray instance corresponding to that
740	* user. This of course is only possible if that user is logged in (over
741	* physical console or terminal services).
742	* - If we found the user's Explorer/VBoxTray app, use and modify the token to
743	* use it in order to allow the newly started process acess the user's
744	* desktop. If there's no Explorer/VBoxTray app we cannot display the started
745	* process (but run it without UI).
746	*
747	* The following restrictions apply:
748	* - A process only can show its UI when the user the process should run
749	* under is logged in (has a desktop).
750	* - We do not want to display a process of user A run on the desktop
751	* of user B on multi session systems.
752	*
753	* The following rights are needed in order to use LogonUserW and
754	* CreateProcessAsUserW, so the local policy has to be modified to:
755	* - SE_TCB_NAME = Act as part of the operating system
756	* - SE_ASSIGNPRIMARYTOKEN_NAME = Create/replace a token object
757	* - SE_INCREASE_QUOTA_NAME
758	*
759	* We may fail here with ERROR_PRIVILEGE_NOT_HELD.
760	*/
761	PHANDLE phToken = NULL;
762	HANDLE hTokenLogon = INVALID_HANDLE_VALUE;
763	rc = rtProcUserLogon(pwszUser, pwszPassword, NULL /* Domain */, &hTokenLogon);
764	if (RT_SUCCESS(rc))
765	{
766	bool fFound = false;
767	HANDLE hTokenUserDesktop = INVALID_HANDLE_VALUE;
768
769	if (fFlags & RTPROC_FLAGS_SERVICE)
770	{
771	DWORD cbName = 0; /* Must be zero to query size! */
772	DWORD cbDomain = 0;
773	SID_NAME_USE sidNameUse = SidTypeUser;
774	fRc = LookupAccountNameW(NULL,
775	pwszUser,
776	NULL,
777	&cbName,
778	NULL,
779	&cbDomain,
780	&sidNameUse);
781	if (!fRc)
782	dwErr = GetLastError();
783	if ( !fRc
784	&& dwErr == ERROR_INSUFFICIENT_BUFFER
785	&& cbName > 0)
786	{
787	dwErr = NO_ERROR;
788
789	PSID pSID = (PSID)RTMemAlloc(cbName * sizeof(wchar_t));
790	AssertPtrReturn(pSID, VERR_NO_MEMORY);
791
792	/** @todo No way to allocate a PRTUTF16 directly? */
793	PRTUTF16 pwszDomain = NULL;
794	if (cbDomain > 0)
795	{
796	pwszDomain = (PRTUTF16)RTMemAlloc(cbDomain * sizeof(RTUTF16));
797	AssertPtrReturn(pwszDomain, VERR_NO_MEMORY);
798	}
799
800	/* Note: Also supports FQDNs! */
801	if ( LookupAccountNameW(NULL, /* lpSystemName */
802	pwszUser,
803	pSID,
804	&cbName,
805	pwszDomain,
806	&cbDomain,
807	&sidNameUse)
808	&& IsValidSid(pSID))
809	{
810	/* Array of process names we want to look for. */
811	static const char * const s_papszProcNames[] =
812	{
813	#ifdef VBOX /* The explorer entry is a fallback in case GA aren't installed. */
814	{ "VBoxTray.exe" },
815	#endif
816	{ "explorer.exe" },
817	NULL
818	};
819	fFound = rtProcFindProcessByName(s_papszProcNames, pSID, &hTokenUserDesktop);
820	}
821	else
822	dwErr = GetLastError(); /* LookupAccountNameW() failed. */
823	RTMemFree(pSID);
824	if (pwszDomain != NULL)
825	RTUtf16Free(pwszDomain);
826	}
827	}
828	else /* !RTPROC_FLAGS_SERVICE */
829	{
830	/* Nothing to do here right now. */
831	}
832
833	/*
834	* If we didn't find a matching VBoxTray, just use the token we got
835	* above from LogonUserW(). This enables us to at least run processes with
836	* desktop interaction without UI.
837	*/
838	phToken = fFound ? &hTokenUserDesktop : &hTokenLogon;
839
840	RTLDRMOD hUserenv;
841	int rc = RTLdrLoad("Userenv.dll", &hUserenv);
842	if (RT_SUCCESS(rc))
843	{
844	PFNLOADUSERPROFILEW pfnLoadUserProfileW;
845	rc = RTLdrGetSymbol(hUserenv, "LoadUserProfileW", (void**)&pfnLoadUserProfileW);
846	if (RT_SUCCESS(rc))
847	{
848	PFNUNLOADUSERPROFILE pfnUnloadUserProfile;
849	rc = RTLdrGetSymbol(hUserenv, "UnloadUserProfile", (void**)&pfnUnloadUserProfile);
850	if (RT_SUCCESS(rc))
851	{
852	PROFILEINFOW profileInfo;
853	ZeroMemory(&profileInfo, sizeof(profileInfo));
854	profileInfo.dwSize = sizeof(profileInfo);
855	profileInfo.lpUserName = pwszUser;
856	profileInfo.dwFlags = PI_NOUI; /* Prevents the display of profile error messages. */
857
858	if (pfnLoadUserProfileW(*phToken, &profileInfo))
859	{
860	PRTUTF16 pwszzBlock;
861	rc = rtProcEnvironmentCreateFromToken(*phToken, hEnv, &pwszzBlock);
862	if (RT_SUCCESS(rc))
863	{
864	/*
865	* Useful KB articles:
866	* http://support.microsoft.com/kb/165194/
867	* http://support.microsoft.com/kb/184802/
868	* http://support.microsoft.com/kb/327618/
869	*/
870	fRc = CreateProcessAsUserW(*phToken,
871	pwszExec,
872	pwszCmdLine,
873	NULL, /* pProcessAttributes */
874	NULL, /* pThreadAttributes */
875	TRUE, /* fInheritHandles */
876	dwCreationFlags,
877	pwszzBlock,
878	NULL, /* pCurrentDirectory */
879	pStartupInfo,
880	pProcInfo);
881	if (fRc)
882	dwErr = NO_ERROR;
883	else
884	dwErr = GetLastError(); /* CreateProcessAsUserW() failed. */
885	rtProcEnvironmentDestroy(pwszzBlock);
886	}
887	else
888	dwErr = rc;
889	pfnUnloadUserProfile(*phToken, profileInfo.hProfile);
890	}
891	else
892	dwErr = GetLastError(); /* LoadUserProfileW() failed. */
893	}
894	}
895	RTLdrClose(hUserenv);
896	}
897	if (hTokenUserDesktop != INVALID_HANDLE_VALUE)
898	CloseHandle(hTokenUserDesktop);
899	rtProcUserLogoff(hTokenLogon);
900	}
901	else
902	dwErr = rc; /* rtProcUserLogon() failed. */
903	}
904
905	if (dwErr != NO_ERROR)
906	{
907	/*
908	* Map some important or much used Windows error codes
909	* to our error codes.
910	*/
911	switch (dwErr)
912	{
913	case ERROR_NOACCESS:
914	case ERROR_PRIVILEGE_NOT_HELD:
915	rc = VERR_PERMISSION_DENIED;
916	break;
917
918	case ERROR_PASSWORD_EXPIRED:
919	case ERROR_ACCOUNT_RESTRICTION: /* See: http://support.microsoft.com/kb/303846/ */
920	rc = VERR_AUTHENTICATION_FAILURE;
921	break;
922
923	default:
924	/* Could trigger a debug assertion! */
925	rc = RTErrConvertFromWin32(dwErr);
926	break;
927	}
928	}
929	else
930	rc = VINF_SUCCESS;
931	return rc;
932	}
933
934	RTR3DECL(int) RTProcCreateEx(const char pszExec, const char const *papszArgs, RTENV hEnv, uint32_t fFlags,
935	PCRTHANDLE phStdIn, PCRTHANDLE phStdOut, PCRTHANDLE phStdErr, const char *pszAsUser,
936	const char *pszPassword, PRTPROCESS phProcess)
937	{
938	/*
939	* Input validation
940	*/
941	AssertPtrReturn(pszExec, VERR_INVALID_POINTER);
942	AssertReturn(*pszExec, VERR_INVALID_PARAMETER);
943	AssertReturn(!(fFlags & ~(RTPROC_FLAGS_DAEMONIZE_DEPRECATED \| RTPROC_FLAGS_DETACHED \| RTPROC_FLAGS_SERVICE)), VERR_INVALID_PARAMETER);
944	AssertReturn(!(fFlags & RTPROC_FLAGS_DETACHED) \|\| !phProcess, VERR_INVALID_PARAMETER);
945	AssertReturn(hEnv != NIL_RTENV, VERR_INVALID_PARAMETER);
946	AssertPtrReturn(papszArgs, VERR_INVALID_PARAMETER);
947	AssertPtrNullReturn(pszAsUser, VERR_INVALID_POINTER);
948	AssertReturn(!pszAsUser \|\| *pszAsUser, VERR_INVALID_PARAMETER);
949	AssertReturn(!pszPassword \|\| pszAsUser, VERR_INVALID_PARAMETER);
950	AssertPtrNullReturn(pszPassword, VERR_INVALID_POINTER);
951	/** @todo search the PATH (add flag for this). */
952
953	/*
954	* Initialize the globals.
955	*/
956	int rc = RTOnce(&g_rtProcWinInitOnce, rtProcWinInitOnce, NULL, NULL);
957	AssertRCReturn(rc, rc);
958
959	/*
960	* Get the file descriptors for the handles we've been passed.
961	*
962	* It seems there is no point in trying to convince a child process's CRT
963	* that any of the standard file handles is non-TEXT. So, we don't...
964	*/
965	STARTUPINFOW StartupInfo;
966	RT_ZERO(StartupInfo);
967	StartupInfo.cb = sizeof(StartupInfo);
968	StartupInfo.dwFlags = STARTF_USESTDHANDLES;
969	#if 1 /* The CRT should keep the standard handles up to date. */
970	StartupInfo.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
971	StartupInfo.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
972	StartupInfo.hStdError = GetStdHandle(STD_ERROR_HANDLE);
973	#else
974	StartupInfo.hStdInput = _get_osfhandle(0);
975	StartupInfo.hStdOutput = _get_osfhandle(1);
976	StartupInfo.hStdError = _get_osfhandle(2);
977	#endif
978	PCRTHANDLE paHandles[3] = { phStdIn, phStdOut, phStdErr };
979	HANDLE *aphStds[3] = { &StartupInfo.hStdInput, &StartupInfo.hStdOutput, &StartupInfo.hStdError };
980	DWORD afInhStds[3] = { 0xffffffff, 0xffffffff, 0xffffffff };
981	for (int i = 0; i < 3; i++)
982	{
983	if (paHandles[i])
984	{
985	AssertPtrReturn(paHandles[i], VERR_INVALID_POINTER);
986	switch (paHandles[i]->enmType)
987	{
988	case RTHANDLETYPE_FILE:
989	*aphStds[i] = paHandles[i]->u.hFile != NIL_RTFILE
990	? (HANDLE)RTFileToNative(paHandles[i]->u.hFile)
991	: INVALID_HANDLE_VALUE;
992	break;
993
994	case RTHANDLETYPE_PIPE:
995	*aphStds[i] = paHandles[i]->u.hPipe != NIL_RTPIPE
996	? (HANDLE)RTPipeToNative(paHandles[i]->u.hPipe)
997	: INVALID_HANDLE_VALUE;
998	break;
999
1000	case RTHANDLETYPE_SOCKET:
1001	*aphStds[i] = paHandles[i]->u.hSocket != NIL_RTSOCKET
1002	? (HANDLE)RTSocketToNative(paHandles[i]->u.hSocket)
1003	: INVALID_HANDLE_VALUE;
1004	break;
1005
1006	default:
1007	AssertMsgFailedReturn(("%d: %d\n", i, paHandles[i]->enmType), VERR_INVALID_PARAMETER);
1008	}
1009
1010	/* Get the inheritability of the handle. */
1011	if (*aphStds[i] != INVALID_HANDLE_VALUE)
1012	{
1013	if (!GetHandleInformation(*aphStds[i], &afInhStds[i]))
1014	{
1015	rc = RTErrConvertFromWin32(GetLastError());
1016	AssertMsgFailedReturn(("%Rrc %p\n", rc, *aphStds[i]), rc);
1017	}
1018	}
1019	}
1020	}
1021
1022	/*
1023	* Set the inheritability any handles we're handing the child.
1024	*/
1025	rc = VINF_SUCCESS;
1026	for (int i = 0; i < 3; i++)
1027	if ( (afInhStds[i] != 0xffffffff)
1028	&& !(afInhStds[i] & HANDLE_FLAG_INHERIT))
1029	{
1030	if (!SetHandleInformation(*aphStds[i], HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT))
1031	{
1032	rc = RTErrConvertFromWin32(GetLastError());
1033	AssertMsgFailedBreak(("%Rrc %p\n", rc, *aphStds[i]));
1034	}
1035	}
1036
1037	/*
1038	* Create the environment block, command line and convert the executable
1039	* name.
1040	*/
1041	PRTUTF16 pwszzBlock;
1042	if (RT_SUCCESS(rc))
1043	rc = RTEnvQueryUtf16Block(hEnv, &pwszzBlock);
1044	if (RT_SUCCESS(rc))
1045	{
1046	PRTUTF16 pwszCmdLine;
1047	rc = RTGetOptArgvToUtf16String(&pwszCmdLine, papszArgs, RTGETOPTARGV_CNV_QUOTE_MS_CRT);
1048	if (RT_SUCCESS(rc))
1049	{
1050	PRTUTF16 pwszExec;
1051	rc = RTStrToUtf16(pszExec, &pwszExec);
1052	if (RT_SUCCESS(rc))
1053	{
1054	/*
1055	* Get going...
1056	*/
1057	PROCESS_INFORMATION ProcInfo;
1058	RT_ZERO(ProcInfo);
1059	DWORD dwCreationFlags = CREATE_UNICODE_ENVIRONMENT;
1060	if (fFlags & RTPROC_FLAGS_DETACHED)
1061	dwCreationFlags \|= DETACHED_PROCESS;
1062
1063	/*
1064	* Only use the normal CreateProcess stuff if we have no user name
1065	* and we are not running from a (Windows) service. Otherwise use
1066	* the more advanced version in rtProcCreateAsUserHlp().
1067	*/
1068	if ( pszAsUser == NULL
1069	&& !(fFlags & RTPROC_FLAGS_SERVICE))
1070	{
1071	if (CreateProcessW(pwszExec,
1072	pwszCmdLine,
1073	NULL, /* pProcessAttributes */
1074	NULL, /* pThreadAttributes */
1075	TRUE, /* fInheritHandles */
1076	dwCreationFlags,
1077	pwszzBlock,
1078	NULL, /* pCurrentDirectory */
1079	&StartupInfo,
1080	&ProcInfo))
1081	rc = VINF_SUCCESS;
1082	else
1083	rc = RTErrConvertFromWin32(GetLastError());
1084	}
1085	else
1086	{
1087	/*
1088	* Convert the additional parameters and use a helper
1089	* function to do the actual work.
1090	*/
1091	PRTUTF16 pwszUser;
1092	rc = RTStrToUtf16(pszAsUser, &pwszUser);
1093	if (RT_SUCCESS(rc))
1094	{
1095	PRTUTF16 pwszPassword;
1096	rc = RTStrToUtf16(pszPassword ? pszPassword : "", &pwszPassword);
1097	if (RT_SUCCESS(rc))
1098	{
1099	rc = rtProcCreateAsUserHlp(pwszUser, pwszPassword,
1100	pwszExec, pwszCmdLine, hEnv, dwCreationFlags,
1101	&StartupInfo, &ProcInfo, fFlags);
1102
1103	RTUtf16Free(pwszPassword);
1104	}
1105	RTUtf16Free(pwszUser);
1106	}
1107	}
1108	if (RT_SUCCESS(rc))
1109	{
1110	CloseHandle(ProcInfo.hThread);
1111	if (phProcess)
1112	{
1113	/*
1114	* Add the process to the child process list so
1115	* RTProcWait can reuse and close the process handle.
1116	*/
1117	rtProcWinAddPid(ProcInfo.dwProcessId, ProcInfo.hProcess);
1118	*phProcess = ProcInfo.dwProcessId;
1119	}
1120	else
1121	CloseHandle(ProcInfo.hProcess);
1122	rc = VINF_SUCCESS;
1123	}
1124	RTUtf16Free(pwszExec);
1125	}
1126	RTUtf16Free(pwszCmdLine);
1127	}
1128	RTEnvFreeUtf16Block(pwszzBlock);
1129	}
1130
1131	/* Undo any handle inherit changes. */
1132	for (int i = 0; i < 3; i++)
1133	if ( (afInhStds[i] != 0xffffffff)
1134	&& !(afInhStds[i] & HANDLE_FLAG_INHERIT))
1135	{
1136	if (!SetHandleInformation(*aphStds[i], HANDLE_FLAG_INHERIT, 0))
1137	AssertMsgFailed(("%Rrc %p\n", RTErrConvertFromWin32(GetLastError()), *aphStds[i]));
1138	}
1139
1140	return rc;
1141	}
1142
1143
1144
1145	RTR3DECL(int) RTProcWait(RTPROCESS Process, unsigned fFlags, PRTPROCSTATUS pProcStatus)
1146	{
1147	AssertReturn(!(fFlags & ~(RTPROCWAIT_FLAGS_BLOCK \| RTPROCWAIT_FLAGS_NOBLOCK)), VERR_INVALID_PARAMETER);
1148	int rc = RTOnce(&g_rtProcWinInitOnce, rtProcWinInitOnce, NULL, NULL);
1149	AssertRCReturn(rc, rc);
1150
1151	/*
1152	* Try find the process among the ones we've spawned, otherwise, attempt
1153	* opening the specified process.
1154	*/
1155	HANDLE hProcess = rtProcWinFindPid(Process);
1156	if (hProcess == NULL)
1157	hProcess = OpenProcess(PROCESS_QUERY_INFORMATION \| SYNCHRONIZE, FALSE, Process);
1158	if (hProcess != NULL)
1159	{
1160	/*
1161	* Wait for it to terminate.
1162	*/
1163	DWORD Millies = fFlags == RTPROCWAIT_FLAGS_BLOCK ? INFINITE : 0;
1164	DWORD WaitRc = WaitForSingleObjectEx(hProcess, Millies, TRUE);
1165	while (WaitRc == WAIT_IO_COMPLETION)
1166	WaitRc = WaitForSingleObjectEx(hProcess, Millies, TRUE);
1167	switch (WaitRc)
1168	{
1169	/*
1170	* It has terminated.
1171	*/
1172	case WAIT_OBJECT_0:
1173	{
1174	DWORD dwExitCode;
1175	if (GetExitCodeProcess(hProcess, &dwExitCode))
1176	{
1177	/** @todo the exit code can be special statuses. */
1178	if (pProcStatus)
1179	{
1180	pProcStatus->enmReason = RTPROCEXITREASON_NORMAL;
1181	pProcStatus->iStatus = (int)dwExitCode;
1182	}
1183	rtProcWinRemovePid(Process);
1184	return VINF_SUCCESS;
1185	}
1186	break;
1187	}
1188
1189	/*
1190	* It hasn't terminated just yet.
1191	*/
1192	case WAIT_TIMEOUT:
1193	return VERR_PROCESS_RUNNING;
1194
1195	/*
1196	* Something went wrong...
1197	*/
1198	case WAIT_FAILED:
1199	break;
1200	case WAIT_ABANDONED:
1201	AssertFailed();
1202	return VERR_GENERAL_FAILURE;
1203	default:
1204	AssertMsgFailed(("WaitRc=%RU32\n", WaitRc));
1205	return VERR_GENERAL_FAILURE;
1206	}
1207	}
1208	DWORD dwErr = GetLastError();
1209	if (dwErr == ERROR_INVALID_PARAMETER)
1210	return VERR_PROCESS_NOT_FOUND;
1211	return RTErrConvertFromWin32(dwErr);
1212	}
1213
1214
1215	RTR3DECL(int) RTProcWaitNoResume(RTPROCESS Process, unsigned fFlags, PRTPROCSTATUS pProcStatus)
1216	{
1217	/** @todo this isn't quite right. */
1218	return RTProcWait(Process, fFlags, pProcStatus);
1219	}
1220
1221
1222	RTR3DECL(int) RTProcTerminate(RTPROCESS Process)
1223	{
1224	int rc = RTOnce(&g_rtProcWinInitOnce, rtProcWinInitOnce, NULL, NULL);
1225	AssertRCReturn(rc, rc);
1226
1227	/*
1228	* Try find the process among the ones we've spawned, otherwise, attempt
1229	* opening the specified process.
1230	*/
1231	HANDLE hProcess = rtProcWinFindPid(Process);
1232	if (hProcess != NULL)
1233	{
1234	if (!TerminateProcess(hProcess, 127))
1235	rc = RTErrConvertFromWin32(GetLastError());
1236	}
1237	else
1238	{
1239	hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, Process);
1240	if (hProcess != NULL)
1241	{
1242	BOOL fRc = TerminateProcess(hProcess, 127);
1243	DWORD dwErr = GetLastError();
1244	CloseHandle(hProcess);
1245	if (!fRc)
1246	rc = RTErrConvertFromWin32(dwErr);
1247	}
1248	}
1249	return rc;
1250	}
1251
1252
1253	RTR3DECL(uint64_t) RTProcGetAffinityMask(void)
1254	{
1255	DWORD_PTR dwProcessAffinityMask = 0xffffffff;
1256	DWORD_PTR dwSystemAffinityMask;
1257
1258	BOOL fRc = GetProcessAffinityMask(GetCurrentProcess(), &dwProcessAffinityMask, &dwSystemAffinityMask);
1259	Assert(fRc);
1260
1261	return dwProcessAffinityMask;
1262	}
1263

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

source: vbox/trunk/src/VBox/Runtime/r3/win/process-win.cpp@ 32657

Download in other formats: