1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
# 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')
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.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)
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())
|
