OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2014 Google Inc. All rights reserved. | 2 # Copyright (c) 2014 Google Inc. All rights reserved. |
3 # | 3 # |
4 # Redistribution and use in source and binary forms, with or without | 4 # Redistribution and use in source and binary forms, with or without |
5 # modification, are permitted provided that the following conditions are | 5 # modification, are permitted provided that the following conditions are |
6 # met: | 6 # met: |
7 # | 7 # |
8 # * Redistributions of source code must retain the above copyright | 8 # * Redistributions of source code must retain the above copyright |
9 # notice, this list of conditions and the following disclaimer. | 9 # notice, this list of conditions and the following disclaimer. |
10 # * Redistributions in binary form must reproduce the above | 10 # * Redistributions in binary form must reproduce the above |
(...skipping 19 matching lines...) Expand all Loading... |
30 import re | 30 import re |
31 import sys | 31 import sys |
32 import os | 32 import os |
33 | 33 |
34 | 34 |
35 def validate_injected_script(fileName): | 35 def validate_injected_script(fileName): |
36 f = open(fileName, "r") | 36 f = open(fileName, "r") |
37 lines = f.readlines() | 37 lines = f.readlines() |
38 f.close() | 38 f.close() |
39 | 39 |
40 array_proto_functions = "|".join(["concat", "every", "filter", "forEach", "i
ndexOf", "join", "lastIndexOf", "map", "pop", "push", "reduce", "reduceRight", "
reverse", "shift", "slice", "some", "sort", "splice", "toLocaleString", "toStrin
g", "unshift"]) | 40 proto_functions = "|".join([ |
| 41 # Array.prototype.* |
| 42 "concat", "every", "filter", "forEach", "indexOf", "join", "lastIndexOf"
, "map", "pop", |
| 43 "push", "reduce", "reduceRight", "reverse", "shift", "slice", "some", "s
ort", "splice", "toLocaleString", "toString", "unshift", |
| 44 # Function.prototype.* |
| 45 "apply", "bind", "call", "isGenerator", "toSource", |
| 46 ]) |
41 | 47 |
42 # Black list: | 48 # Black list: |
43 # - Function.prototype.bind() | |
44 # - Object.prototype.toString() | 49 # - Object.prototype.toString() |
45 # - Array.prototype.* | 50 # - Array.prototype.* |
| 51 # - Function.prototype.* |
46 # - Math.* | 52 # - Math.* |
47 black_list_call_regex = re.compile(r"\bMath\.\w+\(|\.(bind|toString|" + arra
y_proto_functions + r")\(") | 53 black_list_call_regex = re.compile(r"\bMath\.\w+\(|\.(toString|" + proto_fun
ctions + r")\(") |
48 | 54 |
49 errors_found = False | 55 errors_found = False |
50 for i, line in enumerate(lines): | 56 for i, line in enumerate(lines): |
51 for match in re.finditer(black_list_call_regex, line): | 57 for match in re.finditer(black_list_call_regex, line): |
52 errors_found = True | 58 errors_found = True |
53 print "ERROR: Black list function call in %s at line %02d column %02
d: %s" % (os.path.basename(fileName), i + 1, match.start(), match.group(0)) | 59 print "ERROR: Black list function call in %s at line %02d column %02
d: %s" % (os.path.basename(fileName), i + 1, match.start(), match.group(0)) |
54 | 60 |
55 if not errors_found: | 61 if not errors_found: |
56 print "OK" | 62 print "OK" |
57 | 63 |
58 | 64 |
59 def main(argv): | 65 def main(argv): |
60 if len(argv) < 2: | 66 if len(argv) < 2: |
61 print('ERROR: Usage: %s path/to/InjectedScriptSource.js' % argv[0]) | 67 print('ERROR: Usage: %s path/to/InjectedScriptSource.js' % argv[0]) |
62 return 1 | 68 return 1 |
63 | 69 |
64 validate_injected_script(argv[1]) | 70 validate_injected_script(argv[1]) |
65 | 71 |
66 if __name__ == '__main__': | 72 if __name__ == '__main__': |
67 sys.exit(main(sys.argv)) | 73 sys.exit(main(sys.argv)) |
OLD | NEW |