Make sure task unwaits from tasking directory

Author: maxdev1

kernel/src/kernel/calls/syscall_tasking.cpp

@@ -297,6 +297,12 @@ void syscallAwaitTaskByIdentifier(g_task* task, g_syscall_task_await_by_id* data
 			taskingDirectoryWaitForRegister(data->identifier, task->id);
 		});
 		clockUnwaitForTime(task->id);
+		taskingDirectoryUnwaitForRegister(data->identifier, task->id);
+
+		taskingWait(task, __func__, [data, task]()
+		{
+			clockWaitForTime(task->id, clockGetLocal()->time + 500);
+		});
 	}
 	data->task = target;
 }

kernel/src/kernel/tasking/tasking_directory.cpp

@@ -58,6 +58,16 @@ void taskingDirectoryWaitForRegister(const char* name, g_tid task)
 	mutexRelease(&entryLock);
 }
 
+void taskingDirectoryUnwaitForRegister(const char* name, g_tid task)
+{
+	mutexAcquire(&entryLock);
+
+	auto entry = _taskingDirectoryGetOrCreateEntry(name);
+	waitQueueRemove(&entry->waitQueue, task);
+
+	mutexRelease(&entryLock);
+}
+
 bool taskingDirectoryRegister(const char* name, g_tid tid, g_security_level priority)
 {
 	bool success = false;

kernel/src/kernel/tasking/tasking_directory.hpp

@@ -56,5 +56,6 @@ const char* taskingDirectoryGetIdentifier(g_tid tid);
  * Adds the task to the wait queue for when another task registers with this identifier.
  */
 void taskingDirectoryWaitForRegister(const char* name, g_tid task);
+void taskingDirectoryUnwaitForRegister(const char* name, g_tid task);
 
 #endif