mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-01 00:31:21 +00:00
85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/ledgerwatch/erigon-lib/kv"
|
|
"github.com/ledgerwatch/erigon-lib/kv/mdbx"
|
|
"github.com/ledgerwatch/erigon/core/rawdb"
|
|
"github.com/ledgerwatch/erigon/eth/stagedsync"
|
|
"github.com/ledgerwatch/erigon/ethdb/snapshotdb"
|
|
"github.com/ledgerwatch/log/v3"
|
|
"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 {
|
|
logger := log.New()
|
|
return VerifyStateSnapshot(cmd.Context(), logger, chaindata, snapshotFile, block)
|
|
},
|
|
}
|
|
|
|
func VerifyStateSnapshot(ctx context.Context, logger log.Logger, dbPath, snapshotPath string, block uint64) error {
|
|
var snkv, tmpDB kv.RwDB
|
|
tmpPath, err := ioutil.TempDir(os.TempDir(), "vrf*")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
snkv = mdbx.NewMDBX(logger).WithTablessCfg(func(defaultBuckets kv.TableCfg) kv.TableCfg {
|
|
return kv.TableCfg{
|
|
kv.PlainState: kv.ChaindataTablesCfg[kv.PlainState],
|
|
kv.PlainContractCode: kv.ChaindataTablesCfg[kv.PlainContractCode],
|
|
kv.Code: kv.ChaindataTablesCfg[kv.Code],
|
|
}
|
|
}).Path(snapshotPath).Readonly().MustOpen()
|
|
tmpDB = mdbx.NewMDBX(logger).Path(tmpPath).MustOpen()
|
|
|
|
defer os.RemoveAll(tmpPath)
|
|
defer tmpDB.Close()
|
|
snkv = snapshotdb.NewSnapshotKV().StateSnapshot(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
|
|
}
|