You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
1.9 KiB
109 lines
1.9 KiB
#pragma once
|
|
#include <Windows.h>
|
|
#include <map>
|
|
#include <memory>
|
|
|
|
#define JMP_CODE_SIZE 14
|
|
#define OFFSET_TO_ADDRESS 0x2
|
|
|
|
namespace hook
|
|
{
|
|
class detour
|
|
{
|
|
public:
|
|
detour(void* addrToHook, void* jmpTo);
|
|
~detour();
|
|
void install();
|
|
void uninstall();
|
|
bool installed();
|
|
uintptr_t hook_address();
|
|
uintptr_t detour_address();
|
|
private:
|
|
bool hook_installed{ false };
|
|
uintptr_t hook_addr, detour_addr;
|
|
unsigned char jmp_code[JMP_CODE_SIZE] = {
|
|
0x48, 0xb8, //movabs rax, &jmpTo
|
|
0x0, //jmpTo address will be here in these 0's
|
|
0x0,
|
|
0x0,
|
|
0x0,
|
|
0x0,
|
|
0x0,
|
|
0x0,
|
|
0x0,
|
|
0xff, 0xe0, //jmp rax
|
|
0x90, 0x90 //nop, nop
|
|
};
|
|
char org_bytes[JMP_CODE_SIZE];
|
|
};
|
|
|
|
static std::map<uintptr_t, std::unique_ptr<detour>> hooks{};
|
|
|
|
__forceinline void write_to_readonly(void* addr, void* data, int size)
|
|
{
|
|
DWORD old_flags;
|
|
VirtualProtect((LPVOID)addr, size, PAGE_READWRITE, &old_flags);
|
|
memcpy((void*)addr, data, size);
|
|
VirtualProtect((LPVOID)addr, size, old_flags, &old_flags);
|
|
}
|
|
|
|
/*
|
|
Author: xerox
|
|
Date: 12/19/2019
|
|
|
|
Create Hook without needing to deal with objects
|
|
*/
|
|
__forceinline void install(void* addr_to_hook, void* jmp_to_addr) {
|
|
|
|
if (!addr_to_hook)
|
|
return;
|
|
|
|
hooks.insert({
|
|
(std::uintptr_t)addr_to_hook,
|
|
std::make_unique<detour>(
|
|
addr_to_hook,
|
|
jmp_to_addr
|
|
)}
|
|
);
|
|
}
|
|
|
|
/*
|
|
Author: xerox
|
|
Date: 12/19/2019
|
|
|
|
Enable hook given the address to hook
|
|
*/
|
|
__forceinline void enable(void* addr)
|
|
{
|
|
if (!addr)
|
|
return;
|
|
hooks.at((std::uintptr_t)addr)->install();
|
|
}
|
|
|
|
/*
|
|
Author: xerox
|
|
Date: 12/19/2019
|
|
|
|
Disable hook givent the address of the hook
|
|
*/
|
|
__forceinline void disable(void* addr)
|
|
{
|
|
if (!addr)
|
|
return;
|
|
hooks.at((std::uintptr_t)addr)->uninstall();
|
|
}
|
|
|
|
|
|
/*
|
|
Author: xerox
|
|
Date: 12/19/2019
|
|
|
|
Remove hook completely from vector
|
|
*/
|
|
__forceinline void remove(void* addr)
|
|
{
|
|
if (!addr)
|
|
return;
|
|
hooks.erase((std::uintptr_t)addr);
|
|
}
|
|
} |