- fix launcher pinning a thread

- add fall back lua state listener

src/Memory/BeamNG.cpp

@@ -10,10 +10,19 @@
 std::unique_ptr<atomic_queue<std::string, 1000>> RCVQueue, SendQueue;
 
 int BeamNG::lua_open_jit_D(lua_State* State) {
-   Memory::Print("Got lua State");
-   GELua::State = State;
-   RegisterGEFunctions();
-   return OpenJITDetour->Original(State);
+    Memory::Print("Got lua State -> " + Memory::GetHex(reinterpret_cast<uint64_t>(State)));
+    GELua::State = State;
+    RegisterGEFunctions();
+    return OpenJITDetour->Original(State);
+}
+
+uint64_t BeamNG::update_D(lua_State* State) {
+    if(GELua::State != State) {
+        Memory::Print("Got lua State -> " + Memory::GetHex(reinterpret_cast<uint64_t>(State)));
+        GELua::State = State;
+        RegisterGEFunctions();
+    }
+    return UpdateDetour->Original(State);
 }
 
 void BeamNG::EntryPoint() {
@@ -27,6 +36,11 @@ void BeamNG::EntryPoint() {
    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();
@@ -94,7 +108,7 @@ void BeamNG::IPCListener() {
          IPCFromLauncher->confirm_receive();
       } else TimeOuts++;
    }
-   Memory::Print("IPC Listener System shutting down");
+   Memory::Print("IPC Listener System shutting down (timeout)");
 }
 
 uint32_t BeamNG::IPCSender(void* LP) {
@@ -105,8 +119,10 @@ uint32_t BeamNG::IPCSender(void* LP) {
          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");
+   Memory::Print("IPC Sender System shutting down (timeout)");
    return 0;
 }

src/Memory/GELua.cpp

@@ -10,61 +10,60 @@
 const char* GameModule = "BeamNG.drive.x64.exe";
 const char* DllModule  = "libbeamng.x64.dll";
 
-std::string GetHex(uint64_t num) {
-   char buffer[30];
-   sprintf(buffer, "%llx", num);
-   return std::string{buffer};
-}
-
 void GELua::FindAddresses() {
    GELua::State = nullptr;
    auto Base    = Memory::GetModuleBase(GameModule);
    GetTickCount = reinterpret_cast<def::GetTickCount>(
        Memory::FindPattern(GameModule, Patterns::GetTickCount));
    Memory::Print("GetTickCount -> " +
-                 GetHex(reinterpret_cast<uint64_t>(GetTickCount) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(GetTickCount) - Base));
    lua_open_jit = reinterpret_cast<def::lua_open_jit>(
        Memory::FindPattern(GameModule, Patterns::open_jit));
    Memory::Print("lua_open_jit -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_open_jit) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_open_jit) - Base));
    lua_push_fstring = reinterpret_cast<def::lua_push_fstring>(
        Memory::FindPattern(GameModule, Patterns::push_fstring));
    Memory::Print("lua_push_fstring -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_push_fstring) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_push_fstring) - Base));
    lua_get_field = reinterpret_cast<def::lua_get_field>(
        Memory::FindPattern(GameModule, Patterns::get_field));
    Memory::Print("lua_get_field -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_get_field) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_get_field) - Base));
    lua_p_call = reinterpret_cast<def::lua_p_call>(
        Memory::FindPattern(GameModule, Patterns::p_call));
    Memory::Print("lua_p_call -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_p_call) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_p_call) - Base));
    lua_createtable = reinterpret_cast<def::lua_createtable>(
        Memory::FindPattern(GameModule, Patterns::lua_createtable));
    Memory::Print("lua_createtable -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_createtable) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_createtable) - Base));
    lua_pushcclosure = reinterpret_cast<def::lua_pushcclosure>(
        Memory::FindPattern(GameModule, Patterns::lua_pushcclosure));
    Memory::Print("lua_pushcclosure -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_pushcclosure) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_pushcclosure) - Base));
    lua_setfield = reinterpret_cast<def::lua_setfield>(
        Memory::FindPattern(GameModule, Patterns::lua_setfield));
    Memory::Print("lua_setfield -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_setfield) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_setfield) - Base));
    lua_settable = reinterpret_cast<def::lua_settable>(
        Memory::FindPattern(GameModule, Patterns::lua_settable));
    Memory::Print("lua_settable -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_settable) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_settable) - Base));
    lua_tolstring = reinterpret_cast<def::lua_tolstring>(
        Memory::FindPattern(GameModule, Patterns::lua_tolstring));
    Memory::Print("lua_tolstring -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_tolstring) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_tolstring) - Base));
    GEUpdate = reinterpret_cast<def::GEUpdate>(
        Memory::FindPattern(GameModule, Patterns::GEUpdate));
    Memory::Print("GEUpdate -> " +
-                 GetHex(reinterpret_cast<uint64_t>(GEUpdate) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(GEUpdate) - Base));
    lua_settop = reinterpret_cast<def::lua_settop>(
        Memory::FindPattern(GameModule, Patterns::lua_settop));
    Memory::Print("lua_settop -> " +
-                 GetHex(reinterpret_cast<uint64_t>(lua_settop) - Base));
+                         Memory::GetHex(reinterpret_cast<uint64_t>(lua_settop) - Base));
+
+    update_function = reinterpret_cast<def::update_function>(
+            Memory::FindPattern(GameModule, Patterns::update_function));
+    Memory::Print("testupdatefunction -> " +
+                          Memory::GetHex(reinterpret_cast<uint64_t>(update_function) - Base));
 }

src/Memory/Memory.cpp

@@ -83,6 +83,12 @@ uint64_t Memory::FindPattern(const char* module, const char* Pattern[]) {
    return 0;
 }
 
+std::string Memory::GetHex(uint64_t num) {
+    char buffer[30];
+    sprintf(buffer, "%llx", num);
+    return std::string{buffer};
+}
+
 void* operator new(size_t size) {
    return GlobalAlloc(GPTR, size);
 }