forked from cseagle/blc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
inject_sleigh.hh
112 lines (102 loc) · 4.2 KB
/
inject_sleigh.hh
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
106
107
108
109
110
111
112
/* ###
* IP: GHIDRA
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef __INJECT_SLEIGH__
#define __INJECT_SLEIGH__
#include "pcodeinject.hh"
#include "sleigh.hh"
class InjectContextSleigh : public InjectContext {
public:
PcodeCacher cacher;
ParserContext *pos;
InjectContextSleigh(void) { pos = (ParserContext *)0; }
virtual ~InjectContextSleigh(void);
virtual void encode(Encoder &encoder) const {} // We don't need this functionality for sleigh
};
class InjectPayloadSleigh : public InjectPayload {
friend class PcodeInjectLibrarySleigh;
ConstructTpl *tpl;
string parsestring;
string source;
protected:
void decodeBody(Decoder &decoder); ///< Parse the <body> tag
public:
InjectPayloadSleigh(const string &src,const string &nm,int4 tp);
virtual ~InjectPayloadSleigh(void);
virtual void inject(InjectContext &context,PcodeEmit &emit) const;
virtual void decode(Decoder &decoder);
virtual void printTemplate(ostream &s) const;
virtual string getSource(void) const { return source; }
static void checkParameterRestrictions(InjectContextSleigh &con,const vector<InjectParameter> &inputlist,
const vector<InjectParameter> &output,const string &source);
static void setupParameters(InjectContextSleigh &con,ParserWalkerChange &walker,
const vector<InjectParameter> &inputlist,const vector<InjectParameter> &output,
const string &source);
};
class InjectPayloadCallfixup : public InjectPayloadSleigh {
vector<string> targetSymbolNames;
public:
InjectPayloadCallfixup(const string &sourceName);
virtual void decode(Decoder &decoder);
};
class InjectPayloadCallother : public InjectPayloadSleigh {
public:
InjectPayloadCallother(const string &sourceName);
virtual void decode(Decoder &decoder);
};
class ExecutablePcodeSleigh : public ExecutablePcode {
friend class PcodeInjectLibrarySleigh;
protected:
string parsestring;
ConstructTpl *tpl;
public:
ExecutablePcodeSleigh(Architecture *g,const string &src,const string &nm);
virtual ~ExecutablePcodeSleigh(void);
virtual void inject(InjectContext &context,PcodeEmit &emit) const;
virtual void decode(Decoder &decoder);
virtual void printTemplate(ostream &s) const;
};
class InjectPayloadDynamic : public InjectPayload {
Architecture *glb;
map<Address,Document *> addrMap; // Map from address to specific inject
public:
InjectPayloadDynamic(Architecture *g,const string &nm,int4 tp) : InjectPayload(nm,tp) { glb = g; dynamic = true; }
virtual ~InjectPayloadDynamic(void);
void decodeEntry(Decoder &decoder);
virtual void inject(InjectContext &context,PcodeEmit &emit) const;
virtual void decode(Decoder &decoder) { throw LowlevelError("decode not supported for InjectPayloadDynamic"); }
virtual void printTemplate(ostream &s) const { s << "dynamic"; }
virtual string getSource(void) const { return "dynamic"; }
};
class PcodeInjectLibrarySleigh : public PcodeInjectLibrary {
const SleighBase *slgh;
vector<OpBehavior *> inst;
InjectContextSleigh contextCache;
int4 registerDynamicInject(InjectPayload *payload);
InjectPayloadDynamic *forceDebugDynamic(int4 injectid);
void parseInject(InjectPayload *payload);
protected:
virtual int4 allocateInject(const string &sourceName,const string &name,int4 type);
virtual void registerInject(int4 injectid);
public:
PcodeInjectLibrarySleigh(Architecture *g);
virtual void decodeDebug(Decoder &decoder);
virtual int4 manualCallFixup(const string &name,const string &snippetstring);
virtual int4 manualCallOtherFixup(const string &name,const string &outname,const vector<string> &inname,
const string &snippet);
virtual InjectContext &getCachedContext(void) { return contextCache; }
virtual const vector<OpBehavior *> &getBehaviors(void);
};
#endif