73 lines
2.5 KiB
Rust
73 lines
2.5 KiB
Rust
use std::env;
|
|
use std::path::PathBuf;
|
|
|
|
fn main() {
|
|
let libdir_path = PathBuf::from("mlkem-native")
|
|
.canonicalize()
|
|
.expect("cannot canonicalize path");
|
|
|
|
let headers_path = libdir_path.join("mlkem/mlkem_native.h");
|
|
let headers_path_str = headers_path.to_str().expect("Path is not a valid string");
|
|
|
|
println!(
|
|
"cargo:rustc-link-search={}",
|
|
libdir_path.join("test/build").to_str().unwrap()
|
|
);
|
|
|
|
println!("cargo:rustc-link-lib=mlkem512");
|
|
println!("cargo:rustc-link-lib=mlkem768");
|
|
println!("cargo:rustc-link-lib=mlkem1024");
|
|
|
|
let mlkem_native_path = PathBuf::from("mlkem-native")
|
|
.canonicalize()
|
|
.expect("cannot canonicalize path");
|
|
if !make_cmd::make()
|
|
.current_dir(mlkem_native_path)
|
|
.output()
|
|
.expect("could not compile mlkem-native")
|
|
.status
|
|
.success()
|
|
{
|
|
panic!("could not compile mlkem-native");
|
|
}
|
|
|
|
let bindings_level2 = bindgen::Builder::default()
|
|
.header(headers_path_str)
|
|
.clang_arg("-DMLKEM_K=2")
|
|
.blocklist_function("PQCP_MLKEM_NATIVE_MLKEM512_keypair|PQCP_MLKEM_NATIVE_MLKEM512_enc")
|
|
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
|
|
.generate()
|
|
.expect("Unable to generate bindings");
|
|
|
|
let bindings_level3 = bindgen::Builder::default()
|
|
.header(headers_path_str)
|
|
.clang_arg("-DMLKEM_K=3")
|
|
.blocklist_function("PQCP_MLKEM_NATIVE_MLKEM768_keypair|PQCP_MLKEM_NATIVE_MLKEM768_enc")
|
|
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
|
|
.generate()
|
|
.expect("Unable to generate bindings");
|
|
|
|
let bindings_level4 = bindgen::Builder::default()
|
|
.header(headers_path_str)
|
|
.clang_arg("-DMLKEM_K=4")
|
|
.blocklist_function("PQCP_MLKEM_NATIVE_MLKEM1024_keypair|PQCP_MLKEM_NATIVE_MLKEM1024_enc")
|
|
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
|
|
.generate()
|
|
.expect("Unable to generate bindings");
|
|
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("bindings_level2.rs");
|
|
bindings_level2
|
|
.write_to_file(out_path)
|
|
.expect("Couldn't write bindings!");
|
|
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("bindings_level3.rs");
|
|
bindings_level3
|
|
.write_to_file(out_path)
|
|
.expect("Couldn't write bindings!");
|
|
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("bindings_level4.rs");
|
|
bindings_level4
|
|
.write_to_file(out_path)
|
|
.expect("Couldn't write bindings!");
|
|
}
|