process-win.cpp revision c77f724884bdfe5a7376dc72de92dfae3476f5e3
5b281ba489ca18f0380d7efc7a5108b606cce449vboxsync * IPRT - Process, Windows.
8e3f7f941e446e203bd8f3d0782d761aeb5af6c9vboxsync * Copyright (C) 2006-2010 Sun Microsystems, Inc.
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * available from http://www.virtualbox.org. This file is free software;
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * you can redistribute it and/or modify it under the terms of the GNU
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * General Public License (GPL) as published by the Free Software
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * The contents of this file may alternatively be used under the terms
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * of the Common Development and Distribution License Version 1.0
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * VirtualBox OSE distribution, in which case the provisions of the
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * CDDL are applicable instead of those of the GPL.
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * You may elect to license modified versions of this file under the
bd8e360cd1db83dcb2694ea9122ce3bc5bae678avboxsync * terms and conditions of either the GPL or the CDDL or both.
4d6dcfe00aab559241d9ed05b89f803ab5ddf611vboxsync * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
4d6dcfe00aab559241d9ed05b89f803ab5ddf611vboxsync * Clara, CA 95054 USA or visit http://www.sun.com if you need
4d6dcfe00aab559241d9ed05b89f803ab5ddf611vboxsync * additional information or have any questions.
aa4bcf0a4b2db3ac352b56a291d49cb8d4b66d32vboxsync/*******************************************************************************
aa4bcf0a4b2db3ac352b56a291d49cb8d4b66d32vboxsync* Header Files *
dccbafa70b5a9a6f933c5566e4542caf9f379b97vboxsync*******************************************************************************/
dccbafa70b5a9a6f933c5566e4542caf9f379b97vboxsync/*******************************************************************************
ea6c70405e39fa563a55780ef25e0933d8c73a1avboxsync* Structures and Typedefs *
ea6c70405e39fa563a55780ef25e0933d8c73a1avboxsync*******************************************************************************/
ea6c70405e39fa563a55780ef25e0933d8c73a1avboxsynctypedef WINADVAPI BOOL WINAPI FNCREATEPROCESSWITHLOGON(LPCWSTR,
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsynctypedef FNCREATEPROCESSWITHLOGON *PFNCREATEPROCESSWITHLOGON;
ea6c70405e39fa563a55780ef25e0933d8c73a1avboxsync/*******************************************************************************
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync* Global Variables *
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync*******************************************************************************/
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync/** Init once structure. */
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsyncstatic RTONCE g_rtProcWinInitOnce = RTONCE_INITIALIZER;
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync/** Critical section protecting the process array. */
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync/** The number of processes in the array. */
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync/** The current allocation size. */
ea6c70405e39fa563a55780ef25e0933d8c73a1avboxsync/** Array containing the live or non-reaped child processes. */
ea6c70405e39fa563a55780ef25e0933d8c73a1avboxsync /** The process ID. */
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync /** The process handle. */
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * Clean up the globals.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * @param enmReason Ignored.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * @param iStatus Ignored.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * @param pvUser Ignored.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsyncstatic DECLCALLBACK(void) rtProcWinTerm(RTTERMREASON enmReason, int32_t iStatus, void *pvUser)
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync while (i-- > 0)
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * Initialize the globals.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * @returns IPRT status code.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * @param pvUser1 Ignored.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsync * @param pvUser2 Ignored.
c882ddf98a60a4aab1218108b334083f98b7d66avboxsyncstatic DECLCALLBACK(int32_t) rtProcWinInitOnce(void *pvUser1, void *pvUser2)
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync /** @todo init once, terminate once - this is a generic thing which should
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync * have some kind of static and simpler setup! */
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync * Gets the process handle for a process from g_paProcesses.
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync * @returns Process handle if found, NULL if not.
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync * @param pid The process to remove (pid).
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync while (i-- > 0)
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync * Removes a process from g_paProcesses.
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync * @param pid The process to remove (pid).
b6097213cd07c5383c49168bfb61f915dbd4587avboxsync while (i-- > 0)
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync memmove(&g_paProcesses[i], &g_paProcesses[i + 1], cToMove * sizeof(g_paProcesses[0]));
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync * Adds a process to g_paProcesses.
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync * @returns IPRT status code.
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync * @param pid The process id.
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync * @param hProcess The process handle.
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsyncstatic int rtProcWinAddPid(RTPROCESS pid, HANDLE hProcess)
fd546afe09073de92e5422c1334f1c27b4108687vboxsync void *pvNew = RTMemRealloc(g_paProcesses, (i + 16) * sizeof(g_paProcesses[0]));
fd546afe09073de92e5422c1334f1c27b4108687vboxsyncRTR3DECL(int) RTProcCreate(const char *pszExec, const char * const *papszArgs, RTENV Env, unsigned fFlags, PRTPROCESS pProcess)
fd546afe09073de92e5422c1334f1c27b4108687vboxsync return RTProcCreateEx(pszExec, papszArgs, Env, fFlags,
8e3f7f941e446e203bd8f3d0782d761aeb5af6c9vboxsyncstatic int rtProcCreateAsUserHlp(PRTUTF16 pwszUser, PRTUTF16 pwszPassword, PRTUTF16 pwszExec, PRTUTF16 pwszCmdLine,
fd546afe09073de92e5422c1334f1c27b4108687vboxsync PRTUTF16 pwszzBlock, STARTUPINFOW *pStartupInfo, PROCESS_INFORMATION *pProcInfo)
fd546afe09073de92e5422c1334f1c27b4108687vboxsync /** @todo On NT4 we need to enable the SeTcbPrivilege to act as part of the operating system. Otherwise
fd546afe09073de92e5422c1334f1c27b4108687vboxsync * we will get error 1314 (priviledge not held) as a response. */
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync * The following rights are needed in order to use LogonUserW and
8e3f7f941e446e203bd8f3d0782d761aeb5af6c9vboxsync * CreateProcessAsUserW, so the local policy has to be modified to:
8e3f7f941e446e203bd8f3d0782d761aeb5af6c9vboxsync * - SE_TCB_NAME = Act as part of the operating system
8e3f7f941e446e203bd8f3d0782d761aeb5af6c9vboxsync * - SE_ASSIGNPRIMARYTOKEN_NAME = Create/replace a token object
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync * - SE_INCREASE_QUOTA_NAME
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync * We may fail here with ERROR_PRIVILEGE_NOT_HELD.
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * If we don't hold enough priviledges to spawn a new process with
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * different credentials we have to use CreateProcessWithLogonW here. This
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * API is W2K+ and uses a system service for spawning the process.
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync /** @todo Use fFlags to either use this feature or just fail. */
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync /* This may fail on too old (NT4) platforms. */
45e9c1c72518aeba6673332bdd4d70b59e1c11a4vboxsync PFNCREATEPROCESSWITHLOGON pfnCreateProcessWithLogonW;
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync rc = RTLdrGetSymbol(hAdvAPI32, "CreateProcessWithLogonW", (void**)&pfnCreateProcessWithLogonW);
4fa08e9052161e65bfe8d2e3bb38fa227af628c0vboxsyncRTR3DECL(int) RTProcCreateEx(const char *pszExec, const char * const *papszArgs, RTENV hEnv, uint32_t fFlags,
4fa08e9052161e65bfe8d2e3bb38fa227af628c0vboxsync PCRTHANDLE phStdIn, PCRTHANDLE phStdOut, PCRTHANDLE phStdErr, const char *pszAsUser,
4fa08e9052161e65bfe8d2e3bb38fa227af628c0vboxsync * Input validation
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync AssertReturn(!(fFlags & ~RTPROC_FLAGS_DAEMONIZE), VERR_INVALID_PARAMETER);
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync AssertReturn(hEnv != NIL_RTENV, VERR_INVALID_PARAMETER);
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync AssertPtrReturn(papszArgs, VERR_INVALID_PARAMETER);
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync /** @todo search the PATH (add flag for this). */
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync * Initialize the globals.
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync int rc = RTOnce(&g_rtProcWinInitOnce, rtProcWinInitOnce, NULL, NULL);
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync * Get the file descriptors for the handles we've been passed.
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync * It seems there is no point in trying to convince a child process's CRT
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync * that any of the standard file handles is non-TEXT. So, we don't...
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync#if 1 /* The CRT should keep the standard handles up to date. */
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync StartupInfo.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync StartupInfo.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync StartupInfo.hStdError = GetStdHandle(STD_ERROR_HANDLE);
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync PCRTHANDLE paHandles[3] = { phStdIn, phStdOut, phStdErr };
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync HANDLE *aphStds[3] = { &StartupInfo.hStdInput, &StartupInfo.hStdOutput, &StartupInfo.hStdError };
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync DWORD afInhStds[3] = { 0xffffffff, 0xffffffff, 0xffffffff };
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync for (int i = 0; i < 3; i++)
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync AssertPtrReturn(paHandles[i], VERR_INVALID_POINTER);
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync *aphStds[i] = paHandles[i]->u.hSocket != NIL_RTSOCKET
1a05eb58618c8f646bd1304d10c73792e0034f20vboxsync ? (HANDLE)RTSocketToNative(paHandles[i]->u.hSocket)
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync AssertMsgFailedReturn(("%d: %d\n", i, paHandles[i]->enmType), VERR_INVALID_PARAMETER);
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync /* Get the inheritability of the handle. */
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync if (!GetHandleInformation(*aphStds[i], &afInhStds[i]))
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync AssertMsgFailedReturn(("%Rrc %p\n", rc, *aphStds[i]), rc);
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync * Set the inheritability any handles we're handing the child.
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync for (int i = 0; i < 3; i++)
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync if (!SetHandleInformation(*aphStds[i], HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT))
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync AssertMsgFailedBreak(("%Rrc %p\n", rc, *aphStds[i]));
f6f8bfbbbc6a59ba94b01886ed5a8d6e5813073bvboxsync * Create the environment block, command line and convert the executable
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync rc = RTGetOptArgvToUtf16String(&pwszCmdLine, papszArgs, RTGETOPTARGV_CNV_QUOTE_MS_CRT);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * Get going...
81587231c9c584851518872e197f6f02dffe68cavboxsync * Convert the additional parameters and use a helper
81587231c9c584851518872e197f6f02dffe68cavboxsync * function to do the actual work.
81587231c9c584851518872e197f6f02dffe68cavboxsync rc = RTStrToUtf16(pszPassword ? pszPassword : "", &pwszPassword);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * Add the process to the child process list so
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * RTProcWait can reuse and close the process handle.
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync rtProcWinAddPid(ProcInfo.dwProcessId, ProcInfo.hProcess);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync /* Undo any handle inherit changes. */
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync for (int i = 0; i < 3; i++)
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync if (!SetHandleInformation(*aphStds[i], HANDLE_FLAG_INHERIT, 0))
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync AssertMsgFailed(("%Rrc %p\n", RTErrConvertFromWin32(GetLastError()), *aphStds[i]));
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsyncRTR3DECL(int) RTProcWait(RTPROCESS Process, unsigned fFlags, PRTPROCSTATUS pProcStatus)
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync AssertReturn(!(fFlags & ~(RTPROCWAIT_FLAGS_BLOCK | RTPROCWAIT_FLAGS_NOBLOCK)), VERR_INVALID_PARAMETER);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync int rc = RTOnce(&g_rtProcWinInitOnce, rtProcWinInitOnce, NULL, NULL);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * Try find the process among the ones we've spawned, otherwise, attempt
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * opening the specified process.
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | SYNCHRONIZE, FALSE, Process);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * Wait for it to terminate.
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync DWORD Millies = fFlags == RTPROCWAIT_FLAGS_BLOCK ? INFINITE : 0;
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync DWORD WaitRc = WaitForSingleObjectEx(hProcess, Millies, TRUE);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync WaitRc = WaitForSingleObjectEx(hProcess, Millies, TRUE);
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync * It has terminated.
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsync /** @todo the exit code can be special statuses. */
81587231c9c584851518872e197f6f02dffe68cavboxsync * It hasn't terminated just yet.
dccbafa70b5a9a6f933c5566e4542caf9f379b97vboxsync * Something went wrong...
b1cf57acefb5d1ce6ae2466f43e225a81083ff34vboxsyncRTR3DECL(int) RTProcWaitNoResume(RTPROCESS Process, unsigned fFlags, PRTPROCSTATUS pProcStatus)
da97b7ac000d0f02e86c31a4f2767a00d83c6167vboxsync /** @todo this isn't quite right. */
81587231c9c584851518872e197f6f02dffe68cavboxsync HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, Process);