lms/definitions: Minor rework

Fraunhofer-AISEC · Sep 3, 2024 · 11d52e2 · 11d52e2
1 parent 382b3b6
commit 11d52e2
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/lms/definitions.rs b/src/lms/definitions.rs
@@ -49,15 +49,15 @@ impl<H: HashChain> LmsPrivateKey<H> {
 
     pub fn use_lmots_private_key(&mut self) -> Result<LmotsPrivateKey<H>, ()> {
         let number_of_lm_ots_keys = {
-            if let Some(my_sst_ext) = &self.sst_ext {
-                // our last leafs function returns 0..total_num-1, but here we need 1..total_num
-                1 + get_sst_last_leaf_idx(&self.lms_parameter, my_sst_ext)
+            if let Some(sst_ext) = &self.sst_ext {
+                // last leaf idx is total_num-1, hence add 1
+                get_sst_last_leaf_idx(&self.lms_parameter, sst_ext) as usize + 1
             } else {
-                self.lms_parameter.number_of_lm_ots_keys() as u32
+                self.lms_parameter.number_of_lm_ots_keys()
             }
         };
 
-        if self.used_leafs_index >= number_of_lm_ots_keys {
+        if self.used_leafs_index as usize >= number_of_lm_ots_keys {
             return Err(());
         }