fix: use correct state for eth_getCode and eth_getStorageAt

Author: LesnyRumcajs

CHANGELOG.md

@@ -37,6 +37,8 @@
 
 - [#6327](https://github.com/ChainSafe/forest/pull/6327) Fixed: Forest returns 404 for all invalid api paths.
 
+- [#6140](https://github.com/ChainSafe/forest/pull/6140) Fixed the `eth_getLogs` RPC method to accept `None` as the `address` parameter.
+
 ## Forest v0.30.5 "Dulce de Leche"
 
 Non-mandatory release supporting new API methods and addressing a critical panic issue.
@@ -119,7 +121,6 @@ The release includes new CLI commands for snapshot management and state inspecti
 
 - [#6103](https://github.com/ChainSafe/forest/pull/6103) Fixed `eth_getTransactionCount` to return the nonce of the requested tipset and not its parent.
 
-- [#6140](https://github.com/ChainSafe/forest/pull/6140) Fixed the `eth_getLogs` RPC method to accept `None` as the `address` parameter.
 
 ## Forest v0.30.1 "Laurelin"
 

src/rpc/methods/eth.rs

@@ -2157,9 +2157,10 @@ impl RpcMethod<2> for EthGetCode {
             ..Default::default()
         };
 
+        let (state, _) = ctx.state_manager.tipset_state(&ts).await?;
         let api_invoc_result = 'invoc: {
             for ts in ts.chain(ctx.store()) {
-                match ctx.state_manager.call(&message, Some(ts)) {
+                match ctx.state_manager.call_on_state(state, &message, Some(ts)) {
                     Ok(res) => {
                         break 'invoc res;
                     }
@@ -2208,10 +2209,8 @@ impl RpcMethod<3> for EthGetStorageAt {
             ResolveNullTipset::TakeOlder,
         )?;
         let to_address = FilecoinAddress::try_from(&eth_address)?;
-        let Some(actor) = ctx
-            .state_manager
-            .get_actor(&to_address, *ts.parent_state())?
-        else {
+        let (state, _) = ctx.state_manager.tipset_state(&ts).await?;
+        let Some(actor) = ctx.state_manager.get_actor(&to_address, state)? else {
             return Ok(make_empty_result());
         };
 
@@ -2230,7 +2229,7 @@ impl RpcMethod<3> for EthGetStorageAt {
         };
         let api_invoc_result = 'invoc: {
             for ts in ts.chain(ctx.store()) {
-                match ctx.state_manager.call(&message, Some(ts)) {
+                match ctx.state_manager.call_on_state(state, &message, Some(ts)) {
                     Ok(res) => {
                         break 'invoc res;
                     }

src/state_manager/mod.rs

@@ -582,13 +582,14 @@ where
     #[instrument(skip(self, rand))]
     fn call_raw(
         &self,
+        state_cid: Option<Cid>,
         msg: &Message,
         rand: ChainRand<DB>,
         tipset: &Tipset,
     ) -> Result<ApiInvocResult, Error> {
         let mut msg = msg.clone();
 
-        let state_cid = tipset.parent_state();
+        let state_cid = state_cid.unwrap_or(*tipset.parent_state());
 
         let tipset_messages = self
             .chain_store()
@@ -606,14 +607,14 @@ where
         let mut vm = VM::new(
             ExecutionContext {
                 heaviest_tipset: tipset.clone(),
-                state_tree_root: *state_cid,
+                state_tree_root: state_cid,
                 epoch: height,
                 rand: Box::new(rand),
                 base_fee: tipset.block_headers().first().parent_base_fee.clone(),
                 circ_supply: genesis_info.get_vm_circulating_supply(
                     height,
                     self.blockstore(),
-                    state_cid,
+                    &state_cid,
                 )?,
                 chain_config: self.chain_config().clone(),
                 chain_index: self.chain_index().clone(),
@@ -660,7 +661,20 @@ where
     pub fn call(&self, message: &Message, tipset: Option<Tipset>) -> Result<ApiInvocResult, Error> {
         let ts = tipset.unwrap_or_else(|| self.heaviest_tipset());
         let chain_rand = self.chain_rand(ts.clone());
-        self.call_raw(message, chain_rand, &ts)
+        self.call_raw(None, message, chain_rand, &ts)
+    }
+
+    /// Same as [`StateManager::call`] but runs the message on the given state and not
+    /// on the parent state of the tipset.
+    pub fn call_on_state(
+        &self,
+        state_cid: Cid,
+        message: &Message,
+        tipset: Option<Tipset>,
+    ) -> Result<ApiInvocResult, Error> {
+        let ts = tipset.unwrap_or_else(|| self.cs.heaviest_tipset());
+        let chain_rand = self.chain_rand(ts.clone());
+        self.call_raw(Some(state_cid), message, chain_rand, &ts)
     }
 
     pub async fn apply_on_state_with_gas(