From ec436b4c2b525f8bc5b1138147bbce59026a8148 Mon Sep 17 00:00:00 2001
From: Thomas Piellard <thomas.piellard@consensys.net>
Date: Wed, 18 Sep 2024 18:54:28 +0200
Subject: [PATCH] fix: fixed panic

---
 ecc/bls12-377/shplonk/shplonk_test.go                    | 6 +++++-
 ecc/bls12-378/shplonk/shplonk_test.go                    | 6 +++++-
 ecc/bls12-381/shplonk/shplonk_test.go                    | 6 +++++-
 ecc/bls24-315/shplonk/shplonk_test.go                    | 6 +++++-
 ecc/bls24-317/shplonk/shplonk_test.go                    | 6 +++++-
 ecc/bn254/shplonk/shplonk_test.go                        | 6 +++++-
 ecc/bw6-633/shplonk/shplonk_test.go                      | 6 +++++-
 ecc/bw6-756/shplonk/shplonk_test.go                      | 6 +++++-
 ecc/bw6-761/shplonk/shplonk_test.go                      | 6 +++++-
 internal/generator/shplonk/template/shplonk.test.go.tmpl | 6 +++++-
 10 files changed, 50 insertions(+), 10 deletions(-)

diff --git a/ecc/bls12-377/shplonk/shplonk_test.go b/ecc/bls12-377/shplonk/shplonk_test.go
index 69a26be1b..0b69e54a7 100644
--- a/ecc/bls12-377/shplonk/shplonk_test.go
+++ b/ecc/bls12-377/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bls12-378/shplonk/shplonk_test.go b/ecc/bls12-378/shplonk/shplonk_test.go
index dcb5f6942..ac3d707cf 100644
--- a/ecc/bls12-378/shplonk/shplonk_test.go
+++ b/ecc/bls12-378/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bls12-381/shplonk/shplonk_test.go b/ecc/bls12-381/shplonk/shplonk_test.go
index 043238b47..12468961c 100644
--- a/ecc/bls12-381/shplonk/shplonk_test.go
+++ b/ecc/bls12-381/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bls24-315/shplonk/shplonk_test.go b/ecc/bls24-315/shplonk/shplonk_test.go
index 3e282d99a..9bc663dcf 100644
--- a/ecc/bls24-315/shplonk/shplonk_test.go
+++ b/ecc/bls24-315/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bls24-317/shplonk/shplonk_test.go b/ecc/bls24-317/shplonk/shplonk_test.go
index 14dd5af29..72c89c962 100644
--- a/ecc/bls24-317/shplonk/shplonk_test.go
+++ b/ecc/bls24-317/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bn254/shplonk/shplonk_test.go b/ecc/bn254/shplonk/shplonk_test.go
index 6d2c2841b..be2a812dc 100644
--- a/ecc/bn254/shplonk/shplonk_test.go
+++ b/ecc/bn254/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bw6-633/shplonk/shplonk_test.go b/ecc/bw6-633/shplonk/shplonk_test.go
index 807b263e0..cde6f766c 100644
--- a/ecc/bw6-633/shplonk/shplonk_test.go
+++ b/ecc/bw6-633/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bw6-756/shplonk/shplonk_test.go b/ecc/bw6-756/shplonk/shplonk_test.go
index c72929436..2ada895df 100644
--- a/ecc/bw6-756/shplonk/shplonk_test.go
+++ b/ecc/bw6-756/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/ecc/bw6-761/shplonk/shplonk_test.go b/ecc/bw6-761/shplonk/shplonk_test.go
index e962fe6df..3fb7e0715 100644
--- a/ecc/bw6-761/shplonk/shplonk_test.go
+++ b/ecc/bw6-761/shplonk/shplonk_test.go
@@ -34,7 +34,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {
diff --git a/internal/generator/shplonk/template/shplonk.test.go.tmpl b/internal/generator/shplonk/template/shplonk.test.go.tmpl
index d3f6611b6..c7e3514cf 100644
--- a/internal/generator/shplonk/template/shplonk.test.go.tmpl
+++ b/internal/generator/shplonk/template/shplonk.test.go.tmpl
@@ -16,7 +16,11 @@ var bAlpha *big.Int
 func init() {
 	const srsSize = 230
 	bAlpha = new(big.Int).SetInt64(42) // randomise ?
-	testSrs, _ = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	var err error
+	testSrs, err = kzg.NewSRS(ecc.NextPowerOfTwo(srsSize), bAlpha)
+	if err!=nil {
+		panic(err)
+	}
 }
 
 func TestOpening(t *testing.T) {