Use correct pragma in the 64-bit case

CollaboraOnline · Oct 3, 2018 · e7ce4ab · e7ce4ab
1 parent f742833
commit e7ce4ab
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/injecteddll/injecteddll.cpp b/injecteddll/injecteddll.cpp
@@ -1031,7 +1031,11 @@ static bool hook(bool bMandatory, ThreadProcParam* pParam, const wchar_t* sModul
 extern "C" DWORD WINAPI InjectedDllMainFunction(ThreadProcParam* pParam)
 {
 // Magic to export this function using a plain undecorated name despite it being WINAPI
+#ifdef _WIN64
+#pragma comment(linker, "/EXPORT:InjectedDllMainFunction=InjectedDllMainFunction")
+#else
 #pragma comment(linker, "/EXPORT:InjectedDllMainFunction=_InjectedDllMainFunction@4")
+#endif
 
     if (pParam->mnSize != sizeof(*pParam))
     {