Files
mlkem-native-rs/build.rs
2025-03-15 13:59:20 +01:00

58 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=mlkem-native/test/build");
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!");
}