From 7c8c7f64b6845e01d763bf14d00b727ebbb9073f Mon Sep 17 00:00:00 2001 From: Darryl Pogue Date: Fri, 16 Jun 2023 21:46:59 -0700 Subject: [PATCH] Set up cross-platform ModDLL loading --- .../Plasma/Apps/plClient/Mac-Cocoa/main.mm | 2 - Sources/Plasma/Apps/plClient/linux/main.cpp | 2 - Sources/Plasma/Apps/plClient/main.cpp | 2 - Sources/Plasma/Apps/plClient/plClient.cpp | 56 +++++++++++++++++++ Sources/Plasma/Apps/plClient/plClient.h | 4 +- .../Apps/plClient/win32/plClient_Win.cpp | 30 ---------- Sources/Plasma/CoreLib/HeadSpin.h | 2 + 7 files changed, 61 insertions(+), 37 deletions(-) diff --git a/Sources/Plasma/Apps/plClient/Mac-Cocoa/main.mm b/Sources/Plasma/Apps/plClient/Mac-Cocoa/main.mm index a861f94403..300a4195de 100644 --- a/Sources/Plasma/Apps/plClient/Mac-Cocoa/main.mm +++ b/Sources/Plasma/Apps/plClient/Mac-Cocoa/main.mm @@ -120,8 +120,6 @@ @interface AppDelegate : NSWindowController // For ModDLL loading +#endif #define MSG_LOADING_BAR @@ -370,6 +374,58 @@ bool plClient::Shutdown() return false; } +void plClient::InitDLLs() { + hsStatusMessage("Init dlls client\n"); + + std::vector dlls = plFileSystem::ListDir("ModDLL", +#if defined(HS_BUILD_FOR_WIN32) + "*.dll" +#elif defined(HS_BUILD_FOR_APPLE) + "*.dylib" +#else + "*.so" +#endif + ); + + for (auto iter = dlls.begin(); iter != dlls.end(); ++iter) + { +#ifdef HS_BUILD_FOR_WIN32 + hsLibraryHndl mod = LoadLibraryW(iter->WideString().data()); +#else + hsLibraryHndl mod = dlopen(iter->AsString().c_str(), RTLD_LAZY | RTLD_LOCAL); +#endif + + if (mod) + { +#ifdef HS_BUILD_FOR_WIN32 + pInitGlobalsFunc initGlobals = (pInitGlobalsFunc)GetProcAddress(mod, "InitGlobals"); +#else + pInitGlobalsFunc initGlobals = (pInitGlobalsFunc)dlsym(mod, "InitGlobals"); +#endif + + (*initGlobals)(hsgResMgr::ResMgr(), plFactory::GetTheFactory(), plgTimerCallbackMgr::Mgr(), + hsTimer::GetTheTimer(), plNetClientApp::GetInstance()); + fLoadedDLLs.emplace_back(mod); + } + } +} + +void plClient::ShutdownDLLs() +{ + for (hsLibraryHndl mod : fLoadedDLLs) + { +#ifdef HS_BUILD_FOR_WIN32 + BOOL ret = FreeLibrary(mod); + if (!ret) + hsStatusMessage("Failed to free lib\n"); +#else + dlclose(mod); +#endif + } + + fLoadedDLLs.clear(); +} + void plClient::InitAuxInits() { // Use another init directory specified in Command line Arg -i diff --git a/Sources/Plasma/Apps/plClient/plClient.h b/Sources/Plasma/Apps/plClient/plClient.h index dd839cb5c8..5baafffa45 100644 --- a/Sources/Plasma/Apps/plClient/plClient.h +++ b/Sources/Plasma/Apps/plClient/plClient.h @@ -178,7 +178,9 @@ class plClient : public hsKeyedObject int fNumPostLoadMsgs; float fPostLoadMsgInc; - + + std::vector fLoadedDLLs; + void ICompleteInit (); void IOnAsyncInitComplete (); void IHandlePatcherMsg (plResPatcherMsg * msg); diff --git a/Sources/Plasma/Apps/plClient/win32/plClient_Win.cpp b/Sources/Plasma/Apps/plClient/win32/plClient_Win.cpp index 0604e448c5..b8aaa4215a 100644 --- a/Sources/Plasma/Apps/plClient/win32/plClient_Win.cpp +++ b/Sources/Plasma/Apps/plClient/win32/plClient_Win.cpp @@ -51,12 +51,10 @@ You can contact Cyan Worlds, Inc. by email legal@cyan.com #include "plClient.h" #include "plWinDpi/plWinDpi.h" -#include "pnFactory/plFactory.h" #include "pnNetCommon/plNetApp.h" #include "plProgressMgr/plProgressMgr.h" extern ITaskbarList3* gTaskbarList; -static std::vector fLoadedDLLs; void plClient::IResizeNativeDisplayDevice(int width, int height, bool windowed) { @@ -146,34 +144,6 @@ void plClient::IUpdateProgressIndicator(plOperationProgress* progress) } } -void plClient::InitDLLs() -{ - hsStatusMessage("Init dlls client\n"); - std::vector dlls = plFileSystem::ListDir("ModDLL", "*.dll"); - for (auto iter = dlls.begin(); iter != dlls.end(); ++iter) - { - HMODULE hMod = LoadLibraryW(iter->WideString().data()); - if (hMod) - { - pInitGlobalsFunc initGlobals = (pInitGlobalsFunc)GetProcAddress(hMod, "InitGlobals"); - (*initGlobals)(hsgResMgr::ResMgr(), plFactory::GetTheFactory(), plgTimerCallbackMgr::Mgr(), - hsTimer::GetTheTimer(), plNetClientApp::GetInstance()); - fLoadedDLLs.emplace_back(hMod); - } - } -} - -void plClient::ShutdownDLLs() -{ - for (HMODULE dll : fLoadedDLLs) - { - BOOL ret = FreeLibrary(dll); - if (!ret) - hsStatusMessage("Failed to free lib\n"); - } - fLoadedDLLs.clear(); -} - // Show the client window void plClient::ShowClientWindow() { diff --git a/Sources/Plasma/CoreLib/HeadSpin.h b/Sources/Plasma/CoreLib/HeadSpin.h index 0f0ce3517e..9016773c51 100644 --- a/Sources/Plasma/CoreLib/HeadSpin.h +++ b/Sources/Plasma/CoreLib/HeadSpin.h @@ -71,11 +71,13 @@ You can contact Cyan Worlds, Inc. by email legal@cyan.com typedef HWND hsWindowHndl; typedef HINSTANCE hsWindowInst; typedef HINSTANCE HMODULE; + typedef HMODULE hsLibraryHndl; typedef long HRESULT; typedef void* HANDLE; #else typedef int32_t* hsWindowHndl; typedef int32_t* hsWindowInst; + typedef void* hsLibraryHndl; #endif // HS_BUILD_FOR_WIN32 //======================================