From 26460c016793da7d0bf25de20e1885b29c2c7e5f Mon Sep 17 00:00:00 2001
From: Gent <gentsemaj@live.com>
Date: Sat, 7 Nov 2020 21:51:16 -0500
Subject: [PATCH] Add battery caps for combat rewards & commands

---
 src/MobManager.cpp    | 7 ++++++-
 src/PlayerManager.cpp | 6 ++++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/MobManager.cpp b/src/MobManager.cpp
index 0283712..a8decba 100644
--- a/src/MobManager.cpp
+++ b/src/MobManager.cpp
@@ -191,6 +191,11 @@ void MobManager::giveReward(CNSocket *sock, Mob* mob) {
         if (rand() % 3 == 0)
             plr->batteryW += drop.boosts;
     }
+    // caps
+    if (plr->batteryW > 9999)
+        plr->batteryW = 9999;
+    if (plr->batteryN > 9999)
+        plr->batteryN = 9999;
 
     // simple rewards
     reward->m_iCandy = plr->money;
@@ -1154,4 +1159,4 @@ void MobManager::clearDebuff(Mob *mob) {
     pkt1.iID = mob->appearanceData.iNPC_ID;
     pkt1.iConditionBitFlag = mob->appearanceData.iConditionBitFlag;
     NPCManager::sendToViewable(mob, &pkt1, P_FE2CL_CHAR_TIME_BUFF_TIME_OUT, sizeof(sP_FE2CL_CHAR_TIME_BUFF_TIME_OUT));
-}
\ No newline at end of file
+}
diff --git a/src/PlayerManager.cpp b/src/PlayerManager.cpp
index 90c0d7a..7b9f889 100644
--- a/src/PlayerManager.cpp
+++ b/src/PlayerManager.cpp
@@ -746,9 +746,15 @@ void PlayerManager::setSpecialPlayer(CNSocket* sock, CNPacketData* data) {
         break;
     case 2:
         plr->batteryW = setData->iSetValue;
+        // caps
+        if (plr->batteryW > 9999)
+            plr->batteryW = 9999;
         break;
     case 3:
         plr->batteryN = setData->iSetValue;
+        // caps
+        if (plr->batteryN > 9999)
+            plr->batteryN = 9999;
         break;
     case 4:
         plr->fusionmatter = setData->iSetValue;