tutorial-secdev-2021/lookup.h at main · salmanyam/tutorial-secdev-2021 · GitHub

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
#ifndef JITROP_LOOKUP_H
#define JITROP_LOOKUP_H

#include <vector>
#include <string>
#include <set>

#define TC_GADGETS 11
#define MOVTC_GADGETS 8
#define PRIORITY_GADGETS 13
#define PAYLOAD_ONE_GADGETS 11-1

#define ALL_GADGETS 29

typedef enum {
	JITROP_TIME_TC = 1,
	JITROP_TIME_PRIORITY,
	JITROP_TIME_MOVTC,
	JITROP_TIME_MOVTC_COUNT,
	JITROP_TIME_PAYLOAD1
} jitrop_timing;


int get_gadget_size(int which);

void get_min_tc_set(std::vector<std::string> gadgets, int *tc_set);
void get_priority_set(std::vector<std::string> gadgets, int *tc_set);
void get_mov_tc_set(std::vector<std::string> gadgets, int *tc_set, std::set<std::string> *regset);
void get_mov_tc_count(std::vector<std::string> gadgets, int *tc_set, std::set<std::string> *gadget_set, std::set<std::string> *regset);
void get_payload_set_one(std::vector<std::string> gadgets, int *tc_set);

void lookup_gadgets(std::vector<std::string> gadgets);
void lookup_priority_gadgets(std::vector<std::string> gadgets);
void lookup_movtc_gadgets(std::vector<std::string> gadgets, std::set<std::string> *gadget_set, std::set<std::string> *regset);

#endif //JITROP_LOOKUP_H