| Index: chrome/tools/build/win/syzygy/reorder.py
|
| diff --git a/chrome/tools/build/win/syzygy/reorder.py b/chrome/tools/build/win/syzygy/reorder.py
|
| new file mode 100755
|
| index 0000000000000000000000000000000000000000..8cdd6d7ce6e572b8a6575cc4f22d9378c22ce1a0
|
| --- /dev/null
|
| +++ b/chrome/tools/build/win/syzygy/reorder.py
|
| @@ -0,0 +1,111 @@
|
| +#!/usr/bin/env python
|
| +# Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +"""A utility script to help building Syzygy-reordered Chrome binaries."""
|
| +
|
| +import logging
|
| +import optparse
|
| +import os
|
| +import subprocess
|
| +import sys
|
| +
|
| +
|
| +# The default relink executable to use to reorder binaries.
|
| +_DEFAULT_RELINKER = os.path.join(
|
| + os.path.join(os.path.dirname(__file__), '../../../../..'),
|
| + 'third_party/syzygy/binaries/exe/relink.exe')
|
| +
|
| +_LOGGER = logging.getLogger()
|
| +
|
| +# We use the same seed for all random reorderings to get a deterministic build.
|
| +_RANDOM_SEED = 1347344
|
| +
|
| +
|
| +def _Shell(*cmd, **kw):
|
| + """Shells out to "cmd". Returns a tuple of cmd's stdout, stderr."""
|
| + _LOGGER.info('Running command "%s".', cmd)
|
| + prog = subprocess.Popen(cmd, **kw)
|
| +
|
| + stdout, stderr = prog.communicate()
|
| + if prog.returncode != 0:
|
| + raise RuntimeError('Command "%s" returned %d.' % (cmd, prog.returncode))
|
| +
|
| + return stdout, stderr
|
| +
|
| +
|
| +def _ReorderBinary(relink_exe, executable, symbol, destination_dir):
|
| + """Reorders the executable found in input_dir, and writes the resultant
|
| + reordered executable and symbol files to destination_dir.
|
| +
|
| + If a file named <executable>-order.json exists, imposes that order on the
|
| + output binaries, otherwise orders them randomly.
|
| + """
|
| + cmd = [relink_exe,
|
| + '--overwrite',
|
| + '--input-image=%s' % executable,
|
| + '--input-pdb=%s' % symbol,
|
| + '--output-image=%s' % os.path.abspath(
|
| + os.path.join(destination_dir, os.path.basename(executable))),
|
| + '--output-pdb=%s' % os.path.abspath(
|
| + os.path.join(destination_dir, os.path.basename(symbol))),]
|
| +
|
| + # Check whether there's an order file available for the executable.
|
| + order_file = '%s-order.json' % executable
|
| + if os.path.exists(order_file):
|
| + # The ordering file exists, let's use that.
|
| + _LOGGER.info('Reordering "%s" according to "%s".',
|
| + os.path.basename(executable),
|
| + os.path.basename(order_file))
|
| + cmd.append('--order-file=%s' % order_file)
|
| + else:
|
| + # No ordering file, we randomize the output.
|
| + _LOGGER.info('Randomly reordering "%s"', executable)
|
| + cmd.append('--seed=%d' % _RANDOM_SEED)
|
| +
|
| + return _Shell(*cmd)
|
| +
|
| +
|
| +def main(options):
|
| + logging.basicConfig(level=logging.INFO)
|
| +
|
| + # Make sure the destination directory exists.
|
| + if not os.path.isdir(options.destination_dir):
|
| + _LOGGER.info('Creating destination directory "%s".',
|
| + options.destination_dir)
|
| + os.makedirs(options.destination_dir)
|
| +
|
| + # Reorder the binaries into the destination directory.
|
| + _ReorderBinary(options.relinker,
|
| + options.input_executable,
|
| + options.input_symbol,
|
| + options.destination_dir)
|
| +
|
| +
|
| +def _ParseOptions():
|
| + option_parser = optparse.OptionParser()
|
| + option_parser.add_option('--input_executable',
|
| + help='The path to the input executable.')
|
| + option_parser.add_option('--input_symbol',
|
| + help='The path to the input symbol file.')
|
| + option_parser.add_option('--relinker', default=_DEFAULT_RELINKER,
|
| + help='Relinker exectuable to use, defaults to "%s"' % _DEFAULT_RELINKER)
|
| + option_parser.add_option('-d', '--destination_dir',
|
| + help='Destination directory for reordered files, defaults to '
|
| + 'the subdirectory "reordered" in the output_dir.')
|
| + options, args = option_parser.parse_args()
|
| +
|
| + if not options.input_executable:
|
| + option_parser.error('You must provide an input executable.')
|
| + if not options.input_symbol:
|
| + option_parser.error('You must provide an input symbol file.')
|
| +
|
| + if not options.destination_dir:
|
| + options.destination_dir = os.path.join(options.output_dir, 'reordered')
|
| +
|
| + return options
|
| +
|
| +
|
| +if '__main__' == __name__:
|
| + sys.exit(main(_ParseOptions()))
|
|
|