blob: d750bb55e53746d066336eaed34d1f78fa5e6c96 (
plain)
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
|
#pragma once
#include <iterator>
namespace inject {
// small wrapper for pe files
class c_pe_file {
// wew lad
bool is_valid_pe() {
IMAGE_DOS_HEADER *dos_header;
IMAGE_NT_HEADERS *nt_headers;
// dos header, "mz"
dos_header = reinterpret_cast<decltype(dos_header)>(m_file.data());
if(!dos_header || dos_header->e_magic != IMAGE_DOS_SIGNATURE)
return false;
// nt headers, "pe00"
nt_headers = reinterpret_cast<decltype(nt_headers)>(m_file.data() + dos_header->e_lfanew);
if(!nt_headers || nt_headers->Signature != IMAGE_NT_SIGNATURE)
return false;
return true;
}
public:
std::vector<uint8_t> m_file;
c_pe_file() = default;
bool get(const char *name) {
std::ifstream file_handle(name, std::ios::in | std::ios::binary);
// do not skip whitespace
file_handle.unsetf(std::ios::skipws);
if(!file_handle.is_open())
return false;
// read file contents
m_file.insert(
m_file.begin(),
std::istream_iterator<uint8_t>(file_handle),
std::istream_iterator<uint8_t>()
);
file_handle.close();
// we read the file, check if it's a pe
if(m_file.size() > 0)
return is_valid_pe();
// empty file or not pe
return false;
}
// homo wrappers
size_t size() {
return m_file.size();
}
uint8_t *data() {
return m_file.data();
}
};
// handles reloc, overwriting pe header with cheat header
class c_inject_transaction {
c_pe_file m_file;
public:
c_inject_transaction() = default;
bool get(c_pe_file &file) {
if(!file.size())
return false;
m_file = file;
return true;
}
void process_pe_header(std::vector<uint8_t> &cheat_header) {
std::memcpy(m_file.data(), cheat_header.data(), cheat_header.size());
}
bool process_reloc() {
return true;
}
};
}
|
