make_merkle_tree

Author: behzadnouri

ledger/src/shred/merkle.rs

@@ -382,6 +382,24 @@ fn verify_merkle_proof(index: usize, node: Hash, merkle_branch: &MerkleBranch) -
     }) == (0, merkle_branch.root)
 }
 
+fn make_merkle_tree(nodes: Vec<Hash>) -> Vec<Hash> {
+    let mut tree = nodes;
+    let mut size = tree.len();
+    while size != 1 {
+        let offset = tree.len() - size;
+        for i in (offset..offset + size).step_by(2) {
+            let node = &tree[i];
+            let next = tree.get(i + 1).unwrap_or(&node);
+            let node = &node.as_ref()[..SIZE_OF_MERKLE_PROOF_ENTRY];
+            let next = &next.as_ref()[..SIZE_OF_MERKLE_PROOF_ENTRY];
+            let hash = hashv(&[MERKLE_HASH_PREFIX_NODE, node, next]);
+            tree.push(hash);
+        }
+        size = tree.len() - offset - size;
+    }
+    tree
+}
+
 #[cfg(test)]
 mod test {
     use super::*;