mirror of
https://github.com/paradigmxyz/reth.git
synced 2026-04-08 03:01:12 -04:00
78 lines
2.8 KiB
Rust
78 lines
2.8 KiB
Rust
use alloy_primitives::{
|
|
map::{HashMap, HashSet},
|
|
Address, Bytes, B256,
|
|
};
|
|
use reth_storage_errors::provider::ProviderResult;
|
|
use reth_trie::{
|
|
updates::TrieUpdates, AccountProof, HashedPostState, HashedStorage, MultiProof, TrieInput,
|
|
};
|
|
|
|
/// A type that can compute the state root of a given post state.
|
|
#[auto_impl::auto_impl(&, Box, Arc)]
|
|
pub trait StateRootProvider: Send + Sync {
|
|
/// Returns the state root of the `BundleState` on top of the current state.
|
|
///
|
|
/// # Note
|
|
///
|
|
/// It is recommended to provide a different implementation from
|
|
/// `state_root_with_updates` since it affects the memory usage during state root
|
|
/// computation.
|
|
fn state_root(&self, hashed_state: HashedPostState) -> ProviderResult<B256>;
|
|
|
|
/// Returns the state root of the `HashedPostState` on top of the current state but re-uses the
|
|
/// intermediate nodes to speed up the computation. It's up to the caller to construct the
|
|
/// prefix sets and inform the provider of the trie paths that have changes.
|
|
fn state_root_from_nodes(&self, input: TrieInput) -> ProviderResult<B256>;
|
|
|
|
/// Returns the state root of the `HashedPostState` on top of the current state with trie
|
|
/// updates to be committed to the database.
|
|
fn state_root_with_updates(
|
|
&self,
|
|
hashed_state: HashedPostState,
|
|
) -> ProviderResult<(B256, TrieUpdates)>;
|
|
|
|
/// Returns state root and trie updates.
|
|
/// See [`StateRootProvider::state_root_from_nodes`] for more info.
|
|
fn state_root_from_nodes_with_updates(
|
|
&self,
|
|
input: TrieInput,
|
|
) -> ProviderResult<(B256, TrieUpdates)>;
|
|
}
|
|
|
|
/// A type that can compute the storage root for a given account.
|
|
#[auto_impl::auto_impl(&, Box, Arc)]
|
|
pub trait StorageRootProvider: Send + Sync {
|
|
/// Returns the storage root of the `HashedStorage` for target address on top of the current
|
|
/// state.
|
|
fn storage_root(&self, address: Address, hashed_storage: HashedStorage)
|
|
-> ProviderResult<B256>;
|
|
}
|
|
|
|
/// A type that can generate state proof on top of a given post state.
|
|
#[auto_impl::auto_impl(&, Box, Arc)]
|
|
pub trait StateProofProvider: Send + Sync {
|
|
/// Get account and storage proofs of target keys in the `HashedPostState`
|
|
/// on top of the current state.
|
|
fn proof(
|
|
&self,
|
|
input: TrieInput,
|
|
address: Address,
|
|
slots: &[B256],
|
|
) -> ProviderResult<AccountProof>;
|
|
|
|
/// Generate [`MultiProof`] for target hashed account and corresponding
|
|
/// hashed storage slot keys.
|
|
fn multiproof(
|
|
&self,
|
|
input: TrieInput,
|
|
targets: HashMap<B256, HashSet<B256>>,
|
|
) -> ProviderResult<MultiProof>;
|
|
|
|
/// Get trie witness for provided state.
|
|
fn witness(
|
|
&self,
|
|
input: TrieInput,
|
|
target: HashedPostState,
|
|
) -> ProviderResult<HashMap<B256, Bytes>>;
|
|
}
|