trevm 0.34.2

A typestate API wrapper for the revm EVM implementation
Documentation 
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

use crate::fill::traits::{Cfg, Tx};
use revm::context::{BlockEnv, CfgEnv, TxEnv};

/// A [`Cfg`] that disables gas-related checks and payment of the
/// beneficiary reward, while leaving other cfg options unchanged.
///
/// ## Warning
///
/// This filler relies on the following optional features:
/// - `optional_balance_check`
/// - `optional_beneficiary_reward`
/// - `optional_gas_refund`
/// - `optional_no_base_fee`
///
/// It will disable the corresponding checks if the features are enabled. **If
/// none of the features are enabled, this filler will do nothing.**
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
pub struct DisableGasChecks;

impl Cfg for DisableGasChecks {
    #[allow(unused_variables)]
    fn fill_cfg_env(&self, cfg_env: &mut CfgEnv) {
        #[cfg(feature = "optional_balance_check")]
        {
            cfg_env.disable_balance_check = true;
        }
        #[cfg(feature = "optional_no_base_fee")]
        {
            cfg_env.disable_base_fee = true;
        }
    }
}

/// A [`Cfg`] that disables the chain ID check, while leaving other [`CfgEnv`]
/// attributes untouched.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
pub struct DisableChainIdCheck;

impl Cfg for DisableChainIdCheck {
    fn fill_cfg_env(&self, cfg_env: &mut CfgEnv) {
        {
            cfg_env.tx_chain_id_check = false;
        }
    }
}

/// A [`Cfg`] that disables the nonce check, while leaving other [`CfgEnv`]
/// attributes untouched.
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
pub struct DisableNonceCheck;

impl Cfg for DisableNonceCheck {
    #[allow(unused_variables)]
    fn fill_cfg_env(&self, cfg_env: &mut CfgEnv) {
        cfg_env.disable_nonce_check = true;
    }
}

/// Prime the EVM for gas estimation. This filler is all of [`Cfg`], and
/// [`Tx`]. It is used internally by [`crate::Trevm::estimate_gas`], and is
/// considered a low-level API. Generally it is not correct to import this
/// type.
#[cfg(feature = "estimate_gas")]
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
#[repr(transparent)]
pub(crate) struct GasEstimationFiller {
    pub(crate) gas_limit: u64,
}

#[cfg(feature = "estimate_gas")]
impl From<u64> for GasEstimationFiller {
    fn from(gas_limit: u64) -> Self {
        Self { gas_limit }
    }
}

#[cfg(feature = "estimate_gas")]
impl Cfg for GasEstimationFiller {
    fn fill_cfg_env(&self, cfg_env: &mut CfgEnv) {
        cfg_env.disable_base_fee = true;
        cfg_env.disable_eip3607 = true;
        DisableNonceCheck.fill_cfg_env(cfg_env);
    }

    fn fill_cfg<Db: revm::Database, Insp, Inst, Prec, Frame>(
        &self,
        evm: &mut revm::context::Evm<crate::helpers::Ctx<Db>, Insp, Inst, Prec, Frame>,
    ) {
        evm.ctx.modify_cfg(|cfg_env| self.fill_cfg_env(cfg_env));

        let chain_id = evm.ctx.cfg.chain_id;

        evm.ctx.modify_tx(|tx_env| {
            tx_env.chain_id = Some(chain_id);
        });
    }
}

#[cfg(feature = "estimate_gas")]
impl Tx for GasEstimationFiller {
    fn fill_tx_env(&self, tx_env: &mut TxEnv) {
        tx_env.gas_limit = self.gas_limit;
    }
}

#[cfg(feature = "call")]
pub(crate) struct CallFiller {
    pub gas_limit: u64,
}

#[cfg(feature = "call")]
impl Cfg for CallFiller {
    fn fill_cfg_env(&self, cfg_env: &mut CfgEnv) {
        cfg_env.disable_base_fee = true;
        cfg_env.disable_eip3607 = true;
        DisableNonceCheck.fill_cfg_env(cfg_env);
    }

    fn fill_cfg<Db: revm::Database, Insp, Inst, Prec, Frame>(
        &self,
        evm: &mut revm::context::Evm<crate::helpers::Ctx<Db>, Insp, Inst, Prec, Frame>,
    ) {
        evm.ctx.modify_cfg(|cfg_env| self.fill_cfg_env(cfg_env));

        let chain_id = evm.ctx.cfg.chain_id;

        evm.ctx.modify_tx(|tx_env| {
            tx_env.chain_id = Some(chain_id);
        });
    }
}

#[cfg(feature = "call")]
impl crate::Block for CallFiller {
    fn fill_block_env(&self, block_env: &mut BlockEnv) {
        block_env.gas_limit = self.gas_limit;
    }
}