extern crate bindgen;
use std::env;
use std::path::{Path, PathBuf};
use std::process::Command;
fn main() {
println!("cargo:include=vendor/rdma-core/build/include");
println!("cargo:rustc-link-search=native=vendor/rdma-core/build/lib");
println!("cargo:rustc-link-lib=ibverbs");
if Path::new(".git").is_dir() {
Command::new("git")
.args(&["submodule", "update", "--init"])
.status()
.expect("Failed to update submodules.");
} else {
assert!(
Path::new("vendor/rdma-core").is_dir(),
"vendor source not included"
);
}
Command::new("bash")
.current_dir("vendor/rdma-core/")
.args(&["build.sh"])
.status()
.expect("Failed to build vendor/rdma-core using build.sh");
let bindings = bindgen::Builder::default()
.header("vendor/rdma-core/libibverbs/verbs.h")
.clang_arg("-Ivendor/rdma-core/build/include/")
.blacklist_type("max_align_t")
.whitelist_function("ibv_.*")
.whitelist_type("ibv_.*")
.bitfield_enum("ibv_access_flags")
.bitfield_enum("ibv_qp_attr_mask")
.bitfield_enum("ibv_wc_flags")
.bitfield_enum("ibv_send_flags")
.bitfield_enum("ibv_port_cap_flags")
.constified_enum_module("ibv_qp_type")
.constified_enum_module("ibv_qp_state")
.constified_enum_module("ibv_port_state")
.constified_enum_module("ibv_wc_opcode")
.constified_enum_module("ibv_wr_opcode")
.constified_enum_module("ibv_wc_status")
.derive_default(true)
.derive_debug(true)
.prepend_enum_name(false)
.blacklist_type("ibv_wc")
.generate()
.expect("Unable to generate bindings");
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
bindings
.write_to_file(out_path.join("bindings.rs"))
.expect("Could not write bindings");
}