[staking] reject endorse action before HF activated (#4187)

iotexproject · Mar 14, 2024 · 855d8ba · 855d8ba
1 parent a5b7ef8
commit 855d8ba
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/action/protocol/staking/protocol.go b/action/protocol/staking/protocol.go
@@ -469,6 +469,10 @@ func (p *Protocol) Validate(ctx context.Context, act action.Action, sr protocol.
 		return p.validateCandidateRegister(ctx, act)
 	case *action.CandidateUpdate:
 		return p.validateCandidateUpdate(ctx, act)
+	case *action.CandidateActivate:
+		return p.validateCandidateActivate(ctx, act)
+	case *action.CandidateEndorsement:
+		return p.validateCandidateEndorsement(ctx, act)
 	}
 	return nil
 }

diff --git a/action/protocol/staking/validations.go b/action/protocol/staking/validations.go
@@ -84,3 +84,17 @@ func (p *Protocol) validateCandidateUpdate(ctx context.Context, act *action.Cand
 	}
 	return nil
 }
+
+func (p *Protocol) validateCandidateEndorsement(ctx context.Context, act *action.CandidateEndorsement) error {
+	if protocol.MustGetFeatureCtx(ctx).DisableDelegateEndorsement {
+		return errors.Wrap(action.ErrInvalidAct, "candidate endorsement is disabled")
+	}
+	return nil
+}
+
+func (p *Protocol) validateCandidateActivate(ctx context.Context, act *action.CandidateActivate) error {
+	if protocol.MustGetFeatureCtx(ctx).DisableDelegateEndorsement {
+		return errors.Wrap(action.ErrInvalidAct, "candidate activate is disabled")
+	}
+	return nil
+}