eToPid(HANDLE handle)
{
PROCESS_BASIC_INFORMATION ProcessBasicInfor;
// 初始化字符串,并获取动态地址
UNICODE_STRING UtrZwQueryInformationProcessName = RTL_CONSTANT_STRING(L"ZwQueryInformationProcess");
ZwQueryInformationProcess = (PfnZwQueryInformationProcess)MmGetSystemRoutineAddress(&UtrZwQueryInformationProcessName);
// 调用查询
ZwQueryInformationProcess(
handle,
ProcessBasicInformation,
(PVOID)&ProcessBasicInfor,
sizeof(ProcessBasicInfor),
NULL);
// 返回进程PID
return ProcessBasicInfor.UniqueProcessId;
}
VOID UnDriver(PDRIVER_OBJECT driver)
{
DbgPrint("[-] 驱动卸载 \n");
}
NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{
DbgPrint("Hello LyShark \n");
// 将PID转换为HANDLE
HANDLE ptr = PidToHandle(6932);
DbgPrint("[*] PID --> HANDLE = %p \n", ptr);
// 句柄转为PID
ULONG pid = HandleToPid(ptr);
DbgPrint("[*] HANDLE --> PID = %d \n", pid);
Driver->DriverUnload = UnDriver;
return STATUS_SUCCESS;
}
编译并运行如上这段代码片段,将把进程PID转为HANDLE句柄,再通过句柄将其转为PID,输出效果图如下所示;
进程PID转换为EProcess结构: 通过PsLookUpProcessByProcessId
函数,该函数传入一个PID
则可获取到该PID的EProcess
结构体,具体转换实现方法如下所示;
本段代码展示了如何使用Windows内核API函数PsLookupProcessByProcessId
将一个PID(Process ID)转换为对应的EProcess
结构体,EProcess是Windows内核中描述进程的数据结构之一。
代码段中定义了一个名为PidToObject
的函数,该函数的输入参数是一个PID
,输出参数是对应的EProcess
结构体。
在函数中,通过调用PsLookupProcessByProcessId
函数来获取对应PID的EProcess
结构体,如果获取成功,则调用ObDereferenceObject
函数来减少EProcess
对象的引用计数,并返回获取到的EProcess
指针;否则返回0。
在DriverEntry
函数中,调用了PidToObject
函数将PID 6932转换为对应的EProcess
结构体,并使用DbgPrint
函数输出了转换结果。最后设置了驱动程序卸载函数为UnDriver
,当驱动程序被卸载时,UnDriver
函数会被调用。
#include <ntifs.h>
#include <windef.h>
// 将Pid转换为Object or EProcess
PEPROCESS PidToObject(ULONG Pid)
{
PEPROCESS pEprocess;
NTSTATUS status = PsLookupProcessByProcessId((HANDLE)Pid, &pEprocess);
if (status == STATUS_SUCCESS)
{
ObDereferenceObject(pEprocess);
return pEprocess;
}
return 0;
}
VOID UnDriver(PDRIVER_OBJECT driver)
{
DbgPrint("[-] 驱动卸载 \n");
}
NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{
DbgPrint("Hello LyShark \n");
// 将PID转换为PEPROCESS
PEPROCESS ptr = PidToObject(6932);
DbgPrint("[*] PID --> PEPROCESS = %p \n", ptr);
Driver->DriverUnload = UnDriver;
return STATUS_SUCCESS;
}
编译并运行如上这段代码片段,将把进程PID转为EProcess结构,输出效果图如下所示;
进程HANDLE与EPROCESS互相转换: 将Handle
转换为EProcess
结构可使用内核函数ObReferenceObjectByHandle
实现,反过来EProcess
转换为Handle
句柄可使用ObOpenObjectByPointer
内核函数实现,具体转换实现方法如下所示;
首先,将Handle
转换为EProcess
结构体,可以使用ObReferenceObjectByHandle
内核函数。该函数接受一个Handle
参数,以及对应的对象类型(这里为EProcess),并返回对应对象的指针。此函数会对返回的对象增加引用计数,因此在使用完毕后,需要使用ObDereferenceObject
将引用计数减少。
其次,将EProcess
结构体转换为Handle
句柄,可以使用ObOpenObjectByPointer
内核函数。该函数接受一个指向对象的指针(这里为EProcess结构体的指针),以及所需的访问权限和对象类型,并返回对应的Handle
句柄。此函数会将返回的句柄添加到当前进程的句柄表中,因此在使用完毕后,需要使用CloseHandle
函数将句柄关闭,以避免资源泄漏。
综上所述,我们可以通过这两个内核函数实现Handle
和EProcess
之间的相互转换,转换代码如下所示;
#include <ntifs.h>
#include <windef.h>
// 传入PID传出HANDLE句柄
HANDLE PidToHandle(ULONG PID)
{
HANDLE hProcessHandle;
OBJECT_ATTRIBUTES obj;
CLIENT_ID clientid;
clientid.UniqueProc