WIP minor

Fraunhofer-AISEC · Sep 6, 2024 · 5324ad7 · 5324ad7
1 parent 00990a0
commit 5324ad7
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/lms/helper.rs b/src/lms/helper.rs
@@ -1,6 +1,4 @@
-use tinyvec::ArrayVec;
-
-use crate::constants::{D_INTR, D_LEAF, MAX_HASH_SIZE};
+use crate::constants::{Node, D_INTR, D_LEAF, MAX_HASH_SIZE};
 use crate::hasher::HashChain;
 use crate::hss::aux::{hss_extract_aux_data, hss_save_aux_data, MutableExpandedAuxData};
 use crate::lm_ots;
@@ -11,8 +9,7 @@ pub fn get_tree_element<H: HashChain>(
     index: usize,
     private_key: &LmsPrivateKey<H>,
     aux_data: &mut Option<MutableExpandedAuxData>,
-) -> ArrayVec<[u8; MAX_HASH_SIZE]> {
-    // TODO/Rework: use "Node" struct as return value?
+) -> Node {
     // Check if we already have the value cached
     if let Some(aux_data) = aux_data {
         if let Some(result) = hss_extract_aux_data::<H>(aux_data, index) {