2 * Process environment management
4 * Copyright 1996, 1998 Alexandre Julliard
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 #include "wine/port.h"
34 #include "wine/server.h"
35 #include "wine/library.h"
37 #include "wine/unicode.h"
38 #include "wine/debug.h"
40 WINE_DEFAULT_DEBUG_CHANNEL(environ
);
43 * - contrary to Microsoft docs, the environment strings do not appear
44 * to be sorted on Win95 (although they are on NT); so we don't bother
45 * to sort them either.
48 static STARTUPINFOW startup_infoW
;
49 static STARTUPINFOA startup_infoA
;
52 /***********************************************************************
53 * GetCommandLineA (KERNEL32.@)
55 * WARNING: there's a Windows incompatibility lurking here !
56 * Win32s always includes the full path of the program file,
57 * whereas Windows NT only returns the full file path plus arguments
58 * in case the program has been started with a full path.
59 * Win9x seems to have inherited NT behaviour.
61 * Note that both Start Menu Execute and Explorer start programs with
62 * fully specified quoted app file paths, which is why probably the only case
63 * where you'll see single file names is in case of direct launch
64 * via CreateProcess or WinExec.
66 * Perhaps we should take care of Win3.1 programs here (Win32s "feature").
68 * References: MS KB article q102762.txt (special Win32s handling)
70 LPSTR WINAPI
GetCommandLineA(void)
72 static char *cmdlineA
; /* ASCII command line */
74 if (!cmdlineA
) /* make an ansi version if we don't have it */
79 cmdlineA
= (RtlUnicodeStringToAnsiString( &ansi
, &NtCurrentTeb()->Peb
->ProcessParameters
->CommandLine
, TRUE
) == STATUS_SUCCESS
) ?
86 /***********************************************************************
87 * GetCommandLineW (KERNEL32.@)
89 LPWSTR WINAPI
GetCommandLineW(void)
91 return NtCurrentTeb()->Peb
->ProcessParameters
->CommandLine
.Buffer
;
95 /***********************************************************************
96 * GetEnvironmentStrings (KERNEL32.@)
97 * GetEnvironmentStringsA (KERNEL32.@)
99 LPSTR WINAPI
GetEnvironmentStringsA(void)
109 ptrW
= NtCurrentTeb()->Peb
->ProcessParameters
->Environment
;
112 slen
= strlenW(ptrW
) + 1;
113 len
+= WideCharToMultiByte( CP_ACP
, 0, ptrW
, slen
, NULL
, 0, NULL
, NULL
);
117 if ((ret
= HeapAlloc( GetProcessHeap(), 0, len
)) != NULL
)
119 ptrW
= NtCurrentTeb()->Peb
->ProcessParameters
->Environment
;
123 slen
= strlenW(ptrW
) + 1;
124 WideCharToMultiByte( CP_ACP
, 0, ptrW
, slen
, ptrA
, len
, NULL
, NULL
);
126 ptrA
+= strlen(ptrA
) + 1;
136 /***********************************************************************
137 * GetEnvironmentStringsW (KERNEL32.@)
139 LPWSTR WINAPI
GetEnvironmentStringsW(void)
141 return NtCurrentTeb()->Peb
->ProcessParameters
->Environment
;
145 /***********************************************************************
146 * FreeEnvironmentStringsA (KERNEL32.@)
148 BOOL WINAPI
FreeEnvironmentStringsA( LPSTR ptr
)
150 return HeapFree( GetProcessHeap(), 0, ptr
);
154 /***********************************************************************
155 * FreeEnvironmentStringsW (KERNEL32.@)
157 BOOL WINAPI
FreeEnvironmentStringsW( LPWSTR ptr
)
163 /***********************************************************************
164 * GetEnvironmentVariableA (KERNEL32.@)
166 DWORD WINAPI
GetEnvironmentVariableA( LPCSTR name
, LPSTR value
, DWORD size
)
168 UNICODE_STRING us_name
;
174 SetLastError(ERROR_ENVVAR_NOT_FOUND
);
178 if (!(valueW
= HeapAlloc(GetProcessHeap(), 0, size
* sizeof(WCHAR
))))
181 RtlCreateUnicodeStringFromAsciiz( &us_name
, name
);
183 ret
= GetEnvironmentVariableW( us_name
.Buffer
, valueW
, size
);
184 if (ret
&& ret
< size
)
186 WideCharToMultiByte( CP_ACP
, 0, valueW
, ret
+ 1, value
, size
, NULL
, NULL
);
188 /* this is needed to tell, with 0 as a return value, the difference between:
189 * - an error (GetLastError() != 0)
190 * - returning an empty string (in this case, we need to update the buffer)
192 if (ret
== 0 && size
&& GetLastError() == 0)
195 RtlFreeUnicodeString( &us_name
);
196 HeapFree(GetProcessHeap(), 0, valueW
);
202 /***********************************************************************
203 * GetEnvironmentVariableW (KERNEL32.@)
205 DWORD WINAPI
GetEnvironmentVariableW( LPCWSTR name
, LPWSTR val
, DWORD size
)
207 UNICODE_STRING us_name
;
208 UNICODE_STRING us_value
;
212 TRACE("(%s %p %lu)\n", debugstr_w(name
), val
, size
);
216 SetLastError(ERROR_ENVVAR_NOT_FOUND
);
220 RtlInitUnicodeString(&us_name
, name
);
222 us_value
.MaximumLength
= (size
? size
- 1 : 0) * sizeof(WCHAR
);
223 us_value
.Buffer
= val
;
225 status
= RtlQueryEnvironmentVariable_U(NULL
, &us_name
, &us_value
);
226 len
= us_value
.Length
/ sizeof(WCHAR
);
227 if (status
!= STATUS_SUCCESS
)
229 SetLastError( RtlNtStatusToDosError(status
) );
230 return (status
== STATUS_BUFFER_TOO_SMALL
) ? len
+ 1 : 0;
232 if (size
) val
[len
] = '\0';
234 return us_value
.Length
/ sizeof(WCHAR
);
238 /***********************************************************************
239 * SetEnvironmentVariableA (KERNEL32.@)
241 BOOL WINAPI
SetEnvironmentVariableA( LPCSTR name
, LPCSTR value
)
243 UNICODE_STRING us_name
;
248 SetLastError(ERROR_ENVVAR_NOT_FOUND
);
252 RtlCreateUnicodeStringFromAsciiz( &us_name
, name
);
255 UNICODE_STRING us_value
;
257 RtlCreateUnicodeStringFromAsciiz( &us_value
, value
);
258 ret
= SetEnvironmentVariableW( us_name
.Buffer
, us_value
.Buffer
);
259 RtlFreeUnicodeString( &us_value
);
261 else ret
= SetEnvironmentVariableW( us_name
.Buffer
, NULL
);
263 RtlFreeUnicodeString( &us_name
);
269 /***********************************************************************
270 * SetEnvironmentVariableW (KERNEL32.@)
272 BOOL WINAPI
SetEnvironmentVariableW( LPCWSTR name
, LPCWSTR value
)
274 UNICODE_STRING us_name
;
277 TRACE("(%s %s)\n", debugstr_w(name
), debugstr_w(value
));
281 SetLastError(ERROR_ENVVAR_NOT_FOUND
);
285 RtlInitUnicodeString(&us_name
, name
);
288 UNICODE_STRING us_value
;
290 RtlInitUnicodeString(&us_value
, value
);
291 status
= RtlSetEnvironmentVariable(NULL
, &us_name
, &us_value
);
293 else status
= RtlSetEnvironmentVariable(NULL
, &us_name
, NULL
);
295 if (status
!= STATUS_SUCCESS
)
297 SetLastError( RtlNtStatusToDosError(status
) );
304 /***********************************************************************
305 * ExpandEnvironmentStringsA (KERNEL32.@)
307 * Note: overlapping buffers are not supported; this is how it should be.
308 * FIXME: return value is wrong for MBCS
310 DWORD WINAPI
ExpandEnvironmentStringsA( LPCSTR src
, LPSTR dst
, DWORD count
)
312 UNICODE_STRING us_src
;
316 RtlCreateUnicodeStringFromAsciiz( &us_src
, src
);
319 if (!(dstW
= HeapAlloc(GetProcessHeap(), 0, count
* sizeof(WCHAR
))))
321 ret
= ExpandEnvironmentStringsW( us_src
.Buffer
, dstW
, count
);
323 WideCharToMultiByte( CP_ACP
, 0, dstW
, ret
, dst
, count
, NULL
, NULL
);
325 else ret
= ExpandEnvironmentStringsW( us_src
.Buffer
, NULL
, 0);
327 RtlFreeUnicodeString( &us_src
);
328 HeapFree(GetProcessHeap(), 0, dstW
);
334 /***********************************************************************
335 * ExpandEnvironmentStringsW (KERNEL32.@)
337 DWORD WINAPI
ExpandEnvironmentStringsW( LPCWSTR src
, LPWSTR dst
, DWORD len
)
339 UNICODE_STRING us_src
;
340 UNICODE_STRING us_dst
;
344 TRACE("(%s %p %lu)\n", debugstr_w(src
), dst
, len
);
346 RtlInitUnicodeString(&us_src
, src
);
348 us_dst
.MaximumLength
= len
* sizeof(WCHAR
);
352 status
= RtlExpandEnvironmentStrings_U(NULL
, &us_src
, &us_dst
, &res
);
353 res
/= sizeof(WCHAR
);
354 if (status
!= STATUS_SUCCESS
)
356 SetLastError( RtlNtStatusToDosError(status
) );
357 if (status
!= STATUS_BUFFER_TOO_SMALL
) return 0;
358 if (len
&& dst
) dst
[len
- 1] = '\0';
365 /***********************************************************************
366 * GetStdHandle (KERNEL32.@)
368 HANDLE WINAPI
GetStdHandle( DWORD std_handle
)
372 case STD_INPUT_HANDLE
: return NtCurrentTeb()->Peb
->ProcessParameters
->hStdInput
;
373 case STD_OUTPUT_HANDLE
: return NtCurrentTeb()->Peb
->ProcessParameters
->hStdOutput
;
374 case STD_ERROR_HANDLE
: return NtCurrentTeb()->Peb
->ProcessParameters
->hStdError
;
376 SetLastError( ERROR_INVALID_PARAMETER
);
377 return INVALID_HANDLE_VALUE
;
381 /***********************************************************************
382 * SetStdHandle (KERNEL32.@)
384 BOOL WINAPI
SetStdHandle( DWORD std_handle
, HANDLE handle
)
388 case STD_INPUT_HANDLE
: NtCurrentTeb()->Peb
->ProcessParameters
->hStdInput
= handle
; return TRUE
;
389 case STD_OUTPUT_HANDLE
: NtCurrentTeb()->Peb
->ProcessParameters
->hStdOutput
= handle
; return TRUE
;
390 case STD_ERROR_HANDLE
: NtCurrentTeb()->Peb
->ProcessParameters
->hStdError
= handle
; return TRUE
;
392 SetLastError( ERROR_INVALID_PARAMETER
);
396 /***********************************************************************
397 * GetStartupInfoA (KERNEL32.@)
399 VOID WINAPI
GetStartupInfoA( LPSTARTUPINFOA info
)
401 assert(startup_infoA
.cb
);
402 memcpy(info
, &startup_infoA
, sizeof(startup_infoA
));
406 /***********************************************************************
407 * GetStartupInfoW (KERNEL32.@)
409 VOID WINAPI
GetStartupInfoW( LPSTARTUPINFOW info
)
411 assert(startup_infoW
.cb
);
412 memcpy(info
, &startup_infoW
, sizeof(startup_infoW
));
415 /******************************************************************
416 * ENV_CopyStartupInformation (internal)
418 * Creates the STARTUPINFO information from the ntdll information
420 void ENV_CopyStartupInformation(void)
422 RTL_USER_PROCESS_PARAMETERS
* rupp
;
427 rupp
= NtCurrentTeb()->Peb
->ProcessParameters
;
429 startup_infoW
.cb
= sizeof(startup_infoW
);
430 startup_infoW
.lpReserved
= NULL
;
431 startup_infoW
.lpDesktop
= rupp
->Desktop
.Buffer
;
432 startup_infoW
.lpTitle
= rupp
->WindowTitle
.Buffer
;
433 startup_infoW
.dwX
= rupp
->dwX
;
434 startup_infoW
.dwY
= rupp
->dwY
;
435 startup_infoW
.dwXSize
= rupp
->dwXSize
;
436 startup_infoW
.dwYSize
= rupp
->dwYSize
;
437 startup_infoW
.dwXCountChars
= rupp
->dwXCountChars
;
438 startup_infoW
.dwYCountChars
= rupp
->dwYCountChars
;
439 startup_infoW
.dwFillAttribute
= rupp
->dwFillAttribute
;
440 startup_infoW
.dwFlags
= rupp
->dwFlags
;
441 startup_infoW
.wShowWindow
= rupp
->wShowWindow
;
442 startup_infoW
.cbReserved2
= rupp
->RuntimeInfo
.MaximumLength
;
443 startup_infoW
.lpReserved2
= rupp
->RuntimeInfo
.MaximumLength
? (void*)rupp
->RuntimeInfo
.Buffer
: NULL
;
444 startup_infoW
.hStdInput
= rupp
->hStdInput
;
445 startup_infoW
.hStdOutput
= rupp
->hStdOutput
;
446 startup_infoW
.hStdError
= rupp
->hStdError
;
448 startup_infoA
.cb
= sizeof(startup_infoA
);
449 startup_infoA
.lpReserved
= NULL
;
450 startup_infoA
.lpDesktop
= (rupp
->Desktop
.Length
&&
451 RtlUnicodeStringToAnsiString( &ansi
, &rupp
->Desktop
, TRUE
) == STATUS_SUCCESS
) ?
453 startup_infoA
.lpTitle
= (rupp
->WindowTitle
.Length
&&
454 RtlUnicodeStringToAnsiString( &ansi
, &rupp
->WindowTitle
, TRUE
) == STATUS_SUCCESS
) ?
456 startup_infoA
.dwX
= rupp
->dwX
;
457 startup_infoA
.dwY
= rupp
->dwY
;
458 startup_infoA
.dwXSize
= rupp
->dwXSize
;
459 startup_infoA
.dwYSize
= rupp
->dwYSize
;
460 startup_infoA
.dwXCountChars
= rupp
->dwXCountChars
;
461 startup_infoA
.dwYCountChars
= rupp
->dwYCountChars
;
462 startup_infoA
.dwFillAttribute
= rupp
->dwFillAttribute
;
463 startup_infoA
.dwFlags
= rupp
->dwFlags
;
464 startup_infoA
.wShowWindow
= rupp
->wShowWindow
;
465 startup_infoA
.cbReserved2
= rupp
->RuntimeInfo
.MaximumLength
;
466 startup_infoA
.lpReserved2
= rupp
->RuntimeInfo
.MaximumLength
? (void*)rupp
->RuntimeInfo
.Buffer
: NULL
;
467 startup_infoA
.hStdInput
= rupp
->hStdInput
;
468 startup_infoA
.hStdOutput
= rupp
->hStdOutput
;
469 startup_infoA
.hStdError
= rupp
->hStdError
;