fix: buffer sufficient size (#551)

Consensys · Oct 10, 2024 · 441b06e · 441b06e
1 parent ef45936
commit 441b06e
Show file tree

Hide file tree

Showing 8 changed files with 40 additions and 0 deletions.
diff --git a/ecc/bls12-377/shplonk/shplonk.go b/ecc/bls12-377/shplonk/shplonk.go
diff --git a/ecc/bls12-381/shplonk/shplonk.go b/ecc/bls12-381/shplonk/shplonk.go
diff --git a/ecc/bls24-315/shplonk/shplonk.go b/ecc/bls24-315/shplonk/shplonk.go
diff --git a/ecc/bls24-317/shplonk/shplonk.go b/ecc/bls24-317/shplonk/shplonk.go
diff --git a/ecc/bn254/shplonk/shplonk.go b/ecc/bn254/shplonk/shplonk.go
diff --git a/ecc/bw6-633/shplonk/shplonk.go b/ecc/bw6-633/shplonk/shplonk.go
diff --git a/ecc/bw6-761/shplonk/shplonk.go b/ecc/bw6-761/shplonk/shplonk.go
diff --git a/internal/generator/shplonk/template/shplonk.go.tmpl b/internal/generator/shplonk/template/shplonk.go.tmpl
@@ -62,6 +62,11 @@ func BatchOpen(polynomials [][]fr.Element, digests []kzg.Digest, points [][]fr.E
 			maxSizePolys = len(polynomials[i])
 		}
 	}
+	for i := range points {
+		if len(points[i])+1 > maxSizePolys {
+			maxSizePolys = len(points[i]) + 1
+		}
+	}
 	nbPoints := 0
 	sizeSi := make([]int, len(points))
 	for i := 0; i < nbPolynomials; i++ {