blob: e0649ddcf7fc8908102165a5ce9b7ecac80bc609 (
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
|
#ifndef CODEGENERATOR_H
#define CODEGENERATOR_H
#include <string>
#include <vector>
#include "CodeInjector.h"
std::vector<unsigned char> x86_relJump(unsigned long dst,
unsigned long src);
class CodeGenerator
{
public:
CodeGenerator(const native_data& nd);
virtual ~CodeGenerator();
void clear()
{
codes.clear();
}
bool hasCode(int index);
CodeGenerator& addCode(const std::vector<unsigned char>& code);
CodeGenerator& setCode(int index, const std::vector<unsigned char>& code);
CodeGenerator& setCodeSized(int index, const std::vector<unsigned char>& code);
CodeGenerator& setRel32JMP(int index, unsigned long dst, unsigned long src, bool reversed = false);
std::vector<unsigned char>::size_type buildSize(int maxCodes = -1);
std::vector<unsigned char> build();
std::vector<unsigned char> buildAndClear();
std::string toString();
private:
const native_data& nd;
std::vector<std::vector<unsigned char>> codes;
unsigned long diffRel32JMP(bool reversed, int index = -1)
{
return (!reversed ? buildSize(index) - 0x5 : buildSize(index));
}
};
#endif // CODEGENERATOR_H
|