-
-
Notifications
You must be signed in to change notification settings - Fork 5.5k
/
fixup-libgfortran.sh
executable file
·162 lines (141 loc) · 6.03 KB
/
fixup-libgfortran.sh
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
#!/bin/sh
# This file is a part of Julia. License is MIT: https://julialang.org/license
# Run as: fixup-libgfortran.sh [--verbose] <$private_libdir>
FC=${FC:-gfortran}
PATCHELF=${PATCHELF:-patchelf}
# If we're invoked with "--verbose", create a `debug` function that prints stuff out
if [ "$1" = "--verbose" ] || [ "$1" = "-v" ]; then
shift 1
debug() { echo "$*" >&2; }
else
debug() { :; }
fi
if [ -z "$1" ]; then
echo "Usage: $0 [--verbose] <private_libdir>"
exit 1
fi
UNAME="$(uname -s)"
if [ "$UNAME" = "Linux" ]; then
SHLIB_EXT="so"
elif [ "$UNAME" = "Darwin" ]; then
SHLIB_EXT="dylib"
else
echo "WARNING: Could not autodetect platform type ('uname -s' = $UNAME); assuming Linux" >&2
UNAME="Linux"
SHLIB_EXT="so"
fi
private_libdir=$1
find_shlib()
{
lib_path="$1"
if [ -f "$lib_path" ]; then
if [ "$UNAME" = "Linux" ]; then
${PATCHELF} --print-needed "$lib_path" | grep "$2" | xargs
else # $UNAME is "Darwin", we only have two options, see above
otool -L "$lib_path" | grep "$2" | cut -d' ' -f1 | xargs
fi
fi
}
find_shlib_dir()
{
# Usually, on platforms like OSX we get full paths when linking. However,
# if we are inspecting, say, BinaryBuilder-built OpenBLAS libraries, we will
# only get something like `@rpath/libgfortran.5.dylib` when inspecting the
# libraries. We can, as a last resort, ask `$FC` directly what the full
# filepath for this library is, but only if we don't have a direct path to it:
if [ "$(dirname "$1")" = "@rpath" ] || [ "$(dirname "$1")" = "." ]; then
dirname "$($FC -print-file-name="$(basename "$1")" 2>/dev/null)"
else
dirname "$1" 2>/dev/null
fi
}
# First, discover all the places where libgfortran/libgcc is, as well as their true SONAMES
for lib in lapack blas openblas; do
for private_libname in ${private_libdir}/lib$lib*.$SHLIB_EXT*; do
# Find the paths to the libraries we're interested in. These are almost
# always within the same directory, but we like to be general.
LIBGFORTRAN_PATH="$(find_shlib "$private_libname" libgfortran)"
if [ -z "$LIBGFORTRAN_PATH" ]; then continue; fi
# Take the directories, add them onto LIBGFORTRAN_DIRS, which we use to
# search for these libraries in the future. If there is no directory, try
# asking `$FC` where such a file could be found.
LIBGFORTRAN_DIR="$(find_shlib_dir "$LIBGFORTRAN_PATH")"
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $LIBGFORTRAN_DIR"
# Save the SONAME
LIBGFORTRAN_SONAME="$(basename "$LIBGFORTRAN_PATH")"
LIBGFORTRAN_SONAMES="$LIBGFORTRAN_SONAMES $LIBGFORTRAN_SONAME"
# Now that we've (maybe) found a libgfortran, ask _it_ for things like libgcc_s and libquadmath:
LIBGCC_PATH="$(find_shlib "${LIBGFORTRAN_DIR}/${LIBGFORTRAN_SONAME}" libgcc_s)"
LIBQUADMATH_PATH="$(find_shlib "${LIBGFORTRAN_DIR}/${LIBGFORTRAN_SONAME}" libquadmath)"
if [ ! -z "$LIBGCC_PATH" ]; then
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(find_shlib_dir $LIBGCC_PATH)"
LIBGCC_SONAMES="$LIBGCC_SONAMES $(basename "$LIBGCC_PATH")"
fi
if [ ! -z "$LIBQUADMATH_PATH" ]; then
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(find_shlib_dir $LIBQUADMATH_PATH)"
LIBQUADMATH_SONAMES="$LIBQUADMATH_SONAMES $(basename "$LIBQUADMATH_PATH")"
fi
done
done
# Take in a list of space-separated tokens, return a deduplicated list of the same
uniquify()
{
echo "$1" | tr " " "\n" | sort | uniq | grep -v '^$' | tr "\n" " "
}
LIBGFORTRAN_DIRS=$(uniquify "$LIBGFORTRAN_DIRS")
SONAMES="$(uniquify "$LIBGFORTRAN_SONAMES $LIBGCC_SONAMES $LIBQUADMATH_SONAMES")"
debug "Discovered traces of libgfortran within $LIBGFORTRAN_DIRS"
debug "Got SONAMES of $SONAMES"
# Copy the SONAMEs we identified above into our private_libdir
for soname in $SONAMES; do
for dir in $LIBGFORTRAN_DIRS; do
if [ ! -f "$private_libdir/$soname" ] && [ -f "$dir/$soname" ]; then
cp -v "$dir/$soname" "$private_libdir"
chmod 755 "$private_libdir/$soname"
if [ "$UNAME" = "Darwin" ]; then
debug "Rewriting identity of ${private_libdir}/${soname} to @rpath/${soname}"
install_name_tool -id "@rpath/$soname" "$private_libdir/$soname"
fi
fi
done
done
# On OSX, we need to change the old link (which is usually a full path)
# to point to `@rpath/${soname}` explicitly, so we use `find_shlib()`
# to dynamically find the full path we want to change. On Linux, we
# don't care about full paths, we just set the rpath to `$ORIGIN`.
change_linkage()
{
# This is the path of the library we want to edit
lib_path="$1"
# If it doesn't exist, exit quietly
if [ ! -f "$lib_path" ]; then
debug " $lib_path doesn't exist, skipping"
return
fi
# This is the soname of the dependency we want to swap out
soname="$2"
if [ "$UNAME" = "Darwin" ]; then
old_link=$(find_shlib "$lib_path" "$soname")
echo " $old_link"
install_name_tool -change "$old_link" "@rpath/$soname" "$lib_path"
else # $UNAME is "Linux", we only have two options, see above
${PATCHELF} --set-rpath \$ORIGIN "$lib_path"
fi
}
# For every library that remotely touches libgfortran stuff (the libraries we
# have copied in ourselves) we must
# update the linkage to point to @rpath (on OSX) or $ORIGIN (on Linux) so
# that direct links to the old libgfortran directories are instead directed
# to the proper location, which is our $private_libdir.
for lib in libopenblas libcholmod liblapack $SONAMES; do
# Grab every incarnation of that library that exists within $private_libdir
# (e.g. "libopenblas.so", and "libopenblas.so.0", etc...)
for lib_path in $private_libdir/$lib*; do
# Iterate over dependency names that need to be changed
for soname in $SONAMES; do
debug "changing linkage of $lib_path to $soname"
chmod 755 "$lib_path"
change_linkage "$lib_path" "$soname"
done
done
done