From 3f048f3f326c79f0ad0d041c1949c35b26ecad13 Mon Sep 17 00:00:00 2001
From: momo5502 <mauriceheumann@gmail.com>
Date: Sun, 2 Oct 2016 18:11:30 +0200
Subject: [PATCH] Some more gfxworld stuff

---
 src/Components/Modules/AssetHandler.cpp |  4 ++++
 src/Components/Modules/Maps.cpp         |  6 ++++++
 src/Components/Modules/Zones.cpp        | 22 ++++++++++++++++++----
 3 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/src/Components/Modules/AssetHandler.cpp b/src/Components/Modules/AssetHandler.cpp
index 756a3354..882b7303 100644
--- a/src/Components/Modules/AssetHandler.cpp
+++ b/src/Components/Modules/AssetHandler.cpp
@@ -188,6 +188,10 @@ namespace Components
 		}
 
 		offset->pointer = *reinterpret_cast<void**>(pointer);
+
+#ifdef DEBUG
+		Game::XAssetHeader zob{ offset->pointer };
+#endif
  	}
 
 	void AssetHandler::ZoneSave(Game::XAsset asset, ZoneBuilder::Zone* builder)
diff --git a/src/Components/Modules/Maps.cpp b/src/Components/Modules/Maps.cpp
index 3dca67d9..4979ad8e 100644
--- a/src/Components/Modules/Maps.cpp
+++ b/src/Components/Modules/Maps.cpp
@@ -117,6 +117,12 @@ namespace Components
 			// This doesn't work, entities are spawned before the patch file is loaded
 			//Maps::OverrideMapEnts(asset.mapEnts);
 		}
+
+		if (type == Game::XAssetType::ASSET_TYPE_MENU && Zones::Version() >= 359)
+		{
+			*restrict = true;
+			return;
+		}
 	}
 
 	void Maps::GetBSPName(char* buffer, size_t size, const char* format, const char* mapname)
diff --git a/src/Components/Modules/Zones.cpp b/src/Components/Modules/Zones.cpp
index c9a3b562..75d46516 100644
--- a/src/Components/Modules/Zones.cpp
+++ b/src/Components/Modules/Zones.cpp
@@ -954,9 +954,12 @@ namespace Components
 
 	bool Zones::LoadmenuDef_t(bool atStreamStart, char* buffer, int size)
 	{
-		bool result = Game::Load_Stream(atStreamStart, buffer, size + 4);
-		std::memmove(buffer + 168, buffer + 172, 232);
-		AssetHandler::Relocate(buffer + 172, buffer + 168, 232);
+		if (Zones::ZoneVersion < 359) size += 4;
+
+		bool result = Game::Load_Stream(atStreamStart, buffer, size);
+		std::memmove(buffer + 168, buffer + 172, (Zones::ZoneVersion < 359 ? 232 : 228));
+		AssetHandler::Relocate(buffer + 172, buffer + 168, (Zones::ZoneVersion < 359 ? 232 : 228));
+
 		return result;
 	}
 
@@ -1160,7 +1163,8 @@ namespace Components
 	{
 		bool result = Game::Load_Stream(atStreamStart, buffer, size + 968);
 
-		std::memmove(buffer + 348, buffer + 1316, 280);
+		int sunDiff = 8;
+		std::memmove(buffer + 348 + sunDiff, buffer + 1316 + sunDiff, 280 - sunDiff);
 		AssetHandler::Relocate(buffer + 1316, buffer + 348, 280);
 
 		return result;
@@ -1179,6 +1183,9 @@ namespace Components
 		Game::Load_MaterialHandle(atStreamStart);
 
 		std::memmove(varsunflare_t + 12, varsunflare_t + 20, 84);
+
+		// Copy the remaining struct data we couldn't copy in LoadGfxWorld
+		std::memmove(varsunflare_t + 96, varsunflare_t + 104, 8);
 	}
 
 	void Zones::InstallPatches(int version)
@@ -1273,6 +1280,9 @@ namespace Components
 				Zones::LoadMaterialHook.Install();
 				Zones::LoadGfxWorldHook.Install();
 				Zones::Loadsunflare_tHook.Install();
+
+				// menu stuff
+				Utils::Hook::Nop(0x41A590, 5);
 			}
 			else
 			{
@@ -1281,6 +1291,8 @@ namespace Components
 				Zones::LoadMaterialHook.Uninstall();
 				Zones::LoadGfxWorldHook.Uninstall();
 				Zones::Loadsunflare_tHook.Uninstall();
+
+				Utils::Hook(0x41A590, 0x4AF680, HOOK_CALL).Install()->Quick();
 			}
 
 			Zones::LoadMaterialShaderArgumentArrayHook.Install();
@@ -1320,6 +1332,8 @@ namespace Components
 			Zones::LoadMaterialHook.Uninstall();
 			Zones::LoadGfxWorldHook.Uninstall();
 			Zones::Loadsunflare_tHook.Uninstall();
+
+			Utils::Hook(0x41A590, 0x4AF680, HOOK_CALL).Install()->Quick();
 		}
 
 		AntiCheat::EmptyHash();