From 816d3020fa442107e7d83fcc893b18d507518512 Mon Sep 17 00:00:00 2001
From: Alex <93446519+AlexOn1ine@users.noreply.github.com>
Date: Wed, 25 Dec 2024 22:42:14 +0100
Subject: [PATCH] Fixes UB in caps.c (#5878)

---
 src/caps.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/caps.c b/src/caps.c
index 9c30e55527..941509c2a4 100644
--- a/src/caps.c
+++ b/src/caps.c
@@ -54,7 +54,7 @@ u32 GetSoftLevelCapExpValue(u32 level, u32 expValue)
         if (B_LEVEL_CAP_EXP_UP)
         {
             levelDifference = currentLevelCap - level;
-            if (levelDifference > ARRAY_COUNT(sExpScalingUp))
+            if (levelDifference > ARRAY_COUNT(sExpScalingUp) - 1)
                 return expValue + (expValue / sExpScalingUp[ARRAY_COUNT(sExpScalingUp) - 1]);
             else
                 return expValue + (expValue / sExpScalingUp[levelDifference]);
@@ -71,7 +71,7 @@ u32 GetSoftLevelCapExpValue(u32 level, u32 expValue)
     else if (B_EXP_CAP_TYPE == EXP_CAP_SOFT)
     {
         levelDifference = level - currentLevelCap;
-        if (levelDifference > ARRAY_COUNT(sExpScalingDown))
+        if (levelDifference > ARRAY_COUNT(sExpScalingDown) - 1)
             return expValue / sExpScalingDown[ARRAY_COUNT(sExpScalingDown) - 1];
         else
             return expValue / sExpScalingDown[levelDifference];