-
Notifications
You must be signed in to change notification settings - Fork 1
/
python-gdb.py
109 lines (83 loc) · 3.39 KB
/
python-gdb.py
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
import gdb.printing
def pprinter(f):
class Printer:
def __init__(self,val):
self.val = val
def to_string(self):
return f(self.val)
return Printer
def v_array_str(val):
store = val.type.template_argument(0)
store_v = val['store']
items = store_v['items']
if store.tag.startswith('fixed::item_store'):
size = int(store.template_argument(0))
items = items.cast(items.type.target().pointer())
else:
size = int(store_v['size'])
return ','.join(str((items + i).dereference()) for i in range(size))
@pprinter
def vectorPrinter(val):
return '<{0}>'.format(v_array_str(val))
@pprinter
def varrayPrinter(val):
return '{{{0}}}'.format(v_array_str(val))
@pprinter
def vtypePrinter(val):
return str(val['data']['p'])
@pprinter
def fixedinitarrayPrinter(val):
type = val.type.template_argument(0)
items = val['data']['__data'].cast(val.type.template_argument(0).pointer())
return '{{{0}}}'.format(','.join(str((items + i).dereference()) for i in range(val.type.template_argument(1))))
def build_pretty_printer():
pp = gdb.printing.RegexpCollectionPrettyPrinter('ntracer')
pp.add_printer('vector','^impl::vector<.+>$',vectorPrinter)
pp.add_printer('v_array','^impl::v_array<.+>$',varrayPrinter)
pp.add_printer('v_type','^simd::(?:float|double|int(?:8|16|32|64))_v_(?:128|256|512)$',vtypePrinter)
pp.add_printer('fixed_init_array','^fixed::init_array<.+>$',fixedinitarrayPrinter)
return pp
gdb.printing.register_pretty_printer(
gdb.current_objfile(),
build_pretty_printer(),
True)
try:
import gdb.xmethod
except ImportError:
pass
else:
class XMethod(gdb.xmethod.XMethod):
def __init__(self,name,meth,worker):
gdb.xmethod.XMethod.__init__(self,name)
self.meth = meth
self.worker = worker
class XMethodWorker(gdb.xmethod.XMethodWorker):
def __init__(self,ctype):
gdb.xmethod.XMethodWorker.__init__(self)
self.ctype = ctype
def get_arg_types(self):
return ()
class PyPtrMatcher(gdb.xmethod.XMethodMatcher):
def __init__(self,name,type_prefix,methods):
gdb.xmethod.XMethodMatcher.__init__(self,name)
self.methods = list(methods)
self.mmap = {}
for m in self.methods:
if m.meth not in self.mmap: self.mmap[m.meth] = []
self.mmap[m.meth].append(m)
def match(self,ctype,mname):
if not ctype.tag.startswith(self.type_prefix): return None
return [m.worker(ctype) for m in self.mmap[mname] if m.enabled]
class PyPtrWorker_arrow(XMethodWorker):
def __call__(self,obj):
return obj['_obj']['_ptr'].cast(self.ctype.template_argument(0).pointer())
class TProtoBase_getbase(XMethodWorker):
def __call__(self,obj):
r = obj['base']
if self.ctype.template_argument(2) == False:
r = r['_M_t']['_M_head_impl'].dereference()
return r
gdb.xmethod.register_xmethod_matcher(None,
PyPtrMatcher('PyPtrMatcher','py::pyptr<',[XMethod('arrow','operator->',PyPtrWorker_arrow)]),True)
gdb.xmethod.register_xmethod_matcher(None,
PyPtrMatcher('TProtoBaseMatcher','tproto_base<',[XMethod('getbase','get_base',TProtoBase_getbase)]),True)