OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2014 The Chromium Authors. All rights reserved. | 2 # Copyright 2014 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 import glob | 6 import glob |
7 import optparse | 7 import optparse |
8 import os | 8 import os |
9 import shutil | 9 import shutil |
10 import subprocess | 10 import subprocess |
11 import sys | 11 import sys |
12 | 12 |
13 def reorder_imports(input_dir, output_dir, architecture): | 13 def reorder_imports(input_dir, output_dir, architecture): |
14 """Run swapimports.exe on the initial chrome.exe, and write to the output | 14 """Run swapimports.exe on the initial chrome.exe, and write to the output |
15 directory. Also copy over any related files that might be needed | 15 directory. Also copy over any related files that might be needed |
16 (pdbs, manifests etc.). | 16 (pdbs, manifests etc.). |
17 """ | 17 """ |
18 | 18 |
19 input_image = os.path.join(input_dir, 'chrome.exe') | 19 input_image = os.path.join(input_dir, 'chrome.exe') |
20 output_image = os.path.join(output_dir, 'chrome.exe') | 20 output_image = os.path.join(output_dir, 'chrome.exe') |
21 | 21 |
22 # TODO(caitkp): Remove this once swapimport works on x64 builds. | 22 swap_exe = os.path.join( |
| 23 __file__, |
| 24 '..\\..\\..\\third_party\\syzygy\\binaries\\exe\\swapimport.exe') |
| 25 |
| 26 args = [swap_exe, '--input-image=%s' % input_image, |
| 27 '--output-image=%s' % output_image, '--overwrite'] |
| 28 |
23 if architecture == 'x64': | 29 if architecture == 'x64': |
24 shutil.copy(input_image, output_image) | 30 args.append('--x64'); |
25 else: | 31 |
26 swap_exe = os.path.join( | 32 args.append('chrome_elf.dll'); |
27 __file__, | 33 |
28 '..\\..\\..\\third_party\\syzygy\\binaries\\exe\\swapimport.exe') | 34 subprocess.call(args) |
29 subprocess.call( | |
30 [swap_exe, '--input-image=%s' % input_image, | |
31 '--output-image=%s' % output_image, '--overwrite', 'chrome_elf.dll']) | |
32 | 35 |
33 for fname in glob.iglob(os.path.join(input_dir, 'chrome.exe.*')): | 36 for fname in glob.iglob(os.path.join(input_dir, 'chrome.exe.*')): |
34 shutil.copy(fname, os.path.join(output_dir, os.path.basename(fname))) | 37 shutil.copy(fname, os.path.join(output_dir, os.path.basename(fname))) |
35 return 0 | 38 return 0 |
36 | 39 |
37 | 40 |
38 def main(argv): | 41 def main(argv): |
39 usage = 'reorder_imports.py -i <input_dir> -o <output_dir> -a <target_arch>' | 42 usage = 'reorder_imports.py -i <input_dir> -o <output_dir> -a <target_arch>' |
40 parser = optparse.OptionParser(usage=usage) | 43 parser = optparse.OptionParser(usage=usage) |
41 parser.add_option('-i', '--input', help='reorder chrome.exe in DIR', | 44 parser.add_option('-i', '--input', help='reorder chrome.exe in DIR', |
42 metavar='DIR') | 45 metavar='DIR') |
43 parser.add_option('-o', '--output', help='write new chrome.exe to DIR', | 46 parser.add_option('-o', '--output', help='write new chrome.exe to DIR', |
44 metavar='DIR') | 47 metavar='DIR') |
45 parser.add_option('-a', '--arch', help='architecture of build (optional)', | 48 parser.add_option('-a', '--arch', help='architecture of build (optional)', |
46 default='ia32') | 49 default='ia32') |
47 opts, args = parser.parse_args() | 50 opts, args = parser.parse_args() |
48 | 51 |
49 if not opts.input or not opts.output: | 52 if not opts.input or not opts.output: |
50 parser.error('Please provide and input and output directory') | 53 parser.error('Please provide and input and output directory') |
51 return reorder_imports(opts.input, opts.output, opts.arch) | 54 return reorder_imports(opts.input, opts.output, opts.arch) |
52 | 55 |
53 if __name__ == "__main__": | 56 if __name__ == "__main__": |
54 sys.exit(main(sys.argv[1:])) | 57 sys.exit(main(sys.argv[1:])) |
OLD | NEW |