rixxc/mlkem-native-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add bindings for level 2 and 4; generate randomness from provided CSRNG

Browse Source
This commit is contained in:
Aaron Kaiser
2025-03-19 11:07:26 +01:00
parent 6f3e5163b6
commit 7d0667d1c0
8 changed files with 285 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
build.rs
View File

@@ -2,25 +2,19 @@ use std::env;
use std::path::PathBuf;
fn main() {
// This is the directory where the `c` library is located.
let libdir_path = PathBuf::from("mlkem-native")
// 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/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.join("test/build").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=mlkem512");
println!("cargo:rustc-link-lib=mlkem768");
println!("cargo:rustc-link-lib=mlkem1024");
@@ -37,24 +31,39 @@ fn main() {
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.
let bindings_level2 = bindgen::Builder::default()
.header(headers_path_str)
// Tell cargo to invalidate the built crate whenever any of the
// included header files changed.
.clang_arg("-DMLKEM_K=2")
.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
let bindings_level3 = bindgen::Builder::default()
.header(headers_path_str)
.clang_arg("-DMLKEM_K=3")
.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")
.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!");
}