Ps回调函数.拦截驱动模块原理+实现.

#include <ntimage.h>
#include <ntddk.h>

#include <wdm.h>

ULONG_PTR g_LoadPtr = 0;


void WPONx64(KIRQL irql)
{
    UINT64 cr0 = __readcr0();
    cr0 |= 0x10000;
    _enable();
    __writecr0(cr0);
    KeLowerIrql(irql);
}



KIRQL WPOFFx64()
{
    KIRQL irql = KeRaiseIrqlToDpcLevel();
    UINT64 cr0 = __readcr0();
    cr0 &= 0xfffffffffffeffff;
    __writecr0(cr0);
    _disable();
    return irql;

}


BOOLEAN Fuck(PVOID pOep)
{
    KIRQL kirql;

    if (NULL == pOep)
        return FALSE;

    //写入字节.
    
    UCHAR FuckCode[] = { "\xB8\x22\x00\x00\xC0\xC3" };
    kirql = WPOFFx64();
    KdBreakPoint();
    memcpy(pOep, FuckCode, sizeof(FuckCode) / sizeof(FuckCode[0])); //开头写入拒绝访问错误码让其无法加载即可.
    WPONx64(kirql);
    return TRUE;
}
//根据ImageBase来获取加载模块的入口点
PVOID GetImageOep(PVOID ImageBase)
{

    PVOID pAddressOfEntryPoint = NULL;
    PIMAGE_DOS_HEADER pDosHead = NULL;
    PIMAGE_NT_HEADERS pNtHead = NULL;
    PIMAGE_FILE_HEADER pFileHead = NULL;
    PIMAGE_OPTIONAL_HEADER pOptHead = NULL;

    //开始解析
    if (ImageBase == NULL)
        return NULL;

    pDosHead = (PIMAGE_DOS_HEADER)ImageBase;
    pNtHead = (PIMAGE_NT_HEADERS)(pDosHead->e_lfanew + (char *)ImageBase);
    pFileHead = (PIMAGE_FILE_HEADER)&pNtHead->FileHeader;
    pOptHead = (PIMAGE_OPTIONAL_HEADER)&pNtHead->OptionalHeader;

    //判断是否是PE头
    if (pDosHead->e_magic != 0x5A4D && pNtHead->Signature != 0x5045)
        return NULL;
    pAddressOfEntryPoint = pOptHead->AddressOfEntryPoint + (char *)ImageBase;

    return pAddressOfEntryPoint;
}

char * MyWideUnicodeStringToMutilString(PUNICODE_STRING uString)
{
    ANSI_STRING asStr;
    char *Buffer = NULL; ;
    RtlUnicodeStringToAnsiString(&asStr, uString, TRUE);
    Buffer = ExAllocatePoolWithTag(NonPagedPool, uString->MaximumLength * sizeof(wchar_t), 0);
    if (Buffer == NULL)
        return NULL;
    RtlCopyMemory(Buffer, asStr.Buffer, asStr.Length);
    return Buffer;
}
VOID pSfFilterModule(
    _In_opt_ PUNICODE_STRING FullImageName,
    _In_ HANDLE ProcessId,                // pid into which image is being mapped
    _In_ PIMAGE_INFO ImageInfo
)
{
    /*
    模块拦截思路:
    1.通过 PIMAGE_INFO->ImageBase 得到模块的ImageBase
    2.解析PE头.
    3.可选头的OEP + ImageBase = 程序运行位置
    4.在程序开头写入 ret.
    5.通过参数一,判断是不是我们想要拦截的模块.然后进行1 2 3 4步
    6.如何判断加载驱动模块还是DLL 根据参数2.ProcessId高低位判断即可. == 0加载驱动模块.否则加载DLL
    */

    
    
    PVOID pDriverOep = NULL;
    char *ComPareString = NULL;
    if (MmIsAddressValid(FullImageName))
    {
        
        if (ProcessId == 0)
        {
            //代表拦截驱动模块

            

            //判断名字是否是你想要拦截的.
            KdBreakPoint();
            
            ComPareString = MyWideUnicodeStringToMutilString(FullImageName);
            if (ComPareString == NULL)
                return;

            if (strstr(ComPareString,"1.sys"))
            {
                KdBreakPoint();
                KdPrint(("你要拦截的驱动模块名字为: %wZ \r\n", FullImageName));
                pDriverOep = GetImageOep(ImageInfo->ImageBase);

                //判断拦截的名字是否是你想要拦截的
                if (pDriverOep != NULL)
                {
                    Fuck(pDriverOep);
                }
            }
            
        }
        //输出调试
        
    }
    
}

NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObj, PUNICODE_STRING pRegPath)
{
    ULONG iCount = 0;
    NTSTATUS ntStatus;
    pDriverObj->DriverUnload = DriverUnLoad;
    ntStatus = InitDeviceAnSybolicLinkName(pDriverObj);
    if (!NT_SUCCESS(ntStatus))
    {
        return ntStatus;
    }

    ntStatus = InitDisPatchFunction(pDriverObj);
    if (!NT_SUCCESS(ntStatus))
    {
        return ntStatus;
    }


    PsSetLoadImageNotifyRoutine(pSfFilterModule);
    return STATUS_SUCCESS;
}