update code_hash query

Signed-off-by: smtmfft <[email protected]>
taikoxyz · Dec 26, 2024 · 4a9b703 · 4a9b703
1 parent 4ad92f9
commit 4a9b703
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/core/src/preflight/reth_preflight.rs b/core/src/preflight/reth_preflight.rs
@@ -4,7 +4,7 @@ use crate::{
     interfaces::{RaikoError, RaikoResult},
     provider::{db::ProviderDb, BlockDataProvider},
 };
-use alloy_primitives::{Address, Bytes, B256};
+use alloy_primitives::{Bytes, B256};
 use alloy_rpc_types::eth::{
     Block as AlloyEthBlock, EIP1186AccountProofResponse, Header as AlloyEthHeader,
 };
@@ -31,8 +31,6 @@ pub struct PreFlightRpcData {
     pub parent_header: AlloyEthHeader,
     /// The account proofs.
     pub account_proofs: Vec<EIP1186AccountProofResponse>,
-    /// The account code bytes.
-    pub account_codes: HashMap<Address, Bytes>,
     /// The parent account proofs.
     pub parent_account_proofs: Vec<EIP1186AccountProofResponse>,
     /// The contracts used.

diff --git a/core/src/provider/preflight_rpc.rs b/core/src/provider/preflight_rpc.rs
@@ -118,8 +118,8 @@ impl BlockDataProvider for RethPreflightBlockDataProvider {
                 let nonce = account_proof.nonce;
                 let code_hash = account_proof.code_hash;
                 let code = preflight_data
-                    .account_codes
-                    .get(&account_proof.address)
+                    .contracts
+                    .get(&account_proof.code_hash)
                     .map(|code| Bytecode::new_raw(code.clone()))
                     .unwrap();
                 assert_eq!(code_hash, code.hash_slow());