arnaucube/miden-crypto
1
0
Fork 0
mirror of https://github.com/arnaucube/miden-crypto.git synced 2026-01-12 00:51:29 +01:00
Code Issues Projects Releases Wiki Activity

Merge pull request #113 from 0xPolygonMiden/hacka-merkle-store-fix-empty-roots

Browse Source 
bugfix: fix internal nodes of for empty leafs of a SMT
This commit is contained in:
Augusto Hack
2023-03-24 23:26:48 +01:00
committed by GitHub
parent bf15e1331a 9e6c8ff700
commit 75af3d474b
2 changed files with 80 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/merkle/store/mod.rs
View File

@@ -44,7 +44,7 @@ pub struct Node {
/// let mut store = MerkleStore::new();
///
/// // the store is initialized with the SMT empty nodes
/// assert_eq!(store.num_internal_nodes(), 64);
/// assert_eq!(store.num_internal_nodes(), 255);
///
/// // populates the store with two merkle trees, common nodes are shared
/// store.add_merkle_tree([A, B, C, D, E, F, G, H0]);
@@ -66,7 +66,7 @@ pub struct Node {
///
/// // Common internal nodes are shared, the two added trees have a total of 30, but the store has
/// // only 10 new entries, corresponding to the 10 unique internal nodes of these trees.
/// assert_eq!(store.num_internal_nodes() - 64, 10);
/// assert_eq!(store.num_internal_nodes() - 255, 10);
/// ```
#[derive(Debug, Clone, Eq, PartialEq)]
pub struct MerkleStore {
@@ -86,11 +86,12 @@ impl MerkleStore {
/// Creates an empty `MerkleStore` instance.
pub fn new() -> MerkleStore {
// pre-populate the store with the empty hashes
let subtrees = EmptySubtreeRoots::empty_hashes(64);
let subtrees = EmptySubtreeRoots::empty_hashes(255);
let nodes = subtrees
.iter()
.rev()
.copied()
.zip(subtrees.iter().skip(1).copied())
.zip(subtrees.iter().rev().skip(1).copied())
.map(|(child, parent)| {
(
parent,