add:实现GetModulePath接口

GengGode · Feb 9, 2024 · e73ee86 · e73ee86
1 parent 45b5708
commit e73ee86
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/include/cvAutoTrack.h b/include/cvAutoTrack.h
@@ -292,6 +292,7 @@ extern "C"
     bool CVAUTOTRACK_API GetMapIsEmbedded();
     bool CVAUTOTRACK_API GetCompileVersion(char* version_buff, int buff_size);
     bool CVAUTOTRACK_API GetCompileTime(char* time_buff, int buff_size);
+    bool CVAUTOTRACK_API GetModulePath(char* path_buff, int buff_size);
 
     /// @brief
     /// @param doc_buff

diff --git a/source/dllmain.cpp b/source/dllmain.cpp
@@ -1,8 +1,11 @@
 #include "pch.h"
 
-BOOL APIENTRY DllMain(HMODULE /*hModule*/, DWORD ul_reason_for_call, LPVOID /*lpReserved*/
+static HMODULE g_library_module = NULL;
+
+BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID /*lpReserved*/
 )
 {
+    g_library_module = hModule;
     switch (ul_reason_for_call)
     {
         case DLL_PROCESS_ATTACH:
@@ -11,4 +14,16 @@ BOOL APIENTRY DllMain(HMODULE /*hModule*/, DWORD ul_reason_for_call, LPVOID /*lp
         case DLL_PROCESS_DETACH: break;
     }
     return TRUE;
+}
+
+bool GetModulePath(char* path_buff, int buff_size)
+{
+    if (g_library_module == NULL)
+    {
+        return false;
+    }
+    char module_path_char[MAX_PATH] = { 0 };
+    GetModuleFileNameA(g_library_module, module_path_char, MAX_PATH);
+    strcpy_s(path_buff, buff_size, module_path_char);
+    return true;
 }