20.driverbase-CONTAINING_RECODE和双向链表

CONTAINING_RECODE

#define CONTAINING_RECORD(address, type, field) ((type *)( \
                                                  (PCHAR)(address) - \
                                                  (ULONG_PTR)(&((type *)0)->field)))
addr:  结构体中某个成员变量的地址
type:  结构体的原型
field: 结构体的某个成员(与前面相同)

根据结构体中的某成员的地址来推算出该结构体整体的地址,这是个万能公式，如下例：

typedef struct _MYDATASTRUCT 
{
	ULONG number;
	LIST_ENTRY ListEntry;
} MYDATASTRUCT, *PMYDATASTRUCT;
PLIST_ENTRY pEntry = RemoveTailList(&linkListHead);
PMYDATASTRUCT	pData = CONTAINING_RECORD(pEntry,
                              MYDATASTRUCT,
                              ListEntry);

含义如下:pEntry代表MYDATASTRUCT的变量中的ListEntry的地址，pData为MYDATASTRUCT的变量地址
      

双向链表

参看WRK：public\sdk\inc\ntrtl.h

BLINK指向前一个元素,FINK指向下一个元素，注意，指向的总是LIST_ENTRY,而LIST_ENTRY可以在自定义结构的任意位置

typedef struct _LIST_ENTRY {
   struct _LIST_ENTRY *Flink;
   struct _LIST_ENTRY *Blink;
} LIST_ENTRY, *PLIST_ENTRY, *RESTRICTED_POINTER PRLIST_ENTRY;

1.首先初始化链表，即Flink和Blink指向自身

VOID FORCEINLINE InitializeListHead(IN PLIST_ENTRY ListHead)
{
    ListHead->Flink = ListHead->Blink = ListHead;
}

2.判断链表是否为空，即Flink和Blink指向自身

#define IsListEmpty(ListHead) \
    ((ListHead)->Flink == (ListHead))

3.向头部插入链表，即插入到ListHead以后

VOID
FORCEINLINE
InsertHeadList(
    IN PLIST_ENTRY ListHead,
    IN PLIST_ENTRY Entry
    )
{
    PLIST_ENTRY Flink;

    Flink = ListHead->Flink;
    Entry->Flink = Flink;
    Entry->Blink = ListHead;
    Flink->Blink = Entry;
    ListHead->Flink = Entry;// 插入到ListHead以后
}

前向插入Entry2

4.向尾部插入链表

VOID
FORCEINLINE
InsertTailList(
    IN PLIST_ENTRY ListHead,
    IN PLIST_ENTRY Entry
    )
{
    PLIST_ENTRY Blink;

    Blink = ListHead->Blink;// 取得原始尾部
    Entry->Flink = ListHead;// 新的尾部的后一元素是头（环）
    Entry->Blink = Blink;   // 新的尾部前一元素是原始尾部
    Blink->Flink = Entry;   // 原始尾部的后一元素是新的尾部
    ListHead->Blink = Entry;// 头的前一元素是新的尾部（环）
}

后向插入Entry3

演示代码

VOID LinkListTest() 
{
	LIST_ENTRY linkListHead;
	//初始化链表
	InitializeListHead(&linkListHead);

	PMYDATASTRUCT pData;
	ULONG i = 0;
	//在链表中插入10个元素
	KdPrint(("Begin insert to link list"));
	for (i=0 ; i<10 ; i++)
	{
		pData = (PMYDATASTRUCT)
			ExAllocatePool(PagedPool,sizeof(MYDATASTRUCT));
		pData->number = i;
		InsertHeadList(&linkListHead,&pData->ListEntry);
	}

	//从链表中取出，并显示
	KdPrint(("Begin remove from link list\n"));
	while(!IsListEmpty(&linkListHead))
	{
		PLIST_ENTRY pEntry = RemoveTailList(&linkListHead);
		pData = CONTAINING_RECORD(pEntry,
                              MYDATASTRUCT,
                              ListEntry);
		KdPrint(("%d\n",pData->number));
		ExFreePool(pData);
	}
}