Add two-way storage compare

crates/telos/bin/Cargo.toml

@@ -16,6 +16,8 @@ reth-chainspec.workspace = true
 reth-provider.workspace = true
 reth-node-telos.workspace = true
 reth-telos-rpc.workspace = true
+reth-db.workspace = true
+alloy-primitives.workspace = true
 
 
 clap = { workspace = true, features = ["derive", "env"] }

crates/telos/bin/src/main.rs

@@ -4,13 +4,18 @@
 static ALLOC: reth_cli_util::allocator::Allocator = reth_cli_util::allocator::new_allocator();
 
 use clap::Parser;
+use tracing::{info, warn};
 use reth::args::utils::EthereumChainSpecParser;
 use reth_node_builder::{engine_tree_config::TreeConfig, EngineNodeLauncher};
 use reth::cli::Cli;
 use reth_node_telos::{TelosArgs, TelosNode};
 use reth_node_telos::node::TelosAddOns;
 use reth_provider::providers::BlockchainProvider2;
 use reth_telos_rpc::TelosClient;
+use reth::primitives::BlockId;
+use reth::rpc::types::BlockNumberOrTag;
+use reth_provider::{DatabaseProviderFactory, StateProviderFactory};
+use reth_db::{PlainAccountState, PlainStorageState};
 
 
 #[cfg(feature = "telos")]
@@ -45,6 +50,10 @@ fn main() {
                 handle.node_exit_future.await
             },
             false => {
+                let two_way_storage_compare = telos_args.two_way_storage_compare.clone();
+                let telos_rpc = telos_args.telos_endpoint.clone();
+                let block_delta = telos_args.block_delta.clone();
+
                 let handle = builder
                     .node(TelosNode::new(telos_args.clone()))
                     .extend_rpc_modules(move |ctx| {
@@ -59,6 +68,32 @@ fn main() {
                     .launch()
                     .await?;
 
+                match two_way_storage_compare {
+                    true => {
+                        if telos_rpc.is_none() {
+                            warn!("Telos RPC Endpoint is not specified, skipping two-way storage compare");
+                        } else if block_delta.is_none() {
+                            warn!("Block delta is not specified, skipping two-way storage compare");
+                        } else {
+                            info!("Fetching account and accountstate from Telos native RPC (Can take a long time)...");
+
+                            let (account_table, accountstate_table, block_number) = reth_node_telos::two_way_storage_compare::get_telos_tables(telos_rpc.unwrap().as_str(), block_delta.unwrap()).await;
+
+                            info!("Two-way comparing state (Reth vs. Telos) at height: {:?}", block_number);
+
+                            let state_at_specific_height = handle.node.provider.state_by_block_id(BlockId::Number(BlockNumberOrTag::Number(block_number.as_u64().unwrap()))).unwrap();
+                            let plain_account_state = handle.node.provider.database_provider_ro().unwrap().table::<PlainAccountState>().unwrap();
+                            let plain_storage_state = handle.node.provider.database_provider_ro().unwrap().table::<PlainStorageState>().unwrap();
+
+                            let match_counter = reth_node_telos::two_way_storage_compare::two_side_state_compare(account_table, accountstate_table, state_at_specific_height, plain_account_state, plain_storage_state).await;
+                            match_counter.print();
+
+                            info!("Comparing done");
+                        }
+                    }
+                    _ => {}
+                }
+
                 handle.node_exit_future.await
             }
         }

crates/telos/node/Cargo.toml

@@ -35,6 +35,8 @@ reth-telos-rpc.workspace = true
 reth-tracing.workspace = true
 reth-transaction-pool.workspace = true
 reth-telos-rpc-engine-api.workspace = true
+alloy-primitives.workspace = true
+reth-db.workspace = true
 
 clap.workspace = true
 serde = { workspace = true, features = ["derive"] }

crates/telos/node/src/args.rs

@@ -45,6 +45,14 @@ pub struct TelosArgs {
     /// a batch of downloaded blocks.
     #[arg(long = "engine.max-execute-block-batch-size", requires = "experimental", default_value_t = DEFAULT_MAX_EXECUTE_BLOCK_BATCH_SIZE)]
     pub max_execute_block_batch_size: usize,
+
+    /// Enable Two-way storage compare between reth and telos
+    #[arg(long = "telos.two_way_storage_compare", default_value = "false")]
+    pub two_way_storage_compare: bool,
+
+    /// Block delta between native and EVM
+    #[arg(long = "telos.block_delta")]
+    pub block_delta: Option<u32>,
 }
 
 impl From<TelosArgs> for TelosClientArgs {

crates/telos/node/src/lib.rs

@@ -11,6 +11,7 @@
 
 pub mod args;
 pub mod node;
+pub mod two_way_storage_compare;
 
 pub use crate::args::TelosArgs;
 pub use crate::node::TelosNode;