From 459801331055f2ea8920747cd95a5a0dee9859c7 Mon Sep 17 00:00:00 2001
From: narzoul <narzoul@users.noreply.github.com>
Date: Sun, 25 Aug 2024 00:58:51 +0200
Subject: [PATCH] Added support for ddrawex

See issue #349.
---
 DDrawCompat/DDrawCompat.vcxproj |  4 +-
 DDrawCompat/Win32/Version.cpp   | 90 +++++++++++++++++++++++++++++++++
 2 files changed, 92 insertions(+), 2 deletions(-)

diff --git a/DDrawCompat/DDrawCompat.vcxproj b/DDrawCompat/DDrawCompat.vcxproj
index 9bb8e5e..e0958d9 100644
--- a/DDrawCompat/DDrawCompat.vcxproj
+++ b/DDrawCompat/DDrawCompat.vcxproj
@@ -72,7 +72,7 @@
       <AdditionalIncludeDirectories>$(ProjectDir);$(IntDir)</AdditionalIncludeDirectories>
     </ClCompile>
     <Link>
-      <AdditionalDependencies>dwmapi.lib;dxguid.lib;imm32.lib;msimg32.lib;oleacc.lib;uxtheme.lib;winmm.lib;%(AdditionalDependencies)</AdditionalDependencies>
+      <AdditionalDependencies>dwmapi.lib;dxguid.lib;imm32.lib;msimg32.lib;oleacc.lib;uxtheme.lib;version.lib;winmm.lib;%(AdditionalDependencies)</AdditionalDependencies>
       <GenerateDebugInformation>DebugFull</GenerateDebugInformation>
       <OptimizeReferences>true</OptimizeReferences>
       <ImportLibrary>$(IntDir)$(TargetName).lib</ImportLibrary>
@@ -110,7 +110,7 @@
       <AdditionalIncludeDirectories>$(ProjectDir);$(IntDir)</AdditionalIncludeDirectories>
     </ClCompile>
     <Link>
-      <AdditionalDependencies>dwmapi.lib;dxguid.lib;imm32.lib;msimg32.lib;oleacc.lib;uxtheme.lib;winmm.lib;%(AdditionalDependencies)</AdditionalDependencies>
+      <AdditionalDependencies>dwmapi.lib;dxguid.lib;imm32.lib;msimg32.lib;oleacc.lib;uxtheme.lib;version.lib;winmm.lib;%(AdditionalDependencies)</AdditionalDependencies>
       <GenerateDebugInformation>DebugFull</GenerateDebugInformation>
       <OptimizeReferences>true</OptimizeReferences>
       <ImportLibrary>$(IntDir)$(TargetName).lib</ImportLibrary>
diff --git a/DDrawCompat/Win32/Version.cpp b/DDrawCompat/Win32/Version.cpp
index 8a01c39..b359489 100644
--- a/DDrawCompat/Win32/Version.cpp
+++ b/DDrawCompat/Win32/Version.cpp
@@ -2,7 +2,9 @@
 
 #include <Common/Hook.h>
 #include <Common/Log.h>
+#include <Common/Path.h>
 #include <Config/Settings/WinVersionLie.h>
+#include <Dll/Dll.h>
 
 #include <Win32/Version.h>
 
@@ -10,6 +12,55 @@
 
 namespace
 {
+	DWORD getModuleFileName(HMODULE mod, char* filename, DWORD size)
+	{
+		return GetModuleFileNameA(mod, filename, size);
+	}
+
+	DWORD getModuleFileName(HMODULE mod, wchar_t* filename, DWORD size)
+	{
+		return GetModuleFileNameW(mod, filename, size);
+	}
+
+	HMODULE getModuleHandle(const char* moduleName)
+	{
+		return GetModuleHandleA(moduleName);
+	}
+
+	HMODULE getModuleHandle(const wchar_t* moduleName)
+	{
+		return GetModuleHandleW(moduleName);
+	}
+
+	template <typename Char>
+	void fixVersionInfoFileName(const Char*& filename)
+	{
+		if (getModuleHandle(filename) == Dll::g_currentModule)
+		{
+			static Char path[MAX_PATH];
+			if (0 != getModuleFileName(Dll::g_origDDrawModule, path, MAX_PATH))
+			{
+				filename = path;
+			}
+		}
+	}
+
+	template <auto func, typename Result, typename Char, typename... Params>
+	Result WINAPI getFileVersionInfoFunc(const Char* filename ,Params... params)
+	{
+		LOG_FUNC(Compat::g_origFuncName<func>.c_str(), filename, params...);
+		fixVersionInfoFileName(filename);
+		return LOG_RESULT(CALL_ORIG_FUNC(func)(filename, params...));
+	}
+
+	template <auto func, typename Result, typename Char, typename... Params>
+	Result WINAPI getFileVersionInfoFunc(DWORD flags, const Char* filename, Params... params)
+	{
+		LOG_FUNC(Compat::g_origFuncName<func>.c_str(), filename, params...);
+		fixVersionInfoFileName(filename);
+		return LOG_RESULT(CALL_ORIG_FUNC(func)(flags, filename, params...));
+	}
+
 	DWORD WINAPI getVersion()
 	{
 		LOG_FUNC("GetVersion");
@@ -64,14 +115,53 @@ namespace
 	{
 		return getVersionInfo<OSVERSIONINFOEXW>(lpVersionInformation, CALL_ORIG_FUNC(GetVersionExW), "GetVersionExW");
 	}
+
+	template <auto origFunc>
+	bool hookVersionInfoFunc(const char* moduleName, const char* funcName)
+	{
+		HMODULE mod = GetModuleHandle(moduleName);
+		if (mod)
+		{
+			FARPROC func = Compat::getProcAddress(mod, funcName);
+			if (func)
+			{
+				Compat::hookFunction<origFunc>(moduleName, funcName, getFileVersionInfoFunc<origFunc>);
+				return true;
+			}
+		}
+		return false;
+	}
+
+	template <auto origFunc>
+	void hookVersionInfoFunc(const char* funcName)
+	{
+		hookVersionInfoFunc<origFunc>("kernelbase", funcName) || hookVersionInfoFunc<origFunc>("version", funcName);
+	}
 }
 
+namespace Compat
+{
+	template<> decltype(&GetFileVersionInfoExA) g_origFuncPtr<GetFileVersionInfoExA> = nullptr;
+	template<> decltype(&GetFileVersionInfoSizeExA) g_origFuncPtr<GetFileVersionInfoSizeExA> = nullptr;
+}
+
+#define HOOK_VERSION_INFO_FUNCTION(func) hookVersionInfoFunc<func>(#func)
+
 namespace Win32
 {
 	namespace Version
 	{
 		void installHooks()
 		{
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoA);
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoW);
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoExA);
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoExW);
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoSizeA);
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoSizeW);
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoSizeExA);
+			HOOK_VERSION_INFO_FUNCTION(GetFileVersionInfoSizeExW);
+
 			HOOK_FUNCTION(kernel32, GetVersion, getVersion);
 			HOOK_FUNCTION(kernel32, GetVersionExA, getVersionExA);
 			HOOK_FUNCTION(kernel32, GetVersionExW, getVersionExW);