diff --git a/coordinator/internal/logic/submitproof/proof_receiver.go b/coordinator/internal/logic/submitproof/proof_receiver.go index ffc5df4147..176a68c336 100644 --- a/coordinator/internal/logic/submitproof/proof_receiver.go +++ b/coordinator/internal/logic/submitproof/proof_receiver.go @@ -169,17 +169,21 @@ func (m *ProofReceiverLogic) HandleZkProof(ctx *gin.Context, proofParameter coor if getHardForkErr != nil { return ErrGetHardForkNameFailed } - // Post-Bernoulli we do not verify chunk proofs. - // Verify batch proof - if message.ProofType(proofParameter.TaskType) == message.ProofTypeBatch { + + switch message.ProofType(proofParameter.TaskType) { + case message.ProofTypeChunk: + var chunkProof message.ChunkProof + if unmarshalErr := json.Unmarshal([]byte(proofParameter.Proof), &chunkProof); unmarshalErr != nil { + return unmarshalErr + } + success, verifyErr = m.verifier.VerifyChunkProof(&chunkProof, hardForkName) + case message.ProofTypeBatch: var batchProof message.BatchProof if unmarshalErr := json.Unmarshal([]byte(proofParameter.Proof), &batchProof); unmarshalErr != nil { return unmarshalErr } success, verifyErr = m.verifier.VerifyBatchProof(&batchProof, hardForkName) - } - // Verify bundle proof - if message.ProofType(proofParameter.TaskType) == message.ProofTypeBundle { + case message.ProofTypeBundle: var bundleProof message.BundleProof if unmarshalErr := json.Unmarshal([]byte(proofParameter.Proof), &bundleProof); unmarshalErr != nil { return unmarshalErr diff --git a/coordinator/internal/logic/verifier/mock.go b/coordinator/internal/logic/verifier/mock.go index 722a4778c6..7615c51b6e 100644 --- a/coordinator/internal/logic/verifier/mock.go +++ b/coordinator/internal/logic/verifier/mock.go @@ -16,7 +16,7 @@ func NewVerifier(cfg *config.VerifierConfig) (*Verifier, error) { } // VerifyChunkProof return a mock verification result for a ChunkProof. -func (v *Verifier) VerifyChunkProof(proof *message.ChunkProof) (bool, error) { +func (v *Verifier) VerifyChunkProof(proof *message.ChunkProof, forkName string) (bool, error) { if string(proof.Proof) == InvalidTestProof { return false, nil }