fix build

1 files changed, 15 insertions, 4 deletions

diff --git a/build.rs b/build.rs
index 1c29bfb..0f01f5f 100644
--- a/build.rs
+++ b/build.rs
@@ -1,6 +1,17 @@
+use std::env;
+
 fn main() {
-    println!("{:?}", std::env::var("LIBPCAP_LIBDIR"));
-    println!("cargo:rustc-link-arg=-lc");
-    println!("cargo:rustc-link-arg=-lnl-3");
-    println!("cargo:rustc-link-arg=-lnl-genl-3");
+    // #[cfg(arch = "x86_64")]
+    // println!("cargo:rustc-link-search=./libpcap/x86_64");
+
+    // #[cfg(arch = "aarch64")]
+    // println!("cargo:rustc-link-search=./libpcap/rpi");
+
+    match env::var("CARGO_CFG_TARGET_ARCH").unwrap_or(String::new()).as_str() {
+        "x86_64" => println!("cargo:rustc-link-search=./libpcap/x86_64"),
+        "aarch64" => println!("cargo:rustc-link-search=./libpcap/rpi"),
+        _ => println!("cargo:warning=Building for unsupported architecture, library search path may be incorrect"),
+    }
+
+    println!("cargo:rustc-link-lib=static=pcap");
 }