Newer
Older
Johnny Chen
committed
#
Johnny Chen
committed
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
#
# This script modifies the lldb module (which was automatically generated via
# running swig) to support iteration for certain lldb objects, adds a global
# variable 'debugger_unique_id' and initializes it to 0.
#
# It also calls SBDebugger.Initialize() to initialize the lldb debugger
# subsystem.
#
import sys, re, StringIO
if len (sys.argv) != 2:
output_name = "./lldb.py"
else:
output_name = sys.argv[1] + "/lldb.py"
# print "output_name is '" + output_name + "'"
#
# lldb_iter() should appear before the our first SB* class definition.
#
lldb_iter_def = '''
# ===================================
# Iterator for lldb container objects
# ===================================
def lldb_iter(obj, getsize, getelem):
"""A generator adaptor to support iteration for lldb container objects."""
size = getattr(obj, getsize)
elem = getattr(obj, getelem)
for i in range(size()):
yield elem(i)
'''
#
# This supports the iteration protocol.
#
iter_def = " def __iter__(self): return lldb_iter(self, '%s', '%s')"
module_iter = " def module_iter(self): return lldb_iter(self, '%s', '%s')"
breakpoint_iter = " def breakpoint_iter(self): return lldb_iter(self, '%s', '%s')"
#
# The dictionary defines a mapping from classname to (getsize, getelem) tuple.
#
d = { 'SBBreakpoint': ('GetNumLocations', 'GetLocationAtIndex'),
'SBCompileUnit': ('GetNumLineEntries', 'GetLineEntryAtIndex'),
'SBDebugger': ('GetNumTargets', 'GetTargetAtIndex'),
'SBModule': ('GetNumSymbols', 'GetSymbolAtIndex'),
'SBProcess': ('GetNumThreads', 'GetThreadAtIndex'),
'SBThread': ('GetNumFrames', 'GetFrameAtIndex'),
'SBInstructionList': ('GetSize', 'GetInstructionAtIndex'),
'SBStringList': ('GetSize', 'GetStringAtIndex',),
'SBSymbolContextList': ('GetSize', 'GetContextAtIndex'),
'SBValueList': ('GetSize', 'GetValueAtIndex'),
'SBType': ('GetNumberChildren', 'GetChildAtIndex'),
'SBValue': ('GetNumChildren', 'GetChildAtIndex'),
'SBTarget': {'module': ('GetNumModules', 'GetModuleAtIndex'),
'breakpoint': ('GetNumBreakpoints', 'GetBreakpointAtIndex')
}
}
# The new content will have the iteration protocol defined for our lldb objects.
new_content = StringIO.StringIO()
with open(output_name, 'r') as f_in:
content = f_in.read()
# The pattern for recognizing the beginning of an SB class definition.
class_pattern = re.compile("^class (SB.*)\(_object\):$")
# The pattern for recognizing the beginning of the __init__ method definition.
init_pattern = re.compile("^ def __init__\(self, \*args\):")
# These define the states of our state machine.
NORMAL = 0
DEFINING_ITERATOR = 1
DEFINING_TARGET_ITERATOR = 2
# The lldb_iter_def only needs to be inserted once.
lldb_iter_defined = False;
state = NORMAL
for line in content.splitlines():
if state == NORMAL:
match = class_pattern.search(line)
if not lldb_iter_defined and match:
print >> new_content, lldb_iter_def
lldb_iter_defined = True
if match and match.group(1) in d:
# Adding support for iteration for the matched SB class.
cls = match.group(1)
# Next state will be DEFINING_ITERATOR.
state = DEFINING_ITERATOR
elif state == DEFINING_ITERATOR:
match = init_pattern.search(line)
if match:
# We found the beginning of the __init__ method definition.
# This is a good spot to insert the iteration support.
#
# But note that SBTarget has two types of iterations.
if cls == "SBTarget":
print >> new_content, module_iter % (d[cls]['module'])
print >> new_content, breakpoint_iter % (d[cls]['breakpoint'])
else:
print >> new_content, iter_def % d[cls]
# Next state will be NORMAL.
state = NORMAL
# Pass the original line of content to the ew_content.
print >> new_content, line
with open(output_name, 'w') as f_out:
f_out.write(new_content.getvalue())
f_out.write("debugger_unique_id = 0\n")
f_out.write("SBDebugger.Initialize()\n")