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

fix(tree): root task storage leaf removal (#12950)

Browse Source
This commit is contained in:
Roman Krasiuk
2024-11-28 11:28:07 +01:00
committed by GitHub
parent 064f9393ff
commit b2e1da3ebc
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
crates/engine/tree/src/tree/root.rs
View File

@ -514,12 +514,17 @@ fn update_sparse_trie(
}
for (slot, value) in storage.storage {
let slot_path = Nibbles::unpack(slot);
trie.update_storage_leaf(
address,
slot_path,
alloy_rlp::encode_fixed_size(&value).to_vec(),
)?;
let slot_nibbles = Nibbles::unpack(slot);
if value.is_zero() {
// TODO: handle blinded node error
trie.remove_storage_leaf(address, &slot_nibbles)?;
} else {
trie.update_storage_leaf(
address,
slot_nibbles,
alloy_rlp::encode_fixed_size(&value).to_vec(),
)?;
}
}
storage_roots.insert(address, trie.storage_root(address).unwrap());
@ -527,7 +532,7 @@ fn update_sparse_trie(
// Update accounts with new values and include updated storage roots
for (address, account) in state.accounts {
let path = Nibbles::unpack(address);
let account_nibbles = Nibbles::unpack(address);
if let Some(account) = account {
let storage_root = storage_roots
@ -538,9 +543,10 @@ fn update_sparse_trie(
let mut encoded = Vec::with_capacity(128);
TrieAccount::from((account, storage_root)).encode(&mut encoded as &mut dyn BufMut);
trie.update_account_leaf(path, encoded)?;
trie.update_account_leaf(account_nibbles, encoded)?;
} else {
trie.remove_account_leaf(&path)?;
// TODO: handle blinded node error
trie.remove_account_leaf(&account_nibbles)?;
}
}