speeds up dh multi encrypt by using chunks_exact

seanwatters · Feb 27, 2024 · cc45f06 · cc45f06
1 parent bbe49da
commit cc45f06
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -14,7 +14,7 @@ license = "MIT"
 name = "rgp"
 readme = "README.md"
 repository = "https://github.com/seanwatters/rgp"
-version = "0.3.1"
+version = "0.3.2"
 
 [dependencies]
 blake2 = "0.10.6"

diff --git a/src/crypto/modes/dh.rs b/src/crypto/modes/dh.rs
@@ -115,9 +115,9 @@ fn dh_encrypt_keys(
     let mut keys = vec![0u8; KEY_SIZE * keys_count];
 
     #[cfg(feature = "multi-thread")]
-    let chunks = keys.par_chunks_mut(KEY_SIZE);
+    let chunks = keys.par_chunks_exact_mut(KEY_SIZE);
     #[cfg(not(feature = "multi-thread"))]
-    let chunks = keys.chunks_mut(KEY_SIZE);
+    let chunks = keys.chunks_exact_mut(KEY_SIZE);
 
     chunks.enumerate().for_each(|(i, chunk)| {
         let mut key = GenericArray::from(priv_key.diffie_hellman(&pub_keys[i].into()).to_bytes());

diff --git a/src/crypto/modes/kem.rs b/src/crypto/modes/kem.rs
@@ -107,7 +107,7 @@ pub fn generate_kem_keys() -> ([u8; KEM_SECRET_KEY_SIZE], [u8; KEM_PUB_KEY_SIZE]
     (*secret_key.as_array(), *pub_key.as_array())
 }
 
-/// for reading a large volume of McEliece public keys
+/// for reading a large volume of Classic McEliece public keys
 pub struct KemKeyReader<R: Read> {
     pub reader: BufReader<R>,