|
|
|
@ -18,41 +18,58 @@ class KernelCore; |
|
|
|
|
|
|
|
class KLightConditionVariable { |
|
|
|
public: |
|
|
|
explicit KLightConditionVariable(KernelCore& kernel_) |
|
|
|
: thread_queue(kernel_), kernel(kernel_) {} |
|
|
|
explicit KLightConditionVariable(KernelCore& kernel_) : kernel{kernel_} {} |
|
|
|
|
|
|
|
void Wait(KLightLock* lock, s64 timeout = -1) { |
|
|
|
WaitImpl(lock, timeout); |
|
|
|
lock->Lock(); |
|
|
|
void Wait(KLightLock* lock, s64 timeout = -1, bool allow_terminating_thread = true) { |
|
|
|
WaitImpl(lock, timeout, allow_terminating_thread); |
|
|
|
} |
|
|
|
|
|
|
|
void Broadcast() { |
|
|
|
KScopedSchedulerLock lk{kernel}; |
|
|
|
while (thread_queue.WakeupFrontThread() != nullptr) { |
|
|
|
// We want to signal all threads, and so should continue waking up until there's nothing |
|
|
|
// to wake. |
|
|
|
|
|
|
|
// Signal all threads. |
|
|
|
for (auto& thread : wait_list) { |
|
|
|
thread.SetState(ThreadState::Runnable); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
private: |
|
|
|
void WaitImpl(KLightLock* lock, s64 timeout) { |
|
|
|
void WaitImpl(KLightLock* lock, s64 timeout, bool allow_terminating_thread) { |
|
|
|
KThread* owner = GetCurrentThreadPointer(kernel); |
|
|
|
|
|
|
|
// Sleep the thread. |
|
|
|
{ |
|
|
|
KScopedSchedulerLockAndSleep lk(kernel, owner, timeout); |
|
|
|
lock->Unlock(); |
|
|
|
KScopedSchedulerLockAndSleep lk{kernel, owner, timeout}; |
|
|
|
|
|
|
|
if (!thread_queue.SleepThread(owner)) { |
|
|
|
if (!allow_terminating_thread && owner->IsTerminationRequested()) { |
|
|
|
lk.CancelSleep(); |
|
|
|
return; |
|
|
|
} |
|
|
|
|
|
|
|
lock->Unlock(); |
|
|
|
|
|
|
|
// Set the thread as waiting. |
|
|
|
GetCurrentThread(kernel).SetState(ThreadState::Waiting); |
|
|
|
|
|
|
|
// Add the thread to the queue. |
|
|
|
wait_list.push_back(GetCurrentThread(kernel)); |
|
|
|
} |
|
|
|
|
|
|
|
// Remove the thread from the wait list. |
|
|
|
{ |
|
|
|
KScopedSchedulerLock sl{kernel}; |
|
|
|
|
|
|
|
wait_list.erase(wait_list.iterator_to(GetCurrentThread(kernel))); |
|
|
|
} |
|
|
|
|
|
|
|
// Cancel the task that the sleep setup. |
|
|
|
kernel.TimeManager().UnscheduleTimeEvent(owner); |
|
|
|
|
|
|
|
// Re-acquire the lock. |
|
|
|
lock->Lock(); |
|
|
|
} |
|
|
|
KThreadQueue thread_queue; |
|
|
|
|
|
|
|
KernelCore& kernel; |
|
|
|
KThread::WaiterList wait_list{}; |
|
|
|
}; |
|
|
|
} // namespace Kernel |