From 17d4e4d35ae71b85127afb29a3661c454ce8f638 Mon Sep 17 00:00:00 2001
From: Federico Cecchetto <fedecek3@gmail.com>
Date: Fri, 24 Jun 2022 22:38:18 +0200
Subject: [PATCH] Add entity overflow patch

---
 src/client/component/security.cpp | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/src/client/component/security.cpp b/src/client/component/security.cpp
index 12285f85..f1300cc6 100644
--- a/src/client/component/security.cpp
+++ b/src/client/component/security.cpp
@@ -16,6 +16,29 @@ namespace security
 				utils::hook::invoke<void>(0x61A9D0_b, localclient, index1, index2);
 			}
 		}
+
+		void remap_cached_entities(game::mp::cachedSnapshot_t& snapshot)
+		{
+			static bool printed = false;
+			if (snapshot.num_clients > 1200 && !printed)
+			{
+				printed = true;
+				printf("Too many entities (%d)... remapping!\n", snapshot.num_clients);
+			}
+
+			snapshot.num_clients = std::min(snapshot.num_clients, 1200);
+		}
+
+		void remap_cached_entities_stub(utils::hook::assembler& a)
+		{
+			a.pushad64();
+
+			a.mov(rcx, rbx);
+			a.call_aligned(remap_cached_entities);
+
+			a.popad64();
+			a.jmp(0x55E4D8_b);
+		}
 	}
 
 	class component final : public component_interface
@@ -30,6 +53,9 @@ namespace security
 
 			// Patch vulnerability in PlayerCards_SetCachedPlayerData
 			utils::hook::call(0xF4632_b, set_cached_playerdata_stub);
+
+			// Patch entity overflow
+			utils::hook::jump(0x55E4C7_b, assemble(remap_cached_entities_stub), true);
 		}
 	};
 }