| Index: build/toolchain/win/setup_toolchain.py
|
| diff --git a/build/toolchain/win/setup_toolchain.py b/build/toolchain/win/setup_toolchain.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..bc9bd1e7b43bf6fe65f5c34f36ac37aa71442c25
|
| --- /dev/null
|
| +++ b/build/toolchain/win/setup_toolchain.py
|
| @@ -0,0 +1,154 @@
|
| +# Copyright (c) 2013 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.
|
| +#
|
| +# Copies the given "win tool" (which the toolchain uses to wrap compiler
|
| +# invocations) and the environment blocks for the 32-bit and 64-bit builds on
|
| +# Windows to the build directory.
|
| +#
|
| +# The arguments are the visual studio install location and the location of the
|
| +# win tool. The script assumes that the root build directory is the current dir
|
| +# and the files will be written to the current directory.
|
| +
|
| +import errno
|
| +import os
|
| +import re
|
| +import subprocess
|
| +import sys
|
| +
|
| +
|
| +def _ExtractImportantEnvironment(output_of_set):
|
| + """Extracts environment variables required for the toolchain to run from
|
| + a textual dump output by the cmd.exe 'set' command."""
|
| + envvars_to_save = (
|
| + 'goma_.*', # TODO(scottmg): This is ugly, but needed for goma.
|
| + 'include',
|
| + 'lib',
|
| + 'libpath',
|
| + 'path',
|
| + 'pathext',
|
| + 'systemroot',
|
| + 'temp',
|
| + 'tmp',
|
| + )
|
| + env = {}
|
| + for line in output_of_set.splitlines():
|
| + for envvar in envvars_to_save:
|
| + if re.match(envvar + '=', line.lower()):
|
| + var, setting = line.split('=', 1)
|
| + if envvar == 'path':
|
| + # Our own rules (for running gyp-win-tool) and other actions in
|
| + # Chromium rely on python being in the path. Add the path to this
|
| + # python here so that if it's not in the path when ninja is run
|
| + # later, python will still be found.
|
| + setting = os.path.dirname(sys.executable) + os.pathsep + setting
|
| + env[var.upper()] = setting
|
| + break
|
| + for required in ('SYSTEMROOT', 'TEMP', 'TMP'):
|
| + if required not in env:
|
| + raise Exception('Environment variable "%s" '
|
| + 'required to be set to valid path' % required)
|
| + return env
|
| +
|
| +
|
| +def _SetupScript(target_cpu, sdk_dir):
|
| + """Returns a command (with arguments) to be used to set up the
|
| + environment."""
|
| + # Check if we are running in the SDK command line environment and use
|
| + # the setup script from the SDK if so. |target_cpu| should be either
|
| + # 'x86' or 'x64'.
|
| + assert target_cpu in ('x86', 'x64')
|
| + if bool(int(os.environ.get('DEPOT_TOOLS_WIN_TOOLCHAIN', 1))) and sdk_dir:
|
| + return [os.path.normpath(os.path.join(sdk_dir, 'Bin/SetEnv.Cmd')),
|
| + '/' + target_cpu]
|
| + else:
|
| + # We only support x64-hosted tools.
|
| + # TODO(scottmg|dpranke): Non-depot_tools toolchain: need to get Visual
|
| + # Studio install location from registry.
|
| + return [os.path.normpath(os.path.join(os.environ['GYP_MSVS_OVERRIDE_PATH'],
|
| + 'VC/vcvarsall.bat')),
|
| + 'amd64_x86' if target_cpu == 'x86' else 'amd64']
|
| +
|
| +
|
| +def _FormatAsEnvironmentBlock(envvar_dict):
|
| + """Format as an 'environment block' directly suitable for CreateProcess.
|
| + Briefly this is a list of key=value\0, terminated by an additional \0. See
|
| + CreateProcess documentation for more details."""
|
| + block = ''
|
| + nul = '\0'
|
| + for key, value in envvar_dict.iteritems():
|
| + block += key + '=' + value + nul
|
| + block += nul
|
| + return block
|
| +
|
| +
|
| +def _CopyTool(source_path):
|
| + """Copies the given tool to the current directory, including a warning not
|
| + to edit it."""
|
| + with open(source_path) as source_file:
|
| + tool_source = source_file.readlines()
|
| +
|
| + # Add header and write it out to the current directory (which should be the
|
| + # root build dir).
|
| + with open("gyp-win-tool", 'w') as tool_file:
|
| + tool_file.write(''.join([tool_source[0],
|
| + '# Generated by setup_toolchain.py do not edit.\n']
|
| + + tool_source[1:]))
|
| +
|
| +
|
| +def main():
|
| + if len(sys.argv) != 6:
|
| + print('Usage setup_toolchain.py '
|
| + '<visual studio path> <win tool path> <win sdk path> '
|
| + '<runtime dirs> <target_cpu>')
|
| + sys.exit(2)
|
| + tool_source = sys.argv[2]
|
| + win_sdk_path = sys.argv[3]
|
| + runtime_dirs = sys.argv[4]
|
| + target_cpu = sys.argv[5]
|
| +
|
| + _CopyTool(tool_source)
|
| +
|
| + cpus = ('x86', 'x64')
|
| + assert target_cpu in cpus
|
| + vc_bin_dir = ''
|
| +
|
| + # TODO(scottmg|goma): Do we need an equivalent of
|
| + # ninja_use_custom_environment_files?
|
| +
|
| + for cpu in cpus:
|
| + # Extract environment variables for subprocesses.
|
| + args = _SetupScript(cpu, win_sdk_path)
|
| + args.extend(('&&', 'set'))
|
| + popen = subprocess.Popen(
|
| + args, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
| + variables, _ = popen.communicate()
|
| + env = _ExtractImportantEnvironment(variables)
|
| + env['PATH'] = runtime_dirs + ';' + env['PATH']
|
| +
|
| + if cpu == target_cpu:
|
| + for path in env['PATH'].split(os.pathsep):
|
| + if os.path.exists(os.path.join(path, 'cl.exe')):
|
| + vc_bin_dir = os.path.realpath(path)
|
| + break
|
| +
|
| + # The Windows SDK include directories must be first. They both have a sal.h,
|
| + # and the SDK one is newer and the SDK uses some newer features from it not
|
| + # present in the Visual Studio one.
|
| +
|
| + if win_sdk_path:
|
| + additional_includes = ('{sdk_dir}\\Include\\shared;' +
|
| + '{sdk_dir}\\Include\\um;' +
|
| + '{sdk_dir}\\Include\\winrt;').format(
|
| + sdk_dir=win_sdk_path)
|
| + env['INCLUDE'] = additional_includes + env['INCLUDE']
|
| + env_block = _FormatAsEnvironmentBlock(env)
|
| + with open('environment.' + cpu, 'wb') as f:
|
| + f.write(env_block)
|
| +
|
| + assert vc_bin_dir
|
| + print 'vc_bin_dir = "%s"' % vc_bin_dir
|
| +
|
| +
|
| +if __name__ == '__main__':
|
| + main()
|
|
|