feat: patch material memory allocation

Author: ThisAMJ

src/Modules/MaterialSystem.cpp

@@ -10,6 +10,8 @@
 #include "Utils.hpp"
 #include "Features/OverlayRender.hpp"
 
+#define RENDERCONTEXT_ALLOC_SIZE 0x672000 // 6.45 MB
+
 class MemTexRegen : public ITextureRegenerator {
 	uint8_t *bgra;
 	int w, h;
@@ -92,10 +94,29 @@ bool MaterialSystem::Init() {
 		OverlayRender::initMaterials();
 	}
 
+	this->renderContextSize = Memory::Scan<uint32_t *>(this->Name(), Offsets::RenderContextSize, Offsets::RenderContextSizeOff);
+	this->RenderContextShutdown = Memory::Scan<_RenderContextShutdown>(this->Name(), Offsets::RenderContextShutdown);
+	this->RenderContextInit = Memory::Scan<_RenderContextInit>(this->Name(), Offsets::RenderContextInit);
+	if (this->renderContextSize && this->RenderContextShutdown && this->RenderContextInit) {
+		if (*this->renderContextSize != RENDERCONTEXT_ALLOC_SIZE) {
+			Memory::UnProtect((void *)this->renderContextSize, sizeof(uint32_t));
+			this->origRenderContextSize = *this->renderContextSize;
+			*this->renderContextSize = RENDERCONTEXT_ALLOC_SIZE;
+			this->RenderContextShutdown();
+			this->RenderContextInit();
+		}
+	}
+
 	return this->hasLoaded = this->materials;
 }
 void MaterialSystem::Shutdown() {
 	Interface::Delete(this->materials);
+
+	if (origRenderContextSize) {
+		*renderContextSize = origRenderContextSize;
+		RenderContextShutdown();
+		RenderContextInit();
+	}
 }
 IMaterial *MaterialSystem::FindMaterial(const char *materialName, const char *textureGroupName) {
 	auto func = (IMaterial *(__rescall *)(void *, const char *, const char *, bool, const char *))this->materials->Current(Offsets::FindMaterial);

src/Modules/MaterialSystem.hpp

@@ -11,6 +11,15 @@ class MaterialSystem : public Module {
 public:
 	Interface *materials = nullptr;
 
+	uint32_t *renderContextSize = nullptr;
+	uint32_t origRenderContextSize = 0;
+
+    using _RenderContextShutdown = void(__cdecl *)(void);
+	_RenderContextShutdown RenderContextShutdown = nullptr;
+
+    using _RenderContextInit = void(__cdecl *)(void);
+	_RenderContextInit RenderContextInit = nullptr;
+
 	using _RemoveMaterial = void(__rescall*)(void* thisptr, IMaterialInternal* pMaterial);
 	_RemoveMaterial RemoveMaterial = nullptr;
 

src/Offsets/Portal 2 9568.hpp

@@ -467,7 +467,13 @@ SIGSCAN_DEFAULT(UpdateLeaderboardData, "55 8B EC 83 EC 08 53 8B D9 8B 03 8B 50 0
 // MaterialSystem
 SIGSCAN_DEFAULT(KeyValues_SetString, "55 8B EC 8B 45 08 6A 01 50 E8 ? ? ? ? 85 C0 74 0B",
                                      "53 83 EC ? 8B 5C 24 ? 6A ? FF 74 24 ? FF 74 24 ? E8 ? ? ? ? 83 C4 ? 85 C0 74 ? 89 5C 24")
-
+SIGSCAN_DEFAULT(RenderContextSize, "56 BE ? ? ? ? E8",
+                                   "81 C3 ? ? ? ? E9") // "portal2" AND "infested" AND "swarm" -> CMatRenderContextBase::Init -> nSize initial value (linux is weird)
+OFFSET_DEFAULT(RenderContextSizeOff, 2, 2)
+SIGSCAN_DEFAULT(RenderContextShutdown, "FF 0D ? ? ? ? 75 ? B9",
+                                       "83 3D ? ? ? ? 00 0F 85 ? ? ? ? 56") // CMatRenderContextBase::Init -> sm_nInitializeCount usage -> decrement, check for 0 -> CMatRenderContextBase::Shutdown
+SIGSCAN_DEFAULT(RenderContextInit, "83 3D ? ? ? ? 00 0F 85 ? ? ? ? 56",
+                                   "A1 ? ? ? ? 85 C0 74 ? 83 C0 01") // CMatRenderContextBase::Init
 
 // Server
 SIGSCAN_DEFAULT(GlobalEntity_GetIndex, "55 8B EC 51 8B 45 08 50 8D 4D FC 51 B9 ? ? ? ? E8 ? ? ? ? 66 8B 55 FC B8 FF FF 00 00",