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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use crate::try_loaded;
use cfx_statedb::Result as DbResult;
use cfx_types::{AddressSpaceUtil, AddressWithSpace, Space, H256};
use cfx_vm_types::ActionParams;
use primitives::AccessListItem;

use super::State;

impl State {
    pub fn is_warm_account(&self, address: &AddressWithSpace) -> bool {
        if self
            .tx_access_list
            .as_ref()
            .map_or(false, |x| x.contains_key(address))
        {
            return true;
        }
        self.cache.read().get(address).map_or(false, |x| x.warm)
    }

    pub fn is_warm_storage_entry(
        &self, address: &AddressWithSpace, key: &H256,
    ) -> DbResult<bool> {
        if self.is_warm_storage_entry_in_access_list(address, key) {
            return Ok(true);
        }

        let acc = try_loaded!(self.read_account_lock(address));
        Ok(acc.is_warm_storage_entry(&key[..]))
    }

    fn is_warm_storage_entry_in_access_list(
        &self, address: &AddressWithSpace, key: &H256,
    ) -> bool {
        let Some(access_list) = self.tx_access_list.as_ref() else {
            return false;
        };
        let Some(account) = access_list.get(address) else {
            return false;
        };
        account.contains(key)
    }

    pub fn set_tx_access_list(
        &mut self, space: Space, access_list: &[AccessListItem],
    ) {
        let access_list = access_list
            .iter()
            .map(|x| {
                (
                    x.address.with_space(space),
                    x.storage_keys.iter().cloned().collect(),
                )
            })
            .collect();

        self.tx_access_list = Some(access_list);
    }

    pub fn touch_tx_addresses(&self, params: &ActionParams) -> DbResult<()> {
        self.touch(&params.address.with_space(params.space))?;
        self.touch(&params.sender.with_space(params.space))?;
        Ok(())
    }

    pub fn clear_tx_access_list(&mut self) { self.tx_access_list = None; }

    pub fn update_state_post_tx_execution(
        &mut self, retain_transient_storage: bool,
    ) {
        self.clear_tx_access_list();
        self.commit_cache(retain_transient_storage);
    }
}