gbe_fork/overlay_experimental/Base_Hook.cpp
2019-09-01 20:53:16 +02:00

73 lines
1.5 KiB
C++

#include "Base_Hook.h"
#include "Hook_Manager.h"
#include <algorithm>
#ifndef NO_OVERLAY
#ifdef STEAM_WIN32
#include "../detours/detours.h"
#define DETOUR_HOOKBEGIN DetourTransactionBegin
#define DETOUR_UPDATETHREAD() DetourUpdateThread(GetCurrentThread())
#define DETOUR_ENDHOOK DetourTransactionCommit
#define DETOUR_HOOK DetourAttach
#define DETOUR_UNHOOK DetourDetach
#elif defined(__LINUX__)
#include "linux/Linux_Detour.h"
#define DETOUR_HOOKBEGIN Linux_Detour::transaction_begin
#define DETOUR_UPDATETHREAD() Linux_Detour::update_thread(pthread_self())
#define DETOUR_ENDHOOK Linux_Detour::transaction_commit
#define DETOUR_HOOK Linux_Detour::hook_func
#define DETOUR_UNHOOK Linux_Detour::unhook_func
#endif
Base_Hook::Base_Hook():
_library(nullptr)
{}
Base_Hook::~Base_Hook()
{
UnhookAll();
}
const char* Base_Hook::get_lib_name() const
{
return "<no_name>";
}
void Base_Hook::BeginHook()
{
DETOUR_HOOKBEGIN();
DETOUR_UPDATETHREAD();
}
void Base_Hook::EndHook()
{
DETOUR_ENDHOOK();
}
void Base_Hook::HookFunc(std::pair<void**, void*> hook)
{
if( DETOUR_HOOK(hook.first, hook.second) == 0 )
_hooked_funcs.emplace_back(hook);
}
void Base_Hook::UnhookAll()
{
if (_hooked_funcs.size())
{
BeginHook();
std::for_each(_hooked_funcs.begin(), _hooked_funcs.end(), [](std::pair<void**, void*>& hook) {
DETOUR_UNHOOK(hook.first, hook.second);
});
EndHook();
_hooked_funcs.clear();
}
}
#endif//NO_OVERLAY