authorgravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2023-02-28 01:25:05-05:00
committergravatar for noreply@github.comGitHub <noreply@github.com> 2023-02-28 01:25:05-05:00
log98508a12cfae15a895d3e829539f907e9d88d0b6
tree64667cc40b95097f2d834fad1f0971ac8f9f9971
parentf6c934677315665c140151b8dd28a56f948205e2
parent6bed45b87358975da97fec45fd30eb095ad1dfc9
signaturebadge-question-mark Signed by PGP key 4AEE18F83AFDEB23

Merge pull request #14741 from jacobly0/lldb

13 files changed, 789 insertions(+), 191 deletions(-)

lib/std/hash_map.zig+5-4
...@@ -1595,16 +1595,17 @@ pub fn HashMapUnmanaged(...@@ -1595,16 +1595,17 @@ pub fn HashMapUnmanaged(
1595 self.available = 0;1595 self.available = 0;
1596 }1596 }
15971597
1598 /// This function is used in tools/zig-gdb.py to fetch the header type to facilitate1598 /// This function is used in the debugger pretty formatters in tools/ to fetch the
1599 /// fancy debug printing for this type.1599 /// header type to facilitate fancy debug printing for this type.
1600 fn gdbHelper(self: *Self, hdr: *Header) void {1600 fn dbHelper(self: *Self, hdr: *Header, entry: *Entry) void {
1601 _ = self;1601 _ = self;
1602 _ = hdr;1602 _ = hdr;
1603 _ = entry;
1603 }1604 }
16041605
1605 comptime {1606 comptime {
1606 if (builtin.mode == .Debug) {1607 if (builtin.mode == .Debug) {
1607 _ = gdbHelper;1608 _ = dbHelper;
1608 }1609 }
1609 }1610 }
1610 };1611 };
lib/std/multi_array_list.zig+23-6
...@@ -131,8 +131,8 @@ pub fn MultiArrayList(comptime S: type) type {...@@ -131,8 +131,8 @@ pub fn MultiArrayList(comptime S: type) type {
131 .capacity = self.capacity,131 .capacity = self.capacity,
132 };132 };
133 var ptr: [*]u8 = self.bytes;133 var ptr: [*]u8 = self.bytes;
134 for (sizes.bytes, 0..) |field_size, i| {134 for (sizes.bytes, sizes.fields) |field_size, i| {
135 result.ptrs[sizes.fields[i]] = ptr;135 result.ptrs[i] = ptr;
136 ptr += field_size * self.capacity;136 ptr += field_size * self.capacity;
137 }137 }
138 return result;138 return result;
...@@ -446,16 +446,33 @@ pub fn MultiArrayList(comptime S: type) type {...@@ -446,16 +446,33 @@ pub fn MultiArrayList(comptime S: type) type {
446 return meta.fieldInfo(S, field).type;446 return meta.fieldInfo(S, field).type;
447 }447 }
448448
449 /// This function is used in tools/zig-gdb.py to fetch the child type to facilitate449 const Entry = entry: {
450 /// fancy debug printing for this type.450 var entry_fields: [fields.len]std.builtin.Type.StructField = undefined;
451 fn gdbHelper(self: *Self, child: *S) void {451 for (&entry_fields, sizes.fields) |*entry_field, i| entry_field.* = .{
452 .name = fields[i].name ++ "_ptr",
453 .type = *fields[i].type,
454 .default_value = null,
455 .is_comptime = fields[i].is_comptime,
456 .alignment = fields[i].alignment,
457 };
458 break :entry @Type(.{ .Struct = .{
459 .layout = .Extern,
460 .fields = &entry_fields,
461 .decls = &.{},
462 .is_tuple = false,
463 } });
464 };
465 /// This function is used in the debugger pretty formatters in tools/ to fetch the
466 /// child type to facilitate fancy debug printing for this type.
467 fn dbHelper(self: *Self, child: *S, entry: *Entry) void {
452 _ = self;468 _ = self;
453 _ = child;469 _ = child;
470 _ = entry;
454 }471 }
455472
456 comptime {473 comptime {
457 if (builtin.mode == .Debug) {474 if (builtin.mode == .Debug) {
458 _ = gdbHelper;475 _ = dbHelper;
459 }476 }
460 }477 }
461 };478 };
src/arch/aarch64/CodeGen.zig+4-2
...@@ -4177,8 +4177,10 @@ fn airFieldParentPtr(self: *Self, inst: Air.Inst.Index) !void {...@@ -4177,8 +4177,10 @@ fn airFieldParentPtr(self: *Self, inst: Air.Inst.Index) !void {
4177}4177}
41784178
4179fn airArg(self: *Self, inst: Air.Inst.Index) !void {4179fn airArg(self: *Self, inst: Air.Inst.Index) !void {
4180 const arg_index = self.arg_index;4180 // skip zero-bit arguments as they don't have a corresponding arg instruction
4181 self.arg_index += 1;4181 var arg_index = self.arg_index;
4182 while (self.args[arg_index] == .none) arg_index += 1;
4183 self.arg_index = arg_index + 1;
41824184
4183 const ty = self.air.typeOfIndex(inst);4185 const ty = self.air.typeOfIndex(inst);
4184 const tag = self.air.instructions.items(.tag)[inst];4186 const tag = self.air.instructions.items(.tag)[inst];
src/arch/arm/CodeGen.zig+4-2
...@@ -4125,8 +4125,10 @@ fn genInlineMemsetCode(...@@ -4125,8 +4125,10 @@ fn genInlineMemsetCode(
4125}4125}
41264126
4127fn airArg(self: *Self, inst: Air.Inst.Index) !void {4127fn airArg(self: *Self, inst: Air.Inst.Index) !void {
4128 const arg_index = self.arg_index;4128 // skip zero-bit arguments as they don't have a corresponding arg instruction
4129 self.arg_index += 1;4129 var arg_index = self.arg_index;
4130 while (self.args[arg_index] == .none) arg_index += 1;
4131 self.arg_index = arg_index + 1;
41304132
4131 const ty = self.air.typeOfIndex(inst);4133 const ty = self.air.typeOfIndex(inst);
4132 const tag = self.air.instructions.items(.tag)[inst];4134 const tag = self.air.instructions.items(.tag)[inst];
src/arch/x86_64/CodeGen.zig+4-2
...@@ -3827,8 +3827,10 @@ fn genIntMulComplexOpMir(self: *Self, dst_ty: Type, dst_mcv: MCValue, src_mcv: M...@@ -3827,8 +3827,10 @@ fn genIntMulComplexOpMir(self: *Self, dst_ty: Type, dst_mcv: MCValue, src_mcv: M
3827}3827}
38283828
3829fn airArg(self: *Self, inst: Air.Inst.Index) !void {3829fn airArg(self: *Self, inst: Air.Inst.Index) !void {
3830 const arg_index = self.arg_index;3830 // skip zero-bit arguments as they don't have a corresponding arg instruction
3831 self.arg_index += 1;3831 var arg_index = self.arg_index;
3832 while (self.args[arg_index] == .none) arg_index += 1;
3833 self.arg_index = arg_index + 1;
38323834
3833 const ty = self.air.typeOfIndex(inst);3835 const ty = self.air.typeOfIndex(inst);
3834 const mcv = self.args[arg_index];3836 const mcv = self.args[arg_index];
src/codegen/llvm.zig+7-2
...@@ -2214,11 +2214,16 @@ pub const Object = struct {...@@ -2214,11 +2214,16 @@ pub const Object = struct {
2214 ));2214 ));
2215 }2215 }
22162216
2217 const union_name = if (layout.tag_size == 0) name.ptr else "AnonUnion";2217 var union_name_buf: ?[:0]const u8 = null;
2218 defer if (union_name_buf) |buf| gpa.free(buf);
2219 const union_name = if (layout.tag_size == 0) name else name: {
2220 union_name_buf = try std.fmt.allocPrintZ(gpa, "{s}:Payload", .{name});
2221 break :name union_name_buf.?;
2222 };
22182223
2219 const union_di_ty = dib.createUnionType(2224 const union_di_ty = dib.createUnionType(
2220 compile_unit_scope,2225 compile_unit_scope,
2221 union_name,2226 union_name.ptr,
2222 null, // file2227 null, // file
2223 0, // line2228 0, // line
2224 ty.abiSize(target) * 8, // size in bits2229 ty.abiSize(target) * 8, // size in bits
src/type.zig+30
...@@ -1,4 +1,5 @@...@@ -1,4 +1,5 @@
1const std = @import("std");1const std = @import("std");
2const builtin = @import("builtin");
2const Value = @import("value.zig").Value;3const Value = @import("value.zig").Value;
3const assert = std.debug.assert;4const assert = std.debug.assert;
4const Allocator = std.mem.Allocator;5const Allocator = std.mem.Allocator;
...@@ -6694,4 +6695,33 @@ pub const Type = extern union {...@@ -6694,4 +6695,33 @@ pub const Type = extern union {
6694 /// This is only used for comptime asserts. Bump this number when you make a change6695 /// This is only used for comptime asserts. Bump this number when you make a change
6695 /// to packed struct layout to find out all the places in the codebase you need to edit!6696 /// to packed struct layout to find out all the places in the codebase you need to edit!
6696 pub const packed_struct_layout_version = 2;6697 pub const packed_struct_layout_version = 2;
6698
6699 /// This function is used in the debugger pretty formatters in tools/ to fetch the
6700 /// Tag to Payload mapping to facilitate fancy debug printing for this type.
6701 fn dbHelper(self: *Type, tag_to_payload_map: *map: {
6702 const tags = @typeInfo(Tag).Enum.fields;
6703 var fields: [tags.len]std.builtin.Type.StructField = undefined;
6704 for (&fields, tags) |*field, t| field.* = .{
6705 .name = t.name,
6706 .type = *if (t.value < Tag.no_payload_count) void else @field(Tag, t.name).Type(),
6707 .default_value = null,
6708 .is_comptime = false,
6709 .alignment = 0,
6710 };
6711 break :map @Type(.{ .Struct = .{
6712 .layout = .Extern,
6713 .fields = &fields,
6714 .decls = &.{},
6715 .is_tuple = false,
6716 } });
6717 }) void {
6718 _ = self;
6719 _ = tag_to_payload_map;
6720 }
6721
6722 comptime {
6723 if (builtin.mode == .Debug) {
6724 _ = dbHelper;
6725 }
6726 }
6697};6727};
src/value.zig+30
...@@ -1,4 +1,5 @@...@@ -1,4 +1,5 @@
1const std = @import("std");1const std = @import("std");
2const builtin = @import("builtin");
2const Type = @import("type.zig").Type;3const Type = @import("type.zig").Type;
3const log2 = std.math.log2;4const log2 = std.math.log2;
4const assert = std.debug.assert;5const assert = std.debug.assert;
...@@ -5584,6 +5585,35 @@ pub const Value = extern union {...@@ -5584,6 +5585,35 @@ pub const Value = extern union {
5584 ri.* = @intToEnum(RuntimeIndex, @enumToInt(ri.*) + 1);5585 ri.* = @intToEnum(RuntimeIndex, @enumToInt(ri.*) + 1);
5585 }5586 }
5586 };5587 };
5588
5589 /// This function is used in the debugger pretty formatters in tools/ to fetch the
5590 /// Tag to Payload mapping to facilitate fancy debug printing for this type.
5591 fn dbHelper(self: *Value, tag_to_payload_map: *map: {
5592 const tags = @typeInfo(Tag).Enum.fields;
5593 var fields: [tags.len]std.builtin.Type.StructField = undefined;
5594 for (&fields, tags) |*field, t| field.* = .{
5595 .name = t.name,
5596 .type = *if (t.value < Tag.no_payload_count) void else @field(Tag, t.name).Type(),
5597 .default_value = null,
5598 .is_comptime = false,
5599 .alignment = 0,
5600 };
5601 break :map @Type(.{ .Struct = .{
5602 .layout = .Extern,
5603 .fields = &fields,
5604 .decls = &.{},
5605 .is_tuple = false,
5606 } });
5607 }) void {
5608 _ = self;
5609 _ = tag_to_payload_map;
5610 }
5611
5612 comptime {
5613 if (builtin.mode == .Debug) {
5614 _ = dbHelper;
5615 }
5616 }
5587};5617};
55885618
5589var negative_one_payload: Value.Payload.I64 = .{5619var negative_one_payload: Value.Payload.I64 = .{
tools/lldb_pretty_printers.py created+575
...@@ -0,0 +1,575 @@
1# pretty printing for the zig language, zig standard library, and zig stage 2 compiler.
2# put commands in ~/.lldbinit to run them automatically when starting lldb
3# `command script /path/to/stage2_lldb_pretty_printers.py` to import this file
4# `type category enable zig` to enable pretty printing for the zig language
5# `type category enable zig.std` to enable pretty printing for the zig standard library
6# `type category enable zig.stage2` to enable pretty printing for the zig stage 2 compiler
7import lldb
8import re
9
10page_size = 1 << 12
11
12def log2_int(i): return i.bit_length() - 1
13
14# Define Zig Language
15
16zig_keywords = {
17 'addrspace',
18 'align',
19 'allowzero',
20 'and',
21 'anyframe',
22 'anytype',
23 'asm',
24 'async',
25 'await',
26 'break',
27 'callconv',
28 'catch',
29 'comptime',
30 'const',
31 'continue',
32 'defer',
33 'else',
34 'enum',
35 'errdefer',
36 'error',
37 'export',
38 'extern',
39 'fn',
40 'for',
41 'if',
42 'inline',
43 'noalias',
44 'noinline',
45 'nosuspend',
46 'opaque',
47 'or',
48 'orelse',
49 'packed',
50 'pub',
51 'resume',
52 'return',
53 'linksection',
54 'struct',
55 'suspend',
56 'switch',
57 'test',
58 'threadlocal',
59 'try',
60 'union',
61 'unreachable',
62 'usingnamespace',
63 'var',
64 'volatile',
65 'while',
66}
67zig_primitives = {
68 'anyerror',
69 'anyframe',
70 'anyopaque',
71 'bool',
72 'c_int',
73 'c_long',
74 'c_longdouble',
75 'c_longlong',
76 'c_short',
77 'c_uint',
78 'c_ulong',
79 'c_ulonglong',
80 'c_ushort',
81 'comptime_float',
82 'comptime_int',
83 'f128',
84 'f16',
85 'f32',
86 'f64',
87 'f80',
88 'false',
89 'isize',
90 'noreturn',
91 'null',
92 'true',
93 'type',
94 'undefined',
95 'usize',
96 'void',
97}
98zig_integer_type = re.compile('[iu][1-9][0-9]+')
99zig_identifier_regex = re.compile('[A-Z_a-z][0-9A-Z_a-z]*')
100def zig_IsVariableName(string): return string != '_' and string not in zig_keywords and string not in zig_primitives and not zig_integer_type.fullmatch(string) and zig_identifier_regex.fullmatch(string)
101def zig_IsFieldName(string): return string not in zig_keywords and zig_identifier_regex.fullmatch(string)
102
103class zig_Slice_SynthProvider:
104 def __init__(self, value, _=None): self.value = value
105 def update(self):
106 try:
107 self.ptr = self.value.GetChildMemberWithName('ptr')
108 self.len = self.value.GetChildMemberWithName('len').unsigned if self.ptr.unsigned > page_size else 0
109 self.elem_type = self.ptr.type.GetPointeeType()
110 self.elem_size = self.elem_type.size
111 except: pass
112 def has_children(self): return True
113 def num_children(self): return self.len or 0
114 def get_child_index(self, name):
115 try: return int(name.removeprefix('[').removesuffix(']'))
116 except: return -1
117 def get_child_at_index(self, index):
118 if index < 0 or index >= self.len: return None
119 try: return self.ptr.CreateChildAtOffset('[%d]' % index, index * self.elem_size, self.elem_type)
120 except: return None
121
122def zig_String_decode(value, offset=0, length=None):
123 try:
124 value = value.GetNonSyntheticValue()
125 data = value.GetChildMemberWithName('ptr').GetPointeeData(offset, length if length is not None else value.GetChildMemberWithName('len').unsigned)
126 b = bytes(data.uint8)
127 b = b.replace(b'\\', b'\\\\')
128 b = b.replace(b'\n', b'\\n')
129 b = b.replace(b'\r', b'\\r')
130 b = b.replace(b'\t', b'\\t')
131 b = b.replace(b'"', b'\\"')
132 b = b.replace(b'\'', b'\\\'')
133 s = b.decode(encoding='ascii', errors='backslashreplace')
134 return s if s.isprintable() else ''.join((c if c.isprintable() else '\\x%02x' % ord(c) for c in s))
135 except: return None
136def zig_String_SummaryProvider(value, _=None): return '"%s"' % zig_String_decode(value)
137def zig_String_AsIdentifier(value, pred):
138 string = zig_String_decode(value)
139 return string if pred(string) else '@"%s"' % string
140
141class zig_Optional_SynthProvider:
142 def __init__(self, value, _=None): self.value = value
143 def update(self):
144 try:
145 self.child = self.value.GetChildMemberWithName('some').unsigned == 1 and self.value.GetChildMemberWithName('data').Clone('child')
146 except: pass
147 def has_children(self): return bool(self.child)
148 def num_children(self): return int(self.child)
149 def get_child_index(self, name): return 0 if self.child and (name == 'child' or name == '?') else -1
150 def get_child_at_index(self, index): return self.child if self.child and index == 0 else None
151def zig_Optional_SummaryProvider(value, _=None):
152 child = value.GetChildMemberWithName('child')
153 return child or 'null'
154
155class zig_ErrorUnion_SynthProvider:
156 def __init__(self, value, _=None): self.value = value
157 def update(self):
158 try:
159 self.error_set = self.value.GetChildMemberWithName('tag').Clone('error_set')
160 self.payload = self.value.GetChildMemberWithName('value').Clone('payload') if self.error_set.unsigned == 0 else None
161 except: pass
162 def has_children(self): return True
163 def num_children(self): return 1
164 def get_child_index(self, name): return 0 if name == ('payload' if self.payload else 'error_set') else -1
165 def get_child_at_index(self, index): return self.payload or self.error_set if index == 0 else None
166
167# Define Zig Standard Library
168
169class std_SegmentedList_SynthProvider:
170 def __init__(self, value, _=None): self.value = value
171 def update(self):
172 try:
173 self.prealloc_segment = self.value.GetChildMemberWithName('prealloc_segment')
174 self.dynamic_segments = zig_Slice_SynthProvider(self.value.GetChildMemberWithName('dynamic_segments'))
175 self.dynamic_segments.update()
176 self.len = self.value.GetChildMemberWithName('len').unsigned
177 except: pass
178 def has_children(self): return True
179 def num_children(self): return self.len
180 def get_child_index(self, name):
181 try: return int(name.removeprefix('[').removesuffix(']'))
182 except: return -1
183 def get_child_at_index(self, index):
184 try:
185 if index < 0 or index >= self.len: return None
186 prealloc_item_count = len(self.prealloc_segment)
187 if index < prealloc_item_count: return self.prealloc_segment.child[index]
188 prealloc_exp = prealloc_item_count.bit_length() - 1
189 shelf_index = log2_int(index + 1) if prealloc_item_count == 0 else log2_int(index + prealloc_item_count) - prealloc_exp - 1
190 shelf = self.dynamic_segments.get_child_at_index(shelf_index)
191 box_index = (index + 1) - (1 << shelf_index) if prealloc_item_count == 0 else index + prealloc_item_count - (1 << ((prealloc_exp + 1) + shelf_index))
192 elem_type = shelf.type.GetPointeeType()
193 return shelf.CreateChildAtOffset('[%d]' % index, box_index * elem_type.size, elem_type)
194 except: return None
195
196class std_MultiArrayList_SynthProvider:
197 def __init__(self, value, _=None): self.value = value
198 def update(self):
199 try:
200 self.len = 0
201
202 value_type = self.value.type
203 for helper in self.value.target.FindFunctions('%s.dbHelper' % value_type.name, lldb.eFunctionNameTypeFull):
204 ptr_self_type, ptr_child_type, ptr_entry_type = helper.function.type.GetFunctionArgumentTypes()
205 if ptr_self_type.GetPointeeType() == value_type: break
206 else: return
207
208 self.entry_type = ptr_entry_type.GetPointeeType()
209 self.bytes = self.value.GetChildMemberWithName('bytes')
210 self.len = self.value.GetChildMemberWithName('len').unsigned
211 self.capacity = self.value.GetChildMemberWithName('capacity').unsigned
212 except: pass
213 def has_children(self): return True
214 def num_children(self): return self.len
215 def get_child_index(self, name):
216 try: return int(name.removeprefix('[').removesuffix(']'))
217 except: return -1
218 def get_child_at_index(self, index):
219 try:
220 if index < 0 or index >= self.len: return None
221 offset = 0
222 data = lldb.SBData()
223 for field in self.entry_type.fields:
224 ptr_field_type = field.type
225 field_size = ptr_field_type.GetPointeeType().size
226 data.Append(self.bytes.CreateChildAtOffset(field.name, offset + index * field_size, ptr_field_type).address_of.data)
227 offset += self.capacity * field_size
228 return self.bytes.CreateValueFromData('[%d]' % index, data, self.entry_type)
229 except: return None
230
231class std_HashMapUnmanaged_SynthProvider:
232 def __init__(self, value, _=None): self.value = value
233 def update(self):
234 try:
235 self.capacity = 0
236 self.indices = tuple()
237
238 self.metadata = self.value.GetChildMemberWithName('metadata')
239 if not self.metadata.unsigned: return
240
241 value_type = self.value.type
242 for helper in self.value.target.FindFunctions('%s.dbHelper' % value_type.name, lldb.eFunctionNameTypeFull):
243 ptr_self_type, ptr_hdr_type, ptr_entry_type = helper.function.type.GetFunctionArgumentTypes()
244 if ptr_self_type.GetPointeeType() == value_type: break
245 else: return
246 self.entry_type = ptr_entry_type.GetPointeeType()
247
248 hdr_type = ptr_hdr_type.GetPointeeType()
249 hdr = self.metadata.CreateValueFromAddress('header', self.metadata.deref.load_addr - hdr_type.size, hdr_type)
250 self.values = hdr.GetChildMemberWithName('values')
251 self.keys = hdr.GetChildMemberWithName('keys')
252 self.capacity = hdr.GetChildMemberWithName('capacity').unsigned
253
254 self.indices = tuple(i for i, value in enumerate(self.metadata.GetPointeeData(0, self.capacity).sint8) if value < 0)
255 except: pass
256 def has_children(self): return True
257 def num_children(self): return len(self.indices)
258 def get_capacity(self): return self.capacity
259 def get_child_index(self, name):
260 try: return int(name.removeprefix('[').removesuffix(']'))
261 except: return -1
262 def get_child_at_index(self, index):
263 try:
264 fields = {name: base.CreateChildAtOffset(name, self.indices[index] * pointee_type.size, pointee_type).address_of.data for name, base, pointee_type in ((name, base, base.type.GetPointeeType()) for name, base in (('key_ptr', self.keys), ('value_ptr', self.values)))}
265 data = lldb.SBData()
266 for field in self.entry_type.fields: data.Append(fields[field.name])
267 return self.metadata.CreateValueFromData('[%d]' % index, data, self.entry_type)
268 except: return None
269def std_HashMapUnmanaged_SummaryProvider(value, _=None):
270 synth = std_HashMapUnmanaged_SynthProvider(value.GetNonSyntheticValue(), _)
271 synth.update()
272 return 'len=%d capacity=%d' % (synth.num_children(), synth.get_capacity())
273
274# formats a struct of fields of the form `name_ptr: *Type` by auto dereferencing its fields
275class std_Entry_SynthProvider:
276 def __init__(self, value, _=None): self.value = value
277 def update(self):
278 try:
279 self.children = tuple(child.Clone(child.name.removesuffix('_ptr')) for child in self.value.children if child.type.GetPointeeType().size != 0)
280 self.indices = {child.name: i for i, child in enumerate(self.children)}
281 except: pass
282 def has_children(self): return self.num_children() != 0
283 def num_children(self): return len(self.children)
284 def get_child_index(self, name): return self.indices.get(name)
285 def get_child_at_index(self, index): return self.children[index].deref if index >= 0 and index < len(self.children) else None
286
287# Define Zig Stage2 Compiler
288
289class TagAndPayload_SynthProvider:
290 def __init__(self, value, _=None): self.value = value
291 def update(self):
292 try:
293 self.tag = self.value.GetChildMemberWithName('tag') or self.value.GetChildMemberWithName('tag_ptr').deref.Clone('tag')
294 data = self.value.GetChildMemberWithName('data_ptr') or self.value.GetChildMemberWithName('data')
295 self.payload = data.GetChildMemberWithName('payload').GetChildMemberWithName(data.GetChildMemberWithName('tag').value)
296 except: pass
297 def has_children(self): return True
298 def num_children(self): return 2
299 def get_child_index(self, name):
300 try: return ('tag', 'payload').index(name)
301 except: return -1
302 def get_child_at_index(self, index): return (self.tag, self.payload)[index] if index >= 0 and index < 2 else None
303
304def Inst_Ref_SummaryProvider(value, _=None):
305 members = value.type.enum_members
306 return value if any(value.unsigned == member.unsigned for member in members) else 'instructions[%d]' % (value.unsigned - len(members))
307
308class Module_Decl__Module_Decl_Index_SynthProvider:
309 def __init__(self, value, _=None): self.value = value
310 def update(self):
311 try:
312 for frame in self.value.thread:
313 mod = frame.FindVariable('mod') or frame.FindVariable('module')
314 if mod: break
315 else: return
316 self.ptr = mod.GetChildMemberWithName('allocated_decls').GetChildAtIndex(self.value.unsigned).Clone('decl')
317 except: pass
318 def has_children(self): return True
319 def num_children(self): return 1
320 def get_child_index(self, name): return 0 if name == 'decl' else -1
321 def get_child_at_index(self, index): return self.ptr if index == 0 else None
322
323class TagOrPayloadPtr_SynthProvider:
324 def __init__(self, value, _=None): self.value = value
325 def update(self):
326 try:
327 value_type = self.value.type
328 for helper in self.value.target.FindFunctions('%s.dbHelper' % value_type.name, lldb.eFunctionNameTypeFull):
329 ptr_self_type, ptr_tag_to_payload_map_type = helper.function.type.GetFunctionArgumentTypes()
330 self_type = ptr_self_type.GetPointeeType()
331 if self_type == value_type: break
332 else: return
333 tag_to_payload_map = {field.name: field.type for field in ptr_tag_to_payload_map_type.GetPointeeType().fields}
334
335 tag = self.value.GetChildMemberWithName('tag_if_small_enough')
336 if tag.unsigned < page_size:
337 self.tag = tag.Clone('tag')
338 self.payload = None
339 else:
340 ptr_otherwise = self.value.GetChildMemberWithName('ptr_otherwise')
341 self.tag = ptr_otherwise.GetChildMemberWithName('tag')
342 self.payload = ptr_otherwise.Cast(tag_to_payload_map[self.tag.value]).GetChildMemberWithName('data').Clone('payload')
343 except: pass
344 def has_children(self): return True
345 def num_children(self): return 1 + (self.payload is not None)
346 def get_child_index(self, name):
347 try: return ('tag', 'payload').index(name)
348 except: return -1
349 def get_child_at_index(self, index): return (self.tag, self.payload)[index] if index >= 0 and index < 2 else None
350
351def Module_Decl_name(decl):
352 error = lldb.SBError()
353 return decl.process.ReadCStringFromMemory(decl.GetChildMemberWithName('name').deref.load_addr, 256, error)
354
355def Module_Namespace_RenderFullyQualifiedName(namespace):
356 parent = namespace.GetChildMemberWithName('parent')
357 if parent.unsigned < page_size: return zig_String_decode(namespace.GetChildMemberWithName('file_scope').GetChildMemberWithName('sub_file_path')).removesuffix('.zig').replace('/', '.')
358 return '.'.join((Module_Namespace_RenderFullyQualifiedName(parent), Module_Decl_name(namespace.GetChildMemberWithName('ty').GetChildMemberWithName('payload').GetChildMemberWithName('owner_decl').GetChildMemberWithName('decl'))))
359
360def Module_Decl_RenderFullyQualifiedName(decl): return '.'.join((Module_Namespace_RenderFullyQualifiedName(decl.GetChildMemberWithName('src_namespace')), Module_Decl_name(decl)))
361
362def OwnerDecl_RenderFullyQualifiedName(payload): return Module_Decl_RenderFullyQualifiedName(payload.GetChildMemberWithName('owner_decl').GetChildMemberWithName('decl'))
363
364def type_Type_pointer(payload):
365 pointee_type = payload.GetChildMemberWithName('pointee_type')
366 sentinel = payload.GetChildMemberWithName('sentinel').GetChildMemberWithName('child')
367 align = payload.GetChildMemberWithName('align').unsigned
368 addrspace = payload.GetChildMemberWithName('addrspace').value
369 bit_offset = payload.GetChildMemberWithName('bit_offset').unsigned
370 host_size = payload.GetChildMemberWithName('host_size').unsigned
371 vector_index = payload.GetChildMemberWithName('vector_index')
372 allowzero = payload.GetChildMemberWithName('allowzero').unsigned
373 const = not payload.GetChildMemberWithName('mutable').unsigned
374 volatile = payload.GetChildMemberWithName('volatile').unsigned
375 size = payload.GetChildMemberWithName('size').value
376
377 if size == 'One': summary = '*'
378 elif size == 'Many': summary = '[*'
379 elif size == 'Slice': summary = '['
380 elif size == 'C': summary = '[*c'
381 if sentinel: summary += ':%s' % value_Value_SummaryProvider(sentinel)
382 if size != 'One': summary += ']'
383 if allowzero: summary += 'allowzero '
384 if align != 0 or host_size != 0 or vector_index.value != 'none': summary += 'align(%d%s%s) ' % (align, ':%d:%d' % (bit_offset, host_size) if bit_offset != 0 or host_size != 0 else '', ':?' if vector_index.value == 'runtime' else ':%d' % vector_index.unsigned if vector_index.value != 'none' else '')
385 if addrspace != 'generic': summary += 'addrspace(.%s) ' % addrspace
386 if const: summary += 'const '
387 if volatile: summary += 'volatile '
388 summary += type_Type_SummaryProvider(pointee_type)
389 return summary
390
391def type_Type_function(payload):
392 param_types = payload.GetChildMemberWithName('param_types').children
393 comptime_params = payload.GetChildMemberWithName('comptime_params').GetPointeeData(0, len(param_types)).uint8
394 return_type = payload.GetChildMemberWithName('return_type')
395 alignment = payload.GetChildMemberWithName('alignment').unsigned
396 noalias_bits = payload.GetChildMemberWithName('noalias_bits').unsigned
397 cc = payload.GetChildMemberWithName('cc').value
398 is_var_args = payload.GetChildMemberWithName('is_var_args').unsigned
399
400 return 'fn(%s)%s%s %s' % (', '.join(tuple(''.join(('comptime ' if comptime_param else '', 'noalias ' if noalias_bits & 1 << i else '', type_Type_SummaryProvider(param_type))) for i, (comptime_param, param_type) in enumerate(zip(comptime_params, param_types))) + (('...',) if is_var_args else ())), ' align(%d)' % alignment if alignment != 0 else '', ' callconv(.%s)' % cc if cc != 'Unspecified' else '', type_Type_SummaryProvider(return_type))
401
402def type_Type_SummaryProvider(value, _=None):
403 tag = value.GetChildMemberWithName('tag').value
404 return type_tag_handlers.get(tag, lambda payload: tag)(value.GetChildMemberWithName('payload'))
405
406type_tag_handlers = {
407 'atomic_order': lambda payload: 'std.builtin.AtomicOrder',
408 'atomic_rmw_op': lambda payload: 'std.builtin.AtomicRmwOp',
409 'calling_convention': lambda payload: 'std.builtin.CallingConvention',
410 'address_space': lambda payload: 'std.builtin.AddressSpace',
411 'float_mode': lambda payload: 'std.builtin.FloatMode',
412 'reduce_op': lambda payload: 'std.builtin.ReduceOp',
413 'modifier': lambda payload: 'std.builtin.CallModifier',
414 'prefetch_options': lambda payload: 'std.builtin.PrefetchOptions',
415 'export_options': lambda payload: 'std.builtin.ExportOptions',
416 'extern_options': lambda payload: 'std.builtin.ExternOptions',
417 'type_info': lambda payload: 'std.builtin.Type',
418
419 'enum_literal': lambda payload: '@TypeOf(.enum_literal)',
420 'null': lambda payload: '@TypeOf(null)',
421 'undefined': lambda payload: '@TypeOf(undefined)',
422 'empty_struct_literal': lambda payload: '@TypeOf(.{})',
423
424 'anyerror_void_error_union': lambda payload: 'anyerror!void',
425 'const_slice_u8': lambda payload: '[]const u8',
426 'const_slice_u8_sentinel_0': lambda payload: '[:0]const u8',
427 'fn_noreturn_no_args': lambda payload: 'fn() noreturn',
428 'fn_void_no_args': lambda payload: 'fn() void',
429 'fn_naked_noreturn_no_args': lambda payload: 'fn() callconv(.Naked) noreturn',
430 'fn_ccc_void_no_args': lambda payload: 'fn() callconv(.C) void',
431 'single_const_pointer_to_comptime_int': lambda payload: '*const comptime_int',
432 'manyptr_u8': lambda payload: '[*]u8',
433 'manyptr_const_u8': lambda payload: '[*]const u8',
434 'manyptr_const_u8_sentinel_0': lambda payload: '[*:0]const u8',
435
436 'function': type_Type_function,
437 'error_union': lambda payload: '%s!%s' % (type_Type_SummaryProvider(payload.GetChildMemberWithName('error_set')), type_Type_SummaryProvider(payload.GetChildMemberWithName('payload'))),
438 'array_u8': lambda payload: '[%d]u8' % payload.unsigned,
439 'array_u8_sentinel_0': lambda payload: '[%d:0]u8' % payload.unsigned,
440 'vector': lambda payload: '@Vector(%d, %s)' % (payload.GetChildMemberWithName('len').unsigned, type_Type_SummaryProvider(payload.GetChildMemberWithName('elem_type'))),
441 'array': lambda payload: '[%d]%s' % (payload.GetChildMemberWithName('len').unsigned, type_Type_SummaryProvider(payload.GetChildMemberWithName('elem_type'))),
442 'array_sentinel': lambda payload: '[%d:%s]%s' % (payload.GetChildMemberWithName('len').unsigned, value_Value_SummaryProvider(payload.GetChildMemberWithName('sentinel')), type_Type_SummaryProvider(payload.GetChildMemberWithName('elem_type'))),
443 'tuple': lambda payload: 'tuple{%s}' % ', '.join(('comptime %%s = %s' % value_Value_SummaryProvider(value) if value.GetChildMemberWithName('tag').value != 'unreachable_value' else '%s') % type_Type_SummaryProvider(type) for type, value in zip(payload.GetChildMemberWithName('types').children, payload.GetChildMemberWithName('values').children)),
444 'anon_struct': lambda payload: 'struct{%s}' % ', '.join(('comptime %%s: %%s = %s' % value_Value_SummaryProvider(value) if value.GetChildMemberWithName('tag').value != 'unreachable_value' else '%s: %s') % (zig_String_AsIdentifier(name, zig_IsFieldName), type_Type_SummaryProvider(type)) for name, type, value in zip(payload.GetChildMemberWithName('names').children, payload.GetChildMemberWithName('types').children, payload.GetChildMemberWithName('values').children)),
445 'pointer': type_Type_pointer,
446 'single_const_pointer': lambda payload: '*const %s' % type_Type_SummaryProvider(payload),
447 'single_mut_pointer': lambda payload: '*%s' % type_Type_SummaryProvider(payload),
448 'many_const_pointer': lambda payload: '[*]const %s' % type_Type_SummaryProvider(payload),
449 'many_mut_pointer': lambda payload: '[*]%s' % type_Type_SummaryProvider(payload),
450 'c_const_pointer': lambda payload: '[*c]const %s' % type_Type_SummaryProvider(payload),
451 'c_mut_pointer': lambda payload: '[*c]%s' % type_Type_SummaryProvider(payload),
452 'const_slice': lambda payload: '[]const %s' % type_Type_SummaryProvider(payload),
453 'mut_slice': lambda payload: '[]%s' % type_Type_SummaryProvider(payload),
454 'int_signed': lambda payload: 'i%d' % payload.unsigned,
455 'int_unsigned': lambda payload: 'u%d' % payload.unsigned,
456 'optional': lambda payload: '?%s' % type_Type_SummaryProvider(payload),
457 'optional_single_mut_pointer': lambda payload: '?*%s' % type_Type_SummaryProvider(payload),
458 'optional_single_const_pointer': lambda payload: '?*const %s' % type_Type_SummaryProvider(payload),
459 'anyframe_T': lambda payload: 'anyframe->%s' % type_Type_SummaryProvider(payload),
460 'error_set': lambda payload: type_tag_handlers['error_set_merged'](payload.GetChildMemberWithName('names')),
461 'error_set_single': lambda payload: 'error{%s}' % zig_String_AsIdentifier(payload, zig_IsFieldName),
462 'error_set_merged': lambda payload: 'error{%s}' % ','.join(zig_String_AsIdentifier(child.GetChildMemberWithName('key'), zig_IsFieldName) for child in payload.GetChildMemberWithName('entries').children),
463 'error_set_inferred': lambda payload: '@typeInfo(@typeInfo(@TypeOf(%s)).Fn.return_type.?).ErrorUnion.error_set' % OwnerDecl_RenderFullyQualifiedName(payload.GetChildMemberWithName('func')),
464
465 'enum_full': OwnerDecl_RenderFullyQualifiedName,
466 'enum_nonexhaustive': OwnerDecl_RenderFullyQualifiedName,
467 'enum_numbered': OwnerDecl_RenderFullyQualifiedName,
468 'enum_simple': OwnerDecl_RenderFullyQualifiedName,
469 'struct': OwnerDecl_RenderFullyQualifiedName,
470 'union': OwnerDecl_RenderFullyQualifiedName,
471 'union_safety_tagged': OwnerDecl_RenderFullyQualifiedName,
472 'union_tagged': OwnerDecl_RenderFullyQualifiedName,
473 'opaque': OwnerDecl_RenderFullyQualifiedName,
474}
475
476def value_Value_str_lit(payload):
477 for frame in payload.thread:
478 mod = frame.FindVariable('mod') or frame.FindVariable('module')
479 if mod: break
480 else: return
481 return '"%s"' % zig_String_decode(mod.GetChildMemberWithName('string_literal_bytes').GetChildMemberWithName('items'), payload.GetChildMemberWithName('index').unsigned, payload.GetChildMemberWithName('len').unsigned)
482
483def value_Value_SummaryProvider(value, _=None):
484 tag = value.GetChildMemberWithName('tag').value
485 return value_tag_handlers.get(tag, lambda payload: tag.removesuffix('_type'))(value.GetChildMemberWithName('payload'))
486
487value_tag_handlers = {
488 'undef': lambda payload: 'undefined',
489 'zero': lambda payload: '0',
490 'one': lambda payload: '1',
491 'void_value': lambda payload: '{}',
492 'unreachable_value': lambda payload: 'unreachable',
493 'null_value': lambda payload: 'null',
494 'bool_true': lambda payload: 'true',
495 'bool_false': lambda payload: 'false',
496
497 'empty_struct_value': lambda payload: '.{}',
498 'empty_array': lambda payload: '.{}',
499
500 'ty': type_Type_SummaryProvider,
501 'int_type': lambda payload: '%c%d' % (payload.GetChildMemberWithName('bits').unsigned, 's' if payload.GetChildMemberWithName('signed').unsigned == 1 else 'u'),
502 'int_u64': lambda payload: '%d' % payload.unsigned,
503 'int_i64': lambda payload: '%d' % payload.signed,
504 'int_big_positive': lambda payload: sum(child.unsigned << i * child.type.size * 8 for i, child in enumerate(payload.children)),
505 'int_big_negative': lambda payload: '-%s' % value_tag_handlers['int_big_positive'](payload),
506 'function': OwnerDecl_RenderFullyQualifiedName,
507 'extern_fn': OwnerDecl_RenderFullyQualifiedName,
508 'variable': lambda payload: value_Value_SummaryProvider(payload.GetChildMemberWithName('decl').GetChildMemberWithName('val')),
509 'runtime_value': value_Value_SummaryProvider,
510 'decl_ref': lambda payload: value_Value_SummaryProvider(payload.GetChildMemberWithName('decl').GetChildMemberWithName('val')),
511 'decl_ref_mut': lambda payload: value_Value_SummaryProvider(payload.GetChildMemberWithName('decl_index').GetChildMemberWithName('decl').GetChildMemberWithName('val')),
512 'comptime_field_ptr': lambda payload: '&%s' % value_Value_SummaryProvider(payload.GetChildMemberWithName('field_val')),
513 'elem_ptr': lambda payload: '(%s)[%d]' % (value_Value_SummaryProvider(payload.GetChildMemberWithName('array_ptr')), payload.GetChildMemberWithName('index').unsigned),
514 'field_ptr': lambda payload: '(%s).field[%d]' % (value_Value_SummaryProvider(payload.GetChildMemberWithName('container_ptr')), payload.GetChildMemberWithName('field_index').unsigned),
515 'bytes': lambda payload: '"%s"' % zig_String_decode(payload),
516 'str_lit': value_Value_str_lit,
517 'repeated': lambda payload: '.{%s} ** _' % value_Value_SummaryProvider(payload),
518 'empty_array_sentinel': lambda payload: '.{%s}' % value_Value_SummaryProvider(payload),
519 'slice': lambda payload: '(%s)[0..%s]' % tuple(value_Value_SummaryProvider(payload.GetChildMemberWithName(name)) for name in ('ptr', 'len')),
520 'float_16': lambda payload: payload.value,
521 'float_32': lambda payload: payload.value,
522 'float_64': lambda payload: payload.value,
523 'float_80': lambda payload: payload.value,
524 'float_128': lambda payload: payload.value,
525 'enum_literal': lambda payload: '.%s' % zig_String_AsIdentifier(payload, zig_IsFieldName),
526 'enum_field_index': lambda payload: 'field[%d]' % payload.unsigned,
527 'error': lambda payload: 'error.%s' % zig_String_AsIdentifier(payload.GetChildMemberWithName('name'), zig_IsFieldName),
528 'eu_payload': value_Value_SummaryProvider,
529 'eu_payload_ptr': lambda payload: '&((%s).* catch unreachable)' % value_Value_SummaryProvider(payload.GetChildMemberWithName('container_ptr')),
530 'opt_payload': value_Value_SummaryProvider,
531 'opt_payload_ptr': lambda payload: '&(%s).*.?' % value_Value_SummaryProvider(payload.GetChildMemberWithName('container_ptr')),
532 'aggregate': lambda payload: '.{%s}' % ', '.join(map(value_Value_SummaryProvider, payload.children)),
533 'union': lambda payload: '.{.%s = %s}' % tuple(value_Value_SummaryProvider(payload.GetChildMemberWithName(name)) for name in ('tag', 'val')),
534
535 'lazy_align': lambda payload: '@alignOf(%s)' % type_Type_SummaryProvider(payload),
536 'lazy_size': lambda payload: '@sizeOf(%s)' % type_Type_SummaryProvider(payload),
537}
538
539# Initialize
540
541def add(debugger, *, category, regex=False, type, identifier=None, synth=False, inline_children=False, expand=False, summary=False):
542 prefix = '.'.join((__name__, (identifier or type).replace('.', '_').replace(':', '_')))
543 if summary: debugger.HandleCommand('type summary add --category %s%s%s "%s"' % (category, ' --inline-children' if inline_children else ''.join((' --expand' if expand else '', ' --python-function %s_SummaryProvider' % prefix if summary == True else ' --summary-string "%s"' % summary)), ' --regex' if regex else '', type))
544 if synth: debugger.HandleCommand('type synthetic add --category %s%s --python-class %s_SynthProvider "%s"' % (category, ' --regex' if regex else '', prefix, type))
545
546def MultiArrayList_Entry(type): return '^multi_array_list\\.MultiArrayList\\(%s\\)\\.Entry__struct_[1-9][0-9]*$' % type
547
548def __lldb_init_module(debugger, _=None):
549 # Initialize Zig Language
550 add(debugger, category='zig', regex=True, type='^\\[\\]', identifier='zig_Slice', synth=True, expand=True, summary='len=${svar%#}')
551 add(debugger, category='zig', type='[]u8', identifier='zig_String', summary=True)
552 add(debugger, category='zig', regex=True, type='^\\?', identifier='zig_Optional', synth=True, summary=True)
553 add(debugger, category='zig', regex=True, type='^(error{.*}|anyerror)!', identifier='zig_ErrorUnion', synth=True, inline_children=True, summary=True)
554
555 # Initialize Zig Standard Library
556 add(debugger, category='zig.std', type='mem.Allocator', summary='${var.ptr}')
557 add(debugger, category='zig.std', regex=True, type='^segmented_list\\.SegmentedList\\(.*\\)$', identifier='std_SegmentedList', synth=True, expand=True, summary='len=${var.len}')
558 add(debugger, category='zig.std', regex=True, type='^multi_array_list\\.MultiArrayList\\(.*\\)$', identifier='std_MultiArrayList', synth=True, expand=True, summary='len=${var.len} capacity=${var.capacity}')
559 add(debugger, category='zig.std', regex=True, type=MultiArrayList_Entry('.*'), identifier='std_Entry', synth=True, inline_children=True, summary=True)
560 add(debugger, category='zig.std', regex=True, type='^hash_map\\.HashMapUnmanaged\\(.*\\)$', identifier='std_HashMapUnmanaged', synth=True, expand=True, summary=True)
561 add(debugger, category='zig.std', regex=True, type='^hash_map\\.HashMapUnmanaged\\(.*\\)\\.Entry$', identifier = 'std_Entry', synth=True, inline_children=True, summary=True)
562
563 # Initialize Zig Stage2 Compiler
564 add(debugger, category='zig.stage2', type='Zir.Inst', identifier='TagAndPayload', synth=True, inline_children=True, summary=True)
565 add(debugger, category='zig.stage2', regex=True, type=MultiArrayList_Entry('Zir\\.Inst'), identifier='TagAndPayload', synth=True, inline_children=True, summary=True)
566 add(debugger, category='zig.stage2', regex=True, type='^Zir\\.Inst\\.Data\\.Data__struct_[1-9][0-9]*$', inline_children=True, summary=True)
567 add(debugger, category='zig.stage2', type='Zir.Inst::Zir.Inst.Ref', identifier='Inst_Ref', summary=True)
568 add(debugger, category='zig.stage2', type='Air.Inst', identifier='TagAndPayload', synth=True, inline_children=True, summary=True)
569 add(debugger, category='zig.stage2', regex=True, type=MultiArrayList_Entry('Air\\.Inst'), identifier='TagAndPayload', synth=True, inline_children=True, summary=True)
570 add(debugger, category='zig.stage2', regex=True, type='^Air\\.Inst\\.Data\\.Data__struct_[1-9][0-9]*$', inline_children=True, summary=True)
571 add(debugger, category='zig.stage2', type='Module.Decl::Module.Decl.Index', synth=True)
572 add(debugger, category='zig.stage2', type='type.Type', identifier='TagOrPayloadPtr', synth=True)
573 add(debugger, category='zig.stage2', type='type.Type', summary=True)
574 add(debugger, category='zig.stage2', type='value.Value', identifier='TagOrPayloadPtr', synth=True)
575 add(debugger, category='zig.stage2', type='value.Value', summary=True)
tools/stage2_gdb_pretty_printers.py+105-14
...@@ -3,13 +3,55 @@...@@ -3,13 +3,55 @@
3import re3import re
4import gdb.printing4import gdb.printing
55
6import sys
7from pathlib import Path
8sys.path.insert(0, str(Path(__file__).parent))
9import stage2_pretty_printers_common as common
10
11
12class TypePrinter:6class TypePrinter:
7 no_payload_count = 4096
8
9 # Keep in sync with src/type.zig
10 # Types which have no payload do not need to be entered here.
11 payload_type_names = {
12 'array_u8': 'Type.Payload.Len',
13 'array_u8_sentinel_0': 'Type.Payload.Len',
14
15 'single_const_pointer': 'Type.Payload.ElemType',
16 'single_mut_pointer': 'Type.Payload.ElemType',
17 'many_const_pointer': 'Type.Payload.ElemType',
18 'many_mut_pointer': 'Type.Payload.ElemType',
19 'c_const_pointer': 'Type.Payload.ElemType',
20 'c_mut_pointer': 'Type.Payload.ElemType',
21 'const_slice': 'Type.Payload.ElemType',
22 'mut_slice': 'Type.Payload.ElemType',
23 'optional': 'Type.Payload.ElemType',
24 'optional_single_mut_pointer': 'Type.Payload.ElemType',
25 'optional_single_const_pointer': 'Type.Payload.ElemType',
26 'anyframe_T': 'Type.Payload.ElemType',
27
28 'int_signed': 'Type.Payload.Bits',
29 'int_unsigned': 'Type.Payload.Bits',
30
31 'error_set': 'Type.Payload.ErrorSet',
32 'error_set_inferred': 'Type.Payload.ErrorSetInferred',
33 'error_set_merged': 'Type.Payload.ErrorSetMerged',
34
35 'array': 'Type.Payload.Array',
36 'vector': 'Type.Payload.Array',
37
38 'array_sentinel': 'Type.Payload.ArraySentinel',
39 'pointer': 'Type.Payload.Pointer',
40 'function': 'Type.Payload.Function',
41 'error_union': 'Type.Payload.ErrorUnion',
42 'error_set_single': 'Type.Payload.Name',
43 'opaque': 'Type.Payload.Opaque',
44 'struct': 'Type.Payload.Struct',
45 'union': 'Type.Payload.Union',
46 'union_tagged': 'Type.Payload.Union',
47 'enum_full, .enum_nonexhaustive': 'Type.Payload.EnumFull',
48 'enum_simple': 'Type.Payload.EnumSimple',
49 'enum_numbered': 'Type.Payload.EnumNumbered',
50 'empty_struct': 'Type.Payload.ContainerScope',
51 'tuple': 'Type.Payload.Tuple',
52 'anon_struct': 'Type.Payload.AnonStruct',
53 }
54
13 def __init__(self, val):55 def __init__(self, val):
14 self.val = val56 self.val = val
1557
...@@ -17,7 +59,7 @@ class TypePrinter:...@@ -17,7 +59,7 @@ class TypePrinter:
17 tag_if_small_enough = self.val['tag_if_small_enough']59 tag_if_small_enough = self.val['tag_if_small_enough']
18 tag_type = tag_if_small_enough.type60 tag_type = tag_if_small_enough.type
1961
20 if tag_if_small_enough < common.Type.no_payload_count:62 if tag_if_small_enough < TypePrinter.no_payload_count:
21 return tag_if_small_enough63 return tag_if_small_enough
22 else:64 else:
23 return self.val['ptr_otherwise'].dereference()['tag']65 return self.val['ptr_otherwise'].dereference()['tag']
...@@ -27,7 +69,7 @@ class TypePrinter:...@@ -27,7 +69,7 @@ class TypePrinter:
27 if tag is None:69 if tag is None:
28 return None70 return None
2971
30 type_name = common.Type.payload_type_names.get(str(tag))72 type_name = TypePrinter.payload_type_names.get(str(tag))
31 if type_name is None:73 if type_name is None:
32 return None74 return None
33 return gdb.lookup_type('struct type.%s' % type_name)75 return gdb.lookup_type('struct type.%s' % type_name)
...@@ -36,12 +78,12 @@ class TypePrinter:...@@ -36,12 +78,12 @@ class TypePrinter:
36 tag = self.tag()78 tag = self.tag()
37 if tag is None:79 if tag is None:
38 return '(invalid type)'80 return '(invalid type)'
39 if self.val['tag_if_small_enough'] < common.Type.no_payload_count:81 if self.val['tag_if_small_enough'] < TypePrinter.no_payload_count:
40 return '.%s' % str(tag)82 return '.%s' % str(tag)
41 return None83 return None
4284
43 def children(self):85 def children(self):
44 if self.val['tag_if_small_enough'] < common.Type.no_payload_count:86 if self.val['tag_if_small_enough'] < TypePrinter.no_payload_count:
45 return87 return
4688
47 yield ('tag', '.%s' % str(self.tag()))89 yield ('tag', '.%s' % str(self.tag()))
...@@ -51,6 +93,55 @@ class TypePrinter:...@@ -51,6 +93,55 @@ class TypePrinter:
51 yield ('payload', self.val['ptr_otherwise'].cast(payload_type.pointer()).dereference()['data'])93 yield ('payload', self.val['ptr_otherwise'].cast(payload_type.pointer()).dereference()['data'])
5294
53class ValuePrinter:95class ValuePrinter:
96 no_payload_count = 4096
97
98 # Keep in sync with src/value.zig
99 # Values which have no payload do not need to be entered here.
100 payload_type_names = {
101 'big_int_positive': 'Value.Payload.BigInt',
102 'big_int_negative': 'Value.Payload.BigInt',
103
104 'extern_fn': 'Value.Payload.ExternFn',
105
106 'decl_ref': 'Value.Payload.Decl',
107
108 'repeated': 'Value.Payload.SubValue',
109 'eu_payload': 'Value.Payload.SubValue',
110 'opt_payload': 'Value.Payload.SubValue',
111 'empty_array_sentinel': 'Value.Payload.SubValue',
112
113 'eu_payload_ptr': 'Value.Payload.PayloadPtr',
114 'opt_payload_ptr': 'Value.Payload.PayloadPtr',
115
116 'bytes': 'Value.Payload.Bytes',
117 'enum_literal': 'Value.Payload.Bytes',
118
119 'slice': 'Value.Payload.Slice',
120
121 'enum_field_index': 'Value.Payload.U32',
122
123 'ty': 'Value.Payload.Ty',
124 'int_type': 'Value.Payload.IntType',
125 'int_u64': 'Value.Payload.U64',
126 'int_i64': 'Value.Payload.I64',
127 'function': 'Value.Payload.Function',
128 'variable': 'Value.Payload.Variable',
129 'decl_ref_mut': 'Value.Payload.DeclRefMut',
130 'elem_ptr': 'Value.Payload.ElemPtr',
131 'field_ptr': 'Value.Payload.FieldPtr',
132 'float_16': 'Value.Payload.Float_16',
133 'float_32': 'Value.Payload.Float_32',
134 'float_64': 'Value.Payload.Float_64',
135 'float_80': 'Value.Payload.Float_80',
136 'float_128': 'Value.Payload.Float_128',
137 'error': 'Value.Payload.Error',
138 'inferred_alloc': 'Value.Payload.InferredAlloc',
139 'inferred_alloc_comptime': 'Value.Payload.InferredAllocComptime',
140 'aggregate': 'Value.Payload.Aggregate',
141 'union': 'Value.Payload.Union',
142 'bound_fn': 'Value.Payload.BoundFn',
143 }
144
54 def __init__(self, val):145 def __init__(self, val):
55 self.val = val146 self.val = val
56147
...@@ -58,7 +149,7 @@ class ValuePrinter:...@@ -58,7 +149,7 @@ class ValuePrinter:
58 tag_if_small_enough = self.val['tag_if_small_enough']149 tag_if_small_enough = self.val['tag_if_small_enough']
59 tag_type = tag_if_small_enough.type150 tag_type = tag_if_small_enough.type
60151
61 if tag_if_small_enough < common.Value.no_payload_count:152 if tag_if_small_enough < ValuePrinter.no_payload_count:
62 return tag_if_small_enough153 return tag_if_small_enough
63 else:154 else:
64 return self.val['ptr_otherwise'].dereference()['tag']155 return self.val['ptr_otherwise'].dereference()['tag']
...@@ -68,7 +159,7 @@ class ValuePrinter:...@@ -68,7 +159,7 @@ class ValuePrinter:
68 if tag is None:159 if tag is None:
69 return None160 return None
70161
71 type_name = Comman.Value.payload_type_names.get(str(tag))162 type_name = ValuePrinter.payload_type_names.get(str(tag))
72 if type_name is None:163 if type_name is None:
73 return None164 return None
74 return gdb.lookup_type('struct value.%s' % type_name)165 return gdb.lookup_type('struct value.%s' % type_name)
...@@ -77,12 +168,12 @@ class ValuePrinter:...@@ -77,12 +168,12 @@ class ValuePrinter:
77 tag = self.tag()168 tag = self.tag()
78 if tag is None:169 if tag is None:
79 return '(invalid value)'170 return '(invalid value)'
80 if self.val['tag_if_small_enough'] < common.Value.no_payload_count:171 if self.val['tag_if_small_enough'] < ValuePrinter.no_payload_count:
81 return '.%s' % str(tag)172 return '.%s' % str(tag)
82 return None173 return None
83174
84 def children(self):175 def children(self):
85 if self.val['tag_if_small_enough'] < common.Value.no_payload_count:176 if self.val['tag_if_small_enough'] < ValuePrinter.no_payload_count:
86 return177 return
87178
88 yield ('tag', '.%s' % str(self.tag()))179 yield ('tag', '.%s' % str(self.tag()))
tools/stage2_lldb_pretty_printers.py deleted-59
...@@ -1,59 +0,0 @@
1# pretty printing for stage 2.
2# put "command script /path/to/stage2_lldb_pretty_printers.py" and "type category enable stage2" in ~/.lldbinit to load it automatically.
3import lldb
4import stage2_pretty_printers_common as common
5
6category = 'stage2'
7module = category + '_lldb_pretty_printers'
8
9class type_Type_SynthProvider:
10 def __init__(self, type, _=None):
11 self.type = type
12
13 def update(self):
14 self.tag = self.type.GetChildMemberWithName('tag_if_small_enough').Clone('tag')
15 self.payload = None
16 if self.tag.GetValueAsUnsigned() >= common.Type.no_payload_count:
17 ptr_otherwise = self.type.GetChildMemberWithName('ptr_otherwise')
18 self.tag = ptr_otherwise.Dereference().GetChildMemberWithName('tag')
19 payload_type = self.type.target.FindFirstType('type.' + common.Type.payload_type_names[self.tag.GetValue()])
20 self.payload = ptr_otherwise.Cast(payload_type.GetPointerType()).Dereference().GetChildMemberWithName('data').Clone('payload')
21
22 def num_children(self):
23 return 1 + (self.payload is not None)
24
25 def get_child_index(self, name):
26 return ['tag', 'payload'].index(name)
27
28 def get_child_at_index(self, index):
29 return [self.tag, self.payload][index]
30
31class value_Value_SynthProvider:
32 def __init__(self, value, _=None):
33 self.value = value
34
35 def update(self):
36 self.tag = self.value.GetChildMemberWithName('tag_if_small_enough').Clone('tag')
37 self.payload = None
38 if self.tag.GetValueAsUnsigned() >= common.Value.no_payload_count:
39 ptr_otherwise = self.value.GetChildMemberWithName('ptr_otherwise')
40 self.tag = ptr_otherwise.Dereference().GetChildMemberWithName('tag')
41 payload_type = self.value.target.FindFirstType('value.' + common.Value.payload_type_names[self.tag.GetValue()])
42 self.payload = ptr_otherwise.Cast(payload_type.GetPointerType()).Dereference().GetChildMemberWithName('data').Clone('payload')
43
44 def num_children(self):
45 return 1 + (self.payload is not None)
46
47 def get_child_index(self, name):
48 return ['tag', 'payload'].index(name)
49
50 def get_child_at_index(self, index):
51 return [self.tag, self.payload][index]
52
53def add(debugger, type, summary=False, synth=False):
54 if summary: debugger.HandleCommand('type summary add --python-function ' + module + '.' + type.replace('.', '_') + '_SummaryProvider "' + type + '" --category ' + category)
55 if synth: debugger.HandleCommand('type synthetic add --python-class ' + module + '.' + type.replace('.', '_') + '_SynthProvider "' + type + '" --category ' + category)
56
57def __lldb_init_module(debugger, _=None):
58 add(debugger, 'type.Type', synth=True)
59 add(debugger, 'value.Value', synth=True)
tools/stage2_pretty_printers_common.py deleted-98
...@@ -1,98 +0,0 @@
1class Type:
2 no_payload_count = 4096
3
4 # Keep in sync with src/type.zig
5 # Types which have no payload do not need to be entered here.
6 payload_type_names = {
7 'array_u8': 'Type.Payload.Len',
8 'array_u8_sentinel_0': 'Type.Payload.Len',
9
10 'single_const_pointer': 'Type.Payload.ElemType',
11 'single_mut_pointer': 'Type.Payload.ElemType',
12 'many_const_pointer': 'Type.Payload.ElemType',
13 'many_mut_pointer': 'Type.Payload.ElemType',
14 'c_const_pointer': 'Type.Payload.ElemType',
15 'c_mut_pointer': 'Type.Payload.ElemType',
16 'const_slice': 'Type.Payload.ElemType',
17 'mut_slice': 'Type.Payload.ElemType',
18 'optional': 'Type.Payload.ElemType',
19 'optional_single_mut_pointer': 'Type.Payload.ElemType',
20 'optional_single_const_pointer': 'Type.Payload.ElemType',
21 'anyframe_T': 'Type.Payload.ElemType',
22
23 'int_signed': 'Type.Payload.Bits',
24 'int_unsigned': 'Type.Payload.Bits',
25
26 'error_set': 'Type.Payload.ErrorSet',
27 'error_set_inferred': 'Type.Payload.ErrorSetInferred',
28 'error_set_merged': 'Type.Payload.ErrorSetMerged',
29
30 'array': 'Type.Payload.Array',
31 'vector': 'Type.Payload.Array',
32
33 'array_sentinel': 'Type.Payload.ArraySentinel',
34 'pointer': 'Type.Payload.Pointer',
35 'function': 'Type.Payload.Function',
36 'error_union': 'Type.Payload.ErrorUnion',
37 'error_set_single': 'Type.Payload.Name',
38 'opaque': 'Type.Payload.Opaque',
39 'struct': 'Type.Payload.Struct',
40 'union': 'Type.Payload.Union',
41 'union_tagged': 'Type.Payload.Union',
42 'enum_full, .enum_nonexhaustive': 'Type.Payload.EnumFull',
43 'enum_simple': 'Type.Payload.EnumSimple',
44 'enum_numbered': 'Type.Payload.EnumNumbered',
45 'empty_struct': 'Type.Payload.ContainerScope',
46 'tuple': 'Type.Payload.Tuple',
47 'anon_struct': 'Type.Payload.AnonStruct',
48 }
49
50class Value:
51 no_payload_count = 4096
52
53 # Keep in sync with src/value.zig
54 # Values which have no payload do not need to be entered here.
55 payload_type_names = {
56 'big_int_positive': 'Value.Payload.BigInt',
57 'big_int_negative': 'Value.Payload.BigInt',
58
59 'extern_fn': 'Value.Payload.ExternFn',
60
61 'decl_ref': 'Value.Payload.Decl',
62
63 'repeated': 'Value.Payload.SubValue',
64 'eu_payload': 'Value.Payload.SubValue',
65 'opt_payload': 'Value.Payload.SubValue',
66 'empty_array_sentinel': 'Value.Payload.SubValue',
67
68 'eu_payload_ptr': 'Value.Payload.PayloadPtr',
69 'opt_payload_ptr': 'Value.Payload.PayloadPtr',
70
71 'bytes': 'Value.Payload.Bytes',
72 'enum_literal': 'Value.Payload.Bytes',
73
74 'slice': 'Value.Payload.Slice',
75
76 'enum_field_index': 'Value.Payload.U32',
77
78 'ty': 'Value.Payload.Ty',
79 'int_type': 'Value.Payload.IntType',
80 'int_u64': 'Value.Payload.U64',
81 'int_i64': 'Value.Payload.I64',
82 'function': 'Value.Payload.Function',
83 'variable': 'Value.Payload.Variable',
84 'decl_ref_mut': 'Value.Payload.DeclRefMut',
85 'elem_ptr': 'Value.Payload.ElemPtr',
86 'field_ptr': 'Value.Payload.FieldPtr',
87 'float_16': 'Value.Payload.Float_16',
88 'float_32': 'Value.Payload.Float_32',
89 'float_64': 'Value.Payload.Float_64',
90 'float_80': 'Value.Payload.Float_80',
91 'float_128': 'Value.Payload.Float_128',
92 'error': 'Value.Payload.Error',
93 'inferred_alloc': 'Value.Payload.InferredAlloc',
94 'inferred_alloc_comptime': 'Value.Payload.InferredAllocComptime',
95 'aggregate': 'Value.Payload.Aggregate',
96 'union': 'Value.Payload.Union',
97 'bound_fn': 'Value.Payload.BoundFn',
98 }
tools/std_gdb_pretty_printers.py+2-2
...@@ -26,7 +26,7 @@ class MultiArrayListPrinter:...@@ -26,7 +26,7 @@ class MultiArrayListPrinter:
26 self.val = val26 self.val = val
2727
28 def child_type(self):28 def child_type(self):
29 (helper_fn, _) = gdb.lookup_symbol('%s.gdbHelper' % self.val.type.name)29 (helper_fn, _) = gdb.lookup_symbol('%s.dbHelper' % self.val.type.name)
30 return helper_fn.type.fields()[1].type.target()30 return helper_fn.type.fields()[1].type.target()
3131
32 def to_string(self):32 def to_string(self):
...@@ -65,7 +65,7 @@ class HashMapPrinter:...@@ -65,7 +65,7 @@ class HashMapPrinter:
65 self.val = val['unmanaged'] if is_managed else val65 self.val = val['unmanaged'] if is_managed else val
6666
67 def header_ptr_type(self):67 def header_ptr_type(self):
68 (helper_fn, _) = gdb.lookup_symbol('%s.gdbHelper' % self.val.type.name)68 (helper_fn, _) = gdb.lookup_symbol('%s.dbHelper' % self.val.type.name)
69 return helper_fn.type.fields()[1].type69 return helper_fn.type.fields()[1].type
7070
71 def header(self):71 def header(self):