feat: integrate BLS precompile with zkevm

Author: ivokub

prover/zkevm/settings.go

@@ -3,6 +3,7 @@ package zkevm
 import (
 	"github.com/consensys/linea-monorepo/prover/protocol/wizard"
 	"github.com/consensys/linea-monorepo/prover/zkevm/arithmetization"
+	"github.com/consensys/linea-monorepo/prover/zkevm/prover/bls"
 	"github.com/consensys/linea-monorepo/prover/zkevm/prover/ecarith"
 	"github.com/consensys/linea-monorepo/prover/zkevm/prover/ecdsa"
 	"github.com/consensys/linea-monorepo/prover/zkevm/prover/ecpair"
@@ -27,6 +28,7 @@ type Settings struct {
 	Ecadd, Ecmul     ecarith.Limits
 	Ecpair           ecpair.Limits
 	Sha2             sha2.Settings
+	Bls              bls.Limits
 	PublicInput      publicInput.Settings
 	CompilationSuite CompilationSuite
 	Metadata         wizard.VersionMetadata

prover/zkevm/zkevm.go

@@ -5,6 +5,7 @@ import (
 	"github.com/consensys/linea-monorepo/prover/protocol/serialization"
 	"github.com/consensys/linea-monorepo/prover/protocol/wizard"
 	"github.com/consensys/linea-monorepo/prover/zkevm/arithmetization"
+	"github.com/consensys/linea-monorepo/prover/zkevm/prover/bls"
 	"github.com/consensys/linea-monorepo/prover/zkevm/prover/ecarith"
 	"github.com/consensys/linea-monorepo/prover/zkevm/prover/ecdsa"
 	"github.com/consensys/linea-monorepo/prover/zkevm/prover/ecpair"
@@ -45,6 +46,22 @@ type ZkEvm struct {
 	// Sha2 is the module responsible for doing the computation of the Sha2
 	// precompile.
 	Sha2 *sha2.Sha2SingleProvider `json:"sha2"`
+	// BlsG1Add is responsible for BLS G1 addition precompile.
+	BlsG1Add *bls.BlsAdd `json:"blsG1Add"`
+	// BlsG2Add is responsible for BLS G2 addition precompile.
+	BlsG2Add *bls.BlsAdd `json:"blsG2Add"`
+	// BlsG1Msm is responsible for BLS G1 multi-scalar multiplicaton precompile.
+	BlsG1Msm *bls.BlsMsm `json:"blsG1Msm"`
+	// BlsG2Msm is responsible for BLS G2 multi-scalar multiplication precompile.
+	BlsG2Msm *bls.BlsMsm `json:"blsG2Msm"`
+	// BlsG1Map is responsible for BLS Fp map to G1 precompile.
+	BlsG1Map *bls.BlsMap `json:"blsG1Map"`
+	// BlsG2Map is responsible for BLS Fp2 map to G2 precompile.
+	BlsG2Map *bls.BlsMap `json:"blsG2Map"`
+	// BlsPairingCheck is responsible for BLS pairing check precompile.
+	BlsPairingCheck *bls.BlsPair `json:"blsPairingCheck"`
+	// PointEval is responsible for EIP-4844 point evaluation precompile.
+	PointEval *bls.BlsPointEval `json:"pointEval"`
 	// Contains the actual wizard-IOP compiled object. This object is called to
 	// generate the inner-proof.
 	WizardIOP *wizard.CompiledIOP `json:"wizardIOP"`
@@ -93,17 +110,25 @@ func (z *ZkEvm) VerifyInner(proof wizard.Proof) error {
 func newZkEVM(b *wizard.Builder, s *Settings) *ZkEvm {
 
 	var (
-		comp         = b.CompiledIOP
-		arith        = arithmetization.NewArithmetization(b, s.Arithmetization)
-		ecdsa        = ecdsa.NewEcdsaZkEvm(comp, &s.Ecdsa)
-		stateManager = statemanager.NewStateManager(comp, s.Statemanager)
-		keccak       = keccak.NewKeccakZkEVM(comp, s.Keccak, ecdsa.GetProviders())
-		modexp       = modexp.NewModuleZkEvm(comp, s.Modexp)
-		ecadd        = ecarith.NewEcAddZkEvm(comp, &s.Ecadd)
-		ecmul        = ecarith.NewEcMulZkEvm(comp, &s.Ecmul)
-		ecpair       = ecpair.NewECPairZkEvm(comp, &s.Ecpair)
-		sha2         = sha2.NewSha2ZkEvm(comp, s.Sha2)
-		publicInput  = publicInput.NewPublicInputZkEVM(comp, &s.PublicInput, &stateManager.StateSummary)
+		comp            = b.CompiledIOP
+		arith           = arithmetization.NewArithmetization(b, s.Arithmetization)
+		ecdsa           = ecdsa.NewEcdsaZkEvm(comp, &s.Ecdsa)
+		stateManager    = statemanager.NewStateManager(comp, s.Statemanager)
+		keccak          = keccak.NewKeccakZkEVM(comp, s.Keccak, ecdsa.GetProviders())
+		modexp          = modexp.NewModuleZkEvm(comp, s.Modexp)
+		ecadd           = ecarith.NewEcAddZkEvm(comp, &s.Ecadd)
+		ecmul           = ecarith.NewEcMulZkEvm(comp, &s.Ecmul)
+		ecpair          = ecpair.NewECPairZkEvm(comp, &s.Ecpair)
+		sha2            = sha2.NewSha2ZkEvm(comp, s.Sha2)
+		blsG1Add        = bls.NewG1AddZkEvm(comp, &s.Bls)
+		blsG1Msm        = bls.NewG1MsmZkEvm(comp, &s.Bls)
+		blsG1Map        = bls.NewG1MapZkEvm(comp, &s.Bls)
+		blsG2Add        = bls.NewG2AddZkEvm(comp, &s.Bls)
+		blsG2Msm        = bls.NewG2MsmZkEvm(comp, &s.Bls)
+		blsG2Map        = bls.NewG2MapZkEvm(comp, &s.Bls)
+		blsPairingCheck = bls.NewPairingZkEvm(comp, &s.Bls)
+		pointEval       = bls.NewPointEvalZkEvm(comp, &s.Bls)
+		publicInput     = publicInput.NewPublicInputZkEVM(comp, &s.PublicInput, &stateManager.StateSummary)
 	)
 
 	return &ZkEvm{
@@ -116,6 +141,14 @@ func newZkEVM(b *wizard.Builder, s *Settings) *ZkEvm {
 		Ecmul:           ecmul,
 		Ecpair:          ecpair,
 		Sha2:            sha2,
+		BlsG1Add:        blsG1Add,
+		BlsG2Add:        blsG2Add,
+		BlsG1Msm:        blsG1Msm,
+		BlsG2Msm:        blsG2Msm,
+		BlsG1Map:        blsG1Map,
+		BlsG2Map:        blsG2Map,
+		BlsPairingCheck: blsPairingCheck,
+		PointEval:       pointEval,
 		PublicInput:     &publicInput,
 	}
 }
@@ -139,6 +172,14 @@ func (z *ZkEvm) GetMainProverStep(input *Witness) (prover wizard.MainProverStep)
 		z.Ecmul.Assign(run)
 		z.Ecpair.Assign(run)
 		z.Sha2.Run(run)
+		z.BlsG1Add.Assign(run)
+		z.BlsG2Add.Assign(run)
+		z.BlsG1Msm.Assign(run)
+		z.BlsG2Msm.Assign(run)
+		z.BlsG1Map.Assign(run)
+		z.BlsG2Map.Assign(run)
+		z.BlsPairingCheck.Assign(run)
+		z.PointEval.Assign(run)
 		z.PublicInput.Assign(run, input.L2BridgeAddress, input.BlockHashList)
 	}
 }