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
105
|
#include "CodeGenerator.h"
#include <assert.h>
#include <sstream>
#include <iomanip>
#include "native.h"
#include "utils.h"
std::vector<unsigned char> x86_relJump(unsigned long dst,
unsigned long src)
{
std::vector<unsigned char> code(5);
code[0] = 0xE9;
unsigned long addr = dst - src;
code[1] = (*((unsigned char *)(&addr)+0));
code[2] = (*((unsigned char *)(&addr)+1));
code[3] = (*((unsigned char *)(&addr)+2));
code[4] = (*((unsigned char *)(&addr)+3));
return code;
}
CodeGenerator::CodeGenerator(const native_data& nd)
: nd(nd), codes()
{
}
CodeGenerator::~CodeGenerator()
{
}
CodeGenerator& CodeGenerator::addCode(const std::vector<unsigned char>& code)
{
codes.push_back(code);
return *this;
}
CodeGenerator& CodeGenerator::addCode(const std::string& code)
{
return *this;
}
CodeGenerator& CodeGenerator::setCode(int index, const std::vector<unsigned char>& code)
{
codes.at(index) = code;
return *this;
}
CodeGenerator& CodeGenerator::setCodeSized(int index, const std::vector<unsigned char>& code)
{
assert(codes.at(index).size() == code.size());
return setCode(index, code);
}
CodeGenerator& CodeGenerator::setRel32JMP(int index, unsigned long dst, unsigned long src, bool reversed)
{
if (!reversed)
{
dst += (unsigned long) nd.proc.modbase - diffRel32JMP(reversed, index);
}
else
{
src += (unsigned long) nd.proc.modbase + diffRel32JMP(reversed, index);
}
auto jmp = x86_relJump(dst, src);
setCodeSized(index, jmp);
return *this;
}
std::vector<unsigned char>::size_type CodeGenerator::buildSize(int maxCodes)
{
std::vector<unsigned char>::size_type total = 0;
for (auto& code : codes)
{
total += code.size();
if (maxCodes-- == 0)
break;
}
return total;
}
std::vector<unsigned char> CodeGenerator::build()
{
std::vector<unsigned char> result;
for (auto& code : codes)
{
result.insert(result.end(), code.begin(), code.end());
}
return result;
}
std::vector<unsigned char> CodeGenerator::buildAndClear()
{
auto result = build();
clear();
return result;
}
std::string CodeGenerator::toString()
{
std::stringstream out;
out << "CodeBin: " << utils::convertBinToHexstr(build()) << std::endl;
return out.str();
}
|