在筆者上一篇文章《驅動開發:內核註冊並監控對象回調》
介紹瞭如何運用ObRegisterCallbacks
註冊進程與線程
回調,並通過該回調實現了攔截
指定進行運行的效果,本章LyShark
將帶大家繼續探索一個新的回調註冊函數,PsSetLoadImageNotifyRoutine
常用於註冊LoadImage
映像監視,當有模塊被系統加載時則可以第一時間獲取到加載模塊信息,需要注意的是該回調函數內無法進行攔截,如需要攔截則需寫入返回指令這部分內容將在下一章進行講解,本章將主要實現對模塊的監視功能。
監視模塊加載與卸載需要費別使用兩個函數,這兩個函數的參數傳遞都是自己的回調地址。
- PsSetLoadImageNotifyRoutine 設置回調
- PsRemoveLoadImageNotifyRoutine 移除回調
此處MyLySharkLoadImageNotifyRoutine
回調地址必須有三個參數傳遞組成,其中FullImageName
代表完整路徑,ModuleStyle
代表模塊類型,一般來說ModuleStyle=0
表示加載SYS驅動,如果ModuleStyle=1
則表示加載的是DLL
,最後一個參數ImageInfo
則是映像的詳細參數結構體。
VOID MyLySharkLoadImageNotifyRoutine(PUNICODE_STRING FullImageName, HANDLE ModuleStyle, PIMAGE_INFO ImageInfo)
那麼如何實現監視映像加載呢,來看如下完整代碼片段,首先PsSetLoadImageNotifyRoutine
註冊回調,當有模塊被加載則自動執行MyLySharkLoadImageNotifyRoutine
回調函數,其內部首先判斷ModuleStyle
得出是什麼類型的模塊,然後再通過GetDriverEntryByImageBase
拿到當前進程詳細參數並打印輸出。
// 署名權
// right to sign one's name on a piece of work
// PowerBy: LyShark
// Email: [email protected]
#include <ntddk.h>
#include <ntimage.h>
// 未導出函數聲明
PUCHAR PsGetProcessImageFileName(PEPROCESS pEProcess);
// 獲取到鏡像裝載基地址
PVOID GetDriverEntryByImageBase(PVOID ImageBase)
{
PIMAGE_DOS_HEADER pDOSHeader;
PIMAGE_NT_HEADERS64 pNTHeader;
PVOID pEntryPoint;
pDOSHeader = (PIMAGE_DOS_HEADER)ImageBase;
pNTHeader = (PIMAGE_NT_HEADERS64)((ULONG64)ImageBase + pDOSHeader->e_lfanew);
pEntryPoint = (PVOID)((ULONG64)ImageBase + pNTHeader->OptionalHeader.AddressOfEntryPoint);
return pEntryPoint;
}
// 獲取當前進程名
UCHAR* GetCurrentProcessName()
{
PEPROCESS pEProcess = PsGetCurrentProcess();
if (NULL != pEProcess)
{
UCHAR *lpszProcessName = PsGetProcessImageFileName(pEProcess);
if (NULL != lpszProcessName)
{
return lpszProcessName;
}
}
return NULL;
}
// 設置自己的回調函數
VOID MyLySharkLoadImageNotifyRoutine(PUNICODE_STRING FullImageName, HANDLE ModuleStyle, PIMAGE_INFO ImageInfo)
{
PVOID pDrvEntry;
// MmIsAddress 驗證地址可用性
if (FullImageName != NULL && MmIsAddressValid(FullImageName))
{
// ModuleStyle爲零表示加載sys
if (ModuleStyle == 0)
{
// 得到裝載主進程名
UCHAR *load_name = GetCurrentProcessName();
pDrvEntry = GetDriverEntryByImageBase(ImageInfo->ImageBase);
DbgPrint("[LyShark SYS加載] 模塊名稱:%wZ --> 裝載基址:%p --> 鏡像長度: %d --> 裝載主進程: %s \n", FullImageName, pDrvEntry, ImageInfo->ImageSize, load_name);
}
// ModuleStyle非零表示加載DLL
else
{
// 得到裝載主進程名
UCHAR *load_name = GetCurrentProcessName();
pDrvEntry = GetDriverEntryByImageBase(ImageInfo->ImageBase);
DbgPrint("[LyShark DLL加載] 模塊名稱:%wZ --> 裝載基址:%p --> 鏡像長度: %d --> 裝載主進程: %s \n", FullImageName, pDrvEntry, ImageInfo->ImageSize, load_name);
}
}
}
VOID UnDriver(PDRIVER_OBJECT driver)
{
PsRemoveLoadImageNotifyRoutine((PLOAD_IMAGE_NOTIFY_ROUTINE)MyLySharkLoadImageNotifyRoutine);
DbgPrint("[LyShark.com] 驅動卸載完成...");
}
NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)
{
DbgPrint("hello lyshark.com \n");
PsSetLoadImageNotifyRoutine((PLOAD_IMAGE_NOTIFY_ROUTINE)MyLySharkLoadImageNotifyRoutine);
DbgPrint("[LyShark.com] 驅動加載完成...");
Driver->DriverUnload = UnDriver;
return STATUS_SUCCESS;
}
運行這個驅動程序,則會輸出被加載的驅動詳細參數。