fix fetching callee onchain balance

fuzzland · Sep 17, 2024 · f9052c6 · f9052c6
1 parent 6b1c2d8
commit f9052c6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/evm/onchain/mod.rs b/src/evm/onchain/mod.rs
@@ -268,13 +268,14 @@ where
                 let address = match *interp.instruction_pointer {
                     0xf1 | 0xf2 => {
                         // CALL | CALLCODE
+                        let calee = interp.stack.peek(1).unwrap();
                         #[cfg(feature = "real_balance")]
                         {
                             // Get balance of the callee
-                            host.next_slot = self.endpoint.get_balance(caller);
+                            host.next_slot = self.endpoint.get_balance(convert_u256_to_h160(calee));
                         }
 
-                        interp.stack.peek(1).unwrap()
+                        calee
                     }
                     0xf4 | 0xfa => interp.stack.peek(1).unwrap(),
                     0x3b | 0x3c | 0x3f => interp.stack.peek(0).unwrap(),