刘收获

导航

驱动程序多线程 PsCreateSystemThread

  内核函数PsCreateSystemThread负责创建新线程。该函数可以创建两种线程,一种是用户线程,它属于当前进程中的线程。另一种是系统线程,系统线程不属于当前用户进程,而是属于系统进程,一般PID为4,名字为“System”的进程。

    NTSTATUS 
    PsCreateSystemThread(
      OUT PHANDLE  ThreadHandle, //新创建的线程句柄
      IN ULONG  DesiredAccess, //创建的权限
      IN POBJECT_ATTRIBUTES  ObjectAttributes  OPTIONAL,//线程的属性,一般设为NULL
      IN HANDLE  ProcessHandle  OPTIONAL,//指定创建用户线程还是系统线程。如果为NULL,则为系统进程,如果该值是一个进程句柄,则新创建的线程属于这个指定的进程。DDK提供的NTCurrentProcess可以得到当前进程的句柄。
      OUT PCLIENT_ID  ClientId  OPTIONAL,
      IN PKSTART_ROUTINE  StartRoutine,//新线程的运行地址
      IN PVOID  StartContext //新线程接收的参数
      );

 

  

          在内核模式下创建的线程是无法自动退出的,必须使用PsTerminateSystemThread强制结束线程

 

         通过内核事件KEVENT和内核等待KeWaitForSingleObject来演示事件的创建过程。

   

	KEVENT Event;

	//创建一个同步事件
	KeInitializeEvent(
		&Event,               //要初始化的KEVENT结构对象
		SynchronizationEvent, //事件类型
		FALSE);       

      KeSetEvent(
        &Event, //被激活的事件
        IO_NO_INCREMENT, //被唤醒线程临时提升线程优先级的增量,传0
        FALSE); //If TRUE, the KeSetEvent call must be followed by a call to KeWaitForMultipleObjects, KeWaitForMutexObject, or KeWaitForSingleObject.

  

	KeWaitForSingleObject(
		&Event,      //同步对象的指针,
		Executive,   //等待的原因,一般为Executive
		KernelMode,  //等待模式,一般为KernelMode
		FALSE,       //指明等待是否为“警惕”的,一般为FALSE
		NULL);       //等待时间,如果为NULL,就表示无限期等待,直到同步对象变为激发态
	DbgPrint("ThreadProcedure() Exit\r\n");
	PsTerminateSystemThread(STATUS_SUCCESS);

  

   总结一下代码流程:

    1.驱动程序KeInitializeEvent创建了一个内核事件

                2.PsCreateSystemThread创建新的系统线程,将创建好的KEVENT结构传参给新创建的线程。 再调用KeSetEvent事件激活

                3.线程中调用KeWaitForSingleObject等待事件激活状态(受信),执行后续代码,最后PsTerminateSystemThread结束线程

 

  源代码:

 

KEvent.h

 

#pragma once
#include <ntifs.h>
#define DELAY_ONE_MICROSECOND 	(-10)
#define DELAY_ONE_MILLISECOND	(DELAY_ONE_MICROSECOND*1000)





VOID DriverUnload(PDRIVER_OBJECT DriverObject);
VOID KSleep(LONG MilliSecond);
void ThreadProcedure(PVOID ParameterData);

 

 

KEvent.c

#include "KEvent.h"
//http://www.cnblogs.com/dacainiao/p/5923147.html
//同步事件(SynchronizationEvent)
//当事件对象为激发时,如遇到KeWaitForXX等内核函数,事件对象则自动变回未激发态
//通知事件(NotificationEvent)
//当事件对象为激发时,如遇到KeWaitForXX等内核函数,事件对象则不会变回未激发态 
NTSTATUS DriverEntry(PDRIVER_OBJECT DriverObject, PUNICODE_STRING RegisterPath)
{
	int i = 0;
	NTSTATUS Status = STATUS_SUCCESS;
	HANDLE   ThreadHandle = NULL;
	PDEVICE_OBJECT  DeviceObject = NULL;
	KEVENT Event;

	CLIENT_ID       ClientID = {0};

	DriverObject->DriverUnload = DriverUnload;

	//创建一个同步事件
	KeInitializeEvent(
		&Event,               //要初始化的KEVENT结构对象
		SynchronizationEvent, //事件类型
		FALSE);               //如果为真,初始为激发态


	//创建一个通知事件
	//KeInitializeEvent(&Event, NotificationEvent, 
	//	FALSE);

	


	//创建一个System进程下运行的线程	
	for (i=0;i<2;i++)
	{
		Status = PsCreateSystemThread(
			&ThreadHandle,                     //新创建的线程句柄
			0,                                 //创建的权限
			NULL,                              //线程的属性,一般设为NULL
			NtCurrentProcess(),                //指定创建用户线程还是系统线程。如果为NULL,则为系统进程,如果该值是一个进程句柄,则新创建的线程属于这个指定的进程。DDK的NTCurrentProcess可以得到当前进程的句柄。
			&ClientID,
			(PKSTART_ROUTINE)ThreadProcedure,  //新线程的运行地址
			(PVOID)&Event);                    //新线程接收的参数
	}
	
	
	KSleep(3000);
	KeSetEvent(
		&Event,               //被激活的事件
		IO_NO_INCREMENT,      //被唤醒线程临时提升线程优先级的增量,传0
		FALSE);               //If TRUE, the KeSetEvent call must be followed by a call to KeWaitForMultipleObjects, KeWaitForMutexObject, or KeWaitForSingleObject.


	


	return Status;
}



void ThreadProcedure(PVOID ParameterData)
{
	KEVENT Event = *((PKEVENT)ParameterData);
	NTSTATUS Status;
	KeWaitForSingleObject(
		&Event,      //同步对象的指针,
		Executive,   //等待的原因,一般为Executive
		KernelMode,  //等待模式,一般为KernelMode
		FALSE,       //指明等待是否为“警惕”的,一般为FALSE
		NULL);       //等待时间,如果为NULL,就表示无限期等待,直到同步对象变为激发态
	DbgPrint("ThreadProcedure() Exit\r\n");
	PsTerminateSystemThread(STATUS_SUCCESS);
}

VOID DriverUnload(PDRIVER_OBJECT DriverObject)
{
	DbgPrint("DriverUnload()\r\n");
}


VOID KSleep(LONG MilliSecond)
{
	LARGE_INTEGER Interval = {0};
	Interval.QuadPart = DELAY_ONE_MILLISECOND;
	Interval.QuadPart *= MilliSecond;
	KeDelayExecutionThread(KernelMode, 0, &Interval);
}

  

 

posted on 2017-08-14 14:02  沉疴  阅读(7710)  评论(2编辑  收藏  举报