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
|
#pragma once
#include <cstdint>
#include <ostream>
namespace torch::unwind {
enum {
A_UNDEFINED = 0x0,
A_REG_PLUS_DATA = 0x1, // exp = REG[reg] + data0
A_LOAD_CFA_OFFSET = 0x2, // exp = *(cfa + data0)
A_REG_PLUS_DATA_DEREF = 0x3 // exp = *(REG[reg] + data0)
};
// register numbers in dwarf info
enum {
D_UNDEFINED = -1,
D_RBP = 6,
D_RSP = 7,
D_RIP = 16,
D_REG_SIZE = 17,
};
struct Action {
uint8_t kind = A_UNDEFINED;
int32_t reg = -1;
int64_t data = 0;
static Action undefined() {
return Action{A_UNDEFINED};
}
static Action regPlusData(int32_t reg, int64_t offset) {
return Action{A_REG_PLUS_DATA, reg, offset};
}
static Action regPlusDataDeref(int32_t reg, int64_t offset) {
return Action{A_REG_PLUS_DATA_DEREF, reg, offset};
}
static Action loadCfaOffset(int64_t offset) {
return Action{A_LOAD_CFA_OFFSET, D_UNDEFINED, offset};
}
friend std::ostream& operator<<(std::ostream& out, const Action& self) {
switch (self.kind) {
case A_UNDEFINED:
out << "u";
break;
case A_REG_PLUS_DATA:
out << "r" << (int)self.reg << " + " << self.data;
break;
case A_REG_PLUS_DATA_DEREF:
out << "*(r" << (int)self.reg << " + " << self.data << ")";
break;
case A_LOAD_CFA_OFFSET:
out << "*(cfa + " << self.data << ")";
break;
}
return out;
}
};
} // namespace torch::unwind
|
