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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
|
//
//
//
#pragma once
#include <vector>
#include <memory>
#include <cstdint>
#include <cassert>
namespace HIR {
struct TypeRef;
struct Path;
}
class Allocation;
struct Value;
class AllocationPtr
{
friend class Allocation;
void* m_ptr;
public:
enum class Ty
{
Allocation,
Function, // m_ptr is a pointer to the function.
Unused1,
Unused2,
};
private:
AllocationPtr(Allocation* p):
m_ptr(p)
{
}
public:
AllocationPtr(): m_ptr(nullptr) {}
AllocationPtr(AllocationPtr&& x): m_ptr(x.m_ptr) {
x.m_ptr = nullptr;
}
AllocationPtr(const AllocationPtr& x);
~AllocationPtr();
static AllocationPtr new_fcn(::HIR::Path p);
AllocationPtr& operator=(const AllocationPtr& x) = delete;
AllocationPtr& operator=(AllocationPtr&& x) {
this->~AllocationPtr();
this->m_ptr = x.m_ptr;
x.m_ptr = nullptr;
return *this;
}
operator bool() const { return m_ptr != 0; }
bool is_alloc() {
return *this && get_ty() == Ty::Allocation;
}
Allocation& alloc() {
assert(*this);
assert(get_ty() == Ty::Allocation);
return *static_cast<Allocation*>(get_ptr());
}
const Allocation& alloc() const {
assert(*this);
assert(get_ty() == Ty::Allocation);
return *static_cast<Allocation*>(get_ptr());
}
const ::HIR::Path& fcn() const {
assert(*this);
assert(get_ty() == Ty::Function);
return *static_cast<const ::HIR::Path*>(get_ptr());
}
Ty get_ty() const {
return static_cast<Ty>( reinterpret_cast<uintptr_t>(m_ptr) & 3 );
}
friend ::std::ostream& operator<<(::std::ostream& os, const AllocationPtr& x);
private:
void* get_ptr() const {
return reinterpret_cast<void*>( reinterpret_cast<uintptr_t>(m_ptr) & ~3 );
}
};
struct Relocation
{
// Offset within parent allocation where this relocation is performed.
// TODO: Size?
size_t slot_ofs;
AllocationPtr backing_alloc;
};
class Allocation
{
friend class AllocationPtr;
size_t refcount;
// TODO: Read-only flag?
public:
static AllocationPtr new_alloc(size_t size);
const uint8_t* data_ptr() const { return reinterpret_cast<const uint8_t*>(this->data.data()); }
uint8_t* data_ptr() { return reinterpret_cast< uint8_t*>(this->data.data()); }
size_t size() const { return this->data.size() * 8; }
::std::vector<uint64_t> data;
::std::vector<uint8_t> mask;
::std::vector<Relocation> relocations;
AllocationPtr get_relocation(size_t ofs) const {
for(const auto& r : relocations) {
if(r.slot_ofs == ofs)
return r.backing_alloc;
}
return AllocationPtr();
}
void check_bytes_valid(size_t ofs, size_t size) const;
void mark_bytes_valid(size_t ofs, size_t size);
Value read_value(size_t ofs, size_t size) const;
void read_bytes(size_t ofs, void* dst, size_t count) const;
void write_value(size_t ofs, Value v);
void write_bytes(size_t ofs, const void* src, size_t count);
// TODO: Make this block common
void write_u8 (size_t ofs, uint8_t v) { write_bytes(ofs, &v, 1); }
void write_u16(size_t ofs, uint16_t v) { write_bytes(ofs, &v, 2); }
void write_u32(size_t ofs, uint32_t v) { write_bytes(ofs, &v, 4); }
void write_u64(size_t ofs, uint64_t v) { write_bytes(ofs, &v, 8); }
void write_i8 (size_t ofs, int8_t v) { write_u8 (ofs, static_cast<uint8_t >(v)); }
void write_i16(size_t ofs, int16_t v) { write_u16(ofs, static_cast<uint16_t>(v)); }
void write_i32(size_t ofs, int32_t v) { write_u32(ofs, static_cast<uint32_t>(v)); }
void write_i64(size_t ofs, int64_t v) { write_u64(ofs, static_cast<uint64_t>(v)); }
void write_f32(size_t ofs, float v) { write_bytes(ofs, &v, 4); }
void write_f64(size_t ofs, double v) { write_bytes(ofs, &v, 8); }
void write_usize(size_t ofs, uint64_t v);
void write_isize(size_t ofs, int64_t v) { write_usize(ofs, static_cast<uint64_t>(v)); }
uint8_t read_u8(size_t ofs) const { uint8_t rv; read_bytes(ofs, &rv, 1); return rv; }
uint16_t read_u16(size_t ofs) const { uint16_t rv; read_bytes(ofs, &rv, 2); return rv; }
uint32_t read_u32(size_t ofs) const { uint32_t rv; read_bytes(ofs, &rv, 4); return rv; }
uint64_t read_u64(size_t ofs) const { uint64_t rv; read_bytes(ofs, &rv, 8); return rv; }
int8_t read_i8(size_t ofs) const { return static_cast<int8_t>(read_u8(ofs)); }
int16_t read_i16(size_t ofs) const { return static_cast<int16_t>(read_u16(ofs)); }
int32_t read_i32(size_t ofs) const { return static_cast<int32_t>(read_u32(ofs)); }
int64_t read_i64(size_t ofs) const { return static_cast<int64_t>(read_u64(ofs)); }
float read_f32(size_t ofs) const { float rv; read_bytes(ofs, &rv, 4); return rv; }
double read_f64(size_t ofs) const { double rv; read_bytes(ofs, &rv, 8); return rv; }
uint64_t read_usize(size_t ofs) const;
int64_t read_isize(size_t ofs) const { return static_cast<int64_t>(read_usize(ofs)); }
};
extern ::std::ostream& operator<<(::std::ostream& os, const Allocation& x);
struct Value
{
// If NULL, data is direct
AllocationPtr allocation;
struct {
uint8_t data[2*sizeof(size_t)-3]; // 16-3 = 13, fits in 16 bits of mask
uint8_t mask[2];
uint8_t size;
} direct_data;
Value();
Value(::HIR::TypeRef ty);
static Value new_fnptr(const ::HIR::Path& fn_path);
void create_allocation();
size_t size() const { return allocation ? allocation.alloc().size() : direct_data.size; }
void check_bytes_valid(size_t ofs, size_t size) const;
void mark_bytes_valid(size_t ofs, size_t size);
Value read_value(size_t ofs, size_t size) const;
void read_bytes(size_t ofs, void* dst, size_t count) const;
void write_value(size_t ofs, Value v);
void write_bytes(size_t ofs, const void* src, size_t count);
// TODO: Make this block common
void write_u8 (size_t ofs, uint8_t v) { write_bytes(ofs, &v, 1); }
void write_u16(size_t ofs, uint16_t v) { write_bytes(ofs, &v, 2); }
void write_u32(size_t ofs, uint32_t v) { write_bytes(ofs, &v, 4); }
void write_u64(size_t ofs, uint64_t v) { write_bytes(ofs, &v, 8); }
void write_i8 (size_t ofs, int8_t v) { write_u8 (ofs, static_cast<uint8_t >(v)); }
void write_i16(size_t ofs, int16_t v) { write_u16(ofs, static_cast<uint16_t>(v)); }
void write_i32(size_t ofs, int32_t v) { write_u32(ofs, static_cast<uint32_t>(v)); }
void write_i64(size_t ofs, int64_t v) { write_u64(ofs, static_cast<uint64_t>(v)); }
void write_f32(size_t ofs, float v) { write_bytes(ofs, &v, 4); }
void write_f64(size_t ofs, double v) { write_bytes(ofs, &v, 8); }
void write_usize(size_t ofs, uint64_t v);
void write_isize(size_t ofs, int64_t v) { write_usize(ofs, static_cast<uint64_t>(v)); }
uint8_t read_u8(size_t ofs) const { uint8_t rv; read_bytes(ofs, &rv, 1); return rv; }
uint16_t read_u16(size_t ofs) const { uint16_t rv; read_bytes(ofs, &rv, 2); return rv; }
uint32_t read_u32(size_t ofs) const { uint32_t rv; read_bytes(ofs, &rv, 4); return rv; }
uint64_t read_u64(size_t ofs) const { uint64_t rv; read_bytes(ofs, &rv, 8); return rv; }
int8_t read_i8(size_t ofs) const { return static_cast<int8_t>(read_u8(ofs)); }
int16_t read_i16(size_t ofs) const { return static_cast<int16_t>(read_u16(ofs)); }
int32_t read_i32(size_t ofs) const { return static_cast<int32_t>(read_u32(ofs)); }
int64_t read_i64(size_t ofs) const { return static_cast<int64_t>(read_u64(ofs)); }
float read_f32(size_t ofs) const { float rv; read_bytes(ofs, &rv, 4); return rv; }
double read_f64(size_t ofs) const { double rv; read_bytes(ofs, &rv, 8); return rv; }
uint64_t read_usize(size_t ofs) const;
int64_t read_isize(size_t ofs) const { return static_cast<int64_t>(read_usize(ofs)); }
};
extern ::std::ostream& operator<<(::std::ostream& os, const Value& v);
// A read-only reference to a value (to write, you have to go through it)
struct ValueRef
{
// Either an AllocationPtr, or a Value pointer
AllocationPtr m_alloc;
Value* m_value;
size_t m_offset;
size_t m_size;
ValueRef(AllocationPtr ptr, size_t ofs, size_t size):
m_alloc(ptr),
m_offset(ofs),
m_size(size)
{
}
ValueRef(Value& val, size_t ofs, size_t size):
m_value(&val),
m_offset(ofs),
m_size(size)
{
}
Value read_value(size_t ofs, size_t size) const {
if( size == 0 )
return Value();
assert(ofs < m_size);
assert(size <= m_size);
assert(ofs+size <= m_size);
if( m_alloc ) {
return m_alloc.alloc().read_value(m_offset + ofs, size);
}
else {
return m_value->read_value(m_offset + ofs, size);
}
}
void read_bytes(size_t ofs, void* dst, size_t size) const {
if( size == 0 )
return ;
assert(ofs < m_size);
assert(size <= m_size);
assert(ofs+size <= m_size);
if( m_alloc ) {
m_alloc.alloc().read_bytes(m_offset + ofs, dst, size);
}
else {
m_value->read_bytes(m_offset + ofs, dst, size);
}
}
uint8_t read_u8(size_t ofs) const { uint8_t rv; read_bytes(ofs, &rv, 1); return rv; }
uint16_t read_u16(size_t ofs) const { uint16_t rv; read_bytes(ofs, &rv, 2); return rv; }
uint32_t read_u32(size_t ofs) const { uint32_t rv; read_bytes(ofs, &rv, 4); return rv; }
uint64_t read_u64(size_t ofs) const { uint64_t rv; read_bytes(ofs, &rv, 8); return rv; }
int8_t read_i8(size_t ofs) const { return static_cast<int8_t>(read_u8(ofs)); }
int16_t read_i16(size_t ofs) const { return static_cast<int16_t>(read_u16(ofs)); }
int32_t read_i32(size_t ofs) const { return static_cast<int32_t>(read_u32(ofs)); }
int64_t read_i64(size_t ofs) const { return static_cast<int64_t>(read_u64(ofs)); }
float read_f32(size_t ofs) const { float rv; read_bytes(ofs, &rv, 4); return rv; }
double read_f64(size_t ofs) const { double rv; read_bytes(ofs, &rv, 8); return rv; }
uint64_t read_usize(size_t ofs) const;
int64_t read_isize(size_t ofs) const { return static_cast<int64_t>(read_usize(ofs)); }
};
|
