nvnflinger: fix lost wakeup
This commit is contained in:
		@@ -23,15 +23,17 @@ void BufferQueueCore::NotifyShutdown() {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void BufferQueueCore::SignalDequeueCondition() {
 | 
			
		||||
    dequeue_possible.store(true);
 | 
			
		||||
    dequeue_condition.notify_all();
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool BufferQueueCore::WaitForDequeueCondition() {
 | 
			
		||||
bool BufferQueueCore::WaitForDequeueCondition(std::unique_lock<std::mutex>& lk) {
 | 
			
		||||
    if (is_shutting_down) {
 | 
			
		||||
        return false;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    dequeue_condition.wait(mutex);
 | 
			
		||||
    dequeue_condition.wait(lk, [&] { return dequeue_possible.load(); });
 | 
			
		||||
    dequeue_possible.store(false);
 | 
			
		||||
 | 
			
		||||
    return true;
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -38,7 +38,7 @@ public:
 | 
			
		||||
 | 
			
		||||
private:
 | 
			
		||||
    void SignalDequeueCondition();
 | 
			
		||||
    bool WaitForDequeueCondition();
 | 
			
		||||
    bool WaitForDequeueCondition(std::unique_lock<std::mutex>& lk);
 | 
			
		||||
 | 
			
		||||
    s32 GetMinUndequeuedBufferCountLocked(bool async) const;
 | 
			
		||||
    s32 GetMinMaxBufferCountLocked(bool async) const;
 | 
			
		||||
@@ -60,7 +60,8 @@ private:
 | 
			
		||||
    BufferQueueDefs::SlotsType slots{};
 | 
			
		||||
    std::vector<BufferItem> queue;
 | 
			
		||||
    s32 override_max_buffer_count{};
 | 
			
		||||
    mutable std::condition_variable_any dequeue_condition;
 | 
			
		||||
    std::condition_variable dequeue_condition;
 | 
			
		||||
    std::atomic<bool> dequeue_possible{};
 | 
			
		||||
    const bool use_async_buffer{}; // This is always disabled on HOS
 | 
			
		||||
    bool dequeue_buffer_cannot_block{};
 | 
			
		||||
    PixelFormat default_buffer_format{PixelFormat::Rgba8888};
 | 
			
		||||
 
 | 
			
		||||
@@ -121,8 +121,8 @@ Status BufferQueueProducer::SetBufferCount(s32 buffer_count) {
 | 
			
		||||
    return Status::NoError;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
Status BufferQueueProducer::WaitForFreeSlotThenRelock(bool async, s32* found,
 | 
			
		||||
                                                      Status* return_flags) const {
 | 
			
		||||
Status BufferQueueProducer::WaitForFreeSlotThenRelock(bool async, s32* found, Status* return_flags,
 | 
			
		||||
                                                      std::unique_lock<std::mutex>& lk) const {
 | 
			
		||||
    bool try_again = true;
 | 
			
		||||
 | 
			
		||||
    while (try_again) {
 | 
			
		||||
@@ -214,7 +214,7 @@ Status BufferQueueProducer::WaitForFreeSlotThenRelock(bool async, s32* found,
 | 
			
		||||
                return Status::WouldBlock;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if (!core->WaitForDequeueCondition()) {
 | 
			
		||||
            if (!core->WaitForDequeueCondition(lk)) {
 | 
			
		||||
                // We are no longer running
 | 
			
		||||
                return Status::NoError;
 | 
			
		||||
            }
 | 
			
		||||
@@ -237,7 +237,7 @@ Status BufferQueueProducer::DequeueBuffer(s32* out_slot, Fence* out_fence, bool
 | 
			
		||||
    Status return_flags = Status::NoError;
 | 
			
		||||
    bool attached_by_consumer = false;
 | 
			
		||||
    {
 | 
			
		||||
        std::scoped_lock lock{core->mutex};
 | 
			
		||||
        std::unique_lock lock{core->mutex};
 | 
			
		||||
        core->WaitWhileAllocatingLocked();
 | 
			
		||||
 | 
			
		||||
        if (format == PixelFormat::NoFormat) {
 | 
			
		||||
@@ -248,7 +248,7 @@ Status BufferQueueProducer::DequeueBuffer(s32* out_slot, Fence* out_fence, bool
 | 
			
		||||
        usage |= core->consumer_usage_bit;
 | 
			
		||||
 | 
			
		||||
        s32 found{};
 | 
			
		||||
        Status status = WaitForFreeSlotThenRelock(async, &found, &return_flags);
 | 
			
		||||
        Status status = WaitForFreeSlotThenRelock(async, &found, &return_flags, lock);
 | 
			
		||||
        if (status != Status::NoError) {
 | 
			
		||||
            return status;
 | 
			
		||||
        }
 | 
			
		||||
@@ -400,13 +400,13 @@ Status BufferQueueProducer::AttachBuffer(s32* out_slot,
 | 
			
		||||
        return Status::BadValue;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    std::scoped_lock lock{core->mutex};
 | 
			
		||||
    std::unique_lock lock{core->mutex};
 | 
			
		||||
    core->WaitWhileAllocatingLocked();
 | 
			
		||||
 | 
			
		||||
    Status return_flags = Status::NoError;
 | 
			
		||||
    s32 found{};
 | 
			
		||||
 | 
			
		||||
    const auto status = WaitForFreeSlotThenRelock(false, &found, &return_flags);
 | 
			
		||||
    const auto status = WaitForFreeSlotThenRelock(false, &found, &return_flags, lock);
 | 
			
		||||
    if (status != Status::NoError) {
 | 
			
		||||
        return status;
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
@@ -70,7 +70,8 @@ public:
 | 
			
		||||
private:
 | 
			
		||||
    BufferQueueProducer(const BufferQueueProducer&) = delete;
 | 
			
		||||
 | 
			
		||||
    Status WaitForFreeSlotThenRelock(bool async, s32* found, Status* return_flags) const;
 | 
			
		||||
    Status WaitForFreeSlotThenRelock(bool async, s32* found, Status* return_flags,
 | 
			
		||||
                                     std::unique_lock<std::mutex>& lk) const;
 | 
			
		||||
 | 
			
		||||
    Kernel::KEvent* buffer_wait_event{};
 | 
			
		||||
    Service::KernelHelpers::ServiceContext& service_context;
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user