historysegment: add testnet&mainnet hard code;

cmd/geth/chaincmd.go

@@ -726,29 +726,28 @@ func exportSegment(ctx *cli.Context) error {
 	}
 	latest := headerChain.CurrentHeader()
 	if _, ok := engine.(consensus.PoSA); !ok {
-		return errors.New("current chain is not POSA consensus, cannot generate history segment")
+		return errors.New("cannot generate history segment because consensus engine is not PoSA")
 	}
 	if !chainConfig.IsLuban(latest.Number) {
-		return errors.New("current chain is not enable Luban hard fork, cannot generate history segment")
+		return errors.New("luban hard fork is not enabled , cannot generate history segment")
 	}
 
 	var (
 		boundStartBlock      = params.BoundStartBlock
 		historySegmentLength = params.HistorySegmentLength
 	)
-	// TODO(0xbundler): for testing
-	//switch genesisHash {
-	//case params.BSCGenesisHash, params.ChapelGenesisHash, params.RialtoGenesisHash:
-	//	boundStartBlock = params.BoundStartBlock
-	//	historySegmentLength = params.HistorySegmentLength
-	//default:
-	if ctx.IsSet(utils.BoundStartBlockFlag.Name) {
-		boundStartBlock = ctx.Uint64(utils.BoundStartBlockFlag.Name)
-	}
-	if ctx.IsSet(utils.HistorySegmentLengthFlag.Name) {
-		historySegmentLength = ctx.Uint64(utils.HistorySegmentLengthFlag.Name)
-	}
-	//}
+	switch genesisHash {
+	case params.BSCGenesisHash, params.ChapelGenesisHash, params.RialtoGenesisHash:
+		boundStartBlock = params.BoundStartBlock
+		historySegmentLength = params.HistorySegmentLength
+	default:
+		if ctx.IsSet(utils.BoundStartBlockFlag.Name) {
+			boundStartBlock = ctx.Uint64(utils.BoundStartBlockFlag.Name)
+		}
+		if ctx.IsSet(utils.HistorySegmentLengthFlag.Name) {
+			historySegmentLength = ctx.Uint64(utils.HistorySegmentLengthFlag.Name)
+		}
+	}
 	if boundStartBlock == 0 || historySegmentLength == 0 {
 		return fmt.Errorf("wrong params, boundStartBlock: %v, historySegmentLength: %v", boundStartBlock, historySegmentLength)
 	}
@@ -812,7 +811,7 @@ func exportSegment(ctx *cli.Context) error {
 		}
 		segments = append(segments, segment)
 	}
-	if err = params.ValidateHisSegments(params.NewHistoryBlock(0, genesisHash, td.Uint64()), segments); err != nil {
+	if err = params.ValidateHistorySegments(params.NewHistoryBlock(0, genesisHash, td.Uint64()), segments); err != nil {
 		return err
 	}
 	output, err := json.MarshalIndent(segments, "", "    ")
@@ -903,9 +902,6 @@ func simpleHeaderChain(db ethdb.Database, genesisHash common.Hash) (*params.Chai
 	if err != nil {
 		return nil, nil, nil, err
 	}
-	if _, ok := engine.(consensus.PoSA); !ok {
-		return nil, nil, nil, errors.New("current chain is not POSA, cannot generate history segment")
-	}
 	headerChain, err := core.NewHeaderChain(db, chainConfig, engine, func() bool {
 		return true
 	})

cmd/geth/main.go

@@ -166,7 +166,7 @@ var (
 		utils.BLSPasswordFileFlag,
 		utils.BLSWalletDirFlag,
 		utils.VoteJournalDirFlag,
-		utils.HistorySegEnableFlag,
+		utils.HistorySegEnabledFlag,
 		utils.HistorySegCustomFlag,
 	}, utils.NetworkFlags, utils.DatabasePathFlags)
 

cmd/utils/flags.go

@@ -1109,7 +1109,7 @@ Please note that --` + MetricsHTTPFlag.Name + ` must be set to start the server.
 	}
 
 	// History segment
-	HistorySegEnableFlag = &cli.BoolFlag{
+	HistorySegEnabledFlag = &cli.BoolFlag{
 		Name:     "history-segment",
 		Usage:    "Enable history segment feature, it will auto prune history segments by hard-code segment hash",
 		Value:    false,
@@ -2152,8 +2152,8 @@ func SetEthConfig(ctx *cli.Context, stack *node.Node, cfg *ethconfig.Config) {
 	}
 
 	// parse History Segment flags
-	if ctx.IsSet(HistorySegEnableFlag.Name) {
-		cfg.HistorySegmentEnable = ctx.Bool(HistorySegEnableFlag.Name)
+	if ctx.IsSet(HistorySegEnabledFlag.Name) {
+		cfg.HistorySegmentEnabled = ctx.Bool(HistorySegEnabledFlag.Name)
 	}
 	if ctx.IsSet(HistorySegCustomFlag.Name) {
 		cfg.HistorySegmentCustomFile = ctx.String(HistorySegCustomFlag.Name)

core/rawdb/database.go

@@ -201,7 +201,7 @@ func (db *nofreezedb) AncientDatadir() (string, error) {
 }
 
 func (db *nofreezedb) AncientReset(tail, head uint64) error {
-	return errNotSupported
+	return nil
 }
 
 // NewDatabase creates a high level database on top of a given key-value data

core/rawdb/prunedfreezer.go

@@ -189,7 +189,7 @@ func (f *prunedfreezer) TruncateTail(tail uint64) (uint64, error) {
 }
 
 func (f *prunedfreezer) AncientReset(tail, head uint64) error {
-	return errNotSupported
+	return nil
 }
 
 // Sync flushes meta data tables to disk.

eth/backend.go

@@ -196,7 +196,7 @@ func New(stack *node.Node, config *ethconfig.Config) (*Ethereum, error) {
 		hsm         *params.HistorySegmentManager
 		lastSegment *params.HistorySegment
 	)
-	if config.HistorySegmentEnable {
+	if config.HistorySegmentEnabled {
 		hsm, lastSegment, err = GetHistorySegmentAndLastSegment(chainDb, genesisHash, config.HistorySegmentCustomFile)
 		if err != nil {
 			return nil, err
@@ -267,12 +267,12 @@ func New(stack *node.Node, config *ethconfig.Config) (*Ethereum, error) {
 	txLookupLimit := &config.TransactionHistory
 	// if enable HistorySegment, just skip txLookupLimit params,
 	// may cause regenerate tx index, but it will also generate new block index
-	if config.HistorySegmentEnable {
+	if config.HistorySegmentEnabled {
 		txLookupLimit = nil
 	}
 	bcOps = append(bcOps, func(bc *core.BlockChain) (*core.BlockChain, error) {
 		// if enable history segment, try prune ancient data when restart
-		if config.HistorySegmentEnable {
+		if config.HistorySegmentEnabled {
 			if err = truncateAncientTail(chainDb, lastSegment); err != nil {
 				return nil, err
 			}

eth/ethconfig/config.go

@@ -199,7 +199,7 @@ type Config struct {
 	OverrideVerkle *uint64 `toml:",omitempty"`
 
 	// History Segment
-	HistorySegmentEnable     bool   `toml:",omitempty"`
+	HistorySegmentEnabled    bool   `toml:",omitempty"`
 	HistorySegmentCustomFile string `toml:",omitempty"`
 }