OLD | NEW |
---|---|
(Empty) | |
1 #!/usr/bin/env python | |
scottmg
2015/10/30 16:29:41
We should really move most of this file to depot_t
Michael Achenbach
2015/10/31 18:37:15
How about to buildtools for a start?
| |
2 # Copyright 2015 the V8 project authors. All rights reserved. | |
3 # Copyright 2014 The Chromium Authors. All rights reserved. | |
4 # Use of this source code is governed by a BSD-style license that can be | |
5 # found in the LICENSE file. | |
6 | |
7 import json | |
8 import os | |
9 import pipes | |
10 import shutil | |
11 import subprocess | |
12 import sys | |
13 import vs_toolchain | |
14 | |
15 | |
16 script_dir = os.path.dirname(os.path.realpath(__file__)) | |
17 chrome_src = os.path.abspath(os.path.join(script_dir, os.pardir)) | |
18 SRC_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | |
19 sys.path.insert(1, os.path.join(chrome_src, 'tools')) | |
20 sys.path.insert(0, os.path.join(chrome_src, 'build', 'gyp', 'pylib')) | |
21 json_data_file = os.path.join(script_dir, 'win_toolchain.json') | |
22 | |
23 | |
24 import gyp | |
25 | |
26 | |
27 def SetEnvironmentAndGetRuntimeDllDirs(): | |
28 """Sets up os.environ to use the depot_tools VS toolchain with gyp, and | |
29 returns the location of the VS runtime DLLs so they can be copied into | |
30 the output directory after gyp generation. | |
31 """ | |
32 vs2013_runtime_dll_dirs = None | |
33 depot_tools_win_toolchain = \ | |
34 bool(int(os.environ.get('DEPOT_TOOLS_WIN_TOOLCHAIN', '1'))) | |
35 # When running on a non-Windows host, only do this if the SDK has explicitly | |
36 # been downloaded before (in which case json_data_file will exist). | |
37 if ((sys.platform in ('win32', 'cygwin') or os.path.exists(json_data_file)) | |
38 and depot_tools_win_toolchain): | |
39 if not os.path.exists(json_data_file): | |
40 Update() | |
41 with open(json_data_file, 'r') as tempf: | |
42 toolchain_data = json.load(tempf) | |
43 | |
44 toolchain = toolchain_data['path'] | |
45 version = toolchain_data['version'] | |
46 win_sdk = toolchain_data.get('win_sdk') | |
47 if not win_sdk: | |
48 win_sdk = toolchain_data['win8sdk'] | |
49 wdk = toolchain_data['wdk'] | |
50 # TODO(scottmg): The order unfortunately matters in these. They should be | |
51 # split into separate keys for x86 and x64. (See CopyVsRuntimeDlls call | |
52 # below). http://crbug.com/345992 | |
53 vs2013_runtime_dll_dirs = toolchain_data['runtime_dirs'] | |
54 | |
55 os.environ['GYP_MSVS_OVERRIDE_PATH'] = toolchain | |
56 os.environ['GYP_MSVS_VERSION'] = version | |
57 # We need to make sure windows_sdk_path is set to the automated | |
58 # toolchain values in GYP_DEFINES, but don't want to override any | |
59 # otheroptions.express | |
60 # values there. | |
61 gyp_defines_dict = gyp.NameValueListToDict(gyp.ShlexEnv('GYP_DEFINES')) | |
62 gyp_defines_dict['windows_sdk_path'] = win_sdk | |
63 os.environ['GYP_DEFINES'] = ' '.join('%s=%s' % (k, pipes.quote(str(v))) | |
64 for k, v in gyp_defines_dict.iteritems()) | |
65 os.environ['WINDOWSSDKDIR'] = win_sdk | |
66 os.environ['WDK_DIR'] = wdk | |
67 # Include the VS runtime in the PATH in case it's not machine-installed. | |
68 runtime_path = ';'.join(vs2013_runtime_dll_dirs) | |
69 os.environ['PATH'] = runtime_path + ';' + os.environ['PATH'] | |
70 return vs2013_runtime_dll_dirs | |
71 | |
72 | |
73 def _VersionNumber(): | |
74 """Gets the standard version number ('120', '140', etc.) based on | |
75 GYP_MSVS_VERSION.""" | |
76 if os.environ['GYP_MSVS_VERSION'] == '2013': | |
77 return '120' | |
78 elif os.environ['GYP_MSVS_VERSION'] == '2015': | |
79 return '140' | |
80 else: | |
81 raise ValueError('Unexpected GYP_MSVS_VERSION') | |
82 | |
83 | |
84 def _CopyRuntimeImpl(target, source): | |
85 """Copy |source| to |target| if it doesn't already exist or if it | |
86 needs to be updated. | |
87 """ | |
88 if (os.path.isdir(os.path.dirname(target)) and | |
89 (not os.path.isfile(target) or | |
90 os.stat(target).st_mtime != os.stat(source).st_mtime)): | |
91 print 'Copying %s to %s...' % (source, target) | |
92 if os.path.exists(target): | |
93 os.unlink(target) | |
94 shutil.copy2(source, target) | |
95 | |
96 | |
97 def _CopyRuntime2013(target_dir, source_dir, dll_pattern): | |
98 """Copy both the msvcr and msvcp runtime DLLs, only if the target doesn't | |
99 exist, but the target directory does exist.""" | |
100 for file_part in ('p', 'r'): | |
101 dll = dll_pattern % file_part | |
102 target = os.path.join(target_dir, dll) | |
103 source = os.path.join(source_dir, dll) | |
104 _CopyRuntimeImpl(target, source) | |
105 | |
106 | |
107 def _CopyRuntime2015(target_dir, source_dir, dll_pattern): | |
108 """Copy both the msvcp and vccorlib runtime DLLs, only if the target doesn't | |
109 exist, but the target directory does exist.""" | |
110 for file_part in ('msvcp', 'vccorlib'): | |
111 dll = dll_pattern % file_part | |
112 target = os.path.join(target_dir, dll) | |
113 source = os.path.join(source_dir, dll) | |
114 _CopyRuntimeImpl(target, source) | |
115 | |
116 | |
117 def CopyVsRuntimeDlls(output_dir, runtime_dirs): | |
118 """Copies the VS runtime DLLs from the given |runtime_dirs| to the output | |
119 directory so that even if not system-installed, built binaries are likely to | |
120 be able to run. | |
121 | |
122 This needs to be run after gyp has been run so that the expected target | |
123 output directories are already created. | |
124 """ | |
125 x86, x64 = runtime_dirs | |
126 out_debug = os.path.join(output_dir, 'Debug') | |
127 out_debug_nacl64 = os.path.join(output_dir, 'Debug', 'x64') | |
128 out_release = os.path.join(output_dir, 'Release') | |
129 out_release_nacl64 = os.path.join(output_dir, 'Release', 'x64') | |
130 out_debug_x64 = os.path.join(output_dir, 'Debug_x64') | |
131 out_release_x64 = os.path.join(output_dir, 'Release_x64') | |
132 | |
133 if os.path.exists(out_debug) and not os.path.exists(out_debug_nacl64): | |
134 os.makedirs(out_debug_nacl64) | |
135 if os.path.exists(out_release) and not os.path.exists(out_release_nacl64): | |
136 os.makedirs(out_release_nacl64) | |
137 if os.environ.get('GYP_MSVS_VERSION') == '2015': | |
138 _CopyRuntime2015(out_debug, x86, '%s140d.dll') | |
139 _CopyRuntime2015(out_release, x86, '%s140.dll') | |
140 _CopyRuntime2015(out_debug_x64, x64, '%s140d.dll') | |
141 _CopyRuntime2015(out_release_x64, x64, '%s140.dll') | |
142 _CopyRuntime2015(out_debug_nacl64, x64, '%s140d.dll') | |
143 _CopyRuntime2015(out_release_nacl64, x64, '%s140.dll') | |
144 else: | |
145 # VS2013 is the default. | |
146 _CopyRuntime2013(out_debug, x86, 'msvc%s120d.dll') | |
147 _CopyRuntime2013(out_release, x86, 'msvc%s120.dll') | |
148 _CopyRuntime2013(out_debug_x64, x64, 'msvc%s120d.dll') | |
149 _CopyRuntime2013(out_release_x64, x64, 'msvc%s120.dll') | |
150 _CopyRuntime2013(out_debug_nacl64, x64, 'msvc%s120d.dll') | |
151 _CopyRuntime2013(out_release_nacl64, x64, 'msvc%s120.dll') | |
152 | |
153 # Copy the PGO runtime library to the release directories. | |
154 if os.environ.get('GYP_MSVS_OVERRIDE_PATH'): | |
155 pgo_x86_runtime_dir = os.path.join(os.environ.get('GYP_MSVS_OVERRIDE_PATH'), | |
156 'VC', 'bin') | |
157 pgo_x64_runtime_dir = os.path.join(pgo_x86_runtime_dir, 'amd64') | |
158 pgo_runtime_dll = 'pgort' + _VersionNumber() + '.dll' | |
159 source_x86 = os.path.join(pgo_x86_runtime_dir, pgo_runtime_dll) | |
160 if os.path.exists(source_x86): | |
161 _CopyRuntimeImpl(os.path.join(out_release, pgo_runtime_dll), source_x86) | |
162 source_x64 = os.path.join(pgo_x64_runtime_dir, pgo_runtime_dll) | |
163 if os.path.exists(source_x64): | |
164 _CopyRuntimeImpl(os.path.join(out_release_x64, pgo_runtime_dll), | |
165 source_x64) | |
166 | |
167 | |
168 def CopyDlls(target_dir, configuration, target_cpu): | |
169 """Copy the VS runtime DLLs into the requested directory as needed. | |
170 | |
171 configuration is one of 'Debug' or 'Release'. | |
172 target_cpu is one of 'x86' or 'x64'. | |
173 | |
174 The debug configuration gets both the debug and release DLLs; the | |
175 release config only the latter. | |
176 """ | |
177 vs2013_runtime_dll_dirs = SetEnvironmentAndGetRuntimeDllDirs() | |
178 if not vs2013_runtime_dll_dirs: | |
179 return | |
180 | |
181 x64_runtime, x86_runtime = vs2013_runtime_dll_dirs | |
182 runtime_dir = x64_runtime if target_cpu == 'x64' else x86_runtime | |
183 _CopyRuntime2013( | |
184 target_dir, runtime_dir, 'msvc%s' + _VersionNumber() + '.dll') | |
185 if configuration == 'Debug': | |
186 _CopyRuntime2013( | |
187 target_dir, runtime_dir, 'msvc%s' + _VersionNumber() + 'd.dll') | |
188 | |
189 | |
190 def _GetDesiredVsToolchainHashes(): | |
191 """Load a list of SHA1s corresponding to the toolchains that we want installed | |
192 to build with.""" | |
193 if os.environ.get('GYP_MSVS_VERSION') == '2015': | |
194 return ['49ae4b60d898182fc3f521c2fcda82c453915011'] | |
195 else: | |
196 # Default to VS2013. | |
197 return ['ee7d718ec60c2dc5d255bbe325909c2021a7efef'] | |
198 | |
199 | |
200 def Update(force=False): | |
201 """Requests an update of the toolchain to the specific hashes we have at | |
202 this revision. The update outputs a .json of the various configuration | |
203 information required to pass to gyp which we use in |GetToolchainDir()|. | |
204 """ | |
205 if force != False and force != '--force': | |
206 print >>sys.stderr, 'Unknown parameter "%s"' % force | |
207 return 1 | |
208 if force == '--force' or os.path.exists(json_data_file): | |
209 force = True | |
210 | |
211 depot_tools_win_toolchain = \ | |
212 bool(int(os.environ.get('DEPOT_TOOLS_WIN_TOOLCHAIN', '1'))) | |
213 if ((sys.platform in ('win32', 'cygwin') or force) and | |
214 depot_tools_win_toolchain): | |
215 import find_depot_tools | |
216 depot_tools_path = find_depot_tools.add_depot_tools_to_path() | |
217 get_toolchain_args = [ | |
218 sys.executable, | |
219 os.path.join(depot_tools_path, | |
220 'win_toolchain', | |
221 'get_toolchain_if_necessary.py'), | |
222 '--output-json', json_data_file, | |
223 ] + _GetDesiredVsToolchainHashes() | |
224 if force: | |
225 get_toolchain_args.append('--force') | |
226 subprocess.check_call(get_toolchain_args) | |
227 | |
228 return 0 | |
229 | |
230 | |
231 def GetToolchainDir(): | |
232 """Gets location information about the current toolchain (must have been | |
233 previously updated by 'update'). This is used for the GN build.""" | |
234 runtime_dll_dirs = SetEnvironmentAndGetRuntimeDllDirs() | |
235 | |
236 # If WINDOWSSDKDIR is not set, search the default SDK path and set it. | |
237 if not 'WINDOWSSDKDIR' in os.environ: | |
238 default_sdk_path = 'C:\\Program Files (x86)\\Windows Kits\\8.1' | |
239 if os.path.isdir(default_sdk_path): | |
240 os.environ['WINDOWSSDKDIR'] = default_sdk_path | |
241 | |
242 print '''vs_path = "%s" | |
243 sdk_path = "%s" | |
244 vs_version = "%s" | |
245 wdk_dir = "%s" | |
246 runtime_dirs = "%s" | |
247 ''' % ( | |
248 os.environ['GYP_MSVS_OVERRIDE_PATH'], | |
249 os.environ['WINDOWSSDKDIR'], | |
250 os.environ['GYP_MSVS_VERSION'], | |
251 os.environ.get('WDK_DIR', ''), | |
252 ';'.join(runtime_dll_dirs or ['None'])) | |
253 | |
254 | |
255 def main(): | |
256 commands = { | |
257 'update': Update, | |
258 'get_toolchain_dir': GetToolchainDir, | |
259 'copy_dlls': CopyDlls, | |
260 } | |
261 if len(sys.argv) < 2 or sys.argv[1] not in commands: | |
262 print >>sys.stderr, 'Expected one of: %s' % ', '.join(commands) | |
263 return 1 | |
264 return commands[sys.argv[1]](*sys.argv[2:]) | |
265 | |
266 | |
267 if __name__ == '__main__': | |
268 sys.exit(main()) | |
OLD | NEW |