split_link tool, config, and scripts for windows build

tools/win/split_link/install_split_link.py

+#!/usr/bin/env python
+# Copyright (c) 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.
+
+import os
+import shutil
+import subprocess
+import sys
+
+
+if sys.platform == 'win32':
+  import _winreg
+elif sys.platform == 'cygwin':
+  try:
+    import cygwinreg as _winreg
+  except ImportError:
+    print "Please install cygwinreg so I can access the registry."
+    print "Install setuptools and run 'easy_install cygwinreg'."
+
+
+def IsExe(fpath):
+  return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
+
+
+def FindInPath(program):
+  fpath, fname = os.path.split(program)
+  if fpath:
+    if IsExe(program):
+      return program
+  else:
+    for path in os.environ['PATH'].split(os.pathsep):
+      path = path.strip('"')
+      exe_file = os.path.join(path, program)
+      if IsExe(exe_file):
+        return exe_file
+  return None
+
+
+def Run(command):
+  p = subprocess.Popen([], shell=True, stdout=subprocess.PIPE,
+        universal_newlines=True)
+  out, err = p.communicate()
+  return p.returncode, out
+
+
+def EscapeForCommandLineAndCString(path):
+  """Quoted sufficiently to be passed on the compile command line as a define
+  to be turned into a string in the target C program."""
+  path = '"' + path + '"'
+  return path.replace('\\', '\\\\').replace('"', '\\"')
+
+
+def main():
+  # Switch to our own dir.
+  os.chdir(os.path.dirname(os.path.abspath(__file__)))
+
+  link = FindInPath('link.exe')
+  if not link:
+    print ("Couldn't find link.exe in PATH. Must run from Administrator "
+           "Visual Studio Command Prompt.")
+    return 1
+
+  link_backup = os.path.join(os.path.split(link)[0], 'link.exe.split_link.exe')
+
+  # Don't re-backup link.exe, so only copy link.exe to backup if it's
+  # not there already.
+  if not os.path.exists(link_backup):
+    try:
+      print 'Saving original link.exe...'
+      shutil.copyfile(link, link_backup)
+    except IOError:
+      print (('Wasn\'t able to back up %s to %s. '
+              'Not running with Administrator privileges?')
+              % (link, link_backup))
+      return 1
+
+  # Build our linker shim.
+  print 'Building split_link.exe...'
+  split_link_py = os.path.abspath('split_link.py')
+  script_path = EscapeForCommandLineAndCString(split_link_py)
+  python = EscapeForCommandLineAndCString(sys.executable)
+  subprocess.check_call('cl /nologo /Ox /Zi /W4 /WX /D_UNICODE /DUNICODE'
+                        ' /D_CRT_SECURE_NO_WARNINGS /EHsc split_link.cc'
+                        ' /DPYTHON_PATH="%s"'
+                        ' /DSPLIT_LINK_SCRIPT_PATH="%s"'
+                        ' /link shell32.lib shlwapi.lib /out:split_link.exe' % (
+                            python, script_path),
+                        shell=True)
+
+  # Copy shim into place.
+  print 'Copying split_link.exe over link.exe...'
+  try:
+    shutil.copyfile('split_link.exe', link)
+    _winreg.SetValue(_winreg.HKEY_CURRENT_USER,
+                     'Software\\Chromium\\split_link_installed',
+                     _winreg.REG_SZ,
+                     link_backup)
+  except IOError:
+    print ('Wasn\'t able to copy split_link.exe over %s. '
+            'Not running with Administrator privileges?' % link)
+    return 1
+
+  return 0
+
+
+if __name__ == '__main__':
+  sys.exit(main())