#include "../MemoryModule/stdafx.h"
#include <cstdio>

static PVOID ReadDllFile(LPCSTR FileName) {
    LPVOID buffer;
    size_t size;
    FILE* f;
    fopen_s(&f, FileName, "rb");
    if (!f)return 0;
    _fseeki64(f, 0, SEEK_END);
    if (!(size = _ftelli64(f))) {
        fclose(f);
        return 0;
    }
    _fseeki64(f, 0, SEEK_SET);
    fread(buffer = new char[size], 1, size, f);
    fclose(f);
    return buffer;
}

int test() {
    HMODULE hModule;
    NTSTATUS status;
    PVOID buffer = ReadDllFile("a.dll");
    if (!buffer) return 0;

    status = LdrLoadDllMemoryExW(
        &hModule,                               // ModuleHandle
        nullptr,                                // LdrEntry
        0,                                      // Flags
        buffer,                                 // Buffer
        0,                                      // Reserved
        nullptr,                               // DllBaseName
        nullptr         // DllFullName
    );
    if (NT_SUCCESS(status) && status != STATUS_IMAGE_MACHINE_TYPE_MISMATCH) {

        typedef int(__stdcall* func)();
        func test_user32 = (func)GetProcAddress(hModule, "test_user32");
        test_user32();

        //
        // After calling MessageBox, we can't free it.
        // 
        //LdrUnloadDllMemory(hModule);
    }

    return 0;
}

int main() {
    if (MmpGlobalDataPtr->WindowsVersion == WINDOWS_VERSION::win11) {
        auto head = &NtCurrentPeb()->Ldr->InLoadOrderModuleList;
        auto entry = head->Flink;
        while (entry != head) {
            PLDR_DATA_TABLE_ENTRY_WIN11 __entry = CONTAINING_RECORD(entry, LDR_DATA_TABLE_ENTRY_WIN11, InLoadOrderLinks);
            wprintf(L"%s\t0x%08X, 0x%08X, 0x%p, %d\n",
                __entry->BaseDllName.Buffer,
                __entry->CheckSum,
                RtlImageNtHeader(__entry->DllBase)->OptionalHeader.CheckSum,
                __entry->ActivePatchImageBase,
                __entry->HotPatchState
            );

            entry = entry->Flink;
        }
    }

    return 0;
}