1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
#include "Memory.h"
#include <psapi.h>
#include <tlhelp32.h>
#include <iostream>
#undef PROCESSENTRY32
#undef Process32Next
Memory::Memory(const std::string& processName) {
// First, get the handle of the process
PROCESSENTRY32 entry;
entry.dwSize = sizeof(entry);
HANDLE snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
while (Process32Next(snapshot, &entry)) {
if (processName == entry.szExeFile) {
_handle = OpenProcess(PROCESS_ALL_ACCESS, FALSE, entry.th32ProcessID);
break;
}
}
if (!_handle) {
std::cerr << "Couldn't find " << processName.c_str() << ", is it open?" << std::endl;
throw std::exception("Unable to find process!");
}
// Next, get the process base address
DWORD numModules;
std::vector<HMODULE> moduleList(1024);
EnumProcessModulesEx(_handle, &moduleList[0], static_cast<DWORD>(moduleList.size()), &numModules, 3);
std::string name(64, '\0');
for (DWORD i = 0; i < numModules / sizeof(HMODULE); i++) {
int length = GetModuleBaseNameA(_handle, moduleList[i], &name[0], static_cast<DWORD>(name.size()));
name.resize(length);
if (processName == name) {
_baseAddress = (uintptr_t)moduleList[i];
break;
}
}
if (_baseAddress == 0) {
throw std::exception("Couldn't find the base process address!");
}
// Unprotect regions of memory
DWORD oldProtect;
VirtualProtectEx(_handle, (LPVOID)_baseAddress, sizeof(DWORD), PAGE_READWRITE, &oldProtect);
}
Memory::~Memory() {
CloseHandle(_handle);
}
int Memory::GetCurrentFrame()
{
int SCRIPT_FRAMES;
if (GLOBALS == 0x5B28C0) {
SCRIPT_FRAMES = 0x5BE3B0;
} else if (GLOBALS == 0x62A080) {
SCRIPT_FRAMES = 0x63651C;
} else {
throw std::exception("Unknown value for Globals!");
}
return ReadData<int>({SCRIPT_FRAMES}, 1)[0];
}
void Memory::SigScan(std::function<void(int offset, const std::vector<byte>& data)> scanFunc)
{
for (int i=0; i<0x200000; i+=0x1000) {
std::vector<byte> data = ReadData<byte>({i}, 0x1100);
scanFunc(i, data);
}
}
void Memory::ThrowError() {
std::string message(256, '\0');
int length = FormatMessageA(4096, nullptr, GetLastError(), 1024, &message[0], static_cast<DWORD>(message.size()), nullptr);
message.resize(length);
throw std::exception(message.c_str());
}
void* Memory::ComputeOffset(std::vector<int> offsets)
{
// Leave off the last offset, since it will be either read/write, and may not be of type unitptr_t.
int final_offset = offsets.back();
offsets.pop_back();
uintptr_t cumulativeAddress = _baseAddress;
for (const int offset : offsets) {
cumulativeAddress += offset;
const auto search = _computedAddresses.find(cumulativeAddress);
if (search == std::end(_computedAddresses)) {
// If the address is not yet computed, then compute it.
uintptr_t computedAddress = 0;
if (!ReadProcessMemory(_handle, reinterpret_cast<LPVOID>(cumulativeAddress), &computedAddress, sizeof(uintptr_t), NULL)) {
ThrowError();
}
_computedAddresses[cumulativeAddress] = computedAddress;
}
cumulativeAddress = _computedAddresses[cumulativeAddress];
}
return reinterpret_cast<void*>(cumulativeAddress + final_offset);
}
|
