feat(taiko-client): catch raiko deserialization errors (#18644)

taikoxyz · Dec 25, 2024 · 98a98fd · 98a98fd
1 parent 59d4f10
commit 98a98fd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/packages/taiko-client/prover/proof_producer/sgx_producer.go b/packages/taiko-client/prover/proof_producer/sgx_producer.go
@@ -308,14 +308,18 @@ func (s *SGXProofProducer) requestBatchProof(
 		return nil, fmt.Errorf("failed to get batch proof, msg: %s", output.ErrorMessage)
 	}
 
+	if output.Data == nil {
+		return nil, fmt.Errorf("unexpected structure error, response: %s", string(resBytes))
+	}
 	if output.Data.Status == ErrProofInProgress.Error() {
 		return nil, ErrProofInProgress
 	}
 	if output.Data.Status == StatusRegistered {
 		return nil, ErrRetry
 	}
 
-	if len(output.Data.Proof.Proof) == 0 {
+	if output.Data.Proof == nil ||
+		len(output.Data.Proof.Proof) == 0 {
 		return nil, errEmptyProof
 	}
 	proof = common.Hex2Bytes(output.Data.Proof.Proof[2:])

diff --git a/packages/taiko-client/prover/proof_producer/zkvm_producer.go b/packages/taiko-client/prover/proof_producer/zkvm_producer.go
@@ -479,14 +479,18 @@ func (s *ZKvmProofProducer) requestBatchProof(
 	if len(output.ErrorMessage) > 0 || len(output.Error) > 0 {
 		return nil, fmt.Errorf("failed to get batch proof, msg: %s", output.ErrorMessage)
 	}
+	if output.Data == nil {
+		return nil, fmt.Errorf("unexpected structure error, response: %s", string(resBytes))
+	}
+
 	if output.Data.Status == ErrProofInProgress.Error() {
 		return nil, ErrProofInProgress
 	}
 	if output.Data.Status == StatusRegistered {
 		return nil, ErrRetry
 	}
 
-	if len(output.Data.Proof.Proof) == 0 {
+	if output.Data.Proof == nil || len(output.Data.Proof.Proof) == 0 {
 		return nil, errEmptyProof
 	}
 	proof = common.Hex2Bytes(output.Data.Proof.Proof[2:])