smt: change SimpleSmt::open() to return a sparse path

This commit is contained in:
Qyriad 2025-04-04 15:14:27 +02:00
parent 0c436503e9
commit fadbc74bb4

View file

@ -1,5 +1,7 @@
use alloc::collections::BTreeSet;
use crate::merkle::{SparseMerklePath, SparseValuePath};
use super::{
super::ValuePath, EMPTY_WORD, EmptySubtreeRoots, InnerNode, InnerNodeInfo, InnerNodes,
LeafIndex, MerkleError, MerklePath, MutationSet, NodeIndex, RpoDigest, SMT_MAX_DEPTH,
@ -169,8 +171,15 @@ impl<const DEPTH: u8> SimpleSmt<DEPTH> {
/// Returns an opening of the leaf associated with `key`. Conceptually, an opening is a Merkle
/// path to the leaf, as well as the leaf itself.
pub fn open(&self, key: &LeafIndex<DEPTH>) -> ValuePath {
<Self as SparseMerkleTree<DEPTH>>::open(self, key)
pub fn open(&self, key: &LeafIndex<DEPTH>) -> SparseValuePath {
let value = RpoDigest::new(self.get_value(key));
let nodes = key.index.proof_indices().map(|index| self.get_hash(index));
// `from_sized_iter()` returns an error if there are more nodes than `SMT_MAX_DEPTH`, but
// this could only happen if we have more levels than `SMT_MAX_DEPTH` ourselves, which is
// guarded against in `SimpleSmt::new()`.
let path = SparseMerklePath::from_sized_iter(nodes).unwrap();
SparseValuePath { value, path }
}
/// Returns a boolean value indicating whether the SMT is empty.