Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-09-12 05:43:24 +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 * COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS system libraries * PROJECT: ReactOS system libraries
@ -18,20 +18,63 @@
static BOOL static BOOL
SetUserEnvironmentVariable (LPVOID *Environment, SetUserEnvironmentVariable (LPVOID *Environment,
LPWSTR lpName, LPWSTR lpName,
LPWSTR lpValue) LPWSTR lpValue,
BOOL bExpand)
{ {
UNICODE_STRING Name; UNICODE_STRING Name;
UNICODE_STRING Value; UNICODE_STRING SrcValue;
UNICODE_STRING DstValue;
ULONG Length;
NTSTATUS Status; 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, RtlInitUnicodeString (&Name,
lpName); lpName);
RtlInitUnicodeString (&Value,
lpValue); DPRINT("Value: %wZ\n", &DstValue);
Status = RtlSetEnvironmentVariable ((PWSTR*)Environment, Status = RtlSetEnvironmentVariable ((PWSTR*)Environment,
&Name, &Name,
&Value); &DstValue);
if (bExpand)
{
LocalFree(DstValue.Buffer);
}
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
DPRINT1 ("RtlSetEnvironmentVariable() failed (Status %lx)\n", Status); DPRINT1 ("RtlSetEnvironmentVariable() failed (Status %lx)\n", Status);
@ -214,7 +257,8 @@ SetUserEnvironment (LPVOID *lpEnvironment,
/* Set environment variable */ /* Set environment variable */
SetUserEnvironmentVariable (lpEnvironment, SetUserEnvironmentVariable (lpEnvironment,
lpValueName, lpValueName,
lpValueData); lpValueData,
(dwType == REG_EXPAND_SZ));
} }
} }
@ -254,9 +298,10 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
if (GetComputerNameW (Buffer, if (GetComputerNameW (Buffer,
&Length)) &Length))
{ {
SetUserEnvironmentVariable (lpEnvironment, SetUserEnvironmentVariable(lpEnvironment,
L"COMPUTERNAME", L"COMPUTERNAME",
Buffer); Buffer,
FALSE);
} }
if (hToken == NULL) if (hToken == NULL)
@ -275,9 +320,10 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
if (GetAllUsersProfileDirectoryW (Buffer, if (GetAllUsersProfileDirectoryW (Buffer,
&Length)) &Length))
{ {
SetUserEnvironmentVariable (lpEnvironment, SetUserEnvironmentVariable(lpEnvironment,
L"ALLUSERSPROFILE", L"ALLUSERSPROFILE",
Buffer); Buffer,
FALSE);
} }
/* Set 'USERPROFILE' variable */ /* Set 'USERPROFILE' variable */
@ -286,9 +332,10 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
Buffer, Buffer,
&Length)) &Length))
{ {
SetUserEnvironmentVariable (lpEnvironment, SetUserEnvironmentVariable(lpEnvironment,
L"USERPROFILE", L"USERPROFILE",
Buffer); Buffer,
FALSE);
} }
/* FIXME: Set 'USERDOMAIN' variable */ /* FIXME: Set 'USERDOMAIN' variable */