[Runner] Workaround if we can't open the parent process

src/STDInclude.hpp

@@ -17,6 +17,7 @@
 #include <Wininet.h>
 #include <d3d9.h>
 #include <Aclapi.h>
+#include <Psapi.h>
 
 #pragma warning(push)
 #pragma warning(disable: 4996)

src/Worker/Runner.cpp

@@ -18,32 +18,58 @@ namespace Worker
 	{
 		printf("Attaching to parent process %d...\n", this->processId);
 		HANDLE processHandle = OpenProcess(SYNCHRONIZE, FALSE, this->processId);
-		if (!processHandle || processHandle == INVALID_HANDLE_VALUE)
+
+		if (!processHandle || processHandle == INVALID_HANDLE_VALUE && !Runner::isProcessAlive())
 		{
 			printf("Unable to attach to parent process\n");
+			return;
 		}
-		else
+
-		{
 		printf("Successfully attached to parent process\n");
 		printf("Starting worker...\n");
 
 		std::thread workerThread(&Runner::worker, this);
 
+		if (!processHandle || processHandle == INVALID_HANDLE_VALUE)
+		{
+			while(this->isProcessAlive())
+			{
+				std::this_thread::sleep_for(100ms);
+			}
+		}
+		else
+		{
 			WaitForSingleObject(processHandle, INFINITE);
 			CloseHandle(processHandle);
+		}
 
 		printf("Awaiting worker termination...\n");
 		this->terminate = true;
 		if (workerThread.joinable()) workerThread.join();
 		printf("Worker terminated\n");
 	}
-	}
 
 	void Runner::attachHandler(Runner::Handler* handler)
 	{
 		this->handlers[handler->getCommand()] = std::shared_ptr<Runner::Handler>(handler);
 	}
 
+	bool Runner::isProcessAlive()
+	{
+		DWORD aProcesses[1024], cbNeeded;
+		if (!EnumProcesses(aProcesses, sizeof(aProcesses), &cbNeeded)) return false;
+
+		for(DWORD i = 0; i < cbNeeded / sizeof(DWORD); ++i)
+		{
+			if(aProcesses[i] == static_cast<DWORD>(this->processId))
+			{
+				return true;
+			}
+		}
+
+		return false;
+	}
+
 	void Runner::worker()
 	{
 		printf("Worker started\n");

src/Worker/Runner.hpp

@@ -47,6 +47,7 @@ namespace Worker
 
 	private:
 		void worker();
+		bool isProcessAlive();
 
 		int processId;
 		bool terminate;