Merge pull request #339 from kcalvinalvin/forest-provebatch-bugfix

accumulator/forestproofs: Fix bug in ProveBatch()
mit-dci · Feb 14, 2022 · df7b3aa · df7b3aa
2 parents 3f771bb + 56d86eb
commit df7b3aa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/accumulator/forestproofs.go b/accumulator/forestproofs.go
@@ -142,7 +142,9 @@ func (f *Forest) ProveBatch(hs []Hash) (BatchProof, error) {
 	if len(hs) == 0 {
 		return bp, nil
 	}
-	if f.data.size() < 2 {
+	// When there is only 1 leaf in the entire forest, the leaf is the proof.
+	// When there are no leaves, there's nothing to prove.
+	if f.numLeaves <= 1 {
 		return bp, nil
 	}