2 * Ntdll environment functions
4 * Copyright 1996, 1998 Alexandre Julliard
5 * Copyright 2003 Eric Pouech
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
30 #include "wine/unicode.h"
31 #include "wine/debug.h"
32 #include "ntdll_misc.h"
34 WINE_DEFAULT_DEBUG_CHANNEL(environ
);
36 /******************************************************************************
37 * RtlCreateEnvironment [NTDLL.@]
39 NTSTATUS WINAPI
RtlCreateEnvironment(BOOLEAN inherit
, PWSTR
* env
)
43 TRACE("(%u,%p)!\n", inherit
, env
);
47 MEMORY_BASIC_INFORMATION mbi
;
51 nts
= NtQueryVirtualMemory(NtCurrentProcess(),
52 NtCurrentTeb()->Peb
->ProcessParameters
->Environment
,
53 0, &mbi
, sizeof(mbi
), NULL
);
54 if (nts
== STATUS_SUCCESS
)
57 nts
= NtAllocateVirtualMemory(NtCurrentProcess(), (void**)env
, 0, &mbi
.RegionSize
,
58 MEM_RESERVE
| MEM_COMMIT
, PAGE_READWRITE
);
59 if (nts
== STATUS_SUCCESS
)
60 memcpy(*env
, NtCurrentTeb()->Peb
->ProcessParameters
->Environment
, mbi
.RegionSize
);
68 nts
= NtAllocateVirtualMemory(NtCurrentProcess(), (void**)env
, 0, &size
,
69 MEM_RESERVE
| MEM_COMMIT
, PAGE_READWRITE
);
70 if (nts
== STATUS_SUCCESS
)
71 memset(*env
, 0, size
);
77 /******************************************************************************
78 * RtlDestroyEnvironment [NTDLL.@]
80 NTSTATUS WINAPI
RtlDestroyEnvironment(PWSTR env
)
84 TRACE("(%p)!\n", env
);
86 return NtFreeVirtualMemory(NtCurrentProcess(), (void**)&env
, &size
, MEM_RELEASE
);
89 static LPCWSTR
ENV_FindVariable(PCWSTR var
, PCWSTR name
, unsigned namelen
)
91 for (; *var
; var
+= strlenW(var
) + 1)
93 /* match var names, but avoid setting a var with a name including a '='
94 * (a starting '=' is valid though)
96 if (strncmpiW(var
, name
, namelen
) == 0 && var
[namelen
] == '=' &&
97 strchrW(var
+ 1, '=') == var
+ namelen
)
99 return var
+ namelen
+ 1;
105 /******************************************************************
106 * RtlQueryEnvironmentVariable_U [NTDLL.@]
108 * NOTES: when the buffer is too small, the string is not written, but if the
109 * terminating null char is the only char that cannot be written, then
110 * all chars (except the null) are written and success is returned
111 * (behavior of Win2k at least)
113 NTSTATUS WINAPI
RtlQueryEnvironmentVariable_U(PWSTR env
,
114 PUNICODE_STRING name
,
115 PUNICODE_STRING value
)
117 NTSTATUS nts
= STATUS_VARIABLE_NOT_FOUND
;
121 TRACE("%s %s %p\n", debugstr_w(env
), debugstr_w(name
->Buffer
), value
);
124 namelen
= name
->Length
/ sizeof(WCHAR
);
125 if (!namelen
) return nts
;
130 var
= NtCurrentTeb()->Peb
->ProcessParameters
->Environment
;
134 var
= ENV_FindVariable(var
, name
->Buffer
, namelen
);
137 value
->Length
= strlenW(var
) * sizeof(WCHAR
);
139 if (value
->Length
<= value
->MaximumLength
)
141 memmove(value
->Buffer
, var
, min(value
->Length
+ sizeof(WCHAR
), value
->MaximumLength
));
142 nts
= STATUS_SUCCESS
;
144 else nts
= STATUS_BUFFER_TOO_SMALL
;
147 if (!env
) RtlReleasePebLock();
152 /******************************************************************
153 * RtlSetCurrentEnvironment [NTDLL.@]
156 void WINAPI
RtlSetCurrentEnvironment(PWSTR new_env
, PWSTR
* old_env
)
158 TRACE("(%p %p)\n", new_env
, old_env
);
162 if (old_env
) *old_env
= NtCurrentTeb()->Peb
->ProcessParameters
->Environment
;
163 NtCurrentTeb()->Peb
->ProcessParameters
->Environment
= new_env
;
169 /******************************************************************************
170 * RtlSetEnvironmentVariable [NTDLL.@]
172 NTSTATUS WINAPI
RtlSetEnvironmentVariable(PWSTR
* penv
, PUNICODE_STRING name
,
173 PUNICODE_STRING value
)
177 NTSTATUS nts
= STATUS_VARIABLE_NOT_FOUND
;
178 MEMORY_BASIC_INFORMATION mbi
;
180 TRACE("(%p,%s,%s)\n",
181 penv
, debugstr_w(name
->Buffer
),
182 value
? debugstr_w(value
->Buffer
) : "--nil--");
184 if (!name
|| !name
->Buffer
|| !name
->Buffer
[0])
185 return STATUS_INVALID_PARAMETER_1
;
186 /* variable names can't contain a '=' except as a first character */
187 if (strchrW(name
->Buffer
+ 1, '=')) return STATUS_INVALID_PARAMETER
;
192 env
= NtCurrentTeb()->Peb
->ProcessParameters
->Environment
;
195 len
= name
->Length
/ sizeof(WCHAR
);
197 /* compute current size of environment */
198 for (p
= env
; *p
; p
+= strlenW(p
) + 1);
199 old_size
= p
+ 1 - env
;
201 /* Find a place to insert the string */
202 for (p
= env
; *p
; p
+= strlenW(p
) + 1)
204 if (!strncmpiW(name
->Buffer
, p
, len
) && (p
[len
] == '=')) break;
206 if (!value
&& !*p
) goto done
; /* Value to remove doesn't exist */
208 /* Realloc the buffer */
209 len
= value
? len
+ value
->Length
/ sizeof(WCHAR
) + 2 : 0;
210 if (*p
) len
-= strlenW(p
) + 1; /* The name already exists */
214 LPWSTR next
= p
+ strlenW(p
) + 1; /* We know there is a next one */
215 memmove(next
+ len
, next
, (old_size
- (next
- env
)) * sizeof(WCHAR
));
218 nts
= NtQueryVirtualMemory(NtCurrentProcess(), env
, 0,
219 &mbi
, sizeof(mbi
), NULL
);
220 if (nts
!= STATUS_SUCCESS
) goto done
;
222 if ((old_size
+ len
) * sizeof(WCHAR
) > mbi
.RegionSize
)
225 ULONG new_size
= (old_size
+ len
) * sizeof(WCHAR
);
228 nts
= NtAllocateVirtualMemory(NtCurrentProcess(), (void**)&new_env
, 0,
229 &new_size
, MEM_RESERVE
| MEM_COMMIT
, PAGE_READWRITE
);
230 if (nts
!= STATUS_SUCCESS
) goto done
;
232 memmove(new_env
, env
, (p
- env
) * sizeof(WCHAR
));
234 memmove(new_env
+ (p
- env
) + len
, p
, (old_size
- (p
- env
)) * sizeof(WCHAR
));
235 p
= new_env
+ (p
- env
);
237 RtlDestroyEnvironment(env
);
238 if (!penv
) NtCurrentTeb()->Peb
->ProcessParameters
->Environment
= new_env
;
239 else *penv
= new_env
;
244 if (len
> 0) memmove(p
+ len
, p
, (old_size
- (p
- env
)) * sizeof(WCHAR
));
247 /* Set the new string */
250 static const WCHAR equalW
[] = {'=',0};
252 strcpyW(p
, name
->Buffer
);
254 strcatW(p
, value
->Buffer
);
257 if (!penv
) RtlReleasePebLock();
262 /******************************************************************
263 * RtlExpandEnvironmentStrings_U (NTDLL.@)
266 NTSTATUS WINAPI
RtlExpandEnvironmentStrings_U(PWSTR renv
, const UNICODE_STRING
* us_src
,
267 PUNICODE_STRING us_dst
, PULONG plen
)
269 DWORD len
, count
, total_size
= 1; /* 1 for terminating '\0' */
270 LPCWSTR env
, src
, p
, var
;
273 src
= us_src
->Buffer
;
274 count
= us_dst
->MaximumLength
/ sizeof(WCHAR
);
275 dst
= count
? us_dst
->Buffer
: NULL
;
280 env
= NtCurrentTeb()->Peb
->ProcessParameters
->Environment
;
288 if ((p
= strchrW( src
, '%' ))) len
= p
- src
;
289 else len
= strlenW(src
);
293 else /* we are at the start of a variable */
295 if ((p
= strchrW( src
+ 1, '%' )))
297 len
= p
- src
- 1; /* Length of the variable name */
298 if ((var
= ENV_FindVariable( env
, src
+ 1, len
)))
300 src
+= len
+ 2; /* Skip the variable name */
305 var
= src
; /* Copy original name instead */
310 else /* unfinished variable name, ignore it */
313 len
= strlenW(src
); /* Copy whole string */
320 if (count
< len
) len
= count
;
321 memcpy(dst
, var
, len
* sizeof(WCHAR
));
327 if (!renv
) RtlReleasePebLock();
329 /* Null-terminate the string */
330 if (dst
&& count
) *dst
= '\0';
332 us_dst
->Length
= (dst
) ? (dst
- us_dst
->Buffer
) * sizeof(WCHAR
) : 0;
333 if (plen
) *plen
= total_size
* sizeof(WCHAR
);
335 return (count
) ? STATUS_SUCCESS
: STATUS_BUFFER_TOO_SMALL
;
339 static inline void normalize( void *base
, WCHAR
**ptr
)
341 if (*ptr
) *ptr
= (WCHAR
*)((char *)base
+ (UINT_PTR
)*ptr
);
344 /******************************************************************************
345 * RtlNormalizeProcessParams [NTDLL.@]
347 PRTL_USER_PROCESS_PARAMETERS WINAPI
RtlNormalizeProcessParams( RTL_USER_PROCESS_PARAMETERS
*params
)
349 if (params
&& !(params
->Flags
& PROCESS_PARAMS_FLAG_NORMALIZED
))
351 normalize( params
, ¶ms
->CurrentDirectoryName
.Buffer
);
352 normalize( params
, ¶ms
->DllPath
.Buffer
);
353 normalize( params
, ¶ms
->ImagePathName
.Buffer
);
354 normalize( params
, ¶ms
->CommandLine
.Buffer
);
355 normalize( params
, ¶ms
->WindowTitle
.Buffer
);
356 normalize( params
, ¶ms
->Desktop
.Buffer
);
357 normalize( params
, ¶ms
->ShellInfo
.Buffer
);
358 normalize( params
, ¶ms
->RuntimeInfo
.Buffer
);
359 params
->Flags
|= PROCESS_PARAMS_FLAG_NORMALIZED
;
365 static inline void denormalize( void *base
, WCHAR
**ptr
)
367 if (*ptr
) *ptr
= (WCHAR
*)(UINT_PTR
)((char *)*ptr
- (char *)base
);
370 /******************************************************************************
371 * RtlDeNormalizeProcessParams [NTDLL.@]
373 PRTL_USER_PROCESS_PARAMETERS WINAPI
RtlDeNormalizeProcessParams( RTL_USER_PROCESS_PARAMETERS
*params
)
375 if (params
&& (params
->Flags
& PROCESS_PARAMS_FLAG_NORMALIZED
))
377 denormalize( params
, ¶ms
->CurrentDirectoryName
.Buffer
);
378 denormalize( params
, ¶ms
->DllPath
.Buffer
);
379 denormalize( params
, ¶ms
->ImagePathName
.Buffer
);
380 denormalize( params
, ¶ms
->CommandLine
.Buffer
);
381 denormalize( params
, ¶ms
->WindowTitle
.Buffer
);
382 denormalize( params
, ¶ms
->Desktop
.Buffer
);
383 denormalize( params
, ¶ms
->ShellInfo
.Buffer
);
384 denormalize( params
, ¶ms
->RuntimeInfo
.Buffer
);
385 params
->Flags
&= ~PROCESS_PARAMS_FLAG_NORMALIZED
;
391 /* append a unicode string to the process params data; helper for RtlCreateProcessParameters */
392 static void append_unicode_string( void **data
, const UNICODE_STRING
*src
,
393 UNICODE_STRING
*dst
)
395 dst
->Length
= src
->Length
;
396 dst
->MaximumLength
= src
->MaximumLength
;
398 memcpy( dst
->Buffer
, src
->Buffer
, dst
->MaximumLength
);
399 *data
= (char *)dst
->Buffer
+ dst
->MaximumLength
;
403 /******************************************************************************
404 * RtlCreateProcessParameters [NTDLL.@]
406 NTSTATUS WINAPI
RtlCreateProcessParameters( RTL_USER_PROCESS_PARAMETERS
**result
,
407 const UNICODE_STRING
*ImagePathName
,
408 const UNICODE_STRING
*DllPath
,
409 const UNICODE_STRING
*CurrentDirectoryName
,
410 const UNICODE_STRING
*CommandLine
,
412 const UNICODE_STRING
*WindowTitle
,
413 const UNICODE_STRING
*Desktop
,
414 const UNICODE_STRING
*ShellInfo
,
415 const UNICODE_STRING
*RuntimeInfo
)
417 static const WCHAR empty
[] = {0};
418 static const UNICODE_STRING empty_str
= { 0, sizeof(empty
), (WCHAR
*)empty
};
420 const RTL_USER_PROCESS_PARAMETERS
*cur_params
;
421 ULONG size
, total_size
;
426 cur_params
= NtCurrentTeb()->Peb
->ProcessParameters
;
427 if (!DllPath
) DllPath
= &cur_params
->DllPath
;
428 if (!CurrentDirectoryName
) CurrentDirectoryName
= &cur_params
->CurrentDirectoryName
;
429 if (!CommandLine
) CommandLine
= ImagePathName
;
430 if (!Environment
) Environment
= cur_params
->Environment
;
431 if (!WindowTitle
) WindowTitle
= &empty_str
;
432 if (!Desktop
) Desktop
= &empty_str
;
433 if (!ShellInfo
) ShellInfo
= &empty_str
;
434 if (!RuntimeInfo
) RuntimeInfo
= &empty_str
;
436 size
= (sizeof(RTL_USER_PROCESS_PARAMETERS
)
437 + ImagePathName
->MaximumLength
438 + DllPath
->MaximumLength
439 + CurrentDirectoryName
->MaximumLength
440 + CommandLine
->MaximumLength
441 + WindowTitle
->MaximumLength
442 + Desktop
->MaximumLength
443 + ShellInfo
->MaximumLength
444 + RuntimeInfo
->MaximumLength
);
447 if ((status
= NtAllocateVirtualMemory( NtCurrentProcess(), &ptr
, NULL
, &total_size
,
448 MEM_COMMIT
, PAGE_READWRITE
)) == STATUS_SUCCESS
)
450 RTL_USER_PROCESS_PARAMETERS
*params
= ptr
;
451 params
->AllocationSize
= total_size
;
453 params
->Flags
= PROCESS_PARAMS_FLAG_NORMALIZED
;
454 params
->ProcessGroup
= cur_params
->ProcessGroup
;
455 params
->Environment
= Environment
;
456 /* all other fields are zero */
459 append_unicode_string( &ptr
, CurrentDirectoryName
, ¶ms
->CurrentDirectoryName
);
460 append_unicode_string( &ptr
, DllPath
, ¶ms
->DllPath
);
461 append_unicode_string( &ptr
, ImagePathName
, ¶ms
->ImagePathName
);
462 append_unicode_string( &ptr
, CommandLine
, ¶ms
->CommandLine
);
463 append_unicode_string( &ptr
, WindowTitle
, ¶ms
->WindowTitle
);
464 append_unicode_string( &ptr
, Desktop
, ¶ms
->Desktop
);
465 append_unicode_string( &ptr
, ShellInfo
, ¶ms
->ShellInfo
);
466 append_unicode_string( &ptr
, RuntimeInfo
, ¶ms
->RuntimeInfo
);
467 *result
= RtlDeNormalizeProcessParams( params
);
474 /******************************************************************************
475 * RtlDestroyProcessParameters [NTDLL.@]
477 void WINAPI
RtlDestroyProcessParameters( RTL_USER_PROCESS_PARAMETERS
*params
)
481 NtFreeVirtualMemory( NtCurrentProcess(), &ptr
, &size
, MEM_RELEASE
);