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
|
import sys
import re
import os
import subprocess as subp
known_rpaths = {
'@rpath/libc++.1.dylib': "/usr/lib/libc++.1.dylib",
}
def run_shell(cmd):
out = subp.check_output(cmd, shell=True)
return out
def main(whl):
# Find shared libraries
run_shell('wheel unpack {}'.format(whl))
thedir = '-'.join(whl.split('-')[:2])
sharedlibs = run_shell('find {} -name "*.so"'.format(thedir)).decode().splitlines()
# Scan paths
regex_rpath = re.compile(r'@rpath\/[^ ]+')
updated_files = set()
for path in sharedlibs:
# Check rpaths
otool_out = run_shell('otool -L {}'.format(path)).decode()
rpaths = set(regex_rpath.findall(otool_out))
unknown = rpaths - set(known_rpaths.keys())
if unknown:
#raise AssertionError('unknown rpath {}'.format(unknown))
print('unknown', unknown)
# Patch
for rpath in rpaths & set(known_rpaths.keys()):
print('patch', rpath, 'in', path)
run_shell('install_name_tool -change {} {} {}'.format(
rpath, known_rpaths[rpath], path))
updated_files.add(path)
print('updated_files', updated_files)
# Update whl
if updated_files:
run_shell('wheel pack {}'.format(thedir))
else:
print('nothing to do')
if __name__ == '__main__':
main(*sys.argv[1:])
|
