Merge pull request #1580 from 0chain/hotfix/vt-finalize

check leaf index in finalize
0chain · Aug 16, 2024 · 95591c8 · 95591c8
2 parents 9830d99 + 94877b5
commit 95591c8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/core/util/validation_tree.go b/core/util/validation_tree.go
@@ -155,6 +155,12 @@ func (v *ValidationTree) Finalize() error {
 	v.isFinalized = true
 
 	if v.writeCount > 0 {
+		if v.leafIndex == len(v.leaves) {
+			// increase leaves size
+			leaves := make([][]byte, len(v.leaves)+1)
+			copy(leaves, v.leaves)
+			v.leaves = leaves
+		}
 		v.leaves[v.leafIndex] = v.h.Sum(nil)
 	} else {
 		v.leafIndex--