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
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#!/usr/bin/env python
# Copyright (c) 2011 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
import tempfile
import _winreg
VSVARS_PATH = ('C:\\Program Files (x86)\\Microsoft Visual Studio 9.0\\'
'Common7\\Tools\\vsvars32.bat')
def run_with_vsvars(cmd):
(fd, filename) = tempfile.mkstemp('.bat', text=True)
f = os.fdopen(fd, "w")
f.write('@echo off\n')
f.write('call "%s"\n' % VSVARS_PATH)
f.write(cmd + '\n')
f.close()
try:
p = subprocess.Popen([filename], shell=True, stdout=subprocess.PIPE,
universal_newlines=True)
out, err = p.communicate()
return p.returncode, out
finally:
os.unlink(filename)
def get_vc_dir():
rc, out = run_with_vsvars('echo VCINSTALLDIR=%VCINSTALLDIR%')
for line in out.splitlines():
if line.startswith('VCINSTALLDIR='):
return line[len('VCINSTALLDIR='):]
return None
def main():
vcdir = os.environ.get('VCINSTALLDIR')
if not vcdir:
vcdir = get_vc_dir()
if not vcdir:
print 'Couldn\'t get VCINSTALLDIR. Run vsvars32.bat?'
return 1
os.environ['PATH'] += (';' + os.path.join(vcdir, 'bin') +
';' + os.path.join(vcdir, '..\\Common7\\IDE'))
# Switch to our own dir.
os.chdir(os.path.dirname(os.path.abspath(__file__)))
# Verify that we can find link.exe.
link = os.path.join(vcdir, 'bin', 'link.exe')
link_backup = os.path.join(vcdir, 'bin', 'link.exe.supalink_orig.exe')
if not os.path.exists(link):
print 'link.exe not found at %s' % link
return 1
# 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 supalink.exe but only if it's out of date.
cpptime = os.path.getmtime('supalink.cpp')
if (not os.path.exists('supalink.exe') or
cpptime > os.path.getmtime('supalink.exe')):
print 'Building supalink.exe...'
rc, out = run_with_vsvars('cl /nologo /Ox /Zi /W4 /WX /D_UNICODE /DUNICODE'
' /D_CRT_SECURE_NO_WARNINGS /EHsc supalink.cpp'
' /link /out:supalink.exe')
if rc:
print out
print 'Failed to build supalink.exe'
return 1
# Copy supalink into place if it's been updated since last time we ran.
exetime = os.path.getmtime('supalink.exe')
if exetime > os.path.getmtime(link):
print 'Copying supalink.exe over link.exe...'
try:
shutil.copyfile('supalink.exe', link)
except IOError:
print ('Wasn\'t able to copy supalink.exe over %s. '
'Not running with Administrator privileges?' % link)
return 1
_winreg.SetValue(_winreg.HKEY_CURRENT_USER,
'Software\\Chromium\\supalink_installed',
_winreg.REG_SZ,
link_backup)
print 'Linker shim installed. Regenerate via gyp: "gclient runhooks".'
return 0
if __name__ == '__main__':
sys.exit(main())
|
