WIP add new sk test

Fraunhofer-AISEC · Sep 6, 2024 · 8b66ae4 · 8b66ae4
1 parent c464898
commit 8b66ae4
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/src/hss/reference_impl_private_key.rs b/src/hss/reference_impl_private_key.rs
@@ -389,6 +389,7 @@ impl CompressedUsedLeafsIndexes {
 #[cfg(test)]
 mod tests {
     use super::{CompressedParameterSet, ReferenceImplPrivateKey};
+    use crate::sst::parameters::SstsParameter;
     use crate::util::helper::test_helper::gen_random_seed;
     use crate::{
         constants::MAX_ALLOWED_HSS_LEVELS, hss::definitions::HssPrivateKey, HssParameter,
@@ -398,6 +399,36 @@ mod tests {
 
     type Hasher = Sha256_256;
 
+    #[test]
+    fn generate_sst_impl_private_key() {
+        const L0_TOP_DIV: u32 = 4;
+        const SIGNING_ENTITY_IDX: u32 = 3;
+
+        let seed = gen_random_seed::<Hasher>();
+        let mut hss_parameters = ArrayVec::new();
+        hss_parameters.extend_from_slice(&[
+            HssParameter::construct_default_parameters(),
+            HssParameter::construct_default_parameters(),
+        ]);
+        let ssts_parameter = SstsParameter::<Hasher>::new(
+            hss_parameters,
+            L0_TOP_DIV as u8,
+            SIGNING_ENTITY_IDX as u8,
+        );
+        let impl_private_key = ReferenceImplPrivateKey::generate(
+            ssts_parameter.get_hss_parameters(),
+            &seed,
+            Some(ssts_parameter.sst_extension().clone()),
+        )
+        .unwrap();
+        assert_eq!(
+            impl_private_key
+                .compressed_used_leafs_indexes
+                .to(&hss_parameters)[0],
+            2u32.pow(SIGNING_ENTITY_IDX - 1)
+        );
+    }
+
     #[test]
     fn exhaust_state() {
         let lmots = LmotsAlgorithm::LmotsW4;