123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898 |
- /* Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include "win32/apr_arch_threadproc.h"
- #include "win32/apr_arch_file_io.h"
- #include "apr_thread_proc.h"
- #include "apr_file_io.h"
- #include "apr_general.h"
- #include "apr_strings.h"
- #include "apr_portable.h"
- #include "apr_lib.h"
- #include <stdlib.h>
- #if APR_HAVE_SIGNAL_H
- #include <signal.h>
- #endif
- #include <string.h>
- #if APR_HAVE_PROCESS_H
- #include <process.h>
- #endif
- #ifdef _WIN32_WCE
- #ifndef DETACHED_PROCESS
- #define DETACHED_PROCESS 0
- #endif
- #ifndef CREATE_UNICODE_ENVIRONMENT
- #define CREATE_UNICODE_ENVIRONMENT 0
- #endif
- #ifndef STARTF_USESHOWWINDOW
- #define STARTF_USESHOWWINDOW 0
- #endif
- #ifndef SW_HIDE
- #define SW_HIDE 0
- #endif
- #endif
- /*
- * some of the ideas expressed herein are based off of Microsoft
- * Knowledge Base article: Q190351
- *
- */
- APR_DECLARE(apr_status_t) apr_procattr_create(apr_procattr_t **new,
- apr_pool_t *pool)
- {
- (*new) = (apr_procattr_t *)apr_pcalloc(pool, sizeof(apr_procattr_t));
- (*new)->pool = pool;
- (*new)->cmdtype = APR_PROGRAM;
- return APR_SUCCESS;
- }
- APR_DECLARE(apr_status_t) apr_procattr_io_set(apr_procattr_t *attr,
- apr_int32_t in,
- apr_int32_t out,
- apr_int32_t err)
- {
- apr_status_t stat = APR_SUCCESS;
- if (in) {
- /* APR_CHILD_BLOCK maps to APR_WRITE_BLOCK, while
- * APR_PARENT_BLOCK maps to APR_READ_BLOCK, so we
- * must transpose the CHILD/PARENT blocking flags
- * only for the stdin pipe. stdout/stderr naturally
- * map to the correct mode.
- */
- if (in == APR_CHILD_BLOCK)
- in = APR_READ_BLOCK;
- else if (in == APR_PARENT_BLOCK)
- in = APR_WRITE_BLOCK;
- stat = apr_create_nt_pipe(&attr->child_in, &attr->parent_in, in,
- attr->pool);
- if (stat == APR_SUCCESS)
- stat = apr_file_inherit_unset(attr->parent_in);
- }
- if (out && stat == APR_SUCCESS) {
- stat = apr_create_nt_pipe(&attr->parent_out, &attr->child_out, out,
- attr->pool);
- if (stat == APR_SUCCESS)
- stat = apr_file_inherit_unset(attr->parent_out);
- }
- if (err && stat == APR_SUCCESS) {
- stat = apr_create_nt_pipe(&attr->parent_err, &attr->child_err, err,
- attr->pool);
- if (stat == APR_SUCCESS)
- stat = apr_file_inherit_unset(attr->parent_err);
- }
- return stat;
- }
- APR_DECLARE(apr_status_t) apr_procattr_child_in_set(apr_procattr_t *attr,
- apr_file_t *child_in,
- apr_file_t *parent_in)
- {
- apr_status_t rv = APR_SUCCESS;
- if (child_in) {
- if (attr->child_in == NULL)
- rv = apr_file_dup(&attr->child_in, child_in, attr->pool);
- else
- rv = apr_file_dup2(attr->child_in, child_in, attr->pool);
- if (rv == APR_SUCCESS)
- rv = apr_file_inherit_set(attr->child_in);
- }
- if (parent_in && rv == APR_SUCCESS) {
- if (attr->parent_in == NULL)
- rv = apr_file_dup(&attr->parent_in, parent_in, attr->pool);
- else
- rv = apr_file_dup2(attr->parent_in, parent_in, attr->pool);
- }
- return rv;
- }
- APR_DECLARE(apr_status_t) apr_procattr_child_out_set(apr_procattr_t *attr,
- apr_file_t *child_out,
- apr_file_t *parent_out)
- {
- apr_status_t rv = APR_SUCCESS;
- if (child_out) {
- if (attr->child_out == NULL)
- rv = apr_file_dup(&attr->child_out, child_out, attr->pool);
- else
- rv = apr_file_dup2(attr->child_out, child_out, attr->pool);
- if (rv == APR_SUCCESS)
- rv = apr_file_inherit_set(attr->child_out);
- }
- if (parent_out && rv == APR_SUCCESS) {
- if (attr->parent_out == NULL)
- rv = apr_file_dup(&attr->parent_out, parent_out, attr->pool);
- else
- rv = apr_file_dup2(attr->parent_out, parent_out, attr->pool);
- }
- return rv;
- }
- APR_DECLARE(apr_status_t) apr_procattr_child_err_set(apr_procattr_t *attr,
- apr_file_t *child_err,
- apr_file_t *parent_err)
- {
- apr_status_t rv = APR_SUCCESS;
- if (child_err) {
- if (attr->child_err == NULL)
- rv = apr_file_dup(&attr->child_err, child_err, attr->pool);
- else
- rv = apr_file_dup2(attr->child_err, child_err, attr->pool);
- if (rv == APR_SUCCESS)
- rv = apr_file_inherit_set(attr->child_err);
- }
- if (parent_err && rv == APR_SUCCESS) {
- if (attr->parent_err == NULL)
- rv = apr_file_dup(&attr->parent_err, parent_err, attr->pool);
- else
- rv = apr_file_dup2(attr->parent_err, parent_err, attr->pool);
- }
- return rv;
- }
- APR_DECLARE(apr_status_t) apr_procattr_dir_set(apr_procattr_t *attr,
- const char *dir)
- {
- /* curr dir must be in native format, there are all sorts of bugs in
- * the NT library loading code that flunk the '/' parsing test.
- */
- return apr_filepath_merge(&attr->currdir, NULL, dir,
- APR_FILEPATH_NATIVE, attr->pool);
- }
- APR_DECLARE(apr_status_t) apr_procattr_cmdtype_set(apr_procattr_t *attr,
- apr_cmdtype_e cmd)
- {
- attr->cmdtype = cmd;
- return APR_SUCCESS;
- }
- APR_DECLARE(apr_status_t) apr_procattr_detach_set(apr_procattr_t *attr,
- apr_int32_t det)
- {
- attr->detached = det;
- return APR_SUCCESS;
- }
- static apr_status_t attr_cleanup(void *theattr)
- {
- apr_procattr_t *attr = (apr_procattr_t *)theattr;
- if (attr->user_token)
- CloseHandle(attr->user_token);
- attr->user_token = NULL;
- return APR_SUCCESS;
- }
- APR_DECLARE(apr_status_t) apr_procattr_user_set(apr_procattr_t *attr,
- const char *username,
- const char *password)
- {
- #ifdef _WIN32_WCE
- return APR_ENOTIMPL;
- #else
- HANDLE user;
- apr_wchar_t *wusername = NULL;
- apr_wchar_t *wpassword = NULL;
- apr_status_t rv;
- apr_size_t len, wlen;
- if (apr_os_level >= APR_WIN_NT_4)
- {
- if (attr->user_token) {
- /* Cannot set that twice */
- if (attr->errfn) {
- attr->errfn(attr->pool, 0,
- apr_pstrcat(attr->pool,
- "function called twice"
- " on username: ", username, NULL));
- }
- return APR_EINVAL;
- }
- len = strlen(username) + 1;
- wlen = len;
- wusername = apr_palloc(attr->pool, wlen * sizeof(apr_wchar_t));
- if ((rv = apr_conv_utf8_to_ucs2(username, &len, wusername, &wlen))
- != APR_SUCCESS) {
- if (attr->errfn) {
- attr->errfn(attr->pool, rv,
- apr_pstrcat(attr->pool,
- "utf8 to ucs2 conversion failed"
- " on username: ", username, NULL));
- }
- return rv;
- }
- if (password) {
- len = strlen(password) + 1;
- wlen = len;
- wpassword = apr_palloc(attr->pool, wlen * sizeof(apr_wchar_t));
- if ((rv = apr_conv_utf8_to_ucs2(password, &len, wpassword, &wlen))
- != APR_SUCCESS) {
- if (attr->errfn) {
- attr->errfn(attr->pool, rv,
- apr_pstrcat(attr->pool,
- "utf8 to ucs2 conversion failed"
- " on password: ", password, NULL));
- }
- return rv;
- }
- }
- if (!LogonUserW(wusername,
- NULL,
- wpassword ? wpassword : L"",
- LOGON32_LOGON_NETWORK,
- LOGON32_PROVIDER_DEFAULT,
- &user)) {
- /* Logon Failed */
- return apr_get_os_error();
- }
- if (wpassword)
- memset(wpassword, 0, wlen * sizeof(apr_wchar_t));
- /* Get the primary token for user */
- if (!DuplicateTokenEx(user,
- TOKEN_QUERY | TOKEN_DUPLICATE | TOKEN_ASSIGN_PRIMARY,
- NULL,
- SecurityImpersonation,
- TokenPrimary,
- &(attr->user_token))) {
- /* Failed to duplicate the user token */
- rv = apr_get_os_error();
- CloseHandle(user);
- return rv;
- }
- CloseHandle(user);
- attr->sd = apr_pcalloc(attr->pool, SECURITY_DESCRIPTOR_MIN_LENGTH);
- InitializeSecurityDescriptor(attr->sd, SECURITY_DESCRIPTOR_REVISION);
- SetSecurityDescriptorDacl(attr->sd, -1, 0, 0);
- attr->sa = apr_palloc(attr->pool, sizeof(SECURITY_ATTRIBUTES));
- attr->sa->nLength = sizeof (SECURITY_ATTRIBUTES);
- attr->sa->lpSecurityDescriptor = attr->sd;
- attr->sa->bInheritHandle = TRUE;
- /* register the cleanup */
- apr_pool_cleanup_register(attr->pool, (void *)attr,
- attr_cleanup,
- apr_pool_cleanup_null);
- return APR_SUCCESS;
- }
- else
- return APR_ENOTIMPL;
- #endif
- }
- APR_DECLARE(apr_status_t) apr_procattr_group_set(apr_procattr_t *attr,
- const char *groupname)
- {
- /* Always return SUCCESS cause groups are irrelevant */
- return APR_SUCCESS;
- }
- static const char* has_space(const char *str)
- {
- const char *ch;
- for (ch = str; *ch; ++ch) {
- if (apr_isspace(*ch)) {
- return ch;
- }
- }
- return NULL;
- }
- static char *apr_caret_escape_args(apr_pool_t *p, const char *str)
- {
- char *cmd;
- unsigned char *d;
- const unsigned char *s;
- cmd = apr_palloc(p, 2 * strlen(str) + 1); /* Be safe */
- d = (unsigned char *)cmd;
- s = (const unsigned char *)str;
- for (; *s; ++s) {
- /*
- * Newlines to Win32/OS2 CreateProcess() are ill advised.
- * Convert them to spaces since they are effectively white
- * space to most applications
- */
- if (*s == '\r' || *s == '\n') {
- *d++ = ' ';
- continue;
- }
- if (IS_SHCHAR(*s)) {
- *d++ = '^';
- }
- *d++ = *s;
- }
- *d = '\0';
- return cmd;
- }
- APR_DECLARE(apr_status_t) apr_procattr_child_errfn_set(apr_procattr_t *attr,
- apr_child_errfn_t *errfn)
- {
- attr->errfn = errfn;
- return APR_SUCCESS;
- }
- APR_DECLARE(apr_status_t) apr_procattr_error_check_set(apr_procattr_t *attr,
- apr_int32_t chk)
- {
- attr->errchk = chk;
- return APR_SUCCESS;
- }
- APR_DECLARE(apr_status_t) apr_procattr_addrspace_set(apr_procattr_t *attr,
- apr_int32_t addrspace)
- {
- /* won't ever be used on this platform, so don't save the flag */
- return APR_SUCCESS;
- }
- APR_DECLARE(apr_status_t) apr_proc_create(apr_proc_t *new,
- const char *progname,
- const char * const *args,
- const char * const *env,
- apr_procattr_t *attr,
- apr_pool_t *pool)
- {
- apr_status_t rv;
- apr_size_t i;
- const char *argv0;
- char *cmdline;
- char *pEnvBlock;
- PROCESS_INFORMATION pi;
- DWORD dwCreationFlags = 0;
- new->in = attr->parent_in;
- new->out = attr->parent_out;
- new->err = attr->parent_err;
- if (attr->detached) {
- /* If we are creating ourselves detached, Then we should hide the
- * window we are starting in. And we had better redfine our
- * handles for STDIN, STDOUT, and STDERR. Do not set the
- * detached attribute for Win9x. We have found that Win9x does
- * not manage the stdio handles properly when running old 16
- * bit executables if the detached attribute is set.
- */
- if (apr_os_level >= APR_WIN_NT) {
- /*
- * XXX DETACHED_PROCESS won't on Win9x at all; on NT/W2K
- * 16 bit executables fail (MS KB: Q150956)
- */
- dwCreationFlags |= DETACHED_PROCESS;
- }
- }
- /* progname must be unquoted, in native format, as there are all sorts
- * of bugs in the NT library loader code that fault when parsing '/'.
- * XXX progname must be NULL if this is a 16 bit app running in WOW
- */
- if (progname[0] == '\"') {
- progname = apr_pstrndup(pool, progname + 1, strlen(progname) - 2);
- }
- if (attr->cmdtype == APR_PROGRAM || attr->cmdtype == APR_PROGRAM_ENV) {
- char *fullpath = NULL;
- if ((rv = apr_filepath_merge(&fullpath, attr->currdir, progname,
- APR_FILEPATH_NATIVE, pool)) != APR_SUCCESS) {
- if (attr->errfn) {
- attr->errfn(pool, rv,
- apr_pstrcat(pool, "filepath_merge failed.",
- " currdir: ", attr->currdir,
- " progname: ", progname, NULL));
- }
- return rv;
- }
- progname = fullpath;
- }
- else {
- /* Do not fail if the path isn't parseable for APR_PROGRAM_PATH
- * or APR_SHELLCMD. We only invoke apr_filepath_merge (with no
- * left hand side expression) in order to correct the path slash
- * delimiters. But the filename doesn't need to be in the CWD,
- * nor does it need to be a filename at all (it could be a
- * built-in shell command.)
- */
- char *fullpath = NULL;
- if ((rv = apr_filepath_merge(&fullpath, "", progname,
- APR_FILEPATH_NATIVE, pool)) == APR_SUCCESS) {
- progname = fullpath;
- }
- }
- if (has_space(progname)) {
- argv0 = apr_pstrcat(pool, "\"", progname, "\"", NULL);
- }
- else {
- argv0 = progname;
- }
- /* Handle the args, seperate from argv0 */
- cmdline = "";
- for (i = 1; args && args[i]; ++i) {
- if (has_space(args[i]) || !args[i][0]) {
- cmdline = apr_pstrcat(pool, cmdline, " \"", args[i], "\"", NULL);
- }
- else {
- cmdline = apr_pstrcat(pool, cmdline, " ", args[i], NULL);
- }
- }
- #ifndef _WIN32_WCE
- if (attr->cmdtype == APR_SHELLCMD || attr->cmdtype == APR_SHELLCMD_ENV) {
- char *shellcmd = getenv("COMSPEC");
- if (!shellcmd) {
- if (attr->errfn) {
- attr->errfn(pool, APR_EINVAL, "COMSPEC envar is not set");
- }
- return APR_EINVAL;
- }
- if (shellcmd[0] == '"') {
- progname = apr_pstrndup(pool, shellcmd + 1, strlen(shellcmd) - 2);
- }
- else {
- progname = shellcmd;
- if (has_space(shellcmd)) {
- shellcmd = apr_pstrcat(pool, "\"", shellcmd, "\"", NULL);
- }
- }
- /* Command.com does not support a quoted command, while cmd.exe demands one.
- */
- i = strlen(progname);
- if (i >= 11 && strcasecmp(progname + i - 11, "command.com") == 0) {
- cmdline = apr_pstrcat(pool, shellcmd, " /C ", argv0, cmdline, NULL);
- }
- else {
- cmdline = apr_pstrcat(pool, shellcmd, " /C \"", argv0, cmdline, "\"", NULL);
- }
- }
- else
- #endif
- {
- /* Win32 is _different_ than unix. While unix will find the given
- * program since it's already chdir'ed, Win32 cannot since the parent
- * attempts to open the program with it's own path.
- * ###: This solution isn't much better - it may defeat path searching
- * when the path search was desired. Open to further discussion.
- */
- i = strlen(progname);
- if (i >= 4 && (strcasecmp(progname + i - 4, ".bat") == 0
- || strcasecmp(progname + i - 4, ".cmd") == 0))
- {
- char *shellcmd = getenv("COMSPEC");
- if (!shellcmd) {
- if (attr->errfn) {
- attr->errfn(pool, APR_EINVAL, "COMSPEC envar is not set");
- }
- return APR_EINVAL;
- }
- if (shellcmd[0] == '"') {
- progname = apr_pstrndup(pool, shellcmd + 1, strlen(shellcmd) - 2);
- }
- else {
- progname = shellcmd;
- if (has_space(shellcmd)) {
- shellcmd = apr_pstrcat(pool, "\"", shellcmd, "\"", NULL);
- }
- }
- i = strlen(progname);
- if (i >= 11 && strcasecmp(progname + i - 11, "command.com") == 0) {
- /* XXX: Still insecure - need doubled-quotes on each individual
- * arg of cmdline. Suspect we need to postpone cmdline parsing
- * until this moment in all four code paths, with some flags
- * to toggle 'which flavor' is needed.
- */
- cmdline = apr_pstrcat(pool, shellcmd, " /C ", argv0, cmdline, NULL);
- }
- else {
- /* We must protect the cmdline args from any interpolation - this
- * is not a shellcmd, and the source of argv[] is untrusted.
- * Notice we escape ALL the cmdline args, including the quotes
- * around the individual args themselves. No sense in allowing
- * the shift-state to be toggled, and the application will
- * not see the caret escapes.
- */
- cmdline = apr_caret_escape_args(pool, cmdline);
- /*
- * Our app name must always be quoted so the quotes surrounding
- * the entire /c "command args" are unambigious.
- */
- if (*argv0 != '"') {
- cmdline = apr_pstrcat(pool, shellcmd, " /C \"\"", argv0, "\"", cmdline, "\"", NULL);
- }
- else {
- cmdline = apr_pstrcat(pool, shellcmd, " /C \"", argv0, cmdline, "\"", NULL);
- }
- }
- }
- else {
- /* A simple command we are directly invoking. Do not pass
- * the first arg to CreateProc() for APR_PROGRAM_PATH
- * invocation, since it would need to be a literal and
- * complete file path. That is; "c:\bin\aprtest.exe"
- * would succeed, but "c:\bin\aprtest" or "aprtest.exe"
- * can fail.
- */
- cmdline = apr_pstrcat(pool, argv0, cmdline, NULL);
- if (attr->cmdtype == APR_PROGRAM_PATH) {
- progname = NULL;
- }
- }
- }
- if (!env || attr->cmdtype == APR_PROGRAM_ENV ||
- attr->cmdtype == APR_SHELLCMD_ENV) {
- pEnvBlock = NULL;
- }
- else {
- apr_size_t iEnvBlockLen;
- /*
- * Win32's CreateProcess call requires that the environment
- * be passed in an environment block, a null terminated block of
- * null terminated strings.
- */
- i = 0;
- iEnvBlockLen = 1;
- while (env[i]) {
- iEnvBlockLen += strlen(env[i]) + 1;
- i++;
- }
- if (!i)
- ++iEnvBlockLen;
- #if APR_HAS_UNICODE_FS
- IF_WIN_OS_IS_UNICODE
- {
- apr_wchar_t *pNext;
- pEnvBlock = (char *)apr_palloc(pool, iEnvBlockLen * 2);
- dwCreationFlags |= CREATE_UNICODE_ENVIRONMENT;
- i = 0;
- pNext = (apr_wchar_t*)pEnvBlock;
- while (env[i]) {
- apr_size_t in = strlen(env[i]) + 1;
- if ((rv = apr_conv_utf8_to_ucs2(env[i], &in,
- pNext, &iEnvBlockLen))
- != APR_SUCCESS) {
- if (attr->errfn) {
- attr->errfn(pool, rv,
- apr_pstrcat(pool,
- "utf8 to ucs2 conversion failed"
- " on this string: ", env[i], NULL));
- }
- return rv;
- }
- pNext = wcschr(pNext, L'\0') + 1;
- i++;
- }
- if (!i)
- *(pNext++) = L'\0';
- *pNext = L'\0';
- }
- #endif /* APR_HAS_UNICODE_FS */
- #if APR_HAS_ANSI_FS
- ELSE_WIN_OS_IS_ANSI
- {
- char *pNext;
- pEnvBlock = (char *)apr_palloc(pool, iEnvBlockLen);
-
- i = 0;
- pNext = pEnvBlock;
- while (env[i]) {
- strcpy(pNext, env[i]);
- pNext = strchr(pNext, '\0') + 1;
- i++;
- }
- if (!i)
- *(pNext++) = '\0';
- *pNext = '\0';
- }
- #endif /* APR_HAS_ANSI_FS */
- }
- new->invoked = cmdline;
- #if APR_HAS_UNICODE_FS
- IF_WIN_OS_IS_UNICODE
- {
- STARTUPINFOW si;
- apr_wchar_t *wprg = NULL;
- apr_wchar_t *wcmd = NULL;
- apr_wchar_t *wcwd = NULL;
- if (progname) {
- apr_size_t nprg = strlen(progname) + 1;
- apr_size_t nwprg = nprg + 6;
- wprg = apr_palloc(pool, nwprg * sizeof(wprg[0]));
- if ((rv = apr_conv_utf8_to_ucs2(progname, &nprg, wprg, &nwprg))
- != APR_SUCCESS) {
- if (attr->errfn) {
- attr->errfn(pool, rv,
- apr_pstrcat(pool,
- "utf8 to ucs2 conversion failed"
- " on progname: ", progname, NULL));
- }
- return rv;
- }
- }
- if (cmdline) {
- apr_size_t ncmd = strlen(cmdline) + 1;
- apr_size_t nwcmd = ncmd;
- wcmd = apr_palloc(pool, nwcmd * sizeof(wcmd[0]));
- if ((rv = apr_conv_utf8_to_ucs2(cmdline, &ncmd, wcmd, &nwcmd))
- != APR_SUCCESS) {
- if (attr->errfn) {
- attr->errfn(pool, rv,
- apr_pstrcat(pool,
- "utf8 to ucs2 conversion failed"
- " on cmdline: ", cmdline, NULL));
- }
- return rv;
- }
- }
- if (attr->currdir)
- {
- apr_size_t ncwd = strlen(attr->currdir) + 1;
- apr_size_t nwcwd = ncwd;
- wcwd = apr_palloc(pool, ncwd * sizeof(wcwd[0]));
- if ((rv = apr_conv_utf8_to_ucs2(attr->currdir, &ncwd,
- wcwd, &nwcwd))
- != APR_SUCCESS) {
- if (attr->errfn) {
- attr->errfn(pool, rv,
- apr_pstrcat(pool,
- "utf8 to ucs2 conversion failed"
- " on currdir: ", attr->currdir, NULL));
- }
- return rv;
- }
- }
- memset(&si, 0, sizeof(si));
- si.cb = sizeof(si);
- if (attr->detached) {
- si.dwFlags |= STARTF_USESHOWWINDOW;
- si.wShowWindow = SW_HIDE;
- }
- #ifndef _WIN32_WCE
- if ((attr->child_in && attr->child_in->filehand)
- || (attr->child_out && attr->child_out->filehand)
- || (attr->child_err && attr->child_err->filehand))
- {
- si.dwFlags |= STARTF_USESTDHANDLES;
- si.hStdInput = (attr->child_in)
- ? attr->child_in->filehand
- : INVALID_HANDLE_VALUE;
- si.hStdOutput = (attr->child_out)
- ? attr->child_out->filehand
- : INVALID_HANDLE_VALUE;
- si.hStdError = (attr->child_err)
- ? attr->child_err->filehand
- : INVALID_HANDLE_VALUE;
- }
- if (attr->user_token) {
- si.lpDesktop = L"Winsta0\\Default";
- if (!ImpersonateLoggedOnUser(attr->user_token)) {
- /* failed to impersonate the logged user */
- rv = apr_get_os_error();
- CloseHandle(attr->user_token);
- attr->user_token = NULL;
- return rv;
- }
- rv = CreateProcessAsUserW(attr->user_token,
- wprg, wcmd,
- attr->sa,
- NULL,
- TRUE,
- dwCreationFlags,
- pEnvBlock,
- wcwd,
- &si, &pi);
- RevertToSelf();
- }
- else {
- rv = CreateProcessW(wprg, wcmd, /* Executable & Command line */
- NULL, NULL, /* Proc & thread security attributes */
- TRUE, /* Inherit handles */
- dwCreationFlags, /* Creation flags */
- pEnvBlock, /* Environment block */
- wcwd, /* Current directory name */
- &si, &pi);
- }
- #else
- rv = CreateProcessW(wprg, wcmd, /* Executable & Command line */
- NULL, NULL, /* Proc & thread security attributes */
- FALSE, /* must be 0 */
- dwCreationFlags, /* Creation flags */
- NULL, /* Environment block must be NULL */
- NULL, /* Current directory name must be NULL*/
- NULL, /* STARTUPINFO not supported */
- &pi);
- #endif
- }
- #endif /* APR_HAS_UNICODE_FS */
- #if APR_HAS_ANSI_FS
- ELSE_WIN_OS_IS_ANSI
- {
- STARTUPINFOA si;
- memset(&si, 0, sizeof(si));
- si.cb = sizeof(si);
- if (attr->detached) {
- si.dwFlags |= STARTF_USESHOWWINDOW;
- si.wShowWindow = SW_HIDE;
- }
- if ((attr->child_in && attr->child_in->filehand)
- || (attr->child_out && attr->child_out->filehand)
- || (attr->child_err && attr->child_err->filehand))
- {
- si.dwFlags |= STARTF_USESTDHANDLES;
- si.hStdInput = (attr->child_in)
- ? attr->child_in->filehand
- : INVALID_HANDLE_VALUE;
- si.hStdOutput = (attr->child_out)
- ? attr->child_out->filehand
- : INVALID_HANDLE_VALUE;
- si.hStdError = (attr->child_err)
- ? attr->child_err->filehand
- : INVALID_HANDLE_VALUE;
- }
- rv = CreateProcessA(progname, cmdline, /* Command line */
- NULL, NULL, /* Proc & thread security attributes */
- TRUE, /* Inherit handles */
- dwCreationFlags, /* Creation flags */
- pEnvBlock, /* Environment block */
- attr->currdir, /* Current directory name */
- &si, &pi);
- }
- #endif /* APR_HAS_ANSI_FS */
- /* Check CreateProcess result
- */
- if (!rv)
- return apr_get_os_error();
- /* XXX Orphaned handle warning - no fix due to broken apr_proc_t api.
- */
- new->hproc = pi.hProcess;
- new->pid = pi.dwProcessId;
- if (attr->child_in) {
- apr_file_close(attr->child_in);
- }
- if (attr->child_out) {
- apr_file_close(attr->child_out);
- }
- if (attr->child_err) {
- apr_file_close(attr->child_err);
- }
- CloseHandle(pi.hThread);
- return APR_SUCCESS;
- }
- APR_DECLARE(apr_status_t) apr_proc_wait_all_procs(apr_proc_t *proc,
- int *exitcode,
- apr_exit_why_e *exitwhy,
- apr_wait_how_e waithow,
- apr_pool_t *p)
- {
- /* Unix does apr_proc_wait(proc(-1), exitcode, exitwhy, waithow)
- * but Win32's apr_proc_wait won't work that way. We can either
- * register all APR created processes in some sort of AsyncWait
- * thread, or simply walk from the global process pool for all
- * apr_pool_note_subprocess()es registered with APR.
- */
- return APR_ENOTIMPL;
- }
- static apr_exit_why_e why_from_exit_code(DWORD exit) {
- /* See WinNT.h STATUS_ACCESS_VIOLATION and family for how
- * this class of failures was determined
- */
- if (((exit & 0xC0000000) == 0xC0000000)
- && !(exit & 0x3FFF0000))
- return APR_PROC_SIGNAL;
- else
- return APR_PROC_EXIT;
- /* ### No way to tell if Dr Watson grabbed a core, AFAICT. */
- }
- APR_DECLARE(apr_status_t) apr_proc_wait(apr_proc_t *proc,
- int *exitcode, apr_exit_why_e *exitwhy,
- apr_wait_how_e waithow)
- {
- DWORD stat;
- DWORD time;
- if (waithow == APR_WAIT)
- time = INFINITE;
- else
- time = 0;
- if ((stat = WaitForSingleObject(proc->hproc, time)) == WAIT_OBJECT_0) {
- if (GetExitCodeProcess(proc->hproc, &stat)) {
- if (exitcode)
- *exitcode = stat;
- if (exitwhy)
- *exitwhy = why_from_exit_code(stat);
- CloseHandle(proc->hproc);
- proc->hproc = NULL;
- return APR_CHILD_DONE;
- }
- }
- else if (stat == WAIT_TIMEOUT) {
- return APR_CHILD_NOTDONE;
- }
- return apr_get_os_error();
- }
- APR_DECLARE(apr_status_t) apr_proc_detach(int daemonize)
- {
- return APR_ENOTIMPL;
- }
|