reactos操作系统实现(74)

来源：互联网发布：淘宝客类目佣金设置编辑：程序博客网时间：2024/06/05 02:56

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

#001 NTSTATUS INIT_FUNCTION

#002 IoCreateDriverList(VOID)

#003 {

#004 RTL_QUERY_REGISTRY_TABLEQueryTable[2];

#005 PKEY_BASIC_INFORMATIONKeyInfo = NULL;

#006 OBJECT_ATTRIBUTESObjectAttributes;

注册表里的键。

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

#008 UNICODE_STRINGSubKeyName;

#009 HANDLEKeyHandle;

#010 NTSTATUSStatus;

#011 ULONGIndex;

#012

#013 ULONGKeyInfoLength = 0;

#014 ULONGReturnedLength;

#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