reactos操作系统实现(74)

很多驱动程序，都是通过硬件检测时写到注册表里保存起来的，因此读取注册表里键值列表，就知道有多少设备需要需要加载了。下面就从注册表里创建一组驱动程序列表，实现的代码如下：

#001 NTSTATUS INIT_FUNCTION

#002 IoCreateDriverList(VOID)

#003 {

#004 RTL_QUERY_REGISTRY_TABLE
QueryTable[2];

#005 PKEY_BASIC_INFORMATION
KeyInfo = NULL;

#006 OBJECT_ATTRIBUTES
ObjectAttributes;

注册表里的键。

#007 UNICODE_STRING
ServicesKeyName =
RTL_CONSTANT_STRING(L"//Registry//Machine//System//CurrentControlSet//Services");

#008 UNICODE_STRING
SubKeyName;

#009 HANDLE
KeyHandle;

#010 NTSTATUS
Status;

#011 ULONG
Index;

#012

#013 ULONG
KeyInfoLength = 0;

#014 ULONG
ReturnedLength;

#015

#016 DPRINT("IoCreateDriverList()
called/n");

#017

初始化全局变量的驱动程序组和服务列表。

#018 /*
Initialize basic variables */

#019 InitializeListHead(&GroupListHead);

#020 InitializeListHead(&ServiceListHead);

#021

清空查询表的数据。

#022 /*
Build group order list */

#023 RtlZeroMemory(&QueryTable,

#024 sizeof(QueryTable));

#025

设置查询的名称。

#026 QueryTable[0].Name
= L"List";

设置查询的回调函数。

#027 QueryTable[0].QueryRoutine
= IopCreateGroupListEntry;

#028

建立组列表。

#029 Status
= RtlQueryRegistryValues(RTL_REGISTRY_CONTROL,

#030 L"ServiceGroupOrder",

#031 QueryTable,

#032 NULL,

#033 NULL);

#034 if
(!NT_SUCCESS(Status))

#035 return(Status);

#036

枚举服务和创建服务列表。

#037 /*
Enumerate services and create the service list */

#038 InitializeObjectAttributes(&ObjectAttributes,

#039 &ServicesKeyName,

#040
OBJ_CASE_INSENSITIVE,

#041 NULL,

#042 NULL);

#043

打开注册表的键。

#044 Status
= ZwOpenKey(&KeyHandle,

#045 KEY_ENUMERATE_SUB_KEYS,

#046 &ObjectAttributes);

#047 if
(!NT_SUCCESS(Status))

#048 {

#049 return(Status);

#050 }

#051

#052 KeyInfoLength
= sizeof(KEY_BASIC_INFORMATION) + MAX_PATH * sizeof(WCHAR);

#053 KeyInfo
= ExAllocatePool(NonPagedPool, KeyInfoLength);

#054 if
(KeyInfo == NULL)

#055 {

#056 ZwClose(KeyHandle);

#057 return(STATUS_INSUFFICIENT_RESOURCES);

#058 }

#059

#060 Index
= 0;

循环地枚举所有注册表里的键。

#061 while
(TRUE)

#062 {

#063 Status
= ZwEnumerateKey(KeyHandle,

#064 Index,

#065 KeyBasicInformation,

#066 KeyInfo,

#067 KeyInfoLength,

#068 &ReturnedLength);

#069 if
(NT_SUCCESS(Status))

#070 {

#071 if
(KeyInfo->NameLength < MAX_PATH * sizeof(WCHAR))

#072 {

#073

#074 SubKeyName.Length
= (USHORT)KeyInfo->NameLength;

#075 SubKeyName.MaximumLength
= (USHORT)KeyInfo->NameLength + sizeof(WCHAR);

#076 SubKeyName.Buffer
= KeyInfo->Name;

#077 SubKeyName.Buffer[SubKeyName.Length
/ sizeof(WCHAR)] = 0;

#078

#079 DPRINT("KeyName:
'%wZ'/n", &SubKeyName);

每个子键都创建一个入口，保存到全局变量ServiceListHead里。

#080 IopCreateServiceListEntry(&SubKeyName);

#081 }

#082 }

#083

#084 if
(!NT_SUCCESS(Status))

#085 break;

#086

#087 Index++;

#088 }

#089

#090 ExFreePool(KeyInfo);

#091 ZwClose(KeyHandle);

#092

#093 DPRINT("IoCreateDriverList()
done/n");

#094

#095 return(STATUS_SUCCESS);

#096 }

#097