erigon-pulse/cmd/hack/hack.go
Alex Sharov 305f8aff02
Integration tests2 (#714)
```
go run ./cmd/integration reset --chaindata=...
go run ./cmd/integration state_stages -h
go run ./cmd/integration state_stages  --chaindata=... --verbosity=3 --block=2_000_000 --unwind=10 --unwind_every=1_000 
```
Also, it inherits flags from geth:
```
--pprof.cpuprofile=./cpu.out   // to file
--pprof --pprof.port=6060     // launch pprof server
--metrics                  //  sends to prometheus 
```
2020-07-07 11:00:25 +07:00

2521 lines
71 KiB
Go

package main
import (
"bufio"
"bytes"
"context"
"encoding/binary"
"flag"
"fmt"
"io"
"io/ioutil"
"math/big"
"os"
"os/signal"
"runtime"
"runtime/pprof"
"sort"
"strconv"
"strings"
"syscall"
"time"
"github.com/AskAlexSharov/lmdb-go/lmdb"
"github.com/ledgerwatch/bolt"
"github.com/ledgerwatch/turbo-geth/common"
"github.com/ledgerwatch/turbo-geth/common/changeset"
"github.com/ledgerwatch/turbo-geth/common/dbutils"
"github.com/ledgerwatch/turbo-geth/common/etl"
"github.com/ledgerwatch/turbo-geth/consensus/ethash"
"github.com/ledgerwatch/turbo-geth/core"
"github.com/ledgerwatch/turbo-geth/core/rawdb"
"github.com/ledgerwatch/turbo-geth/core/state"
"github.com/ledgerwatch/turbo-geth/core/types"
"github.com/ledgerwatch/turbo-geth/core/types/accounts"
"github.com/ledgerwatch/turbo-geth/core/vm"
"github.com/ledgerwatch/turbo-geth/crypto"
"github.com/ledgerwatch/turbo-geth/eth/stagedsync"
"github.com/ledgerwatch/turbo-geth/eth/stagedsync/stages"
"github.com/ledgerwatch/turbo-geth/ethdb"
"github.com/ledgerwatch/turbo-geth/log"
"github.com/ledgerwatch/turbo-geth/node"
"github.com/ledgerwatch/turbo-geth/params"
"github.com/ledgerwatch/turbo-geth/rlp"
"github.com/ledgerwatch/turbo-geth/trie"
"github.com/mattn/go-colorable"
"github.com/mattn/go-isatty"
"github.com/wcharczuk/go-chart"
"github.com/wcharczuk/go-chart/util"
)
var emptyCodeHash = crypto.Keccak256(nil)
var verbosity = flag.Uint("verbosity", 3, "Logging verbosity: 0=silent, 1=error, 2=warn, 3=info, 4=debug, 5=detail (default 3)")
var action = flag.String("action", "", "action to execute")
var cpuprofile = flag.String("cpuprofile", "", "write cpu profile `file`")
var rewind = flag.Int("rewind", 1, "rewind to given number of blocks")
var block = flag.Int("block", 1, "specifies a block number for operation")
var account = flag.String("account", "0x", "specifies account to investigate")
var name = flag.String("name", "", "name to add to the file names")
var chaindata = flag.String("chaindata", "chaindata", "path to the chaindata database file")
var bucket = flag.String("bucket", "", "bucket in the database")
var hash = flag.String("hash", "0x00", "image for preimage or state root for testBlockHashes action")
var preImage = flag.String("preimage", "0x00", "preimage")
func check(e error) {
if e != nil {
panic(e)
}
}
func parseFloat64(str string) float64 {
v, _ := strconv.ParseFloat(str, 64)
return v
}
func readData(filename string) (blocks []float64, hours []float64, dbsize []float64, trienodes []float64, heap []float64) {
err := util.File.ReadByLines(filename, func(line string) error {
parts := strings.Split(line, ",")
blocks = append(blocks, parseFloat64(strings.Trim(parts[0], " ")))
hours = append(hours, parseFloat64(strings.Trim(parts[1], " ")))
dbsize = append(dbsize, parseFloat64(strings.Trim(parts[2], " ")))
trienodes = append(trienodes, parseFloat64(strings.Trim(parts[3], " ")))
heap = append(heap, parseFloat64(strings.Trim(parts[4], " ")))
return nil
})
if err != nil {
fmt.Println(err.Error())
}
return
}
func notables() []chart.GridLine {
return []chart.GridLine{
{Value: 1.0},
{Value: 2.0},
{Value: 3.0},
{Value: 4.0},
{Value: 5.0},
{Value: 6.0},
}
}
func days() []chart.GridLine {
return []chart.GridLine{
{Value: 24.0},
{Value: 48.0},
{Value: 72.0},
{Value: 96.0},
{Value: 120.0},
{Value: 144.0},
{Value: 168.0},
{Value: 192.0},
{Value: 216.0},
{Value: 240.0},
{Value: 264.0},
{Value: 288.0},
}
}
func mychart() {
blocks, hours, dbsize, trienodes, heap := readData("bolt.csv")
blocks0, hours0, dbsize0, _, _ := readData("badger.csv")
mainSeries := &chart.ContinuousSeries{
Name: "Cumulative sync time (bolt)",
Style: chart.Style{
Show: true,
StrokeColor: chart.ColorBlue,
FillColor: chart.ColorBlue.WithAlpha(100),
},
XValues: blocks,
YValues: hours,
}
badgerSeries := &chart.ContinuousSeries{
Name: "Cumulative sync time (badger)",
Style: chart.Style{
Show: true,
StrokeColor: chart.ColorRed,
FillColor: chart.ColorRed.WithAlpha(100),
},
XValues: blocks0,
YValues: hours0,
}
dbsizeSeries := &chart.ContinuousSeries{
Name: "Database size (bolt)",
Style: chart.Style{
Show: true,
StrokeColor: chart.ColorBlack,
},
YAxis: chart.YAxisSecondary,
XValues: blocks,
YValues: dbsize,
}
dbsizeSeries0 := &chart.ContinuousSeries{
Name: "Database size (badger)",
Style: chart.Style{
Show: true,
StrokeColor: chart.ColorOrange,
},
YAxis: chart.YAxisSecondary,
XValues: blocks,
YValues: dbsize0,
}
graph1 := chart.Chart{
Width: 1280,
Height: 720,
Background: chart.Style{
Padding: chart.Box{
Top: 50,
},
},
YAxis: chart.YAxis{
Name: "Elapsed time",
NameStyle: chart.StyleShow(),
Style: chart.StyleShow(),
TickStyle: chart.Style{
TextRotationDegrees: 45.0,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%d h", int(v.(float64)))
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.ColorBlue,
StrokeWidth: 1.0,
},
GridLines: days(),
},
YAxisSecondary: chart.YAxis{
NameStyle: chart.StyleShow(),
Style: chart.StyleShow(),
TickStyle: chart.Style{
TextRotationDegrees: 45.0,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%d G", int(v.(float64)))
},
},
XAxis: chart.XAxis{
Name: "Blocks, million",
Style: chart.Style{
Show: true,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%.3fm", v.(float64))
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.ColorAlternateGray,
StrokeWidth: 1.0,
},
GridLines: notables(),
},
Series: []chart.Series{
mainSeries,
badgerSeries,
dbsizeSeries,
dbsizeSeries0,
},
}
graph1.Elements = []chart.Renderable{chart.LegendThin(&graph1)}
buffer := bytes.NewBuffer([]byte{})
err := graph1.Render(chart.PNG, buffer)
check(err)
err = ioutil.WriteFile("chart1.png", buffer.Bytes(), 0644)
check(err)
heapSeries := &chart.ContinuousSeries{
Name: "Allocated heap",
Style: chart.Style{
Show: true,
StrokeColor: chart.ColorYellow,
FillColor: chart.ColorYellow.WithAlpha(100),
},
XValues: blocks,
YValues: heap,
}
trienodesSeries := &chart.ContinuousSeries{
Name: "Trie nodes",
Style: chart.Style{
Show: true,
StrokeColor: chart.ColorGreen,
},
YAxis: chart.YAxisSecondary,
XValues: blocks,
YValues: trienodes,
}
graph2 := chart.Chart{
Width: 1280,
Height: 720,
Background: chart.Style{
Padding: chart.Box{
Top: 50,
},
},
YAxis: chart.YAxis{
Name: "Allocated heap",
NameStyle: chart.StyleShow(),
Style: chart.StyleShow(),
TickStyle: chart.Style{
TextRotationDegrees: 45.0,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%.1f G", v.(float64))
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.ColorYellow,
StrokeWidth: 1.0,
},
GridLines: days(),
},
YAxisSecondary: chart.YAxis{
NameStyle: chart.StyleShow(),
Style: chart.StyleShow(),
TickStyle: chart.Style{
TextRotationDegrees: 45.0,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%.1f m", v.(float64))
},
},
XAxis: chart.XAxis{
Name: "Blocks, million",
Style: chart.Style{
Show: true,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%.3fm", v.(float64))
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.ColorAlternateGray,
StrokeWidth: 1.0,
},
GridLines: notables(),
},
Series: []chart.Series{
heapSeries,
trienodesSeries,
},
}
graph2.Elements = []chart.Renderable{chart.LegendThin(&graph2)}
buffer.Reset()
err = graph2.Render(chart.PNG, buffer)
check(err)
err = ioutil.WriteFile("chart2.png", buffer.Bytes(), 0644)
check(err)
}
func accountSavings(db *bolt.DB) (int, int) {
emptyRoots := 0
emptyCodes := 0
db.View(func(tx *bolt.Tx) error {
b := tx.Bucket(dbutils.CurrentStateBucket)
c := b.Cursor()
for k, v := c.First(); k != nil; k, v = c.Next() {
if len(k) != 32 {
continue
}
if bytes.Contains(v, trie.EmptyRoot.Bytes()) {
emptyRoots++
}
if bytes.Contains(v, emptyCodeHash) {
emptyCodes++
}
}
return nil
})
return emptyRoots, emptyCodes
}
func printBuckets(db *bolt.DB) {
for _, bucket := range dbutils.Buckets {
fmt.Printf("%s\n", bucket)
}
}
func bucketStats(chaindata string) {
t := ethdb.Lmdb
bucketList := dbutils.Buckets
switch t {
case ethdb.Bolt:
db, err := bolt.Open(chaindata, 0600, &bolt.Options{ReadOnly: true})
check(err)
fmt.Printf(",BranchPageN,BranchOverflowN,LeafPageN,LeafOverflowN,KeyN,Depth,BranchAlloc,BranchInuse,LeafAlloc,LeafInuse,BucketN,InlineBucketN,InlineBucketInuse\n")
_ = db.View(func(tx *bolt.Tx) error {
for _, bucket := range bucketList {
b := tx.Bucket(bucket)
bs := b.Stats()
fmt.Printf("%s,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d\n", string(bucket),
bs.BranchPageN, bs.BranchOverflowN, bs.LeafPageN, bs.LeafOverflowN, bs.KeyN, bs.Depth, bs.BranchAlloc, bs.BranchInuse,
bs.LeafAlloc, bs.LeafInuse, bs.BucketN, bs.InlineBucketN, bs.InlineBucketInuse)
}
return nil
})
case ethdb.Lmdb:
env, err := lmdb.NewEnv()
check(err)
err = env.SetMaxDBs(100)
check(err)
err = env.Open(chaindata, lmdb.Readonly, 0664)
check(err)
fmt.Printf(",BranchPageN,LeafPageN,OverflowN,Entries\n")
_ = env.View(func(tx *lmdb.Txn) error {
for _, bucket := range bucketList {
dbi, bucketErr := tx.OpenDBI(string(bucket), 0)
if bucketErr != nil {
fmt.Printf("opening bucket %s: %v\n", bucket, bucketErr)
continue
}
bs, statErr := tx.Stat(dbi)
check(statErr)
fmt.Printf("%s,%d,%d,%d,%d\n", string(bucket),
bs.BranchPages, bs.LeafPages, bs.OverflowPages, bs.Entries)
}
return nil
})
}
}
func readTrieLog() ([]float64, map[int][]float64, []float64) {
data, err := ioutil.ReadFile("dust/hack.log")
check(err)
thresholds := []float64{}
counts := map[int][]float64{}
for i := 2; i <= 16; i++ {
counts[i] = []float64{}
}
shorts := []float64{}
lines := bytes.Split(data, []byte("\n"))
for _, line := range lines {
if bytes.HasPrefix(line, []byte("Threshold:")) {
tokens := bytes.Split(line, []byte(" "))
if len(tokens) == 23 {
wei := parseFloat64(string(tokens[1]))
thresholds = append(thresholds, wei)
for i := 2; i <= 16; i++ {
pair := bytes.Split(tokens[i+3], []byte(":"))
counts[i] = append(counts[i], parseFloat64(string(pair[1])))
}
pair := bytes.Split(tokens[21], []byte(":"))
shorts = append(shorts, parseFloat64(string(pair[1])))
}
}
}
return thresholds, counts, shorts
}
func ts() []chart.GridLine {
return []chart.GridLine{
{Value: 420.0},
}
}
func trieChart() {
thresholds, counts, shorts := readTrieLog()
fmt.Printf("%d %d %d\n", len(thresholds), len(counts), len(shorts))
shortsSeries := &chart.ContinuousSeries{
Name: "Short nodes",
Style: chart.Style{
Show: true,
StrokeColor: chart.ColorBlue,
FillColor: chart.ColorBlue.WithAlpha(100),
},
XValues: thresholds,
YValues: shorts,
}
countSeries := make(map[int]*chart.ContinuousSeries)
for i := 2; i <= 16; i++ {
countSeries[i] = &chart.ContinuousSeries{
Name: fmt.Sprintf("%d-nodes", i),
Style: chart.Style{
Show: true,
StrokeColor: chart.GetAlternateColor(i),
},
XValues: thresholds,
YValues: counts[i],
}
}
xaxis := &chart.XAxis{
Name: "Dust theshold",
Style: chart.Style{
Show: true,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%d wei", int(v.(float64)))
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.DefaultStrokeColor,
StrokeWidth: 1.0,
},
Range: &chart.ContinuousRange{
Min: thresholds[0],
Max: thresholds[len(thresholds)-1],
},
Ticks: []chart.Tick{
{Value: 0.0, Label: "0"},
{Value: 1.0, Label: "wei"},
{Value: 10.0, Label: "10"},
{Value: 100.0, Label: "100"},
{Value: 1e3, Label: "1e3"},
{Value: 1e4, Label: "1e4"},
{Value: 1e5, Label: "1e5"},
{Value: 1e6, Label: "1e6"},
{Value: 1e7, Label: "1e7"},
{Value: 1e8, Label: "1e8"},
{Value: 1e9, Label: "1e9"},
{Value: 1e10, Label: "1e10"},
//{1e15, "finney"},
//{1e18, "ether"},
},
}
graph3 := chart.Chart{
Width: 1280,
Height: 720,
Background: chart.Style{
Padding: chart.Box{
Top: 50,
},
},
XAxis: *xaxis,
YAxis: chart.YAxis{
Name: "Node count",
NameStyle: chart.StyleShow(),
Style: chart.StyleShow(),
TickStyle: chart.Style{
TextRotationDegrees: 45.0,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%dm", int(v.(float64)/1e6))
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.DefaultStrokeColor,
StrokeWidth: 1.0,
},
},
Series: []chart.Series{
shortsSeries,
},
}
graph3.Elements = []chart.Renderable{chart.LegendThin(&graph3)}
buffer := bytes.NewBuffer([]byte{})
err := graph3.Render(chart.PNG, buffer)
check(err)
err = ioutil.WriteFile("chart3.png", buffer.Bytes(), 0644)
check(err)
graph4 := chart.Chart{
Width: 1280,
Height: 720,
Background: chart.Style{
Padding: chart.Box{
Top: 50,
},
},
XAxis: *xaxis,
YAxis: chart.YAxis{
Name: "Node count",
NameStyle: chart.StyleShow(),
Style: chart.StyleShow(),
TickStyle: chart.Style{
TextRotationDegrees: 45.0,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%.2fm", v.(float64)/1e6)
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.DefaultStrokeColor,
StrokeWidth: 1.0,
},
},
Series: []chart.Series{
countSeries[2],
countSeries[3],
},
}
graph4.Elements = []chart.Renderable{chart.LegendThin(&graph4)}
buffer = bytes.NewBuffer([]byte{})
err = graph4.Render(chart.PNG, buffer)
check(err)
err = ioutil.WriteFile("chart4.png", buffer.Bytes(), 0644)
check(err)
graph5 := chart.Chart{
Width: 1280,
Height: 720,
Background: chart.Style{
Padding: chart.Box{
Top: 50,
},
},
XAxis: *xaxis,
YAxis: chart.YAxis{
Name: "Node count",
NameStyle: chart.StyleShow(),
Style: chart.StyleShow(),
TickStyle: chart.Style{
TextRotationDegrees: 45.0,
},
ValueFormatter: func(v interface{}) string {
return fmt.Sprintf("%.2fk", v.(float64)/1e3)
},
GridMajorStyle: chart.Style{
Show: true,
StrokeColor: chart.DefaultStrokeColor,
StrokeWidth: 1.0,
},
},
Series: []chart.Series{
countSeries[4],
countSeries[5],
countSeries[6],
countSeries[7],
countSeries[8],
countSeries[9],
countSeries[10],
countSeries[11],
countSeries[12],
countSeries[13],
countSeries[14],
countSeries[15],
countSeries[16],
},
}
graph5.Elements = []chart.Renderable{chart.LegendThin(&graph5)}
buffer = bytes.NewBuffer([]byte{})
err = graph5.Render(chart.PNG, buffer)
check(err)
err = ioutil.WriteFile("chart5.png", buffer.Bytes(), 0644)
check(err)
}
func execToBlock(chaindata string, block uint64, fromScratch bool) {
state.MaxTrieCacheSize = 100 * 1024
blockDb := ethdb.MustOpen(chaindata)
defer blockDb.Close()
bcb, err := core.NewBlockChain(blockDb, nil, params.MainnetChainConfig, ethash.NewFaker(), vm.Config{}, nil, nil, nil)
check(err)
if fromScratch {
os.Remove("statedb")
}
stateDB := ethdb.MustOpen("statedb")
defer stateDB.Close()
//_, _, _, err = core.SetupGenesisBlock(stateDB, core.DefaultGenesisBlock())
_, _, _, err = core.SetupGenesisBlock(stateDB, nil, false /* history */)
check(err)
bc, err := core.NewBlockChain(stateDB, nil, params.MainnetChainConfig, ethash.NewFaker(), vm.Config{}, nil, nil, nil)
check(err)
tds, err := bc.GetTrieDbState()
check(err)
importedBn := tds.GetBlockNr()
if importedBn == 0 {
importedBn = 1
}
//bc.SetNoHistory(true)
blocks := types.Blocks{}
var lastBlock *types.Block
now := time.Now()
Loop:
for i := importedBn; i <= block; i++ {
lastBlock = bcb.GetBlockByNumber(i)
blocks = append(blocks, lastBlock)
if len(blocks) >= 1000 || i == block {
_, err = bc.InsertChain(context.Background(), blocks)
if err != nil {
log.Error("Could not insert blocks (group)", "number", len(blocks), "error", err)
// Try to insert blocks one by one to keep the latest state
for j := 0; j < len(blocks); j++ {
if _, err1 := bc.InsertChain(context.Background(), blocks[j:j+1]); err1 != nil {
log.Error("Could not insert block", "error", err1)
break Loop
}
}
break
}
blocks = types.Blocks{}
}
if i%10000 == 0 {
fmt.Printf("Inserted %dK, %s \n", i/1000, time.Since(now))
}
}
root := tds.LastRoot()
fmt.Printf("Root hash: %x\n", root)
fmt.Printf("Last block root hash: %x\n", lastBlock.Root())
filename := fmt.Sprintf("right_%d.txt", lastBlock.NumberU64())
fmt.Printf("Generating deep snapshot of the right tries... %s\n", filename)
f, err := os.Create(filename)
if err == nil {
defer f.Close()
tds.PrintTrie(f)
}
}
func extractTrie(block int) {
stateDb := ethdb.MustOpen("statedb")
defer stateDb.Close()
bc, err := core.NewBlockChain(stateDb, nil, params.RopstenChainConfig, ethash.NewFaker(), vm.Config{}, nil, nil, nil)
check(err)
baseBlock := bc.GetBlockByNumber(uint64(block))
tds := state.NewTrieDbState(baseBlock.Root(), stateDb, baseBlock.NumberU64())
rebuiltRoot := tds.LastRoot()
fmt.Printf("Rebuit root hash: %x\n", rebuiltRoot)
filename := fmt.Sprintf("right_%d.txt", baseBlock.NumberU64())
fmt.Printf("Generating deep snapshot of the right tries... %s\n", filename)
f, err := os.Create(filename)
if err == nil {
defer f.Close()
tds.PrintTrie(f)
}
}
func testRewind(chaindata string, block, rewind int) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
bc, err := core.NewBlockChain(ethDb, nil, params.MainnetChainConfig, ethash.NewFaker(), vm.Config{}, nil, nil, nil)
check(err)
currentBlock := bc.CurrentBlock()
currentBlockNr := currentBlock.NumberU64()
if block == 1 {
block = int(currentBlockNr)
}
baseBlock := bc.GetBlockByNumber(uint64(block))
baseBlockNr := baseBlock.NumberU64()
fmt.Printf("Base block number: %d\n", baseBlockNr)
fmt.Printf("Base block root hash: %x\n", baseBlock.Root())
db := ethDb.NewBatch()
defer db.Rollback()
tds := state.NewTrieDbState(baseBlock.Root(), db, baseBlockNr)
tds.SetHistorical(baseBlockNr != currentBlockNr)
rebuiltRoot := tds.LastRoot()
fmt.Printf("Rebuit root hash: %x\n", rebuiltRoot)
startTime := time.Now()
rewindLen := uint64(rewind)
err = tds.UnwindTo(baseBlockNr - rewindLen)
fmt.Printf("Unwind done in %v\n", time.Since(startTime))
check(err)
rewoundBlock1 := bc.GetBlockByNumber(baseBlockNr - rewindLen + 1)
fmt.Printf("Rewound+1 block number: %d\n", rewoundBlock1.NumberU64())
fmt.Printf("Rewound+1 block hash: %x\n", rewoundBlock1.Hash())
fmt.Printf("Rewound+1 block root hash: %x\n", rewoundBlock1.Root())
fmt.Printf("Rewound+1 block parent hash: %x\n", rewoundBlock1.ParentHash())
rewoundBlock := bc.GetBlockByNumber(baseBlockNr - rewindLen)
fmt.Printf("Rewound block number: %d\n", rewoundBlock.NumberU64())
fmt.Printf("Rewound block hash: %x\n", rewoundBlock.Hash())
fmt.Printf("Rewound block root hash: %x\n", rewoundBlock.Root())
fmt.Printf("Rewound block parent hash: %x\n", rewoundBlock.ParentHash())
rewoundRoot := tds.LastRoot()
fmt.Printf("Calculated rewound root hash: %x\n", rewoundRoot)
/*
filename := fmt.Sprintf("root_%d.txt", rewoundBlock.NumberU64())
fmt.Printf("Generating deep snapshot of the wront tries... %s\n", filename)
f, err := os.Create(filename)
if err == nil {
defer f.Close()
tds.PrintTrie(f)
}
{
tds, err = state.NewTrieDbState(rewoundBlock.Root(), db, rewoundBlock.NumberU64())
tds.SetHistorical(true)
check(err)
rebuiltRoot, err := tds.TrieRoot()
fmt.Printf("Rebuilt root: %x\n", rebuiltRoot)
check(err)
}
*/
}
func testStartup() {
startTime := time.Now()
//ethDb := ethdb.MustOpen(node.DefaultDataDir() + "/geth/chaindata")
ethDb := ethdb.MustOpen("/home/akhounov/.ethereum/geth/chaindata")
defer ethDb.Close()
bc, err := core.NewBlockChain(ethDb, nil, params.MainnetChainConfig, ethash.NewFaker(), vm.Config{}, nil, nil, nil)
check(err)
currentBlock := bc.CurrentBlock()
currentBlockNr := currentBlock.NumberU64()
fmt.Printf("Current block number: %d\n", currentBlockNr)
fmt.Printf("Current block root hash: %x\n", currentBlock.Root())
l := trie.NewSubTrieLoader(currentBlockNr)
rl := trie.NewRetainList(0)
subTries, err1 := l.LoadSubTries(ethDb, currentBlockNr, rl, nil /* HashCollector */, [][]byte{nil}, []int{0}, false)
if err1 != nil {
fmt.Printf("%v\n", err1)
}
if subTries.Hashes[0] != currentBlock.Root() {
fmt.Printf("Hash mismatch, got %x, expected %x\n", subTries.Hashes[0], currentBlock.Root())
}
fmt.Printf("Took %v\n", time.Since(startTime))
}
func dbSlice(chaindata string, bucket []byte, prefix []byte) {
db := ethdb.MustOpen(chaindata)
defer db.Close()
if err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
st := tx.Bucket(bucket)
c := st.Cursor()
for k, v, err := c.Seek(prefix); k != nil && bytes.HasPrefix(k, prefix); k, v, err = c.Next() {
if err != nil {
return err
}
fmt.Printf("db.Put([]byte(\"%s\"), common.FromHex(\"%x\"), common.FromHex(\"%x\"))\n", bucket, k, v)
}
return nil
}); err != nil {
panic(err)
}
}
func testResolve(chaindata string) {
startTime := time.Now()
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
//bc, err := core.NewBlockChain(ethDb, nil, params.MainnetChainConfig, ethash.NewFaker(), vm.Config{}, nil, nil)
//check(err)
/*
currentBlock := bc.CurrentBlock()
currentBlockNr := currentBlock.NumberU64()
fmt.Printf("Current block number: %d\n", currentBlockNr)
fmt.Printf("Current block root hash: %x\n", currentBlock.Root())
prevBlock := bc.GetBlockByNumber(currentBlockNr - 2)
fmt.Printf("Prev block root hash: %x\n", prevBlock.Root())
*/
currentBlockNr := uint64(286798)
//var contract []byte
//contract = common.FromHex("8416044c93d8fdf2d06a5bddbea65234695a3d4d278d5c824776c8b31702505dfffffffffffffffe")
resolveHash := common.HexToHash("321131c74d582ebe29075d573023accd809234e4dbdee29e814bacedd3467279")
l := trie.NewSubTrieLoader(currentBlockNr)
var key []byte
key = common.FromHex("0a080d05070c0604040302030508050100020105040e05080c0a0f030d0d050f08070a050b0c08090b02040e0e0200030f0c0b0f0704060a0d0703050009010f")
rl := trie.NewRetainList(0)
rl.AddHex(key[:3])
subTries, err1 := l.LoadSubTries(ethDb, currentBlockNr, rl, nil /* HashCollector */, [][]byte{{0xa8, 0xd0}}, []int{12}, true)
if err1 != nil {
fmt.Printf("Resolve error: %v\n", err1)
}
if subTries.Hashes[0] != resolveHash {
fmt.Printf("Has mismatch, got %x, expected %x\n", subTries.Hashes[0], resolveHash)
}
/*
var filename string
if err == nil {
filename = fmt.Sprintf("right_%d.txt", currentBlockNr)
} else {
filename = fmt.Sprintf("root_%d.txt", currentBlockNr)
}
fmt.Printf("Generating deep snapshot of the tries... %s\n", filename)
f, err := os.Create(filename)
if err == nil {
defer f.Close()
t.Print(f)
}
if err != nil {
fmt.Printf("%v\n", err)
}
*/
fmt.Printf("Took %v\n", time.Since(startTime))
}
func hashFile() {
f, err := os.Open("/Users/alexeyakhunov/mygit/go-ethereum/geth.log")
check(err)
defer f.Close()
w, err := os.Create("/Users/alexeyakhunov/mygit/go-ethereum/geth_read.log")
check(err)
defer w.Close()
scanner := bufio.NewScanner(f)
count := 0
for scanner.Scan() {
line := scanner.Text()
if strings.HasPrefix(line, "ResolveWithDb") || strings.HasPrefix(line, "Error") ||
strings.HasPrefix(line, "0000000000000000000000000000000000000000000000000000000000000000") ||
strings.HasPrefix(line, "ERROR") || strings.HasPrefix(line, "tc{") {
fmt.Printf("%d %s\n", count, line)
count++
} else if count == 66 {
w.WriteString(line)
w.WriteString("\n")
}
}
fmt.Printf("%d lines scanned\n", count)
}
func rlpIndices() {
keybuf := new(bytes.Buffer)
for i := 0; i < 512; i++ {
keybuf.Reset()
rlp.Encode(keybuf, uint(i))
fmt.Printf("Encoding of %d is %x\n", i, keybuf.Bytes())
}
}
func printFullNodeRLPs() {
trie.FullNode1()
trie.FullNode2()
trie.FullNode3()
trie.FullNode4()
trie.ShortNode1()
trie.ShortNode2()
trie.Hash1()
trie.Hash2()
trie.Hash3()
trie.Hash4()
trie.Hash5()
trie.Hash6()
trie.Hash7()
}
func testDifficulty() {
genesisBlock, _, _, err := core.DefaultGenesisBlock().ToBlock(nil, false /* history */)
check(err)
d1 := ethash.CalcDifficulty(params.MainnetChainConfig, 100000, genesisBlock.Header())
fmt.Printf("Block 1 difficulty: %d\n", d1)
}
// Searches 1000 blocks from the given one to try to find the one with the given state root hash
func testBlockHashes(chaindata string, block int, stateRoot common.Hash) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
blocksToSearch := 10000000
for i := uint64(block); i < uint64(block+blocksToSearch); i++ {
hash := rawdb.ReadCanonicalHash(ethDb, i)
header := rawdb.ReadHeader(ethDb, hash, i)
if header.Root == stateRoot || stateRoot == (common.Hash{}) {
fmt.Printf("\n===============\nCanonical hash for %d: %x\n", i, hash)
fmt.Printf("Header.Root: %x\n", header.Root)
fmt.Printf("Header.TxHash: %x\n", header.TxHash)
fmt.Printf("Header.UncleHash: %x\n", header.UncleHash)
}
}
}
func printCurrentBlockNumber(chaindata string) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
hash := rawdb.ReadHeadBlockHash(ethDb)
number := rawdb.ReadHeaderNumber(ethDb, hash)
fmt.Printf("Block number: %d\n", *number)
}
func printTxHashes() {
ethDb := ethdb.MustOpen(node.DefaultDataDir() + "/geth/chaindata")
defer ethDb.Close()
for b := uint64(0); b < uint64(100000); b++ {
hash := rawdb.ReadCanonicalHash(ethDb, b)
block := rawdb.ReadBlock(ethDb, hash, b)
if block == nil {
break
}
for _, tx := range block.Transactions() {
fmt.Printf("%x\n", tx.Hash())
}
}
}
func relayoutKeys() {
//db := ethdb.MustOpen("/home/akhounov/.ethereum/geth/chaindata")
db := ethdb.MustOpen(node.DefaultDataDir() + "/geth/chaindata")
defer db.Close()
var accountChangeSetCount, storageChangeSetCount int
err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
b := tx.Bucket(dbutils.AccountChangeSetBucket)
if b == nil {
return nil
}
c := b.Cursor()
for k, _, _ := c.First(); k != nil; k, _, _ = c.Next() {
accountChangeSetCount++
}
return nil
})
check(err)
err = db.KV().View(context.Background(), func(tx ethdb.Tx) error {
b := tx.Bucket(dbutils.StorageChangeSetBucket)
if b == nil {
return nil
}
c := b.Cursor()
for k, _, _ := c.First(); k != nil; k, _, _ = c.Next() {
storageChangeSetCount++
}
return nil
})
check(err)
fmt.Printf("Account changeset: %d\n", accountChangeSetCount)
fmt.Printf("Storage changeset: %d\n", storageChangeSetCount)
fmt.Printf("Total: %d\n", accountChangeSetCount+storageChangeSetCount)
}
func readTrie(filename string) *trie.Trie {
f, err := os.Open(filename)
check(err)
defer f.Close()
t, err := trie.Load(f)
check(err)
return t
}
func invTree(wrong, right, diff string, name string) {
fmt.Printf("Reading trie...\n")
t1 := readTrie(fmt.Sprintf("%s_%s.txt", wrong, name))
fmt.Printf("Root hash: %x\n", t1.Hash())
fmt.Printf("Reading trie 2...\n")
t2 := readTrie(fmt.Sprintf("%s_%s.txt", right, name))
fmt.Printf("Root hash: %x\n", t2.Hash())
c, err := os.Create(fmt.Sprintf("%s_%s.txt", diff, name))
check(err)
defer c.Close()
t1.PrintDiff(t2, c)
}
func preimage(chaindata string, image common.Hash) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
p, err := ethDb.Get(dbutils.PreimagePrefix, image[:])
check(err)
fmt.Printf("%x\n", p)
}
func addPreimage(chaindata string, image common.Hash, preimage []byte) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
err := ethDb.Put(dbutils.PreimagePrefix, image[:], preimage)
check(err)
}
func printBranches(block uint64) {
//ethDb := ethdb.MustOpen("/home/akhounov/.ethereum/geth/chaindata")
ethDb := ethdb.MustOpen(node.DefaultDataDir() + "/testnet/geth/chaindata")
defer ethDb.Close()
fmt.Printf("All headers at the same height %d\n", block)
{
var hashes []common.Hash
numberEnc := make([]byte, 8)
binary.BigEndian.PutUint64(numberEnc, block)
if err := ethDb.Walk([]byte("h"), numberEnc, 8*8, func(k, v []byte) (bool, error) {
if len(k) == 8+32 {
hashes = append(hashes, common.BytesToHash(k[8:]))
}
return true, nil
}); err != nil {
panic(err)
}
for _, hash := range hashes {
h := rawdb.ReadHeader(ethDb, hash, block)
fmt.Printf("block hash: %x, root hash: %x\n", h.Hash(), h.Root)
}
}
}
func readPlainAccount(chaindata string, address common.Address) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
var acc accounts.Account
enc, err := ethDb.Get(dbutils.PlainStateBucket, address[:])
if err != nil {
panic(err)
} else if enc == nil {
panic("acc not found")
}
if err = acc.DecodeForStorage(enc); err != nil {
panic(err)
}
fmt.Printf("%x\n%x\n%x\n%d\n", address, acc.Root, acc.CodeHash, acc.Incarnation)
}
func readAccount(chaindata string, account common.Address, block uint64, rewind uint64) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
secKey := crypto.Keccak256(account[:])
var a accounts.Account
ok, err := rawdb.ReadAccount(ethDb, common.BytesToHash(secKey), &a)
if err != nil {
panic(err)
} else if !ok {
panic("acc not found")
}
fmt.Printf("%x\n%x\n%x\n%d\n", secKey, a.Root, a.CodeHash, a.Incarnation)
//var addrHash common.Hash
//copy(addrHash[:], secKey)
//codeHash, err := ethDb.Get(dbutils.ContractCodeBucket, dbutils.GenerateStoragePrefix(addrHash, a.Incarnation))
//check(err)
//fmt.Printf("codeHash: %x\n", codeHash)
timestamp := block
for i := uint64(0); i < rewind; i++ {
var printed bool
encodedTS := dbutils.EncodeTimestamp(timestamp)
var v []byte
v, _ = ethDb.Get(dbutils.StorageChangeSetBucket, encodedTS)
if v != nil {
err = changeset.StorageChangeSetBytes(v).Walk(func(key, value []byte) error {
if bytes.HasPrefix(key, secKey) {
incarnation := ^binary.BigEndian.Uint64(key[common.HashLength : common.HashLength+common.IncarnationLength])
if !printed {
fmt.Printf("Changes for block %d\n", timestamp)
printed = true
}
fmt.Printf("%d %x %x\n", incarnation, key[common.HashLength+common.IncarnationLength:], value)
}
return nil
})
check(err)
}
timestamp--
}
}
func fixAccount(chaindata string, addrHash common.Hash, storageRoot common.Hash) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
var a accounts.Account
if ok, err := rawdb.ReadAccount(ethDb, addrHash, &a); err != nil {
panic(err)
} else if !ok {
panic("acc not found")
}
a.Root = storageRoot
if err := rawdb.WriteAccount(ethDb, addrHash, a); err != nil {
panic(err)
}
}
func nextIncarnation(chaindata string, addrHash common.Hash) {
ethDb := ethdb.MustOpen(chaindata)
defer ethDb.Close()
var found bool
var incarnationBytes [common.IncarnationLength]byte
startkey := make([]byte, common.HashLength+common.IncarnationLength+common.HashLength)
var fixedbits int = 8 * common.HashLength
copy(startkey, addrHash[:])
if err := ethDb.Walk(dbutils.CurrentStateBucket, startkey, fixedbits, func(k, v []byte) (bool, error) {
copy(incarnationBytes[:], k[common.HashLength:])
found = true
return false, nil
}); err != nil {
fmt.Printf("Incarnation(z): %d\n", 0)
return
}
if found {
fmt.Printf("Incarnation: %d\n", (^binary.BigEndian.Uint64(incarnationBytes[:]))+1)
return
}
fmt.Printf("Incarnation(f): %d\n", state.FirstContractIncarnation)
}
func repairCurrent() {
historyDb := ethdb.MustOpen("/Volumes/tb4/turbo-geth/ropsten/geth/chaindata")
defer historyDb.Close()
currentDb := ethdb.MustOpen("statedb")
defer currentDb.Close()
check(historyDb.ClearBuckets(dbutils.CurrentStateBucket))
check(historyDb.KV().Update(context.Background(), func(tx ethdb.Tx) error {
newB := tx.Bucket(dbutils.CurrentStateBucket)
count := 0
if err := currentDb.KV().View(context.Background(), func(ctx ethdb.Tx) error {
b := ctx.Bucket(dbutils.CurrentStateBucket)
c := b.Cursor()
for k, v, err := c.First(); k != nil; k, v, err = c.Next() {
if err != nil {
return err
}
if len(k) == 32 {
continue
}
check(newB.Put(k, v))
count++
if count == 10000 {
fmt.Printf("Copied %d storage items\n", count)
}
}
return nil
}); err != nil {
return err
}
return nil
}))
}
func dumpStorage() {
db := ethdb.MustOpen(node.DefaultDataDir() + "/geth/chaindata")
defer db.Close()
if err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
return tx.Bucket(dbutils.StorageHistoryBucket).Cursor().Walk(func(k, v []byte) (bool, error) {
fmt.Printf("%x %x\n", k, v)
return true, nil
})
}); err != nil {
panic(err)
}
}
func printBucket(chaindata string) {
db := ethdb.MustOpen(chaindata)
defer db.Close()
f, err := os.Create("bucket.txt")
check(err)
defer f.Close()
fb := bufio.NewWriter(f)
defer fb.Flush()
if err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
b := tx.Bucket(dbutils.StorageHistoryBucket)
c := b.Cursor()
for k, v, err := c.First(); k != nil; k, v, err = c.Next() {
if err != nil {
return err
}
fmt.Fprintf(fb, "%x %x\n", k, v)
}
return nil
}); err != nil {
panic(err)
}
}
func ValidateTxLookups2(chaindata string) {
startTime := time.Now()
db := ethdb.MustOpen(chaindata)
defer db.Close()
//nolint: errcheck
startTime = time.Now()
sigs := make(chan os.Signal, 1)
interruptCh := make(chan bool, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
go func() {
<-sigs
interruptCh <- true
}()
var blockNum uint64 = 1
validateTxLookups2(db, blockNum, interruptCh)
log.Info("All done", "duration", time.Since(startTime))
}
func validateTxLookups2(db rawdb.DatabaseReader, startBlock uint64, interruptCh chan bool) {
blockNum := startBlock
iterations := 0
var interrupt bool
// Validation Process
blockBytes := big.NewInt(0)
for !interrupt {
blockHash := rawdb.ReadCanonicalHash(db, blockNum)
body := rawdb.ReadBody(db, blockHash, blockNum)
if body == nil {
break
}
select {
case interrupt = <-interruptCh:
log.Info("interrupted, please wait for cleanup...")
default:
}
blockBytes.SetUint64(blockNum)
bn := blockBytes.Bytes()
for _, tx := range body.Transactions {
val, err := db.Get(dbutils.TxLookupPrefix, tx.Hash().Bytes())
iterations++
if iterations%100000 == 0 {
log.Info("Validated", "entries", iterations, "number", blockNum)
}
if bytes.Compare(val, bn) != 0 {
check(err)
panic(fmt.Sprintf("Validation process failed(%d). Expected %b, got %b", iterations, bn, val))
}
}
blockNum++
}
}
func getModifiedAccounts(chaindata string) {
db := ethdb.MustOpen(chaindata)
defer db.Close()
addrs, err := ethdb.GetModifiedAccounts(db, 49300, 49400)
check(err)
fmt.Printf("Len(addrs)=%d\n", len(addrs))
}
func resetState(chaindata string) {
db := ethdb.MustOpen(chaindata)
defer db.Close()
check(db.ClearBuckets(
dbutils.CurrentStateBucket,
dbutils.AccountChangeSetBucket,
dbutils.StorageChangeSetBucket,
dbutils.ContractCodeBucket,
dbutils.PlainStateBucket,
dbutils.PlainAccountChangeSetBucket,
dbutils.PlainStorageChangeSetBucket,
dbutils.PlainContractCodeBucket,
dbutils.IncarnationMapBucket,
dbutils.CodeBucket,
))
core.UsePlainStateExecution = true
_, _, err := core.DefaultGenesisBlock().CommitGenesisState(db, false)
check(err)
err = stages.SaveStageProgress(db, stages.Execution, 0, nil)
check(err)
fmt.Printf("Reset state done\n")
}
func resetHashedState(chaindata string) {
db := ethdb.MustOpen(chaindata)
defer db.Close()
check(db.ClearBuckets(
dbutils.IntermediateTrieHashBucket,
dbutils.IntermediateTrieHashBucket,
))
err := stages.SaveStageProgress(db, stages.HashState, 0, nil)
check(err)
fmt.Printf("Reset hashed state done\n")
}
type Receiver struct {
defaultReceiver *trie.DefaultReceiver
accountMap map[string]*accounts.Account
storageMap map[string][]byte
unfurlList []string
currentIdx int
}
func (r *Receiver) Receive(
itemType trie.StreamItem,
accountKey []byte,
storageKey []byte,
accountValue *accounts.Account,
storageValue []byte,
hash []byte,
cutoff int,
) error {
for r.currentIdx < len(r.unfurlList) {
ks := r.unfurlList[r.currentIdx]
k := []byte(ks)
var c int
switch itemType {
case trie.StorageStreamItem, trie.SHashStreamItem:
c = bytes.Compare(k, storageKey)
case trie.AccountStreamItem, trie.AHashStreamItem:
c = bytes.Compare(k, accountKey)
case trie.CutoffStreamItem:
c = -1
}
if c > 0 {
return r.defaultReceiver.Receive(itemType, accountKey, storageKey, accountValue, storageValue, hash, cutoff)
}
if len(k) > common.HashLength {
v := r.storageMap[ks]
if len(v) > 0 {
if err := r.defaultReceiver.Receive(trie.StorageStreamItem, nil, k, nil, v, nil, 0); err != nil {
return err
}
}
} else {
v := r.accountMap[ks]
if v != nil {
if err := r.defaultReceiver.Receive(trie.AccountStreamItem, k, nil, v, nil, nil, 0); err != nil {
return err
}
}
}
r.currentIdx++
if c == 0 {
return nil
}
}
// We ran out of modifications, simply pass through
return r.defaultReceiver.Receive(itemType, accountKey, storageKey, accountValue, storageValue, hash, cutoff)
}
func (r *Receiver) Result() trie.SubTries {
return r.defaultReceiver.Result()
}
func regenerate(chaindata string) error {
var m runtime.MemStats
db := ethdb.MustOpen(chaindata)
defer db.Close()
check(db.ClearBuckets(
dbutils.IntermediateTrieHashBucket,
))
headHash := rawdb.ReadHeadBlockHash(db)
headNumber := rawdb.ReadHeaderNumber(db, headHash)
headHeader := rawdb.ReadHeader(db, headHash, *headNumber)
log.Info("Regeneration started")
collector := etl.NewCollector(".", etl.NewSortableBuffer(etl.BufferOptimalSize))
hashCollector := func(keyHex []byte, hash []byte) error {
if len(keyHex)%2 != 0 || len(keyHex) == 0 {
return nil
}
var k []byte
trie.CompressNibbles(keyHex, &k)
if hash == nil {
return collector.Collect(k, nil)
}
return collector.Collect(k, common.CopyBytes(hash))
}
loader := trie.NewFlatDbSubTrieLoader()
if err := loader.Reset(db, trie.NewRetainList(0), trie.NewRetainList(0), hashCollector /* HashCollector */, [][]byte{nil}, []int{0}, false); err != nil {
return err
}
if subTries, err := loader.LoadSubTries(); err == nil {
runtime.ReadMemStats(&m)
log.Info("Loaded initial trie", "root", fmt.Sprintf("%x", subTries.Hashes[0]),
"expected root", fmt.Sprintf("%x", headHeader.Root),
"alloc", common.StorageSize(m.Alloc), "sys", common.StorageSize(m.Sys), "numGC", int(m.NumGC))
} else {
return err
}
/*
quitCh := make(chan struct{})
if err := collector.Load(db, dbutils.IntermediateTrieHashBucket, etl.IdentityLoadFunc, etl.TransformArgs{Quit: quitCh}); err != nil {
return err
}
*/
log.Info("Regeneration ended")
return nil
}
func testGetProof(chaindata string, address common.Address, rewind int, regen bool) error {
if regen {
if err := regenerate(chaindata); err != nil {
return err
}
}
storageKeys := []string{}
var m runtime.MemStats
runtime.ReadMemStats(&m)
db := ethdb.MustOpen(chaindata)
defer db.Close()
headHash := rawdb.ReadHeadBlockHash(db)
headNumber := rawdb.ReadHeaderNumber(db, headHash)
block := *headNumber - uint64(rewind)
log.Info("GetProof", "address", address, "storage keys", len(storageKeys), "head", *headNumber, "block", block,
"alloc", common.StorageSize(m.Alloc), "sys", common.StorageSize(m.Sys), "numGC", int(m.NumGC))
ts := dbutils.EncodeTimestamp(block + 1)
accountMap := make(map[string]*accounts.Account)
if err := db.Walk(dbutils.AccountChangeSetBucket, ts, 0, func(k, v []byte) (bool, error) {
timestamp, _ := dbutils.DecodeTimestamp(k)
if timestamp > *headNumber {
return false, nil
}
if changeset.Len(v) > 0 {
walker := func(kk, vv []byte) error {
if _, ok := accountMap[string(kk)]; !ok {
if len(vv) > 0 {
var a accounts.Account
if innerErr := a.DecodeForStorage(vv); innerErr != nil {
return innerErr
}
accountMap[string(kk)] = &a
} else {
accountMap[string(kk)] = nil
}
}
return nil
}
v = common.CopyBytes(v) // Making copy because otherwise it will be invalid after the transaction
if innerErr := changeset.AccountChangeSetBytes(v).Walk(walker); innerErr != nil {
return false, innerErr
}
}
return true, nil
}); err != nil {
return err
}
runtime.ReadMemStats(&m)
log.Info("Constructed account map", "size", len(accountMap),
"alloc", common.StorageSize(m.Alloc), "sys", common.StorageSize(m.Sys), "numGC", int(m.NumGC))
storageMap := make(map[string][]byte)
if err := db.Walk(dbutils.StorageChangeSetBucket, ts, 0, func(k, v []byte) (bool, error) {
timestamp, _ := dbutils.DecodeTimestamp(k)
if timestamp > *headNumber {
return false, nil
}
if changeset.Len(v) > 0 {
walker := func(kk, vv []byte) error {
if _, ok := storageMap[string(kk)]; !ok {
storageMap[string(kk)] = vv
}
return nil
}
v = common.CopyBytes(v) // Making copy because otherwise it will be invalid after the transaction
if innerErr := changeset.StorageChangeSetBytes(v).Walk(walker); innerErr != nil {
return false, innerErr
}
}
return true, nil
}); err != nil {
return err
}
runtime.ReadMemStats(&m)
log.Info("Constructed storage map", "size", len(storageMap),
"alloc", common.StorageSize(m.Alloc), "sys", common.StorageSize(m.Sys), "numGC", int(m.NumGC))
var unfurlList = make([]string, len(accountMap)+len(storageMap))
unfurl := trie.NewRetainList(0)
i := 0
for ks, acc := range accountMap {
unfurlList[i] = ks
i++
unfurl.AddKey([]byte(ks))
if acc != nil {
// Fill the code hashes
if acc.Incarnation > 0 && acc.IsEmptyCodeHash() {
if codeHash, err1 := db.Get(dbutils.ContractCodeBucket, dbutils.GenerateStoragePrefix([]byte(ks), acc.Incarnation)); err1 == nil {
copy(acc.CodeHash[:], codeHash)
} else {
return err1
}
}
}
}
for ks := range storageMap {
unfurlList[i] = ks
i++
unfurl.AddKey([]byte(ks))
}
rl := trie.NewRetainList(0)
addrHash, err := common.HashData(address[:])
if err != nil {
return err
}
rl.AddKey(addrHash[:])
unfurl.AddKey(addrHash[:])
for _, key := range storageKeys {
keyAsHash := common.HexToHash(key)
if keyHash, err1 := common.HashData(keyAsHash[:]); err1 == nil {
//TODO Add incarnation in the middle of this
trieKey := append(addrHash[:], keyHash[:]...)
rl.AddKey(trieKey)
unfurl.AddKey(trieKey)
} else {
return err1
}
}
sort.Strings(unfurlList)
runtime.ReadMemStats(&m)
log.Info("Constructed account unfurl lists",
"alloc", common.StorageSize(m.Alloc), "sys", common.StorageSize(m.Sys), "numGC", int(m.NumGC))
loader := trie.NewFlatDbSubTrieLoader()
if err = loader.Reset(db, unfurl, trie.NewRetainList(0), nil /* HashCollector */, [][]byte{nil}, []int{0}, false); err != nil {
return err
}
r := &Receiver{defaultReceiver: trie.NewDefaultReceiver(), unfurlList: unfurlList, accountMap: accountMap, storageMap: storageMap}
r.defaultReceiver.Reset(rl, nil /* HashCollector */, false)
loader.SetStreamReceiver(r)
subTries, err1 := loader.LoadSubTries()
if err1 != nil {
return err1
}
runtime.ReadMemStats(&m)
log.Info("Loaded subtries",
"alloc", common.StorageSize(m.Alloc), "sys", common.StorageSize(m.Sys), "numGC", int(m.NumGC))
hash := rawdb.ReadCanonicalHash(db, block)
header := rawdb.ReadHeader(db, hash, block)
tr := trie.New(common.Hash{})
if err = tr.HookSubTries(subTries, [][]byte{nil}); err != nil {
fmt.Printf("Error hooking: %v\n", err)
}
runtime.ReadMemStats(&m)
log.Info("Constructed trie", "nodes", tr.NumberOfAccounts(),
"alloc", common.StorageSize(m.Alloc), "sys", common.StorageSize(m.Sys), "numGC", int(m.NumGC))
fmt.Printf("Resulting root: %x (subTrie %x), expected root: %x\n", tr.Hash(), subTries.Hashes[0], header.Root)
return nil
}
func testStage3(chaindata string, reset bool, block uint64) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
if reset {
if err := db.ClearBuckets(
dbutils.Senders,
); err != nil {
return err
}
if err := stages.SaveStageProgress(db, stages.Senders, 0, nil); err != nil {
return err
}
}
var err error
var stage2progress uint64
if stage2progress, _, err = stages.GetStageProgress(db, stages.Bodies); err != nil {
return err
}
log.Info("Stage2", "progress", stage2progress)
var stage3progress uint64
if stage3progress, _, err = stages.GetStageProgress(db, stages.Senders); err != nil {
return err
}
log.Info("Stage3", "progress", stage3progress)
ch := make(chan struct{})
s := &stagedsync.StageState{Stage: stages.Senders, BlockNumber: stage3progress}
const batchSize = 10000
const blockSize = 4096
n := runtime.NumCPU()
cfg := stagedsync.Stage3Config{
BatchSize: batchSize,
BlockSize: blockSize,
BufferSize: (blockSize * 10 / 20) * 10000, // 20*4096
StartTrace: false,
Prof: false,
NumOfGoroutines: n,
ReadChLen: 4,
Now: time.Now(),
}
if err = stagedsync.SpawnRecoverSendersStage(cfg, s, db, params.MainnetChainConfig, block, "", ch); err != nil {
return err
}
return nil
}
func testStage4(chaindata string, block uint64) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
var err error
var stage4progress uint64
if stage4progress, _, err = stages.GetStageProgress(db, stages.Execution); err != nil {
return err
}
core.UsePlainStateExecution = true
ch := make(chan struct{})
s := &stagedsync.StageState{Stage: stages.Execution, BlockNumber: stage4progress}
blockchain, _ := core.NewBlockChain(db, nil, params.MainnetChainConfig, ethash.NewFaker(), vm.Config{}, nil, nil, nil)
if err = stagedsync.SpawnExecuteBlocksStage(s, db, blockchain, block, ch, nil, false); err != nil {
return err
}
return nil
}
func testUnwind4(chaindata string, rewind uint64) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
var err error
var stage4progress uint64
if stage4progress, _, err = stages.GetStageProgress(db, stages.Execution); err != nil {
return err
}
log.Info("Stage4", "progress", stage4progress)
core.UsePlainStateExecution = true
u := &stagedsync.UnwindState{Stage: stages.Execution, UnwindPoint: stage4progress - rewind}
s := &stagedsync.StageState{Stage: stages.Execution, BlockNumber: stage4progress}
if err = stagedsync.UnwindExecutionStage(u, s, db); err != nil {
return err
}
return nil
}
func testStage5(chaindata string, reset bool) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
if reset {
if err := db.ClearBuckets(
dbutils.CurrentStateBucket,
dbutils.ContractCodeBucket,
dbutils.IntermediateTrieHashBucket,
); err != nil {
return err
}
if err := stages.SaveStageProgress(db, stages.IntermediateHashes, 0, nil); err != nil {
return err
}
if err := stages.SaveStageProgress(db, stages.HashState, 0, nil); err != nil {
return err
}
}
var err error
var stage4progress uint64
if stage4progress, _, err = stages.GetStageProgress(db, stages.Execution); err != nil {
return err
}
var stage5progress uint64
if stage5progress, _, err = stages.GetStageProgress(db, stages.IntermediateHashes); err != nil {
return err
}
log.Info("Stage4", "progress", stage4progress)
log.Info("Stage5", "progress", stage5progress)
core.UsePlainStateExecution = true
ch := make(chan struct{})
stageState := &stagedsync.StageState{Stage: stages.IntermediateHashes, BlockNumber: stage5progress}
if err = stagedsync.SpawnIntermediateHashesStage(stageState, db, "", ch); err != nil {
return err
}
close(ch)
return nil
}
func testUnwind5(chaindata string, rewind uint64) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
var err error
var stage5progress uint64
if stage5progress, _, err = stages.GetStageProgress(db, stages.IntermediateHashes); err != nil {
return err
}
log.Info("Stage5", "progress", stage5progress)
core.UsePlainStateExecution = true
ch := make(chan struct{})
u := &stagedsync.UnwindState{Stage: stages.IntermediateHashes, UnwindPoint: stage5progress - rewind}
s := &stagedsync.StageState{Stage: stages.IntermediateHashes, BlockNumber: stage5progress}
if err = stagedsync.UnwindHashStateStage(u, s, db, "", ch); err != nil {
return err
}
close(ch)
return nil
}
func testStage6(chaindata string, reset bool) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
if reset {
if err := stages.SaveStageProgress(db, stages.HashState, 0, nil); err != nil {
return err
}
}
var err error
var stage5progress uint64
if stage5progress, _, err = stages.GetStageProgress(db, stages.IntermediateHashes); err != nil {
return err
}
var stage6progress uint64
if stage6progress, _, err = stages.GetStageProgress(db, stages.HashState); err != nil {
return err
}
log.Info("Stage5", "progress", stage5progress)
log.Info("Stage6", "progress", stage6progress)
core.UsePlainStateExecution = true
ch := make(chan struct{})
stageState := &stagedsync.StageState{Stage: stages.HashState, BlockNumber: stage6progress}
if err = stagedsync.SpawnHashStateStage(stageState, db, "", ch); err != nil {
return err
}
close(ch)
return nil
}
func testUnwind6(chaindata string, rewind uint64) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
var err error
var stage6progress uint64
if stage6progress, _, err = stages.GetStageProgress(db, stages.HashState); err != nil {
return err
}
log.Info("Stage6", "progress", stage6progress)
core.UsePlainStateExecution = true
ch := make(chan struct{})
u := &stagedsync.UnwindState{Stage: stages.HashState, UnwindPoint: stage6progress - rewind}
s := &stagedsync.StageState{Stage: stages.HashState, BlockNumber: stage6progress}
if err = stagedsync.UnwindIntermediateHashesStage(u, s, db, "", ch); err != nil {
return err
}
close(ch)
return nil
}
func testStage78(chaindata string, reset bool) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
if reset {
if err := db.ClearBuckets(
dbutils.AccountsHistoryBucket,
dbutils.StorageHistoryBucket,
); err != nil {
return err
}
if err := stages.SaveStageProgress(db, stages.AccountHistoryIndex, 0, nil); err != nil {
return err
}
if err := stages.SaveStageProgress(db, stages.StorageHistoryIndex, 0, nil); err != nil {
return err
}
}
var err error
var stage4progress uint64
if stage4progress, _, err = stages.GetStageProgress(db, stages.Execution); err != nil {
return err
}
var stage7progress uint64
if stage7progress, _, err = stages.GetStageProgress(db, stages.AccountHistoryIndex); err != nil {
return err
}
var stage8progress uint64
if stage8progress, _, err = stages.GetStageProgress(db, stages.StorageHistoryIndex); err != nil {
return err
}
log.Info("Stage4", "progress", stage4progress)
log.Info("Stage7", "progress", stage7progress)
log.Info("Stage8", "progress", stage8progress)
core.UsePlainStateExecution = true
ch := make(chan struct{})
stageState := &stagedsync.StageState{Stage: stages.AccountHistoryIndex, BlockNumber: stage7progress}
if err = stagedsync.SpawnAccountHistoryIndex(stageState, db, "", ch); err != nil {
return err
}
stageState = &stagedsync.StageState{Stage: stages.StorageHistoryIndex, BlockNumber: stage8progress}
if err = stagedsync.SpawnStorageHistoryIndex(stageState, db, "", ch); err != nil {
return err
}
close(ch)
return nil
}
func findPreimage(chaindata string, hash common.Hash) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
if err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
b := tx.Bucket(dbutils.PlainStateBucket)
c := b.Cursor()
k, _, e := c.First()
if e != nil {
return e
}
for k != nil {
if bytes.Equal(crypto.Keccak256(k), hash[:]) {
fmt.Printf("preimage: %x\n", k)
break
}
k, _, e = c.Next()
if e != nil {
return e
}
}
return nil
}); err != nil {
return err
}
return nil
}
/*
CurrentStateBucket layout which utilises DupSort feature of LMDB (store multiple values inside 1 key).
LMDB stores multiple values of 1 key in B+ tree (as keys which have no values).
It allows do Seek by values and other features of LMDB.
You can think about it as Sub-Database which doesn't store any values.
-------------------------------------------------------------
key | value
-------------------------------------------------------------
[acc_hash] | [acc_value]
[acc_hash]+[inc] | [storage1_hash]+[storage1_value]
| [storage2_hash]+[storage2_value] // this value has no own key. it's 2nd value of [acc_hash]+[inc] key.
| [storage3_hash]+[storage3_value]
| ...
[acc_hash]+[old_inc] | [storage1_hash]+[storage1_value]
| ...
[acc2_hash] | [acc2_value]
...
On 5M block CurrentState bucket using 5.9Gb. This layout using 2.3Gb. See dupSortState()
Another feature of this layout is:
- DupSort stores values in B+ tree
- To delete key with multiple values - LMDB does add pages (of sub-tree) to the free list
- Same logic used when `drop bucket`
- LMDB doesn't deletes much. Deletion of 17GB (1 key with 1 billions values 8 bytes each) took 2sec on SSD. See deleteLargeDupSortKey()
- It means performance of deletion contracts with huge storage is predictable
- Means we can drop incarnations concept
As further evolution of this idea:
- Can add CodeHash as 2-nd value of [acc_hash] key (just use 1st byte to store type of value)
- And drop ContractCodeBucket
*/
func dupSortState(chaindata string) {
db := ethdb.MustOpen(chaindata)
defer db.Close()
dbFile := "statedb_dupsort"
dbFile2 := "statedb_no_dupsort"
err := os.MkdirAll(dbFile, 0744)
check(err)
env, err := lmdb.NewEnv()
check(err)
err = env.SetMaxDBs(100)
check(err)
err = env.SetMapSize(32 << 40) // 32TB
check(err)
var flags uint = lmdb.NoReadahead
err = env.Open(dbFile, flags, 0664)
check(err)
defer env.Close()
err = os.MkdirAll(dbFile2, 0744)
check(err)
env2, err := lmdb.NewEnv()
check(err)
err = env2.SetMaxDBs(100)
check(err)
err = env2.SetMapSize(32 << 40) // 32TB
check(err)
var flags2 uint = lmdb.NoReadahead
err = env2.Open(dbFile2, flags2, 0664)
check(err)
defer env.Close()
var newStateBucket lmdb.DBI
err = env.Update(func(tx *lmdb.Txn) error {
var createErr error
newStateBucket, createErr = tx.OpenDBI(string(dbutils.CurrentStateBucket), lmdb.Create|lmdb.DupSort)
check(createErr)
return nil
})
check(err)
err = db.KV().View(context.Background(), func(tx ethdb.Tx) error {
b := tx.Bucket(dbutils.CurrentStateBucket)
sz, _ := b.Size()
fmt.Printf("Current State bucket size: %s\n", common.StorageSize(sz))
txn, _ := env.BeginTxn(nil, 0)
err = txn.Drop(newStateBucket, false)
check(err)
txn2, _ := env2.BeginTxn(nil, 0)
err = txn2.Drop(newStateBucket, false)
check(err)
c := b.Cursor()
i := 0
for k, v, err1 := c.First(); k != nil; k, v, err = c.Next() {
check(err1)
i++
if i%1_000_000 == 0 {
err = txn.Commit()
check(err)
txn, _ = env.BeginTxn(nil, 0)
err = txn2.Commit()
check(err)
txn2, _ = env2.BeginTxn(nil, 0)
fmt.Printf("%x\n", k[:2])
}
if len(k) == common.HashLength {
err = txn.Put(newStateBucket, common.CopyBytes(k), common.CopyBytes(v), lmdb.AppendDup)
check(err)
err = txn2.Put(newStateBucket, common.CopyBytes(k), common.CopyBytes(v), lmdb.Append)
check(err)
} else {
prefix := k[:common.HashLength+common.IncarnationLength]
suffix := k[common.HashLength+common.IncarnationLength:]
err = txn.Put(newStateBucket, common.CopyBytes(prefix), append(suffix, v...), lmdb.AppendDup)
check(err)
err = txn2.Put(newStateBucket, common.CopyBytes(k), common.CopyBytes(v), lmdb.Append)
check(err)
}
}
err = txn.Commit()
check(err)
err = txn2.Commit()
check(err)
return nil
})
check(err)
err = env.View(func(txn *lmdb.Txn) (err error) {
st, err := txn.Stat(newStateBucket)
check(err)
fmt.Printf("Current bucket size: %s\n", common.StorageSize((st.LeafPages+st.BranchPages+st.OverflowPages)*uint64(os.Getpagesize())))
cur, err := txn.OpenCursor(newStateBucket)
check(err)
i := 0
for k, v, err := cur.Get(nil, nil, lmdb.First); !lmdb.IsNotFound(err); k, v, err = cur.Get(nil, nil, lmdb.Next) {
check(err)
i++
if i == 100 { // print first 100
return nil
}
if len(k) == 32 {
fmt.Printf("Acc: %x, %x\n", k, v)
continue
}
// try to seek on incarnation=2 and check that no incarnation=1 are visible
for _, v, err := cur.Get(nil, nil, lmdb.FirstDup); !lmdb.IsNotFound(err); _, v, err = cur.Get(nil, nil, lmdb.NextDup) {
fmt.Printf("Storage: %x, %x\n", k, v)
}
}
return nil
})
check(err)
}
func deleteLargeDupSortKey() {
dbFile := "statedb_dupsort_delete"
err := os.MkdirAll(dbFile, 0744)
check(err)
env, err := lmdb.NewEnv()
check(err)
err = env.SetMaxDBs(100)
check(err)
err = env.SetMapSize(32 << 40) // 32TB
check(err)
var flags uint = lmdb.NoReadahead
err = env.Open(dbFile, flags, 0664)
check(err)
defer env.Close()
var newStateBucket lmdb.DBI
err = env.Update(func(tx *lmdb.Txn) error {
var createErr error
newStateBucket, createErr = tx.OpenDBI(string(dbutils.CurrentStateBucket), lmdb.Create|lmdb.DupSort)
check(createErr)
return nil
})
check(err)
k := common.FromHex("000034c2eb874b6125c8a84e69fe77640d468ccef4c02a2d20c284446dbb1460")
txn, _ := env.BeginTxn(nil, 0)
err = txn.Drop(newStateBucket, false)
check(err)
for i := uint64(0); i < 1_000_000_000; i++ {
if i%10_000_000 == 0 {
err = txn.Commit()
check(err)
txn, err = env.BeginTxn(nil, 0)
check(err)
fmt.Printf("%dM\n", i/1_000_000)
}
v := make([]byte, 8)
binary.BigEndian.PutUint64(v, i)
err = txn.Put(newStateBucket, k, v, lmdb.AppendDup)
check(err)
}
err = txn.Commit()
check(err)
//print new bucket size
err = env.View(func(txn *lmdb.Txn) (err error) {
st, err := txn.Stat(newStateBucket)
check(err)
fmt.Printf("Current bucket size: %s\n", common.StorageSize((st.LeafPages+st.BranchPages+st.OverflowPages)*uint64(os.Getpagesize())))
return nil
})
check(err)
// delete single DupSort key
t := time.Now()
err = env.Update(func(txn *lmdb.Txn) (err error) {
cur, err := txn.OpenCursor(newStateBucket)
check(err)
k, _, err = cur.Get(k, nil, lmdb.SetRange)
check(err)
err = cur.Del(lmdb.NoDupData)
check(err)
return nil
})
check(err)
fmt.Printf("Deletion took: %s\n", time.Since(t))
// check that no keys are left in the bucket
err = env.View(func(txn *lmdb.Txn) (err error) {
st, err := txn.Stat(newStateBucket)
check(err)
fmt.Printf("Current bucket size: %s\n", common.StorageSize((st.LeafPages+st.BranchPages+st.OverflowPages)*uint64(os.Getpagesize())))
cur, err := txn.OpenCursor(newStateBucket)
check(err)
for k, v, err := cur.Get(nil, nil, lmdb.First); !lmdb.IsNotFound(err); k, v, err = cur.Get(nil, nil, lmdb.Next) {
check(err)
fmt.Printf("Still see some key: %x, %x\n", k, v)
panic(1)
}
return nil
})
check(err)
}
func compareBucket(chaindata string, chaindataCopy string, bucket string) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
copyDb := ethdb.MustOpen(chaindataCopy)
defer copyDb.Close()
count := 0
if err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
b := tx.Bucket([]byte(bucket))
c := b.Cursor()
k, v, e := c.First()
if e != nil {
return e
}
return copyDb.KV().View(context.Background(), func(copyTx ethdb.Tx) error {
copyB := copyTx.Bucket([]byte(bucket))
copyC := copyB.Cursor()
copyK, copyV, copyE := copyC.First()
if copyE != nil {
return copyE
}
for k != nil || copyK != nil {
count++
if count%100000 == 0 {
fmt.Printf("Compared %d records\n", count)
}
if k == nil {
fmt.Printf("Missing in db: %x [%x]\n", copyK, copyV)
copyK, copyV, copyE = copyC.Next()
if copyE != nil {
return copyE
}
} else if copyK == nil {
fmt.Printf("Missing copyDb: %x [%x]\n", k, v)
k, v, e = c.Next()
if e != nil {
return e
}
} else {
switch bytes.Compare(k, copyK) {
case -1:
fmt.Printf("Missing copyDb: %x [%x]\n", k, v)
k, v, e = c.Next()
if e != nil {
return e
}
case 1:
fmt.Printf("Missing in db: %x [%x]\n", copyK, copyV)
copyK, copyV, copyE = copyC.Next()
if copyE != nil {
return copyE
}
case 0:
if !bytes.Equal(v, copyV) {
fmt.Printf("Different values for %x. db: [%x], copyDb: [%x]\n", k, v, copyV)
}
k, v, e = c.Next()
if e != nil {
return e
}
copyK, copyV, copyE = copyC.Next()
if copyE != nil {
return copyE
}
default:
fmt.Printf("Unexpected result of bytes.Compare: %d\n", bytes.Compare(k, copyK))
}
}
}
return nil
})
}); err != nil {
return err
}
return nil
}
func printStages(chaindata string) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
var err error
var progress uint64
for stage := stages.SyncStage(0); stage < stages.Finish; stage++ {
if progress, _, err = stages.GetStageProgress(db, stage); err != nil {
return err
}
fmt.Printf("Stage: %d, progress: %d\n", stage, progress)
}
return nil
}
func fixStages(chaindata string) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
var err error
var progress uint64
var list []uint64
for stage := stages.SyncStage(0); stage < stages.Finish; stage++ {
if progress, _, err = stages.GetStageProgress(db, stage); err != nil {
return err
}
fmt.Printf("Stage: %d, progress: %d\n", stage, progress)
list = append(list, progress)
}
for stage := stages.IntermediateHashes; stage < stages.TxLookup; stage++ {
if err = stages.SaveStageProgress(db, stage+1, list[int(stage)], nil); err != nil {
return err
}
}
return nil
}
func testStageLoop(chaindata string) error {
for block := uint64(10000000); block < uint64(11000000); block += uint64(100000) {
if err := testStage4(chaindata, block); err != nil {
return err
}
if err := testStage5(chaindata, true /* reset */); err != nil {
return err
}
}
return nil
}
func searchChangeSet(chaindata string, key []byte, block uint64) error {
fmt.Printf("Searching changesets\n")
db := ethdb.MustOpen(chaindata)
defer db.Close()
if err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
st := tx.Bucket(dbutils.PlainAccountChangeSetBucket)
start := dbutils.EncodeTimestamp(block)
c := st.Cursor()
for k, v, err := c.Seek(start); k != nil; k, v, err = c.Next() {
if err != nil {
return err
}
timestamp, _ := dbutils.DecodeTimestamp(k)
//fmt.Printf("timestamp: %d\n", timestamp)
if err1 := changeset.AccountChangeSetPlainBytes(v).Walk(func(kk, vv []byte) error {
if bytes.Equal(kk, key) {
fmt.Printf("Found in block %d with value %x\n", timestamp, vv)
}
return nil
}); err1 != nil {
return err1
}
}
return nil
}); err != nil {
return err
}
return nil
}
func searchStorageChangeSet(chaindata string, key []byte, block uint64) error {
fmt.Printf("Searching storage changesets\n")
db := ethdb.MustOpen(chaindata)
defer db.Close()
if err := db.KV().View(context.Background(), func(tx ethdb.Tx) error {
st := tx.Bucket(dbutils.PlainStorageChangeSetBucket)
start := dbutils.EncodeTimestamp(block)
c := st.Cursor()
for k, v, err := c.Seek(start); k != nil; k, v, err = c.Next() {
if err != nil {
return err
}
timestamp, _ := dbutils.DecodeTimestamp(k)
if err1 := changeset.StorageChangeSetPlainBytes(v).Walk(func(kk, vv []byte) error {
if bytes.Equal(kk, key) {
fmt.Printf("Found in block %d with value %x\n", timestamp, vv)
}
return nil
}); err1 != nil {
return err1
}
}
return nil
}); err != nil {
return err
}
return nil
}
func buildHistory(chaindata string, reset bool) error {
db := ethdb.MustOpen(chaindata)
defer db.Close()
if reset {
if err := db.ClearBuckets(
dbutils.AccountsHistoryBucket,
dbutils.StorageHistoryBucket,
); err != nil {
return err
}
if err := stages.SaveStageProgress(db, stages.AccountHistoryIndex, 0, nil); err != nil {
return err
}
if err := stages.SaveStageProgress(db, stages.StorageHistoryIndex, 0, nil); err != nil {
return err
}
}
var err error
var stage4progress uint64
if stage4progress, _, err = stages.GetStageProgress(db, stages.Execution); err != nil {
return err
}
var stage7progress uint64
if stage7progress, _, err = stages.GetStageProgress(db, stages.AccountHistoryIndex); err != nil {
return err
}
var stage8progress uint64
if stage8progress, _, err = stages.GetStageProgress(db, stages.StorageHistoryIndex); err != nil {
return err
}
log.Info("Stage4", "progress", stage4progress)
log.Info("Stage7", "progress", stage7progress)
log.Info("Stage8", "progress", stage8progress)
core.UsePlainStateExecution = true
ch := make(chan struct{})
stageState := &stagedsync.StageState{Stage: stages.AccountHistoryIndex, BlockNumber: stage7progress}
if err = stagedsync.SpawnAccountHistoryIndex(stageState, db, "", ch); err != nil {
return err
}
stageState = &stagedsync.StageState{Stage: stages.StorageHistoryIndex, BlockNumber: stage8progress}
if err = stagedsync.SpawnStorageHistoryIndex(stageState, db, "", ch); err != nil {
return err
}
close(ch)
return nil
}
func main() {
flag.Parse()
var (
ostream log.Handler
glogger *log.GlogHandler
)
usecolor := (isatty.IsTerminal(os.Stderr.Fd()) || isatty.IsCygwinTerminal(os.Stderr.Fd())) && os.Getenv("TERM") != "dumb"
output := io.Writer(os.Stderr)
if usecolor {
output = colorable.NewColorableStderr()
}
ostream = log.StreamHandler(output, log.TerminalFormat(usecolor))
glogger = log.NewGlogHandler(ostream)
log.Root().SetHandler(glogger)
glogger.Verbosity(log.Lvl(*verbosity))
if *cpuprofile != "" {
f, err := os.Create(*cpuprofile)
if err != nil {
log.Error("could not create CPU profile", "error", err)
return
}
if err := pprof.StartCPUProfile(f); err != nil {
log.Error("could not start CPU profile", "error", err)
return
}
defer pprof.StopCPUProfile()
}
//db := ethdb.MustOpen("/home/akhounov/.ethereum/geth/chaindata")
//db := ethdb.MustOpen(node.DefaultDataDir() + "/geth/chaindata")
//check(err)
//defer db.Close()
if *action == "bucketStats" {
bucketStats(*chaindata)
}
if *action == "syncChart" {
mychart()
}
//testRebuild()
if *action == "testRewind" {
testRewind(*chaindata, *block, *rewind)
}
//hashFile()
//buildHashFromFile()
if *action == "testResolve" {
testResolve(*chaindata)
}
//rlpIndices()
//printFullNodeRLPs()
//testStartup()
//testDifficulty()
//testRewindTests()
//if *reset != -1 {
// testReset(uint64(*reset))
//}
if *action == "testBlockHashes" {
testBlockHashes(*chaindata, *block, common.HexToHash(*hash))
}
//printBuckets(db)
//printTxHashes()
//relayoutKeys()
//testRedis()
//upgradeBlocks()
//compareTries()
if *action == "invTree" {
invTree("root", "right", "diff", *name)
}
//invTree("iw", "ir", "id", *block, true)
//loadAccount()
if *action == "preimage" {
if err := findPreimage(*chaindata, common.HexToHash(*hash)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "addPreimage" {
addPreimage(*chaindata, common.HexToHash(*hash), common.FromHex(*preImage))
}
//printBranches(uint64(*block))
if *action == "execToBlock" {
execToBlock(*chaindata, uint64(*block), false)
}
//extractTrie(*block)
//repair()
if *action == "readAccount" {
readAccount(*chaindata, common.HexToAddress(*account), uint64(*block), uint64(*rewind))
}
if *action == "readPlainAccount" {
readPlainAccount(*chaindata, common.HexToAddress(*account))
}
if *action == "fixAccount" {
fixAccount(*chaindata, common.HexToHash(*account), common.HexToHash(*hash))
}
if *action == "nextIncarnation" {
nextIncarnation(*chaindata, common.HexToHash(*account))
}
//repairCurrent()
//testMemBolt()
//fmt.Printf("\u00b3\n")
if *action == "dumpStorage" {
dumpStorage()
}
if *action == "current" {
printCurrentBlockNumber(*chaindata)
}
if *action == "bucket" {
printBucket(*chaindata)
}
if *action == "val-tx-lookup-2" {
ValidateTxLookups2(*chaindata)
}
if *action == "modiAccounts" {
getModifiedAccounts(*chaindata)
}
if *action == "slice" {
dbSlice(*chaindata, []byte(*bucket), common.FromHex(*hash))
}
if *action == "resetState" {
resetState(*chaindata)
}
if *action == "resetHashed" {
resetHashedState(*chaindata)
}
if *action == "getProof" {
if err := testGetProof(*chaindata, common.HexToAddress(*account), *rewind, false); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "reset3" {
if err := testStage3(*chaindata, true, uint64(*block)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "stage3" {
if err := testStage3(*chaindata, false, uint64(*block)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "stage4" {
if err := testStage4(*chaindata, uint64(*block)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "unwind4" {
if err := testUnwind4(*chaindata, uint64(*rewind)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "stage5" {
if err := testStage5(*chaindata, false /* reset */); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "reset5" {
if err := testStage5(*chaindata, true /* reset */); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "unwind5" {
if err := testUnwind5(*chaindata, uint64(*rewind)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "stage6" {
if err := testStage6(*chaindata, false /* reset */); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "reset6" {
if err := testStage6(*chaindata, true /* reset */); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "unwind6" {
if err := testUnwind6(*chaindata, uint64(*rewind)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "stage78" {
if err := testStage78(*chaindata, false /* reset */); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "reset78" {
if err := testStage78(*chaindata, true /* reset */); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "stageLoop" {
if err := testStageLoop(*chaindata); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "regenerateIH" {
if err := regenerate(*chaindata); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "searchChangeSet" {
if err := searchChangeSet(*chaindata, common.FromHex(*hash), uint64(*block)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "searchStorageChangeSet" {
if err := searchStorageChangeSet(*chaindata, common.FromHex(*hash), uint64(*block)); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "printStages" {
if err := printStages(*chaindata); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "fixStages" {
if err := fixStages(*chaindata); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "deleteLargeDupSortKey" {
deleteLargeDupSortKey()
}
if *action == "dupSortState" {
dupSortState(*chaindata)
}
if *action == "compare" {
if err := compareBucket(*chaindata, *chaindata+"-copy", *bucket); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
if *action == "resetHistory" {
if err := buildHistory(*chaindata, true /* reset */); err != nil {
fmt.Printf("Error: %v\n", err)
}
}
}