| /* |
| * Process environment management |
| * |
| * Copyright 1996, 1998 Alexandre Julliard |
| * |
| * 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include "config.h" |
| #include "wine/port.h" |
| |
| #include <stdarg.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <assert.h> |
| |
| #include "ntstatus.h" |
| #define WIN32_NO_STATUS |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winerror.h" |
| #include "wine/library.h" |
| #include "winternl.h" |
| #include "wine/unicode.h" |
| #include "wine/debug.h" |
| |
| #include "kernel_private.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(environ); |
| |
| /* Notes: |
| * - contrary to Microsoft docs, the environment strings do not appear |
| * to be sorted on Win95 (although they are on NT); so we don't bother |
| * to sort them either. |
| */ |
| |
| static STARTUPINFOW startup_infoW; |
| static STARTUPINFOA startup_infoA; |
| |
| |
| /*********************************************************************** |
| * GetCommandLineA (KERNEL32.@) |
| * |
| * WARNING: there's a Windows incompatibility lurking here ! |
| * Win32s always includes the full path of the program file, |
| * whereas Windows NT only returns the full file path plus arguments |
| * in case the program has been started with a full path. |
| * Win9x seems to have inherited NT behaviour. |
| * |
| * Note that both Start Menu Execute and Explorer start programs with |
| * fully specified quoted app file paths, which is why probably the only case |
| * where you'll see single file names is in case of direct launch |
| * via CreateProcess or WinExec. |
| * |
| * Perhaps we should take care of Win3.1 programs here (Win32s "feature"). |
| * |
| * References: MS KB article q102762.txt (special Win32s handling) |
| */ |
| LPSTR WINAPI GetCommandLineA(void) |
| { |
| static char *cmdlineA; /* ASCII command line */ |
| |
| if (!cmdlineA) /* make an ansi version if we don't have it */ |
| { |
| ANSI_STRING ansi; |
| RtlAcquirePebLock(); |
| |
| cmdlineA = (RtlUnicodeStringToAnsiString( &ansi, &NtCurrentTeb()->Peb->ProcessParameters->CommandLine, TRUE) == STATUS_SUCCESS) ? |
| ansi.Buffer : NULL; |
| RtlReleasePebLock(); |
| } |
| return cmdlineA; |
| } |
| |
| /*********************************************************************** |
| * GetCommandLineW (KERNEL32.@) |
| */ |
| LPWSTR WINAPI GetCommandLineW(void) |
| { |
| return NtCurrentTeb()->Peb->ProcessParameters->CommandLine.Buffer; |
| } |
| |
| |
| /*********************************************************************** |
| * GetEnvironmentStringsA (KERNEL32.@) |
| * GetEnvironmentStrings (KERNEL32.@) |
| */ |
| LPSTR WINAPI GetEnvironmentStringsA(void) |
| { |
| LPWSTR ptrW; |
| unsigned len, slen; |
| LPSTR ret, ptrA; |
| |
| RtlAcquirePebLock(); |
| |
| len = 1; |
| |
| ptrW = NtCurrentTeb()->Peb->ProcessParameters->Environment; |
| while (*ptrW) |
| { |
| slen = strlenW(ptrW) + 1; |
| len += WideCharToMultiByte( CP_ACP, 0, ptrW, slen, NULL, 0, NULL, NULL ); |
| ptrW += slen; |
| } |
| |
| if ((ret = HeapAlloc( GetProcessHeap(), 0, len )) != NULL) |
| { |
| ptrW = NtCurrentTeb()->Peb->ProcessParameters->Environment; |
| ptrA = ret; |
| while (*ptrW) |
| { |
| slen = strlenW(ptrW) + 1; |
| WideCharToMultiByte( CP_ACP, 0, ptrW, slen, ptrA, len, NULL, NULL ); |
| ptrW += slen; |
| ptrA += strlen(ptrA) + 1; |
| } |
| *ptrA = 0; |
| } |
| |
| RtlReleasePebLock(); |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * GetEnvironmentStringsW (KERNEL32.@) |
| */ |
| LPWSTR WINAPI GetEnvironmentStringsW(void) |
| { |
| return NtCurrentTeb()->Peb->ProcessParameters->Environment; |
| } |
| |
| |
| /*********************************************************************** |
| * FreeEnvironmentStringsA (KERNEL32.@) |
| */ |
| BOOL WINAPI FreeEnvironmentStringsA( LPSTR ptr ) |
| { |
| return HeapFree( GetProcessHeap(), 0, ptr ); |
| } |
| |
| |
| /*********************************************************************** |
| * FreeEnvironmentStringsW (KERNEL32.@) |
| */ |
| BOOL WINAPI FreeEnvironmentStringsW( LPWSTR ptr ) |
| { |
| return TRUE; |
| } |
| |
| |
| /*********************************************************************** |
| * GetEnvironmentVariableA (KERNEL32.@) |
| */ |
| DWORD WINAPI GetEnvironmentVariableA( LPCSTR name, LPSTR value, DWORD size ) |
| { |
| UNICODE_STRING us_name; |
| PWSTR valueW; |
| DWORD ret; |
| |
| if (!name || !*name) |
| { |
| SetLastError(ERROR_ENVVAR_NOT_FOUND); |
| return 0; |
| } |
| |
| if (!(valueW = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR)))) |
| return 0; |
| |
| RtlCreateUnicodeStringFromAsciiz( &us_name, name ); |
| SetLastError(0); |
| ret = GetEnvironmentVariableW( us_name.Buffer, valueW, size); |
| if (ret && ret < size) |
| { |
| WideCharToMultiByte( CP_ACP, 0, valueW, ret + 1, value, size, NULL, NULL ); |
| } |
| /* this is needed to tell, with 0 as a return value, the difference between: |
| * - an error (GetLastError() != 0) |
| * - returning an empty string (in this case, we need to update the buffer) |
| */ |
| if (ret == 0 && size && GetLastError() == 0) |
| value[0] = '\0'; |
| |
| RtlFreeUnicodeString( &us_name ); |
| HeapFree(GetProcessHeap(), 0, valueW); |
| |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * GetEnvironmentVariableW (KERNEL32.@) |
| */ |
| DWORD WINAPI GetEnvironmentVariableW( LPCWSTR name, LPWSTR val, DWORD size ) |
| { |
| UNICODE_STRING us_name; |
| UNICODE_STRING us_value; |
| NTSTATUS status; |
| unsigned len; |
| |
| TRACE("(%s %p %u)\n", debugstr_w(name), val, size); |
| |
| if (!name || !*name) |
| { |
| SetLastError(ERROR_ENVVAR_NOT_FOUND); |
| return 0; |
| } |
| |
| RtlInitUnicodeString(&us_name, name); |
| us_value.Length = 0; |
| us_value.MaximumLength = (size ? size - 1 : 0) * sizeof(WCHAR); |
| us_value.Buffer = val; |
| |
| status = RtlQueryEnvironmentVariable_U(NULL, &us_name, &us_value); |
| len = us_value.Length / sizeof(WCHAR); |
| if (status != STATUS_SUCCESS) |
| { |
| SetLastError( RtlNtStatusToDosError(status) ); |
| return (status == STATUS_BUFFER_TOO_SMALL) ? len + 1 : 0; |
| } |
| if (size) val[len] = '\0'; |
| |
| return us_value.Length / sizeof(WCHAR); |
| } |
| |
| |
| /*********************************************************************** |
| * SetEnvironmentVariableA (KERNEL32.@) |
| */ |
| BOOL WINAPI SetEnvironmentVariableA( LPCSTR name, LPCSTR value ) |
| { |
| UNICODE_STRING us_name; |
| BOOL ret; |
| |
| if (!name) |
| { |
| SetLastError(ERROR_ENVVAR_NOT_FOUND); |
| return 0; |
| } |
| |
| RtlCreateUnicodeStringFromAsciiz( &us_name, name ); |
| if (value) |
| { |
| UNICODE_STRING us_value; |
| |
| RtlCreateUnicodeStringFromAsciiz( &us_value, value ); |
| ret = SetEnvironmentVariableW( us_name.Buffer, us_value.Buffer ); |
| RtlFreeUnicodeString( &us_value ); |
| } |
| else ret = SetEnvironmentVariableW( us_name.Buffer, NULL ); |
| |
| RtlFreeUnicodeString( &us_name ); |
| |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * SetEnvironmentVariableW (KERNEL32.@) |
| */ |
| BOOL WINAPI SetEnvironmentVariableW( LPCWSTR name, LPCWSTR value ) |
| { |
| UNICODE_STRING us_name; |
| NTSTATUS status; |
| |
| TRACE("(%s %s)\n", debugstr_w(name), debugstr_w(value)); |
| |
| if (!name) |
| { |
| SetLastError(ERROR_ENVVAR_NOT_FOUND); |
| return 0; |
| } |
| |
| RtlInitUnicodeString(&us_name, name); |
| if (value) |
| { |
| UNICODE_STRING us_value; |
| |
| RtlInitUnicodeString(&us_value, value); |
| status = RtlSetEnvironmentVariable(NULL, &us_name, &us_value); |
| } |
| else status = RtlSetEnvironmentVariable(NULL, &us_name, NULL); |
| |
| if (status != STATUS_SUCCESS) |
| { |
| SetLastError( RtlNtStatusToDosError(status) ); |
| return FALSE; |
| } |
| return TRUE; |
| } |
| |
| |
| /*********************************************************************** |
| * ExpandEnvironmentStringsA (KERNEL32.@) |
| * |
| * Note: overlapping buffers are not supported; this is how it should be. |
| * FIXME: return value is wrong for MBCS |
| */ |
| DWORD WINAPI ExpandEnvironmentStringsA( LPCSTR src, LPSTR dst, DWORD count ) |
| { |
| UNICODE_STRING us_src; |
| PWSTR dstW = NULL; |
| DWORD ret; |
| |
| RtlCreateUnicodeStringFromAsciiz( &us_src, src ); |
| if (count) |
| { |
| if (!(dstW = HeapAlloc(GetProcessHeap(), 0, count * sizeof(WCHAR)))) |
| return 0; |
| ret = ExpandEnvironmentStringsW( us_src.Buffer, dstW, count); |
| if (ret) |
| WideCharToMultiByte( CP_ACP, 0, dstW, ret, dst, count, NULL, NULL ); |
| } |
| else ret = ExpandEnvironmentStringsW( us_src.Buffer, NULL, 0); |
| |
| RtlFreeUnicodeString( &us_src ); |
| HeapFree(GetProcessHeap(), 0, dstW); |
| |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * ExpandEnvironmentStringsW (KERNEL32.@) |
| */ |
| DWORD WINAPI ExpandEnvironmentStringsW( LPCWSTR src, LPWSTR dst, DWORD len ) |
| { |
| UNICODE_STRING us_src; |
| UNICODE_STRING us_dst; |
| NTSTATUS status; |
| DWORD res; |
| |
| TRACE("(%s %p %u)\n", debugstr_w(src), dst, len); |
| |
| RtlInitUnicodeString(&us_src, src); |
| |
| /* make sure we don't overflow maximum UNICODE_STRING size */ |
| if (len > 0x7fff) |
| len = 0x7fff; |
| |
| us_dst.Length = 0; |
| us_dst.MaximumLength = len * sizeof(WCHAR); |
| us_dst.Buffer = dst; |
| |
| res = 0; |
| status = RtlExpandEnvironmentStrings_U(NULL, &us_src, &us_dst, &res); |
| res /= sizeof(WCHAR); |
| if (status != STATUS_SUCCESS) |
| { |
| SetLastError( RtlNtStatusToDosError(status) ); |
| if (status != STATUS_BUFFER_TOO_SMALL) return 0; |
| if (len && dst) dst[len - 1] = '\0'; |
| } |
| |
| return res; |
| } |
| |
| |
| /*********************************************************************** |
| * GetStdHandle (KERNEL32.@) |
| */ |
| HANDLE WINAPI GetStdHandle( DWORD std_handle ) |
| { |
| switch (std_handle) |
| { |
| case STD_INPUT_HANDLE: return NtCurrentTeb()->Peb->ProcessParameters->hStdInput; |
| case STD_OUTPUT_HANDLE: return NtCurrentTeb()->Peb->ProcessParameters->hStdOutput; |
| case STD_ERROR_HANDLE: return NtCurrentTeb()->Peb->ProcessParameters->hStdError; |
| } |
| SetLastError( ERROR_INVALID_PARAMETER ); |
| return INVALID_HANDLE_VALUE; |
| } |
| |
| |
| /*********************************************************************** |
| * SetStdHandle (KERNEL32.@) |
| */ |
| BOOL WINAPI SetStdHandle( DWORD std_handle, HANDLE handle ) |
| { |
| switch (std_handle) |
| { |
| case STD_INPUT_HANDLE: NtCurrentTeb()->Peb->ProcessParameters->hStdInput = handle; return TRUE; |
| case STD_OUTPUT_HANDLE: NtCurrentTeb()->Peb->ProcessParameters->hStdOutput = handle; return TRUE; |
| case STD_ERROR_HANDLE: NtCurrentTeb()->Peb->ProcessParameters->hStdError = handle; return TRUE; |
| } |
| SetLastError( ERROR_INVALID_PARAMETER ); |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * GetStartupInfoA (KERNEL32.@) |
| */ |
| VOID WINAPI GetStartupInfoA( LPSTARTUPINFOA info ) |
| { |
| memcpy(info, &startup_infoA, sizeof(startup_infoA)); |
| } |
| |
| |
| /*********************************************************************** |
| * GetStartupInfoW (KERNEL32.@) |
| */ |
| VOID WINAPI GetStartupInfoW( LPSTARTUPINFOW info ) |
| { |
| memcpy(info, &startup_infoW, sizeof(startup_infoW)); |
| } |
| |
| /****************************************************************** |
| * ENV_CopyStartupInformation (internal) |
| * |
| * Creates the STARTUPINFO information from the ntdll information |
| */ |
| void ENV_CopyStartupInformation(void) |
| { |
| RTL_USER_PROCESS_PARAMETERS* rupp; |
| ANSI_STRING ansi; |
| |
| RtlAcquirePebLock(); |
| |
| rupp = NtCurrentTeb()->Peb->ProcessParameters; |
| |
| startup_infoW.cb = sizeof(startup_infoW); |
| startup_infoW.lpReserved = NULL; |
| startup_infoW.lpDesktop = rupp->Desktop.Buffer; |
| startup_infoW.lpTitle = rupp->WindowTitle.Buffer; |
| startup_infoW.dwX = rupp->dwX; |
| startup_infoW.dwY = rupp->dwY; |
| startup_infoW.dwXSize = rupp->dwXSize; |
| startup_infoW.dwYSize = rupp->dwYSize; |
| startup_infoW.dwXCountChars = rupp->dwXCountChars; |
| startup_infoW.dwYCountChars = rupp->dwYCountChars; |
| startup_infoW.dwFillAttribute = rupp->dwFillAttribute; |
| startup_infoW.dwFlags = rupp->dwFlags; |
| startup_infoW.wShowWindow = rupp->wShowWindow; |
| startup_infoW.cbReserved2 = rupp->RuntimeInfo.MaximumLength; |
| startup_infoW.lpReserved2 = rupp->RuntimeInfo.MaximumLength ? (void*)rupp->RuntimeInfo.Buffer : NULL; |
| startup_infoW.hStdInput = rupp->hStdInput; |
| startup_infoW.hStdOutput = rupp->hStdOutput; |
| startup_infoW.hStdError = rupp->hStdError; |
| |
| startup_infoA.cb = sizeof(startup_infoA); |
| startup_infoA.lpReserved = NULL; |
| startup_infoA.lpDesktop = (rupp->Desktop.Length && |
| RtlUnicodeStringToAnsiString( &ansi, &rupp->Desktop, TRUE) == STATUS_SUCCESS) ? |
| ansi.Buffer : NULL; |
| startup_infoA.lpTitle = (rupp->WindowTitle.Length && |
| RtlUnicodeStringToAnsiString( &ansi, &rupp->WindowTitle, TRUE) == STATUS_SUCCESS) ? |
| ansi.Buffer : NULL; |
| startup_infoA.dwX = rupp->dwX; |
| startup_infoA.dwY = rupp->dwY; |
| startup_infoA.dwXSize = rupp->dwXSize; |
| startup_infoA.dwYSize = rupp->dwYSize; |
| startup_infoA.dwXCountChars = rupp->dwXCountChars; |
| startup_infoA.dwYCountChars = rupp->dwYCountChars; |
| startup_infoA.dwFillAttribute = rupp->dwFillAttribute; |
| startup_infoA.dwFlags = rupp->dwFlags; |
| startup_infoA.wShowWindow = rupp->wShowWindow; |
| startup_infoA.cbReserved2 = rupp->RuntimeInfo.MaximumLength; |
| startup_infoA.lpReserved2 = rupp->RuntimeInfo.MaximumLength ? (void*)rupp->RuntimeInfo.Buffer : NULL; |
| startup_infoA.hStdInput = rupp->hStdInput; |
| startup_infoA.hStdOutput = rupp->hStdOutput; |
| startup_infoA.hStdError = rupp->hStdError; |
| |
| RtlReleasePebLock(); |
| } |
