409 lines
10 KiB
C
409 lines
10 KiB
C
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1992 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
cmcontrl.c
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
The module contains CmGetSystemControlValues, see cmdat.c for data.
|
||
|
|
|
||
|
|
Author:
|
||
|
|
|
||
|
|
Bryan M. Willman (bryanwi) 12-May-92
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include "cmp.h"
|
||
|
|
|
||
|
|
extern WCHAR CmDefaultLanguageId[];
|
||
|
|
extern ULONG CmDefaultLanguageIdLength;
|
||
|
|
extern ULONG CmDefaultLanguageIdType;
|
||
|
|
|
||
|
|
extern WCHAR CmInstallUILanguageId[];
|
||
|
|
extern ULONG CmInstallUILanguageIdLength;
|
||
|
|
extern ULONG CmInstallUILanguageIdType;
|
||
|
|
|
||
|
|
HCELL_INDEX
|
||
|
|
CmpWalkPath(
|
||
|
|
PHHIVE SystemHive,
|
||
|
|
HCELL_INDEX ParentCell,
|
||
|
|
PWSTR Path
|
||
|
|
);
|
||
|
|
|
||
|
|
LANGID
|
||
|
|
CmpConvertLangId(
|
||
|
|
PWSTR LangIdString,
|
||
|
|
ULONG LangIdStringLength
|
||
|
|
);
|
||
|
|
|
||
|
|
#ifdef ALLOC_PRAGMA
|
||
|
|
#pragma alloc_text(INIT,CmGetSystemControlValues)
|
||
|
|
#pragma alloc_text(INIT,CmpWalkPath)
|
||
|
|
#pragma alloc_text(INIT,CmpConvertLangId)
|
||
|
|
#endif
|
||
|
|
|
||
|
|
VOID
|
||
|
|
CmGetSystemControlValues(
|
||
|
|
PVOID SystemHiveBuffer,
|
||
|
|
PCM_SYSTEM_CONTROL_VECTOR ControlVector
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Look for registry values in current control set, as specified
|
||
|
|
by entries in ControlVector. Report data for value entries
|
||
|
|
(if any) to variables ControlVector points to.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
SystemHiveBuffer - pointer to flat image of the system hive
|
||
|
|
|
||
|
|
ControlVector - pointer to structure that describes what values
|
||
|
|
to pull out and store
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
NONE.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
NTSTATUS status;
|
||
|
|
PHHIVE SystemHive;
|
||
|
|
CMHIVE TempHive;
|
||
|
|
HCELL_INDEX RootCell;
|
||
|
|
HCELL_INDEX BaseCell;
|
||
|
|
UNICODE_STRING Name;
|
||
|
|
PHCELL_INDEX Index;
|
||
|
|
HCELL_INDEX KeyCell;
|
||
|
|
HCELL_INDEX ValueCell;
|
||
|
|
PCM_KEY_VALUE ValueBody;
|
||
|
|
PVOID ValueData;
|
||
|
|
ULONG Length;
|
||
|
|
BOOLEAN AutoSelect;
|
||
|
|
BOOLEAN small;
|
||
|
|
ULONG tmplength;
|
||
|
|
PCM_KEY_NODE Node;
|
||
|
|
|
||
|
|
//
|
||
|
|
// set up to read flat system hive image loader passes us
|
||
|
|
//
|
||
|
|
RtlZeroMemory((PVOID)&TempHive, sizeof(TempHive));
|
||
|
|
SystemHive = &(TempHive.Hive);
|
||
|
|
CmpInitHiveViewList((PCMHIVE)SystemHive);
|
||
|
|
CmpInitSecurityCache((PCMHIVE)SystemHive);
|
||
|
|
status = HvInitializeHive(
|
||
|
|
SystemHive,
|
||
|
|
HINIT_FLAT,
|
||
|
|
HIVE_VOLATILE,
|
||
|
|
HFILE_TYPE_PRIMARY,
|
||
|
|
SystemHiveBuffer,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
1,
|
||
|
|
NULL
|
||
|
|
);
|
||
|
|
if (!NT_SUCCESS(status)) {
|
||
|
|
CM_BUGCHECK(BAD_SYSTEM_CONFIG_INFO,BAD_SYSTEM_CONTROL_VALUES,1,SystemHive,status);
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// don't bother locking/releasing cells
|
||
|
|
//
|
||
|
|
ASSERT( SystemHive->ReleaseCellRoutine == NULL );
|
||
|
|
//
|
||
|
|
// get hive.cell of root of current control set
|
||
|
|
//
|
||
|
|
RootCell = ((PHBASE_BLOCK)SystemHiveBuffer)->RootCell;
|
||
|
|
RtlInitUnicodeString(&Name, L"current");
|
||
|
|
BaseCell = CmpFindControlSet(
|
||
|
|
SystemHive,
|
||
|
|
RootCell,
|
||
|
|
&Name,
|
||
|
|
&AutoSelect
|
||
|
|
);
|
||
|
|
if (BaseCell == HCELL_NIL) {
|
||
|
|
CM_BUGCHECK(BAD_SYSTEM_CONFIG_INFO,BAD_SYSTEM_CONTROL_VALUES,2,SystemHive,&Name);
|
||
|
|
}
|
||
|
|
|
||
|
|
Node = (PCM_KEY_NODE)HvGetCell(SystemHive,BaseCell);
|
||
|
|
if( Node == NULL ) {
|
||
|
|
//
|
||
|
|
// we couldn't map a view for the bin containing this cell
|
||
|
|
//
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
RtlInitUnicodeString(&Name, L"control");
|
||
|
|
BaseCell = CmpFindSubKeyByName(SystemHive,
|
||
|
|
Node,
|
||
|
|
&Name);
|
||
|
|
if (BaseCell == HCELL_NIL) {
|
||
|
|
CM_BUGCHECK(BAD_SYSTEM_CONFIG_INFO,BAD_SYSTEM_CONTROL_VALUES,3,Node,&Name);
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// SystemHive.BaseCell = \registry\machine\system\currentcontrolset\control
|
||
|
|
//
|
||
|
|
|
||
|
|
//
|
||
|
|
// step through vector, trying to fetch each value
|
||
|
|
//
|
||
|
|
while (ControlVector->KeyPath != NULL) {
|
||
|
|
|
||
|
|
//
|
||
|
|
// Assume we will fail to find the key or value.
|
||
|
|
//
|
||
|
|
|
||
|
|
Length = (ULONG)-1;
|
||
|
|
|
||
|
|
KeyCell = CmpWalkPath(SystemHive, BaseCell, ControlVector->KeyPath);
|
||
|
|
|
||
|
|
if (KeyCell != HCELL_NIL) {
|
||
|
|
|
||
|
|
//
|
||
|
|
// found the key, look for the value entry
|
||
|
|
//
|
||
|
|
Node = (PCM_KEY_NODE)HvGetCell(SystemHive,KeyCell);
|
||
|
|
if( Node == NULL ) {
|
||
|
|
//
|
||
|
|
// we couldn't map a view for the bin containing this cell
|
||
|
|
//
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
RtlInitUnicodeString(&Name, ControlVector->ValueName);
|
||
|
|
ValueCell = CmpFindValueByName(SystemHive,
|
||
|
|
Node,
|
||
|
|
&Name);
|
||
|
|
if (ValueCell != HCELL_NIL) {
|
||
|
|
|
||
|
|
//
|
||
|
|
// SystemHive.ValueCell is value entry body
|
||
|
|
//
|
||
|
|
|
||
|
|
if (ControlVector->BufferLength == NULL) {
|
||
|
|
tmplength = sizeof(ULONG);
|
||
|
|
} else {
|
||
|
|
tmplength = *(ControlVector->BufferLength);
|
||
|
|
}
|
||
|
|
|
||
|
|
ValueBody = (PCM_KEY_VALUE)HvGetCell(SystemHive, ValueCell);
|
||
|
|
if( ValueBody == NULL ) {
|
||
|
|
//
|
||
|
|
// we couldn't map a view for the bin containing this cell
|
||
|
|
//
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
small = CmpIsHKeyValueSmall(Length, ValueBody->DataLength);
|
||
|
|
|
||
|
|
if (tmplength < Length) {
|
||
|
|
Length = tmplength;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (Length > 0) {
|
||
|
|
|
||
|
|
PCELL_DATA Buffer;
|
||
|
|
BOOLEAN BufferAllocated;
|
||
|
|
ULONG realsize;
|
||
|
|
HCELL_INDEX CellToRelease;
|
||
|
|
|
||
|
|
ASSERT((small ? (Length <= CM_KEY_VALUE_SMALL) : TRUE));
|
||
|
|
//
|
||
|
|
// get the data from source, regardless of the size
|
||
|
|
//
|
||
|
|
if( CmpGetValueData(SystemHive,ValueBody,&realsize,&Buffer,&BufferAllocated,&CellToRelease) == FALSE ) {
|
||
|
|
//
|
||
|
|
// insufficient resources; return NULL
|
||
|
|
//
|
||
|
|
ASSERT( BufferAllocated == FALSE );
|
||
|
|
ASSERT( Buffer == NULL );
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
RtlCopyMemory(
|
||
|
|
ControlVector->Buffer,
|
||
|
|
Buffer,
|
||
|
|
Length
|
||
|
|
);
|
||
|
|
|
||
|
|
//
|
||
|
|
// cleanup the temporary buffer
|
||
|
|
//
|
||
|
|
if( BufferAllocated == TRUE ) {
|
||
|
|
ExFreePool( Buffer );
|
||
|
|
}
|
||
|
|
if( CellToRelease != HCELL_NIL ) {
|
||
|
|
HvReleaseCell(SystemHive,CellToRelease);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ControlVector->Type != NULL) {
|
||
|
|
*(ControlVector->Type) = ValueBody->Type;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Stash the length of result (-1 if nothing was found)
|
||
|
|
//
|
||
|
|
|
||
|
|
if (ControlVector->BufferLength != NULL) {
|
||
|
|
*(ControlVector->BufferLength) = Length;
|
||
|
|
}
|
||
|
|
|
||
|
|
ControlVector++;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get the default locale ID for the system from the registry.
|
||
|
|
//
|
||
|
|
|
||
|
|
if (CmDefaultLanguageIdType == REG_SZ) {
|
||
|
|
PsDefaultSystemLocaleId = (LCID) CmpConvertLangId(
|
||
|
|
CmDefaultLanguageId,
|
||
|
|
CmDefaultLanguageIdLength);
|
||
|
|
} else {
|
||
|
|
PsDefaultSystemLocaleId = 0x00000409;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get the install (native UI) language ID for the system from the registry.
|
||
|
|
//
|
||
|
|
|
||
|
|
if (CmInstallUILanguageIdType == REG_SZ) {
|
||
|
|
PsInstallUILanguageId = CmpConvertLangId(
|
||
|
|
CmInstallUILanguageId,
|
||
|
|
CmInstallUILanguageIdLength);
|
||
|
|
} else {
|
||
|
|
PsInstallUILanguageId = LANGIDFROMLCID(PsDefaultSystemLocaleId);
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Set the default thread locale to the default system locale
|
||
|
|
// for now. This will get changed as soon as somebody logs in.
|
||
|
|
// Use the install (native) language id as our default UI language id.
|
||
|
|
// This also will get changed as soon as somebody logs in.
|
||
|
|
//
|
||
|
|
|
||
|
|
PsDefaultThreadLocaleId = PsDefaultSystemLocaleId;
|
||
|
|
PsDefaultUILanguageId = PsInstallUILanguageId;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HCELL_INDEX
|
||
|
|
CmpWalkPath(
|
||
|
|
PHHIVE SystemHive,
|
||
|
|
HCELL_INDEX ParentCell,
|
||
|
|
PWSTR Path
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Walk the path.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
SystemHive - hive
|
||
|
|
|
||
|
|
ParentCell - where to start
|
||
|
|
|
||
|
|
Path - string to walk
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
HCELL_INDEX of found key cell, or HCELL_NIL for error
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
NTSTATUS status;
|
||
|
|
UNICODE_STRING PathString;
|
||
|
|
UNICODE_STRING NextName;
|
||
|
|
BOOLEAN Last;
|
||
|
|
PHCELL_INDEX Index;
|
||
|
|
HCELL_INDEX KeyCell;
|
||
|
|
PCM_KEY_NODE Node;
|
||
|
|
|
||
|
|
//
|
||
|
|
// don't bother counting/releasing used cells
|
||
|
|
//
|
||
|
|
ASSERT( SystemHive->ReleaseCellRoutine == NULL );
|
||
|
|
|
||
|
|
KeyCell = ParentCell;
|
||
|
|
RtlInitUnicodeString(&PathString, Path);
|
||
|
|
|
||
|
|
while (TRUE) {
|
||
|
|
|
||
|
|
CmpGetNextName(&PathString, &NextName, &Last);
|
||
|
|
|
||
|
|
if (NextName.Length == 0) {
|
||
|
|
return KeyCell;
|
||
|
|
}
|
||
|
|
|
||
|
|
Node = (PCM_KEY_NODE)HvGetCell(SystemHive,KeyCell);
|
||
|
|
if( Node == NULL ) {
|
||
|
|
//
|
||
|
|
// we couldn't map a view for the bin containing this cell
|
||
|
|
//
|
||
|
|
return HCELL_NIL;
|
||
|
|
}
|
||
|
|
KeyCell = CmpFindSubKeyByName(SystemHive,
|
||
|
|
Node,
|
||
|
|
&NextName);
|
||
|
|
|
||
|
|
if (KeyCell == HCELL_NIL) {
|
||
|
|
return HCELL_NIL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
LANGID
|
||
|
|
CmpConvertLangId(
|
||
|
|
PWSTR LangIdString,
|
||
|
|
ULONG LangIdStringLength
|
||
|
|
)
|
||
|
|
{
|
||
|
|
|
||
|
|
|
||
|
|
USHORT i, Digit;
|
||
|
|
WCHAR c;
|
||
|
|
LANGID LangId;
|
||
|
|
|
||
|
|
LangId = 0;
|
||
|
|
LangIdStringLength = LangIdStringLength / sizeof( WCHAR );
|
||
|
|
for (i=0; i < LangIdStringLength; i++) {
|
||
|
|
c = LangIdString[ i ];
|
||
|
|
|
||
|
|
if (c >= L'0' && c <= L'9') {
|
||
|
|
Digit = c - L'0';
|
||
|
|
|
||
|
|
} else if (c >= L'A' && c <= L'F') {
|
||
|
|
Digit = c - L'A' + 10;
|
||
|
|
|
||
|
|
} else if (c >= L'a' && c <= L'f') {
|
||
|
|
Digit = c - L'a' + 10;
|
||
|
|
|
||
|
|
} else {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (Digit >= 16) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
LangId = (LangId << 4) | Digit;
|
||
|
|
}
|
||
|
|
|
||
|
|
return LangId;
|
||
|
|
}
|
||
|
|
|