SP1 adapter: separate network error from ProofGenerationError. (#59)

Author: evgenyzdanovich

adapters/sp1/host/src/prover.rs

@@ -1,6 +1,9 @@
-use sp1_sdk::{network::FulfillmentStrategy, ProverClient};
 #[cfg(feature = "remote-prover")]
 use sp1_sdk::{network::B256, SP1ProofMode};
+use sp1_sdk::{
+    network::{Error as NetworkError, FulfillmentStrategy},
+    ProverClient,
+};
 #[cfg(feature = "remote-prover")]
 use zkaleido::ZkVmRemoteProver;
 use zkaleido::{
@@ -75,11 +78,23 @@ impl ZkVmProver for SP1Host {
                 ProofType::Groth16 => network_prover_builder.groth16(),
             };
 
-            let proof_info = network_prover
-                .run()
-                .map_err(|e| ZkVmError::ProofGenerationError(e.to_string()))?;
+            let proof_result = network_prover.run();
+
+            // Some error handling.
+            // If SP1 network prover returned Network RPC error - transform it to zkaleido
+            // network error, so the users can handle it gracefully.
+            // Otherwise, return a general error message wrapped in ProofGeneratedError.
+            let proof = match proof_result {
+                Ok(proof) => proof,
+                Err(e) => match e.downcast_ref::<NetworkError>() {
+                    Some(NetworkError::RpcError(status)) => {
+                        return Err(ZkVmError::NetworkRetryableError(status.to_string()));
+                    }
+                    _ => return Err(ZkVmError::ProofGenerationError(e.to_string())),
+                },
+            };
 
-            return Ok(proof_info.into());
+            return Ok(proof.into());
         }
 
         let client = ProverClient::from_env();

zkaleido/src/errors.rs

@@ -14,6 +14,10 @@ pub enum ZkVmError {
     #[error("Execution failed: {0}")]
     ExecutionError(String),
 
+    /// Network (or RPC) error that can occur while using prover network.
+    #[error("Network error: {0}")]
+    NetworkRetryableError(String),
+
     /// This error is returned when proof generation fails for any reason.
     #[error("Proof generation failed: {0}")]
     ProofGenerationError(String),