#include "hook.h"

bool detour_32(BYTE *src, BYTE *dst, const uintptr_t len)
{
    if (len < 5)
        return false;

    DWORD curProtection;
    VirtualProtect(src, len, PAGE_EXECUTE_READWRITE, &curProtection);

    memset(src, 0x90, len);

    uintptr_t relativeAddress = dst - src - 5;
    *src = 0xE9;
    *(uintptr_t *)(src + 1) = relativeAddress;

    VirtualProtect(src, len, curProtection, &curProtection);
    return true;
}

BYTE *trampoline_32(BYTE *src, BYTE *dst, const uintptr_t len)
{
    if (len < 5)
        return 0;

    BYTE *gateway = (BYTE *)VirtualAlloc(0, len, MEM_COMMIT | MEM_RESERVE,
                                         PAGE_EXECUTE_READWRITE);

    memcpy_s(gateway, len, src, len);

    uintptr_t gatewayRelativeAddr = src - gateway - 5;
    *(gateway + len) = 0xE9;
    *(uintptr_t *)((uintptr_t)gateway + len + 1) = gatewayRelativeAddr;

    detour_32(src, dst, len);

    return gateway;
}