快捷導(dǎo)航

詳解C語言內(nèi)核中的自旋鎖結(jié)構(gòu)

更新時間：2022年09月29日 15:35:50 作者：lyshark

自旋鎖是內(nèi)核中提供的一種高IRQL鎖，用同步以及獨占的方式訪問某個資源。自旋鎖是為了解決內(nèi)核鏈表讀寫時存在線程同步問題。本文將講解一下自旋鎖的簡單應(yīng)用，感興趣的可以了解一下

提到自旋鎖那就必須要說鏈表，在上一篇《驅(qū)動開發(fā)：內(nèi)核中的鏈表與結(jié)構(gòu)體》文章中簡單實用鏈表結(jié)構(gòu)來存儲進程信息列表，相信讀者應(yīng)該已經(jīng)理解了內(nèi)核鏈表的基本使用，本篇文章將講解自旋鎖的簡單應(yīng)用，自旋鎖是為了解決內(nèi)核鏈表讀寫時存在線程同步問題，解決多線程同步問題必須要用鎖，通常使用自旋鎖，自旋鎖是內(nèi)核中提供的一種高IRQL鎖，用同步以及獨占的方式訪問某個資源。

首先以簡單的鏈表為案例，鏈表主要分為單向鏈表與雙向鏈表，單向鏈表的鏈表節(jié)點中只有一個鏈表指針，其指向后一個鏈表元素，而雙向鏈表節(jié)點中有兩個鏈表節(jié)點指針，其中Blink指向前一個鏈表節(jié)點Flink指向后一個節(jié)點，以雙向鏈表為例。

#include <ntifs.h>
#include <ntstrsafe.h>

/*
// 鏈表節(jié)點指針
typedef struct _LIST_ENTRY
{
  struct _LIST_ENTRY *Flink;   // 當前節(jié)點的后一個節(jié)點
  struct _LIST_ENTRY *Blink;   // 當前節(jié)點的前一個結(jié)點
}LIST_ENTRY, *PLIST_ENTRY;
*/

typedef struct _MyStruct
{
  ULONG x;
  ULONG y;
  LIST_ENTRY lpListEntry;
}MyStruct,*pMyStruct;

VOID UnDriver(PDRIVER_OBJECT driver)
{
  DbgPrint("驅(qū)動卸載成功 \n");
}

// By: LyShark
NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{
  DbgPrint("By:LyShark \n");
  DbgPrint("Email:me@lyshark.com \n");
  // 初始化頭節(jié)點
  LIST_ENTRY ListHeader = { 0 };
  InitializeListHead(&ListHeader);

  // 定義鏈表元素
  MyStruct testA = { 0 };
  MyStruct testB = { 0 };
  MyStruct testC = { 0 };

  testA.x = 100;
  testA.y = 200;

  testB.x = 1000;
  testB.y = 2000;

  testC.x = 10000;
  testC.y = 20000;

  // 分別插入節(jié)點到頭部和尾部
  InsertHeadList(&ListHeader, &testA.lpListEntry);
  InsertTailList(&ListHeader, &testB.lpListEntry);
  InsertTailList(&ListHeader, &testC.lpListEntry);

  // 節(jié)點不為空 則 移除一個節(jié)點
  if (IsListEmpty(&ListHeader) == FALSE)
  {
    RemoveEntryList(&testA.lpListEntry);
  }

  // 輸出鏈表數(shù)據(jù)
  PLIST_ENTRY pListEntry = NULL;
  pListEntry = ListHeader.Flink;

  while (pListEntry != &ListHeader)
  {
    // 計算出成員距離結(jié)構(gòu)體頂部內(nèi)存距離
    pMyStruct ptr = CONTAINING_RECORD(pListEntry, MyStruct, lpListEntry);
    DbgPrint("節(jié)點元素X = %d 節(jié)點元素Y = %d \n", ptr->x, ptr->y);

    // 得到下一個元素地址
    pListEntry = pListEntry->Flink;
  }

  Driver->DriverUnload = UnDriver;
  return STATUS_SUCCESS;
}

鏈表輸出效果如下：

如上所述，內(nèi)核鏈表讀寫時存在線程同步問題，解決多線程同步問題必須要用鎖，通常使用自旋鎖，自旋鎖是內(nèi)核中提供的一種高IRQL鎖，用同步以及獨占的方式訪問某個資源。

#include <ntifs.h>
#include <ntstrsafe.h>

/*
// 鏈表節(jié)點指針
typedef struct _LIST_ENTRY
{
struct _LIST_ENTRY *Flink;   // 當前節(jié)點的后一個節(jié)點
struct _LIST_ENTRY *Blink;   // 當前節(jié)點的前一個結(jié)點
}LIST_ENTRY, *PLIST_ENTRY;
*/

typedef struct _MyStruct
{
	ULONG x;
	ULONG y;
	LIST_ENTRY lpListEntry;
}MyStruct, *pMyStruct;

// 定義全局鏈表和全局鎖
LIST_ENTRY my_list_header;
KSPIN_LOCK my_list_lock;

// 初始化
void Init()
{
	InitializeListHead(&my_list_header);
	KeInitializeSpinLock(&my_list_lock);
}

// 函數(shù)內(nèi)使用鎖
void function_ins()
{
	KIRQL Irql;

	// 加鎖
	KeAcquireSpinLock(&my_list_lock, &Irql);

	DbgPrint("鎖內(nèi)部執(zhí)行 \n");

	// 釋放鎖
	KeReleaseSpinLock(&my_list_lock, Irql);
}

VOID UnDriver(PDRIVER_OBJECT driver)
{
	DbgPrint("驅(qū)動卸載成功 \n");
}

// By: LyShark
NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{
	DbgPrint("By:LyShark \n");
	DbgPrint("Email:me@lyshark.com \n");

	// 初始化鏈表
	Init();

	// 分配鏈表空間
	pMyStruct testA = (pMyStruct)ExAllocatePool(NonPagedPoolExecute, sizeof(pMyStruct));
	pMyStruct testB = (pMyStruct)ExAllocatePool(NonPagedPoolExecute, sizeof(pMyStruct));

	// 賦值
	testA->x = 100;
	testA->y = 200;

	testB->x = 1000;
	testB->y = 2000;

	// 向全局鏈表中插入數(shù)據(jù)
	if (NULL != testA && NULL != testB)
	{
		ExInterlockedInsertHeadList(&my_list_header, (PLIST_ENTRY)&testA->lpListEntry, &my_list_lock);
		ExInterlockedInsertTailList(&my_list_header, (PLIST_ENTRY)&testB->lpListEntry, &my_list_lock);
	}

	function_ins();

	// 移除節(jié)點A并放入到remove_entry中
	PLIST_ENTRY remove_entry = ExInterlockedRemoveHeadList(&testA->lpListEntry, &my_list_lock);

	// 輸出鏈表數(shù)據(jù)
	while (remove_entry != &my_list_header)
	{
		// 計算出成員距離結(jié)構(gòu)體頂部內(nèi)存距離
		pMyStruct ptr = CONTAINING_RECORD(remove_entry, MyStruct, lpListEntry);
		DbgPrint("節(jié)點元素X = %d 節(jié)點元素Y = %d \n", ptr->x, ptr->y);

		// 得到下一個元素地址
		remove_entry = remove_entry->Flink;
	}

	Driver->DriverUnload = UnDriver;
	return STATUS_SUCCESS;
}

加鎖后執(zhí)行效果如下：