OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
3 # Use of this source code is governed by a BSD-style license that can be | |
4 # found in the LICENSE file. | |
5 | |
6 """A utility script to help building Syzygy-reordered Chrome binaries.""" | |
7 | |
8 import logging | |
9 import optparse | |
10 import os | |
11 import subprocess | |
12 import sys | |
13 | |
14 | |
15 # The default relink executable to use to reorder binaries. | |
16 _DEFAULT_RELINKER = os.path.join( | |
17 os.path.join(os.path.dirname(__file__), '../../../../..'), | |
18 'third_party/syzygy/binaries/exe/relink.exe') | |
19 | |
20 _LOGGER = logging.getLogger() | |
21 | |
22 # We use the same seed for all random reorderings to get a deterministic build. | |
23 _RANDOM_SEED = 1347344 | |
24 | |
25 | |
26 def _Shell(*cmd, **kw): | |
27 """Shells out to "cmd". Returns a tuple of cmd's stdout, stderr.""" | |
28 _LOGGER.info('Running command "%s".', cmd) | |
29 prog = subprocess.Popen(cmd, **kw) | |
30 | |
31 stdout, stderr = prog.communicate() | |
32 if prog.returncode != 0: | |
33 raise RuntimeError('Command "%s" returned %d.' % (cmd, prog.returncode)) | |
34 | |
35 return stdout, stderr | |
36 | |
37 | |
38 def _ReorderBinary(relink_exe, executable, symbol, destination_dir): | |
39 """Reorders the executable found in input_dir, and writes the resultant | |
40 reordered executable and symbol files to destination_dir. | |
41 | |
42 If a file named <executable>-order.json exists, imposes that order on the | |
43 output binaries, otherwise orders them randomly. | |
44 """ | |
45 cmd = [relink_exe, | |
46 '--overwrite', | |
47 '--input-image=%s' % executable, | |
48 '--input-pdb=%s' % symbol, | |
49 '--output-image=%s' % os.path.abspath( | |
50 os.path.join(destination_dir, os.path.basename(executable))), | |
51 '--output-pdb=%s' % os.path.abspath( | |
52 os.path.join(destination_dir, os.path.basename(symbol))),] | |
53 | |
54 # Check whether there's an order file available for the executable. | |
55 order_file = '%s-order.json' % executable | |
56 if os.path.exists(order_file): | |
57 # The ordering file exists, let's use that. | |
58 _LOGGER.info('Reordering "%s" according to "%s".', | |
59 os.path.basename(executable), | |
60 os.path.basename(order_file)) | |
61 cmd.append('--order-file=%s' % order_file) | |
62 else: | |
63 # No ordering file, we randomize the output. | |
64 _LOGGER.info('Randomly reordering "%s"', executable) | |
65 cmd.append('--seed=%d' % _RANDOM_SEED) | |
66 | |
67 return _Shell(*cmd) | |
68 | |
69 | |
70 def main(options): | |
71 logging.basicConfig(level=logging.INFO) | |
72 | |
73 # Make sure the destination directory exists. | |
74 if not os.path.isdir(options.destination_dir): | |
75 _LOGGER.info('Creating destination directory "%s".', | |
76 options.destination_dir) | |
77 os.makedirs(options.destination_dir) | |
78 | |
79 # Reorder the binaries into the destination directory. | |
80 _ReorderBinary(options.relinker, | |
81 options.input_executable, | |
82 options.input_symbol, | |
83 options.destination_dir) | |
84 | |
85 | |
86 def _ParseOptions(): | |
87 option_parser = optparse.OptionParser() | |
88 option_parser.add_option('--input_executable', | |
89 help='The path to the input executable.') | |
90 option_parser.add_option('--input_symbol', | |
91 help='The path to the input symbol file.') | |
92 option_parser.add_option('--relinker', default=_DEFAULT_RELINKER, | |
93 help='Relinker exectuable to use, defaults to "%s"' % _DEFAULT_RELINKER) | |
94 option_parser.add_option('-d', '--destination_dir', | |
95 help='Destination directory for reordered files, defaults to ' | |
96 'the subdirectory "reordered" in the output_dir.') | |
97 options, args = option_parser.parse_args() | |
98 | |
99 if not options.input_executable: | |
100 option_parser.error('You must provide an input executable.') | |
101 if not options.input_symbol: | |
102 option_parser.error('You must provide an input symbol file.') | |
103 | |
104 if not options.destination_dir: | |
105 options.destination_dir = os.path.join(options.output_dir, 'reordered') | |
106 | |
107 return options | |
108 | |
109 | |
110 if '__main__' == __name__: | |
111 sys.exit(main(_ParseOptions())) | |
OLD | NEW |