Fix DPF PRG key num

myl7 · Jan 28, 2024 · c557cb3 · c557cb3
1 parent 09512e5
commit c557cb3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/dpf/mod.rs b/src/dpf/mod.rs
@@ -165,10 +165,8 @@ mod tests {
     use crate::group::byte::ByteGroup;
     use crate::prg::Aes256HirosePrg;
 
-    const KEYS: &[&[u8; 32]] = &[
-        b"j9\x1b_\xb3X\xf33\xacW\x15\x1b\x0812K\xb3I\xb9\x90r\x1cN\xb5\xee9W\xd3\xbb@\xc6d",
-        b"\x9b\x15\xc8\x0f\xb7\xbc!q\x9e\x89\xb8\xf7\x0e\xa0S\x9dN\xfa\x0c;\x16\xe4\x98\x82b\xfcdy\xb5\x8c{\xc2",
-    ];
+    const KEYS: &[&[u8; 32]] =
+        &[b"j9\x1b_\xb3X\xf33\xacW\x15\x1b\x0812K\xb3I\xb9\x90r\x1cN\xb5\xee9W\xd3\xbb@\xc6d"];
     const ALPHAS: &[&[u8; 16]] = &[
         b"K\xa9W\xf5\xdd\x05\xe9\xfc?\x04\xf6\xfbUo\xa8C",
         b"\xc2GK\xda\xc6\xbb\x99\x98Fq\"f\xb7\x8csU",

diff --git a/src/prg.rs b/src/prg.rs
@@ -18,6 +18,8 @@ pub struct Aes128MatyasMeyerOseasPrg {
 }
 
 impl Aes128MatyasMeyerOseasPrg {
+    /// `keys` length MUST be the output size divided by 16.
+    /// Otherwise the runtime size check would fail and panic.
     pub fn new(keys: &[&[u8; 16]]) -> Self {
         Self {
             ciphers: keys
@@ -70,6 +72,8 @@ pub struct Aes256HirosePrg {
 }
 
 impl Aes256HirosePrg {
+    /// `keys` length MUST be the output size divided by 32.
+    /// Otherwise the runtime size check would fail and panic.
     pub fn new(keys: &[&[u8; 32]]) -> Self {
         Self {
             ciphers: keys