revmc_statetest/
merkle_trie.rs

1use std::convert::Infallible;
2
3use alloy_trie::{HashBuilder, Nibbles, TrieAccount, root::storage_root_unhashed};
4use revm_context_interface::result::{EVMError, ExecutionResult, HaltReason, InvalidTransaction};
5use revm_database::{EmptyDB, PlainAccount, State, bal::EvmDatabaseError};
6use revm_primitives::{Address, B256, Log, keccak256};
7
8pub struct TestValidationResult {
9    pub logs_root: B256,
10    pub state_root: B256,
11}
12
13pub fn compute_test_roots(
14    exec_result: &Result<
15        ExecutionResult<HaltReason>,
16        EVMError<EvmDatabaseError<Infallible>, InvalidTransaction>,
17    >,
18    db: &State<EmptyDB>,
19) -> TestValidationResult {
20    TestValidationResult {
21        logs_root: log_rlp_hash(exec_result.as_ref().map(|r| r.logs()).unwrap_or_default()),
22        state_root: state_merkle_trie_root(db.cache.trie_account()),
23    }
24}
25
26pub fn log_rlp_hash(logs: &[Log]) -> B256 {
27    let mut out = Vec::with_capacity(alloy_rlp::list_length(logs));
28    alloy_rlp::encode_list(logs, &mut out);
29    keccak256(&out)
30}
31
32pub fn state_merkle_trie_root<'a>(
33    accounts: impl IntoIterator<Item = (Address, &'a PlainAccount)>,
34) -> B256 {
35    let mut vec: Vec<_> = accounts
36        .into_iter()
37        .map(|(address, acc)| {
38            let storage_root = storage_root_unhashed(
39                acc.storage
40                    .iter()
41                    .filter(|(_k, v)| !v.is_zero())
42                    .map(|(k, v)| (B256::from(*k), *v)),
43            );
44            (
45                keccak256(address),
46                TrieAccount {
47                    nonce: acc.info.nonce,
48                    balance: acc.info.balance,
49                    storage_root,
50                    code_hash: acc.info.code_hash,
51                },
52            )
53        })
54        .collect();
55    vec.sort_unstable_by_key(|(key, _)| *key);
56
57    let mut hb = HashBuilder::default();
58    let mut account_rlp_buf = Vec::new();
59    for (hashed_key, account) in vec {
60        account_rlp_buf.clear();
61        alloy_rlp::Encodable::encode(&account, &mut account_rlp_buf);
62        hb.add_leaf(Nibbles::unpack(hashed_key), &account_rlp_buf);
63    }
64    hb.root()
65}
revmc_statetest/merkle_trie.rs

revmc_statetest/
merkle_trie.rs
