Skip to content

Commit

Permalink
Rearchitect this file for the second time, it's much more precise now
Browse files Browse the repository at this point in the history
  • Loading branch information
staticfloat committed May 11, 2017
1 parent b50fae9 commit 116741d
Showing 1 changed file with 55 additions and 38 deletions.
93 changes: 55 additions & 38 deletions contrib/fixup-libgfortran.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,15 @@
#!/bin/sh
# This file is a part of Julia. License is MIT: https://julialang.org/license

# Run as: fixup-libgfortran.sh <$private_libdir>
# Run as: fixup-libgfortran.sh [--verbose] <$private_libdir>

# If we're invoked with "--verbose", create a `debug` function that prints stuff out
if [ "$1" == "--verbose" ] || [ "$1" == "-v" ]; then
shift 1
debug() { echo "$*"; }
else
debug() { :; }
fi

if [ -z "$1" ]; then
echo "Usage: $0 <private_libdir>"
Expand All @@ -14,7 +22,7 @@ if [ "$UNAME" = "Linux" ]; then
elif [ "$UNAME" = "Darwin" ]; then
SHLIB_EXT="dylib"
else
echo "WARNING: Could not autodetect platform type ('uname -s' == $UNAME); assuming Linux"
echo "WARNING: Could not autodetect platform type ('uname -s' == $UNAME); assuming Linux" >&2
UNAME="Linux"
SHLIB_EXT="so"
fi
Expand All @@ -26,27 +34,34 @@ if [ ! -f "$private_libdir/libarpack.$SHLIB_EXT" ]; then
exit 2
fi

find_shlib ()
find_shlib()
{
if [ -f "$private_libdir/lib$1.$SHLIB_EXT" ]; then
lib_path="$1"
if [ -f "$lib_path" ]; then
if [ "$UNAME" = "Linux" ]; then
ldd "$private_libdir/lib$1.$SHLIB_EXT" | grep $2 | cut -d' ' -f3 | xargs
ldd "$lib_path" | grep $2 | cut -d' ' -f3 | xargs
else # $UNAME is "Darwin", we only have two options, see above
otool -L "$private_libdir/lib$1.$SHLIB_EXT" | grep $2 | cut -d' ' -f1 | xargs
otool -L "$lib_path" | grep $2 | cut -d' ' -f1 | xargs
fi
fi
}

private_libname()
{
echo "$private_libdir/lib$1.$SHLIB_EXT"
}

# First, discover all the places where libgfortran/libgcc is, as well as their true SONAMES
for lib in arpack openspecfun lapack; do
if [ -f "$private_libdir/lib$lib.$SHLIB_EXT" ]; then
# 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 $lib libgfortran)
LIBGCC_PATH=$(find_shlib $lib libgcc_s)
LIBQUADMATH_PATH=$(find_shlib $lib libquadmath)
LIBGFORTRAN_PATH=$(find_shlib "$(private_libname $lib)" libgfortran)
LIBGCC_PATH=$(find_shlib "$(private_libname $lib)" libgcc_s)
LIBQUADMATH_PATH=$(find_shlib "$(private_libname $lib)" libquadmath)

# Take the directories, add them onto LIBGFORTRAN_DIRS to search for things later
# Take the directories, add them onto LIBGFORTRAN_DIRS, which we use to
# search for these libraries with in the future.
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(dirname $LIBGFORTRAN_PATH)"
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(dirname $LIBGCC_PATH)"
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(dirname $LIBQUADMATH_PATH)"
Expand All @@ -66,6 +81,8 @@ uniquify()

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
Expand All @@ -78,48 +95,48 @@ for soname in $SONAMES; do
fi
fi
done

# Add possible internal directories to LIBGFORTRAN_DIRS so that we can find all possible ways
# That any of our libraries or these libraries we just copied in link to themselves later.
if [ -f "$private_libdir/$soname" ]; then
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(dirname $(find_shlib $lib libgfortran))"
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(dirname $(find_shlib $lib libgcc_s))"
LIBGFORTRAN_DIRS="$LIBGFORTRAN_DIRS $(dirname $(find_shlib $lib libquadmath))"
fi
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()
{
shlib_path="$1"
old_link="$2"
new_link="$3"
# 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
install_name_tool -change "$old_link" "$new_link" "$shlib_path"
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 "$shlib_path"
patchelf --set-rpath \$ORIGIN "$lib_path"
fi
}

LIBGFORTRAN_DIRS=$(uniquify "$LIBGFORTRAN_DIRS")
echo "Found traces of libgfortran/libgcc in $LIBGFORTRAN_DIRS"

# For every library that remotely touches libgfortran stuff (so the libraries
# we have copied in ourselves as well as arpack, openspecfun, etc...) we must
# For every library that remotely touches libgfortran stuff (the libraries we
# have copied in ourselves as well as arpack, openspecfun, etc...) 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.
cd $private_libdir

# Iterate over possible library names
for soname in libopenblas libarpack libcholmod liblapack libopenspecfun $SONAMES; do
for lib in libopenblas libarpack libcholmod liblapack libopenspecfun $SONAMES; do
# Grab every incarnation of that library that exists within $private_libdir
# (e.g. "libopenspecfun.so", and "libopenspecfun.so.0", etc...)
for lib in $private_libdir/$soname*; do
# Look for links to any of the our three musketeers within ANY of the
# potential LIBGFORTRAN_DIRS we've discovered so far
for dir in $LIBGFORTRAN_DIRS; do
change_linkage "$lib" "$dir/$soname" "@rpath/$soname"
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"
change_linkage "$lib_path" "$soname"
done
done
done
Expand Down

0 comments on commit 116741d

Please sign in to comment.