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
|
# Work around awkwardness in gdb's python printers:
# 1. In src/main-gdb-head.py, define the printer mechanism.
# 2. In src/*/*.py, define all the printer classes.
# 3. In src/main-gdb-tail.py, reflect to actually add the printers.
# gdb sticks everything in one scope.
# This lets us enumerate what *we* added.
initial_globals = {id(v):v for v in globals().values()}
import re
# copied from gdb/types.py for compatibility with old gdb
def get_basic_type(type_):
"""Return the "basic" type of a type.
Arguments:
type_: The type to reduce to its basic type.
Returns:
type_ with const/volatile is stripped away,
and typedefs/references converted to the underlying type.
"""
while (type_.code == gdb.TYPE_CODE_REF or
type_.code == gdb.TYPE_CODE_TYPEDEF):
if type_.code == gdb.TYPE_CODE_REF:
type_ = type_.target()
else:
type_ = type_.strip_typedefs()
return type_.unqualified()
def finish():
global finish, initial_globals, FastPrinters
final_globals = {id(v):v for v in globals().values()}
diff = final_globals.keys() - initial_globals.keys() \
- {'finish', 'initial_globals', 'FastPrinters'}
fp = FastPrinters()
# After this, don't access any more globals in this function.
del finish, initial_globals, FastPrinters
for i in diff:
v = final_globals[i]
if hasattr(v, 'children') or hasattr(v, 'to_string'):
fp.add_printer(v)
gdb.current_objfile().pretty_printers.append(fp)
print('Added %d custom printers for %s'
% (len(fp.printers), gdb.current_objfile().filename))
class FastPrinters(object):
''' printer dispatch the way gdb *should* have done it
'''
__slots__ = ('name', 'enabled', 'printers')
def __init__(self):
self.name = 'tmwa'
self.enabled = True
self.printers = {}
def add_printer(self, cls):
assert hasattr(cls, 'enabled')
# TODO: check if the class name exists
# this is really hard since templates are involved
self.printers[cls.name] = cls
@property
def subprinters(self):
return list(self.printers.values())
def strip_templates(self, name, __pattern=re.compile('<[^<>]*>')):
# TODO what about '<' and '>' as non-type template parameters?
changed = 1
while changed:
name, changed = __pattern.subn('', name)
return name
def __call__(self, value):
stype = get_basic_type(value.type).tag
#dtype = get_basic_type(value.dynamic_type).tag
if stype is None:
return
stype = self.strip_templates(stype)
p = self.printers.get(stype)
if p is not None and p.enabled:
return p(value)
return None
|
