38 lines
749 B
C++
38 lines
749 B
C++
#pragma once
|
|
#include "common.hpp"
|
|
|
|
namespace big
|
|
{
|
|
class vmt_hook
|
|
{
|
|
public:
|
|
explicit vmt_hook(void* obj, std::size_t num_funcs);
|
|
|
|
vmt_hook(vmt_hook&& that) = delete;
|
|
vmt_hook& operator=(vmt_hook&& that) = delete;
|
|
vmt_hook(vmt_hook const&) = delete;
|
|
vmt_hook& operator=(vmt_hook const&) = delete;
|
|
|
|
void hook(std::size_t index, void* func);
|
|
void unhook(std::size_t index);
|
|
|
|
template <typename T>
|
|
T get_original(std::size_t index);
|
|
|
|
void enable();
|
|
void disable();
|
|
private:
|
|
void*** m_object;
|
|
std::size_t m_num_funcs;
|
|
|
|
void** m_original_table;
|
|
std::unique_ptr<void*[]> m_new_table;
|
|
};
|
|
|
|
template<typename T>
|
|
inline T vmt_hook::get_original(std::size_t index)
|
|
{
|
|
return static_cast<T>(m_original_table[index]);
|
|
}
|
|
}
|