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

trie: fix the double reference fix (i.e. roots *can* be double refed)

Browse source
This commit is contained in:
Péter Szilágyi 2018-01-24 17:30:17 +02:00
parent 4315480e09
commit b2d8f148d2
No known key found for this signature in database
GPG key ID: E9AE538CEDF8293D
1 changed files with 14 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
trie/database.go
View file

@ -63,7 +63,7 @@ type Database struct {
type cachedNode struct {
blob []byte // Cached data block of the trie node
parents int // Number of live nodes referencing this one
children map[common.Hash]struct{} // Children referenced by this nodes
children map[common.Hash]int // Children referenced by this nodes
}
// NewDatabase creates a new trie database to store ephemeral trie content before
@ -72,7 +72,7 @@ func NewDatabase(diskdb ethdb.Database) *Database {
return &Database{
diskdb: diskdb,
nodes: map[common.Hash]*cachedNode{
common.Hash{}: {children: make(map[common.Hash]struct{})},
common.Hash{}: {children: make(map[common.Hash]int)},
},
preimages: make(map[common.Hash][]byte),
}
@ -99,7 +99,7 @@ func (db *Database) insert(hash common.Hash, blob []byte) {
}
db.nodes[hash] = &cachedNode{
blob: common.CopyBytes(blob),
children: make(map[common.Hash]struct{}),
children: make(map[common.Hash]int),
}
db.size += common.StorageSize(common.HashLength + len(blob))
}
@ -185,12 +185,12 @@ func (db *Database) reference(child common.Hash, parent common.Hash) {
if !ok {
return
}
// If the reference already exists, don't duplicate it
if _, ok = db.nodes[parent].children[child]; ok {
// If the reference already exists, only duplicate for roots
if _, ok = db.nodes[parent].children[child]; ok && parent != (common.Hash{}) {
return
}
node.parents++
db.nodes[parent].children[child] = struct{}{}
db.nodes[parent].children[child]++
}
// Dereference removes an existing reference from a parent node to a child node.
@ -211,9 +211,13 @@ func (db *Database) Dereference(child common.Hash, parent common.Hash) {
// dereference is the private locked version of Dereference.
func (db *Database) dereference(child common.Hash, parent common.Hash) {
// Dereference the parent-child relationship
delete(db.nodes[parent].children, child)
// Dereference the parent-child
node := db.nodes[parent]
node.children[child]--
if node.children[child] == 0 {
delete(node.children, child)
}
// If the node does not exist, it's a previously comitted node.
node, ok := db.nodes[child]
if !ok {