get initial onchain balance for call receiver

Author: Raz0r

src/evm/host.rs

@@ -79,7 +79,7 @@ use crate::{
         mutator::AccessPattern,
         onchain::{
             abi_decompiler::fetch_abi_evmole,
-            flashloan::{register_borrow_txn, Flashloan},
+            flashloan::{register_borrow_txn, Flashloan}, OnChain,
         },
         types::{as_u64, generate_random_address, is_zero, EVMAddress, EVMU256},
         vm::{is_reverted_or_control_leak, EVMState, SinglePostExecution, IN_DEPLOY, IS_FAST_CALL_STATIC},
@@ -1480,6 +1480,12 @@ where
             if let Some(balance) = self.evmstate.get_balance(&receiver) {
                 self.evmstate.set_balance(receiver, *balance + value);
             } else {
+                let on_chain_middleware = self.middlewares.read().unwrap().iter().find_map(|m| {
+                    m.deref().borrow().as_any().downcast_ref::<OnChain>().cloned()
+                });
+                if let Some(mut on_chain_middleware) = on_chain_middleware {
+                    self.next_slot = on_chain_middleware.endpoint.get_balance(receiver);
+                }
                 self.evmstate.set_balance(receiver, self.next_slot + value);
             };
         }