From 94877b5f16cc0031e9a996d448d2ba1f394c5b54 Mon Sep 17 00:00:00 2001 From: Hitenjain14 Date: Fri, 16 Aug 2024 21:59:15 +0530 Subject: [PATCH] check leaf index in finalize --- core/util/validation_tree.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/core/util/validation_tree.go b/core/util/validation_tree.go index 1bfd6045c..7f4438a4f 100644 --- 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--