Fix race condition

Fix race condition when renderer was found and we were hooking a library func
merge-requests/28/head
Nemirtingas 2019-10-11 00:51:03 +02:00
parent 894300b6f4
commit 4d99b6affd
3 changed files with 19 additions and 11 deletions

View File

@ -522,6 +522,11 @@ void Renderer_Detector::find_renderer_proc(Renderer_Detector* _this)
std::vector<std::string>::const_iterator it = libraries.begin(); std::vector<std::string>::const_iterator it = libraries.begin();
while (it != libraries.end()) while (it != libraries.end())
{ {
{
std::lock_guard<std::mutex> lock(_this->_found_mutex);
if (_this->_renderer_found)
break;
it = std::find_if(it, libraries.end(), [](std::string const& name) { it = std::find_if(it, libraries.end(), [](std::string const& name) {
auto x = GetModuleHandle(name.c_str()); auto x = GetModuleHandle(name.c_str());
if (x != NULL) if (x != NULL)
@ -535,6 +540,7 @@ void Renderer_Detector::find_renderer_proc(Renderer_Detector* _this)
_this->create_hook(it->c_str()); _this->create_hook(it->c_str());
++it; ++it;
} }
}
std::this_thread::sleep_for(std::chrono::milliseconds(500)); std::this_thread::sleep_for(std::chrono::milliseconds(500));
} }
@ -824,6 +830,7 @@ extern "C" void* dlsym(void* handle, const char* name)
void Renderer_Detector::renderer_found(Base_Hook* hook) void Renderer_Detector::renderer_found(Base_Hook* hook)
{ {
std::lock_guard<std::mutex> lock(_found_mutex);
Hook_Manager& hm = Hook_Manager::Inst(); Hook_Manager& hm = Hook_Manager::Inst();
_renderer_found = true; _renderer_found = true;

View File

@ -16,6 +16,7 @@ class Renderer_Detector
private: private:
// Variables // Variables
std::thread* _hook_thread; std::thread* _hook_thread;
std::mutex _found_mutex;
unsigned int _hook_retries; unsigned int _hook_retries;
bool _renderer_found; // Is the renderer hooked ? bool _renderer_found; // Is the renderer hooked ?
bool _dx9_hooked; bool _dx9_hooked;
@ -70,6 +71,7 @@ class Renderer_Detector
{ {
// Variables // Variables
std::thread* _hook_thread; std::thread* _hook_thread;
std::mutex _found_mutex;
unsigned int _hook_retries; unsigned int _hook_retries;
bool _oglx_hooked; bool _oglx_hooked;
bool _renderer_found; // Is the renderer hooked ? bool _renderer_found; // Is the renderer hooked ?

View File

@ -165,7 +165,6 @@ UINT WINAPI Windows_Hook::MyGetRawInputData(HRAWINPUT hRawInput, UINT uiCommand,
RawMouseEvent(*reinterpret_cast<RAWINPUT*>(pData)); RawMouseEvent(*reinterpret_cast<RAWINPUT*>(pData));
//memset(pData, 0, *pcbSize);
*pcbSize = 0; *pcbSize = 0;
return 0; return 0;