From 19a4c626173c38efcece37c536ad00927c16a699 Mon Sep 17 00:00:00 2001
From: momo5502 <mauriceheumann@gmail.com>
Date: Tue, 26 Jan 2016 01:41:21 +0100
Subject: [PATCH] Use static cast and check binary.

---
 src/Components/Modules/AssetInterfaces/IMaterial.cpp | 6 +++---
 src/Main.cpp                                         | 6 ++++++
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/Components/Modules/AssetInterfaces/IMaterial.cpp b/src/Components/Modules/AssetInterfaces/IMaterial.cpp
index fc9f7899..79a30eab 100644
--- a/src/Components/Modules/AssetInterfaces/IMaterial.cpp
+++ b/src/Components/Modules/AssetInterfaces/IMaterial.cpp
@@ -61,12 +61,12 @@ namespace Assets
 
 				if (animCoordX.is_number())
 				{
-					material->textureAtlasColumnCount = (char)animCoordX.number_value() & 0xFF;
+					material->textureAtlasColumnCount = static_cast<char>(animCoordX.number_value()) & 0xFF;
 				}
 
 				if (animCoordY.is_number())
 				{
-					material->textureAtlasRowCount = (char)animCoordY.number_value() & 0xFF;
+					material->textureAtlasRowCount = static_cast<char>(animCoordY.number_value()) & 0xFF;
 				}
 			}
 		}
@@ -121,7 +121,7 @@ namespace Assets
 				material->textureTable = 0;
 			}
 
-			material->textureCount = (char)textureList.size() & 0xFF;
+			material->textureCount = static_cast<char>(textureList.size()) & 0xFF;
 		}
 
 		header->material = material;
diff --git a/src/Main.cpp b/src/Main.cpp
index 1a6f0e07..9aa40193 100644
--- a/src/Main.cpp
+++ b/src/Main.cpp
@@ -20,6 +20,12 @@ BOOL APIENTRY DllMain(HMODULE hModule, DWORD  ul_reason_for_call, LPVOID lpReser
 {
 	if (ul_reason_for_call == DLL_PROCESS_ATTACH)
 	{
+		// Ensure we're working with our desired binary
+		if (Utils::Hook::Get<DWORD>(0x6BAC0F) != 0xF44EE8)
+		{
+			return FALSE;
+		}
+
 		DWORD oldProtect;
 		VirtualProtect(GetModuleHandle(NULL), 0x6C73000, PAGE_EXECUTE_READWRITE, &oldProtect);