OLD | NEW |
| (Empty) |
1 #!/usr/bin/python | |
2 | |
3 # Copyright (c) 2010 The Chromium Authors. All rights reserved. | |
4 # Use of this source code is governed by a BSD-style license that can be | |
5 # found in the LICENSE file. | |
6 | |
7 """This script should be run manually on occasion to make sure all PPAPI types | |
8 have appropriate size checking. | |
9 | |
10 """ | |
11 | |
12 import optparse | |
13 import os | |
14 import subprocess | |
15 import sys | |
16 | |
17 | |
18 # The string that the PrintNamesAndSizes plugin uses to indicate whether or not | |
19 # a type has a pointer. Types that have pointers are not expected to have the | |
20 # same size on 32-bit and 64-bit architectures. | |
21 HAS_POINTER_STRING = "HasPointer" | |
22 | |
23 # These are types that don't include a pointer but nonetheless have | |
24 # architecture-dependent size. They all are ultimately typedefs to 'long' or | |
25 # 'unsigned long'. If there get to be too many types that use 'long' or | |
26 # 'unsigned long', we can add detection of them to the plugin to automate this. | |
27 ARCH_DEPENDENT_TYPES = set(["khronos_intptr_t", | |
28 "khronos_uintptr_t", | |
29 "khronos_ssize_t", | |
30 "khronos_usize_t", | |
31 "GLintptr", | |
32 "GLsizeiptr" | |
33 ]) | |
34 | |
35 | |
36 | |
37 class SourceLocation: | |
38 | |
39 """A class representing the source location of a definiton.""" | |
40 | |
41 def __init__(self, filename="", start_line=-1, end_line=-1): | |
42 self.filename = os.path.normpath(filename) | |
43 self.start_line = start_line | |
44 self.end_line = end_line | |
45 | |
46 | |
47 | |
48 class TypeInfo: | |
49 | |
50 """A class representing information about a C++ type.""" | |
51 | |
52 def __init__(self, info_string, target): | |
53 [self.kind, self.name, self.size, has_pointer_string, source_file, | |
54 start_line, end_line] = info_string.split(',') | |
55 self.target = target | |
56 self.parsed_line = info_string | |
57 # Note that Clang counts line numbers from 1, but we want to count from 0. | |
58 self.source_location = SourceLocation(source_file, | |
59 int(start_line)-1, | |
60 int(end_line)-1) | |
61 # If the type is one of our known special cases, mark it as architecture- | |
62 # dependent. | |
63 if self.name in ARCH_DEPENDENT_TYPES: | |
64 self.arch_dependent = True | |
65 # Otherwise, its size can be architecture dependent if it contains one or | |
66 # more pointers (or is a pointer). | |
67 else: | |
68 self.arch_dependent = (has_pointer_string == HAS_POINTER_STRING) | |
69 self.target = target | |
70 self.parsed_line = info_string | |
71 | |
72 | |
73 | |
74 class FilePatch: | |
75 | |
76 """A class representing a set of line-by-line changes to a particular file. | |
77 None | |
78 of the changes are applied until Apply is called. All line numbers are | |
79 counted from 0. | |
80 """ | |
81 | |
82 def __init__(self, filename): | |
83 self.filename = filename | |
84 self.linenums_to_delete = set() | |
85 # A dictionary from line number to an array of strings to be inserted at | |
86 # that line number. | |
87 self.lines_to_add = {} | |
88 | |
89 def Delete(self, start_line, end_line): | |
90 """Make the patch delete the lines [start_line, end_line).""" | |
91 self.linenums_to_delete |= set(range(start_line, end_line)) | |
92 | |
93 def Add(self, text, line_number): | |
94 """Add the given text before the text on the given line number.""" | |
95 if line_number in self.lines_to_add: | |
96 self.lines_to_add[line_number].append(text) | |
97 else: | |
98 self.lines_to_add[line_number] = [text] | |
99 | |
100 def Apply(self): | |
101 """Apply the patch by writing it to self.filename""" | |
102 # Read the lines of the existing file in to a list. | |
103 sourcefile = open(self.filename, "r") | |
104 file_lines = sourcefile.readlines() | |
105 sourcefile.close() | |
106 # Now apply the patch. Our strategy is to keep the array at the same size, | |
107 # and just edit strings in the file_lines list as necessary. When we delete | |
108 # lines, we just blank the line and keep it in the list. When we add lines, | |
109 # we just prepend the added source code to the start of the existing line at | |
110 # that line number. This way, all the line numbers we cached from calls to | |
111 # Add and Delete remain valid list indices, and we don't have to worry about | |
112 # maintaining any offsets. Each element of file_lines at the end may | |
113 # contain any number of lines (0 or more) delimited by carriage returns. | |
114 for linenum_to_delete in self.linenums_to_delete: | |
115 file_lines[linenum_to_delete] = ""; | |
116 for linenum, sourcelines in self.lines_to_add.items(): | |
117 # Sort the lines we're adding so we get relatively consistent results. | |
118 sourcelines.sort() | |
119 # Prepend the new lines. When we output | |
120 file_lines[linenum] = "".join(sourcelines) + file_lines[linenum] | |
121 newsource = open(self.filename, "w") | |
122 for line in file_lines: | |
123 newsource.write(line) | |
124 newsource.close() | |
125 | |
126 | |
127 def CheckAndInsert(typeinfo, typeinfo_map): | |
128 """Check if a TypeInfo exists already in the given map with the same name. If | |
129 so, make sure the size is consistent. | |
130 - If the name exists but the sizes do not match, print a message and | |
131 exit with non-zero exit code. | |
132 - If the name exists and the sizes match, do nothing. | |
133 - If the name does not exist, insert the typeinfo in to the map. | |
134 | |
135 """ | |
136 # If the type is unnamed, ignore it. | |
137 if typeinfo.name == "": | |
138 return | |
139 # If the size is 0, ignore it. | |
140 elif int(typeinfo.size) == 0: | |
141 return | |
142 # If the type is not defined under ppapi, ignore it. | |
143 elif typeinfo.source_location.filename.find("ppapi") == -1: | |
144 return | |
145 # If the type is defined under GLES2, ignore it. | |
146 elif typeinfo.source_location.filename.find("GLES2") > -1: | |
147 return | |
148 # If the type is an interface (by convention, starts with PPP_ or PPB_), | |
149 # ignore it. | |
150 elif (typeinfo.name[:4] == "PPP_") or (typeinfo.name[:4] == "PPB_"): | |
151 return | |
152 elif typeinfo.name in typeinfo_map: | |
153 if typeinfo.size != typeinfo_map[typeinfo.name].size: | |
154 print "Error: '" + typeinfo.name + "' is", \ | |
155 typeinfo_map[typeinfo.name].size, \ | |
156 "bytes on target '" + typeinfo_map[typeinfo.name].target + \ | |
157 "', but", typeinfo.size, "on target '" + typeinfo.target + "'" | |
158 print typeinfo_map[typeinfo.name].parsed_line | |
159 print typeinfo.parsed_line | |
160 sys.exit(1) | |
161 else: | |
162 # It's already in the map and the sizes match. | |
163 pass | |
164 else: | |
165 typeinfo_map[typeinfo.name] = typeinfo | |
166 | |
167 | |
168 def ProcessTarget(clang_command, target, arch_types, ind_types): | |
169 """Run clang using the given clang_command for the given target string. Parse | |
170 the output to create TypeInfos for each discovered type. Insert each type in | |
171 to the appropriate dictionary. For each type that has architecture-dependent | |
172 size, insert it in to arch_types. Types with independent size go in to | |
173 ind_types. If the type already exists in the appropriate map, make sure that | |
174 the size matches what's already in the map. If not, the script terminates | |
175 with an error message. | |
176 """ | |
177 p = subprocess.Popen(clang_command + " -triple " + target, | |
178 shell=True, | |
179 stdout=subprocess.PIPE) | |
180 lines = p.communicate()[0].split() | |
181 for line in lines: | |
182 typeinfo = TypeInfo(line, target) | |
183 # Put types which have architecture-specific size in to arch_types. All | |
184 # other types are 'architecture-independent' and get put in ind_types. | |
185 # in the appropraite dictionary. | |
186 if typeinfo.arch_dependent: | |
187 CheckAndInsert(typeinfo, arch_types) | |
188 else: | |
189 CheckAndInsert(typeinfo, ind_types) | |
190 | |
191 | |
192 def ToAssertionCode(typeinfo): | |
193 """Convert the TypeInfo to an appropriate C compile assertion. | |
194 If it's a struct (Record in Clang terminology), we want a line like this: | |
195 PP_COMPILE_ASSERT_STRUCT_SIZE_IN_BYTES(<name>, <size>);\n | |
196 Enums: | |
197 PP_COMPILE_ASSERT_ENUM_SIZE_IN_BYTES(<name>, <size>);\n | |
198 Typedefs: | |
199 PP_COMPILE_ASSERT_SIZE_IN_BYTES(<name>, <size>);\n | |
200 | |
201 """ | |
202 line = "PP_COMPILE_ASSERT_" | |
203 if typeinfo.kind == "Enum": | |
204 line += "ENUM_" | |
205 elif typeinfo.kind == "Record": | |
206 line += "STRUCT_" | |
207 line += "SIZE_IN_BYTES(" | |
208 line += typeinfo.name | |
209 line += ", " | |
210 line += typeinfo.size | |
211 line += ");\n" | |
212 return line | |
213 | |
214 | |
215 def IsMacroDefinedName(typename): | |
216 """Return true iff the given typename came from a PPAPI compile assertion.""" | |
217 return typename.find("PP_Dummy_Struct_For_") == 0 | |
218 | |
219 | |
220 COPYRIGHT_STRING_C = \ | |
221 """/* Copyright (c) 2010 The Chromium Authors. All rights reserved. | |
222 * Use of this source code is governed by a BSD-style license that can be | |
223 * found in the LICENSE file. | |
224 * | |
225 * This file has compile assertions for the sizes of types that are dependent | |
226 * on the architecture for which they are compiled (i.e., 32-bit vs 64-bit). | |
227 */ | |
228 | |
229 """ | |
230 | |
231 | |
232 def WriteArchSpecificCode(types, root, filename): | |
233 """Write a header file that contains a compile-time assertion for the size of | |
234 each of the given typeinfos, in to a file named filename rooted at root. | |
235 """ | |
236 assertion_lines = [ToAssertionCode(typeinfo) for typeinfo in types] | |
237 assertion_lines.sort() | |
238 outfile = open(os.path.join(root, filename), "w") | |
239 header_guard = "PPAPI_TESTS_" + filename.upper().replace(".", "_") + "_" | |
240 outfile.write(COPYRIGHT_STRING_C) | |
241 outfile.write('#ifndef ' + header_guard + '\n') | |
242 outfile.write('#define ' + header_guard + '\n\n') | |
243 outfile.write('#include "ppapi/tests/test_struct_sizes.c"\n\n') | |
244 for line in assertion_lines: | |
245 outfile.write(line) | |
246 outfile.write('\n#endif /* ' + header_guard + ' */\n') | |
247 | |
248 | |
249 def main(argv): | |
250 parser = optparse.OptionParser() | |
251 parser.add_option( | |
252 '-c', '--clang-path', dest='clang_path', | |
253 default=(''), | |
254 help='the path to the clang binary (default is to get it from your path)') | |
255 parser.add_option( | |
256 '-p', '--plugin', dest='plugin', | |
257 default='tests/clang/libPrintNamesAndSizes.so', | |
258 help='The path to the PrintNamesAndSizes plugin library.') | |
259 parser.add_option( | |
260 '--targets32', dest='targets32', | |
261 default='i386-pc-linux,arm-pc-linux,i386-pc-win32', | |
262 help='Which 32-bit target triples to provide to clang.') | |
263 parser.add_option( | |
264 '--targets64', dest='targets64', | |
265 default='x86_64-pc-linux,x86_64-pc-win', | |
266 help='Which 32-bit target triples to provide to clang.') | |
267 parser.add_option( | |
268 '-r', '--ppapi-root', dest='ppapi_root', | |
269 default='.', | |
270 help='The root directory of ppapi.') | |
271 options, args = parser.parse_args(argv) | |
272 if args: | |
273 parser.print_help() | |
274 print 'ERROR: invalid argument' | |
275 sys.exit(1) | |
276 | |
277 clang_executable = os.path.join(options.clang_path, 'clang') | |
278 clang_command = clang_executable + " -cc1" \ | |
279 + " -load " + options.plugin \ | |
280 + " -plugin PrintNamesAndSizes" \ | |
281 + " -I" + os.path.join(options.ppapi_root, "..") \ | |
282 + " " \ | |
283 + os.path.join(options.ppapi_root, "tests", "test_struct_sizes.c") | |
284 | |
285 # Dictionaries mapping type names to TypeInfo objects. | |
286 # Types that have size dependent on architecture, for 32-bit | |
287 types32 = {} | |
288 # Types that have size dependent on architecture, for 64-bit | |
289 types64 = {} | |
290 # Note that types32 and types64 should contain the same types, but with | |
291 # different sizes. | |
292 | |
293 # Types whose size should be consistent regardless of architecture. | |
294 types_independent = {} | |
295 | |
296 # Now run clang for each target. Along the way, make sure architecture- | |
297 # dependent types are consistent sizes on all 32-bit platforms and consistent | |
298 # on all 64-bit platforms. Any types in 'types_independent' are checked for | |
299 # all targets to make sure their size is consistent across all of them. | |
300 targets32 = options.targets32.split(','); | |
301 for target in targets32: | |
302 ProcessTarget(clang_command, target, types32, types_independent) | |
303 targets64 = options.targets64.split(','); | |
304 for target in targets64: | |
305 ProcessTarget(clang_command, target, types64, types_independent) | |
306 | |
307 # This dictionary maps file names to FilePatch objects. | |
308 file_patches = {} | |
309 | |
310 # Find locations of existing macros, and just delete them all. | |
311 for name, typeinfo in \ | |
312 types_independent.items() + types32.items() + types64.items(): | |
313 if IsMacroDefinedName(name): | |
314 sourcefile = typeinfo.source_location.filename | |
315 if sourcefile not in file_patches: | |
316 file_patches[sourcefile] = FilePatch(sourcefile) | |
317 file_patches[sourcefile].Delete(typeinfo.source_location.start_line, | |
318 typeinfo.source_location.end_line+1) | |
319 | |
320 # Add a compile-time assertion for each type whose size is independent of | |
321 # architecture. These assertions go immediately after the class definition. | |
322 for name, typeinfo in types_independent.items(): | |
323 # Ignore macros and types that are 0 bytes (i.e., typedefs to void) | |
324 if not IsMacroDefinedName(name) and typeinfo.size > 0: | |
325 sourcefile = typeinfo.source_location.filename | |
326 if sourcefile not in file_patches: | |
327 file_patches[sourcefile] = FilePatch(sourcefile) | |
328 # Add the assertion code just after the definition of the type. | |
329 file_patches[sourcefile].Add(ToAssertionCode(typeinfo), | |
330 typeinfo.source_location.end_line+1) | |
331 | |
332 for filename, patch in file_patches.items(): | |
333 patch.Apply() | |
334 | |
335 c_source_root = os.path.join(options.ppapi_root, "tests") | |
336 WriteArchSpecificCode(types32.values(), | |
337 c_source_root, | |
338 "arch_dependent_sizes_32.h") | |
339 WriteArchSpecificCode(types64.values(), | |
340 c_source_root, | |
341 "arch_dependent_sizes_64.h") | |
342 | |
343 return 0 | |
344 | |
345 | |
346 if __name__ == '__main__': | |
347 sys.exit(main(sys.argv[1:])) | |
348 | |
OLD | NEW |