From d8ab1b3f563bfd5591a28bcac716a90ebf04e682 Mon Sep 17 00:00:00 2001
From: momo5502 <mauriceheumann@gmail.com>
Date: Sat, 19 Oct 2024 12:21:17 +0200
Subject: [PATCH] Fix NtWaitForSingleObject

---
 src/windows-emulator/syscalls.cpp | 46 +++++++++++++++++++++++++++++--
 1 file changed, 43 insertions(+), 3 deletions(-)

diff --git a/src/windows-emulator/syscalls.cpp b/src/windows-emulator/syscalls.cpp
index 0b76d5b5..5b53edf0 100644
--- a/src/windows-emulator/syscalls.cpp
+++ b/src/windows-emulator/syscalls.cpp
@@ -15,7 +15,8 @@ struct syscall_context
 	windows_emulator& win_emu;
 	x64_emulator& emu;
 	process_context& proc;
-	mutable bool write_status;
+	mutable bool write_status{true};
+	mutable bool retrigger_syscall{false};
 };
 
 namespace
@@ -122,7 +123,7 @@ namespace
 		}
 
 		const auto new_ip = c.emu.read_instruction_pointer();
-		if (initial_ip != new_ip)
+		if (initial_ip != new_ip || c.retrigger_syscall)
 		{
 			c.emu.reg(x64_register::rip, new_ip - 2);
 		}
@@ -175,6 +176,37 @@ namespace
 		}
 	}
 
+	bool is_object_signaled(process_context& c, const handle h)
+	{
+		const auto type = h.value.type;
+
+		switch (type)
+		{
+		case handle_types::event:
+			{
+				const auto* e = c.events.get(h);
+				if (e)
+				{
+					return e->signaled;
+				}
+
+				break;
+			}
+
+		case handle_types::thread:
+			{
+				const auto* t = c.threads.get(h);
+				if (t)
+				{
+					return t->exit_status.has_value();
+				}
+
+				break;
+			}
+		}
+		throw std::runtime_error("Bad object");
+	}
+
 	NTSTATUS handle_NtQueryPerformanceCounter(const syscall_context&,
 	                                          const emulator_object<LARGE_INTEGER> performance_counter,
 	                                          const emulator_object<LARGE_INTEGER> performance_frequency)
@@ -2029,11 +2061,19 @@ namespace
 			return STATUS_NOT_SUPPORTED;
 		}
 
+		if (is_object_signaled(c.proc, h))
+		{
+			return STATUS_WAIT_0;
+		}
+
+		c.write_status = false;
+		c.retrigger_syscall = true;
+
 		c.win_emu.current_thread().await_object = h;
 		c.win_emu.switch_thread = true;
 		c.emu.stop();
 
-		return STATUS_WAIT_0;
+		return STATUS_SUCCESS;
 	}
 
 	NTSTATUS handle_NtTerminateThread(const syscall_context& c, uint64_t thread_handle,