|
| 1 | +# This file is a part of Julia. License is MIT: https://julialang.org/license |
| 2 | +module Linking |
| 3 | + |
| 4 | +import Base.Libc: Libdl |
| 5 | + |
| 6 | +# inlined LLD_jll |
| 7 | +# These get calculated in __init__() |
| 8 | +const PATH = Ref("") |
| 9 | +const LIBPATH = Ref("") |
| 10 | +const PATH_list = String[] |
| 11 | +const LIBPATH_list = String[] |
| 12 | +const lld_path = Ref{String}() |
| 13 | +const lld_exe = Sys.iswindows() ? "lld.exe" : "lld" |
| 14 | + |
| 15 | +if Sys.iswindows() |
| 16 | + const LIBPATH_env = "PATH" |
| 17 | + const LIBPATH_default = "" |
| 18 | + const pathsep = ';' |
| 19 | +elseif Sys.isapple() |
| 20 | + const LIBPATH_env = "DYLD_FALLBACK_LIBRARY_PATH" |
| 21 | + const LIBPATH_default = "~/lib:/usr/local/lib:/lib:/usr/lib" |
| 22 | + const pathsep = ':' |
| 23 | +else |
| 24 | + const LIBPATH_env = "LD_LIBRARY_PATH" |
| 25 | + const LIBPATH_default = "" |
| 26 | + const pathsep = ':' |
| 27 | +end |
| 28 | + |
| 29 | +function adjust_ENV!(env::Dict, PATH::String, LIBPATH::String, adjust_PATH::Bool, adjust_LIBPATH::Bool) |
| 30 | + if adjust_LIBPATH |
| 31 | + LIBPATH_base = get(env, LIBPATH_env, expanduser(LIBPATH_default)) |
| 32 | + if !isempty(LIBPATH_base) |
| 33 | + env[LIBPATH_env] = string(LIBPATH, pathsep, LIBPATH_base) |
| 34 | + else |
| 35 | + env[LIBPATH_env] = LIBPATH |
| 36 | + end |
| 37 | + end |
| 38 | + if adjust_PATH && (LIBPATH_env != "PATH" || !adjust_LIBPATH) |
| 39 | + if !isempty(get(env, "PATH", "")) |
| 40 | + env["PATH"] = string(PATH, pathsep, env["PATH"]) |
| 41 | + else |
| 42 | + env["PATH"] = PATH |
| 43 | + end |
| 44 | + end |
| 45 | + return env |
| 46 | +end |
| 47 | + |
| 48 | +function __init_lld_path() |
| 49 | + # Prefer our own bundled lld, but if we don't have one, pick it up off of the PATH |
| 50 | + # If this is an in-tree build, `lld` will live in `tools`. Otherwise, it'll be in `libexec` |
| 51 | + for bundled_lld_path in (joinpath(Sys.BINDIR, Base.LIBEXECDIR, lld_exe), |
| 52 | + joinpath(Sys.BINDIR, "..", "tools", lld_exe), |
| 53 | + joinpath(Sys.BINDIR, lld_exe)) |
| 54 | + if isfile(bundled_lld_path) |
| 55 | + lld_path[] = abspath(bundled_lld_path) |
| 56 | + return |
| 57 | + end |
| 58 | + end |
| 59 | + lld_path[] = something(Sys.which(lld_exe), lld_exe) |
| 60 | + return |
| 61 | +end |
| 62 | + |
| 63 | +const VERBOSE = Ref{Bool}(false) |
| 64 | + |
| 65 | +function __init__() |
| 66 | + VERBOSE[] = parse(Bool, get(ENV, "JULIA_VERBOSE_LINKING", "false")) |
| 67 | + |
| 68 | + __init_lld_path() |
| 69 | + PATH[] = dirname(lld_path[]) |
| 70 | + if Sys.iswindows() |
| 71 | + # On windows, the dynamic libraries (.dll) are in Sys.BINDIR ("usr\\bin") |
| 72 | + append!(LIBPATH_list, [abspath(Sys.BINDIR, Base.LIBDIR, "julia"), Sys.BINDIR]) |
| 73 | + else |
| 74 | + append!(LIBPATH_list, [abspath(Sys.BINDIR, Base.LIBDIR, "julia"), abspath(Sys.BINDIR, Base.LIBDIR)]) |
| 75 | + end |
| 76 | + LIBPATH[] = join(LIBPATH_list, pathsep) |
| 77 | + return |
| 78 | +end |
| 79 | + |
| 80 | +function lld(; adjust_PATH::Bool = true, adjust_LIBPATH::Bool = true) |
| 81 | + env = adjust_ENV!(copy(ENV), PATH[], LIBPATH[], adjust_PATH, adjust_LIBPATH) |
| 82 | + return Cmd(Cmd([lld_path[]]); env) |
| 83 | +end |
| 84 | + |
| 85 | +function ld() |
| 86 | + default_args = `` |
| 87 | + @static if Sys.iswindows() |
| 88 | + # LLD supports mingw style linking |
| 89 | + flavor = "gnu" |
| 90 | + m = Sys.ARCH == :x86_64 ? "i386pep" : "i386pe" |
| 91 | + default_args = `-m $m -Bdynamic --enable-auto-image-base --allow-multiple-definition` |
| 92 | + elseif Sys.isapple() |
| 93 | + flavor = "darwin" |
| 94 | + arch = Sys.ARCH == :aarch64 ? :arm64 : Sys.ARCH |
| 95 | + default_args = `-arch $arch -undefined dynamic_lookup -platform_version macos $(Base.MACOS_PRODUCT_VERSION) $(Base.MACOS_PLATFORM_VERSION)` |
| 96 | + else |
| 97 | + flavor = "gnu" |
| 98 | + end |
| 99 | + |
| 100 | + `$(lld()) -flavor $flavor $default_args` |
| 101 | +end |
| 102 | + |
| 103 | +const WHOLE_ARCHIVE = if Sys.isapple() |
| 104 | + "-all_load" |
| 105 | +else |
| 106 | + "--whole-archive" |
| 107 | +end |
| 108 | + |
| 109 | +const NO_WHOLE_ARCHIVE = if Sys.isapple() |
| 110 | + "" |
| 111 | +else |
| 112 | + "--no-whole-archive" |
| 113 | +end |
| 114 | + |
| 115 | +const SHARED = if Sys.isapple() |
| 116 | + "-dylib" |
| 117 | +else |
| 118 | + "-shared" |
| 119 | +end |
| 120 | + |
| 121 | +is_debug() = ccall(:jl_is_debugbuild, Cint, ()) == 1 |
| 122 | +libdir() = abspath(Sys.BINDIR, Base.LIBDIR) |
| 123 | +private_libdir() = abspath(Sys.BINDIR, Base.PRIVATE_LIBDIR) |
| 124 | +if Sys.iswindows() |
| 125 | + shlibdir() = Sys.BINDIR |
| 126 | +else |
| 127 | + shlibdir() = libdir() |
| 128 | +end |
| 129 | + |
| 130 | +function link_image_cmd(path, out) |
| 131 | + LIBDIR = "-L$(libdir())" |
| 132 | + PRIVATE_LIBDIR = "-L$(private_libdir())" |
| 133 | + SHLIBDIR = "-L$(shlibdir())" |
| 134 | + LIBS = is_debug() ? ("-ljulia-debug", "-ljulia-internal-debug") : ("-ljulia", "-ljulia-internal") |
| 135 | + @static if Sys.iswindows() |
| 136 | + LIBS = (LIBS..., "-lopenlibm", "-lssp", "-lgcc_s", "-lgcc", "-lmsvcrt") |
| 137 | + end |
| 138 | + |
| 139 | + V = VERBOSE[] ? "--verbose" : "" |
| 140 | + `$(ld()) $V $SHARED -o $out $WHOLE_ARCHIVE $path $NO_WHOLE_ARCHIVE $LIBDIR $PRIVATE_LIBDIR $SHLIBDIR $LIBS` |
| 141 | +end |
| 142 | + |
| 143 | +function link_image(path, out, internal_stderr::IO = stderr, internal_stdout::IO = stdout) |
| 144 | + run(link_image_cmd(path, out), Base.DevNull(), stderr, stdout) |
| 145 | +end |
| 146 | + |
| 147 | +end # module Linking |
0 commit comments