show error when DetourGetEntryPoint() fails

Rosalie241 · Jul 7, 2022 · c502410 · c502410
1 parent c7fc11c
commit c502410
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/SporeModLoader/dllmain.cpp b/SporeModLoader/dllmain.cpp
@@ -59,9 +59,11 @@ BOOL WINAPI DllMain(HINSTANCE hinst, DWORD dwReason, LPVOID reserved)
     if (dwReason == DLL_PROCESS_ATTACH)
     {
         DetourRestoreAfterWith();
-
         SporeAppEntry_real = (int (WINAPI*)(VOID))DetourGetEntryPoint(nullptr);
-
+        if (SporeAppEntry_real == nullptr)
+        {
+            MessageBoxW(nullptr, L"DetourGetEntryPoint() Failed!", L"SporeModLoader", MB_OK | MB_ICONERROR);
+        }
         DetourTransactionBegin();
         DetourUpdateThread(GetCurrentThread());
         DetourAttach(&(PVOID&)SporeAppEntry_real, SporeAppEntry_detoured);
@@ -71,7 +73,10 @@ BOOL WINAPI DllMain(HINSTANCE hinst, DWORD dwReason, LPVOID reserved)
     {
         DetourTransactionBegin();
         DetourUpdateThread(GetCurrentThread());
-        DetourDetach(&(PVOID&)SporeAppEntry_real, SporeAppEntry_detoured);
+        if (SporeAppEntry_real != nullptr)
+        {
+            DetourDetach(&(PVOID&)SporeAppEntry_real, SporeAppEntry_detoured);
+        }
         DetourTransactionCommit();
     }