Add L1SLOAD precompile

Cargo.toml

@@ -71,13 +71,13 @@ reth-revm = { git = "https://github.com/paradigmxyz/reth.git", tag = "v1.11.0",
 reth-provider = { git = "https://github.com/paradigmxyz/reth.git", tag = "v1.11.0", default-features = false }
 reth-storage-api = { git = "https://github.com/paradigmxyz/reth.git", tag = "v1.11.0", default-features = false }
 
-alethia-reth-block = { git = "https://github.com/NethermindEth/alethia-reth.git", branch = "main", default-features = false }
-alethia-reth-chainspec = { git = "https://github.com/NethermindEth/alethia-reth.git", branch = "main", default-features = false }
-alethia-reth-consensus = { git = "https://github.com/NethermindEth/alethia-reth.git", branch = "main", default-features = false }
-alethia-reth-evm = { git = "https://github.com/NethermindEth/alethia-reth.git", branch = "main", default-features = false, features = [
+alethia-reth-block = { git = "https://github.com/NethermindEth/alethia-reth.git", rev = "98b32a4693e9fb55b0364e4ae922f3e8149c5c21", default-features = false }
+alethia-reth-chainspec = { git = "https://github.com/NethermindEth/alethia-reth.git", rev = "98b32a4693e9fb55b0364e4ae922f3e8149c5c21", default-features = false }
+alethia-reth-consensus = { git = "https://github.com/NethermindEth/alethia-reth.git", rev = "98b32a4693e9fb55b0364e4ae922f3e8149c5c21", default-features = false }
+alethia-reth-evm = { git = "https://github.com/NethermindEth/alethia-reth.git", rev = "98b32a4693e9fb55b0364e4ae922f3e8149c5c21", default-features = false, features = [
     "serde",
 ] }
-alethia-reth-primitives = { git = "https://github.com/NethermindEth/alethia-reth.git", branch = "main", default-features = false, features = [
+alethia-reth-primitives = { git = "https://github.com/NethermindEth/alethia-reth.git", rev = "98b32a4693e9fb55b0364e4ae922f3e8149c5c21", default-features = false, features = [
     "serde-bincode-compat",
 ] }
 
@@ -99,7 +99,7 @@ risc0-binfmt = { version = "=3.0.4" }
 risc0-zkos-v1compat = { version = "=2.2.2" }
 
 # SP1
-sp1-sdk = { version = "6.0.0", features = ["cuda"] }
+sp1-sdk = { version = "6.0.0" }
 sp1-prover = { version = "6.0.0" }
 sp1-zkvm = { version = "6.0.0" }
 sp1-helper = { version = "6.0.0" }
@@ -141,6 +141,7 @@ alloy-transport-http = { version = "1.0.18", default-features = false, features
 ] }
 alloy-signer = { version = "1.0.18", default-features = false }
 alloy-signer-local = { version = "1.0.18", default-features = false }
+alloy-trie = { version = "0.9.1", default-features = false }
 
 # ethers (TODO: remove)
 ethers-contract = { git = "https://github.com/smtmfft/ethers-rs", branch = "ethers-core-2.0.10" }

Dockerfile.zk

@@ -107,7 +107,8 @@ RUN TARGET=sp1 make guest
 # Verify sp1 guest ELFs were produced
 RUN ls provers/sp1/guest/elf/*
 
-# ── Stage 3: Host build with CUDA (depends on all source code) ──────
+# ── Stage 3: Host build (depends on all source code) ────────────────
+# CUDA is opt-in via --build-arg CARGO_FEATURES="sp1,risc0,cuda"
 FROM nvidia/cuda:12.4.0-devel-ubuntu22.04 AS builder
 
 ENV DEBIAN_FRONTEND=noninteractive
@@ -177,7 +178,8 @@ RUN mkdir -p ~/.cargo/bin
 # ARG CARGO_BUILD_JOBS=2
 # ENV CARGO_BUILD_JOBS=${CARGO_BUILD_JOBS}
 # ENV NUM_JOBS=${CARGO_BUILD_JOBS}
-RUN cargo build --release ${BUILD_FLAGS} --features "sp1,risc0,cuda"
+ARG CARGO_FEATURES="sp1,risc0"
+RUN cargo build --release ${BUILD_FLAGS} --features "${CARGO_FEATURES}"
 
 FROM ubuntu:22.04 AS raiko-zk
 

core/src/lib.rs

@@ -1,4 +1,4 @@
-use std::{collections::HashMap, hint::black_box};
+use std::{collections::HashMap, hint::black_box, sync::MutexGuard};
 
 use alethia_reth_primitives::{TaikoBlock, TaikoTxEnvelope};
 use alloy_primitives::Address;
@@ -8,6 +8,9 @@ use raiko_lib::{
     builder::{create_mem_db, RethBlockBuilder},
     consts::ChainSpec,
     input::{GuestBatchInput, GuestBatchOutput, GuestInput, GuestOutput, TaikoProverData},
+    l1_precompiles::{
+        acquire_l1sload_lock, clear_l1sload_cache, verify_and_populate_l1sload_proofs,
+    },
     protocol_instance::ProtocolInstance,
     prover::{IdStore, IdWrite, Proof, ProofKey},
     utils::txs::{generate_transactions, generate_transactions_for_batch_blocks},
@@ -20,7 +23,9 @@ use crate::{
     interfaces::{
         run_shasta_proposal_prover, ProofRequest, RaikoError, RaikoResult, ShastaProposalCheckpoint,
     },
-    preflight::{batch_preflight, preflight, BatchPreflightData, PreflightData},
+    preflight::{
+        batch_preflight, get_anchor_tx_info_by_fork, preflight, BatchPreflightData, PreflightData,
+    },
     provider::BlockDataProvider,
 };
 
@@ -33,6 +38,50 @@ pub mod provider;
 
 pub type MerkleProof = HashMap<Address, EIP1186AccountProofResponse>;
 
+/// Prepare L1SLOAD state for block execution: clear cache, and if the block has
+/// L1 storage proofs, acquire the execution lock and verify/populate them.
+///
+/// Returns the execution guard that must be held until block execution completes.
+fn prepare_l1sload_for_execution(input: &GuestInput) -> RaikoResult<MutexGuard<'static, ()>> {
+    let guard = acquire_l1sload_lock();
+    clear_l1sload_cache();
+
+    if input.l1_storage_proofs.is_empty() {
+        return Ok(guard);
+    }
+    let anchor_tx = input.taiko.anchor_tx.as_ref().ok_or_else(|| {
+        RaikoError::Guest(raiko_lib::prover::ProverError::GuestError(
+            "No anchor tx for L1SLOAD verification".to_string(),
+        ))
+    })?;
+    let fork = input
+        .chain_spec
+        .active_fork(input.block.header.number, input.block.timestamp)
+        .map_err(|e| {
+            RaikoError::Guest(raiko_lib::prover::ProverError::GuestError(format!(
+                "Failed to determine active fork: {e}"
+            )))
+        })?;
+    let (anchor_block_number, _) = get_anchor_tx_info_by_fork(fork, anchor_tx).map_err(|e| {
+        RaikoError::Guest(raiko_lib::prover::ProverError::GuestError(format!(
+            "Failed to decode anchor tx info: {e}"
+        )))
+    })?;
+    verify_and_populate_l1sload_proofs(
+        &input.l1_storage_proofs,
+        anchor_block_number,
+        &input.taiko.l1_header,
+        &input.l1_headers,
+    )
+    .map_err(|e| {
+        RaikoError::Guest(raiko_lib::prover::ProverError::GuestError(format!(
+            "Failed to verify L1SLOAD proofs: {e}"
+        )))
+    })?;
+
+    Ok(guard)
+}
+
 pub struct Raiko {
     pub l1_chain_spec: ChainSpec,
     pub taiko_chain_spec: ChainSpec,
@@ -138,6 +187,9 @@ impl Raiko {
             &input.taiko.tx_data,
             &input.taiko.anchor_tx,
         );
+
+        let _l1sload_guard = prepare_l1sload_for_execution(input)?;
+
         builder
             .execute_transactions(pool_tx, false)
             .expect("execute");
@@ -216,6 +268,8 @@ impl Raiko {
         let db = create_mem_db(&mut input_owned).unwrap();
         let mut builder = RethBlockBuilder::new(input_owned, db);
 
+        let _l1sload_guard = prepare_l1sload_for_execution(input)?;
+
         let mut pool_txs = vec![input.taiko.anchor_tx.clone().unwrap()];
         pool_txs.extend_from_slice(&origin_pool_txs.as_slice());