drs version moved to bds

app/apptesting/test_suite.go

@@ -122,7 +122,7 @@ func (s *KeeperTestHelper) PostStateUpdateWithDRSVersion(ctx sdk.Context, rollap
 	var bds rollapptypes.BlockDescriptors
 	bds.BD = make([]rollapptypes.BlockDescriptor, numOfBlocks)
 	for k := uint64(0); k < numOfBlocks; k++ {
-		bds.BD[k] = rollapptypes.BlockDescriptor{Height: startHeight + k, Timestamp: time.Now().UTC()}
+		bds.BD[k] = rollapptypes.BlockDescriptor{Height: startHeight + k, Timestamp: time.Now().UTC(), DrsVersion: drsVersion}
 	}
 
 	updateState := rollapptypes.MsgUpdateState{
@@ -133,7 +133,6 @@ func (s *KeeperTestHelper) PostStateUpdateWithDRSVersion(ctx sdk.Context, rollap
 		DAPath:      "",
 		BDs:         bds,
 		Last:        false,
-		DrsVersion:  drsVersion,
 	}
 	msgServer := rollappkeeper.NewMsgServerImpl(*s.App.RollappKeeper)
 	_, err = msgServer.UpdateState(ctx, &updateState)

proto/dymensionxyz/dymension/rollapp/block_descriptor.proto

@@ -15,6 +15,8 @@ message BlockDescriptor {
   bytes stateRoot = 2; 
   // timestamp is the time from the block header
   google.protobuf.Timestamp timestamp = 3 [(gogoproto.nullable) = false, (gogoproto.stdtime) = true];
+  // DrsVersion is a DRS version used by the rollapp.
+  string drs_version = 4;
 }
 
 // BlockDescriptors defines list of BlockDescriptor.

proto/dymensionxyz/dymension/rollapp/state_info.proto

@@ -53,8 +53,6 @@ message StateInfo {
         (gogoproto.nullable) = false,
         (gogoproto.moretags) = "yaml:\"created_at\""
     ];
-    // DrsVersion is a DRS version used by the rollapp.
-    string drs_version = 11;
 }
 
 // StateInfoSummary is a compact representation of StateInfo

proto/dymensionxyz/dymension/rollapp/tx.proto

@@ -89,8 +89,6 @@ message MsgUpdateState {
   BlockDescriptors BDs = 7 [(gogoproto.nullable) = false];
   // last is true if this is the last batch of the sequencer
   bool last = 8;
-  // DrsVersion is a DRS version used by the rollapp.
-  string drs_version = 9;
 }
 
 message MsgUpdateStateResponse {

x/rollapp/keeper/msg_server_mark_vulnerable_rollapps.go

@@ -62,19 +62,22 @@ func (k Keeper) MarkVulnerableRollapps(ctx sdk.Context, drsVersions []string) (i
 			logger.With("rollapp_id", rollapp.RollappId).Info("no latest state info for rollapp")
 			continue
 		}
-		// TODO: this check may be deleted once empty DRS version is marked vulnerable
-		//  https://github.com/dymensionxyz/dymension/issues/1233
-		if info.DrsVersion == "" {
-			logger.With("rollapp_id", rollapp.RollappId).Info("no DRS version set for rollapp")
-		}
 
-		_, vulnerable := vulnerableVersions[info.DrsVersion]
-		if vulnerable {
-			err := k.MarkRollappAsVulnerable(ctx, rollapp.RollappId)
-			if err != nil {
-				return 0, fmt.Errorf("freeze rollapp: %w", err)
+		for _, bd := range info.BDs.BD {
+			// TODO: this check may be deleted once empty DRS version is marked vulnerable
+			//  https://github.com/dymensionxyz/dymension/issues/1233
+			if bd.DrsVersion == "" {
+				logger.With("rollapp_id", rollapp.RollappId).Info("no DRS version set for rollapp")
+			}
+
+			_, vulnerable := vulnerableVersions[bd.DrsVersion]
+			if vulnerable {
+				err := k.MarkRollappAsVulnerable(ctx, rollapp.RollappId)
+				if err != nil {
+					return 0, fmt.Errorf("freeze rollapp: %w", err)
+				}
+				vulnerableNum++
 			}
-			vulnerableNum++
 		}
 	}
 

x/rollapp/keeper/msg_server_update_state.go

@@ -34,17 +34,19 @@ func (k msgServer) UpdateState(goCtx context.Context, msg *types.MsgUpdateState)
 		return nil, errorsmod.Wrap(err, "before update state")
 	}
 
-	// verify the DRS version is not vulnerable
-	if k.IsDRSVersionVulnerable(ctx, msg.DrsVersion) {
-		// the rollapp is not marked as vulnerable yet, mark it now
-		err := k.MarkRollappAsVulnerable(ctx, msg.RollappId)
-		if err != nil {
-			return nil, fmt.Errorf("mark rollapp vulnerable: %w", err)
+	for _, bd := range msg.BDs.BD {
+		// verify the DRS version is not vulnerable
+		if k.IsDRSVersionVulnerable(ctx, bd.DrsVersion) {
+			// the rollapp is not marked as vulnerable yet, mark it now
+			err := k.MarkRollappAsVulnerable(ctx, msg.RollappId)
+			if err != nil {
+				return nil, fmt.Errorf("mark rollapp vulnerable: %w", err)
+			}
+			k.Logger(ctx).With("rollapp_id", msg.RollappId, "drs_version", bd.DrsVersion).
+				Info("non-frozen rollapp tried to submit MsgUpdateState with the vulnerable DRS version, mark the rollapp as vulnerable")
+			// we must return non-error if we want the changes to be saved
+			return &types.MsgUpdateStateResponse{}, nil
 		}
-		k.Logger(ctx).With("rollapp_id", msg.RollappId, "drs_version", msg.DrsVersion).
-			Info("non-frozen rollapp tried to submit MsgUpdateState with the vulnerable DRS version, mark the rollapp as vulnerable")
-		// we must return non-error if we want the changes to be saved
-		return &types.MsgUpdateStateResponse{}, nil
 	}
 
 	// retrieve last updating index
@@ -107,7 +109,6 @@ func (k msgServer) UpdateState(goCtx context.Context, msg *types.MsgUpdateState)
 		creationHeight,
 		msg.BDs,
 		blockTime,
-		msg.DrsVersion,
 	)
 	// Write new state information to the store indexed by <RollappId,LatestStateInfoIndex>
 	k.SetStateInfo(ctx, *stateInfo)

x/rollapp/types/state_info.go

@@ -19,7 +19,6 @@ func NewStateInfo(
 	height uint64,
 	BDs BlockDescriptors,
 	createdAt time.Time,
-	drsVersion string,
 ) *StateInfo {
 	stateInfoIndex := StateInfoIndex{RollappId: rollappId, Index: newIndex}
 	status := common.Status_PENDING
@@ -33,7 +32,6 @@ func NewStateInfo(
 		Status:         status,
 		BDs:            BDs,
 		CreatedAt:      createdAt,
-		DrsVersion:     drsVersion,
 	}
 }