add options to allow generating statically linked binaries

Signed-off-by: Jorge Prendes <[email protected]>
WasmEdge · Aug 1, 2023 · e6d89a7 · e6d89a7
1 parent d8e8f30
commit e6d89a7
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 11 deletions.
diff --git a/crates/wasmedge-sys/build.rs b/crates/wasmedge-sys/build.rs
@@ -87,11 +87,9 @@ fn main() {
         // Tell cargo to tell rustc to link our `wasmedge` library. Cargo will
         // automatically know it must look for a `libwasmedge.a` file.
         println!("cargo:rustc-link-lib=static=wasmedge");
-        println!("cargo:rustc-link-lib=rt");
-        println!("cargo:rustc-link-lib=dl");
-        println!("cargo:rustc-link-lib=pthread");
-        println!("cargo:rustc-link-lib=m");
-        println!("cargo:rustc-link-lib=stdc++");
+        for dep in ["rt", "dl", "pthread", "m", "stdc++"] {
+            link_lib(dep);
+        }
     } else {
         println!("cargo:rustc-env=LD_LIBRARY_PATH={lib_dir}");
         println!("cargo:rustc-link-search={lib_dir}");
@@ -119,9 +117,37 @@ fn main() {
         .expect("failed to write bindings");
 }
 
+fn link_lib(dep: &str) {
+    // Sanitize dependency name for evn-vars, particularly `stdc++`.
+    let dep_slug: String = dep.replace('+', "x").to_uppercase();
+
+    let generic_link_type_var = Env!("WASMEDGE_DEPS_LINK_TYPE");
+    let generic_lib_path_var = Env!("WASMEDGE_DEPS_LIB_PATH");
+    let named_link_type_var = Env!("WASMEDGE_DEP_{dep_slug}_LINK_TYPE");
+    let named_lib_path_var = Env!("WASMEDGE_DEP_{dep_slug}_LIB_PATH");
+
+    let link_type = named_link_type_var
+        .lossy()
+        .or_else(|| generic_link_type_var.lossy())
+        .unwrap_or("dylib".to_string());
+
+    for path_var in [named_lib_path_var, generic_lib_path_var] {
+        if let Some(path) = path_var.lossy() {
+            println!("cargo:rustc-link-search={path}");
+        }
+    }
+
+    println!("cargo:rustc-link-lib={link_type}={dep}");
+}
+
 #[macro_export]
 macro_rules! debug {
     ($($args:expr),+) => {
         println!("cargo:warning=[wasmedge-sys] {}", format!($($args),+))
     };
 }
+
+#[macro_export]
+macro_rules! Env {
+    ($($args:expr),+) => { Env(format!($($args),+)) };
+}
diff --git a/crates/wasmedge-sys/build_paths.rs b/crates/wasmedge-sys/build_paths.rs
@@ -1,10 +1,13 @@
 use crate::debug;
 
-pub struct Env<'a>(pub &'a str);
+pub struct Env<S: AsRef<str>>(pub S);
 
-impl Env<'_> {
+impl<S: AsRef<str>> Env<S> {
     pub fn read<T: From<std::ffi::OsString>>(&self) -> Option<T> {
-        std::env::var_os(self.0).map(T::from)
+        if self.0.as_ref() != "OUT_DIR" && !self.0.as_ref().starts_with("CARGO_") {
+            println!("cargo:rerun-if-env-changed={}", self.0.as_ref());
+        }
+        std::env::var_os(self.0.as_ref()).map(T::from)
     }
 
     pub fn expect<T: TryFrom<std::ffi::OsString>>(&self, msg: &str) -> T
@@ -31,9 +34,8 @@ pub trait AsPath {
     fn as_path(&self) -> Option<std::path::PathBuf>;
 }
 
-impl AsPath for Env<'_> {
+impl<S: AsRef<str>> AsPath for Env<S> {
     fn as_path(&self) -> Option<std::path::PathBuf> {
-        println!("cargo:rerun-if-env-changed={}", self.0);
         self.read()
     }
 }

diff --git a/crates/wasmedge-sys/build_standalone.rs b/crates/wasmedge-sys/build_standalone.rs
@@ -108,7 +108,7 @@ fn get_remote_archive() -> Archive {
 
 fn do_http_request(url: &str) -> impl std::io::Read {
     let builder = reqwest::blocking::Client::builder();
-    let builder = match Env("STANDALONE_PROXY").lossy() {
+    let builder = match Env("WASMEDGE_STANDALONE_PROXY").lossy() {
         Some(proxy) => {
             debug!("using proxy to download archive: {proxy}");
             let proxy = reqwest::Proxy::all(proxy).expect("failed to parse proxy");