Make dylib_flag cross-platform

Cargo.toml

@@ -21,6 +21,7 @@ anyhow = "1.0.65"
 clap = { version = "4.0.29", features = ["derive"] }
 ctrlc = "3.2.5"
 genemichaels = "0.1.21"
+libloading = "0.8.0"
 owo-colors = "3.5.0"
 proc-macro2 = { version = "1.0.48", features = ["span-locations"] }
 quote = "1.0.23"
@@ -33,6 +34,3 @@ tracing = "0.1.37"
 tracing-subscriber = { version = "0.3.16", features = ["env-filter"] }
 tracing-tree = "0.2.2"
 walkdir = "2.3.2"
-
-[target."cfg(unix)".dependencies]
-libc = "0.2.138"

src/dylib_flag.rs

@@ -1,9 +1,10 @@
 //! Handles the --verify-fn flag.
 //! It takes in a Rust closure like `|str| true` that takes in a `&str` and returns a bool.
 
-use std::{fmt::Debug, str::FromStr};
+use std::{fmt::Debug, mem::ManuallyDrop, str::FromStr};
 
 use anyhow::{Context, Result};
+use libloading::Symbol;
 
 #[repr(C)]
 pub struct RawOutput {
@@ -24,7 +25,7 @@ impl FromStr for RustFunction {
     type Err = anyhow::Error;
 
     fn from_str(s: &str) -> Result<Self, Self::Err> {
-        Self::compile(s)
+        Self::compile(s).context("compiling and loading rust function")
     }
 }
 
@@ -83,17 +84,9 @@ fn wrap_func_body(func: &str) -> Result<String> {
 }
 
 impl RustFunction {
-    #[cfg(not(target_os = "linux"))]
-    pub fn compile(body: &str) -> Result<Self> {
-        Err(anyhow::anyhow!("--verify-fn only works on unix"))
-    }
-
-    #[cfg(target_os = "linux")]
     pub fn compile(body: &str) -> Result<Self> {
         use anyhow::bail;
-        use std::io;
         use std::process::Command;
-        use std::{ffi::CString, os::unix::prelude::OsStringExt};
 
         let file = tempfile::tempdir()?;
 
@@ -114,29 +107,20 @@ impl RustFunction {
             bail!("Failed to compile code: {stderr}");
         }
 
-        let dylib_path = file.path().join("libhelper.so");
-
-        let os_str = dylib_path.into_os_string();
-        let vec = os_str.into_vec();
-        let cstr = CString::new(vec)?;
-
-        let dylib = unsafe { libc::dlopen(cstr.as_ptr(), libc::RTLD_LAZY) };
-
-        if dylib.is_null() {
-            bail!("failed to open dylib: {}", io::Error::last_os_error());
-        }
-
-        let symbol = b"cargo_minimize_ffi_function\0";
-
-        let func = unsafe { libc::dlsym(dylib, symbol.as_ptr().cast()) };
-
-        if func.is_null() {
-            bail!("didn't find entrypoint symbol");
-        }
+        // SAFETY: We are loading a simple rust cdylib, which does not do weird things. But we cannot unload Rust dylibs, so we use MD below.
+        let dylib = unsafe {
+            libloading::Library::new(file.path().join(libloading::library_filename("helper")))
+                .context("loading helper shared library")?
+        };
+        let dylib = ManuallyDrop::new(dylib);
 
-        let func = unsafe { std::mem::transmute::<*mut _, CheckerCFn>(func) };
+        let func: Symbol<CheckerCFn> = unsafe {
+            dylib
+                .get(b"cargo_minimize_ffi_function\0")
+                .context("failed to find entrypoint symbol")?
+        };
 
-        Ok(Self { func })
+        Ok(Self { func: *func })
     }
 
     pub fn call(&self, output: &str, code: Option<i32>) -> bool {
@@ -169,7 +153,6 @@ mod tests {
     use super::RustFunction;
 
     #[test]
-    #[cfg_attr(not(target_os = "linux"), ignore)]
     fn basic_contains_work() {
         let code = r#"|output| output.out.contains("test")"#;
 

src/processor/checker.rs

@@ -198,7 +198,8 @@ impl PassController {
     fn next_in_worklist(&mut self) {
         let PassControllerState::Bisecting {
             current, worklist, ..
-        } = &mut self.state else {
+        } = &mut self.state
+        else {
             unreachable!("next_in_worklist called on non-bisecting state");
         };
         match worklist.pop() {

src/processor/reaper.rs

@@ -62,12 +62,11 @@ impl Minimizer {
         suggestions: &HashMap<&Path, Vec<&Suggestion>>,
     ) -> Result<()> {
         for (sugg_file, suggestions) in suggestions {
-            let Some(file) = self
-                .files
-                .iter()
-                .find(|source| source.path.ends_with(sugg_file) || sugg_file.ends_with(&source.path)) else {
-                    continue;
-                };
+            let Some(file) = self.files.iter().find(|source| {
+                source.path.ends_with(sugg_file) || sugg_file.ends_with(&source.path)
+            }) else {
+                continue;
+            };
 
             let changes = &mut Changes::default();