Basic guards on changing jobs/getting quest rewards with a full box.

Author: RyanYappert

Arrowgene.Ddon.GameServer/Characters/JobManager.cs

@@ -31,11 +31,26 @@ public JobManager(DdonGameServer server)
             _Server = server;
         }
 
-        public (IPacketStructure jobRes, IPacketStructure itemNtc, IPacketStructure jobNtc) SetJob(GameClient client, CharacterCommon common, JobId jobId, DbConnection? connectionIn = null)
+        public (IPacketStructure? jobRes, IPacketStructure? itemNtc, IPacketStructure? jobNtc) SetJob(GameClient client, CharacterCommon common, JobId jobId, DbConnection? connectionIn = null)
         {
             // TODO: Reject job change if there's no primary and secondary weapon for the new job in storage
             // (or give a lvl 1 weapon for free?)
 
+            var totalSlots = common.Equipment.GetItems(EquipType.Performance)
+                .Concat(common.Equipment.GetItems(EquipType.Visual))
+                .Where(x => x != null)
+                .ToList()
+                .Count;
+            if (totalSlots > client.Character.Storage.GetStorage(StorageType.StorageBoxNormal).EmptySlots())
+            {
+                return (new S2CJobChangeJobRes()
+                {
+                    Error = (uint)ErrorCode.ERROR_CODE_JOBCHANGE_ITEM_CAPACITY_OVER
+                }, 
+                null,
+                null);
+            }
+
             JobId oldJobId = common.Job;
             common.Job = jobId;
 

Arrowgene.Ddon.GameServer/Handler/JobChangeJobHandler.cs

@@ -1,15 +1,11 @@
+using Arrowgene.Ddon.GameServer.Characters;
 using Arrowgene.Ddon.Server;
-using Arrowgene.Ddon.Server.Network;
 using Arrowgene.Ddon.Shared.Entity.PacketStructure;
 using Arrowgene.Logging;
-using Arrowgene.Ddon.Shared.Network;
-using Arrowgene.Ddon.GameServer.Characters;
-using Arrowgene.Ddon.Shared.Entity.Structure;
-using Arrowgene.Ddon.Shared.Entity;
 
 namespace Arrowgene.Ddon.GameServer.Handler
 {
-    public class JobChangeJobHandler : StructurePacketHandler<GameClient, C2SJobChangeJobReq>
+    public class JobChangeJobHandler : GameRequestPacketHandler<C2SJobChangeJobReq, S2CJobChangeJobRes>
     {
         private static readonly ServerLogger Logger = LogProvider.Logger<ServerLogger>(typeof(JobChangeJobHandler));
 
@@ -20,22 +16,30 @@ public JobChangeJobHandler(DdonGameServer server) : base(server)
             jobManager = server.JobManager;
         }
 
-        public override void Handle(GameClient client, StructurePacket<C2SJobChangeJobReq> packet)
+        public override S2CJobChangeJobRes Handle(GameClient client, C2SJobChangeJobReq request)
         {
             (S2CJobChangeJobRes jobRes, S2CItemUpdateCharacterItemNtc itemNtc, S2CJobChangeJobNtc jobNtc) jobResult = (null, null, null);
 
             Server.Database.ExecuteInTransaction(connection =>
             {
                 jobResult = ((S2CJobChangeJobRes, S2CItemUpdateCharacterItemNtc, S2CJobChangeJobNtc))
-                jobManager.SetJob(client, client.Character, packet.Structure.JobId, connection);
+                jobManager.SetJob(client, client.Character, request.JobId, connection);
             });
 
-            foreach (GameClient otherClient in Server.ClientLookup.GetAll())
+            if (jobResult.jobNtc != null)
             {
-                otherClient.Send(jobResult.jobNtc);
+                foreach (GameClient otherClient in Server.ClientLookup.GetAll())
+                {
+                    otherClient.Send(jobResult.jobNtc);
+                }
             }
-            client.Send(jobResult.itemNtc);
-            client.Send(jobResult.jobRes);
+            
+            if (jobResult.itemNtc != null)
+            {
+                client.Send(jobResult.itemNtc);
+            }
+
+            return jobResult.jobRes;
         }
     }
 }

Arrowgene.Ddon.GameServer/Handler/PawnCreatePawnHandler.cs

@@ -22,6 +22,12 @@ public PawnCreatePawnHandler(DdonGameServer server) : base(server)
 
         public override S2CPawnCreatePawnRes Handle(GameClient client, C2SPawnCreatePawnReq request)
         {
+            // I hate hardcoding this but people legitimately keep finding ways to break this. 
+            if (request.SlotNo > 3)
+            {
+                throw new ResponseErrorException(ErrorCode.ERROR_CODE_PAWN_CREATE_NUM_OVER);
+            }
+
             if (request.SlotNo == 1)
             {
                 const byte myPawnSlotNum = 2;
@@ -62,6 +68,17 @@ public override S2CPawnCreatePawnRes Handle(GameClient client, C2SPawnCreatePawn
                         Error = (uint)ErrorCode.ERROR_CODE_CHARACTER_ITEM_NOT_FOUND
                     };
                 }
+                else
+                {
+                    client.Send(new S2CItemUpdateCharacterItemNtc()
+                    {
+                        UpdateType = ItemNoticeType.CreatePawn,
+                        UpdateItemList = new()
+                        {
+                            result
+                        }
+                    });
+                }
             }           
 
             Pawn pawn = new Pawn(client.Character.CharacterId)

Arrowgene.Ddon.GameServer/Handler/QuestGetRewardBoxItemHandler.cs

@@ -62,6 +62,16 @@ public override S2CQuestGetRewardBoxItemRes Handle(GameClient client, C2SQuestGe
                 }
             }
 
+            var slotCount = coalescedRewards
+                .Where(x => !Server.ItemManager.IsItemWalletPoint(x.Value.ItemId) && x.Value.Num > 0)
+                .ToList()
+                .Count;
+
+            if (slotCount > client.Character.Storage.GetStorage(StorageType.StorageBoxNormal).EmptySlots())
+            {
+                throw new ResponseErrorException(ErrorCode.ERROR_CODE_ITEM_STORAGE_OVERFLOW);
+            }
+
             foreach (var rewardUID in packet.GetRewardBoxItemList.Select(x => x.UID).Distinct().ToList())
             {
                 var reward = coalescedRewards[rewardUID];