diff options
Diffstat (limited to 'loader_wsock32_proxy/hookutils.cpp')
| -rw-r--r-- | loader_wsock32_proxy/hookutils.cpp | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/loader_wsock32_proxy/hookutils.cpp b/loader_wsock32_proxy/hookutils.cpp new file mode 100644 index 00000000..8603cb35 --- /dev/null +++ b/loader_wsock32_proxy/hookutils.cpp @@ -0,0 +1,71 @@ +#include "pch.h" +#include "../NorthstarDedicatedTest/hookutils.h" + +#define ERROR(...) { char err[2048]; sprintf_s(err, __VA_ARGS__); MessageBoxA(GetForegroundWindow(), err, "Northstar Wsock32 Proxy Error", 0); } + +TempReadWrite::TempReadWrite(void* ptr) +{ + m_ptr = ptr; + MEMORY_BASIC_INFORMATION mbi; + VirtualQuery(m_ptr, &mbi, sizeof(mbi)); + VirtualProtect(mbi.BaseAddress, mbi.RegionSize, PAGE_EXECUTE_READWRITE, &mbi.Protect); + m_origProtection = mbi.Protect; +} + +TempReadWrite::~TempReadWrite() +{ + MEMORY_BASIC_INFORMATION mbi; + VirtualQuery(m_ptr, &mbi, sizeof(mbi)); + VirtualProtect(mbi.BaseAddress, mbi.RegionSize, m_origProtection, &mbi.Protect); +} + + +void HookEnabler::CreateHook(LPVOID ppTarget, LPVOID ppDetour, LPVOID* ppOriginal, const char* targetName) +{ + // the macro for this uses ppTarget's name as targetName, and this typically starts with & + // targetname is used for debug stuff and debug output is nicer if we don't have this + if (*targetName == '&') + targetName++; + + if (MH_CreateHook(ppTarget, ppDetour, ppOriginal) == MH_OK) + { + HookTarget* target = new HookTarget; + target->targetAddress = ppTarget; + target->targetName = (char*)targetName; + + m_hookTargets.push_back(target); + } + else + { + if (targetName != nullptr) + { + ERROR("MH_CreateHook failed for function %s", targetName); + } + else + { + ERROR("MH_CreateHook failed for unknown function"); + } + } +} + +HookEnabler::~HookEnabler() +{ + for (auto& hook : m_hookTargets) + { + if (MH_EnableHook(hook->targetAddress) != MH_OK) + { + if (hook->targetName != nullptr) + { + ERROR("MH_EnableHook failed for function %s", hook->targetName); + } + else + { + ERROR("MH_EnableHook failed for unknown function"); + } + } + else + { + //ERROR("Enabling hook %s", hook->targetName); + } + } +}
\ No newline at end of file |