Oops forgot to update KAT and example

rozbb · Nov 11, 2024 · dbdeb08 · dbdeb08
1 parent f582c84
commit dbdeb08
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/examples/agility.rs b/examples/agility.rs
@@ -702,10 +702,7 @@ fn main() {
                 let psk_id = b"preshared key attempt #5, take 2. action";
                 let psk_bundle = {
                     csprng.fill_bytes(&mut psk_bytes);
-                    AgilePskBundle(PskBundle {
-                        psk: &psk_bytes,
-                        psk_id,
-                    })
+                    AgilePskBundle(PskBundle::new(&psk_bytes, psk_id).unwrap())
                 };
 
                 // Make two agreeing OpModes (AuthPsk is the most complicated, so we're just using

diff --git a/src/kat_tests.rs b/src/kat_tests.rs
@@ -222,10 +222,7 @@ fn make_op_mode_r<'a, Kem: KemTrait>(
     psk_id: Option<&'a [u8]>,
 ) -> OpModeR<'a, Kem> {
     // Deserialize the optional bundle
-    let bundle = psk.map(|bytes| PskBundle {
-        psk: bytes,
-        psk_id: psk_id.unwrap(),
-    });
+    let bundle = psk.map(|bytes| PskBundle::new(bytes, psk_id.unwrap()).unwrap());
 
     // These better be set if the mode ID calls for them
     match mode_id {