BeamMP/BeamMP-Launcher
1
0
Fork 0
mirror of https://github.com/BeamMP/BeamMP-Launcher.git synced 2025-07-01 23:46:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

- fixed IPC timeout on inactivity

Browse Source
This commit is contained in:
Anonymous-275 2023-04-19 12:34:58 +01:00
parent 4a4204932d
commit 58e9fef53f
2 changed files with 34 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/Memory/BeamNG.h
View File

@ -14,7 +14,7 @@ class BeamNG {
public:
static void EntryPoint();
static void SendIPC(const std::string& Data);
static inline bool Terminate = false;
private:
static inline std::unique_ptr<Hook<def::update_function>> UpdateDetour;
static inline std::unique_ptr<Hook<def::lua_open_jit>> OpenJITDetour;

61
src/Memory/BeamNG.cpp
View File

@ -6,6 +6,7 @@
#include "Memory/BeamNG.h"
#include "Memory/Memory.h"
#include "atomic_queue.h"
#include <csignal>
std::unique_ptr<atomic_queue<std::string, 1000>> RCVQueue, SendQueue;
@ -28,30 +29,39 @@ uint64_t BeamNG::update_D(lua_State* State) {
return ret;
}
void SignalHandler(int sig) {
Memory::Print( "Got termination signal (" + std::to_string(sig) +")");
BeamNG::Terminate = true;
}
void BeamNG::EntryPoint() {
RCVQueue = std::make_unique<atomic_queue<std::string, 1000>>();
SendQueue = std::make_unique<atomic_queue<std::string, 1000>>();
uint32_t PID = Memory::GetPID();
auto status = MH_Initialize();
if (status != MH_OK)
Memory::Print(std::string("MH Error -> ") + MH_StatusToString(status));
Memory::Print("PID : " + std::to_string(PID));
GELua::FindAddresses();
/*GameBaseAddr = Memory::GetModuleBase(GameModule);
DllBaseAddr = Memory::GetModuleBase(DllModule);*/
signal(SIGINT, SignalHandler);
signal(SIGTERM, SignalHandler);
signal(SIGABRT, SignalHandler);
signal(SIGBREAK, SignalHandler);
RCVQueue = std::make_unique<atomic_queue<std::string, 1000>>();
SendQueue = std::make_unique<atomic_queue<std::string, 1000>>();
uint32_t PID = Memory::GetPID();
auto status = MH_Initialize();
if (status != MH_OK)
Memory::Print(std::string("MH Error -> ") + MH_StatusToString(status));
Memory::Print("PID : " + std::to_string(PID));
GELua::FindAddresses();
/*GameBaseAddr = Memory::GetModuleBase(GameModule);
DllBaseAddr = Memory::GetModuleBase(DllModule);*/
UpdateDetour = std::make_unique<Hook<def::update_function>>(
GELua::update_function, update_D);
UpdateDetour->Enable();
OpenJITDetour = std::make_unique<Hook<def::lua_open_jit>>(
GELua::lua_open_jit, lua_open_jit_D);
OpenJITDetour->Enable();
IPCFromLauncher = std::make_unique<IPC>(PID, 0x1900000);
IPCToLauncher = std::make_unique<IPC>(PID + 1, 0x1900000);
CreateThread(nullptr, 0, LPTHREAD_START_ROUTINE(IPCSender), nullptr, 0,
nullptr);
IPCListener();
OpenJITDetour = std::make_unique<Hook<def::lua_open_jit>>(
GELua::lua_open_jit, lua_open_jit_D);
OpenJITDetour->Enable();
IPCFromLauncher = std::make_unique<IPC>(PID, 0x1900000);
IPCToLauncher = std::make_unique<IPC>(PID + 1, 0x1900000);
CreateThread(nullptr, 0, LPTHREAD_START_ROUTINE(IPCSender), nullptr, 0,
nullptr);
IPCListener();
}
int Core(lua_State* L) {
@ -102,30 +112,25 @@ void BeamNG::SendIPC(const std::string& Data) {
}
void BeamNG::IPCListener() {
int TimeOuts = 0;
while (TimeOuts < 20) {
while (!Terminate) {
IPCFromLauncher->receive();
if (!IPCFromLauncher->receive_timed_out()) {
TimeOuts = 0;
RCVQueue->push(IPCFromLauncher->msg());
IPCFromLauncher->confirm_receive();
} else TimeOuts++;
}
}
Memory::Print("IPC Listener System shutting down (timeout)");
Memory::Print("IPC Listener System shutting down (terminate)");
}
uint32_t BeamNG::IPCSender(void* LP) {
std::string result;
int TimeOuts = 0;
while (TimeOuts < 20) {
while (!Terminate) {
if (SendQueue->try_pop(result)) {
IPCToLauncher->send(result);
if (!IPCToLauncher->send_timed_out()) TimeOuts = 0;
else TimeOuts++;
} else {
Sleep(1); //TODO look into possibly have it wake up on a new message instead
}
}
Memory::Print("IPC Sender System shutting down (timeout)");
Memory::Print("IPC Sender System shutting down (terminate)");
return 0;
}