快捷導(dǎo)航

詳解C語言內(nèi)核中的鏈表與結(jié)構(gòu)體

更新時(shí)間：2022年09月29日 15:02:18 作者：lyshark

Windows內(nèi)核中是無法使用vector容器等數(shù)據(jù)結(jié)構(gòu)的，當(dāng)我們需要保存一個(gè)結(jié)構(gòu)體數(shù)組時(shí)，就需要使用內(nèi)核中提供的專用鏈表結(jié)構(gòu)。本文分享了幾個(gè)內(nèi)核中使用鏈表存儲(chǔ)多個(gè)結(jié)構(gòu)體的通用案例，希望對(duì)你有所幫助

Windows內(nèi)核中是無法使用vector容器等數(shù)據(jù)結(jié)構(gòu)的，當(dāng)我們需要保存一個(gè)結(jié)構(gòu)體數(shù)組時(shí)，就需要使用內(nèi)核中提供的專用鏈表結(jié)構(gòu)LIST_ENTRY通過一些列鏈表操作函數(shù)對(duì)結(jié)構(gòu)體進(jìn)行裝入彈出等操作，如下代碼是本人總結(jié)的內(nèi)核中使用鏈表存儲(chǔ)多個(gè)結(jié)構(gòu)體的通用案例。

首先實(shí)現(xiàn)一個(gè)枚舉用戶進(jìn)程功能，將枚舉到的進(jìn)程存儲(chǔ)到鏈表結(jié)構(gòu)體內(nèi)。

#include <ntifs.h>
#include <windef.h>

extern PVOID PsGetProcessPeb(_In_ PEPROCESS Process);
NTKERNELAPI NTSTATUS PsLookupProcessByProcessId(HANDLE ProcessId, PEPROCESS *Process);
extern NTKERNELAPI PVOID PsGetProcessWow64Process(_In_ PEPROCESS Process);
extern NTKERNELAPI UCHAR* PsGetProcessImageFileName(IN PEPROCESS Process);
extern NTKERNELAPI HANDLE PsGetProcessInheritedFromUniqueProcessId(IN PEPROCESS Process);

typedef struct
{
	DWORD Pid;
	UCHAR ProcessName[2048];
	DWORD Handle;
	LIST_ENTRY ListEntry;
}ProcessList;

// 根據(jù)進(jìn)程ID返回進(jìn)程EPROCESS結(jié)構(gòu)體失敗返回NULL
PEPROCESS LookupProcess(HANDLE Pid)
{
	PEPROCESS eprocess = NULL;
	NTSTATUS Status = STATUS_UNSUCCESSFUL;
	Status = PsLookupProcessByProcessId(Pid, &eprocess);
	if (NT_SUCCESS(Status))
	{
		return eprocess;
	}
	return NULL;
}

// 內(nèi)核鏈表操作
// By: LyShark
BOOLEAN GetAllProcess()
{
	PEPROCESS eproc = NULL;
	LIST_ENTRY linkListHead;

	// 初始化鏈表頭部
	InitializeListHead(&linkListHead);
	ProcessList *pData = NULL;

	for (int temp = 0; temp < 100000; temp += 4)
	{
		eproc = LookupProcess((HANDLE)temp);
		if (eproc != NULL)
		{
			STRING nowProcessnameString = { 0 };
			RtlInitString(&nowProcessnameString, PsGetProcessImageFileName(eproc));

			// DbgPrint("進(jìn)程名: %s --> 進(jìn)程PID = %d --> 父進(jìn)程PPID = %d\r\n", 
			// PsGetProcessImageFileName(eproc), PsGetProcessId(eproc), PsGetProcessInheritedFromUniqueProcessId(eproc));

			// 分配內(nèi)核堆空間
			pData = (ProcessList *)ExAllocatePool(PagedPool, sizeof(ProcessList));
			RtlZeroMemory(pData, sizeof(ProcessList));

			// 設(shè)置變量
			pData->Pid = (DWORD)PsGetProcessId(eproc);
			RtlCopyMemory(pData->ProcessName, PsGetProcessImageFileName(eproc), strlen(PsGetProcessImageFileName(eproc)) * 2);
			pData->Handle = (DWORD)PsGetProcessInheritedFromUniqueProcessId(eproc);

			// 插入元素到
			InsertTailList(&linkListHead, &pData->ListEntry);
			ObDereferenceObject(eproc);
		}
	}

	// 輸出鏈表內(nèi)的數(shù)據(jù)
	while (!IsListEmpty(&linkListHead))
	{
		LIST_ENTRY *pEntry = RemoveHeadList(&linkListHead);
		pData = CONTAINING_RECORD(pEntry, ProcessList, ListEntry);

		DbgPrint("%d \n", pData->Pid);
		DbgPrint("%s \n", pData->ProcessName);
		DbgPrint("%d \n", pData->Handle);
		ExFreePool(pData);
	}
	return TRUE;
}

VOID UnDriver(PDRIVER_OBJECT driver)
{
	DbgPrint(("Uninstall Driver Is OK \n"));
}

NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{
	DbgPrint("hello lyshark.com \n");

	GetAllProcess();

	Driver->DriverUnload = UnDriver;
	return STATUS_SUCCESS;
}

運(yùn)行后將可以在DbgView中看到輸出的進(jìn)程信息：

如果需要返回一個(gè)結(jié)構(gòu)體，則可以這樣來寫代碼。

#include <ntifs.h>
#include <windef.h>

typedef struct
{
	int count;
	char username[256];
	char password[256];
}MyData;

// 模擬返回一個(gè)結(jié)構(gòu)
BOOLEAN GetProcess(PVOID OutPut)
{
	RtlZeroMemory(OutPut, sizeof(MyData));
	MyData *data = OutPut;

	data->count = 100;
	RtlCopyMemory(data->username, "lyshark.com", sizeof("lyshark.com"));
	RtlCopyMemory(data->password, "https://www.cnblogs.com/lyshark", sizeof("https://www.cnblogs.com/lyshark"));
	return TRUE;
}

VOID UnDriver(PDRIVER_OBJECT driver)
{
	DbgPrint(("Uninstall Driver Is OK \n"));
}

NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{
	DbgPrint("hello lyshark.com \n");
	PVOID Ptr = (PVOID)ExAllocatePool(NonPagedPool, sizeof(MyData));


	GetProcess(Ptr);

	MyData *data = (MyData *)Ptr;

	DbgPrint("count = %d \n", data->count);
	DbgPrint("username = %s \n", data->username);
	DbgPrint("password = %s \n", data->password);

	Driver->DriverUnload = UnDriver;
	return STATUS_SUCCESS;
}

輸出效果如下：