《Windows内核安全与驱动开发》阅读笔记 -- 索引目录
《Windows内核安全与驱动开发》 4.4 线程与事件
一、开辟一个线程，参数为（打印内容+打印次数），利用线程实现打印。（申请堆内存防止栈清空）
#include <ntifs.h>
typedef struct {
int num;
UNICODE_STRING str;
}MyContext, * PMyContext;
static KEVENT event; // 全局变量中定义一个事件
/*
函数作用：线程函数，探究Event的作用
参数1 context - 线程参数
*/
void MyThradProc(IN PVOID context) {
PMyContext p = (PMyContext)context;
//
// 将内容打印到日志中
//
DbgPrint("num is %wZ", p->str);
DbgPrint("这是在次线程中");
//
// 激活该事件
//
KeSetEvent(&event, , FALSE);
PsTerminateSystemThread(STATUS_SUCCESS);
}
void TestEvent() {
HANDLE ThreadHandle; // 线程数组
MyContext p; // 线程参数数组
NTSTATUS status; // 状态
//
// 初始化时间，让主线程等待
//
KeInitializeEvent(&event, SynchronizationEvent, FALSE);
//
// 为线程参数数组开辟内存
//
RtlInitUnicodeString(&p.str, L"hello world!");
DbgPrint("这是在主线程中··\n");
DbgPrint("字符串为 :%wZ\n", &p.str);
//
// 开辟线程
//
status = PsCreateSystemThread(&ThreadHandle, , NULL, NULL, NULL, MyThradProc, (PVOID)&p);
if (!NT_SUCCESS(status)) {
return ;
}
//
// 等待各个线程结束之后再退出
//
KeWaitForSingleObject(&event, Executive, KernelMode, , );
}
//提供一个卸载函数，让程序能卸载，如果没有这个函数，驱动将不能卸载。
VOID UnDriver(PDRIVER_OBJECT driver)
{
KdPrint(("卸载驱动成功"));
}
//入口函数，相当于main。
NTSTATUS DriverEntry(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path)
{
TestEvent();
driver->DriverUnload = UnDriver;
return STATUS_SUCCESS;
}
二、实现一个线程挂起函数
/
// 线程睡眠
//
#define DELAY_ONE_MICROSECOND (-10)
#define DELAY_ONE_MILLISECOND (DELAY_ONE_MICROSECOND*1000);
VOID MySleep(LONG msec) {
LARGE_INTEGER my_interval;
my_interval.QuadPart = DELAY_ONE_MILLISECOND;
my_interval.QuadPart *= msec;
KeDelayExecutionThread(KernelMode, , &my_interval);
}
三、利用Event事件来实现线程同步（可以利用栈内存来作为线程参数，线程完成之后，主线程才结束释放栈内存）
#include <ntifs.h>
typedef struct {
int num;
UNICODE_STRING str;
}MyContext, * PMyContext;
static KEVENT event; // 全局变量中定义一个事件
/*
函数作用：线程函数，探究Event的作用
参数1 context - 线程参数
*/
void MyThradProc(IN PVOID context) {
PMyContext p = (PMyContext)context;
//
// 将内容打印到日志中
//
DbgPrint("num is %wZ", p->str);
DbgPrint("这是在次线程中");
//
// 激活该事件
//
KeSetEvent(&event, , FALSE);
PsTerminateSystemThread(STATUS_SUCCESS);
}
void TestEvent() {
HANDLE ThreadHandle; // 线程数组
MyContext p; // 线程参数数组
NTSTATUS status; // 状态
//
// 初始化时间，让主线程等待
//
KeInitializeEvent(&event, SynchronizationEvent, FALSE);
//
// 为线程参数数组开辟内存
//
RtlInitUnicodeString(&p.str, L"hello world!");
DbgPrint("这是在主线程中··\n");
DbgPrint("字符串为 :%wZ\n", &p.str);
//
// 开辟线程
//
status = PsCreateSystemThread(&ThreadHandle, , NULL, NULL, NULL, MyThradProc, (PVOID)&p);
if (!NT_SUCCESS(status)) {
return ;
}
//
// 等待各个线程结束之后再退出
//
KeWaitForSingleObject(&event, Executive, KernelMode, , );
}
//提供一个卸载函数，让程序能卸载，如果没有这个函数，驱动将不能卸载。
VOID UnDriver(PDRIVER_OBJECT driver)
{
KdPrint(("卸载驱动成功"));
}
//入口函数，相当于main。
NTSTATUS DriverEntry(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path)
{
TestEvent();
driver->DriverUnload = UnDriver;
return STATUS_SUCCESS;
}
