Update dllmain.cpp

Maplespe · Jan 10, 2022 · ea6ebaa · ea6ebaa
1 parent de4e9a7
commit ea6ebaa
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/ExplorerBgTool/dllmain.cpp b/ExplorerBgTool/dllmain.cpp
@@ -1,4 +1,4 @@
-/*
+/*
 * 文件资源管理器背景工具扩展
 * 
 * Author: Maple
@@ -66,18 +66,22 @@ bool Random = true;//随机显示图片 Random pictures
 
 #pragma endregion
 
+extern bool InjectionEntryPoint();
+
 BOOL APIENTRY DllMain( HMODULE hModule,
                        DWORD  ul_reason_for_call,
                        LPVOID lpReserved
                      )
 {
-    g_hModule = hModule;
-    DisableThreadLibraryCalls(hModule);
+    if (!g_hModule) {
+        g_hModule = hModule;
+        DisableThreadLibraryCalls(hModule);
+        InjectionEntryPoint();
+    }
     return TRUE;
 }
 
-//Hook入口点 EasyHook EntryPoint
-extern "C" __declspec(dllexport) bool __stdcall InjectionEntryPoint()
+bool InjectionEntryPoint()
 {
     //初始化 Gdiplus Init GdiPlus
     GdiplusStartupInput StartupInput;
@@ -307,4 +311,4 @@ HDC MyCreateCompatibleDC(HDC hDC)
         }
     }
     return retDC;
-}
+}