feat: added fake sig in executor (#11308)

* feat: added fake sig in executor

* feat: added fake sig in executor

Author: Soubhik-10

crates/anvil/src/eth/backend/executor.rs

@@ -2,7 +2,10 @@ use crate::{
     PrecompileFactory,
     eth::{
         backend::{
-            db::Db, env::Env, mem::op_haltreason_to_instruction_result,
+            cheats::{CheatEcrecover, CheatsManager},
+            db::Db,
+            env::Env,
+            mem::op_haltreason_to_instruction_result,
             validate::TransactionValidator,
         },
         error::InvalidTransactionError,
@@ -15,7 +18,11 @@ use alloy_consensus::{
     Receipt, ReceiptWithBloom, constants::EMPTY_WITHDRAWALS, proofs::calculate_receipt_root,
 };
 use alloy_eips::{eip7685::EMPTY_REQUESTS_HASH, eip7840::BlobParams};
-use alloy_evm::{EthEvm, Evm, eth::EthEvmContext, precompiles::PrecompilesMap};
+use alloy_evm::{
+    EthEvm, Evm,
+    eth::EthEvmContext,
+    precompiles::{DynPrecompile, Precompile, PrecompilesMap},
+};
 use alloy_op_evm::OpEvm;
 use alloy_primitives::{B256, Bloom, BloomInput, Log};
 use anvil_core::eth::{
@@ -28,7 +35,7 @@ use foundry_evm::{
     backend::DatabaseError,
     traces::{CallTraceDecoder, CallTraceNode},
 };
-use foundry_evm_core::either_evm::EitherEvm;
+use foundry_evm_core::{either_evm::EitherEvm, precompiles::EC_RECOVER};
 use op_revm::{L1BlockInfo, OpContext, precompiles::OpPrecompiles};
 use revm::{
     Database, DatabaseRef, Inspector, Journal,
@@ -127,6 +134,7 @@ pub struct TransactionExecutor<'a, Db: ?Sized, V: TransactionValidator> {
     /// Precompiles to inject to the EVM.
     pub precompile_factory: Option<Arc<dyn PrecompileFactory>>,
     pub blob_params: BlobParams,
+    pub cheats: CheatsManager,
 }
 
 impl<DB: Db + ?Sized, V: TransactionValidator> TransactionExecutor<'_, DB, V> {
@@ -344,6 +352,14 @@ impl<DB: Db + ?Sized, V: TransactionValidator> Iterator for &mut TransactionExec
                 inject_precompiles(&mut evm, factory.precompiles());
             }
 
+            let cheats = Arc::new(self.cheats.clone());
+            if cheats.has_recover_overrides() {
+                let cheat_ecrecover = CheatEcrecover::new(Arc::clone(&cheats));
+                evm.precompiles_mut().apply_precompile(&EC_RECOVER, move |_| {
+                    Some(DynPrecompile::new_stateful(move |input| cheat_ecrecover.call(input)))
+                });
+            }
+
             trace!(target: "backend", "[{:?}] executing", transaction.hash());
             // transact and commit the transaction
             match evm.transact_commit(env.tx) {

crates/anvil/src/eth/backend/mem/mod.rs

@@ -1277,6 +1277,7 @@ impl Backend {
             odyssey: self.odyssey,
             optimism: self.is_optimism(),
             blob_params: self.blob_params(),
+            cheats: self.cheats().clone(),
         };
 
         // create a new pending block
@@ -1364,6 +1365,7 @@ impl Backend {
                     precompile_factory: self.precompile_factory.clone(),
                     optimism: self.is_optimism(),
                     blob_params: self.blob_params(),
+                    cheats: self.cheats().clone(),
                 };
                 let executed_tx = executor.execute();
 
@@ -2727,6 +2729,7 @@ impl Backend {
             odyssey: self.odyssey,
             optimism: self.is_optimism(),
             blob_params: self.blob_params(),
+            cheats: self.cheats().clone(),
         };
 
         let _ = executor.execute();

crates/anvil/tests/it/anvil.rs

@@ -7,6 +7,7 @@ use alloy_network::{ReceiptResponse, TransactionBuilder};
 use alloy_primitives::{Address, B256, hex};
 use alloy_provider::Provider;
 use alloy_rpc_types::TransactionRequest;
+use alloy_sol_types::SolCall;
 use anvil::{NodeConfig, spawn};
 
 #[tokio::test(flavor = "multi_thread")]
@@ -170,3 +171,38 @@ async fn test_anvil_recover_signature() {
     let result = contract.testRecover(fake_hash, v, r, s, expected).call().await;
     assert!(result.is_ok(), "ecrecover failed: {:?}", result.err());
 }
+
+#[tokio::test(flavor = "multi_thread")]
+async fn test_fake_signature_transaction() {
+    let (api, handle) = spawn(NodeConfig::test()).await;
+    let provider = handle.http_provider();
+    alloy_sol_types::sol! {
+    #[sol(rpc)]
+        contract TestRecover {
+            function testRecover(bytes32 hash, uint8 v, bytes32 r, bytes32 s, address expected) external pure {
+                address recovered = ecrecover(hash, v, r, s);
+                require(recovered == expected, "ecrecover failed: address mismatch");
+            }
+        }
+    }
+    let bytecode = hex::decode(
+        "0x60808060405234601557610125908161001a8239f35b5f80fdfe60808060405260043610156011575f80fd5b5f3560e01c63bff0b743146023575f80fd5b3460eb5760a036600319011260eb5760243560ff811680910360eb576084356001600160a01b038116929083900360eb5760805f916020936004358252848201526044356040820152606435606082015282805260015afa1560e0575f516001600160a01b031603609057005b60405162461bcd60e51b815260206004820152602260248201527f65637265636f766572206661696c65643a2061646472657373206d69736d61746044820152610c6d60f31b6064820152608490fd5b6040513d5f823e3d90fd5b5f80fdfea264697066735822122006368b42bca31c97f2c409a1cc5186dc899d4255ecc28db7bbb0ad285dc82ae464736f6c634300081c0033",
+    ).unwrap();
+
+    let tx = TransactionRequest::default().with_deploy_code(bytecode);
+    let _receipt = provider.send_transaction(tx.into()).await.unwrap().get_receipt().await.unwrap();
+
+    let sig = alloy_primitives::hex::decode("11".repeat(65)).unwrap();
+    let r = B256::from_slice(&sig[0..32]);
+    let s = B256::from_slice(&sig[32..64]);
+    let v = sig[64];
+    let fake_hash = B256::random();
+    let expected = alloy_primitives::address!("0x1234567890123456789012345678901234567890");
+    api.anvil_impersonate_signature(sig.clone().into(), expected).await.unwrap();
+    let calldata = TestRecover::testRecoverCall { hash: fake_hash, v, r, s, expected }.abi_encode();
+    let tx = TransactionRequest::default().with_input(calldata);
+    let pending = provider.send_transaction(tx.into()).await.unwrap();
+    let result = pending.get_receipt().await;
+
+    assert!(result.is_ok(), "ecrecover failed: {:?}", result.err());
+}