You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
419 lines
12 KiB
419 lines
12 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math/big"
|
|
"os"
|
|
"time"
|
|
|
|
lru "github.com/hashicorp/golang-lru"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/common/hexutil"
|
|
ethRawDB "github.com/ethereum/go-ethereum/core/rawdb"
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
"github.com/harmony-one/harmony/block"
|
|
"github.com/harmony-one/harmony/core/rawdb"
|
|
"github.com/harmony-one/harmony/core/state"
|
|
"github.com/harmony-one/harmony/core/types"
|
|
"github.com/harmony-one/harmony/hmy"
|
|
"github.com/harmony-one/harmony/internal/cli"
|
|
|
|
nodeconfig "github.com/harmony-one/harmony/internal/configs/node"
|
|
shardingconfig "github.com/harmony-one/harmony/internal/configs/sharding"
|
|
)
|
|
|
|
var snapdbInfo = rawdb.SnapdbInfo{}
|
|
|
|
var batchFlag = cli.IntFlag{
|
|
Name: "batch",
|
|
Shorthand: "b",
|
|
Usage: "batch size limit in MB",
|
|
DefValue: 512,
|
|
}
|
|
|
|
var dumpDBCmd = &cobra.Command{
|
|
Use: "dumpdb srcdb destdb",
|
|
Short: "dump a snapshot db.",
|
|
Long: "dump a snapshot db.",
|
|
Example: "harmony dumpdb /srcDir/harmony_db_0 /destDir/harmony_db_0",
|
|
Args: cobra.RangeArgs(2, 6),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
srcDBDir, destDBDir := args[0], args[1]
|
|
batchLimitMB := cli.GetIntFlagValue(cmd, batchFlag)
|
|
networkType := getNetworkType(cmd)
|
|
shardSchedule = getShardSchedule(networkType)
|
|
if shardSchedule == nil {
|
|
fmt.Println("unsupported network type")
|
|
os.Exit(-1)
|
|
}
|
|
snapdbInfo.NetworkType = networkType
|
|
fmt.Println(srcDBDir, destDBDir, batchLimitMB)
|
|
dumpMain(srcDBDir, destDBDir, batchLimitMB*MB)
|
|
os.Exit(0)
|
|
},
|
|
}
|
|
|
|
func getShardSchedule(networkType nodeconfig.NetworkType) shardingconfig.Schedule {
|
|
switch networkType {
|
|
case nodeconfig.Mainnet:
|
|
return shardingconfig.MainnetSchedule
|
|
case nodeconfig.Testnet:
|
|
return shardingconfig.TestnetSchedule
|
|
case nodeconfig.Pangaea:
|
|
return shardingconfig.PangaeaSchedule
|
|
case nodeconfig.Localnet:
|
|
return shardingconfig.LocalnetSchedule
|
|
case nodeconfig.Partner:
|
|
return shardingconfig.PartnerSchedule
|
|
case nodeconfig.Stressnet:
|
|
return shardingconfig.StressNetSchedule
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func registerDumpDBFlags() error {
|
|
return cli.RegisterFlags(dumpDBCmd, []cli.Flag{batchFlag, networkTypeFlag})
|
|
}
|
|
|
|
type KakashiDB struct {
|
|
ethdb.Database
|
|
toDB ethdb.Database
|
|
toDBBatch ethdb.Batch
|
|
batchLimit int
|
|
cache *lru.Cache
|
|
}
|
|
|
|
func (db *KakashiDB) GetCanonicalHash(number uint64) common.Hash {
|
|
return rawdb.ReadCanonicalHash(db, number)
|
|
}
|
|
|
|
func (db *KakashiDB) ChainDb() ethdb.Database {
|
|
return db
|
|
}
|
|
|
|
const (
|
|
MB = 1024 * 1024
|
|
BLOCKS_DUMP = 512 // must >= 256
|
|
EPOCHS_DUMP = 10
|
|
STATEDB_CACHE_SIZE = 64 // size in MB
|
|
LEVELDB_CACHE_SIZE = 256
|
|
LEVELDB_HANDLES = 1024
|
|
LRU_CACHE_SIZE = 64 * 1024 * 1024
|
|
)
|
|
|
|
const (
|
|
NONE = iota
|
|
ON_ACCOUNT_START
|
|
ON_ACCOUNT_STATE
|
|
ON_ACCOUNT_END
|
|
)
|
|
|
|
var (
|
|
printSize = uint64(0) // last print dump size
|
|
flushedSize = uint64(0) // size flushed into db
|
|
lastAccount = state.DumpAccount{
|
|
Address: &common.Address{},
|
|
}
|
|
accountState = NONE
|
|
emptyHash = common.Hash{}
|
|
shardSchedule shardingconfig.Schedule // init by cli flag
|
|
)
|
|
|
|
func dumpPrint(prefix string, showAccount bool) {
|
|
if snapdbInfo.DumpedSize-printSize > MB || showAccount {
|
|
now := time.Now().Unix()
|
|
fmt.Println(now, prefix, snapdbInfo.AccountCount, snapdbInfo.DumpedSize/MB, printSize/MB, flushedSize/MB)
|
|
if showAccount {
|
|
fmt.Println("account:", lastAccount.Address.Hex(), lastAccount.Balance, len(lastAccount.Code), accountState, lastAccount.SecureKey.String(), snapdbInfo.LastAccountStateKey.String())
|
|
}
|
|
printSize = snapdbInfo.DumpedSize
|
|
}
|
|
}
|
|
|
|
func (db *KakashiDB) Get(key []byte) ([]byte, error) {
|
|
value, err := db.Database.Get(key)
|
|
if exist, _ := db.cache.ContainsOrAdd(string(key), nil); !exist {
|
|
db.copyKV(key, value)
|
|
}
|
|
return value, err
|
|
}
|
|
func (db *KakashiDB) Put(key []byte, value []byte) error {
|
|
return nil
|
|
}
|
|
|
|
// Delete removes the key from the key-value data store.
|
|
func (db *KakashiDB) Delete(key []byte) error {
|
|
return nil
|
|
}
|
|
|
|
// copy key,value to toDB
|
|
func (db *KakashiDB) copyKV(key, value []byte) {
|
|
db.toDBBatch.Put(key, value)
|
|
snapdbInfo.DumpedSize += uint64(len(key) + len(value))
|
|
dumpPrint("copyKV", false)
|
|
}
|
|
|
|
func (db *KakashiDB) flush() {
|
|
dumpPrint("KakashiDB batch writhing", true)
|
|
if err := rawdb.WriteSnapdbInfo(db.toDBBatch, &snapdbInfo); err != nil {
|
|
panic(err)
|
|
}
|
|
db.toDBBatch.Write()
|
|
db.toDBBatch.Reset()
|
|
flushedSize = snapdbInfo.DumpedSize
|
|
}
|
|
|
|
func (db *KakashiDB) Close() error {
|
|
db.toDBBatch.Reset() // drop dirty cache
|
|
fmt.Println("KakashiDB Close")
|
|
db.toDB.Close()
|
|
return db.Database.Close()
|
|
}
|
|
|
|
func (db *KakashiDB) OnRoot(common.Hash) {}
|
|
|
|
// OnAccount implements DumpCollector interface
|
|
func (db *KakashiDB) OnAccountStart(addr common.Address, acc state.DumpAccount) {
|
|
accountState = ON_ACCOUNT_START
|
|
lastAccount = acc
|
|
lastAccount.Address = &addr
|
|
snapdbInfo.LastAccountKey = acc.SecureKey
|
|
}
|
|
|
|
// OnAccount implements DumpCollector interface
|
|
func (db *KakashiDB) OnAccountState(addr common.Address, StateSecureKey hexutil.Bytes, key, value []byte) {
|
|
accountState = ON_ACCOUNT_STATE
|
|
snapdbInfo.LastAccountStateKey = StateSecureKey
|
|
if snapdbInfo.DumpedSize-flushedSize > uint64(db.batchLimit) {
|
|
db.flush()
|
|
}
|
|
}
|
|
|
|
// OnAccount implements DumpCollector interface
|
|
func (db *KakashiDB) OnAccountEnd(addr common.Address, acc state.DumpAccount) {
|
|
snapdbInfo.AccountCount++
|
|
accountState = ON_ACCOUNT_END
|
|
if snapdbInfo.DumpedSize-flushedSize > uint64(db.batchLimit) {
|
|
db.flush()
|
|
}
|
|
}
|
|
|
|
func (db *KakashiDB) getHashByNumber(number uint64) common.Hash {
|
|
hash := rawdb.ReadCanonicalHash(db, number)
|
|
return hash
|
|
}
|
|
func (db *KakashiDB) GetHeaderByNumber(number uint64) *block.Header {
|
|
hash := db.getHashByNumber(number)
|
|
if hash == (common.Hash{}) {
|
|
return nil
|
|
}
|
|
return db.GetHeader(hash, number)
|
|
}
|
|
|
|
func (db *KakashiDB) GetHeader(hash common.Hash, number uint64) *block.Header {
|
|
header := rawdb.ReadHeader(db, hash, number)
|
|
return header
|
|
}
|
|
|
|
func (db *KakashiDB) GetHeaderByHash(hash common.Hash) *block.Header {
|
|
number := rawdb.ReadHeaderNumber(db, hash)
|
|
return rawdb.ReadHeader(db, hash, *number)
|
|
}
|
|
|
|
// GetBlock retrieves a block from the database by hash and number
|
|
func (db *KakashiDB) GetBlock(hash common.Hash, number uint64) *types.Block {
|
|
block := rawdb.ReadBlock(db, hash, number)
|
|
return block
|
|
}
|
|
|
|
// GetBlockNumber retrieves the block number belonging to the given hash
|
|
// from the database
|
|
func (db *KakashiDB) GetBlockNumber(hash common.Hash) *uint64 {
|
|
return rawdb.ReadHeaderNumber(db, hash)
|
|
}
|
|
|
|
// GetBlockByHash retrieves a block from the database by hash
|
|
func (db *KakashiDB) GetBlockByHash(hash common.Hash) *types.Block {
|
|
number := db.GetBlockNumber(hash)
|
|
return db.GetBlock(hash, *number)
|
|
}
|
|
|
|
// GetBlockByNumber retrieves a block from the database by number
|
|
func (db *KakashiDB) GetBlockByNumber(number uint64) *types.Block {
|
|
hash := rawdb.ReadCanonicalHash(db, number)
|
|
return db.GetBlock(hash, number)
|
|
}
|
|
|
|
func (db *KakashiDB) indexerDataDump(block *types.Block) {
|
|
fmt.Println("indexerDataDump:")
|
|
bloomIndexer := hmy.NewBloomIndexer(db, params.BloomBitsBlocks, params.BloomConfirms)
|
|
bloomIndexer.Close() // just stop event loop
|
|
section, blkno, blkhash := bloomIndexer.Sections()
|
|
bloomIndexer.AddCheckpoint(section-1, blkhash)
|
|
for i := blkno; i <= block.NumberU64(); i++ {
|
|
db.GetHeaderByNumber(i)
|
|
}
|
|
snapdbInfo.IndexerDataDumped = true
|
|
db.flush()
|
|
}
|
|
|
|
func (db *KakashiDB) offchainDataDump(block *types.Block) {
|
|
fmt.Println("offchainDataDump:")
|
|
rawdb.WriteHeadBlockHash(db.toDBBatch, block.Hash())
|
|
rawdb.WriteHeadHeaderHash(db.toDBBatch, block.Hash())
|
|
db.GetHeaderByNumber(0)
|
|
db.GetBlockByNumber(0)
|
|
db.GetHeaderByHash(block.Hash())
|
|
// EVM may access the last 256 block hash
|
|
for i := 0; i <= BLOCKS_DUMP; i++ {
|
|
if block.NumberU64() < uint64(i) {
|
|
break
|
|
}
|
|
latestNumber := block.NumberU64() - uint64(i)
|
|
latestBlock := db.GetBlockByNumber(latestNumber)
|
|
db.GetBlockByHash(latestBlock.Hash())
|
|
db.GetHeaderByHash(latestBlock.Hash())
|
|
db.GetBlockByHash(latestBlock.Hash())
|
|
rawdb.ReadBlockRewardAccumulator(db, latestNumber)
|
|
rawdb.ReadBlockCommitSig(db, latestNumber)
|
|
epoch := block.Epoch()
|
|
epochInstance := shardSchedule.InstanceForEpoch(epoch)
|
|
for shard := 0; shard < int(epochInstance.NumShards()); shard++ {
|
|
rawdb.ReadCrossLinkShardBlock(db, uint32(shard), latestNumber)
|
|
}
|
|
}
|
|
headEpoch := block.Epoch()
|
|
epochInstance := shardSchedule.InstanceForEpoch(headEpoch)
|
|
for shard := 0; shard < int(epochInstance.NumShards()); shard++ {
|
|
rawdb.ReadShardLastCrossLink(db, uint32(shard))
|
|
}
|
|
|
|
rawdb.IteratorValidatorStats(db, func(it ethdb.Iterator, addr common.Address) bool {
|
|
db.copyKV(it.Key(), it.Value())
|
|
return true
|
|
})
|
|
rawdb.ReadPendingCrossLinks(db)
|
|
|
|
rawdb.IteratorDelegatorDelegations(db, func(it ethdb.Iterator, delegator common.Address) bool {
|
|
db.copyKV(it.Key(), it.Value())
|
|
return true
|
|
})
|
|
for i := 0; i < EPOCHS_DUMP; i++ {
|
|
epoch := new(big.Int).Sub(headEpoch, big.NewInt(int64(i)))
|
|
if epoch.Sign() < 0 {
|
|
break
|
|
}
|
|
rawdb.ReadShardState(db, epoch)
|
|
rawdb.ReadEpochBlockNumber(db, epoch)
|
|
rawdb.ReadEpochVrfBlockNums(db, epoch)
|
|
rawdb.ReadEpochVdfBlockNum(db, epoch)
|
|
var validators []common.Address
|
|
rawdb.IteratorValidatorSnapshot(db, func(addr common.Address, _epoch *big.Int) bool {
|
|
if _epoch.Cmp(epoch) == 0 {
|
|
validator, err := rawdb.ReadValidatorSnapshot(db, addr, epoch)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
validators = append(validators, validator.Validator.Address)
|
|
}
|
|
return true
|
|
})
|
|
if i == 0 {
|
|
rawdb.ReadValidatorList(db)
|
|
}
|
|
}
|
|
|
|
rawdb.IteratorCXReceiptsProofSpent(db, func(it ethdb.Iterator, shardID uint32, number uint64) bool {
|
|
db.copyKV(it.Key(), it.Value())
|
|
return true
|
|
})
|
|
snapdbInfo.OffchainDataDumped = true
|
|
db.flush()
|
|
}
|
|
|
|
func (db *KakashiDB) stateDataDump(block *types.Block) {
|
|
fmt.Println("stateDataDump:", snapdbInfo.LastAccountKey.String(), snapdbInfo.LastAccountStateKey.String())
|
|
stateDB0 := state.NewDatabaseWithCache(db, STATEDB_CACHE_SIZE)
|
|
rootHash := block.Root()
|
|
stateDB, err := state.New(rootHash, stateDB0)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
config := new(state.DumpConfig)
|
|
config.Start = snapdbInfo.LastAccountKey
|
|
if len(snapdbInfo.LastAccountStateKey) > 0 {
|
|
stateKey := new(big.Int).SetBytes(snapdbInfo.LastAccountStateKey)
|
|
stateKey.Add(stateKey, big.NewInt(1))
|
|
config.StateStart = stateKey.Bytes()
|
|
if len(config.StateStart) != len(snapdbInfo.LastAccountStateKey) {
|
|
panic("statekey overflow")
|
|
}
|
|
}
|
|
stateDB.DumpToCollector(db, config)
|
|
snapdbInfo.StateDataDumped = true
|
|
db.flush()
|
|
}
|
|
|
|
func dumpMain(srcDBDir, destDBDir string, batchLimit int) {
|
|
fmt.Println("===dumpMain===")
|
|
srcDB, err := ethRawDB.NewLevelDBDatabase(srcDBDir, LEVELDB_CACHE_SIZE, LEVELDB_HANDLES, "")
|
|
if err != nil {
|
|
fmt.Println("open src db error:", err)
|
|
os.Exit(-1)
|
|
}
|
|
destDB, err := ethRawDB.NewLevelDBDatabase(destDBDir, LEVELDB_CACHE_SIZE, LEVELDB_HANDLES, "")
|
|
if err != nil {
|
|
fmt.Println("open dest db error:", err)
|
|
os.Exit(-1)
|
|
}
|
|
|
|
if lastSnapdbInfo := rawdb.ReadSnapdbInfo(destDB); lastSnapdbInfo != nil {
|
|
if lastSnapdbInfo.NetworkType != snapdbInfo.NetworkType {
|
|
fmt.Printf("different network type! last:%s cmd:%s\n", lastSnapdbInfo.NetworkType, snapdbInfo.NetworkType)
|
|
os.Exit(-1)
|
|
}
|
|
snapdbInfo = *lastSnapdbInfo
|
|
flushedSize = snapdbInfo.DumpedSize
|
|
printSize = flushedSize
|
|
}
|
|
headHash := rawdb.ReadHeadBlockHash(srcDB)
|
|
headNumber := rawdb.ReadHeaderNumber(srcDB, headHash)
|
|
block := rawdb.ReadBlock(srcDB, headHash, *headNumber)
|
|
if block == nil || block.Hash() == emptyHash {
|
|
fmt.Println("empty head block")
|
|
os.Exit(-1)
|
|
}
|
|
if snapdbInfo.BlockHeader == nil {
|
|
snapdbInfo.BlockHeader = block.Header()
|
|
}
|
|
if snapdbInfo.BlockHeader.Hash() != block.Hash() {
|
|
fmt.Printf("head block does not match! src: %s dest: %x\n", block.Hash().String(), snapdbInfo.BlockHeader.Hash().String())
|
|
os.Exit(-1)
|
|
}
|
|
|
|
fmt.Println("head-block:", block.Header().Number(), block.Hash().Hex())
|
|
fmt.Println("start copying...")
|
|
cache, _ := lru.New(LRU_CACHE_SIZE)
|
|
copier := &KakashiDB{
|
|
Database: srcDB,
|
|
toDB: destDB,
|
|
toDBBatch: destDB.NewBatch(),
|
|
batchLimit: batchLimit,
|
|
cache: cache,
|
|
}
|
|
defer copier.Close()
|
|
if !snapdbInfo.OffchainDataDumped {
|
|
copier.offchainDataDump(block)
|
|
}
|
|
if !snapdbInfo.IndexerDataDumped {
|
|
copier.indexerDataDump(block)
|
|
}
|
|
if !snapdbInfo.StateDataDumped {
|
|
copier.stateDataDump(block)
|
|
}
|
|
fmt.Println("Dump completed!")
|
|
}
|
|
|