From 81739bc6c15293a280d99a2e15582b69bc7f0640 Mon Sep 17 00:00:00 2001
From: Jade <jadeshrinemaiden@gmail.com>
Date: Thu, 31 Dec 2020 05:54:57 +0000
Subject: [PATCH] Basic Anti rapidfire

---
 src/ItemManager.cpp |  2 ++
 src/ItemManager.hpp |  2 +-
 src/MobManager.cpp  | 31 +++++++++++++++++++++++++++++++
 src/Player.hpp      |  4 ++++
 src/TableData.cpp   |  1 +
 5 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/src/ItemManager.cpp b/src/ItemManager.cpp
index 27f7f43..b4328cb 100644
--- a/src/ItemManager.cpp
+++ b/src/ItemManager.cpp
@@ -1074,6 +1074,7 @@ void ItemManager::setItemStats(Player* plr) {
 
     plr->pointDamage = 8 + plr->level * 2;
     plr->groupDamage = 8 + plr->level * 2;
+    plr->fireRate = 0;
     plr->defense = 16 + plr->level * 4;
 
     Item* itemStatsDat;
@@ -1086,6 +1087,7 @@ void ItemManager::setItemStats(Player* plr) {
         }
         plr->pointDamage += itemStatsDat->pointDamage;
         plr->groupDamage += itemStatsDat->groupDamage;
+        plr->fireRate += itemStatsDat->fireRate;
         plr->defense += itemStatsDat->defense;
     }
 }
diff --git a/src/ItemManager.hpp b/src/ItemManager.hpp
index a5a4588..d726a2e 100644
--- a/src/ItemManager.hpp
+++ b/src/ItemManager.hpp
@@ -23,7 +23,7 @@ namespace ItemManager {
     };
     struct Item {
         bool tradeable, sellable;
-        int buyPrice, sellPrice, stackSize, level, rarity, pointDamage, groupDamage, defense, gender; // TODO: implement more as needed
+        int buyPrice, sellPrice, stackSize, level, rarity, pointDamage, groupDamage, fireRate, defense, gender; // TODO: implement more as needed
     };
     // hopefully this is fine since it's never modified after load
     extern std::map<std::pair<int32_t, int32_t>, Item> ItemData; // <id, type> -> data
diff --git a/src/MobManager.cpp b/src/MobManager.cpp
index 50871ae..0302bc4 100644
--- a/src/MobManager.cpp
+++ b/src/MobManager.cpp
@@ -62,6 +62,23 @@ void MobManager::pcAttackNpcs(CNSocket *sock, CNPacketData *data) {
         return;
     }
 
+    // rapid fire anti-cheat
+    time_t currTime = getTime();
+    if (currTime - plr->lastShot < plr->fireRate * 80) {
+        plr->suspicionRating += plr->fireRate * 100 + plr->lastShot - currTime; // gain suspicion for rapid firing
+        std::cout << "suspicion rating: " << plr->suspicionRating << std::endl;
+    } else if (currTime - plr->lastShot < plr->fireRate * 180 && plr->suspicionRating > 0) {
+        plr->suspicionRating += plr->fireRate * 100 + plr->lastShot - currTime; // lose suspicion for delayed firing
+        std::cout << "suspicion rating: " << plr->suspicionRating << std::endl;
+    }
+    plr->lastShot = currTime;
+
+    if (pkt->iNPCCnt > 3) // 3+ targets should never be possible
+        plr->suspicionRating += 10000;
+
+    if (plr->suspicionRating > 10000) // kill the socket when the player is too suspicious
+        sock->kill();
+
     // initialize response struct
     size_t resplen = sizeof(sP_FE2CL_PC_ATTACK_NPCs_SUCC) + pkt->iNPCCnt * sizeof(sAttackResult);
     uint8_t respbuf[CN_PACKET_BUFFER_SIZE];
@@ -1385,6 +1402,20 @@ void MobManager::projectileHit(CNSocket* sock, CNPacketData* data) {
         return;
     }
 
+    // rapid fire anti-cheat
+    time_t currTime = getTime();
+    if (currTime - plr->lastShot < plr->fireRate * 80) {
+        plr->suspicionRating += plr->fireRate * 100 + plr->lastShot - currTime; // gain suspicion for rapid firing
+        std::cout << "suspicion rating: " << plr->suspicionRating << std::endl;
+    } else if (currTime - plr->lastShot < plr->fireRate * 180 && plr->suspicionRating > 0) {
+        plr->suspicionRating += plr->fireRate * 100 + plr->lastShot - currTime; // lose suspicion for delayed firing
+        std::cout << "suspicion rating: " << plr->suspicionRating << std::endl;
+    }
+    plr->lastShot = currTime;
+
+    if (plr->suspicionRating > 10000) // kill the socket when the player is too suspicious
+        sock->kill();
+
     /*
      * initialize response struct
      * rocket style hit doesn't work properly, so we're always sending this one
diff --git a/src/Player.hpp b/src/Player.hpp
index c4dc729..2d7221e 100644
--- a/src/Player.hpp
+++ b/src/Player.hpp
@@ -57,6 +57,7 @@ struct Player {
 
     int pointDamage;
     int groupDamage;
+    int fireRate;
     int defense;
 
     int64_t aQuestFlag[16];
@@ -85,4 +86,7 @@ struct Player {
     ChunkPos chunkPos;
     std::set<Chunk*>* viewableChunks;
     time_t lastHeartbeat;
+
+    int suspicionRating;
+    time_t lastShot;
 };
diff --git a/src/TableData.cpp b/src/TableData.cpp
index f517dd2..b96aad4 100644
--- a/src/TableData.cpp
+++ b/src/TableData.cpp
@@ -157,6 +157,7 @@ void TableData::init() {
                     itemData.level = item["m_iMinReqLev"];
                     itemData.pointDamage = item["m_iPointRat"];
                     itemData.groupDamage = item["m_iGroupRat"];
+                    itemData.fireRate = item["m_iDelayTime"];
                     itemData.defense = item["m_iDefenseRat"];
                     itemData.gender = item["m_iReqSex"];
                 } else {