forked from swiftlang/swift
-
Notifications
You must be signed in to change notification settings - Fork 0
/
swift-darwin-postprocess.py
executable file
·102 lines (83 loc) · 3.61 KB
/
swift-darwin-postprocess.py
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
#!/usr/bin/env python3
# Postprocess binaries to prepare for their execution on Darwin platforms.
# This includes un-rpathization and codesigning.
import argparse
import os
import re
import subprocess
import sys
utils = os.path.dirname(os.path.realpath(__file__))
get_task_allow_plist = os.path.join(utils, 'get-task-allow.plist')
def main(arguments):
parser = argparse.ArgumentParser(
description='Postprocess binaries to prepare for \
their execution on Darwin platforms')
parser.add_argument('bins', nargs='+', help='one or more binary files')
args = parser.parse_args(arguments)
for bin in args.bins:
unrpathize(bin)
codesign(bin)
# This function rewrites binaries that use these `@rpath`-based load
# commands to use direct /usr/lib/swift paths instead, to work around
# an issue where the DYLD_LIBRARY_PATH override doesn't reliably work
# with runpath-relative loads of system libraries on some dyld versions.
# (rdar://78851265)
def unrpathize(filename):
dylibsOutput = None
try:
# `dyldinfo` has been replaced with `dyld_info`, so we try it first
# before falling back to `dyldinfo`
dylibsOutput = subprocess.check_output(
['xcrun', 'dyld_info', '-dependents', filename],
universal_newlines=True)
except subprocess.CalledProcessError:
sys.stderr.write("falling back to 'xcrun dyldinfo' ...\n")
dylibsOutput = subprocess.check_output(
['xcrun', 'dyldinfo', '-dylibs', filename],
universal_newlines=True)
# Do not rewrite @rpath-relative load commands for these libraries:
# they are test support libraries that are never installed under
# /usr/lib/swift and they aren't loaded via DYLD_LIBRARY_PATH.
allow_list = {
'libswiftDifferentiationUnittest.dylib',
'libswiftLocalizationAnalysisTestHelper.dylib',
'libswiftOSLogTestHelper.dylib',
'libswiftRemoteMirror.dylib',
'libswiftRuntimeUnittest.dylib',
'libswiftStdlibCollectionUnittest.dylib',
'libswiftStdlibUnicodeUnittest.dylib',
'libswiftStdlibUnittest.dylib',
'libswiftStdlibUnittestFoundationExtras.dylib',
'libswiftSwiftPrivate.dylib',
'libswiftSwiftPrivateLibcExtras.dylib',
'libswiftSwiftPrivateThreadExtras.dylib',
'libswiftSwiftReflectionTest.dylib',
'libswiftGenericMetadataBuilder.dylib',
}
# The output from dyldinfo -dylibs is a line of header followed by one
# install name per line, indented with spaces.
dylib_regex = re.compile(
r"(^|.*\s)(?P<path>@rpath/(?P<filename>libswift.*\.dylib))\s*$")
# Build a command to invoke install_name_tool.
command = ['install_name_tool']
for line in dylibsOutput.splitlines():
match = dylib_regex.match(line)
if match and match.group('filename') not in allow_list:
command.append('-change')
command.append(match.group('path'))
command.append('/usr/lib/swift/' + match.group('filename'))
continue
# Don't run the command if we didn't find any dylibs to change:
# it's invalid to invoke install_name_tool without any operations.
if len(command) == 1:
return
# The last argument is the filename to operate on.
command.append(filename)
subprocess.check_call(command)
def codesign(filename):
# "-" is the signing identity for ad-hoc signing.
command = ['/usr/bin/codesign', '--force', '--sign', '-',
'--entitlements', get_task_allow_plist,
filename]
subprocess.check_call(command)
sys.exit(main(sys.argv[1:]))