1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use super::State;
use crate::state::overlay_account::AccountEntry;
use cfx_rpc_eth_types::StateOverride;
use cfx_statedb::{Result as DbResult, StateDbExt, StateDbGeneric as StateDb};
use cfx_types::{AddressWithSpace, Space};

/// Apply the state override to the state object, only used for rpc call eg
/// eth_call, eth_estimateGas etc.
impl State {
    pub fn new_with_override(
        db: StateDb, state_override: &StateOverride, space: Space,
    ) -> DbResult<Self> {
        let mut state = Self::new(db)?;
        state.apply_override(state_override, space)?;
        Ok(state)
    }

    fn apply_override(
        &mut self, state_override: &StateOverride, space: Space,
    ) -> DbResult<()> {
        assert!(self.checkpoints.read().is_empty());

        let mut cache = self.cache.write();
        for (address, account) in state_override.iter() {
            let addr_with_space = AddressWithSpace {
                address: address.to_owned(),
                space,
            };

            let loaded_account = self.db.get_account(&addr_with_space)?;
            let account_entry = AccountEntry::from_loaded_with_override(
                &addr_with_space,
                loaded_account,
                account,
            );

            cache.insert(addr_with_space, account_entry);
        }
        Ok(())
    }
}