build/toolchain/win/setup_toolchain.py

   1 # Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4 #
   5 # Copies the given "win tool" (which the toolchain uses to wrap compiler
   6 # invocations) and the environment blocks for the 32-bit and 64-bit builds on
   7 # Windows to the build directory.
   8 #
   9 # The arguments are the visual studio install location and the location of the
  10 # win tool. The script assumes that the root build directory is the current dir
  11 # and the files will be written to the current directory.
  12
  13 import errno
  14 import os
  15 import re
  16 import subprocess
  17 import sys
  18
  19
  20 def _ExtractImportantEnvironment(output_of_set):
  21   """Extracts environment variables required for the toolchain to run from
  22   a textual dump output by the cmd.exe 'set' command."""
  23   envvars_to_save = (
  24       'goma_.*', # TODO(scottmg): This is ugly, but needed for goma.
  25       'include',
  26       'lib',
  27       'libpath',
  28       'path',
  29       'pathext',
  30       'systemroot',
  31       'temp',
  32       'tmp',
  33       )
  34   env = {}
  35   for line in output_of_set.splitlines():
  36     for envvar in envvars_to_save:
  37       if re.match(envvar + '=', line.lower()):
  38         var, setting = line.split('=', 1)
  39         if envvar == 'path':
  40           # Our own rules (for running gyp-win-tool) and other actions in
  41           # Chromium rely on python being in the path. Add the path to this
  42           # python here so that if it's not in the path when ninja is run
  43           # later, python will still be found.
  44           setting = os.path.dirname(sys.executable) + os.pathsep + setting
  45         env[var.upper()] = setting
  46         break
  47   for required in ('SYSTEMROOT', 'TEMP', 'TMP'):
  48     if required not in env:
  49       raise Exception('Environment variable "%s" '
  50                       'required to be set to valid path' % required)
  51   return env
  52
  53
  54 def _SetupScript(target_cpu, sdk_dir):
  55   """Returns a command (with arguments) to be used to set up the
  56   environment."""
  57   # Check if we are running in the SDK command line environment and use
  58   # the setup script from the SDK if so. |target_cpu| should be either
  59   # 'x86' or 'x64'.
  60   assert target_cpu in ('x86', 'x64')
  61   if bool(int(os.environ.get('DEPOT_TOOLS_WIN_TOOLCHAIN', 1))) and sdk_dir:
  62     return [os.path.normpath(os.path.join(sdk_dir, 'Bin/SetEnv.Cmd')),
  63             '/' + target_cpu]
  64   else:
  65     # We only support x64-hosted tools.
  66     # TODO(scottmg|dpranke): Non-depot_tools toolchain: need to get Visual
  67     # Studio install location from registry.
  68     return [os.path.normpath(os.path.join(os.environ['GYP_MSVS_OVERRIDE_PATH'],
  69                                           'VC/vcvarsall.bat')),
  70             'amd64_x86' if target_cpu == 'x86' else 'amd64']
  71
  72
  73 def _FormatAsEnvironmentBlock(envvar_dict):
  74   """Format as an 'environment block' directly suitable for CreateProcess.
  75   Briefly this is a list of key=value\0, terminated by an additional \0. See
  76   CreateProcess documentation for more details."""
  77   block = ''
  78   nul = '\0'
  79   for key, value in envvar_dict.iteritems():
  80     block += key + '=' + value + nul
  81   block += nul
  82   return block
  83
  84
  85 def _CopyTool(source_path):
  86   """Copies the given tool to the current directory, including a warning not
  87   to edit it."""
  88   with open(source_path) as source_file:
  89     tool_source = source_file.readlines()
  90
  91   # Add header and write it out to the current directory (which should be the
  92   # root build dir).
  93   with open("gyp-win-tool", 'w') as tool_file:
  94     tool_file.write(''.join([tool_source[0],
  95                              '# Generated by setup_toolchain.py do not edit.\n']
  96                             + tool_source[1:]))
  97
  98
  99 def main():
 100   if len(sys.argv) != 6:
 101     print('Usage setup_toolchain.py '
 102           '<visual studio path> <win tool path> <win sdk path> '
 103           '<runtime dirs> <target_cpu>')
 104     sys.exit(2)
 105   tool_source = sys.argv[2]
 106   win_sdk_path = sys.argv[3]
 107   runtime_dirs = sys.argv[4]
 108   target_cpu = sys.argv[5]
 109
 110   _CopyTool(tool_source)
 111
 112   cpus = ('x86', 'x64')
 113   assert target_cpu in cpus
 114   vc_bin_dir = ''
 115
 116   # TODO(scottmg|goma): Do we need an equivalent of
 117   # ninja_use_custom_environment_files?
 118
 119   for cpu in cpus:
 120     # Extract environment variables for subprocesses.
 121     args = _SetupScript(cpu, win_sdk_path)
 122     args.extend(('&&', 'set'))
 123     popen = subprocess.Popen(
 124         args, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
 125     variables, _ = popen.communicate()
 126     env = _ExtractImportantEnvironment(variables)
 127     env['PATH'] = runtime_dirs + ';' + env['PATH']
 128
 129     if cpu == target_cpu:
 130       for path in env['PATH'].split(os.pathsep):
 131         if os.path.exists(os.path.join(path, 'cl.exe')):
 132           vc_bin_dir = os.path.realpath(path)
 133           break
 134
 135     # The Windows SDK include directories must be first. They both have a sal.h,
 136     # and the SDK one is newer and the SDK uses some newer features from it not
 137     # present in the Visual Studio one.
 138
 139     if win_sdk_path:
 140       additional_includes = ('{sdk_dir}\\Include\\shared;' +
 141                              '{sdk_dir}\\Include\\um;' +
 142                              '{sdk_dir}\\Include\\winrt;').format(
 143                                   sdk_dir=win_sdk_path)
 144       env['INCLUDE'] = additional_includes + env['INCLUDE']
 145     env_block = _FormatAsEnvironmentBlock(env)
 146     with open('environment.' + cpu, 'wb') as f:
 147       f.write(env_block)
 148
 149   assert vc_bin_dir
 150   print 'vc_bin_dir = "%s"' % vc_bin_dir
 151
 152
 153 if __name__ == '__main__':
 154   main()