101 lines
3.0 KiB
Python
101 lines
3.0 KiB
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 _winreg
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
def IsExe(fpath):
|
|
return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
|
|
|
|
|
|
def FindInPath(program):
|
|
fpath, _ = 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 not path or not os.path.isabs(path):
|
|
continue
|
|
if IsExe(exe_file):
|
|
return exe_file
|
|
return None
|
|
|
|
|
|
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(BASE_DIR)
|
|
|
|
link = FindInPath('link.exe')
|
|
mt = FindInPath('mt.exe')
|
|
if not link or not mt:
|
|
print("Couldn't find link.exe or mt.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.exe /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))
|
|
|
|
# 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)
|
|
_winreg.SetValue(_winreg.HKEY_CURRENT_USER,
|
|
'Software\\Chromium\\split_link_mt_path',
|
|
_winreg.REG_SZ,
|
|
mt)
|
|
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())
|