ChaosMod: Ported PedModel pool to Enhanced

Author: Regynate

ChaosMod/Components/Failsafe.cpp

@@ -30,7 +30,7 @@ Failsafe::Failsafe() : Component()
 		    if (!ms_SearchedForMissionStateGlobal && !ms_StateGlobalIdx && !strcmp(thread->GetName(), "main"))
 		    {
 			    auto program = Memory::ScriptThreadToProgram(thread);
-			    if (program->m_CodeBlocks)
+			    if (program && program->m_CodeBlocks)
 			    {
 				    ms_SearchedForMissionStateGlobal = true;
 

ChaosMod/Lib/scrThread.h

@@ -1,7 +1,5 @@
 #pragma once
 
-#include <scripthookv/inc/main.h>
-
 #include "game.h"
 
 using DWORD   = unsigned long;

ChaosMod/Memory/PedModels.h

@@ -1,5 +1,7 @@
 #pragma once
 
+#include "game.h"
+
 #include "Handle.h"
 #include "Memory.h"
 
@@ -21,31 +23,60 @@ namespace Memory
 		{
 			Handle handle;
 
-			handle = FindPattern("41 F7 F0 48 8B 05 ? ? ? 00 4C");
-			if (!handle.IsValid())
-				return pedModels;
+			DWORD64 archetypeMap_Map;
+			WORD archetypeMap_Slots;
+			DWORD64 archetypePool_ElemSize;
+			DWORD64 archetypePool_PoolPtr;
+			DWORD archetypePool_Size;
+			DWORD64 archetypePool_AllocatedMap_Bits;
+
+			if (IsLegacy())
+			{
+				handle = FindPattern("41 F7 F0 48 8B 05 ? ? ? 00 4C");
+				if (!handle.IsValid())
+					return pedModels;
+
+				archetypeMap_Map   = handle.At(5).Into().Value<DWORD64>();
+				archetypeMap_Slots = handle.At(-28).Into().Value<WORD>();
 
-			DWORD64 qword_7FF69DB37F30 = handle.At(5).Into().Value<DWORD64>();
+				handle             = FindPattern("73 17 4C 0F AF 05");
+				if (!handle.IsValid())
+					return pedModels;
 
-			WORD word_7FF69DB37F38     = handle.At(-28).Into().Value<WORD>();
+				archetypePool_ElemSize = handle.At(5).Into().Value<DWORD64>();
+				archetypePool_PoolPtr  = handle.At(12).Into().Value<DWORD64>();
 
-			handle                     = FindPattern("73 17 4C 0F AF 05");
-			if (!handle.IsValid())
-				return pedModels;
+				handle                 = FindPattern("3B 05 ? ? ? 00 7D 35");
+				if (!handle.IsValid())
+					return pedModels;
 
-			DWORD64 qword_7FF69DB37EE8 = handle.At(5).Into().Value<DWORD64>();
-			DWORD64 qword_7FF69DB37ED0 = handle.At(12).Into().Value<DWORD64>();
+				archetypePool_Size              = handle.At(1).Into().Value<DWORD>();
+				archetypePool_AllocatedMap_Bits = handle.At(18).Into().Value<DWORD64>();
+			}
+			else
+			{
+				handle = Memory::FindPattern("74 7C 49 89 D0 4C 8B 1D");
 
-			handle                     = FindPattern("3B 05 ? ? ? 00 7D 35");
-			if (!handle.IsValid())
-				return pedModels;
+				if (!handle.IsValid())
+					return pedModels;
 
-			DWORD dword_7FF69DB37ED8   = handle.At(1).Into().Value<DWORD>();
-			DWORD64 qword_7FF69DB37F00 = handle.At(18).Into().Value<DWORD64>();
+				archetypeMap_Map                = handle.At(7).Into().Value<DWORD64>();
+				archetypeMap_Slots              = handle.At(-8).Into().Value<WORD>();
+
+				handle                          = Memory::FindPattern("7D 36 48 8B 05");
+
+				archetypePool_Size              = handle.At(-5).Into().Value<DWORD>();
+				archetypePool_AllocatedMap_Bits = handle.At(4).Into().Value<DWORD64>();
+
+				archetypePool_PoolPtr           = handle.At(27).Into().Value<DWORD64>();
+				archetypePool_ElemSize          = handle.At(43).Into().Value<DWORD64>();
+
+				DEBUG_LOG(archetypeMap_Map << " " << archetypeMap_Slots << " " << archetypePool_Size << " " << archetypePool_AllocatedMap_Bits << " " << archetypePool_PoolPtr << " " << archetypePool_ElemSize);
+			}
 
-			for (WORD i = 0; i < word_7FF69DB37F38; i++)
+			for (WORD i = 0; i < archetypeMap_Slots; i++)
 			{
-				auto model = *reinterpret_cast<Hash **>(qword_7FF69DB37F30 + 8 * i);
+				auto model = *reinterpret_cast<Hash **>(archetypeMap_Map + 8 * i);
 				if (!model)
 					continue;
 
@@ -59,13 +90,13 @@ namespace Memory
 					DWORD64 v5 = *reinterpret_cast<WORD *>(model + 4);
 					LONG v6;
 					// IDA copy paste ftw
-					if (static_cast<DWORD>(v5) < dword_7FF69DB37ED8
-					    && (v6 = *reinterpret_cast<DWORD *>(qword_7FF69DB37F00 + 4 * (v5 >> 5)),
+					if (static_cast<DWORD>(v5) < archetypePool_Size
+					    && (v6 = *reinterpret_cast<DWORD *>(archetypePool_AllocatedMap_Bits + 4 * (v5 >> 5)),
 					        _bittest(&v6, v5 & 0x1F)))
 					{
 						v2 = *reinterpret_cast<__int64 *>(
-						    qword_7FF69DB37ED0
-						    + qword_7FF69DB37EE8 * *reinterpret_cast<WORD *>(reinterpret_cast<__int64>(model) + 4));
+						    archetypePool_PoolPtr
+						    + archetypePool_ElemSize * *reinterpret_cast<WORD *>(reinterpret_cast<__int64>(model) + 4));
 					}
 
 					if (v2 && (*reinterpret_cast<BYTE *>(v2 + 157) & 31) == 6) // is a ped model

ChaosMod/game.h

@@ -8,18 +8,18 @@ enum class GameVersion
     GTA5_ENHANCED
 };
 
-GameVersion GetGame()
+inline GameVersion GetGame()
 {
     static GameVersion result = getGameVersion() > 1000 ? GameVersion::GTA5_ENHANCED : GameVersion::GTA5_LEGACY;
     return result;
 }
 
-bool IsEnhanced()
+inline bool IsEnhanced()
 {
     return GetGame() == GameVersion::GTA5_ENHANCED;
 }
 
-bool IsLegacy()
+inline bool IsLegacy()
 {
     return GetGame() == GameVersion::GTA5_LEGACY;
 }