mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 22:28:36 +00:00
120 lines
3.1 KiB
C
120 lines
3.1 KiB
C
/*
|
|
* PROJECT: ReactOS Session Manager
|
|
* LICENSE: GPL v2 or later - See COPYING in the top level directory
|
|
* FILE: base/system/smss/initenv.c
|
|
* PURPOSE: Environment initialization.
|
|
* PROGRAMMERS: ReactOS Development Team
|
|
*/
|
|
|
|
/* INCLUDES ******************************************************************/
|
|
#include "smss.h"
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
/* GLOBALS */
|
|
|
|
PWSTR SmSystemEnvironment = NULL;
|
|
|
|
|
|
/* FUNCTIONS */
|
|
|
|
NTSTATUS
|
|
SmCreateEnvironment(VOID)
|
|
{
|
|
return RtlCreateEnvironment(FALSE, &SmSystemEnvironment);
|
|
}
|
|
|
|
|
|
static NTSTATUS
|
|
SmpSetEnvironmentVariable(IN PVOID Context,
|
|
IN PWSTR ValueName,
|
|
IN PVOID ValueData)
|
|
{
|
|
UNICODE_STRING EnvVariable;
|
|
UNICODE_STRING EnvValue;
|
|
|
|
RtlInitUnicodeString(&EnvVariable,
|
|
ValueName);
|
|
RtlInitUnicodeString(&EnvValue,
|
|
(PWSTR)ValueData);
|
|
return RtlSetEnvironmentVariable(Context,
|
|
&EnvVariable,
|
|
&EnvValue);
|
|
}
|
|
|
|
|
|
static NTSTATUS NTAPI
|
|
SmpEnvironmentQueryRoutine(IN PWSTR ValueName,
|
|
IN ULONG ValueType,
|
|
IN PVOID ValueData,
|
|
IN ULONG ValueLength,
|
|
IN PVOID Context,
|
|
IN PVOID EntryContext)
|
|
{
|
|
DPRINT("ValueName '%S' Type %lu Length %lu\n", ValueName, ValueType, ValueLength);
|
|
|
|
if (ValueType != REG_SZ && ValueType != REG_EXPAND_SZ)
|
|
return STATUS_SUCCESS;
|
|
|
|
DPRINT("ValueData '%S'\n", (PWSTR)ValueData);
|
|
return SmpSetEnvironmentVariable(Context,ValueName,ValueData);
|
|
}
|
|
|
|
|
|
NTSTATUS
|
|
SmSetEnvironmentVariables(VOID)
|
|
{
|
|
RTL_QUERY_REGISTRY_TABLE QueryTable[2];
|
|
WCHAR ValueBuffer[MAX_PATH];
|
|
NTSTATUS Status;
|
|
|
|
/*
|
|
* The following environment variables must be set prior to reading
|
|
* other variables from the registry.
|
|
*
|
|
* Variables (example):
|
|
* SystemRoot = "C:\reactos"
|
|
* SystemDrive = "C:"
|
|
*/
|
|
|
|
/* Copy system root into value buffer */
|
|
wcscpy(ValueBuffer,
|
|
SharedUserData->NtSystemRoot);
|
|
|
|
/* Set SystemRoot = "C:\reactos" */
|
|
SmpSetEnvironmentVariable(&SmSystemEnvironment, L"SystemRoot", ValueBuffer);
|
|
|
|
/* Cut off trailing path */
|
|
ValueBuffer[2] = 0;
|
|
|
|
/* Set SystemDrive = "C:" */
|
|
SmpSetEnvironmentVariable(&SmSystemEnvironment, L"SystemDrive", ValueBuffer);
|
|
|
|
/* Read system environment from the registry. */
|
|
RtlZeroMemory(&QueryTable,
|
|
sizeof(QueryTable));
|
|
|
|
QueryTable[0].QueryRoutine = SmpEnvironmentQueryRoutine;
|
|
|
|
Status = RtlQueryRegistryValues(RTL_REGISTRY_CONTROL,
|
|
L"Session Manager\\Environment",
|
|
QueryTable,
|
|
&SmSystemEnvironment,
|
|
SmSystemEnvironment);
|
|
|
|
return Status;
|
|
}
|
|
|
|
/**********************************************************************
|
|
* Set environment variables from registry
|
|
*/
|
|
NTSTATUS
|
|
SmUpdateEnvironment(VOID)
|
|
{
|
|
/* TODO */
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
/* EOF */
|