jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(trie): SparseStateTrie::storage_trie_mut (#12913)

Browse Source
This commit is contained in:
Roman Krasiuk
2024-11-27 20:18:31 +01:00
committed by GitHub
parent 00c5b69af4
commit 47245642ca
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
crates/trie/sparse/src/state.rs
View File

@ -1,6 +1,4 @@
use std::iter::Peekable;
use crate::{SparseStateTrieError, SparseStateTrieResult, SparseTrie};
use crate::{RevealedSparseTrie, SparseStateTrieError, SparseStateTrieResult, SparseTrie};
use alloy_primitives::{
map::{HashMap, HashSet},
Bytes, B256,
@ -10,6 +8,7 @@ use reth_trie_common::{
updates::{StorageTrieUpdates, TrieUpdates},
MultiProof, Nibbles, TrieNode,
};
use std::iter::Peekable;
/// Sparse state trie representing lazy-loaded Ethereum state trie.
#[derive(Default, Debug)]
@ -47,6 +46,11 @@ impl SparseStateTrie {
self.revealed.get(account).is_some_and(|slots| slots.contains(slot))
}
/// Returned mutable reference to storage sparse trie if it was revealed.
pub fn storage_trie_mut(&mut self, account: &B256) -> Option<&mut RevealedSparseTrie> {
self.storages.get_mut(account).and_then(|e| e.as_revealed_mut())
}
/// Reveal unknown trie paths from provided leaf path and its proof for the account.
/// NOTE: This method does not extensively validate the proof.
pub fn reveal_account(