7.3 Windows驱动开发：内核监视LoadImage映像回调

《内核注册并监控对象回调》ObRegisterCallbacks进程与线程拦截LySharkPsSetLoadImageNotifyRoutineLoadImage

PsSetLoadImageNotifyRoutine和PsRemoveLoadImageNotifyRoutine是Windows操作系统提供的两个内核API函数，用于注册和取消注册LoadImage映像的回调函数。

LoadImage映像回调函数是一种内核回调函数，它可以用于监视和拦截系统中的模块加载事件，例如进程启动时加载的DLL、驱动程序等。当有新的模块被加载时，操作系统会调用注册的LoadImage映像回调函数，并将加载模块的相关信息传递给回调函数。

PsSetLoadImageNotifyRoutine函数用于注册LoadImage映像的回调函数，而PsRemoveLoadImageNotifyRoutine函数则用于取消注册已经注册的回调函数。开发者可以在LoadImage映像回调函数中执行自定义的逻辑，例如记录日志、过滤敏感数据、或者阻止某些操作。

需要注意的是，LoadImage映像回调函数的注册和取消注册必须在内核模式下进行，并且需要开发者有一定的内核开发经验。同时，LoadImage映像回调函数也需要遵守一些约束条件，例如不能阻塞或挂起进程或线程的创建或访问，不能调用一些内核API函数等。

内核监视LoadImage映像回调在安全软件、系统监控和调试工具等领域有着广泛的应用。开发者可以利用这个机制来监视和拦截系统中的模块加载事件，以保护系统安全。

监视模块加载与卸载需要费别使用两个函数，这两个函数的参数传递都是自己的回调地址。

• PsSetLoadImageNotifyRoutine 设置回调
• PsRemoveLoadImageNotifyRoutine 移除回调

MyLySharkLoadImageNotifyRoutineFullImageNameModuleStyleModuleStyle=0ModuleStyle=1DLLImageInfo

VOID MyLySharkLoadImageNotifyRoutine(PUNICODE_STRING FullImageName, HANDLE ModuleStyle, PIMAGE_INFO ImageInfo)

PsSetLoadImageNotifyRoutineMyLySharkLoadImageNotifyRoutineModuleStyleGetDriverEntryByImageBase

#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] 驱动卸载完成...");

}
NTSTATUS DriverEntry(IN PDRIVER_OBJECT Driver, PUNICODE_STRING RegistryPath)

{

    DbgPrint("hello lyshark \n");
PsSetLoadImageNotifyRoutine((PLOAD_IMAGE_NOTIFY_ROUTINE)MyLySharkLoadImageNotifyRoutine);

    DbgPrint("[lyshark] 驱动加载完成...");

    Driver->DriverUnload = UnDriver;

    return STATUS_SUCCESS;

}

运行这个驱动程序，则会输出被加载的驱动详细参数。