Skip to main content

revme/cmd/blockchaintest/
post_block.rs

1use revm::{
2    context::{Block, ContextTr, JournalTr},
3    handler::EvmTr,
4    primitives::{address, hardfork::SpecId, Address, Bytes, ONE_ETHER, ONE_GWEI, U256},
5    statetest_types::blockchain::Withdrawal,
6    Database, DatabaseCommit, SystemCallCommitEvm,
7};
8
9/// Post block transition that includes:
10///   * Block and uncle rewards before the Merge/Paris hardfork.
11///   * Withdrawals (EIP-4895)
12///   * Post-block system calls: EIP-7002 (withdrawal requests) and EIP-7251 (consolidation requests)
13///
14/// # Note
15///
16/// Uncle rewards are not implemented yet.
17#[inline]
18pub fn post_block_transition<
19    'a,
20    DB: Database + DatabaseCommit + 'a,
21    EVM: SystemCallCommitEvm<Error: core::fmt::Debug> + EvmTr<Context: ContextTr<Db = DB>>,
22>(
23    evm: &mut EVM,
24    block: impl Block,
25    withdrawals: &[Withdrawal],
26    spec: SpecId,
27) -> Result<(), EVM::Error> {
28    // block reward
29    let block_reward = block_reward(spec, 0);
30    if block_reward != 0 {
31        evm.ctx_mut()
32            .journal_mut()
33            .balance_incr(block.beneficiary(), U256::from(block_reward))
34            .expect("Db actions to pass");
35    }
36
37    // withdrawals
38    if spec.is_enabled_in(SpecId::SHANGHAI) {
39        for withdrawal in withdrawals {
40            evm.ctx_mut()
41                .journal_mut()
42                .balance_incr(
43                    withdrawal.address,
44                    withdrawal.amount.saturating_mul(U256::from(ONE_GWEI)),
45                )
46                .expect("Db actions to pass");
47        }
48    }
49
50    evm.commit_inner();
51
52    // EIP-7002: Withdrawal requests system call
53    if spec.is_enabled_in(SpecId::PRAGUE) {
54        system_call_eip7002_withdrawal_request(evm)?;
55    }
56
57    // EIP-7251: Consolidation requests system call
58    if spec.is_enabled_in(SpecId::PRAGUE) {
59        system_call_eip7251_consolidation_request(evm)?;
60    }
61
62    Ok(())
63}
64
65/// Block reward for a block.
66#[inline]
67pub const fn block_reward(spec: SpecId, ommers: usize) -> u128 {
68    if spec.is_enabled_in(SpecId::MERGE) {
69        return 0;
70    }
71
72    let reward = if spec.is_enabled_in(SpecId::CONSTANTINOPLE) {
73        ONE_ETHER * 2
74    } else if spec.is_enabled_in(SpecId::BYZANTIUM) {
75        ONE_ETHER * 3
76    } else {
77        ONE_ETHER * 5
78    };
79
80    reward + (reward >> 5) * ommers as u128
81}
82
83pub const WITHDRAWAL_REQUEST_ADDRESS: Address =
84    address!("0x00000961Ef480Eb55e80D19ad83579A64c007002");
85
86/// EIP-7002: Withdrawal requests system call
87pub(crate) fn system_call_eip7002_withdrawal_request<EVM>(evm: &mut EVM) -> Result<(), EVM::Error>
88where
89    EVM: SystemCallCommitEvm<Error: core::fmt::Debug>,
90{
91    // empty data is valid for EIP-7002
92    evm.system_call_commit(WITHDRAWAL_REQUEST_ADDRESS, Bytes::new())?;
93    Ok(())
94}
95
96pub const CONSOLIDATION_REQUEST_ADDRESS: Address =
97    address!("0x0000BBdDc7CE488642fb579F8B00f3a590007251");
98
99/// EIP-7251: Consolidation requests system call
100pub(crate) fn system_call_eip7251_consolidation_request<EVM>(
101    evm: &mut EVM,
102) -> Result<(), EVM::Error>
103where
104    EVM: SystemCallCommitEvm<Error: core::fmt::Debug>,
105{
106    evm.system_call_commit(CONSOLIDATION_REQUEST_ADDRESS, Bytes::new())?;
107    Ok(())
108}