From 49511611b595db6597f4ab90d621876b067f1dbd Mon Sep 17 00:00:00 2001
From: Kevin Jue <kjue235@gmail.com>
Date: Tue, 19 Dec 2023 13:24:27 -0800
Subject: [PATCH] fix for V-SCT-VUL-031

---
 poseidon/bn254.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/poseidon/bn254.go b/poseidon/bn254.go
index 4975415..25ad83c 100644
--- a/poseidon/bn254.go
+++ b/poseidon/bn254.go
@@ -10,6 +10,7 @@ package poseidon
 import (
 	"math/big"
 
+	"github.com/consensys/gnark-crypto/ecc/bn254"
 	"github.com/consensys/gnark/frontend"
 	gl "github.com/succinctlabs/gnark-plonky2-verifier/goldilocks"
 )
@@ -28,6 +29,10 @@ type BN254State = [BN254_SPONGE_WIDTH]frontend.Variable
 type BN254HashOut = frontend.Variable
 
 func NewBN254Chip(api frontend.API) *BN254Chip {
+	if api.Compiler().Field().Cmp(bn254.ID.ScalarField()) != 0 {
+		panic("Gnark compiler not set to BN254 scalar field")
+	}
+
 	return &BN254Chip{api: api, gl: *gl.New(api)}
 }