1
1
Fork 0
forked from suyu/suyu

buffer_queue: Protect queue_sequence list access with a mutex

fixes a data race as this is an unprotected variable manipulated by multiple threads
This commit is contained in:
ameerj 2021-01-04 01:36:41 -05:00
parent 4801f4250d
commit 6b354ccaee
2 changed files with 21 additions and 13 deletions

View file

@ -26,10 +26,10 @@ void BufferQueue::SetPreallocatedBuffer(u32 slot, const IGBPBuffer& igbp_buffer)
LOG_WARNING(Service, "Adding graphics buffer {}", slot); LOG_WARNING(Service, "Adding graphics buffer {}", slot);
{ {
std::unique_lock lock{queue_mutex}; std::unique_lock lock{free_buffers_mutex};
free_buffers.push_back(slot); free_buffers.push_back(slot);
} }
condition.notify_one(); free_buffers_condition.notify_one();
buffers[slot] = { buffers[slot] = {
.slot = slot, .slot = slot,
@ -48,8 +48,8 @@ std::optional<std::pair<u32, Service::Nvidia::MultiFence*>> BufferQueue::Dequeue
u32 height) { u32 height) {
// Wait for first request before trying to dequeue // Wait for first request before trying to dequeue
{ {
std::unique_lock lock{queue_mutex}; std::unique_lock lock{free_buffers_mutex};
condition.wait(lock, [this] { return !free_buffers.empty() || !is_connect; }); free_buffers_condition.wait(lock, [this] { return !free_buffers.empty() || !is_connect; });
} }
if (!is_connect) { if (!is_connect) {
@ -58,7 +58,7 @@ std::optional<std::pair<u32, Service::Nvidia::MultiFence*>> BufferQueue::Dequeue
return std::nullopt; return std::nullopt;
} }
std::unique_lock lock{queue_mutex}; std::unique_lock lock{free_buffers_mutex};
auto f_itr = free_buffers.begin(); auto f_itr = free_buffers.begin();
auto slot = buffers.size(); auto slot = buffers.size();
@ -100,6 +100,7 @@ void BufferQueue::QueueBuffer(u32 slot, BufferTransformFlags transform,
buffers[slot].crop_rect = crop_rect; buffers[slot].crop_rect = crop_rect;
buffers[slot].swap_interval = swap_interval; buffers[slot].swap_interval = swap_interval;
buffers[slot].multi_fence = multi_fence; buffers[slot].multi_fence = multi_fence;
std::unique_lock lock{queue_sequence_mutex};
queue_sequence.push_back(slot); queue_sequence.push_back(slot);
} }
@ -113,15 +114,16 @@ void BufferQueue::CancelBuffer(u32 slot, const Service::Nvidia::MultiFence& mult
buffers[slot].swap_interval = 0; buffers[slot].swap_interval = 0;
{ {
std::unique_lock lock{queue_mutex}; std::unique_lock lock{free_buffers_mutex};
free_buffers.push_back(slot); free_buffers.push_back(slot);
} }
condition.notify_one(); free_buffers_condition.notify_one();
buffer_wait_event.writable->Signal(); buffer_wait_event.writable->Signal();
} }
std::optional<std::reference_wrapper<const BufferQueue::Buffer>> BufferQueue::AcquireBuffer() { std::optional<std::reference_wrapper<const BufferQueue::Buffer>> BufferQueue::AcquireBuffer() {
std::unique_lock lock{queue_sequence_mutex};
std::size_t buffer_slot = buffers.size(); std::size_t buffer_slot = buffers.size();
// Iterate to find a queued buffer matching the requested slot. // Iterate to find a queued buffer matching the requested slot.
while (buffer_slot == buffers.size() && !queue_sequence.empty()) { while (buffer_slot == buffers.size() && !queue_sequence.empty()) {
@ -147,25 +149,29 @@ void BufferQueue::ReleaseBuffer(u32 slot) {
buffers[slot].status = Buffer::Status::Free; buffers[slot].status = Buffer::Status::Free;
{ {
std::unique_lock lock{queue_mutex}; std::unique_lock lock{free_buffers_mutex};
free_buffers.push_back(slot); free_buffers.push_back(slot);
} }
condition.notify_one(); free_buffers_condition.notify_one();
buffer_wait_event.writable->Signal(); buffer_wait_event.writable->Signal();
} }
void BufferQueue::Connect() { void BufferQueue::Connect() {
std::unique_lock lock{queue_sequence_mutex};
queue_sequence.clear(); queue_sequence.clear();
is_connect = true; is_connect = true;
} }
void BufferQueue::Disconnect() { void BufferQueue::Disconnect() {
buffers.fill({}); buffers.fill({});
queue_sequence.clear(); {
std::unique_lock lock{queue_sequence_mutex};
queue_sequence.clear();
}
buffer_wait_event.writable->Signal(); buffer_wait_event.writable->Signal();
is_connect = false; is_connect = false;
condition.notify_one(); free_buffers_condition.notify_one();
} }
u32 BufferQueue::Query(QueryType type) { u32 BufferQueue::Query(QueryType type) {

View file

@ -129,8 +129,10 @@ private:
std::list<u32> queue_sequence; std::list<u32> queue_sequence;
Kernel::EventPair buffer_wait_event; Kernel::EventPair buffer_wait_event;
std::mutex queue_mutex; std::mutex free_buffers_mutex;
std::condition_variable condition; std::condition_variable free_buffers_condition;
std::mutex queue_sequence_mutex;
}; };
} // namespace Service::NVFlinger } // namespace Service::NVFlinger