113 lines
3.3 KiB
Rust
113 lines
3.3 KiB
Rust
use cmake::Config;
|
|
use dunce::canonicalize;
|
|
use std::env;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
pub enum LinkType {
|
|
Dynamic,
|
|
Static,
|
|
}
|
|
|
|
impl LinkType {
|
|
fn detect() -> LinkType {
|
|
if cfg!(feature = "static") {
|
|
LinkType::Static
|
|
} else {
|
|
LinkType::Dynamic
|
|
}
|
|
}
|
|
|
|
fn link_lib(&self) -> &'static str {
|
|
match self {
|
|
LinkType::Dynamic => "dylib=",
|
|
LinkType::Static => "static=",
|
|
}
|
|
}
|
|
|
|
fn target_name(&self) -> &'static str {
|
|
match self {
|
|
LinkType::Dynamic => "aeron_driver",
|
|
LinkType::Static => "aeron_driver_static",
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn main() {
|
|
println!("cargo:rerun-if-changed=build.rs");
|
|
println!("cargo:rerun-if-changed=bindings.h");
|
|
|
|
let aeron_path = canonicalize(Path::new("./aeron")).unwrap();
|
|
let header_path = aeron_path.join("aeron-driver/src/main/c");
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
|
|
|
|
let link_type = LinkType::detect();
|
|
println!(
|
|
"cargo:rustc-link-lib={}{}",
|
|
link_type.link_lib(),
|
|
link_type.target_name()
|
|
);
|
|
|
|
if let LinkType::Static = link_type {
|
|
// On Windows, there are some extra libraries needed for static link
|
|
// that aren't included by Aeron.
|
|
if cfg!(target_os = "windows") {
|
|
println!("cargo:rustc-link-lib=shell32");
|
|
println!("cargo:rustc-link-lib=iphlpapi");
|
|
}
|
|
}
|
|
|
|
let cmake_output = Config::new(&aeron_path)
|
|
.build_target(link_type.target_name())
|
|
.build();
|
|
|
|
// Trying to figure out the final path is a bit weird;
|
|
// For Linux/OSX, it's just build/lib
|
|
// For Windows, the .lib file is in build/lib/{profile}, but the DLL
|
|
// is shipped in build/binaries/{profile}
|
|
let base_lib_dir = cmake_output.join("build");
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
base_lib_dir.join("lib").display()
|
|
);
|
|
// Because the `cmake_output` path is different for debug/release, we're not worried
|
|
// about accidentally linking in the wrong library
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
base_lib_dir.join("lib/Debug").display()
|
|
);
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
base_lib_dir.join("binaries/Debug").display()
|
|
);
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
base_lib_dir.join("lib/Release").display()
|
|
);
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
base_lib_dir.join("binaries/Release").display()
|
|
);
|
|
|
|
println!("cargo:include={}", header_path.display());
|
|
let bindings = bindgen::Builder::default()
|
|
.clang_arg(format!("-I{}", header_path.display()))
|
|
.clang_arg(format!(
|
|
"-I{}",
|
|
aeron_path.join("aeron-client/src/main/c").display()
|
|
))
|
|
.header("bindings.h")
|
|
.whitelist_function("aeron_.*")
|
|
.whitelist_type("aeron_.*")
|
|
.whitelist_var("AERON_.*")
|
|
.constified_enum_module("aeron_.*_enum")
|
|
// Some padding structures use arrays > 120 elements,
|
|
// so we can't derive Debug implementations
|
|
.derive_debug(false)
|
|
.generate()
|
|
.expect("Unable to generate aeron_driver bindings");
|
|
|
|
bindings
|
|
.write_to_file(out_path.join("bindings.rs"))
|
|
.expect("Couldn't write bindings!");
|
|
}
|