mirror of
https://github.com/Mr-X-GTA/YimMenu.git
synced 2025-01-06 01:23:35 +08:00
97a8c5d60b
* feat(Spoofing): add spoofing * feat(Spoofing): prepare code for player attach * remove(PlayerAttach): isn't going to work due to netsync architecture * fix(GUI): fix scaling * feat(Project): add clang-format file * feat(Classes): update classes * fix(BlackHole): remove unnecessary cleanup * fix(Formatting): fix formatting for initializer lists * feat(clang-format): Set tab width and 1 space before comment Co-authored-by: Yimura <24669514+Yimura@users.noreply.github.com>
80 lines
1.9 KiB
C++
80 lines
1.9 KiB
C++
#include "detour_hook.hpp"
|
|
|
|
#include "common.hpp"
|
|
#include "memory/handle.hpp"
|
|
|
|
#include <MinHook.h>
|
|
|
|
namespace big
|
|
{
|
|
detour_hook::detour_hook(std::string name, void* detour) :
|
|
m_name(std::move(name)),
|
|
m_detour(detour)
|
|
{
|
|
}
|
|
|
|
detour_hook::detour_hook(std::string name, void* target, void* detour) :
|
|
m_name(std::move(name)),
|
|
m_target(target),
|
|
m_detour(detour)
|
|
{
|
|
create_hook();
|
|
}
|
|
|
|
void detour_hook::set_target_and_create_hook(void* target)
|
|
{
|
|
m_target = target;
|
|
create_hook();
|
|
}
|
|
|
|
void detour_hook::create_hook()
|
|
{
|
|
if (!m_target)
|
|
return;
|
|
|
|
fix_hook_address();
|
|
if (auto status = MH_CreateHook(m_target, m_detour, &m_original); status != MH_OK)
|
|
throw std::runtime_error(std::format("Failed to create hook '{}' at 0x{:X} (error: {})", m_name, uintptr_t(m_target), MH_StatusToString(status)));
|
|
}
|
|
|
|
detour_hook::~detour_hook() noexcept
|
|
{
|
|
if (!m_target)
|
|
return;
|
|
|
|
if (auto status = MH_RemoveHook(m_target); status != MH_OK)
|
|
LOG(FATAL) << "Failed to remove hook '" << m_name << "' at 0x" << HEX_TO_UPPER(uintptr_t(m_target)) << "(error: " << m_name << ")";
|
|
}
|
|
|
|
void detour_hook::enable()
|
|
{
|
|
if (!m_target)
|
|
return;
|
|
|
|
if (auto status = MH_QueueEnableHook(m_target); status != MH_OK)
|
|
throw std::runtime_error(std::format("Failed to enable hook 0x{:X} ({})", uintptr_t(m_target), MH_StatusToString(status)));
|
|
}
|
|
|
|
void detour_hook::disable()
|
|
{
|
|
if (!m_target)
|
|
return;
|
|
|
|
if (auto status = MH_QueueDisableHook(m_target); status != MH_OK)
|
|
LOG(WARNING) << "Failed to disable hook '" << m_name << "'.";
|
|
}
|
|
|
|
DWORD exp_handler(PEXCEPTION_POINTERS exp, std::string const& name)
|
|
{
|
|
return exp->ExceptionRecord->ExceptionCode == STATUS_ACCESS_VIOLATION ? EXCEPTION_EXECUTE_HANDLER : EXCEPTION_CONTINUE_SEARCH;
|
|
}
|
|
|
|
void detour_hook::fix_hook_address()
|
|
{
|
|
auto ptr = memory::handle(m_target);
|
|
while (ptr.as<std::uint8_t&>() == 0xE9)
|
|
ptr = ptr.add(1).rip();
|
|
m_target = ptr.as<void*>();
|
|
}
|
|
}
|