mirror of
https://github.com/geode-sdk/geode.git
synced 2024-12-12 09:01:17 -05:00
70 lines
No EOL
1.8 KiB
C++
70 lines
No EOL
1.8 KiB
C++
#include <Windows.h>
|
|
#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;
|
|
}
|
|
|
|
DWORD WINAPI load(PVOID module) {
|
|
std::array<TCHAR, MAX_PATH> szFileName;
|
|
GetModuleFileName(NULL, szFileName.data(), MAX_PATH);
|
|
|
|
ghc::filesystem::path path(szFileName);
|
|
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) {
|
|
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();
|
|
}
|
|
}
|
|
|
|
if (ghc::filesystem::exists(updatesDir / "resources", error) && !error) {
|
|
ghc::filesystem::remove_all(resourcesDir / "geode.loader", error);
|
|
|
|
if (error) {
|
|
showError("Unable to update Geode resources: " + error.message());
|
|
} else {
|
|
ghc::filesystem::rename(
|
|
updatesDir / "resources",
|
|
resourcesDir / "geode.loader", error
|
|
);
|
|
if (error) {
|
|
showError("Unable to update Geode resources: " + error.message());
|
|
}
|
|
}
|
|
}
|
|
|
|
return loadGeode(module);
|
|
}
|
|
|
|
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;
|
|
} |