mirror of
https://github.com/reactos/reactos.git
synced 2024-12-28 10:04:49 +00:00
- Set user-specific environment variables.
- Append 'PATH' environment variable if it already exists. svn path=/trunk/; revision=8862
This commit is contained in:
Eric Kohl
parent
7bd2a160e6
commit
9915a75ce6
1 changed files with 212 additions and 3 deletions
|
|
@ -1,4 +1,4 @@
|
|||
/* $Id: environment.c,v 1.1 2004/03/19 12:40:49 ekohl Exp $
|
||||
/* $Id: environment.c,v 1.2 2004/03/24 16:00:01 ekohl Exp $
|
||||
*
|
||||
* COPYRIGHT: See COPYING in the top level directory
|
||||
* PROJECT: ReactOS system libraries
|
||||
|
|
@ -24,8 +24,10 @@ SetUserEnvironmentVariable (LPVOID *Environment,
|
|||
UNICODE_STRING Value;
|
||||
NTSTATUS Status;
|
||||
|
||||
RtlInitUnicodeString (&Name, lpName);
|
||||
RtlInitUnicodeString (&Value, lpValue);
|
||||
RtlInitUnicodeString (&Name,
|
||||
lpName);
|
||||
RtlInitUnicodeString (&Value,
|
||||
lpValue);
|
||||
|
||||
Status = RtlSetEnvironmentVariable ((PWSTR*)Environment,
|
||||
&Name,
|
||||
|
|
@ -40,6 +42,190 @@ SetUserEnvironmentVariable (LPVOID *Environment,
|
|||
}
|
||||
|
||||
|
||||
static BOOL
|
||||
AppendUserEnvironmentVariable (LPVOID *Environment,
|
||||
LPWSTR lpName,
|
||||
LPWSTR lpValue)
|
||||
{
|
||||
UNICODE_STRING Name;
|
||||
UNICODE_STRING Value;
|
||||
NTSTATUS Status;
|
||||
|
||||
RtlInitUnicodeString (&Name,
|
||||
lpName);
|
||||
|
||||
Value.Length = 0;
|
||||
Value.MaximumLength = 1024 * sizeof(WCHAR);
|
||||
Value.Buffer = LocalAlloc (LPTR,
|
||||
1024 * sizeof(WCHAR));
|
||||
if (Value.Buffer == NULL)
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
Value.Buffer[0] = UNICODE_NULL;
|
||||
|
||||
Status = RtlQueryEnvironmentVariable_U ((PWSTR)*Environment,
|
||||
&Name,
|
||||
&Value);
|
||||
if (NT_SUCCESS(Status))
|
||||
{
|
||||
RtlAppendUnicodeToString (&Value,
|
||||
L";");
|
||||
}
|
||||
|
||||
RtlAppendUnicodeToString (&Value,
|
||||
lpValue);
|
||||
|
||||
Status = RtlSetEnvironmentVariable ((PWSTR*)Environment,
|
||||
&Name,
|
||||
&Value);
|
||||
LocalFree (Value.Buffer);
|
||||
if (!NT_SUCCESS(Status))
|
||||
{
|
||||
DPRINT1 ("RtlSetEnvironmentVariable() failed (Status %lx)\n", Status);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
|
||||
static HKEY
|
||||
GetCurrentUserKey (HANDLE hToken)
|
||||
{
|
||||
UNICODE_STRING SidString;
|
||||
HKEY hKey;
|
||||
|
||||
if (!GetUserSidFromToken (hToken,
|
||||
&SidString))
|
||||
{
|
||||
DPRINT1 ("GetUserSidFromToken() failed\n");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (RegOpenKeyExW (HKEY_USERS,
|
||||
SidString.Buffer,
|
||||
0,
|
||||
KEY_ALL_ACCESS,
|
||||
&hKey))
|
||||
{
|
||||
DPRINT1 ("RegOpenKeyExW() failed (Error %ld)\n", GetLastError());
|
||||
RtlFreeUnicodeString (&SidString);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
RtlFreeUnicodeString (&SidString);
|
||||
|
||||
return hKey;
|
||||
}
|
||||
|
||||
|
||||
static BOOL
|
||||
SetUserEnvironment (LPVOID *lpEnvironment,
|
||||
HKEY hKey,
|
||||
LPWSTR lpSubKeyName)
|
||||
{
|
||||
HKEY hEnvKey;
|
||||
DWORD dwValues;
|
||||
DWORD dwMaxValueNameLength;
|
||||
DWORD dwMaxValueDataLength;
|
||||
DWORD dwValueNameLength;
|
||||
DWORD dwValueDataLength;
|
||||
DWORD dwType;
|
||||
DWORD i;
|
||||
LPWSTR lpValueName;
|
||||
LPWSTR lpValueData;
|
||||
|
||||
if (RegOpenKeyExW (hKey,
|
||||
lpSubKeyName,
|
||||
0,
|
||||
KEY_ALL_ACCESS,
|
||||
&hEnvKey))
|
||||
{
|
||||
DPRINT1 ("RegOpenKeyExW() failed (Error %ld)\n", GetLastError());
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
if (RegQueryInfoKey (hEnvKey,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
NULL,
|
||||
&dwValues,
|
||||
&dwMaxValueNameLength,
|
||||
&dwMaxValueDataLength,
|
||||
NULL,
|
||||
NULL))
|
||||
{
|
||||
DPRINT1 ("RegQueryInforKey() failed (Error %ld)\n", GetLastError());
|
||||
RegCloseKey (hEnvKey);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
if (dwValues == 0)
|
||||
{
|
||||
RegCloseKey (hEnvKey);
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
/* Allocate buffers */
|
||||
lpValueName = LocalAlloc (LPTR,
|
||||
dwMaxValueNameLength * sizeof(WCHAR));
|
||||
if (lpValueName == NULL)
|
||||
{
|
||||
RegCloseKey (hEnvKey);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
lpValueData = LocalAlloc (LPTR,
|
||||
dwMaxValueDataLength);
|
||||
if (lpValueData == NULL)
|
||||
{
|
||||
LocalFree (lpValueName);
|
||||
RegCloseKey (hEnvKey);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/* Enumerate values */
|
||||
for (i = 0; i < dwValues; i++)
|
||||
{
|
||||
dwValueNameLength = dwMaxValueNameLength;
|
||||
dwValueDataLength = dwMaxValueDataLength;
|
||||
RegEnumValueW (hEnvKey,
|
||||
i,
|
||||
lpValueName,
|
||||
&dwValueNameLength,
|
||||
NULL,
|
||||
&dwType,
|
||||
(LPBYTE)lpValueData,
|
||||
&dwValueDataLength);
|
||||
|
||||
if (!_wcsicmp (lpValueName, L"path"))
|
||||
{
|
||||
/* Append 'Path' environment variable */
|
||||
AppendUserEnvironmentVariable (lpEnvironment,
|
||||
lpValueName,
|
||||
lpValueData);
|
||||
}
|
||||
else
|
||||
{
|
||||
/* Set environment variable */
|
||||
SetUserEnvironmentVariable (lpEnvironment,
|
||||
lpValueName,
|
||||
lpValueData);
|
||||
}
|
||||
}
|
||||
|
||||
LocalFree (lpValueData);
|
||||
LocalFree (lpValueName);
|
||||
RegCloseKey (hEnvKey);
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
|
||||
BOOL WINAPI
|
||||
CreateEnvironmentBlock (LPVOID *lpEnvironment,
|
||||
HANDLE hToken,
|
||||
|
|
@ -47,6 +233,7 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
|
|||
{
|
||||
WCHAR Buffer[MAX_PATH];
|
||||
DWORD Length;
|
||||
HKEY hKeyUser;
|
||||
NTSTATUS Status;
|
||||
|
||||
DPRINT1 ("CreateEnvironmentBlock() called\n");
|
||||
|
|
@ -72,6 +259,17 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
|
|||
Buffer);
|
||||
}
|
||||
|
||||
if (hToken == NULL)
|
||||
return TRUE;
|
||||
|
||||
hKeyUser = GetCurrentUserKey (hToken);
|
||||
if (hKeyUser == NULL)
|
||||
{
|
||||
DPRINT1 ("GetCurrentUserKey() failed\n");
|
||||
RtlDestroyEnvironment (*lpEnvironment);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/* Set 'USERPROFILE' variable */
|
||||
Length = MAX_PATH;
|
||||
if (GetUserProfileDirectoryW (hToken,
|
||||
|
|
@ -83,6 +281,17 @@ CreateEnvironmentBlock (LPVOID *lpEnvironment,
|
|||
Buffer);
|
||||
}
|
||||
|
||||
/* FIXME: Set 'USERDOMAIN' variable */
|
||||
|
||||
/* FIXME: Set 'USERNAME' variable */
|
||||
|
||||
/* Set user environment variables */
|
||||
SetUserEnvironment (lpEnvironment,
|
||||
hKeyUser,
|
||||
L"Environment");
|
||||
|
||||
RegCloseKey (hKeyUser);
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue