Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-08-19 18:02:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expand user-specific environment variables.

Browse source 
svn path=/trunk/; revision=9870
This commit is contained in:
Eric Kohl 2004-06-25 11:42:00 +00:00
parent 07420e3dd3
commit 4531d96141
1 changed files with 63 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

79
reactos/lib/userenv/environment.c
View file

@ -1,4 +1,4 @@
/* $Id: environment.c,v 1.3 2004/05/01 11:55:01 ekohl Exp $
/* $Id: environment.c,v 1.4 2004/06/25 11:42:00 ekohl Exp $
*
* COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS system libraries
@ -18,20 +18,63 @@
static BOOL
SetUserEnvironmentVariable (LPVOID *Environment,
LPWSTR lpName,
LPWSTR lpValue)
LPWSTR lpValue,
BOOL bExpand)
{
UNICODE_STRING Name;
UNICODE_STRING Value;
UNICODE_STRING SrcValue;
UNICODE_STRING DstValue;
ULONG Length;
NTSTATUS Status;
if (bExpand)
{
RtlInitUnicodeString(&SrcValue,
lpValue);
Length = 2 * MAX_PATH * sizeof(WCHAR);
DstValue.Length = 0;
DstValue.MaximumLength = Length;
DstValue.Buffer = LocalAlloc (LPTR,
Length);
if (DstValue.Buffer == NULL)
{
DPRINT1("LocalAlloc() failed\n");
return FALSE;
}
Status = RtlExpandEnvironmentStrings_U((PWSTR)*Environment,
&SrcValue,
&DstValue,
&Length);
if (!NT_SUCCESS(Status))
{
DPRINT1 ("RtlExpandEnvironmentStrings_U() failed (Status %lx)\n", Status);
DPRINT1 ("Length %lu\n", Length);
return FALSE;
}
}
else
{
RtlInitUnicodeString(&DstValue,
lpValue);
}
RtlInitUnicodeString (&Name,
lpName);
RtlInitUnicodeString (&Value,
lpValue);
DPRINT("Value: %wZ\n", &DstValue);
Status = RtlSetEnvironmentVariable ((PWSTR*)Environment,
&Name,
&Value);
&DstValue);
if (bExpand)
{
LocalFree(DstValue.Buffer);
}
if (!NT_SUCCESS(Status))
{
DPRINT1 ("RtlSetEnvironmentVariable() failed (Status %lx)\n", Status);
@ -214,7 +257,8 @@ SetUserEnvironment (LPVOID *lpEnvironment,
/* Set environment variable */
SetUserEnvironmentVariable (lpEnvironment,
lpValueName,
lpValueData);
lpValueData,
(dwType == REG_EXPAND_SZ));
}
}
@ -254,9 +298,10 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
if (GetComputerNameW (Buffer,
&Length))
{
SetUserEnvironmentVariable (lpEnvironment,
L"COMPUTERNAME",
Buffer);
SetUserEnvironmentVariable(lpEnvironment,
L"COMPUTERNAME",
Buffer,
FALSE);
}
if (hToken == NULL)
@ -275,9 +320,10 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
if (GetAllUsersProfileDirectoryW (Buffer,
&Length))
{
SetUserEnvironmentVariable (lpEnvironment,
L"ALLUSERSPROFILE",
Buffer);
SetUserEnvironmentVariable(lpEnvironment,
L"ALLUSERSPROFILE",
Buffer,
FALSE);
}
/* Set 'USERPROFILE' variable */
@ -286,9 +332,10 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
Buffer,
&Length))
{
SetUserEnvironmentVariable (lpEnvironment,
L"USERPROFILE",
Buffer);
SetUserEnvironmentVariable(lpEnvironment,
L"USERPROFILE",
Buffer,
FALSE);
}
/* FIXME: Set 'USERDOMAIN' variable */