OLD | NEW |
(Empty) | |
| 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() |
OLD | NEW |