《Windows内核安全与驱动开发》 4.4 线程与事件

《Windows内核安全与驱动开发》阅读笔记 -- 索引目录

 

《Windows内核安全与驱动开发》 4.4 线程与事件

一、开辟一个线程,参数为(打印内容+打印次数),利用线程实现打印。(申请堆内存防止栈清空)

#include 

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, 0, 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, 0, NULL, NULL, NULL, MyThradProc, (PVOID)&p);

    if (!NT_SUCCESS(status)) {
        return ;
    }

    //
    // 等待各个线程结束之后再退出
    //
    KeWaitForSingleObject(&event, Executive, KernelMode, 0, 0);
}

//提供一个卸载函数,让程序能卸载,如果没有这个函数,驱动将不能卸载。
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, 0, &my_interval);
    }

 

三、利用Event事件来实现线程同步(可以利用栈内存来作为线程参数,线程完成之后,主线程才结束释放栈内存)

#include 

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, 0, 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, 0, NULL, NULL, NULL, MyThradProc, (PVOID)&p);

    if (!NT_SUCCESS(status)) {
        return ;
    }

    //
    // 等待各个线程结束之后再退出
    //
    KeWaitForSingleObject(&event, Executive, KernelMode, 0, 0);
}

//提供一个卸载函数,让程序能卸载,如果没有这个函数,驱动将不能卸载。
VOID UnDriver(PDRIVER_OBJECT driver)
{
    KdPrint(("卸载驱动成功"));
}
//入口函数,相当于main。
NTSTATUS DriverEntry(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path)
{
    TestEvent();
    driver->DriverUnload = UnDriver;

    return STATUS_SUCCESS;
}

 

你可能感兴趣的:(《Windows内核安全与驱动开发》 4.4 线程与事件)