mirror of
https://github.com/geode-sdk/geode.git
synced 2024-11-30 11:17:08 -05:00
118 lines
3.6 KiB
C++
118 lines
3.6 KiB
C++
#include <Windows.h>
|
|
#include <iostream>
|
|
#include <array>
|
|
#include <filesystem>
|
|
|
|
std::filesystem::path workingDir;
|
|
std::filesystem::path geodeDir;
|
|
std::filesystem::path updatesDir;
|
|
std::filesystem::path resourcesDir;
|
|
|
|
void showError(std::string const& error) {
|
|
MessageBoxA(nullptr, error.c_str(), "Error Loading Geode", MB_ICONERROR);
|
|
}
|
|
|
|
bool waitForFile(std::filesystem::path const& path) {
|
|
if (!path.has_filename())
|
|
return false;
|
|
|
|
int delay = 10;
|
|
HANDLE hFile;
|
|
while ((hFile = CreateFileA(path.string().c_str(), FILE_GENERIC_WRITE, FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL)) == INVALID_HANDLE_VALUE) {
|
|
if (GetLastError() == ERROR_SHARING_VIOLATION) {
|
|
Sleep(delay);
|
|
// the delay would raise and go up to about 5 seconds, after which it will fail
|
|
if (delay < 5120)
|
|
delay *= 2;
|
|
}
|
|
else
|
|
break;
|
|
}
|
|
if (hFile)
|
|
CloseHandle(hFile);
|
|
else {
|
|
showError("Unable to update Geode: " + path.filename().string() + " is open by another process.");
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void updateFile(std::string const& name) {
|
|
std::error_code error;
|
|
if (!std::filesystem::exists(updatesDir / name, error) || error)
|
|
return;
|
|
if (!waitForFile(workingDir / name))
|
|
return;
|
|
|
|
std::filesystem::rename(updatesDir / name, workingDir / name, error);
|
|
if (error) {
|
|
showError("Unable to update Geode: Unable to move " + name + " - " + error.message());
|
|
return;
|
|
}
|
|
}
|
|
|
|
void removePath(std::filesystem::path const& path) {
|
|
std::error_code error;
|
|
if (!std::filesystem::exists(path, error) || error)
|
|
return;
|
|
if (path.has_filename() && !waitForFile(path))
|
|
return;
|
|
|
|
if (std::filesystem::is_directory(path) && !std::filesystem::is_empty(path))
|
|
std::filesystem::remove_all(path, error);
|
|
std::filesystem::remove(path, error);
|
|
if (error) {
|
|
if (path.has_filename())
|
|
showError("Unable to update Geode: Unable to remove " + path.filename().string() + " - " + error.message());
|
|
else
|
|
showError("Unable to update Geode: Unable to remove " + path.string() + " - " + error.message());
|
|
return;
|
|
}
|
|
}
|
|
|
|
void updateResources() {
|
|
std::error_code error;
|
|
if (!std::filesystem::exists(updatesDir / "resources", error) || error)
|
|
return;
|
|
|
|
std::filesystem::remove_all(resourcesDir / "geode.loader", error);
|
|
if (error) {
|
|
showError("Unable to update Geode resources:" + error.message());
|
|
return;
|
|
}
|
|
|
|
std::filesystem::rename(updatesDir / "resources", resourcesDir / "geode.loader", error);
|
|
if (error) {
|
|
showError("Unable to update Geode resources: " + error.message());
|
|
return;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
workingDir = std::filesystem::current_path();
|
|
geodeDir = workingDir / "geode";
|
|
updatesDir = geodeDir / "update";
|
|
resourcesDir = geodeDir / "resources";
|
|
|
|
if (std::filesystem::exists(workingDir / "GeodeBootstrapper.dll"))
|
|
removePath(workingDir / "GeodeBootstrapper.dll");
|
|
|
|
if (std::filesystem::exists(geodeDir) && std::filesystem::exists(updatesDir)) {
|
|
updateFile("XInput9_1_0.dll");
|
|
updateFile("Geode.dll");
|
|
updateResources();
|
|
removePath(updatesDir);
|
|
}
|
|
|
|
if (argc < 2)
|
|
return 0;
|
|
|
|
if (!waitForFile(workingDir / argv[1])) {
|
|
showError("There was an error restarting GD. Please, restart the game manually.");
|
|
return 0;
|
|
}
|
|
|
|
// restart gd using the provided path
|
|
ShellExecuteA(NULL, "open", (workingDir / argv[1]).string().c_str(), "", workingDir.string().c_str(), TRUE);
|
|
return 0;
|
|
}
|