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

use crate::{helpers::Ctx, EvmExtUnchecked, EvmNeedsTx};
use alloy::{
    eips::eip4895::Withdrawal,
    primitives::{map::HashMap, U256},
};
use revm::{context::result::EVMError, Database, DatabaseCommit, Inspector};

impl<Db, Insp> EvmNeedsTx<Db, Insp>
where
    Db: Database + DatabaseCommit,
    Insp: Inspector<Ctx<Db>>,
{
    /// Apply the withdrawals to the EVM state.
    pub fn apply_withdrawals<'b>(
        &mut self,
        withdrawals: impl IntoIterator<Item = &'b Withdrawal>,
    ) -> Result<(), EVMError<Db::Error>> {
        // We need to apply the withdrawals by incrementing the balances of the
        // respective accounts, then committing the changes to the database.
        let mut changes = HashMap::default();

        let increments = withdrawals
            .into_iter()
            .map(|withdrawal| (withdrawal.address, withdrawal.amount as u128))
            .filter(|(_, amount)| *amount != 0);

        for (address, amount) in increments {
            let mut acct =
                self.inner_mut_unchecked().account(address).map_err(EVMError::Database)?;
            acct.info.balance = acct.info.balance.saturating_add(U256::from(amount));
            acct.mark_touch();
            changes.insert(address, acct);
        }

        self.commit_unchecked(changes);
        Ok(())
    }
}

#[cfg(test)]
mod test {
    use alloy::{
        eips::eip4895::{Withdrawal, GWEI_TO_WEI},
        primitives::{Address, U256},
    };

    use crate::{NoopBlock, NoopCfg};

    const USER: Address = Address::with_last_byte(0x42);

    const WITHDRAWALS: &[Withdrawal] =
        &[Withdrawal { validator_index: 1, index: 1, address: USER, amount: 100 * GWEI_TO_WEI }];

    #[test]
    fn test_eip4895() {
        let mut trevm = crate::test_utils::test_trevm().fill_cfg(&NoopCfg).fill_block(&NoopBlock);

        assert_eq!(trevm.try_read_balance(USER).unwrap(), U256::ZERO);

        trevm.apply_withdrawals(WITHDRAWALS).unwrap();

        assert_eq!(trevm.try_read_balance(USER).unwrap(), U256::from(100 * GWEI_TO_WEI));
    }
}