blob: b25b2afdac1488a2361c134980ed172b1ac83cac (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#pragma once
#include "../Include.hpp"
class VMTHook
{
public:
explicit VMTHook(void* ptr)
{
if (!ptr)
return;
pClass = reinterpret_cast<uintptr_t**>(ptr);
pOriginalVMT = *pClass;
const auto Size = Length();
pHookedVMT = std::make_unique<uint64_t[]>(Size + 1);
memcpy(&pHookedVMT.get()[1], pOriginalVMT, Size * sizeof uint64_t);
pHookedVMT.get()[0] = static_cast<uintptr_t>(pOriginalVMT[-1]);
PatchPtr(pHookedVMT.get());
}
~VMTHook()
{
PatchPtr(pOriginalVMT);
}
template<typename Function, typename Original>
Function HookFunction(uint64_t Index, Original _Function)
{
auto Old = pOriginalVMT[Index];
pHookedVMT.get()[Index + 1] = reinterpret_cast<uintptr_t>(_Function);
return reinterpret_cast<Function>(Old);
}
private:
uint64_t Length()
{
uint64_t Index = 0;
const auto VMT = pOriginalVMT;
for (Index = 0; VMT[Index]; Index++)
if (IS_INTRESOURCE(VMT[Index]))
break;
return Index;
}
void PatchPtr(uintptr_t* Replacement)
{
if (!pClass)
return;
DWORD Old;
VirtualProtect(pClass, sizeof uintptr_t, PAGE_READWRITE, &Old);
Replacement == pOriginalVMT ? *pClass = Replacement : *pClass = &Replacement[1];
VirtualProtect(pClass, sizeof uintptr_t, Old, &Old);
}
std::uintptr_t** pClass;
std::uintptr_t* pOriginalVMT;
std::unique_ptr<uint64_t[]> pHookedVMT;
};
|