Package/Install the Windows 10 Universal C Runtime for VS 2015

win_toolchain/get_toolchain_if_necessary.py

 #!/usr/bin/env python
 # Copyright 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.
 
 """Downloads and unpacks a toolchain for building on Windows. The contents are
 matched by sha1 which will be updated when the toolchain is updated.
 
 Having a toolchain script in depot_tools means that it's not versioned
 directly with the source code. That is, if the toolchain is upgraded, but
 you're trying to build an historical version of Chromium from before the
 toolchain upgrade, this will cause you to build with a newer toolchain than
 was available when that code was committed. This is done for a two main
 reasons: 1) it would likely be annoying to have the up-to-date toolchain
 removed and replaced by one without a service pack applied); 2) it would
 require maintaining scripts that can build older not-up-to-date revisions of
 the toolchain. This is likely to be a poorly tested code path that probably
 won't be properly maintained. See http://crbug.com/323300.
 
 This does not extend to major versions of the toolchain however, on the
 assumption that there are more likely to be source incompatibilities between
 major revisions. This script calls a subscript (currently, toolchain2013.py)
 to do the main work. It is expected that toolchain2013.py will always be able
 to acquire/build the most current revision of a VS2013-based toolchain. In the
 future when a hypothetical VS2015 is released, the 2013 script will be
 maintained, and a new 2015 script would be added.
 """
 
+import _winreg
 import hashlib
 import json
 import optparse
 import os
+import platform
 import shutil
 import subprocess
 import sys
 import tempfile
 import time
 import zipfile
 
 
 BASEDIR = os.path.dirname(os.path.abspath(__file__))
 DEPOT_TOOLS_PATH = os.path.join(BASEDIR, '..')
@@ skipping 167 matching lines @@
   else:
     temp_dir, local_zip = DownloadUsingGsutil(tree_sha1 + '.zip')
   sys.stdout.write('Extracting %s...\n' % local_zip)
   sys.stdout.flush()
   with zipfile.ZipFile(local_zip, 'r', zipfile.ZIP_DEFLATED, True) as zf:
     zf.extractall(target_dir)
   if temp_dir:
     RmDir(temp_dir)
 
 
+def GetInstallerName():
+  """Return the name of the Windows 10 Universal C Runtime installer for the
+  current platform, or None if installer is not needed or not applicable.
+  The registry has to be used instead of sys.getwindowsversion() because
+  Python 2.7 is only manifested as being compatible up to Windows 8, so the
+  version APIs helpfully return a maximum of 6.2 (Windows 8).
+  """
+  key_name = r'Software\Microsoft\Windows NT\CurrentVersion'
+  key = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, key_name)
+  value, keytype = _winreg.QueryValueEx(key, "CurrentVersion")
+  key.Close()
+  if keytype != _winreg.REG_SZ:
+    raise Exception("Unexpected type in registry")
+  if value == '6.1':
+    # Windows 7 and Windows Server 2008 R2
+    return 'Windows6.1-KB2999226-x64.msu'
+  elif value == '6.2':
+    # Windows 8 and Windows Server 2012
+    return 'Windows8-RT-KB2999226-x64.msu'
+  elif value == '6.3':
+    # Windows 8.1, Windows Server 2012 R2, and Windows 10.
+    # The Windows 8.1 installer doesn't work on Windows 10, but it will never
+    # be used because the UCRT is always installed on Windows 10.
+    return 'Windows8.1-KB2999226-x64.msu'
+  else:
+    # Some future OS.
+    return None
+
+
+def InstallUniversalCRTIfNeeded(abs_target_dir):
+  installer_name = GetInstallerName()
+  if not installer_name:
+    return
+
+  bitness = platform.architecture()[0]
+  # When running 64-bit python the x64 DLLs will be in System32
+  x64_path = 'System32' if bitness == '64bit' else 'Sysnative'
+  x64_path = os.path.join(r'C:\Windows', x64_path)
+  sample_crt_file = os.path.join(x64_path, 'ucrtbase.dll')
+
+  if os.path.exists(sample_crt_file):
+    # Nothing to do.
+    return
+
+  print ('%s does not exist - installing Windows 10 Universal C Runtime' %
+         sample_crt_file)
+
+  installer = os.path.join(abs_target_dir, "installers", installer_name)
+  command = r'wusa.exe /quiet "%s"' % installer
+  print 'Running %s' % command
+
+  try:
+    subprocess.check_call(command)
+  # Trap OSError instead of WindowsError so pylint will succeed on Linux.
+  except OSError as e:
+    if e.winerror == 740: # The requested operation requires elevation
+      print
+      print '-'*80
+      print
+      print 'Elevation required. You must manually install this update:'
+      print '  %s' % installer
+      print
+      print '-'*80
+      print
+      raise Exception('Elevation required. You must manually install %s' %
+                      installer)
+    raise e
+
+
 def main():
   parser = optparse.OptionParser(description=sys.modules[__name__].__doc__)
   parser.add_option('--output-json', metavar='FILE',
                     help='write information about toolchain to FILE')
   parser.add_option('--force', action='store_true',
                     help='force script to run on non-Windows hosts')
   options, args = parser.parse_args()
 
   if not (sys.platform.startswith(('cygwin', 'win32')) or options.force):
     return 0
@@ skipping 44 matching lines @@
       print('\n\n\nPlease follow the instructions at '
             'https://www.chromium.org/developers/how-tos/'
             'build-instructions-windows\n\n')
       return 1
     print('Windows toolchain out of date or doesn\'t exist, updating (Pro)...')
     print('  current_hash: %s' % current_hash)
     print('  desired_hashes: %s' % ', '.join(desired_hashes))
     sys.stdout.flush()
     DelayBeforeRemoving(target_dir)
     if sys.platform == 'win32':
-      # This stays resident and will make the rmdir below fail.
-      with open(os.devnull, 'wb') as nul:
-        subprocess.call(['taskkill', '/f', '/im', 'mspdbsrv.exe'],
-                        stdin=nul, stdout=nul, stderr=nul)
+      # These stay resident and will make the rmdir below fail.
+      kill_list = [
+        'mspdbsrv.exe',
+        'vctip.exe', # Compiler and tools experience improvement data uploader.
+      ]
+      for process_name in kill_list:
+        with open(os.devnull, 'wb') as nul:
+          subprocess.call(['taskkill', '/f', '/im', process_name],
+                          stdin=nul, stdout=nul, stderr=nul)
     if os.path.isdir(target_dir):
       RmDir(target_dir)
 
     DoTreeMirror(target_dir, desired_hashes[0])
 
     got_new_toolchain = True
 
   win_sdk = os.path.join(abs_target_dir, 'win_sdk')
   try:
     with open(os.path.join(target_dir, 'VS_VERSION'), 'rb') as f:
@@ skipping 26 matching lines @@
           'Got wrong hash after pulling a new toolchain. '
           'Wanted one of \'%s\', got \'%s\'.' % (
               ', '.join(desired_hashes), current_hash))
       return 1
     SaveTimestampsAndHash(target_dir, current_hash)
 
   if options.output_json:
     shutil.copyfile(os.path.join(target_dir, '..', 'data.json'),
                     options.output_json)
 
+  if os.environ.get('GYP_MSVS_VERSION') == '2015':
+    InstallUniversalCRTIfNeeded(abs_target_dir)
+
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())