| /* |
| * Ntdll environment functions |
| * |
| * Copyright 1996, 1998 Alexandre Julliard |
| * Copyright 2003 Eric Pouech |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| */ |
| #include "config.h" |
| |
| #include <assert.h> |
| #include <stdarg.h> |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winreg.h" |
| #include "winternl.h" |
| #include "wine/unicode.h" |
| #include "wine/debug.h" |
| #include "ntdll_misc.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(environ); |
| |
| /****************************************************************************** |
| * RtlCreateEnvironment [NTDLL.@] |
| */ |
| NTSTATUS WINAPI RtlCreateEnvironment(BOOLEAN inherit, PWSTR* env) |
| { |
| NTSTATUS nts; |
| |
| TRACE("(%u,%p)!\n", inherit, env); |
| |
| if (inherit) |
| { |
| MEMORY_BASIC_INFORMATION mbi; |
| |
| RtlAcquirePebLock(); |
| |
| nts = NtQueryVirtualMemory(NtCurrentProcess(), ntdll_get_process_pmts()->Environment, |
| 0, &mbi, sizeof(mbi), NULL); |
| if (nts == STATUS_SUCCESS) |
| { |
| *env = NULL; |
| nts = NtAllocateVirtualMemory(NtCurrentProcess(), (void**)env, 0, &mbi.RegionSize, |
| MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); |
| if (nts == STATUS_SUCCESS) |
| memcpy(*env, ntdll_get_process_pmts()->Environment, mbi.RegionSize); |
| else *env = NULL; |
| } |
| RtlReleasePebLock(); |
| } |
| else |
| { |
| ULONG size = 1; |
| nts = NtAllocateVirtualMemory(NtCurrentProcess(), (void**)env, 0, &size, |
| MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); |
| if (nts == STATUS_SUCCESS) |
| memset(*env, 0, size); |
| } |
| |
| return nts; |
| } |
| |
| /****************************************************************************** |
| * RtlDestroyEnvironment [NTDLL.@] |
| */ |
| NTSTATUS WINAPI RtlDestroyEnvironment(PWSTR env) |
| { |
| ULONG size = 0; |
| |
| TRACE("(%p)!\n", env); |
| |
| return NtFreeVirtualMemory(NtCurrentProcess(), (void**)&env, &size, MEM_RELEASE); |
| } |
| |
| static LPCWSTR ENV_FindVariable(PCWSTR var, PCWSTR name, unsigned namelen) |
| { |
| for (; *var; var += strlenW(var) + 1) |
| { |
| /* match var names, but avoid setting a var with a name including a '=' |
| * (a starting '=' is valid though) |
| */ |
| if (strncmpiW(var, name, namelen) == 0 && var[namelen] == '=' && |
| strchrW(var + 1, '=') == var + namelen) |
| { |
| return var + namelen + 1; |
| } |
| } |
| return NULL; |
| } |
| |
| /****************************************************************** |
| * RtlQueryEnvironmentVariable_U [NTDLL.@] |
| * |
| * NOTES: when the buffer is too small, the string is not written, but if the |
| * terminating null char is the only char that cannot be written, then |
| * all chars (except the null) are written and success is returned |
| * (behavior of Win2k at least) |
| */ |
| NTSTATUS WINAPI RtlQueryEnvironmentVariable_U(PWSTR env, |
| PUNICODE_STRING name, |
| PUNICODE_STRING value) |
| { |
| NTSTATUS nts = STATUS_VARIABLE_NOT_FOUND; |
| PCWSTR var; |
| unsigned namelen; |
| |
| TRACE("%s %s %p\n", debugstr_w(env), debugstr_w(name->Buffer), value); |
| |
| value->Length = 0; |
| namelen = name->Length / sizeof(WCHAR); |
| if (!namelen) return nts; |
| |
| if (!env) |
| { |
| RtlAcquirePebLock(); |
| var = ntdll_get_process_pmts()->Environment; |
| } |
| else var = env; |
| |
| var = ENV_FindVariable(var, name->Buffer, namelen); |
| if (var != NULL) |
| { |
| value->Length = strlenW(var) * sizeof(WCHAR); |
| |
| if (value->Length <= value->MaximumLength) |
| { |
| memmove(value->Buffer, var, min(value->Length + sizeof(WCHAR), value->MaximumLength)); |
| nts = STATUS_SUCCESS; |
| } |
| else nts = STATUS_BUFFER_TOO_SMALL; |
| } |
| |
| if (!env) RtlReleasePebLock(); |
| |
| return nts; |
| } |
| |
| /****************************************************************** |
| * RtlSetCurrentEnvironment [NTDLL.@] |
| * |
| */ |
| void WINAPI RtlSetCurrentEnvironment(PWSTR new_env, PWSTR* old_env) |
| { |
| TRACE("(%p %p)\n", new_env, old_env); |
| |
| RtlAcquirePebLock(); |
| |
| if (old_env) *old_env = ntdll_get_process_pmts()->Environment; |
| ntdll_get_process_pmts()->Environment = new_env; |
| |
| RtlReleasePebLock(); |
| } |
| |
| |
| /****************************************************************************** |
| * RtlSetEnvironmentVariable [NTDLL.@] |
| */ |
| NTSTATUS WINAPI RtlSetEnvironmentVariable(PWSTR* penv, PUNICODE_STRING name, |
| PUNICODE_STRING value) |
| { |
| INT len, old_size; |
| LPWSTR p, env; |
| NTSTATUS nts = STATUS_VARIABLE_NOT_FOUND; |
| MEMORY_BASIC_INFORMATION mbi; |
| |
| TRACE("(%p,%s,%s)\n", |
| penv, debugstr_w(name->Buffer), |
| value ? debugstr_w(value->Buffer) : "--nil--"); |
| |
| if (!name || !name->Buffer || !name->Buffer[0]) |
| return STATUS_INVALID_PARAMETER_1; |
| /* variable names can't contain a '=' except as a first character */ |
| if (strchrW(name->Buffer + 1, '=')) return STATUS_INVALID_PARAMETER; |
| |
| if (!penv) |
| { |
| RtlAcquirePebLock(); |
| env = ntdll_get_process_pmts()->Environment; |
| } else env = *penv; |
| |
| len = name->Length / sizeof(WCHAR); |
| |
| /* compute current size of environment */ |
| for (p = env; *p; p += strlenW(p) + 1); |
| old_size = p + 1 - env; |
| |
| /* Find a place to insert the string */ |
| for (p = env; *p; p += strlenW(p) + 1) |
| { |
| if (!strncmpiW(name->Buffer, p, len) && (p[len] == '=')) break; |
| } |
| if (!value && !*p) goto done; /* Value to remove doesn't exist */ |
| |
| /* Realloc the buffer */ |
| len = value ? len + value->Length / sizeof(WCHAR) + 2 : 0; |
| if (*p) len -= strlenW(p) + 1; /* The name already exists */ |
| |
| if (len < 0) |
| { |
| LPWSTR next = p + strlenW(p) + 1; /* We know there is a next one */ |
| memmove(next + len, next, (old_size - (next - env)) * sizeof(WCHAR)); |
| } |
| |
| nts = NtQueryVirtualMemory(NtCurrentProcess(), env, 0, |
| &mbi, sizeof(mbi), NULL); |
| if (nts != STATUS_SUCCESS) goto done; |
| |
| if ((old_size + len) * sizeof(WCHAR) > mbi.RegionSize) |
| { |
| LPWSTR new_env; |
| ULONG new_size = (old_size + len) * sizeof(WCHAR); |
| |
| new_env = NULL; |
| nts = NtAllocateVirtualMemory(NtCurrentProcess(), (void**)&new_env, 0, |
| &new_size, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); |
| if (nts != STATUS_SUCCESS) goto done; |
| |
| memmove(new_env, env, (p - env) * sizeof(WCHAR)); |
| assert(len > 0); |
| memmove(new_env + (p - env) + len, p, (old_size - (p - env)) * sizeof(WCHAR)); |
| p = new_env + (p - env); |
| |
| RtlDestroyEnvironment(env); |
| if (!penv) ntdll_get_process_pmts()->Environment = new_env; |
| else *penv = new_env; |
| env = new_env; |
| } |
| else |
| { |
| if (len > 0) memmove(p + len, p, (old_size - (p - env)) * sizeof(WCHAR)); |
| } |
| |
| /* Set the new string */ |
| if (value) |
| { |
| static const WCHAR equalW[] = {'=',0}; |
| |
| strcpyW(p, name->Buffer); |
| strcatW(p, equalW); |
| strcatW(p, value->Buffer); |
| } |
| done: |
| if (!penv) RtlReleasePebLock(); |
| |
| return nts; |
| } |
| |
| /****************************************************************** |
| * RtlExpandEnvironmentStrings_U (NTDLL.@) |
| * |
| */ |
| NTSTATUS WINAPI RtlExpandEnvironmentStrings_U(PWSTR renv, const UNICODE_STRING* us_src, |
| PUNICODE_STRING us_dst, PULONG plen) |
| { |
| DWORD len, count, total_size = 1; /* 1 for terminating '\0' */ |
| LPCWSTR env, src, p, var; |
| LPWSTR dst; |
| |
| src = us_src->Buffer; |
| count = us_dst->MaximumLength / sizeof(WCHAR); |
| dst = count ? us_dst->Buffer : NULL; |
| |
| if (!renv) |
| { |
| RtlAcquirePebLock(); |
| env = ntdll_get_process_pmts()->Environment; |
| } |
| else env = renv; |
| |
| while (*src) |
| { |
| if (*src != '%') |
| { |
| if ((p = strchrW( src, '%' ))) len = p - src; |
| else len = strlenW(src); |
| var = src; |
| src += len; |
| } |
| else /* we are at the start of a variable */ |
| { |
| if ((p = strchrW( src + 1, '%' ))) |
| { |
| len = p - src - 1; /* Length of the variable name */ |
| if ((var = ENV_FindVariable( env, src + 1, len ))) |
| { |
| src += len + 2; /* Skip the variable name */ |
| len = strlenW(var); |
| } |
| else |
| { |
| var = src; /* Copy original name instead */ |
| len += 2; |
| src += len; |
| } |
| } |
| else /* unfinished variable name, ignore it */ |
| { |
| var = src; |
| len = strlenW(src); /* Copy whole string */ |
| src += len; |
| } |
| } |
| total_size += len; |
| if (dst) |
| { |
| if (count < len) len = count; |
| memcpy(dst, var, len * sizeof(WCHAR)); |
| count -= len; |
| dst += len; |
| } |
| } |
| |
| if (!renv) RtlReleasePebLock(); |
| |
| /* Null-terminate the string */ |
| if (dst && count) *dst = '\0'; |
| |
| us_dst->Length = (dst) ? (dst - us_dst->Buffer) * sizeof(WCHAR) : 0; |
| if (plen) *plen = total_size * sizeof(WCHAR); |
| |
| return (count) ? STATUS_SUCCESS : STATUS_BUFFER_TOO_SMALL; |
| } |
| |
| |
| static inline void normalize( void *base, WCHAR **ptr ) |
| { |
| if (*ptr) *ptr = (WCHAR *)((char *)base + (UINT_PTR)*ptr); |
| } |
| |
| /****************************************************************************** |
| * RtlNormalizeProcessParams [NTDLL.@] |
| */ |
| PRTL_USER_PROCESS_PARAMETERS WINAPI RtlNormalizeProcessParams( RTL_USER_PROCESS_PARAMETERS *params ) |
| { |
| if (params && !(params->Flags & PROCESS_PARAMS_FLAG_NORMALIZED)) |
| { |
| normalize( params, ¶ms->CurrentDirectoryName.Buffer ); |
| normalize( params, ¶ms->DllPath.Buffer ); |
| normalize( params, ¶ms->ImagePathName.Buffer ); |
| normalize( params, ¶ms->CommandLine.Buffer ); |
| normalize( params, ¶ms->WindowTitle.Buffer ); |
| normalize( params, ¶ms->Desktop.Buffer ); |
| normalize( params, ¶ms->ShellInfo.Buffer ); |
| normalize( params, ¶ms->RuntimeInfo.Buffer ); |
| params->Flags |= PROCESS_PARAMS_FLAG_NORMALIZED; |
| } |
| return params; |
| } |
| |
| |
| static inline void denormalize( void *base, WCHAR **ptr ) |
| { |
| if (*ptr) *ptr = (WCHAR *)(UINT_PTR)((char *)*ptr - (char *)base); |
| } |
| |
| /****************************************************************************** |
| * RtlDeNormalizeProcessParams [NTDLL.@] |
| */ |
| PRTL_USER_PROCESS_PARAMETERS WINAPI RtlDeNormalizeProcessParams( RTL_USER_PROCESS_PARAMETERS *params ) |
| { |
| if (params && (params->Flags & PROCESS_PARAMS_FLAG_NORMALIZED)) |
| { |
| denormalize( params, ¶ms->CurrentDirectoryName.Buffer ); |
| denormalize( params, ¶ms->DllPath.Buffer ); |
| denormalize( params, ¶ms->ImagePathName.Buffer ); |
| denormalize( params, ¶ms->CommandLine.Buffer ); |
| denormalize( params, ¶ms->WindowTitle.Buffer ); |
| denormalize( params, ¶ms->Desktop.Buffer ); |
| denormalize( params, ¶ms->ShellInfo.Buffer ); |
| denormalize( params, ¶ms->RuntimeInfo.Buffer ); |
| params->Flags &= ~PROCESS_PARAMS_FLAG_NORMALIZED; |
| } |
| return params; |
| } |
| |
| |
| /* append a unicode string to the process params data; helper for RtlCreateProcessParameters */ |
| static void append_unicode_string( void **data, const UNICODE_STRING *src, |
| UNICODE_STRING *dst ) |
| { |
| dst->Length = src->Length; |
| dst->MaximumLength = src->MaximumLength; |
| dst->Buffer = *data; |
| memcpy( dst->Buffer, src->Buffer, dst->MaximumLength ); |
| *data = (char *)dst->Buffer + dst->MaximumLength; |
| } |
| |
| |
| /****************************************************************************** |
| * RtlCreateProcessParameters [NTDLL.@] |
| */ |
| NTSTATUS WINAPI RtlCreateProcessParameters( RTL_USER_PROCESS_PARAMETERS **result, |
| const UNICODE_STRING *ImagePathName, |
| const UNICODE_STRING *DllPath, |
| const UNICODE_STRING *CurrentDirectoryName, |
| const UNICODE_STRING *CommandLine, |
| PWSTR Environment, |
| const UNICODE_STRING *WindowTitle, |
| const UNICODE_STRING *Desktop, |
| const UNICODE_STRING *ShellInfo, |
| const UNICODE_STRING *RuntimeInfo ) |
| { |
| static const WCHAR empty[] = {0}; |
| static const UNICODE_STRING empty_str = { 0, sizeof(empty), (WCHAR *)empty }; |
| |
| const RTL_USER_PROCESS_PARAMETERS *cur_params; |
| ULONG size, total_size; |
| void *ptr; |
| NTSTATUS status; |
| |
| RtlAcquirePebLock(); |
| cur_params = NtCurrentTeb()->Peb->ProcessParameters; |
| if (!DllPath) DllPath = &cur_params->DllPath; |
| if (!CurrentDirectoryName) CurrentDirectoryName = &cur_params->CurrentDirectoryName; |
| if (!CommandLine) CommandLine = ImagePathName; |
| if (!Environment) Environment = cur_params->Environment; |
| if (!WindowTitle) WindowTitle = &empty_str; |
| if (!Desktop) Desktop = &empty_str; |
| if (!ShellInfo) ShellInfo = &empty_str; |
| if (!RuntimeInfo) RuntimeInfo = &empty_str; |
| |
| size = (sizeof(RTL_USER_PROCESS_PARAMETERS) |
| + ImagePathName->MaximumLength |
| + DllPath->MaximumLength |
| + CurrentDirectoryName->MaximumLength |
| + CommandLine->MaximumLength |
| + WindowTitle->MaximumLength |
| + Desktop->MaximumLength |
| + ShellInfo->MaximumLength |
| + RuntimeInfo->MaximumLength); |
| |
| total_size = size; |
| if ((status = NtAllocateVirtualMemory( NtCurrentProcess(), &ptr, NULL, &total_size, |
| MEM_COMMIT, PAGE_READWRITE )) == STATUS_SUCCESS) |
| { |
| RTL_USER_PROCESS_PARAMETERS *params = ptr; |
| params->AllocationSize = total_size; |
| params->Size = size; |
| params->Flags = PROCESS_PARAMS_FLAG_NORMALIZED; |
| params->ProcessGroup = cur_params->ProcessGroup; |
| params->Environment = Environment; |
| /* all other fields are zero */ |
| |
| ptr = params + 1; |
| append_unicode_string( &ptr, CurrentDirectoryName, ¶ms->CurrentDirectoryName ); |
| append_unicode_string( &ptr, DllPath, ¶ms->DllPath ); |
| append_unicode_string( &ptr, ImagePathName, ¶ms->ImagePathName ); |
| append_unicode_string( &ptr, CommandLine, ¶ms->CommandLine ); |
| append_unicode_string( &ptr, WindowTitle, ¶ms->WindowTitle ); |
| append_unicode_string( &ptr, Desktop, ¶ms->Desktop ); |
| append_unicode_string( &ptr, ShellInfo, ¶ms->ShellInfo ); |
| append_unicode_string( &ptr, RuntimeInfo, ¶ms->RuntimeInfo ); |
| *result = RtlDeNormalizeProcessParams( params ); |
| } |
| RtlReleasePebLock(); |
| return status; |
| } |
| |
| |
| /****************************************************************************** |
| * RtlDestroyProcessParameters [NTDLL.@] |
| */ |
| void WINAPI RtlDestroyProcessParameters( RTL_USER_PROCESS_PARAMETERS *params ) |
| { |
| void *ptr = params; |
| ULONG size = 0; |
| NtFreeVirtualMemory( NtCurrentProcess(), &ptr, &size, MEM_RELEASE ); |
| } |
| |
| |
| /*********************************************************************** |
| * build_environment |
| * |
| * Build the Win32 environment from the Unix environment |
| */ |
| static NTSTATUS build_initial_environment(void) |
| { |
| extern char **environ; |
| LPSTR* e, te; |
| LPWSTR p; |
| ULONG size; |
| NTSTATUS nts; |
| int len; |
| |
| /* Compute the total size of the Unix environment */ |
| size = sizeof(BYTE); |
| for (e = environ; *e; e++) |
| { |
| if (!memcmp(*e, "PATH=", 5)) continue; |
| size += strlen(*e) + 1; |
| } |
| size *= sizeof(WCHAR); |
| |
| /* Now allocate the environment */ |
| nts = NtAllocateVirtualMemory(NtCurrentProcess(), (void**)&p, 0, &size, |
| MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); |
| if (nts != STATUS_SUCCESS) return nts; |
| |
| ntdll_get_process_pmts()->Environment = p; |
| |
| /* And fill it with the Unix environment */ |
| for (e = environ; *e; e++) |
| { |
| /* skip Unix PATH and store WINEPATH as PATH */ |
| if (!memcmp(*e, "PATH=", 5)) continue; |
| if (!memcmp(*e, "WINEPATH=", 9 )) te = *e + 4; else te = *e; |
| len = strlen(te); |
| RtlMultiByteToUnicodeN(p, len * sizeof(WCHAR), NULL, te, len); |
| p[len] = 0; |
| p += len + 1; |
| } |
| *p = 0; |
| |
| return STATUS_SUCCESS; |
| } |
| |
| static void init_unicode( UNICODE_STRING* us, const char** src, size_t len) |
| { |
| if (len) |
| { |
| STRING ansi; |
| ansi.Buffer = (char*)*src; |
| ansi.Length = len; |
| ansi.MaximumLength = len; |
| /* FIXME: should check value returned */ |
| RtlAnsiStringToUnicodeString( us, &ansi, TRUE ); |
| *src += len; |
| } |
| } |
| |
| /*********************************************************************** |
| * init_user_process_pmts |
| * |
| * Fill the RTL_USER_PROCESS_PARAMETERS structure from the server. |
| */ |
| BOOL init_user_process_pmts( size_t info_size ) |
| { |
| startup_info_t info; |
| void *data; |
| const char *src; |
| size_t len; |
| RTL_USER_PROCESS_PARAMETERS *rupp; |
| |
| if (build_initial_environment() != STATUS_SUCCESS) return FALSE; |
| if (!info_size) return TRUE; |
| if (!(data = RtlAllocateHeap( ntdll_get_process_heap(), 0, info_size ))) |
| return FALSE; |
| |
| SERVER_START_REQ( get_startup_info ) |
| { |
| wine_server_set_reply( req, data, info_size ); |
| wine_server_call( req ); |
| info_size = wine_server_reply_size( reply ); |
| } |
| SERVER_END_REQ; |
| |
| if (info_size < sizeof(info.size)) goto done; |
| len = min( info_size, ((startup_info_t *)data)->size ); |
| memset( &info, 0, sizeof(info) ); |
| memcpy( &info, data, len ); |
| src = (char *)data + len; |
| info_size -= len; |
| |
| /* fixup the lengths */ |
| if (info.filename_len > info_size) info.filename_len = info_size; |
| info_size -= info.filename_len; |
| if (info.cmdline_len > info_size) info.cmdline_len = info_size; |
| info_size -= info.cmdline_len; |
| if (info.desktop_len > info_size) info.desktop_len = info_size; |
| info_size -= info.desktop_len; |
| if (info.title_len > info_size) info.title_len = info_size; |
| |
| rupp = NtCurrentTeb()->Peb->ProcessParameters; |
| |
| init_unicode( &rupp->ImagePathName, &src, info.filename_len ); |
| init_unicode( &rupp->CommandLine, &src, info.cmdline_len ); |
| init_unicode( &rupp->Desktop, &src, info.desktop_len ); |
| init_unicode( &rupp->WindowTitle, &src, info.title_len ); |
| |
| rupp->dwX = info.x; |
| rupp->dwY = info.y; |
| rupp->dwXSize = info.cx; |
| rupp->dwYSize = info.cy; |
| rupp->dwXCountChars = info.x_chars; |
| rupp->dwYCountChars = info.y_chars; |
| rupp->dwFillAttribute = info.attribute; |
| rupp->wShowWindow = info.cmd_show; |
| rupp->dwFlags = info.flags; |
| |
| done: |
| RtlFreeHeap( ntdll_get_process_heap(), 0, data ); |
| return TRUE; |
| } |
| |
| /*********************************************************************** |
| * set_library_argv |
| * |
| * Set the Wine library argc/argv global variables. |
| */ |
| static void set_library_argv( char **argv ) |
| { |
| int argc; |
| WCHAR *p; |
| WCHAR **wargv; |
| DWORD total = 0, len, reslen; |
| |
| for (argc = 0; argv[argc]; argc++) |
| { |
| len = strlen(argv[argc]) + 1; |
| RtlMultiByteToUnicodeN(NULL, 0, &reslen, argv[argc], len); |
| total += reslen; |
| } |
| wargv = RtlAllocateHeap( ntdll_get_process_heap(), 0, |
| total + (argc + 1) * sizeof(*wargv) ); |
| p = (WCHAR *)(wargv + argc + 1); |
| for (argc = 0; argv[argc]; argc++) |
| { |
| len = strlen(argv[argc]) + 1; |
| RtlMultiByteToUnicodeN(p, total, &reslen, argv[argc], len); |
| wargv[argc] = p; |
| p += reslen / sizeof(WCHAR); |
| total -= reslen; |
| } |
| wargv[argc] = NULL; |
| |
| __wine_main_argc = argc; |
| __wine_main_argv = argv; |
| __wine_main_wargv = wargv; |
| } |
| |
| /*********************************************************************** |
| * build_command_line |
| * |
| * Build the command line of a process from the argv array. |
| * |
| * Note that it does NOT necessarily include the file name. |
| * Sometimes we don't even have any command line options at all. |
| * |
| * We must quote and escape characters so that the argv array can be rebuilt |
| * from the command line: |
| * - spaces and tabs must be quoted |
| * 'a b' -> '"a b"' |
| * - quotes must be escaped |
| * '"' -> '\"' |
| * - if '\'s are followed by a '"', they must be doubled and followed by '\"', |
| * resulting in an odd number of '\' followed by a '"' |
| * '\"' -> '\\\"' |
| * '\\"' -> '\\\\\"' |
| * - '\'s that are not followed by a '"' can be left as is |
| * 'a\b' == 'a\b' |
| * 'a\\b' == 'a\\b' |
| */ |
| BOOL build_command_line( char **argv ) |
| { |
| int len; |
| char **arg; |
| LPWSTR p; |
| RTL_USER_PROCESS_PARAMETERS* rupp; |
| |
| set_library_argv( argv ); |
| |
| rupp = ntdll_get_process_pmts(); |
| if (rupp->CommandLine.Buffer) return TRUE; /* already got it from the server */ |
| |
| len = 0; |
| for (arg = argv; *arg; arg++) |
| { |
| int has_space,bcount; |
| char* a; |
| |
| has_space=0; |
| bcount=0; |
| a=*arg; |
| if( !*a ) has_space=1; |
| while (*a!='\0') { |
| if (*a=='\\') { |
| bcount++; |
| } else { |
| if (*a==' ' || *a=='\t') { |
| has_space=1; |
| } else if (*a=='"') { |
| /* doubling of '\' preceeding a '"', |
| * plus escaping of said '"' |
| */ |
| len+=2*bcount+1; |
| } |
| bcount=0; |
| } |
| a++; |
| } |
| len+=(a-*arg)+1 /* for the separating space */; |
| if (has_space) |
| len+=2; /* for the quotes */ |
| } |
| |
| if (!(rupp->CommandLine.Buffer = RtlAllocateHeap( ntdll_get_process_heap(), 0, len * sizeof(WCHAR)))) |
| return FALSE; |
| |
| p = rupp->CommandLine.Buffer; |
| rupp->CommandLine.Length = (len - 1) * sizeof(WCHAR); |
| rupp->CommandLine.MaximumLength = len * sizeof(WCHAR); |
| for (arg = argv; *arg; arg++) |
| { |
| int has_space,has_quote; |
| char* a; |
| |
| /* Check for quotes and spaces in this argument */ |
| has_space=has_quote=0; |
| a=*arg; |
| if( !*a ) has_space=1; |
| while (*a!='\0') { |
| if (*a==' ' || *a=='\t') { |
| has_space=1; |
| if (has_quote) |
| break; |
| } else if (*a=='"') { |
| has_quote=1; |
| if (has_space) |
| break; |
| } |
| a++; |
| } |
| |
| /* Now transfer it to the command line */ |
| if (has_space) |
| *p++='"'; |
| if (has_quote) { |
| int bcount; |
| char* a; |
| |
| bcount=0; |
| a=*arg; |
| while (*a!='\0') { |
| if (*a=='\\') { |
| *p++=*a; |
| bcount++; |
| } else { |
| if (*a=='"') { |
| int i; |
| |
| /* Double all the '\\' preceeding this '"', plus one */ |
| for (i=0;i<=bcount;i++) |
| *p++='\\'; |
| *p++='"'; |
| } else { |
| *p++=*a; |
| } |
| bcount=0; |
| } |
| a++; |
| } |
| } else { |
| char* x = *arg; |
| while ((*p=*x++)) p++; |
| } |
| if (has_space) |
| *p++='"'; |
| *p++=' '; |
| } |
| if (p > rupp->CommandLine.Buffer) |
| p--; /* remove last space */ |
| *p = '\0'; |
| |
| return TRUE; |
| } |