mirror of
https://github.com/geode-sdk/geode.git
synced 2024-12-14 01:51:14 -05:00
81 lines
2.1 KiB
C++
81 lines
2.1 KiB
C++
#include "Windows.hpp"
|
|
|
|
#include "Core.hpp"
|
|
|
|
#ifdef GEODE_IS_WINDOWS
|
|
|
|
#include <Windows.h>
|
|
|
|
using namespace geode::core::hook;
|
|
using namespace geode::core::impl;
|
|
|
|
namespace {
|
|
LONG WINAPI signalHandler(EXCEPTION_POINTERS* info) {
|
|
#if defined(_WIN64)
|
|
auto current = reinterpret_cast<void*>(info->ContextRecord->Rip);
|
|
#elif defined(_WIN32)
|
|
auto current = reinterpret_cast<void*>(info->ContextRecord->Eip);
|
|
#endif
|
|
|
|
// handleContext(reinterpret_cast<void*>(info), current);
|
|
|
|
return EXCEPTION_CONTINUE_EXECUTION;
|
|
}
|
|
}
|
|
|
|
void* Windows::allocateVM(size_t size) {
|
|
return VirtualAlloc(nullptr, size, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
|
|
}
|
|
|
|
std::vector<std::byte> Windows::jump(void* from, void* to) {
|
|
std::vector<std::byte> ret;
|
|
ret.resize(5, std::byte(0u));
|
|
ret[0] = std::byte(0xe9);
|
|
|
|
// target - src - 5
|
|
|
|
intptr_t offset = reinterpret_cast<intptr_t>(to) - reinterpret_cast<intptr_t>(from) - 5;
|
|
std::byte data[4];
|
|
std::memcpy(data, &offset, 4);
|
|
ret[1] = data[0];
|
|
ret[2] = data[1];
|
|
ret[3] = data[2];
|
|
ret[4] = data[3];
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool Windows::enableSingleStep(void* vcontext) {
|
|
auto info = reinterpret_cast<EXCEPTION_POINTERS*>(vcontext);
|
|
#if defined(_WIN64)
|
|
info->ContextRecord->RFlags |= ((QWORD)0x100);
|
|
#elif defined(_WIN32)
|
|
info->ContextRecord->EFlags |= ((DWORD)0x100);
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
bool Windows::disableSingleStep(void* vcontext) {
|
|
auto info = reinterpret_cast<EXCEPTION_POINTERS*>(vcontext);
|
|
#if defined(_WIN64)
|
|
info->ContextRecord->RFlags &= ~((QWORD)0x100);
|
|
#elif defined(_WIN32)
|
|
info->ContextRecord->EFlags &= ~((DWORD)0x100);
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
bool Windows::writeMemory(void* to, void* from, size_t size) {
|
|
DWORD old;
|
|
VirtualProtect(to, size, PAGE_EXECUTE_READWRITE, &old);
|
|
auto res = WriteProcessMemory(GetCurrentProcess(), to, from, size, nullptr);
|
|
VirtualProtect(to, size, old, &old);
|
|
return res;
|
|
}
|
|
|
|
bool Windows::initialize() {
|
|
// return AddVectoredExceptionHandler(true, signalHandler);
|
|
return true;
|
|
}
|
|
|
|
#endif
|