mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-06 02:52:19 +00:00
0be3044b7e
* rename * rename "make grpc" * rename "abi bindings templates" * rename "abi bindings templates"
92 lines
3.1 KiB
Go
92 lines
3.1 KiB
Go
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/ledgerwatch/erigon/common/dbutils"
|
|
"github.com/ledgerwatch/erigon/core/rawdb"
|
|
"github.com/ledgerwatch/erigon/eth/stagedsync"
|
|
"github.com/ledgerwatch/erigon/ethdb"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func init() {
|
|
withDatadir(verifyStateSnapshotCmd)
|
|
withSnapshotFile(verifyStateSnapshotCmd)
|
|
withBlock(verifyStateSnapshotCmd)
|
|
|
|
rootCmd.AddCommand(verifyStateSnapshotCmd)
|
|
}
|
|
|
|
//
|
|
var verifyStateSnapshotCmd = &cobra.Command{
|
|
Use: "verify_state",
|
|
Short: "Verify state snapshot",
|
|
Example: "go run cmd/snapshots/generator/main.go verify_state --block 11000000 --snapshot /media/b00ris/nvme/snapshots/state/ --datadir /media/b00ris/nvme/backup/snapshotsync/",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return VerifyStateSnapshot(cmd.Context(), chaindata, snapshotFile, block, database)
|
|
},
|
|
}
|
|
|
|
func VerifyStateSnapshot(ctx context.Context, dbPath, snapshotPath string, block uint64, database string) error {
|
|
var snkv, tmpDB ethdb.RwKV
|
|
tmpPath, err := ioutil.TempDir(os.TempDir(), "vrf*")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if database == "lmdb" {
|
|
snkv = ethdb.NewLMDB().WithBucketsConfig(func(defaultBuckets dbutils.BucketsCfg) dbutils.BucketsCfg {
|
|
return dbutils.BucketsCfg{
|
|
dbutils.PlainStateBucket: dbutils.BucketsConfigs[dbutils.PlainStateBucket],
|
|
dbutils.PlainContractCodeBucket: dbutils.BucketsConfigs[dbutils.PlainContractCodeBucket],
|
|
dbutils.CodeBucket: dbutils.BucketsConfigs[dbutils.CodeBucket],
|
|
}
|
|
}).Path(snapshotPath).Readonly().MustOpen()
|
|
tmpDB = ethdb.NewLMDB().Path(tmpPath).MustOpen()
|
|
} else {
|
|
snkv = ethdb.NewMDBX().WithBucketsConfig(func(defaultBuckets dbutils.BucketsCfg) dbutils.BucketsCfg {
|
|
return dbutils.BucketsCfg{
|
|
dbutils.PlainStateBucket: dbutils.BucketsConfigs[dbutils.PlainStateBucket],
|
|
dbutils.PlainContractCodeBucket: dbutils.BucketsConfigs[dbutils.PlainContractCodeBucket],
|
|
dbutils.CodeBucket: dbutils.BucketsConfigs[dbutils.CodeBucket],
|
|
}
|
|
}).Path(snapshotPath).Readonly().MustOpen()
|
|
tmpDB = ethdb.NewMDBX().Path(tmpPath).MustOpen()
|
|
}
|
|
defer os.RemoveAll(tmpPath)
|
|
defer tmpDB.Close()
|
|
snkv = ethdb.NewSnapshotKV().SnapshotDB([]string{dbutils.PlainStateBucket, dbutils.PlainContractCodeBucket, dbutils.CodeBucket}, snkv).DB(tmpDB).Open()
|
|
tx, err := snkv.BeginRw(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer tx.Rollback()
|
|
hash, err := rawdb.ReadCanonicalHash(tx, block)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
syncHeadHeader := rawdb.ReadHeader(tx, hash, block)
|
|
if syncHeadHeader == nil {
|
|
return fmt.Errorf("empty header")
|
|
}
|
|
expectedRootHash := syncHeadHeader.Root
|
|
tt := time.Now()
|
|
err = stagedsync.PromoteHashedStateCleanly("", tx, stagedsync.StageHashStateCfg(snkv, os.TempDir()), ctx.Done())
|
|
fmt.Println("Promote took", time.Since(tt))
|
|
if err != nil {
|
|
return fmt.Errorf("promote state err: %w", err)
|
|
}
|
|
|
|
_, err = stagedsync.RegenerateIntermediateHashes("", tx, stagedsync.StageTrieCfg(snkv, true, true, os.TempDir()), expectedRootHash, ctx.Done())
|
|
if err != nil {
|
|
return fmt.Errorf("regenerateIntermediateHashes err: %w", err)
|
|
}
|
|
return nil
|
|
}
|