|
| 1 | +#include "Event.hpp" |
| 2 | +#include "Modules/Console.hpp" |
| 3 | +#include "Modules/Server.hpp" |
| 4 | +#include "Offsets.hpp" |
| 5 | +#include "Utils.hpp" |
| 6 | +#include "Utils/Memory.hpp" |
| 7 | +#include "Variable.hpp" |
| 8 | + |
| 9 | +namespace { |
| 10 | +constexpr unsigned int MODEL_FLAG_PROTECTED_MASK = 0x7E; |
| 11 | +constexpr unsigned int MODEL_FLAG_SKIP_PROTECTED_CLEAR = 0x40; |
| 12 | +constexpr unsigned int MAX_REASONABLE_MODEL_COUNT = 16384; |
| 13 | + |
| 14 | +DECL_CVAR_CALLBACK(sar_demo_modelcache_clear_protected_flags); |
| 15 | + |
| 16 | +Variable sar_demo_modelcache_clear_protected_flags( |
| 17 | + "sar_demo_modelcache_clear_protected_flags", |
| 18 | + "0", |
| 19 | + "Fix demo model-cache growth by clearing stale CModelLoader protected flags on demo stop. Requires sv_cheats 1 to enable.\n", |
| 20 | + FCVAR_NONE, |
| 21 | + sar_demo_modelcache_clear_protected_flags_callback); |
| 22 | + |
| 23 | +DECL_CVAR_CALLBACK(sar_demo_modelcache_clear_protected_flags) { |
| 24 | + if (sar_demo_modelcache_clear_protected_flags.GetBool() && !sv_cheats.GetBool()) { |
| 25 | + console->Print("sar_demo_modelcache_clear_protected_flags requires sv_cheats 1.\n"); |
| 26 | + sar_demo_modelcache_clear_protected_flags.SetValue(0); |
| 27 | + } |
| 28 | +} |
| 29 | + |
| 30 | +void *GetModelLoader() { |
| 31 | + static uintptr_t global = 0; |
| 32 | + if (!global) { |
| 33 | + auto site = Memory::Scan(MODULE("engine"), Offsets::CModelLoaderModelPrecache, Offsets::CModelLoaderModelPrecacheGlobal); |
| 34 | + if (!site) return nullptr; |
| 35 | + |
| 36 | + global = Memory::Deref<uintptr_t>(site); |
| 37 | + } |
| 38 | + |
| 39 | + return Memory::Deref<void *>(global); |
| 40 | +} |
| 41 | + |
| 42 | +void ClearProtectedModelFlags() { |
| 43 | + auto modelLoader = reinterpret_cast<uintptr_t>(GetModelLoader()); |
| 44 | + if (!modelLoader) { |
| 45 | + static bool warned = false; |
| 46 | + if (!warned) { |
| 47 | + warned = true; |
| 48 | + console->Warning("SAR: Failed to find CModelLoader for repeated-demo model-cache cleanup.\n"); |
| 49 | + } |
| 50 | + return; |
| 51 | + } |
| 52 | + |
| 53 | + auto entries = *reinterpret_cast<uintptr_t *>(modelLoader + Offsets::CModelLoaderEntryArray); |
| 54 | + auto count = *reinterpret_cast<unsigned short *>(modelLoader + Offsets::CModelLoaderEntryCount); |
| 55 | + if (!entries || count > MAX_REASONABLE_MODEL_COUNT) { |
| 56 | + static bool warned = false; |
| 57 | + if (!warned) { |
| 58 | + warned = true; |
| 59 | + console->Warning( |
| 60 | + "SAR: Invalid CModelLoader state (loader=%p entries=%p count=%u).\n", |
| 61 | + reinterpret_cast<void *>(modelLoader), |
| 62 | + reinterpret_cast<void *>(entries), |
| 63 | + count); |
| 64 | + } |
| 65 | + return; |
| 66 | + } |
| 67 | + |
| 68 | + for (unsigned int i = 0; i < count; ++i) { |
| 69 | + auto model = *reinterpret_cast<uintptr_t *>(entries + i * Offsets::CModelLoaderEntryStride + Offsets::CModelLoaderEntryModel); |
| 70 | + if (!model) continue; |
| 71 | + |
| 72 | + auto flags = reinterpret_cast<unsigned int *>(model + Offsets::CModelLoaderModelFlags); |
| 73 | + if ((*flags & MODEL_FLAG_SKIP_PROTECTED_CLEAR) != 0) continue; |
| 74 | + |
| 75 | + *flags &= ~MODEL_FLAG_PROTECTED_MASK; |
| 76 | + } |
| 77 | +} |
| 78 | +} // namespace |
| 79 | + |
| 80 | +ON_EVENT(DEMO_STOP) { |
| 81 | + if (sar_demo_modelcache_clear_protected_flags.GetBool() && sv_cheats.GetBool()) { |
| 82 | + ClearProtectedModelFlags(); |
| 83 | + } |
| 84 | +} |
0 commit comments