Windows内核驱动程序：ZwAllocateVirtualMemory导致线程终止

Question

我正在尝试编写一个APC dll注入驱动程序，我找到了这个示例，并想根据自己的需要对其进行修改。

我使用PcreateProcessNotifyRoutineEx来获取我定位的特定应用程序的ProcessId，在这种情况下为“ iexplore.exe”，然后使用PloadImageNotifyRoutine检查是否已加载和初始化ntdll.dll（根据此建议 ），以及是否已加载ntdll.dll。 ，我称之为“我的” InjectDLL函数。

这是调用InjectDll的PloadImageNotifyRoutine函数：

VOID PloadImageNotifyRoutine(
    _In_ PUNICODE_STRING FullImageName,
    _In_ HANDLE ProcessId,
    _In_ PIMAGE_INFO ImageInfo
)
{
    PEPROCESS Process = NULL;
    PETHREAD Thread = NULL;
    PCHAR pTeb = nullptr;
    DWORD ArbitraryUserPointer = 0;
    PCHAR pszProcessNameA = nullptr;

    pTeb = (PCHAR)__readfsdword(0x18);
    ArbitraryUserPointer = *(DWORD*)(pTeb + 0x014);

    // If ArbitraryUserPointer points to kernel32.dll it means ntdll.dll is done loading.
    if (FALSE == IsStringEndWith((wchar_t*)ArbitraryUserPointer, L"\\kernel32.dll"))
    {
        return;
    }

    if (!NT_SUCCESS(PsLookupProcessByProcessId(ProcessId, &Process)))
    {
        return;
    }

    pszProcessNameA = (PCHAR)PsGetProcessImageFileName(Process);
    if (FALSE == StringNCompare(pszProcessNameA, "iexplore.exe", GetStringLength("iexplore.exe")))
    {
        return;
    }

    Thread = KeGetCurrentThread();
    InjectDll(MODULE_PATH, Process, Thread);
    ObDereferenceObject(Process);
}

这是InjectDll函数：

BOOLEAN InjectDll(PWCHAR pModulePath, PEPROCESS Process, PETHREAD Thread)
{
    PKINJECT mem;
    ULONG size;

    mem = NULL;
    size = 4096;

    if (!NT_SUCCESS(ZwAllocateVirtualMemory(NtCurrentProcess(), (PVOID*)&mem, 0, &size, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE)))
    {
        return FALSE;
    }

    //more code ...
}

我在这里删除了一些检查，这样会更清楚。

我尝试对其进行一些调试，但是ZwAllocateVirtualMemory似乎正在尝试分配内存，有时会失败并终止线程。

在调用ExAllocatePoolWithTag之后，它开始释放内存，取消映射节并终止线程（<=我在调试时在堆栈中看到的调用-并未真正跟踪每个调用，而是在一般视图中进行了查看）。

线程终止之前的堆栈：

nt!ExAllocatePoolWithTag+0x195
nt!NtAllocateVirtualMemory+0x1066
nt!KiSystemServicePostCall
nt!ZwAllocateVirtualMemory+0x11
kernel_apc_dll_injector!InjectDll+0x54
kernel_apc_dll_injector!PloadImageNotifyRoutine+0x2b0
nt!PsCallImageNotifyRoutines+0x62

从任务管理器中仍然可以看到该进程，但是该进程的内存为92k，没有CPU使用率，可能是因为未正确“清理”该进程。

我不知道我的分析是否正确，也许对于这个问题甚至是不必要的。

Answer 1

首先注意-不要从图像通知例程调用PsLookupProcessByProcessId 。 根本不需要。 检查ProcessId == PsGetCurrentProcessId() 。 如果是，请使用当前进程指针（如您所用，并在调用ZwAllocateVirtualMemory - NtCurrentProcess() ），否则将仅存在。

现在有关主-“ ZwAllocateVirtualMemory导致线程终止”-当然没有。 线程未终止 。 它挂了 。 首先，如果线程终止，则-由于此线程在此阶段处于进程中-所有进程均终止。 但是您自己说， 该过程仍然可以从任务管理器中看到 。 还如何查看终止线程的调用堆栈？ 这也表示线程没有终止，而是在ExAllocatePoolWithTag内部ExAllocatePoolWithTag

问题是在某些关键区域执行回调，并且您在此例程中可以执行的操作受到限制 （操作系统在禁用了常规内核APC的关键区域内的PASSIVE_LEVEL内调用驱动程序的进程通知例程）。 限制之一-称为ZwAllocateVirtualMemory它挂在回调中，您可以看到。

因此您不能调用ZwAllocateVirtualMemory并直接从回调中进行注入。 但存在解决方案。 将普通内核apc插入当前线程。 它不会就地执行，因为-回调中禁用了常规内核APC。 但在您退出回调并启用apc之后，您的apc已执行。 在这里（按常规），您已经可以调用ZwAllocateVirtualMemory并进行dll注入