| ... | @@ -1,7 +1,8 @@ | ... | @@ -1,7 +1,8 @@ |
| 1 | # pretty printing for stage1 | 1 | # pretty printing for stage1, stage2 and the standard library |
| 2 | # put "source /path/to/zig-gdb.py" in ~/.gdbinit to load it automatically | 2 | # put "source /path/to/zig-gdb.py" in ~/.gdbinit to load it automatically |
| 3 | | 3 | |
| 4 | import gdb.printing | 4 | import gdb.printing |
| | 5 | import re |
| 5 | | 6 | |
| 6 | class ZigListPrinter: | 7 | class ZigListPrinter: |
| 7 | def __init__(self, val): | 8 | def __init__(self, val): |
| ... | @@ -31,9 +32,59 @@ class BufPrinter: | ... | @@ -31,9 +32,59 @@ class BufPrinter: |
| 31 | def display_hint(self): | 32 | def display_hint(self): |
| 32 | return 'string' | 33 | return 'string' |
| 33 | | 34 | |
| | 35 | # Handles both ArrayList and ArrayListUnmanaged. |
| | 36 | class ArrayListPrinter: |
| | 37 | def __init__(self, val): |
| | 38 | self.val = val |
| | 39 | |
| | 40 | def to_string(self): |
| | 41 | type = self.val.type.name[len('std.array_list.'):] |
| | 42 | type = re.sub(r'ArrayListAligned(Unmanaged)?\((.*),null\)$', r'ArrayList\1(\2)', type) |
| | 43 | return '%s of length %s, capacity %s' % (type, self.val['items']['len'], self.val['capacity']) |
| | 44 | |
| | 45 | def children(self): |
| | 46 | for i in range(int(self.val['items']['len'])): |
| | 47 | item = self.val['items']['ptr'] + i |
| | 48 | yield ('[%d]' % i, item.dereference()) |
| | 49 | |
| | 50 | def display_hint(self): |
| | 51 | return 'array' |
| | 52 | |
| | 53 | # Handles both HashMap and HashMapUnmanaged |
| | 54 | class HashMapPrinter: |
| | 55 | def __init__(self, val): |
| | 56 | self.type = val.type |
| | 57 | is_managed = re.search(r'^std\.hash_map\.HashMap\(', self.type.name) |
| | 58 | self.val = val['unmanaged'] if is_managed else val |
| | 59 | |
| | 60 | def header(self): |
| | 61 | (header_fn, _) = gdb.lookup_symbol('%s.header' % self.val.type.name) |
| | 62 | header_ptr_type = header_fn.type.target() |
| | 63 | return (self.val['metadata'].cast(header_ptr_type) - 1).dereference() |
| | 64 | |
| | 65 | def to_string(self): |
| | 66 | type = self.type.name[len('std.hash_map.'):] |
| | 67 | type = re.sub(r'^HashMap(Unmanaged)?\((.*),std.hash_map.AutoContext\(.*$', r'AutoHashMap\1(\2)', type) |
| | 68 | return '%s of length %s, capacity %s' % (type, self.val['size'], self.header()['capacity']) |
| | 69 | |
| | 70 | def children(self): |
| | 71 | hdr = self.header() |
| | 72 | for i in range(int(hdr['capacity'])): |
| | 73 | metadata = self.val['metadata'] + i |
| | 74 | if metadata.dereference()['used'] == 1: |
| | 75 | yield ('[%d]' % i, (hdr['keys'] + i).dereference()) |
| | 76 | yield ('[%d]' % i, (hdr['values'] + i).dereference()) |
| | 77 | |
| | 78 | def display_hint(self): |
| | 79 | return 'map' |
| | 80 | |
| 34 | pp = gdb.printing.RegexpCollectionPrettyPrinter('Zig stage1 compiler') | 81 | pp = gdb.printing.RegexpCollectionPrettyPrinter('Zig stage1 compiler') |
| 35 | pp.add_printer('Buf', '^Buf$', BufPrinter) | 82 | pp.add_printer('Buf', '^Buf$', BufPrinter) |
| 36 | pp.add_printer('ZigList<char>', '^ZigList<char>$', BufPrinter) | 83 | pp.add_printer('ZigList<char>', '^ZigList<char>$', BufPrinter) |
| 37 | pp.add_printer('ZigList', '^ZigList<.*>$', ZigListPrinter) | 84 | pp.add_printer('ZigList', '^ZigList<.*>$', ZigListPrinter) |
| 38 | | | |
| 39 | gdb.printing.register_pretty_printer(gdb.current_objfile(), pp) | 85 | gdb.printing.register_pretty_printer(gdb.current_objfile(), pp) |
| | 86 | |
| | 87 | ppstd = gdb.printing.RegexpCollectionPrettyPrinter('Zig standard library') |
| | 88 | ppstd.add_printer('ArrayList', r'^std\.array_list\.ArrayListAligned(Unmanaged)?\(.*\)$', ArrayListPrinter) |
| | 89 | ppstd.add_printer('HashMap', r'^std\.hash_map\.HashMap(Unmanaged)?\(.*\)$', HashMapPrinter) |
| | 90 | gdb.printing.register_pretty_printer(gdb.current_objfile(), ppstd) |
Robin Voetter <robin@voetter.nl> 2022-03-16 12:40:08+01:00
tools