added EnhancedSetupFlowTCRevision check

Signed-off-by: DenisRybas <denis.rybas@dsr-corporation.com>

Author: DenisRybas

x/model/keeper/msg_server_model.go

@@ -151,10 +151,16 @@ func (k msgServer) UpdateModel(goCtx context.Context, msg *types.MsgUpdateModel)
 	model.EnhancedSetupFlowOptions = msg.EnhancedSetupFlowOptions
 	if msg.EnhancedSetupFlowOptions&1 == 1 {
 		model.EnhancedSetupFlowTCUrl = msg.EnhancedSetupFlowTCUrl
-		model.EnhancedSetupFlowTCRevision = msg.EnhancedSetupFlowTCRevision
 		model.EnhancedSetupFlowTCDigest = msg.EnhancedSetupFlowTCDigest
 		model.EnhancedSetupFlowTCFileSize = msg.EnhancedSetupFlowTCFileSize
 		model.MaintenanceUrl = msg.MaintenanceUrl
+
+		if model.EnhancedSetupFlowTCRevision == 0 ||
+			msg.EnhancedSetupFlowTCRevision == model.EnhancedSetupFlowTCRevision+1 {
+			model.EnhancedSetupFlowTCRevision = msg.EnhancedSetupFlowTCRevision
+		} else {
+			return nil, types.NewErrEnhancedSetupFlowTCRevisionInvalidIncrement(msg.EnhancedSetupFlowTCRevision, model.EnhancedSetupFlowTCRevision)
+		}
 	}
 
 	model.SchemaVersion = msg.SchemaVersion

x/model/types/errors.go

@@ -142,3 +142,8 @@ func NewErrEnhancedSetupFlowTCDigestIsNotBase64Encoded(enhancedSetupFlowTCDigest
 	return errors.Wrapf(ErrFieldIsNotBase64Encoded,
 		"EnhancedSetupFlowTCDigest %v is not base64 encoded", enhancedSetupFlowTCDigest)
 }
+
+func NewErrEnhancedSetupFlowTCRevisionInvalidIncrement(newEnhancedSetupFlowTCRevision int32, prevEnhancedSetupFlowTCRevision int32) error {
+	return errors.Wrapf(ErrFieldIsNotBase64Encoded,
+		"EnhancedSetupFlowTCRevision %v is not correctly incremented to %v", prevEnhancedSetupFlowTCRevision, newEnhancedSetupFlowTCRevision)
+}