[Products] Don't dlopen libraries built for an incompatible ISA

Author: giordano

Project.toml

@@ -1,7 +1,7 @@
 name = "BinaryBuilderBase"
 uuid = "7f725544-6523-48cd-82d1-3fa08ff4056e"
 authors = ["Elliot Saba <[email protected]>"]
-version = "1.8.0"
+version = "1.8.1"
 
 [deps]
 CodecZlib = "944b1d66-785c-5afd-91f1-9de20f533193"

src/Products.jl

@@ -222,7 +222,7 @@ function locate(lp::LibraryProduct, prefix::Prefix; platform::AbstractPlatform =
                     end
 
                     # If it does, try to `dlopen()` it if the current platform is good
-                    if (!lp.dont_dlopen && !skip_dlopen) && platforms_match(platform, HostPlatform())
+                    if (!lp.dont_dlopen && !skip_dlopen) && platforms_match(platform, augment_microarchitecture!(HostPlatform()))
                         if isolate
                             # Isolated dlopen is a lot slower, but safer
                             if success(`$(Base.julia_cmd()) --startup-file=no -e "import Libdl; Libdl.dlopen(\"$dl_path\")"`)

src/utils.jl

@@ -1,6 +1,9 @@
 # The functions in this file may not be used anywhere in this package but may
 # needed by different modules of BinaryBuilder.jl
 
+using Base.BinaryPlatforms: arch_march_isa_mapping, set_compare_strategy!
+using Base.BinaryPlatforms.CPUID
+
 with_logfile(f::Function, prefix::Prefix, name::String; subdir="") = with_logfile(f, joinpath(logdir(prefix; subdir), name))
 function with_logfile(f::Function, logfile::String)
     mkpath(dirname(logfile))
@@ -60,3 +63,45 @@ end
 # We want the AnyPlatform to look like `default_host_platform`,
 get_concrete_platform(::AnyPlatform, shards::Vector{CompilerShard}) =
     get_concrete_platform(default_host_platform, shards)
+
+function march_comparison_strategy(a::String, b::String, a_requested::Bool, b_requested::Bool)
+    # If both b and a requested, then we fall back to equality:
+    if a_requested && b_requested
+        return a == b
+    end
+
+    function get_arch_isa(isa_name::String)
+        for (arch, isas) in arch_march_isa_mapping
+            for (name, isa) in isas
+                name == isa_name && return arch, isa
+            end
+        end
+        return nothing, nothing
+    end
+
+    a_arch, a_isa = get_arch_isa(a)
+    b_arch, b_isa = get_arch_isa(b)
+    if any(isnothing, (a_arch, b_arch)) || a_arch != b_arch
+        # Architectures are definitely not compatible, exit early
+        return false
+    end
+
+    if a_requested
+        # ISA `b` is compatible with ISA `a` only if it's a subset of `a`
+        return b_isa ≤ a_isa
+    else
+        # ISA `a` is compatible with ISA `b` only if it's a subset of `b`
+        return a_isa ≤ b_isa
+    end
+end
+
+function augment_microarchitecture!(platform::Platform)
+    haskey(platform, "march") && return platform
+
+    host_arch = arch(HostPlatform())
+    host_isas = arch_march_isa_mapping[host_arch]
+    idx = findlast(((name, isa),) -> isa <= CPUID.cpu_isa(), host_isas)
+    platform["march"] = first(host_isas[idx])
+    set_compare_strategy!(platform, "march", march_comparison_strategy)
+    return platform
+end