Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(188)

Side by Side Diff: build/toolchain/win/setup_toolchain.py

Issue 2101243005: Add a snapshot of flutter/engine/src/build to our sdk (Closed) Base URL: git@github.com:dart-lang/sdk.git@master
Patch Set: add README.dart Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « build/toolchain/win/midl.gni ('k') | build/tree_truth.sh » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(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()
OLDNEW
« no previous file with comments | « build/toolchain/win/midl.gni ('k') | build/tree_truth.sh » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698