geode/loader/launcher/windows/Bootstrapper.cpp

71 lines
1.8 KiB
C++
Raw Normal View History

#include <Windows.h>
2022-10-08 06:21:57 -04:00
#include <iostream>
#include <ghc/filesystem.hpp>
void showError(std::string const& error) {
MessageBoxA(nullptr, error.c_str(), "Error Loading Geode", MB_ICONERROR);
}
int loadGeode(PVOID module) {
if (!LoadLibraryW(L"Geode.dll")) {
auto code = GetLastError();
showError("Unable to load Geode (code " + std::to_string(code) + ")");
return code;
}
FreeLibraryAndExitThread(static_cast<HINSTANCE>(module), 0);
return 0;
2022-10-08 06:21:57 -04:00
}
DWORD WINAPI load(PVOID module) {
2023-05-01 07:47:25 -04:00
std::array<TCHAR, MAX_PATH> szFileName;
GetModuleFileName(NULL, szFileName.data(), MAX_PATH);
2023-05-01 08:16:29 -04:00
ghc::filesystem::path path(szFileName.data());
2023-05-01 07:47:25 -04:00
auto workingDir = path.parent_path();
auto updatesDir = workingDir / "geode" / "update";
auto resourcesDir = workingDir / "geode" / "resources";
auto error = std::error_code();
if (ghc::filesystem::exists(updatesDir / "Geode.dll", error) && !error) {
2022-10-11 12:14:13 -04:00
ghc::filesystem::rename(
updatesDir / "Geode.dll",
workingDir / "Geode.dll", error
);
if (error) {
showError("Unable to update Geode: Unable to move Geode.dll - " + error.message());
return error.value();
}
2022-10-11 12:14:13 -04:00
}
2022-10-11 12:14:13 -04:00
if (ghc::filesystem::exists(updatesDir / "resources", error) && !error) {
ghc::filesystem::remove_all(resourcesDir / "geode.loader", error);
2022-10-08 06:21:57 -04:00
2022-10-11 12:14:13 -04:00
if (error) {
showError("Unable to update Geode resources: " + error.message());
2022-10-11 12:14:13 -04:00
} else {
ghc::filesystem::rename(
updatesDir / "resources",
resourcesDir / "geode.loader", error
);
if (error) {
showError("Unable to update Geode resources: " + error.message());
}
}
2022-10-11 12:14:13 -04:00
}
return loadGeode(module);
}
2022-10-11 12:14:13 -04:00
BOOL WINAPI DllMain(HINSTANCE module, DWORD reason, LPVOID) {
if (reason == DLL_PROCESS_ATTACH) {
HANDLE handle = CreateThread(NULL, 0, load, module, 0, NULL);
if (handle) {
CloseHandle(handle);
} else {
return FALSE;
}
}
return TRUE;
2023-05-01 08:16:29 -04:00
}