safety check

OffchainLabs · Sep 18, 2024 · 0c50978 · 0c50978
1 parent 2870567
commit 0c50978
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/state-commitments/optimized/inclusion_proof.go b/state-commitments/optimized/inclusion_proof.go
@@ -13,6 +13,13 @@ func (h *HistoryCommitter) computeMerkleProof(leafIndex uint64, leaves []common.
 	if len(leaves) == 0 {
 		return nil, nil
 	}
+	ok, err := isPowTwo(virtual)
+	if err != nil {
+		return nil, err
+	}
+	if !ok {
+		return nil, errors.New("virtual size must be a power of 2")
+	}
 	if leafIndex >= uint64(len(leaves)) {
 		return nil, errors.New("leaf index out of bounds")
 	}