45 lines
1.1 KiB
C
45 lines
1.1 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
class BranchTrampoline
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
BranchTrampoline();
|
||
|
|
~BranchTrampoline();
|
||
|
|
|
||
|
|
bool Create(size_t len, void * module = NULL);
|
||
|
|
void Destroy();
|
||
|
|
void SetBase(size_t len, void* base);
|
||
|
|
|
||
|
|
// allocate unsized
|
||
|
|
void * StartAlloc();
|
||
|
|
void EndAlloc(const void * end);
|
||
|
|
|
||
|
|
void * Allocate(size_t size = sizeof(void *));
|
||
|
|
|
||
|
|
size_t Remain() { return m_len - m_allocated; }
|
||
|
|
|
||
|
|
// takes 6 bytes of space at src, 8 bytes in trampoline
|
||
|
|
bool Write6Branch(uintptr_t src, uintptr_t dst);
|
||
|
|
bool Write6Call(uintptr_t src, uintptr_t dst);
|
||
|
|
|
||
|
|
// takes 5 bytes of space at src, 14 bytes in trampoline
|
||
|
|
bool Write5Branch(uintptr_t src, uintptr_t dst);
|
||
|
|
bool Write5Call(uintptr_t src, uintptr_t dst);
|
||
|
|
|
||
|
|
private:
|
||
|
|
// takes 6 bytes of space at src, 8 bytes in trampoline
|
||
|
|
bool Write6Branch_Internal(uintptr_t src, uintptr_t dst, UInt8 op);
|
||
|
|
|
||
|
|
// takes 5 bytes of space at src, 14 bytes in trampoline
|
||
|
|
bool Write5Branch_Internal(uintptr_t src, uintptr_t dst, UInt8 op);
|
||
|
|
|
||
|
|
void * m_base;
|
||
|
|
size_t m_len; // bytes
|
||
|
|
size_t m_allocated; // bytes
|
||
|
|
|
||
|
|
void * m_curAlloc; // currently active StartAlloc base
|
||
|
|
};
|
||
|
|
|
||
|
|
extern BranchTrampoline g_branchTrampoline;
|
||
|
|
extern BranchTrampoline g_localTrampoline;
|