ML DSA

Cargo.toml

@@ -14,10 +14,11 @@ categories = ["cryptography", "post-quantum-cryptography"]
 
 [dependencies]
 rand = "0.7.3"
+sha2 = "0.10.8"
 
 [dev-dependencies]
 criterion = { version = "0.4", features = ["html_reports"] }
 
 [[bench]]
 name = "dilithium_benchmarks"
-harness = false
+harness = false

benches/dilithium_benchmarks.rs

@@ -3,7 +3,7 @@
 use criterion::{Criterion, criterion_group};
 
 mod dilithium_benches {
-    use dilithium::dilithium2::Keypair;
+    use crystals_dilithium::dilithium2::Keypair;
 
     use super::*;
 

src/lib.rs

@@ -1,3 +1,6 @@
+pub mod ml_dsa_44;
+pub mod ml_dsa_65;
+pub mod ml_dsa_87;
 pub mod dilithium2;
 pub mod dilithium3;
 pub mod dilithium5;
@@ -11,6 +14,24 @@ pub mod rounding;
 pub mod reduce;
 pub mod sign;
 
+pub enum PH {
+    SHA256,
+    SHA512,
+}
+
+use rand::RngCore;
+/// Generate random bytes.
+/// 
+/// # Arguments
+/// 
+/// * 'bytes' - an array to fill with random data
+/// * 'n' - number of bytes to generate
+fn random_bytes(bytes: &mut [u8], n: usize) {
+    rand::prelude::thread_rng()
+        .try_fill_bytes(&mut bytes[..n])
+        .unwrap();
+}
+
 #[cfg(test)]
 mod tests {
     #[test]