mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-22 19:50:36 +00:00
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package trie
|
|
|
|
import (
|
|
"github.com/ledgerwatch/erigon/common"
|
|
"github.com/ledgerwatch/erigon/core/types/accounts"
|
|
)
|
|
|
|
type keyTransformFunc func([]byte) []byte
|
|
|
|
func transformSubTrie(nd node, hex []byte, newTrie *Trie, transformFunc keyTransformFunc) {
|
|
switch n := nd.(type) {
|
|
case nil:
|
|
return
|
|
case valueNode:
|
|
nCopy := make(valueNode, len(n))
|
|
copy(nCopy, n)
|
|
_, newTrie.root = newTrie.insert(newTrie.root, transformFunc(hex), nCopy)
|
|
return
|
|
case *accountNode:
|
|
accountCopy := accounts.NewAccount()
|
|
accountCopy.Copy(&n.Account)
|
|
var code []byte = nil
|
|
if n.code != nil {
|
|
code = make([]byte, len(n.code))
|
|
copy(code, n.code)
|
|
}
|
|
_, newTrie.root = newTrie.insert(newTrie.root, transformFunc(hex), &accountNode{accountCopy, nil, true, code, n.codeSize})
|
|
aHex := hex
|
|
if aHex[len(aHex)-1] == 16 {
|
|
aHex = aHex[:len(aHex)-1]
|
|
}
|
|
transformSubTrie(n.storage, aHex, newTrie, transformFunc)
|
|
case hashNode:
|
|
_, newTrie.root = newTrie.insert(newTrie.root, transformFunc(hex), hashNode{hash: common.CopyBytes(n.hash)})
|
|
return
|
|
case *shortNode:
|
|
var hexVal []byte
|
|
hexVal = concat(hex, n.Key...)
|
|
transformSubTrie(n.Val, hexVal, newTrie, transformFunc)
|
|
case *duoNode:
|
|
i1, i2 := n.childrenIdx()
|
|
hex1 := make([]byte, len(hex)+1)
|
|
copy(hex1, hex)
|
|
hex1[len(hex)] = i1
|
|
hex2 := make([]byte, len(hex)+1)
|
|
copy(hex2, hex)
|
|
hex2[len(hex)] = i2
|
|
transformSubTrie(n.child1, hex1, newTrie, transformFunc)
|
|
transformSubTrie(n.child2, hex2, newTrie, transformFunc)
|
|
case *fullNode:
|
|
for i, child := range n.Children {
|
|
if child != nil {
|
|
transformSubTrie(child, concat(hex, byte(i)), newTrie, transformFunc)
|
|
}
|
|
}
|
|
default:
|
|
panic("")
|
|
}
|
|
}
|