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
|
#ifndef CODEINJECTOR_H
#define CODEINJECTOR_H
#include <vector>
#include <map>
#include <string>
extern "C" {
#include "native.h"
}
typedef struct code_bin
{
unsigned long addr;
unsigned long siz;
bool operator<(const code_bin& a) const
{
return addr < a.addr;
}
} code_bin;
typedef struct code_seg
{
unsigned long addr;
unsigned long siz;
std::map<const std::string, code_bin> children;
} code_seg;
class CodeInjector
{
public:
CodeInjector(const native_data& nd);
virtual ~CodeInjector();
bool allocCodeSegment(const std::string& name,
unsigned long siz = 4096);
bool addCode(const std::string& name, const std::string& code_name,
const std::vector<unsigned char>& code);
bool addCode(const std::string& name, const std::string& code_name,
const std::string& code);
bool addCode(const std::string& name, const std::string& code_name,
unsigned long siz);
bool setCode(const std::string& name, const std::string& code_name,
const std::vector<unsigned char>& code,
unsigned long offset = 0);
bool delCode(const std::string& name, const std::string& code_name);
unsigned long getCodeAddr(const std::string& name, const std::string& code_name);
bool getCodeSeg(const std::string& name, code_seg *seg);
bool getCodeBin(const std::string& name, const std::string& code_name, code_bin *bin);
std::string toString();
private:
const native_data& nd;
std::map<std::string, code_seg> code_map;
bool codeSegExists(const std::string& name)
{
return code_map.find(name) != code_map.end();
}
bool codeBinExists(const std::string& name, const std::string& code_name);
std::vector<code_bin> convertCodeSegChildren(const std::string& name);
unsigned long findCodeCave(const std::string& name, unsigned long siz);
};
#endif // CODEINJECTOR_H
|