feat(sp1-cc-host): allow custom chains

Author: kien-rise

Cargo.lock

@@ -3,12 +3,12 @@ resolver = "2"
 members = [
     "bin/*",
     "crates/*",
-    "example/preloader",    
+    "example/preloader",
     "canoe/bindings",
     "canoe/provider",
     "canoe/steel/apps",
     "canoe/steel/methods",
-    "canoe/steel/verifier", 
+    "canoe/steel/verifier",
     "canoe/sp1-cc/host",
     "canoe/sp1-cc/client",
     "canoe/sp1-cc/vkey-bin",
@@ -55,12 +55,12 @@ hokulea-zkvm-verification = { path = "crates/zkvm-verification", version = "0.1.
 eigenda-cert = { path = "crates/eigenda-cert" }
 
 # Alloy (Network)
-alloy-rlp = { version = "0.3.12", default-features = false }
-alloy-provider = { version = "1.0.38", default-features = false }
 alloy-consensus = { version = "1.0.38", default-features = false }
-alloy-rpc-types = { version = "1.0.38", default-features = false }
 alloy-genesis = { version = "1.0.38", default-features = false }
 alloy-primitives = { version = "1.3.1", default-features = false }
+alloy-provider = { version = "1.0.38", default-features = false }
+alloy-rlp = { version = "0.3.12", default-features = false }
+alloy-rpc-types = { version = "1.0.38", default-features = false }
 alloy-sol-types = { version = "1.3.1", default-features = false }
 
 # Execution
@@ -137,7 +137,7 @@ sp1-core-executor = "5.2.1"
 sp1-prover = "5.2.1"
 sp1-cc-client-executor = { git = "https://github.com/succinctlabs/sp1-contract-call.git", tag = "reth-1.9.1" }
 sp1-cc-host-executor = { git = "https://github.com/succinctlabs/sp1-contract-call.git", tag = "reth-1.9.1" }
-rsp-primitives = { git = "https://github.com/succinctlabs/rsp", tag = "reth-1.9.1" } 
+rsp-primitives = { git = "https://github.com/succinctlabs/rsp", tag = "reth-1.9.1" }
 
 # ZKVM deps
 hex = "0.4"

Cargo.toml

@@ -8,8 +8,9 @@ sp1-cc-host-executor = { workspace = true }
 sp1-cc-client-executor = { workspace = true }
 rsp-primitives = { workspace = true }
 
+alloy-genesis.workspace = true
 alloy-primitives.workspace = true
-alloy-sol-types = { workspace = true }
+alloy-sol-types.workspace = true
 bincode.workspace = true
 
 anyhow = { workspace = true }
@@ -25,4 +26,4 @@ tracing.workspace = true
 # sp1
 sp1-sdk.workspace = true
 sp1-core-executor.workspace = true
-sp1-prover.workspace = true
+sp1-prover.workspace = true

canoe/sp1-cc/host/Cargo.toml

@@ -1,10 +1,12 @@
+use alloy_genesis::ChainConfig;
 use alloy_primitives::Address;
 use alloy_rpc_types::BlockNumberOrTag;
 use alloy_sol_types::SolType;
 use anyhow::Result;
 use async_trait::async_trait;
 use canoe_bindings::{Journal, StatusCode};
 use canoe_provider::{CanoeInput, CanoeProvider, CertVerifierCall};
+use rsp_primitives::genesis::genesis_from_json;
 use sp1_cc_client_executor::ContractInput;
 use sp1_cc_host_executor::{EvmSketch, Genesis};
 use sp1_sdk::{
@@ -19,8 +21,6 @@ use std::{
 use tracing::{debug, info, warn};
 use url::Url;
 
-use rsp_primitives::genesis::genesis_from_json;
-
 /// The ELF we want to execute inside the zkVM.
 pub const ELF: &[u8] = include_bytes!("../../elf/canoe-sp1-cc-client");
 
@@ -62,6 +62,8 @@ pub struct CanoeSp1CCProvider {
     pub eth_rpc_url: String,
     /// if true, execute and return a mock proof
     pub mock_mode: bool,
+    /// optional custom chain configuration for genesis block
+    pub custom_chain_config: Option<ChainConfig>,
 }
 
 #[async_trait]
@@ -77,7 +79,15 @@ impl CanoeProvider for CanoeSp1CCProvider {
             return None;
         }
 
-        Some(get_sp1_cc_proof(canoe_inputs, &self.eth_rpc_url, self.mock_mode).await)
+        Some(
+            get_sp1_cc_proof(
+                canoe_inputs,
+                &self.eth_rpc_url,
+                self.mock_mode,
+                self.custom_chain_config.clone(),
+            )
+            .await,
+        )
     }
 }
 
@@ -92,6 +102,8 @@ pub struct CanoeSp1CCReducedProofProvider {
     pub eth_rpc_url: String,
     /// if true, execute and return a mock proof
     pub mock_mode: bool,
+    /// optional custom chain configuration for genesis block
+    pub custom_chain_config: Option<ChainConfig>,
 }
 
 #[async_trait]
@@ -107,7 +119,14 @@ impl CanoeProvider for CanoeSp1CCReducedProofProvider {
             return None;
         }
 
-        match get_sp1_cc_proof(canoe_inputs, &self.eth_rpc_url, self.mock_mode).await {
+        match get_sp1_cc_proof(
+            canoe_inputs,
+            &self.eth_rpc_url,
+            self.mock_mode,
+            self.custom_chain_config.clone(),
+        )
+        .await
+        {
             Ok(proof) => {
                 let SP1Proof::Compressed(proof) = proof.proof else {
                     panic!("cannot get Sp1ReducedProof")
@@ -123,6 +142,7 @@ async fn get_sp1_cc_proof(
     canoe_inputs: Vec<CanoeInput>,
     eth_rpc_url: &str,
     mock_mode: bool,
+    custom_chain_config: Option<ChainConfig>,
 ) -> Result<sp1_sdk::SP1ProofWithPublicValues> {
     // ensure chain id and l1 block number across all DAcerts are identical
     let l1_chain_id = canoe_inputs[0].l1_chain_id;
@@ -147,35 +167,26 @@ async fn get_sp1_cc_proof(
 
     let rpc_url = Url::from_str(eth_rpc_url).unwrap();
 
-    let sketch = match Genesis::try_from(l1_chain_id) {
-        Ok(genesis) => {
-            EvmSketch::builder()
-                .at_block(block_number)
-                .with_genesis(genesis)
-                .el_rpc_url(rpc_url)
-                .build()
-                .await?
-        }
-        // if genesis is not available in the sp1-cc library, the code uses custom genesis config
-        Err(_) => {
-            let chain_config = match l1_chain_id {
-                17000 => genesis_from_json(HOLESKY_GENESIS).expect("genesis from json"),
-                3151908 => genesis_from_json(KURTOSIS_DEVNET_GENESIS).expect("genesis from json"),
-                _ => panic!("chain id {l1_chain_id} is not supported by canoe sp1 cc"),
-            };
-
-            let genesis = Genesis::Custom(chain_config.config);
-
-            EvmSketch::builder()
-                .at_block(block_number)
-                .with_genesis(genesis)
-                .el_rpc_url(rpc_url)
-                .build()
-                .await
-                .expect("evm sketch builder")
-        }
+    let genesis = if let Some(chain_config) = custom_chain_config {
+        Genesis::Custom(chain_config)
+    } else if let Ok(genesis) = Genesis::try_from(l1_chain_id) {
+        genesis
+    } else {
+        let chain_config = match l1_chain_id {
+            17000 => genesis_from_json(HOLESKY_GENESIS).expect("genesis from json"),
+            3151908 => genesis_from_json(KURTOSIS_DEVNET_GENESIS).expect("genesis from json"),
+            _ => panic!("chain id {l1_chain_id} is not supported by canoe sp1 cc"),
+        };
+        Genesis::Custom(chain_config.config)
     };
 
+    let sketch = EvmSketch::builder()
+        .at_block(block_number)
+        .with_genesis(genesis)
+        .el_rpc_url(rpc_url)
+        .build()
+        .await?;
+
     let derived_l1_header_hash = sketch.anchor.header().hash_slow();
     assert!(l1_head_block_hash == derived_l1_header_hash);
 

canoe/sp1-cc/host/src/lib.rs

@@ -95,3 +95,13 @@ fn cert_verifier_address_abi_encode_interface(
         }
     }
 }
+
+impl CanoeVerifierAddressFetcher for Address {
+    fn fetch_address(
+        &self,
+        _chain_id: u64,
+        _versioned_cert: &EigenDAVersionedCert,
+    ) -> Result<Address, CanoeVerifierAddressFetcherError> {
+        Ok(*self)
+    }
+}

canoe/verifier-address-fetcher/src/lib.rs

@@ -13,12 +13,13 @@ hokulea-client.workspace = true
 hokulea-compute-proof.workspace = true
 
 # General
+alloy-genesis.workspace = true
 anyhow.workspace = true
-tokio = { workspace = true, features = ["full"] }
-clap = { workspace = true, features = ["derive", "env"] }
 cfg-if = { workspace = true }
-tracing = { workspace = true }
+clap = { workspace = true, features = ["derive", "env"] }
 serde_json.workspace = true
+tokio = { workspace = true, features = ["full"] }
+tracing = { workspace = true }
 
 kona-client.workspace = true
 kona-preimage.workspace = true

example/preloader/Cargo.toml

@@ -87,9 +87,19 @@ async fn main() -> anyhow::Result<()> {
                 .and_then(|v| v.parse::<bool>().ok())
                 .unwrap_or(false);
 
-            let canoe_provider = CanoeSp1CCReducedProofProvider{
+            let custom_chain_config = if let Some(path) = cfg.kona_cfg.l1_config_path {
+                let json = std::fs::read_to_string(&path)
+                    .with_context(|| format!("failed to read genesis file at {}", path.display()))?;
+                let genesis = serde_json::from_str::<alloy_genesis::Genesis>(&json)
+                    .with_context(|| format!("failed to parse L1 genesis from {}", path.display()))?;
+                Some(genesis.config)
+            } else {
+                None
+            };
+            let canoe_provider = CanoeSp1CCReducedProofProvider {
                 eth_rpc_url: cfg.kona_cfg.l1_node_address.clone().unwrap(),
                 mock_mode,
+                custom_chain_config,
             };
             let canoe_verifier = CanoeNoOpVerifier{};
         } else {