Add an option to start re validation from a specified block and end at a specified block

staker/block_validator.go

@@ -172,7 +172,13 @@ func (c *BlockValidatorConfig) Validate() error {
 }
 
 type BlockValidatorDangerousConfig struct {
-	ResetBlockValidation bool `koanf:"reset-block-validation"`
+	ResetBlockValidation bool               `koanf:"reset-block-validation"`
+	Revalidation         RevalidationConfig `koanf:"re-validation"`
+}
+
+type RevalidationConfig struct {
+	Enable     bool   `koanf:"enable"`
+	StartBlock uint64 `koanf:"start-block"`
 }
 
 type BlockValidatorConfigFetcher func() *BlockValidatorConfig
@@ -197,6 +203,12 @@ func BlockValidatorConfigAddOptions(prefix string, f *pflag.FlagSet) {
 
 func BlockValidatorDangerousConfigAddOptions(prefix string, f *pflag.FlagSet) {
 	f.Bool(prefix+".reset-block-validation", DefaultBlockValidatorDangerousConfig.ResetBlockValidation, "resets block-by-block validation, starting again at genesis")
+	RevalidationConfigAddOptions(prefix+".re-validation", f)
+}
+
+func RevalidationConfigAddOptions(prefix string, f *pflag.FlagSet) {
+	f.Bool(prefix+".enable", DefaultRevalidationConfig.Enable, "enable re-validation")
+	f.Uint64(prefix+".start-block", DefaultBlockValidatorDangerousConfig.Revalidation.StartBlock, "start re-validation from this block")
 }
 
 var DefaultBlockValidatorConfig = BlockValidatorConfig{
@@ -237,6 +249,12 @@ var TestBlockValidatorConfig = BlockValidatorConfig{
 
 var DefaultBlockValidatorDangerousConfig = BlockValidatorDangerousConfig{
 	ResetBlockValidation: false,
+	Revalidation:         DefaultRevalidationConfig,
+}
+
+var DefaultRevalidationConfig = RevalidationConfig{
+	Enable:     false,
+	StartBlock: 0,
 }
 
 type valStatusField uint32
@@ -325,6 +343,27 @@ func NewBlockValidator(
 			PosInBatch: 0,
 		}
 	}
+	if config().Dangerous.Revalidation.Enable && config().Dangerous.Revalidation.StartBlock > 0 {
+		startBlock := config().Dangerous.Revalidation.StartBlock
+		messageCount, err := inbox.GetBatchMessageCount(startBlock - 1)
+		if err != nil {
+			return nil, err
+		}
+		res, err := streamer.ResultAtCount(messageCount)
+		if err != nil {
+			return nil, err
+		}
+		gs := validator.GoGlobalState{
+			BlockHash:  res.BlockHash,
+			SendRoot:   res.SendRoot,
+			Batch:      startBlock,
+			PosInBatch: 0,
+		}
+		err = ret.writeLastValidated(gs, nil)
+		if err != nil {
+			return nil, err
+		}
+	}
 	streamer.SetBlockValidator(ret)
 	inbox.SetBlockValidator(ret)
 	if config().MemoryFreeLimit != "" {