webkit.py: Really fix the printing of StringImpl now that 8-bit strings are present.
[WebKit-https.git] / Tools / gdb / webkit.py
1 # Copyright (C) 2010, Google Inc. All rights reserved.
2 #
3 # Redistribution and use in source and binary forms, with or without
4 # modification, are permitted provided that the following conditions are
5 # met:
6 #
7 #     * Redistributions of source code must retain the above copyright
8 # notice, this list of conditions and the following disclaimer.
9 #     * Redistributions in binary form must reproduce the above
10 # copyright notice, this list of conditions and the following disclaimer
11 # in the documentation and/or other materials provided with the
12 # distribution.
13 #     * Neither the name of Google Inc. nor the names of its
14 # contributors may be used to endorse or promote products derived from
15 # this software without specific prior written permission.
16 #
17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
29 """GDB support for WebKit types.
30
31 Add this to your gdb by amending your ~/.gdbinit as follows:
32   python
33   import sys
34   sys.path.insert(0, "/path/to/tools/gdb/")
35   import webkit
36 """
37
38 import gdb
39 import re
40 import struct
41
42
43 def ustring_to_string(ptr, length=None):
44     """Convert a pointer to UTF-16 data into a Python Unicode string.
45
46     ptr and length are both gdb.Value objects.
47     If length is unspecified, will guess at the length."""
48     extra = ''
49     if length is None:
50         # Try to guess at the length.
51         for i in xrange(0, 2048):
52             try:
53                 if int((ptr + i).dereference()) == 0:
54                     length = i
55                     break
56             except RuntimeError:
57                 # We indexed into inaccessible memory; give up.
58                 length = i
59                 extra = u' (gdb hit inaccessible memory)'
60                 break
61         if length is None:
62             length = 256
63             extra = u' (gdb found no trailing NUL)'
64     else:
65         length = int(length)
66
67     char_vals = [int((ptr + i).dereference()) for i in xrange(length)]
68     string = struct.pack('H' * length, *char_vals).decode('utf-16', 'replace')
69
70     return string + extra
71
72
73 class StringPrinter(object):
74     "Shared code between different string-printing classes"
75     def __init__(self, val):
76         self.val = val
77
78     def display_hint(self):
79         return 'string'
80
81
82 class UCharStringPrinter(StringPrinter):
83     "Print a UChar*; we must guess at the length"
84     def to_string(self):
85         return ustring_to_string(self.val)
86
87
88 class WTFAtomicStringPrinter(StringPrinter):
89     "Print a WTF::AtomicString"
90     def to_string(self):
91         return self.val['m_string']
92
93
94 class WTFCStringPrinter(StringPrinter):
95     "Print a WTF::CString"
96     def to_string(self):
97         # The CString holds a buffer, which is a refptr to a WTF::Vector of chars.
98         vector = self.val['m_buffer']['m_ptr']['m_vector']
99         # The vector has two more layers of buffer members.
100         return vector['m_buffer']['m_buffer']
101
102
103 class WTFStringPrinter(StringPrinter):
104     "Print a WTF::String"
105     def get_length(self):
106         if not self.val['m_impl']['m_ptr']:
107             return 0
108         return self.val['m_impl']['m_ptr']['m_length']
109
110     def is_8bit(self):
111         return self.val['m_impl']['m_ptr']['m_hashAndFlags'] & self.val['m_impl']['m_ptr']['s_hashFlag8BitBuffer']
112
113     def to_string(self):
114         if self.get_length() == 0:
115             return '(null)'
116
117         if self.is_8bit():
118             data_member = 'm_data8'
119         else:
120             data_member = 'm_data16'
121
122         return ustring_to_string(self.val['m_impl']['m_ptr'][data_member],
123                                  self.get_length())
124
125
126 class JSCUStringPrinter(StringPrinter):
127     "Print a JSC::UString"
128     def get_length(self):
129         if not self.val['m_impl']['m_ptr']:
130             return 0
131         return self.val['m_impl']['m_ptr']['m_length']
132
133     def is_8bit(self):
134         return self.val['m_impl']['m_ptr']['m_hashAndFlags'] & self.val['m_impl']['m_ptr']['s_hashFlag8BitBuffer']
135
136     def to_string(self):
137         if self.get_length() == 0:
138             return ''
139
140         if self.is_8bit():
141             data_member = 'm_data8'
142         else:
143             data_member = 'm_data16'
144
145         return ustring_to_string(self.val['m_impl']['m_ptr'][data_member],
146                                  self.get_length())
147
148
149 class JSCIdentifierPrinter(StringPrinter):
150     "Print a JSC::Identifier"
151     def to_string(self):
152         return JSCUStringPrinter(self.val['m_string']).to_string()
153
154
155 class JSCJSStringPrinter(StringPrinter):
156     "Print a JSC::JSString"
157     def to_string(self):
158         if self.val['m_length'] == 0:
159             return ''
160
161         return JSCUStringPrinter(self.val['m_value']).to_string()
162
163
164 class WebCoreKURLGooglePrivatePrinter(StringPrinter):
165     "Print a WebCore::KURLGooglePrivate"
166     def to_string(self):
167         return WTFCStringPrinter(self.val['m_utf8']).to_string()
168
169
170 class WebCoreQualifiedNamePrinter(StringPrinter):
171     "Print a WebCore::QualifiedName"
172
173     def __init__(self, val):
174         super(WebCoreQualifiedNamePrinter, self).__init__(val)
175         self.prefix_length = 0
176         self.length = 0
177         if self.val['m_impl']:
178             self.prefix_printer = WTFStringPrinter(
179                 self.val['m_impl']['m_prefix']['m_string'])
180             self.local_name_printer = WTFStringPrinter(
181                 self.val['m_impl']['m_localName']['m_string'])
182             self.prefix_length = self.prefix_printer.get_length()
183             if self.prefix_length > 0:
184                 self.length = (self.prefix_length + 1 +
185                     self.local_name_printer.get_length())
186             else:
187                 self.length = self.local_name_printer.get_length()
188
189     def get_length(self):
190         return self.length
191
192     def to_string(self):
193         if self.get_length() == 0:
194             return "(null)"
195         else:
196             if self.prefix_length > 0:
197                 return (self.prefix_printer.to_string() + ":" +
198                     self.local_name_printer.to_string())
199             else:
200                 return self.local_name_printer.to_string()
201
202
203 class WTFVectorPrinter:
204     """Pretty Printer for a WTF::Vector.
205
206     The output of this pretty printer is similar to the output of std::vector's
207     pretty printer, which is bundled in gcc.
208
209     Example gdb session should look like:
210     (gdb) p v
211     $3 = WTF::Vector of length 7, capacity 16 = {7, 17, 27, 37, 47, 57, 67}
212     (gdb) set print elements 3
213     (gdb) p v
214     $6 = WTF::Vector of length 7, capacity 16 = {7, 17, 27...}
215     (gdb) set print array
216     (gdb) p v
217     $7 = WTF::Vector of length 7, capacity 16 = {
218       7,
219       17,
220       27
221       ...
222     }
223     (gdb) set print elements 200
224     (gdb) p v
225     $8 = WTF::Vector of length 7, capacity 16 = {
226       7,
227       17,
228       27,
229       37,
230       47,
231       57,
232       67
233     }
234     """
235
236     class Iterator:
237         def __init__(self, start, finish):
238             self.item = start
239             self.finish = finish
240             self.count = 0
241
242         def __iter__(self):
243             return self
244
245         def next(self):
246             if self.item == self.finish:
247                 raise StopIteration
248             count = self.count
249             self.count += 1
250             element = self.item.dereference()
251             self.item += 1
252             return ('[%d]' % count, element)
253
254     def __init__(self, val):
255         self.val = val
256
257     def children(self):
258         start = self.val['m_buffer']['m_buffer']
259         return self.Iterator(start, start + self.val['m_size'])
260
261     def to_string(self):
262         return ('%s of length %d, capacity %d'
263                 % ('WTF::Vector', self.val['m_size'], self.val['m_buffer']['m_capacity']))
264
265     def display_hint(self):
266         return 'array'
267
268 def add_pretty_printers():
269     pretty_printers = (
270         (re.compile("^WTF::Vector<.*>$"), WTFVectorPrinter),
271         (re.compile("^WTF::AtomicString$"), WTFAtomicStringPrinter),
272         (re.compile("^WTF::CString$"), WTFCStringPrinter),
273         (re.compile("^WTF::String$"), WTFStringPrinter),
274         (re.compile("^WebCore::KURLGooglePrivate$"), WebCoreKURLGooglePrivatePrinter),
275         (re.compile("^WebCore::QualifiedName$"), WebCoreQualifiedNamePrinter),
276         (re.compile("^JSC::UString$"), JSCUStringPrinter),
277         (re.compile("^JSC::Identifier$"), JSCIdentifierPrinter),
278         (re.compile("^JSC::JSString$"), JSCJSStringPrinter),
279     )
280
281     def lookup_function(val):
282         """Function used to load pretty printers; will be passed to GDB."""
283         type = val.type
284         if type.code == gdb.TYPE_CODE_REF:
285             type = type.target()
286         type = type.unqualified().strip_typedefs()
287         tag = type.tag
288         if tag:
289             for function, pretty_printer in pretty_printers:
290                 if function.search(tag):
291                     return pretty_printer(val)
292
293         if type.code == gdb.TYPE_CODE_PTR:
294             name = str(type.target().unqualified())
295             if name == 'UChar':
296                 return UCharStringPrinter(val)
297         return None
298
299     gdb.pretty_printers.append(lookup_function)
300
301
302 add_pretty_printers()
303
304
305 class PrintPathToRootCommand(gdb.Command):
306     """Command for printing WebKit Node trees.
307
308     Usage: printpathtoroot variable_name"""
309
310     def __init__(self):
311         super(PrintPathToRootCommand, self).__init__("printpathtoroot",
312             gdb.COMMAND_SUPPORT,
313             gdb.COMPLETE_NONE)
314
315     def invoke(self, arg, from_tty):
316         element_type = gdb.lookup_type('WebCore::Element')
317         node_type = gdb.lookup_type('WebCore::Node')
318         frame = gdb.selected_frame()
319         try:
320             val = gdb.Frame.read_var(frame, arg)
321         except:
322             print "No such variable, or invalid type"
323             return
324
325         target_type = str(val.type.target().strip_typedefs())
326         if target_type == str(node_type):
327             stack = []
328             while val:
329                 stack.append([val,
330                     val.cast(element_type.pointer()).dereference()['m_tagName']])
331                 val = val.dereference()['m_parent']
332
333             padding = ''
334             while len(stack) > 0:
335                 pair = stack.pop()
336                 print padding, pair[1], pair[0]
337                 padding = padding + '  '
338         else:
339             print 'Sorry: I don\'t know how to deal with %s yet.' % target_type
340
341
342 PrintPathToRootCommand()