add mutex for overload

- update submodules
ThirteenAG · Jan 7, 2025 · 3ec3632 · 3ec3632
1 parent f999fe9
commit 3ec3632
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 3 deletions.
diff --git a/external/d3d8to9 b/external/d3d8to9
diff --git a/external/injector b/external/injector
diff --git a/external/minidx9 b/external/minidx9
diff --git a/source/dllmain.cpp b/source/dllmain.cpp
@@ -10,6 +10,9 @@
 extern "C" Direct3D8* WINAPI Direct3DCreate8(UINT SDKVersion);
 #endif
 
+extern "C" IMAGE_DOS_HEADER __ImageBase;
+typedef NTSTATUS(NTAPI* LdrAddRefDll_t)(ULONG, HMODULE);
+
 bool WINAPI IsUltimateASILoader()
 {
     return true;
@@ -1561,6 +1564,23 @@ namespace OverloadFromFolder
 
     void HookAPIForOverload()
     {
+        constexpr auto mutexName = L"Ultimate-ASI-Loader-OverloadFromFolder";
+
+        auto hMutex = OpenMutexW(MUTEX_ALL_ACCESS, FALSE, mutexName);
+        if (hMutex)
+        {
+            CloseHandle(hMutex);
+            return;
+        }
+
+        hMutex = CreateMutexW(nullptr, TRUE, mutexName);
+        if (!hMutex || GetLastError() == ERROR_ALREADY_EXISTS)
+        {
+            if (hMutex)
+                CloseHandle(hMutex);
+            return;
+        }
+
         mhLoadLibraryExA = std::make_unique<FunctionHookMinHook>((uintptr_t)LoadLibraryExA, (uintptr_t)shCustomLoadLibraryExA);
         mhLoadLibraryExW = std::make_unique<FunctionHookMinHook>((uintptr_t)LoadLibraryExW, (uintptr_t)shCustomLoadLibraryExW);
         mhCreateFileA = std::make_unique<FunctionHookMinHook>((uintptr_t)CreateFileA, (uintptr_t)shCustomCreateFileA);
@@ -1590,6 +1610,17 @@ namespace OverloadFromFolder
         mhFindNextFileW->create();
         mhFindFirstFileExA->create();
         mhFindFirstFileExW->create();
+
+        // increase the ref count in case this dll is unloaded before the game exit
+        auto hNtdll = GetModuleHandleW(L"ntdll.dll");
+        if (hNtdll)
+        {
+            auto pLdrAddRefDll = (LdrAddRefDll_t)GetProcAddress(hNtdll, "LdrAddRefDll");
+            if (pLdrAddRefDll)
+            {
+                pLdrAddRefDll(0, (HMODULE)&__ImageBase);
+            }
+        }
     }
 }
+3 −0		res/d3d8.def
+114 −0		source/d3d8to9.cpp
+6 −4		source/d3d8to9.hpp
+1 −1		source/d3d8to9_base.cpp
+198 −43		source/d3d8to9_device.cpp
+10 −0		source/d3d8types.cpp
+21 −0		source/d3d8types.hpp
+151 −0		Include/activation.h
+55 −0		Include/hstring.h
+169 −0		Include/inspectable.h
+93 −0		Include/roapi.h
+306 −0		Include/wrl/client.h
+36 −0		Include/wrl/internal.h
+23 −0		Include/wrl/module.h
+117 −0		Include/wrl/wrappers/corewrappers.h