forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-20 02:42:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

core/state: check dirty storage in post-check

Browse source
This commit is contained in:
Martin Holst Swende 2024-04-24 13:10:23 +02:00
parent 3380364a4a
commit 29ef67ccfc
No known key found for this signature in database
GPG key ID: 683B438C05A5DDF0
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
core/state/statedb_test.go
View file

@ -558,10 +558,14 @@ func forEachStorage(s *StateDB, addr common.Address, cb func(key, value common.H
if err != nil { if err != nil {
return err return err
} }
it := trie.NewIterator(trieIt) var (
it = trie.NewIterator(trieIt)
visited = make(map[common.Hash]bool)
)
for it.Next() { for it.Next() {
key := common.BytesToHash(s.trie.GetKey(it.Key)) key := common.BytesToHash(s.trie.GetKey(it.Key))
visited[key] = true
if value, dirty := so.dirtyStorage[key]; dirty { if value, dirty := so.dirtyStorage[key]; dirty {
if !cb(key, value) { if !cb(key, value) {
return nil return nil
@ -579,6 +583,16 @@ func forEachStorage(s *StateDB, addr common.Address, cb func(key, value common.H
} }
} }
} }
// Now visit any remaining dirty storage which is not in trie
for key, value := range so.dirtyStorage {
if visited[key] {
continue
}
if !cb(key, value) {
return nil
}
}
return nil return nil
} }