驱动链表（LIST_ENTRY）

Posted 2020-10-03 刘收获

DDK提供了两种链表的数据结构，双向链表和单向链表，其定义如下：

typedef struct _LIST_ENTRY

{

       struct _LIST_ENTRY *Flink;

       struct _LIST_ENTRY *Blink;

} LIST_ENTRY,*PLIST_ENTRY;

 

typedef struct _SINGLE_LIST_ENTRY {
　　struct _SINGLE_LIST_ENTRY *Next;
} SINGLE_LIST_ENTRY, *PSINGLE_LIST_ENTRY;

 

 

一.双向链表LIST_ENTRY

　　这个结构不能直接存进我们要的数据，如果要把我们的数据存进链表的结构里，需要重新定义一个结构体，结构体里必须要包括一个LIST_ENTRY类型的成员，这个成员可以放在结构体里的任何位置，如：

typedef struct _list
{
       LIST_ENTRY List;
       ULONG data;
} Node,*PNode;

或者：

typedef struct _list
{
    ULONG data;

    LIST_ENTRY List;
} Node,*PNode;

 

　　在使用链表时要定义一个链表头并初始化，其类型为LIST_ENTRY，

　　1.初始化链表函数为InitializeListHead(PLIST_ENTRY pListHeader);

        2.插入链表函数

　　InsertHeadList(

　　　　PLIST_ENTRY pListHeader,     //链头

　　　　PLIST_ENTRY MyListEntry);    //结构体中的LIST_ENTRY类型的成员变量地址

　　这个函数是在链表的开头插入一个结点，第一个参数是链头，第二个参数也是一个PLIST_ENTRY类型，是我们自己定义的结构体中的LIST_ENTRY类型的成员变量地址。类似的还有从尾部插入链表函数InsertTailList,参数与InsertHeadList一致。

　　3.从链表删除结点

　　函数RemoveTailList（PLIST_ENTRY pListHeader)把链表的最后的结点删除，返回删除的结点的指针。

　　函数RemoveHeadList（PLIST_ENTRY pListHeader)把链表的第一个结点删除，返回删除的结点的指针。

 

一.单向链表SINGLE_LIST_ENTRY

　　1.初始化链表函数为InitializeListHead(PLIST_ENTRY pListHeader)，与双向链表相同。

　　2.插入链表函数　　　

　　FORCEINLINE
　　VOID
　　PushEntryList(                                                                                        //从头部插入，成为第一个结点
　　　　_Inout_ PSINGLE_LIST_ENTRY ListHead,                              //链头                  
　　　　_Inout_ __drv_aliasesMem PSINGLE_LIST_ENTRY Entry  //结构体中的LIST_ENTRY类型的成员变量地址
　　)

　　{

　　　　Entry->Next = ListHead->Next;
　　　　ListHead->Next = Entry;
　　　　return;
　　}

　　

　　3.从链表删除结点　　

　　FORCEINLINE
　　PSINGLE_LIST_ENTRY
　　PopEntryList(                                                                 //从链表头下一个删除节点，删除第一个结点
　　　　_Inout_ PSINGLE_LIST_ENTRY ListHead      //链头   
　　)
　　{

　　PSINGLE_LIST_ENTRY FirstEntry;

　　FirstEntry = ListHead->Next;
　　if (FirstEntry != NULL) {
　　ListHead->Next = FirstEntry->Next;
　　}

　　return FirstEntry;
　　}

 

//bp ListEntry!DriverEntry

NTSTATUS DriverEntry(PDRIVER_OBJECT DriverObject, PUNICODE_STRING RegisterPath)
{
	NTSTATUS Status = STATUS_SUCCESS;
	PDEVICE_OBJECT  DeviceObject = NULL;
	DriverObject->DriverUnload = DriverUnload;


	SeListEntry();
	SeSingleListEntry();
	return Status;
}
VOID SeListEntry()
{
	LIST_ENTRY  ListHead;
	PITEM Item = NULL;
	ULONG i = 0;
	//初始化链表
	InitializeListHead(&ListHead);
	for (i = 0; i < 10; i++)
	{
		Item = (PITEM)
			ExAllocatePool(PagedPool, sizeof(ITEM));
		Item->ItemData = i;
		InsertHeadList(&ListHead, &Item->u.ListEntry);
	}
	while (!IsListEmpty(&ListHead))
	{
		PLIST_ENTRY ListEntry = RemoveTailList(&ListHead);
		
		KdPrint(("%d\n", ((PITEM)ListEntry)->ItemData));
		ExFreePool((PITEM)ListEntry);
	}
}
VOID SeSingleListEntry()
{
	SINGLE_LIST_ENTRY  ListHead;
	PITEM Item = NULL;
	ULONG i = 0;
	//初始化链表
	InitializeListHead(&ListHead);
	for (i = 0; i < 10; i++)
	{
		Item = (PITEM)
			ExAllocatePool(PagedPool, sizeof(ITEM));
		Item->ItemData = i;
		PushEntryList(&ListHead, &Item->u.SingleListEntry);
	}
	while (!IsListEmpty(&ListHead))
	{
		PSINGLE_LIST_ENTRY ListEntry = PopEntryList(&ListHead);

		KdPrint(("%d\n", ((PITEM)ListEntry)->ItemData));
		ExFreePool((PITEM)ListEntry);
	}
}

VOID DriverUnload(PDRIVER_OBJECT DriverObject)
{
	DbgPrint("DriverUnload()\r\n");
}