diff --git a/src/util.rs b/src/util.rs index 1c68af0..94ec617 100644 --- a/src/util.rs +++ b/src/util.rs @@ -1,4 +1,4 @@ -use crate::{MMRStore, Merge, Result, MMR}; +use crate::{MMRStore, Merge, MerkleProof, Result, MMR}; use std::cell::RefCell; use std::collections::HashMap; use std::fmt::Debug; @@ -70,4 +70,9 @@ impl> MemMMR { mmr.commit()?; Ok(pos) } + + pub fn gen_proof(&self, pos: u64) -> Result> { + let mmr = MMR::>::new(self.mmr_size, &self.store); + mmr.gen_proof(pos) + } }