59 lines
2.1 KiB
Rust
59 lines
2.1 KiB
Rust
use make_cmd;
|
|
use std::env;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
fn main() {
|
|
// This is the directory where the `c` library is located.
|
|
let libdir_path = PathBuf::from("mlkem-native/mlkem")
|
|
// Canonicalize the path as `rustc-link-search` requires an absolute
|
|
// path.
|
|
.canonicalize()
|
|
.expect("cannot canonicalize path");
|
|
|
|
// This is the path to the `c` headers file.
|
|
let headers_path = libdir_path.join("mlkem_native.h");
|
|
let headers_path_str = headers_path.to_str().expect("Path is not a valid string");
|
|
|
|
// Tell cargo to look for shared libraries in the specified directory
|
|
println!("cargo:rustc-link-search={}", libdir_path.to_str().unwrap());
|
|
|
|
// Tell cargo to tell rustc to link our `hello` library. Cargo will
|
|
// automatically know it must look for a `libhello.a` file.
|
|
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");
|
|
}
|
|
|
|
// The bindgen::Builder is the main entry point
|
|
// to bindgen, and lets you build up options for
|
|
// the resulting bindings.
|
|
let bindings = bindgen::Builder::default()
|
|
// The input header we would like to generate
|
|
// bindings for.
|
|
.header(headers_path_str)
|
|
// Tell cargo to invalidate the built crate whenever any of the
|
|
// included header files changed.
|
|
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
|
|
// Finish the builder and generate the bindings.
|
|
.generate()
|
|
// Unwrap the Result and panic on failure.
|
|
.expect("Unable to generate bindings");
|
|
|
|
// Write the bindings to the $OUT_DIR/bindings.rs file.
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap()).join("bindings.rs");
|
|
bindings
|
|
.write_to_file(out_path)
|
|
.expect("Couldn't write bindings!");
|
|
}
|