mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-03-06 11:25:02 +00:00
21920207e4
Previously, PathDB used a single buffer to aggregate database writes, which needed to be flushed atomically. However, flushing large amounts of data (e.g., 256MB) caused significant overhead, often blocking the system for around 3 seconds during the flush. To mitigate this overhead and reduce performance spikes, a double-buffer mechanism is introduced. When the active buffer fills up, it is marked as frozen and a background flushing process is triggered. Meanwhile, a new buffer is allocated for incoming writes, allowing operations to continue uninterrupted. This approach reduces system blocking times and provides flexibility in adjusting buffer parameters for improved performance.
885 lines
26 KiB
Go
885 lines
26 KiB
Go
// Copyright 2024 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>
|
|
|
|
package pathdb
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/rawdb"
|
|
"github.com/ethereum/go-ethereum/trie/trienode"
|
|
)
|
|
|
|
func newTestLayerTree() *layerTree {
|
|
db := New(rawdb.NewMemoryDatabase(), nil, false)
|
|
l := newDiskLayer(common.Hash{0x1}, 0, db, nil, nil, newBuffer(0, nil, nil, 0), nil)
|
|
t := newLayerTree(l)
|
|
return t
|
|
}
|
|
|
|
func TestLayerCap(t *testing.T) {
|
|
var cases = []struct {
|
|
init func() *layerTree
|
|
head common.Hash
|
|
layers int
|
|
base common.Hash
|
|
snapshot map[common.Hash]struct{}
|
|
}{
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
// Chain:
|
|
// C2->C3->C4 (HEAD)
|
|
head: common.Hash{0x4},
|
|
layers: 2,
|
|
base: common.Hash{0x2},
|
|
snapshot: map[common.Hash]struct{}{
|
|
common.Hash{0x2}: {},
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
// Chain:
|
|
// C3->C4 (HEAD)
|
|
head: common.Hash{0x4},
|
|
layers: 1,
|
|
base: common.Hash{0x3},
|
|
snapshot: map[common.Hash]struct{}{
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
// Chain:
|
|
// C4 (HEAD)
|
|
head: common.Hash{0x4},
|
|
layers: 0,
|
|
base: common.Hash{0x4},
|
|
snapshot: map[common.Hash]struct{}{
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
// ->C2'->C3'->C4'
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2a}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3a}, common.Hash{0x2a}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4a}, common.Hash{0x3a}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x2b}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3b}, common.Hash{0x2b}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4b}, common.Hash{0x3b}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
// Chain:
|
|
// C2->C3->C4 (HEAD)
|
|
head: common.Hash{0x4a},
|
|
layers: 2,
|
|
base: common.Hash{0x2a},
|
|
snapshot: map[common.Hash]struct{}{
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x2a}: {},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
// ->C2'->C3'->C4'
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2a}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3a}, common.Hash{0x2a}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4a}, common.Hash{0x3a}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x2b}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3b}, common.Hash{0x2b}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4b}, common.Hash{0x3b}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
// Chain:
|
|
// C3->C4 (HEAD)
|
|
head: common.Hash{0x4a},
|
|
layers: 1,
|
|
base: common.Hash{0x3a},
|
|
snapshot: map[common.Hash]struct{}{
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x3a}: {},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
// ->C3'->C4'
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3a}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4a}, common.Hash{0x3a}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3b}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4b}, common.Hash{0x3b}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
// Chain:
|
|
// C2->C3->C4 (HEAD)
|
|
// ->C3'->C4'
|
|
head: common.Hash{0x4a},
|
|
layers: 2,
|
|
base: common.Hash{0x2},
|
|
snapshot: map[common.Hash]struct{}{
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4b}: {},
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x2}: {},
|
|
},
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
tr := c.init()
|
|
if err := tr.cap(c.head, c.layers); err != nil {
|
|
t.Fatalf("Failed to cap the layer tree %v", err)
|
|
}
|
|
if tr.bottom().root != c.base {
|
|
t.Fatalf("Unexpected bottom layer tree root, want %v, got %v", c.base, tr.bottom().root)
|
|
}
|
|
if len(c.snapshot) != len(tr.layers) {
|
|
t.Fatalf("Unexpected layer tree size, want %v, got %v", len(c.snapshot), len(tr.layers))
|
|
}
|
|
for h := range tr.layers {
|
|
if _, ok := c.snapshot[h]; !ok {
|
|
t.Fatalf("Unexpected layer %v", h)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestBaseLayer(t *testing.T) {
|
|
tr := newTestLayerTree()
|
|
|
|
var cases = []struct {
|
|
op func()
|
|
base common.Hash
|
|
}{
|
|
// Chain:
|
|
// C1 (HEAD)
|
|
{
|
|
func() {},
|
|
common.Hash{0x1},
|
|
},
|
|
// Chain:
|
|
// C1->C2->C3 (HEAD)
|
|
{
|
|
func() {
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
},
|
|
common.Hash{0x1},
|
|
},
|
|
// Chain:
|
|
// C3 (HEAD)
|
|
{
|
|
func() {
|
|
tr.cap(common.Hash{0x3}, 0)
|
|
},
|
|
common.Hash{0x3},
|
|
},
|
|
// Chain:
|
|
// C4->C5->C6 (HEAD)
|
|
{
|
|
func() {
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x5}, common.Hash{0x4}, 4, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x6}, common.Hash{0x5}, 5, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.cap(common.Hash{0x6}, 2)
|
|
},
|
|
common.Hash{0x4},
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
c.op()
|
|
if tr.base.rootHash() != c.base {
|
|
t.Fatalf("Unexpected base root, want %v, got: %v", c.base, tr.base.rootHash())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDescendant(t *testing.T) {
|
|
var cases = []struct {
|
|
init func() *layerTree
|
|
snapshotA map[common.Hash]map[common.Hash]struct{}
|
|
op func(tr *layerTree)
|
|
snapshotB map[common.Hash]map[common.Hash]struct{}
|
|
}{
|
|
{
|
|
// Chain:
|
|
// C1->C2 (HEAD)
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
snapshotA: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2}: {},
|
|
},
|
|
},
|
|
// Chain:
|
|
// C1->C2->C3 (HEAD)
|
|
op: func(tr *layerTree) {
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
},
|
|
snapshotB: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2}: {},
|
|
common.Hash{0x3}: {},
|
|
},
|
|
common.Hash{0x2}: {
|
|
common.Hash{0x3}: {},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
snapshotA: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2}: {},
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
common.Hash{0x2}: {
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
common.Hash{0x3}: {
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
// Chain:
|
|
// C2->C3->C4 (HEAD)
|
|
op: func(tr *layerTree) {
|
|
tr.cap(common.Hash{0x4}, 2)
|
|
},
|
|
snapshotB: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x2}: {
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
common.Hash{0x3}: {
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
snapshotA: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2}: {},
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
common.Hash{0x2}: {
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
common.Hash{0x3}: {
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
// Chain:
|
|
// C3->C4 (HEAD)
|
|
op: func(tr *layerTree) {
|
|
tr.cap(common.Hash{0x4}, 1)
|
|
},
|
|
snapshotB: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x3}: {
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
snapshotA: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2}: {},
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
common.Hash{0x2}: {
|
|
common.Hash{0x3}: {},
|
|
common.Hash{0x4}: {},
|
|
},
|
|
common.Hash{0x3}: {
|
|
common.Hash{0x4}: {},
|
|
},
|
|
},
|
|
// Chain:
|
|
// C4 (HEAD)
|
|
op: func(tr *layerTree) {
|
|
tr.cap(common.Hash{0x4}, 0)
|
|
},
|
|
snapshotB: map[common.Hash]map[common.Hash]struct{}{},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
// ->C2'->C3'->C4'
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2a}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3a}, common.Hash{0x2a}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4a}, common.Hash{0x3a}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x2b}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3b}, common.Hash{0x2b}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4b}, common.Hash{0x3b}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
snapshotA: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2a}: {},
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x2b}: {},
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
common.Hash{0x2a}: {
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
common.Hash{0x3a}: {
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
common.Hash{0x2b}: {
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
common.Hash{0x3b}: {
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
},
|
|
// Chain:
|
|
// C2->C3->C4 (HEAD)
|
|
op: func(tr *layerTree) {
|
|
tr.cap(common.Hash{0x4a}, 2)
|
|
},
|
|
snapshotB: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x2a}: {
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
common.Hash{0x3a}: {
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
// ->C2'->C3'->C4'
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2a}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3a}, common.Hash{0x2a}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4a}, common.Hash{0x3a}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x2b}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3b}, common.Hash{0x2b}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4b}, common.Hash{0x3b}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
snapshotA: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2a}: {},
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x2b}: {},
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
common.Hash{0x2a}: {
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
common.Hash{0x3a}: {
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
common.Hash{0x2b}: {
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
common.Hash{0x3b}: {
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
},
|
|
// Chain:
|
|
// C3->C4 (HEAD)
|
|
op: func(tr *layerTree) {
|
|
tr.cap(common.Hash{0x4a}, 1)
|
|
},
|
|
snapshotB: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x3a}: {
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
// ->C3'->C4'
|
|
init: func() *layerTree {
|
|
tr := newTestLayerTree()
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3a}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4a}, common.Hash{0x3a}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3b}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4b}, common.Hash{0x3b}, 3, trienode.NewMergedNodeSet(), NewStateSetWithOrigin(nil, nil, nil, nil, false))
|
|
return tr
|
|
},
|
|
snapshotA: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x1}: {
|
|
common.Hash{0x2}: {},
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
common.Hash{0x2}: {
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
common.Hash{0x3a}: {
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
common.Hash{0x3b}: {
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
},
|
|
// Chain:
|
|
// C2->C3->C4 (HEAD)
|
|
// ->C3'->C4'
|
|
op: func(tr *layerTree) {
|
|
tr.cap(common.Hash{0x4a}, 2)
|
|
},
|
|
snapshotB: map[common.Hash]map[common.Hash]struct{}{
|
|
common.Hash{0x2}: {
|
|
common.Hash{0x3a}: {},
|
|
common.Hash{0x4a}: {},
|
|
common.Hash{0x3b}: {},
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
common.Hash{0x3a}: {
|
|
common.Hash{0x4a}: {},
|
|
},
|
|
common.Hash{0x3b}: {
|
|
common.Hash{0x4b}: {},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
check := func(setA, setB map[common.Hash]map[common.Hash]struct{}) bool {
|
|
if len(setA) != len(setB) {
|
|
return false
|
|
}
|
|
for h, subA := range setA {
|
|
subB, ok := setB[h]
|
|
if !ok {
|
|
return false
|
|
}
|
|
if len(subA) != len(subB) {
|
|
return false
|
|
}
|
|
for hh := range subA {
|
|
if _, ok := subB[hh]; !ok {
|
|
return false
|
|
}
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
for _, c := range cases {
|
|
tr := c.init()
|
|
if !check(c.snapshotA, tr.descendants) {
|
|
t.Fatalf("Unexpected descendants")
|
|
}
|
|
c.op(tr)
|
|
if !check(c.snapshotB, tr.descendants) {
|
|
t.Fatalf("Unexpected descendants")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAccountLookup(t *testing.T) {
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
tr := newTestLayerTree() // base = 0x1
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(),
|
|
NewStateSetWithOrigin(randomAccountSet("0xa"), nil, nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(),
|
|
NewStateSetWithOrigin(randomAccountSet("0xb"), nil, nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(),
|
|
NewStateSetWithOrigin(randomAccountSet("0xa", "0xc"), nil, nil, nil, false))
|
|
|
|
var cases = []struct {
|
|
account common.Hash
|
|
state common.Hash
|
|
expect common.Hash
|
|
}{
|
|
{
|
|
// unknown account
|
|
common.HexToHash("0xd"), common.Hash{0x4}, common.Hash{0x1},
|
|
},
|
|
/*
|
|
lookup account from the top
|
|
*/
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x4}, common.Hash{0x4},
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x4}, common.Hash{0x3},
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x4}, common.Hash{0x4},
|
|
},
|
|
/*
|
|
lookup account from the middle
|
|
*/
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x3}, common.Hash{0x2},
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x3}, common.Hash{0x3},
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x3}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x2}, common.Hash{0x2},
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x2}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x2}, common.Hash{0x1}, // not found
|
|
},
|
|
/*
|
|
lookup account from the bottom
|
|
*/
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x1}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x1}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x1}, common.Hash{0x1}, // not found
|
|
},
|
|
}
|
|
for i, c := range cases {
|
|
l, err := tr.lookupAccount(c.account, c.state)
|
|
if err != nil {
|
|
t.Fatalf("%d: %v", i, err)
|
|
}
|
|
if l.rootHash() != c.expect {
|
|
t.Errorf("Unexpected tiphash, %d, want: %x, got: %x", i, c.expect, l.rootHash())
|
|
}
|
|
}
|
|
|
|
// Chain:
|
|
// C3->C4 (HEAD)
|
|
tr.cap(common.Hash{0x4}, 1)
|
|
|
|
cases2 := []struct {
|
|
account common.Hash
|
|
state common.Hash
|
|
expect common.Hash
|
|
expectErr error
|
|
}{
|
|
{
|
|
// unknown account
|
|
common.HexToHash("0xd"), common.Hash{0x4}, common.Hash{0x3}, nil,
|
|
},
|
|
/*
|
|
lookup account from the top
|
|
*/
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x4}, common.Hash{0x4}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x4}, common.Hash{0x3}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x4}, common.Hash{0x4}, nil,
|
|
},
|
|
/*
|
|
lookup account from the bottom
|
|
*/
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x3}, common.Hash{0x3}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x3}, common.Hash{0x3}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x3}, common.Hash{0x3}, nil, // not found
|
|
},
|
|
/*
|
|
stale states
|
|
*/
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x2}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x2}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x2}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0xa"), common.Hash{0x1}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0xb"), common.Hash{0x1}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0xc"), common.Hash{0x1}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
}
|
|
for i, c := range cases2 {
|
|
l, err := tr.lookupAccount(c.account, c.state)
|
|
if c.expectErr != nil {
|
|
if !errors.Is(err, c.expectErr) {
|
|
t.Fatalf("%d: unexpected error, want %v, got %v", i, c.expectErr, err)
|
|
}
|
|
}
|
|
if c.expectErr == nil {
|
|
if err != nil {
|
|
t.Fatalf("%d: %v", i, err)
|
|
}
|
|
if l.rootHash() != c.expect {
|
|
t.Errorf("Unexpected tiphash, %d, want: %x, got: %x", i, c.expect, l.rootHash())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestStorageLookup(t *testing.T) {
|
|
// Chain:
|
|
// C1->C2->C3->C4 (HEAD)
|
|
tr := newTestLayerTree() // base = 0x1
|
|
tr.add(common.Hash{0x2}, common.Hash{0x1}, 1, trienode.NewMergedNodeSet(),
|
|
NewStateSetWithOrigin(randomAccountSet("0xa"), randomStorageSet([]string{"0xa"}, [][]string{{"0x1"}}, nil), nil, nil, false))
|
|
tr.add(common.Hash{0x3}, common.Hash{0x2}, 2, trienode.NewMergedNodeSet(),
|
|
NewStateSetWithOrigin(randomAccountSet("0xa"), randomStorageSet([]string{"0xa"}, [][]string{{"0x2"}}, nil), nil, nil, false))
|
|
tr.add(common.Hash{0x4}, common.Hash{0x3}, 3, trienode.NewMergedNodeSet(),
|
|
NewStateSetWithOrigin(randomAccountSet("0xa"), randomStorageSet([]string{"0xa"}, [][]string{{"0x1", "0x3"}}, nil), nil, nil, false))
|
|
|
|
var cases = []struct {
|
|
storage common.Hash
|
|
state common.Hash
|
|
expect common.Hash
|
|
}{
|
|
{
|
|
// unknown storage slot
|
|
common.HexToHash("0x4"), common.Hash{0x4}, common.Hash{0x1},
|
|
},
|
|
/*
|
|
lookup storage slot from the top
|
|
*/
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x4}, common.Hash{0x4},
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x4}, common.Hash{0x3},
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x4}, common.Hash{0x4},
|
|
},
|
|
/*
|
|
lookup storage slot from the middle
|
|
*/
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x3}, common.Hash{0x2},
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x3}, common.Hash{0x3},
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x3}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x2}, common.Hash{0x2},
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x2}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x2}, common.Hash{0x1}, // not found
|
|
},
|
|
/*
|
|
lookup storage slot from the bottom
|
|
*/
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x1}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x1}, common.Hash{0x1}, // not found
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x1}, common.Hash{0x1}, // not found
|
|
},
|
|
}
|
|
for i, c := range cases {
|
|
l, err := tr.lookupStorage(common.HexToHash("0xa"), c.storage, c.state)
|
|
if err != nil {
|
|
t.Fatalf("%d: %v", i, err)
|
|
}
|
|
if l.rootHash() != c.expect {
|
|
t.Errorf("Unexpected tiphash, %d, want: %x, got: %x", i, c.expect, l.rootHash())
|
|
}
|
|
}
|
|
|
|
// Chain:
|
|
// C3->C4 (HEAD)
|
|
tr.cap(common.Hash{0x4}, 1)
|
|
|
|
cases2 := []struct {
|
|
storage common.Hash
|
|
state common.Hash
|
|
expect common.Hash
|
|
expectErr error
|
|
}{
|
|
{
|
|
// unknown storage slot
|
|
common.HexToHash("0x4"), common.Hash{0x4}, common.Hash{0x3}, nil,
|
|
},
|
|
/*
|
|
lookup account from the top
|
|
*/
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x4}, common.Hash{0x4}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x4}, common.Hash{0x3}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x4}, common.Hash{0x4}, nil,
|
|
},
|
|
/*
|
|
lookup account from the bottom
|
|
*/
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x3}, common.Hash{0x3}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x3}, common.Hash{0x3}, nil,
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x3}, common.Hash{0x3}, nil, // not found
|
|
},
|
|
/*
|
|
stale states
|
|
*/
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x2}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x2}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x2}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0x1"), common.Hash{0x1}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0x2"), common.Hash{0x1}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
{
|
|
common.HexToHash("0x3"), common.Hash{0x1}, common.Hash{}, errSnapshotStale,
|
|
},
|
|
}
|
|
for i, c := range cases2 {
|
|
l, err := tr.lookupStorage(common.HexToHash("0xa"), c.storage, c.state)
|
|
if c.expectErr != nil {
|
|
if !errors.Is(err, c.expectErr) {
|
|
t.Fatalf("%d: unexpected error, want %v, got %v", i, c.expectErr, err)
|
|
}
|
|
}
|
|
if c.expectErr == nil {
|
|
if err != nil {
|
|
t.Fatalf("%d: %v", i, err)
|
|
}
|
|
if l.rootHash() != c.expect {
|
|
t.Errorf("Unexpected tiphash, %d, want: %x, got: %x", i, c.expect, l.rootHash())
|
|
}
|
|
}
|
|
}
|
|
}
|