use std::env;
use std::path::{Path, PathBuf};
use std::process::Command;
fn main() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").expect("failed to get current directory");
println!("cargo:include={manifest_dir}/vendor/rdma-core/build/include");
println!("cargo:rustc-link-search=native={manifest_dir}/vendor/rdma-core/build/lib");
println!("cargo:rustc-link-lib=ibverbs");
if Path::new("vendor/rdma-core/CMakeLists.txt").exists() {
} else 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"
);
}
eprintln!("run cmake");
let built_in = cmake::Config::new("vendor/rdma-core")
.define("NO_MAN_PAGES", "1")
.define("CMAKE_INSTALL_PREFIX", "/usr")
.no_build_target(true)
.build();
let built_in = built_in
.to_str()
.expect("build directory path is not valid UTF-8");
eprintln!("run bindgen");
let bindings = bindgen::Builder::default()
.header("vendor/rdma-core/libibverbs/verbs.h")
.clang_arg(format!("-I{built_in}/include/"))
.allowlist_function("ibv_.*")
.allowlist_type("ibv_.*")
.allowlist_var("IBV_LINK_LAYER_.*")
.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)
.blocklist_type("ibv_wc")
.size_t_is_usize(true)
.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");
}