blob: 55034b39d52b36658123d7e2da7c5e2c30de7210 [file] [log] [blame]
Corey Tabaka52ea25c2017-09-13 18:02:48 -07001#include <sys/epoll.h>
2#include <sys/eventfd.h>
Alex Vakulenkoe4eec202017-01-27 14:41:04 -08003#include <sys/poll.h>
Alex Vakulenkoe4eec202017-01-27 14:41:04 -08004
5#include <algorithm>
6#include <atomic>
7#include <thread>
8
Fan Xu74df4902018-09-20 16:40:51 -07009#include <log/log.h>
Fan Xu74df4902018-09-20 16:40:51 -070010#include <private/dvr/buffer_channel.h>
Tianyu0c6b7ff2018-10-03 11:30:04 -070011#include <private/dvr/bufferhub_rpc.h>
Fan Xu74df4902018-09-20 16:40:51 -070012#include <private/dvr/consumer_channel.h>
13#include <private/dvr/producer_channel.h>
14#include <sync/sync.h>
15#include <utils/Trace.h>
Alex Vakulenkoe4eec202017-01-27 14:41:04 -080016
17using android::pdx::BorrowedHandle;
Corey Tabakacd52dd92017-04-07 18:03:57 -070018using android::pdx::ErrorStatus;
Alex Vakulenkoe4eec202017-01-27 14:41:04 -080019using android::pdx::Message;
20using android::pdx::RemoteChannelHandle;
Corey Tabakacd52dd92017-04-07 18:03:57 -070021using android::pdx::Status;
Alex Vakulenkoe4eec202017-01-27 14:41:04 -080022using android::pdx::rpc::BufferWrapper;
23using android::pdx::rpc::DispatchRemoteMethod;
24using android::pdx::rpc::WrapBuffer;
25
26namespace android {
27namespace dvr {
28
Jiwen 'Steve' Cai0728fa92018-04-24 19:03:14 -070029ProducerChannel::ProducerChannel(BufferHubService* service, int buffer_id,
30 int channel_id, IonBuffer buffer,
31 IonBuffer metadata_buffer,
32 size_t user_metadata_size, int* error)
33 : BufferHubChannel(service, buffer_id, channel_id, kProducerType),
34 buffer_(std::move(buffer)),
35 metadata_buffer_(std::move(metadata_buffer)),
36 user_metadata_size_(user_metadata_size),
37 metadata_buf_size_(BufferHubDefs::kMetadataHeaderSize +
38 user_metadata_size) {
39 if (!buffer_.IsValid()) {
40 ALOGE("ProducerChannel::ProducerChannel: Invalid buffer.");
41 *error = -EINVAL;
42 return;
43 }
44 if (!metadata_buffer_.IsValid()) {
45 ALOGE("ProducerChannel::ProducerChannel: Invalid metadata buffer.");
46 *error = -EINVAL;
47 return;
48 }
49
50 *error = InitializeBuffer();
51}
52
Alex Vakulenkoe4eec202017-01-27 14:41:04 -080053ProducerChannel::ProducerChannel(BufferHubService* service, int channel_id,
Corey Tabakacd52dd92017-04-07 18:03:57 -070054 uint32_t width, uint32_t height,
Hendrik Wagenaar108e84f2017-05-07 22:19:17 -070055 uint32_t layer_count, uint32_t format,
Corey Tabaka52ea25c2017-09-13 18:02:48 -070056 uint64_t usage, size_t user_metadata_size,
Hendrik Wagenaar108e84f2017-05-07 22:19:17 -070057 int* error)
Alex Vakulenkoe4eec202017-01-27 14:41:04 -080058 : BufferHubChannel(service, channel_id, channel_id, kProducerType),
59 pending_consumers_(0),
Alex Vakulenkoe4eec202017-01-27 14:41:04 -080060 producer_owns_(true),
Corey Tabaka52ea25c2017-09-13 18:02:48 -070061 user_metadata_size_(user_metadata_size),
62 metadata_buf_size_(BufferHubDefs::kMetadataHeaderSize +
63 user_metadata_size) {
64 if (int ret = buffer_.Alloc(width, height, layer_count, format, usage)) {
Hendrik Wagenaar4d3590f2017-05-06 22:36:04 -070065 ALOGE("ProducerChannel::ProducerChannel: Failed to allocate buffer: %s",
66 strerror(-ret));
67 *error = ret;
68 return;
Alex Vakulenkoe4eec202017-01-27 14:41:04 -080069 }
70
Corey Tabaka52ea25c2017-09-13 18:02:48 -070071 if (int ret = metadata_buffer_.Alloc(metadata_buf_size_, /*height=*/1,
72 /*layer_count=*/1,
73 BufferHubDefs::kMetadataFormat,
74 BufferHubDefs::kMetadataUsage)) {
75 ALOGE("ProducerChannel::ProducerChannel: Failed to allocate metadata: %s",
76 strerror(-ret));
77 *error = ret;
78 return;
79 }
80
Jiwen 'Steve' Cai0728fa92018-04-24 19:03:14 -070081 *error = InitializeBuffer();
82}
83
84int ProducerChannel::InitializeBuffer() {
Corey Tabaka52ea25c2017-09-13 18:02:48 -070085 void* metadata_ptr = nullptr;
86 if (int ret = metadata_buffer_.Lock(BufferHubDefs::kMetadataUsage, /*x=*/0,
87 /*y=*/0, metadata_buf_size_,
88 /*height=*/1, &metadata_ptr)) {
89 ALOGE("ProducerChannel::ProducerChannel: Failed to lock metadata.");
Jiwen 'Steve' Cai0728fa92018-04-24 19:03:14 -070090 return ret;
Corey Tabaka52ea25c2017-09-13 18:02:48 -070091 }
92 metadata_header_ =
93 reinterpret_cast<BufferHubDefs::MetadataHeader*>(metadata_ptr);
94
95 // Using placement new here to reuse shared memory instead of new allocation
96 // and also initialize the value to zero.
97 buffer_state_ =
98 new (&metadata_header_->buffer_state) std::atomic<uint64_t>(0);
Tianyu0c6b7ff2018-10-03 11:30:04 -070099 fence_state_ = new (&metadata_header_->fence_state) std::atomic<uint64_t>(0);
Tianyu67053492018-10-04 14:00:22 -0700100 active_clients_bit_mask_ =
101 new (&metadata_header_->active_clients_bit_mask) std::atomic<uint64_t>(0);
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700102
Tianyu Jiang63dd7c32018-10-30 18:35:06 -0700103 // Producer channel is never created after consumer channel, and one buffer
104 // only have one fixed producer for now. Thus, it is correct to assume
105 // producer state bit is kFirstClientBitMask for now.
106 active_clients_bit_mask_->store(BufferHubDefs::kFirstClientBitMask,
107 std::memory_order_release);
108
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700109 acquire_fence_fd_.Reset(epoll_create1(EPOLL_CLOEXEC));
110 release_fence_fd_.Reset(epoll_create1(EPOLL_CLOEXEC));
111 if (!acquire_fence_fd_ || !release_fence_fd_) {
112 ALOGE("ProducerChannel::ProducerChannel: Failed to create shared fences.");
Jiwen 'Steve' Cai0728fa92018-04-24 19:03:14 -0700113 return -EIO;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700114 }
115
116 dummy_fence_fd_.Reset(eventfd(0, EFD_CLOEXEC | EFD_NONBLOCK));
117 if (!dummy_fence_fd_) {
118 ALOGE("ProducerChannel::ProducerChannel: Failed to create dummy fences.");
Jiwen 'Steve' Cai0728fa92018-04-24 19:03:14 -0700119 return EIO;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700120 }
121
122 epoll_event event;
123 event.events = 0;
124 event.data.u64 = 0ULL;
125 if (epoll_ctl(release_fence_fd_.Get(), EPOLL_CTL_ADD, dummy_fence_fd_.Get(),
126 &event) < 0) {
127 ALOGE(
128 "ProducerChannel::ProducerChannel: Failed to modify the shared "
129 "release fence to include the dummy fence: %s",
130 strerror(errno));
Jiwen 'Steve' Cai0728fa92018-04-24 19:03:14 -0700131 return -EIO;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700132 }
133
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800134 // Success.
Jiwen 'Steve' Cai0728fa92018-04-24 19:03:14 -0700135 return 0;
136}
137
138std::unique_ptr<ProducerChannel> ProducerChannel::Create(
139 BufferHubService* service, int buffer_id, int channel_id, IonBuffer buffer,
140 IonBuffer metadata_buffer, size_t user_metadata_size) {
141 int error = 0;
142 std::unique_ptr<ProducerChannel> producer(new ProducerChannel(
143 service, buffer_id, channel_id, std::move(buffer),
144 std::move(metadata_buffer), user_metadata_size, &error));
145
146 if (error < 0)
147 return nullptr;
148 else
149 return producer;
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800150}
151
Corey Tabakacd52dd92017-04-07 18:03:57 -0700152Status<std::shared_ptr<ProducerChannel>> ProducerChannel::Create(
153 BufferHubService* service, int channel_id, uint32_t width, uint32_t height,
Hendrik Wagenaar108e84f2017-05-07 22:19:17 -0700154 uint32_t layer_count, uint32_t format, uint64_t usage,
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700155 size_t user_metadata_size) {
Corey Tabakacd52dd92017-04-07 18:03:57 -0700156 int error;
Jiwen 'Steve' Cai0057fdd2017-05-02 11:21:18 -0700157 std::shared_ptr<ProducerChannel> producer(
Hendrik Wagenaar108e84f2017-05-07 22:19:17 -0700158 new ProducerChannel(service, channel_id, width, height, layer_count,
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700159 format, usage, user_metadata_size, &error));
Corey Tabakacd52dd92017-04-07 18:03:57 -0700160 if (error < 0)
161 return ErrorStatus(-error);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800162 else
Corey Tabakacd52dd92017-04-07 18:03:57 -0700163 return {std::move(producer)};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800164}
165
166ProducerChannel::~ProducerChannel() {
Corey Tabaka3079cb72017-01-19 15:07:26 -0800167 ALOGD_IF(TRACE,
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700168 "ProducerChannel::~ProducerChannel: channel_id=%d buffer_id=%d "
169 "state=%" PRIx64 ".",
Tianyu Jiange2cdec92018-10-30 17:40:58 -0700170 channel_id(), buffer_id(),
171 buffer_state_->load(std::memory_order_acquire));
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700172 for (auto consumer : consumer_channels_) {
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800173 consumer->OnProducerClosed();
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700174 }
175 Hangup();
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800176}
177
178BufferHubChannel::BufferInfo ProducerChannel::GetBufferInfo() const {
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700179 // Derive the mask of signaled buffers in this producer / consumer set.
Tianyu Jiang63dd7c32018-10-30 18:35:06 -0700180 uint64_t signaled_mask = signaled() ? BufferHubDefs::kFirstClientBitMask : 0;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700181 for (const ConsumerChannel* consumer : consumer_channels_) {
Tianyu Jiang83a991f2018-10-30 16:59:29 -0700182 signaled_mask |= consumer->signaled() ? consumer->client_state_mask() : 0;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700183 }
184
Hendrik Wagenaar4d3590f2017-05-06 22:36:04 -0700185 return BufferInfo(buffer_id(), consumer_channels_.size(), buffer_.width(),
Hendrik Wagenaar108e84f2017-05-07 22:19:17 -0700186 buffer_.height(), buffer_.layer_count(), buffer_.format(),
Tianyu Jiange2cdec92018-10-30 17:40:58 -0700187 buffer_.usage(), pending_consumers_,
188 buffer_state_->load(std::memory_order_acquire),
Jiwen 'Steve' Cai2f260332018-02-15 18:39:47 -0800189 signaled_mask, metadata_header_->queue_index);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800190}
191
192void ProducerChannel::HandleImpulse(Message& message) {
193 ATRACE_NAME("ProducerChannel::HandleImpulse");
194 switch (message.GetOp()) {
195 case BufferHubRPC::ProducerGain::Opcode:
196 OnProducerGain(message);
197 break;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700198 case BufferHubRPC::ProducerPost::Opcode:
199 OnProducerPost(message, {});
200 break;
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800201 }
202}
203
204bool ProducerChannel::HandleMessage(Message& message) {
205 ATRACE_NAME("ProducerChannel::HandleMessage");
206 switch (message.GetOp()) {
207 case BufferHubRPC::GetBuffer::Opcode:
208 DispatchRemoteMethod<BufferHubRPC::GetBuffer>(
209 *this, &ProducerChannel::OnGetBuffer, message);
210 return true;
211
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800212 case BufferHubRPC::NewConsumer::Opcode:
213 DispatchRemoteMethod<BufferHubRPC::NewConsumer>(
214 *this, &ProducerChannel::OnNewConsumer, message);
215 return true;
216
217 case BufferHubRPC::ProducerPost::Opcode:
218 DispatchRemoteMethod<BufferHubRPC::ProducerPost>(
219 *this, &ProducerChannel::OnProducerPost, message);
220 return true;
221
222 case BufferHubRPC::ProducerGain::Opcode:
223 DispatchRemoteMethod<BufferHubRPC::ProducerGain>(
224 *this, &ProducerChannel::OnProducerGain, message);
225 return true;
226
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800227 default:
228 return false;
229 }
230}
231
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700232BufferDescription<BorrowedHandle> ProducerChannel::GetBuffer(
Tianyu Jiang7e204b72018-10-26 15:39:18 -0700233 uint64_t client_state_mask) {
Jiwen 'Steve' Cai57ae3ee2018-05-03 17:51:52 -0700234 return {buffer_,
235 metadata_buffer_,
236 buffer_id(),
237 channel_id(),
Tianyu Jiang7e204b72018-10-26 15:39:18 -0700238 client_state_mask,
Jiwen 'Steve' Cai57ae3ee2018-05-03 17:51:52 -0700239 acquire_fence_fd_.Borrow(),
240 release_fence_fd_.Borrow()};
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700241}
242
243Status<BufferDescription<BorrowedHandle>> ProducerChannel::OnGetBuffer(
Corey Tabakad53870c2017-07-06 18:04:27 -0700244 Message& /*message*/) {
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800245 ATRACE_NAME("ProducerChannel::OnGetBuffer");
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700246 ALOGD_IF(TRACE, "ProducerChannel::OnGetBuffer: buffer=%d, state=%" PRIx64 ".",
Tianyu Jiange2cdec92018-10-30 17:40:58 -0700247 buffer_id(), buffer_state_->load(std::memory_order_acquire));
Tianyu Jiang63dd7c32018-10-30 18:35:06 -0700248 return {GetBuffer(BufferHubDefs::kFirstClientBitMask)};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800249}
250
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800251Status<uint64_t> ProducerChannel::CreateConsumerStateMask() {
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700252 // Try find the next consumer state bit which has not been claimed by any
253 // consumer yet.
Tianyu67053492018-10-04 14:00:22 -0700254 // memory_order_acquire is chosen here because all writes in other threads
255 // that release active_clients_bit_mask_ need to be visible here.
256 uint64_t current_active_clients_bit_mask =
257 active_clients_bit_mask_->load(std::memory_order_acquire);
Tianyuf669f6a2018-10-10 15:34:32 -0700258 uint64_t consumer_state_mask =
259 BufferHubDefs::FindNextAvailableClientStateMask(
260 current_active_clients_bit_mask | orphaned_consumer_bit_mask_);
261 if (consumer_state_mask == 0ULL) {
262 ALOGE("%s: reached the maximum mumber of consumers per producer: 63.",
263 __FUNCTION__);
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700264 return ErrorStatus(E2BIG);
265 }
Tianyu67053492018-10-04 14:00:22 -0700266 uint64_t updated_active_clients_bit_mask =
Tianyuf669f6a2018-10-10 15:34:32 -0700267 current_active_clients_bit_mask | consumer_state_mask;
Tianyu67053492018-10-04 14:00:22 -0700268 // Set the updated value only if the current value stays the same as what was
269 // read before. If the comparison succeeds, update the value without
270 // reordering anything before or after this read-modify-write in the current
271 // thread, and the modification will be visible in other threads that acquire
272 // active_clients_bit_mask_. If the comparison fails, load the result of
273 // all writes from all threads to updated_active_clients_bit_mask.
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800274 // Keep on finding the next available slient state mask until succeed or out
275 // of memory.
276 while (!active_clients_bit_mask_->compare_exchange_weak(
277 current_active_clients_bit_mask, updated_active_clients_bit_mask,
278 std::memory_order_acq_rel, std::memory_order_acquire)) {
Tianyuf669f6a2018-10-10 15:34:32 -0700279 ALOGE("%s: Current active clients bit mask is changed to %" PRIx64
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800280 ", which was expected to be %" PRIx64
281 ". Trying to generate a new client state mask to resolve race "
282 "condition.",
Tianyuf669f6a2018-10-10 15:34:32 -0700283 __FUNCTION__, updated_active_clients_bit_mask,
284 current_active_clients_bit_mask);
285 consumer_state_mask = BufferHubDefs::FindNextAvailableClientStateMask(
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800286 current_active_clients_bit_mask | orphaned_consumer_bit_mask_);
Tianyuf669f6a2018-10-10 15:34:32 -0700287 if (consumer_state_mask == 0ULL) {
288 ALOGE("%s: reached the maximum mumber of consumers per producer: %d.",
289 __FUNCTION__, (BufferHubDefs::kMaxNumberOfClients - 1));
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800290 return ErrorStatus(E2BIG);
291 }
292 updated_active_clients_bit_mask =
Tianyuf669f6a2018-10-10 15:34:32 -0700293 current_active_clients_bit_mask | consumer_state_mask;
Tianyu67053492018-10-04 14:00:22 -0700294 }
295
Tianyuf669f6a2018-10-10 15:34:32 -0700296 return {consumer_state_mask};
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800297}
298
299void ProducerChannel::RemoveConsumerClientMask(uint64_t consumer_state_mask) {
300 // Clear up the buffer state and fence state in case there is already
301 // something there due to possible race condition between producer post and
302 // consumer failed to create channel.
303 buffer_state_->fetch_and(~consumer_state_mask, std::memory_order_release);
304 fence_state_->fetch_and(~consumer_state_mask, std::memory_order_release);
305
306 // Restore the consumer state bit and make it visible in other threads that
307 // acquire the active_clients_bit_mask_.
308 active_clients_bit_mask_->fetch_and(~consumer_state_mask,
309 std::memory_order_release);
310}
311
312Status<RemoteChannelHandle> ProducerChannel::CreateConsumer(
313 Message& message, uint64_t consumer_state_mask) {
Tianyuf669f6a2018-10-10 15:34:32 -0700314 ATRACE_NAME(__FUNCTION__);
315 ALOGD_IF(TRACE, "%s: buffer_id=%d, producer_owns=%d", __FUNCTION__,
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800316 buffer_id(), producer_owns_);
317
318 int channel_id;
319 auto status = message.PushChannel(0, nullptr, &channel_id);
320 if (!status) {
Tianyuf669f6a2018-10-10 15:34:32 -0700321 ALOGE("%s: Failed to push consumer channel: %s", __FUNCTION__,
322 status.GetErrorMessage().c_str());
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800323 RemoveConsumerClientMask(consumer_state_mask);
324 return ErrorStatus(ENOMEM);
325 }
326
327 auto consumer = std::make_shared<ConsumerChannel>(
328 service(), buffer_id(), channel_id, consumer_state_mask,
329 shared_from_this());
Alex Vakulenkof0a7bd02017-03-31 18:06:19 -0700330 const auto channel_status = service()->SetChannel(channel_id, consumer);
331 if (!channel_status) {
Tianyuf669f6a2018-10-10 15:34:32 -0700332 ALOGE("%s: failed to set new consumer channel: %s.", __FUNCTION__,
333 channel_status.GetErrorMessage().c_str());
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800334 RemoveConsumerClientMask(consumer_state_mask);
Corey Tabakacd52dd92017-04-07 18:03:57 -0700335 return ErrorStatus(ENOMEM);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800336 }
337
Tianyu Jiangca9201a2018-11-06 12:40:33 -0800338 uint64_t current_buffer_state =
339 buffer_state_->load(std::memory_order_acquire);
Tianyuf669f6a2018-10-10 15:34:32 -0700340 if (BufferHubDefs::IsBufferReleased(current_buffer_state) ||
341 BufferHubDefs::AnyClientGained(current_buffer_state)) {
342 return {status.take()};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800343 }
344
Tianyuf669f6a2018-10-10 15:34:32 -0700345 // Signal the new consumer when adding it to a posted producer.
346 bool update_buffer_state = true;
347 if (!BufferHubDefs::IsClientPosted(current_buffer_state,
348 consumer_state_mask)) {
349 uint64_t updated_buffer_state =
350 current_buffer_state ^
351 (consumer_state_mask & BufferHubDefs::kHighBitsMask);
352 while (!buffer_state_->compare_exchange_weak(
353 current_buffer_state, updated_buffer_state, std::memory_order_acq_rel,
354 std::memory_order_acquire)) {
355 ALOGI(
356 "%s: Failed to post to the new consumer. "
357 "Current buffer state was changed to %" PRIx64
358 " when trying to acquire the buffer and modify the buffer state to "
359 "%" PRIx64
360 ". About to try again if the buffer is still not gained nor fully "
361 "released.",
362 __FUNCTION__, current_buffer_state, updated_buffer_state);
363 if (BufferHubDefs::IsBufferReleased(current_buffer_state) ||
364 BufferHubDefs::AnyClientGained(current_buffer_state)) {
365 ALOGI("%s: buffer is gained or fully released, state=%" PRIx64 ".",
366 __FUNCTION__, current_buffer_state);
367 update_buffer_state = false;
368 break;
369 }
370 updated_buffer_state =
371 current_buffer_state ^
372 (consumer_state_mask & BufferHubDefs::kHighBitsMask);
373 }
374 }
375 if (update_buffer_state && consumer->OnProducerPosted())
376 pending_consumers_++;
377
Corey Tabakacd52dd92017-04-07 18:03:57 -0700378 return {status.take()};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800379}
380
Corey Tabakacd52dd92017-04-07 18:03:57 -0700381Status<RemoteChannelHandle> ProducerChannel::OnNewConsumer(Message& message) {
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800382 ATRACE_NAME("ProducerChannel::OnNewConsumer");
383 ALOGD_IF(TRACE, "ProducerChannel::OnNewConsumer: buffer_id=%d", buffer_id());
Tianyu Jiang50f7ad12018-11-08 11:13:35 -0800384 auto status = CreateConsumerStateMask();
385 if (!status.ok()) {
386 return status.error_status();
387 }
388 return CreateConsumer(message, /*consumer_state_mask=*/status.get());
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800389}
390
Tianyu0c6b7ff2018-10-03 11:30:04 -0700391Status<void> ProducerChannel::OnProducerPost(Message&,
392 LocalFence acquire_fence) {
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800393 ATRACE_NAME("ProducerChannel::OnProducerPost");
394 ALOGD_IF(TRACE, "ProducerChannel::OnProducerPost: buffer_id=%d", buffer_id());
395 if (!producer_owns_) {
396 ALOGE("ProducerChannel::OnProducerPost: Not in gained state!");
Corey Tabakacd52dd92017-04-07 18:03:57 -0700397 return ErrorStatus(EBUSY);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800398 }
399
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700400 epoll_event event;
401 event.events = 0;
402 event.data.u64 = 0ULL;
403 int ret = epoll_ctl(release_fence_fd_.Get(), EPOLL_CTL_MOD,
404 dummy_fence_fd_.Get(), &event);
405 ALOGE_IF(ret < 0,
Tianyu0c6b7ff2018-10-03 11:30:04 -0700406 "ProducerChannel::OnProducerPost: Failed to modify the shared "
407 "release fence to include the dummy fence: %s",
408 strerror(errno));
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700409
410 eventfd_t dummy_fence_count = 0ULL;
411 if (eventfd_read(dummy_fence_fd_.Get(), &dummy_fence_count) < 0) {
412 const int error = errno;
413 if (error != EAGAIN) {
414 ALOGE(
415 "ProducerChannel::ProducerChannel: Failed to read dummy fence, "
416 "error: %s",
417 strerror(error));
418 return ErrorStatus(error);
419 }
Corey Tabaka8a4e6a92017-04-20 13:42:02 -0700420 }
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800421
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700422 ALOGW_IF(dummy_fence_count > 0,
423 "ProducerChannel::ProducerChannel: %" PRIu64
424 " dummy fence(s) was signaled during last release/gain cycle "
425 "buffer_id=%d.",
426 dummy_fence_count, buffer_id());
427
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800428 post_fence_ = std::move(acquire_fence);
429 producer_owns_ = false;
430
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700431 // Signal any interested consumers. If there are none, the buffer will stay
432 // in posted state until a consumer comes online. This behavior guarantees
433 // that no frame is silently dropped.
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800434 pending_consumers_ = 0;
435 for (auto consumer : consumer_channels_) {
436 if (consumer->OnProducerPosted())
437 pending_consumers_++;
438 }
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800439 ALOGD_IF(TRACE, "ProducerChannel::OnProducerPost: %d pending consumers",
440 pending_consumers_);
441
Corey Tabakacd52dd92017-04-07 18:03:57 -0700442 return {};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800443}
444
Corey Tabakad53870c2017-07-06 18:04:27 -0700445Status<LocalFence> ProducerChannel::OnProducerGain(Message& /*message*/) {
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800446 ATRACE_NAME("ProducerChannel::OnGain");
447 ALOGD_IF(TRACE, "ProducerChannel::OnGain: buffer_id=%d", buffer_id());
448 if (producer_owns_) {
449 ALOGE("ProducerChanneL::OnGain: Already in gained state: channel=%d",
450 channel_id());
Tianyuf669f6a2018-10-10 15:34:32 -0700451 // TODO(b/119331650): remove this if check if producer_owns_ is removed.
452 // return ErrorStatus(EALREADY);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800453 }
454
455 // There are still pending consumers, return busy.
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700456 if (pending_consumers_ > 0) {
457 ALOGE(
458 "ProducerChannel::OnGain: Producer (id=%d) is gaining a buffer that "
459 "still has %d pending consumer(s).",
460 buffer_id(), pending_consumers_);
Tianyuf669f6a2018-10-10 15:34:32 -0700461 // TODO(b/77153033): add gain_posted_buffer to the impulse args, and
462 // return busy if the function does not allow gaining posted buffer.
463 // TODO(b/119331650): remove this if check if pending_consumers_ is removed.
464 // return ErrorStatus(EBUSY);
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700465 }
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800466
467 ClearAvailable();
468 producer_owns_ = true;
Alex Vakulenko052f3ae2017-03-31 09:10:43 -0700469 post_fence_.close();
Corey Tabakacd52dd92017-04-07 18:03:57 -0700470 return {std::move(returned_fence_)};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800471}
472
Fan Xub0eec512018-10-30 11:33:15 -0700473// TODO(b/112338294) Keep here for reference. Remove it after new logic is
474// written.
475/* Status<RemoteChannelHandle> ProducerChannel::OnProducerDetach(
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700476 Message& message) {
477 ATRACE_NAME("ProducerChannel::OnProducerDetach");
478 ALOGD_IF(TRACE, "ProducerChannel::OnProducerDetach: buffer_id=%d",
479 buffer_id());
480
Tianyu Jiange2cdec92018-10-30 17:40:58 -0700481 uint64_t buffer_state = buffer_state_->load(std::memory_order_acquire);
Tianyuf669f6a2018-10-10 15:34:32 -0700482 if (!BufferHubDefs::IsClientGained(
483 buffer_state, BufferHubDefs::kFirstClientStateMask)) {
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700484 // Can only detach a BufferProducer when it's in gained state.
485 ALOGW(
Tianyuf669f6a2018-10-10 15:34:32 -0700486 "ProducerChannel::OnProducerDetach: The buffer (id=%d, state=%"
487 PRIx64
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700488 ") is not in gained state.",
489 buffer_id(), buffer_state);
490 return {};
491 }
492
493 int channel_id;
494 auto status = message.PushChannel(0, nullptr, &channel_id);
495 if (!status) {
496 ALOGE(
497 "ProducerChannel::OnProducerDetach: Failed to push detached buffer "
498 "channel: %s",
499 status.GetErrorMessage().c_str());
500 return ErrorStatus(ENOMEM);
501 }
502
503 // Make sure we unlock the buffer.
504 if (int ret = metadata_buffer_.Unlock()) {
505 ALOGE("ProducerChannel::OnProducerDetach: Failed to unlock metadata.");
506 return ErrorStatus(-ret);
507 };
508
Jiwen 'Steve' Cai9004b8c2018-10-03 18:52:23 -0700509 std::unique_ptr<BufferChannel> channel =
510 BufferChannel::Create(service(), buffer_id(), channel_id,
511 std::move(buffer_), user_metadata_size_);
Jiwen 'Steve' Caia8049a22018-03-28 15:14:02 -0700512 if (!channel) {
513 ALOGE("ProducerChannel::OnProducerDetach: Invalid buffer.");
514 return ErrorStatus(EINVAL);
515 }
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700516
Jiwen 'Steve' Caia8049a22018-03-28 15:14:02 -0700517 const auto channel_status =
518 service()->SetChannel(channel_id, std::move(channel));
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700519 if (!channel_status) {
Tianyuf669f6a2018-10-10 15:34:32 -0700520 // Technically, this should never fail, as we just pushed the channel.
521 // Note that LOG_FATAL will be stripped out in non-debug build.
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700522 LOG_FATAL(
Tianyuf669f6a2018-10-10 15:34:32 -0700523 "ProducerChannel::OnProducerDetach: Failed to set new detached "
524 "buffer channel: %s.", channel_status.GetErrorMessage().c_str());
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700525 }
526
527 return status;
Fan Xub0eec512018-10-30 11:33:15 -0700528} */
Jiwen 'Steve' Cai23c1a732018-03-12 12:16:47 -0700529
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700530Status<LocalFence> ProducerChannel::OnConsumerAcquire(Message& /*message*/) {
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800531 ATRACE_NAME("ProducerChannel::OnConsumerAcquire");
532 ALOGD_IF(TRACE, "ProducerChannel::OnConsumerAcquire: buffer_id=%d",
533 buffer_id());
534 if (producer_owns_) {
535 ALOGE("ProducerChannel::OnConsumerAcquire: Not in posted state!");
Corey Tabakacd52dd92017-04-07 18:03:57 -0700536 return ErrorStatus(EBUSY);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800537 }
538
Tianyuf669f6a2018-10-10 15:34:32 -0700539 // Return a borrowed fd to avoid unnecessary duplication of the underlying
540 // fd. Serialization just needs to read the handle.
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700541 return {std::move(post_fence_)};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800542}
543
Corey Tabakacd52dd92017-04-07 18:03:57 -0700544Status<void> ProducerChannel::OnConsumerRelease(Message&,
545 LocalFence release_fence) {
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800546 ATRACE_NAME("ProducerChannel::OnConsumerRelease");
547 ALOGD_IF(TRACE, "ProducerChannel::OnConsumerRelease: buffer_id=%d",
548 buffer_id());
549 if (producer_owns_) {
550 ALOGE("ProducerChannel::OnConsumerRelease: Not in acquired state!");
Corey Tabakacd52dd92017-04-07 18:03:57 -0700551 return ErrorStatus(EBUSY);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800552 }
553
554 // Attempt to merge the fences if necessary.
555 if (release_fence) {
556 if (returned_fence_) {
Corey Tabaka3079cb72017-01-19 15:07:26 -0800557 LocalFence merged_fence(sync_merge("bufferhub_merged",
558 returned_fence_.get_fd(),
559 release_fence.get_fd()));
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800560 const int error = errno;
561 if (!merged_fence) {
562 ALOGE("ProducerChannel::OnConsumerRelease: Failed to merge fences: %s",
563 strerror(error));
Corey Tabakacd52dd92017-04-07 18:03:57 -0700564 return ErrorStatus(error);
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800565 }
566 returned_fence_ = std::move(merged_fence);
567 } else {
568 returned_fence_ = std::move(release_fence);
569 }
570 }
571
Tianyu0c6b7ff2018-10-03 11:30:04 -0700572 DecrementPendingConsumers();
Tianyuf669f6a2018-10-10 15:34:32 -0700573 if (pending_consumers_ == 0 && orphaned_consumer_bit_mask_) {
574 ALOGW(
575 "%s: orphaned buffer detected during the this acquire/release cycle: "
576 "id=%d orphaned=0x%" PRIx64 " queue_index=%" PRIx64 ".",
577 __FUNCTION__, buffer_id(), orphaned_consumer_bit_mask_,
578 metadata_header_->queue_index);
579 orphaned_consumer_bit_mask_ = 0;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700580 }
581
Tianyu Jiange2cdec92018-10-30 17:40:58 -0700582 ALOGE_IF(
583 pending_consumers_ && BufferHubDefs::IsBufferReleased(
584 buffer_state_->load(std::memory_order_acquire)),
585 "ProducerChannel::OnConsumerRelease: buffer state inconsistent: "
586 "pending_consumers=%d, buffer buffer is in releaed state.",
587 pending_consumers_);
Corey Tabakacd52dd92017-04-07 18:03:57 -0700588 return {};
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800589}
590
Tianyu0c6b7ff2018-10-03 11:30:04 -0700591void ProducerChannel::DecrementPendingConsumers() {
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700592 if (pending_consumers_ == 0) {
Tianyu0c6b7ff2018-10-03 11:30:04 -0700593 ALOGE("ProducerChannel::DecrementPendingConsumers: no pending consumer.");
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700594 return;
595 }
596
597 --pending_consumers_;
Tianyuf669f6a2018-10-10 15:34:32 -0700598 ALOGD_IF(TRACE, "%s: buffer_id=%d %d consumers left", __FUNCTION__,
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800599 buffer_id(), pending_consumers_);
Tianyuf669f6a2018-10-10 15:34:32 -0700600
601 if (pending_consumers_ == 0) {
602 ALOGD_IF(TRACE,
603 "%s: releasing last consumer: buffer_id=%d state=%" PRIx64 ".",
604 __FUNCTION__, buffer_id(),
605 buffer_state_->load(std::memory_order_acquire));
606 SignalAvailable();
607 }
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800608}
609
Tianyuf669f6a2018-10-10 15:34:32 -0700610void ProducerChannel::OnConsumerOrphaned(const uint64_t& consumer_state_mask) {
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700611 // Ignore the orphaned consumer.
Tianyu0c6b7ff2018-10-03 11:30:04 -0700612 DecrementPendingConsumers();
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700613
Tianyuf669f6a2018-10-10 15:34:32 -0700614 // Remember the ignored consumer so that newly added consumer won't be
615 // taking the same state mask as this orphaned consumer.
616 ALOGE_IF(orphaned_consumer_bit_mask_ & consumer_state_mask,
617 "%s: Consumer (consumer_state_mask=%" PRIx64
618 ") is already orphaned.",
619 __FUNCTION__, consumer_state_mask);
620 orphaned_consumer_bit_mask_ |= consumer_state_mask;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700621
622 // Atomically clear the fence state bit as an orphaned consumer will never
Tianyuf669f6a2018-10-10 15:34:32 -0700623 // signal a release fence.
624 fence_state_->fetch_and(~consumer_state_mask, std::memory_order_release);
625
626 // Atomically set the buffer state of this consumer to released state.
627 buffer_state_->fetch_and(~consumer_state_mask, std::memory_order_release);
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700628
629 ALOGW(
Tianyuf669f6a2018-10-10 15:34:32 -0700630 "%s: detected new orphaned consumer buffer_id=%d "
631 "consumer_state_mask=%" PRIx64 " queue_index=%" PRIx64
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700632 " buffer_state=%" PRIx64 " fence_state=%" PRIx64 ".",
Tianyuf669f6a2018-10-10 15:34:32 -0700633 __FUNCTION__, buffer_id(), consumer_state_mask,
634 metadata_header_->queue_index,
Tianyu Jiange2cdec92018-10-30 17:40:58 -0700635 buffer_state_->load(std::memory_order_acquire),
636 fence_state_->load(std::memory_order_acquire));
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700637}
638
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800639void ProducerChannel::AddConsumer(ConsumerChannel* channel) {
640 consumer_channels_.push_back(channel);
641}
642
643void ProducerChannel::RemoveConsumer(ConsumerChannel* channel) {
644 consumer_channels_.erase(
645 std::find(consumer_channels_.begin(), consumer_channels_.end(), channel));
Tianyu67053492018-10-04 14:00:22 -0700646 // Restore the consumer state bit and make it visible in other threads that
647 // acquire the active_clients_bit_mask_.
Tianyuf669f6a2018-10-10 15:34:32 -0700648 uint64_t consumer_state_mask = channel->client_state_mask();
649 uint64_t current_active_clients_bit_mask =
650 active_clients_bit_mask_->load(std::memory_order_acquire);
651 uint64_t updated_active_clients_bit_mask =
652 current_active_clients_bit_mask & (~consumer_state_mask);
653 while (!active_clients_bit_mask_->compare_exchange_weak(
654 current_active_clients_bit_mask, updated_active_clients_bit_mask,
655 std::memory_order_acq_rel, std::memory_order_acquire)) {
656 ALOGI(
657 "%s: Failed to remove consumer state mask. Current active clients bit "
658 "mask is changed to %" PRIu64
659 " when trying to acquire and modify it to %" PRIu64
660 ". About to try again.",
661 __FUNCTION__, current_active_clients_bit_mask,
662 updated_active_clients_bit_mask);
663 updated_active_clients_bit_mask =
664 current_active_clients_bit_mask & (~consumer_state_mask);
665 }
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700666
Tianyuf669f6a2018-10-10 15:34:32 -0700667 const uint64_t current_buffer_state =
668 buffer_state_->load(std::memory_order_acquire);
669 if (BufferHubDefs::IsClientPosted(current_buffer_state,
670 consumer_state_mask) ||
671 BufferHubDefs::IsClientAcquired(current_buffer_state,
672 consumer_state_mask)) {
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700673 // The consumer client is being destoryed without releasing. This could
674 // happen in corner cases when the consumer crashes. Here we mark it
675 // orphaned before remove it from producer.
Tianyuf669f6a2018-10-10 15:34:32 -0700676 OnConsumerOrphaned(consumer_state_mask);
677 return;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700678 }
679
Tianyuf669f6a2018-10-10 15:34:32 -0700680 if (BufferHubDefs::IsClientReleased(current_buffer_state,
681 consumer_state_mask) ||
682 BufferHubDefs::AnyClientGained(current_buffer_state)) {
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700683 // The consumer is being close while it is suppose to signal a release
684 // fence. Signal the dummy fence here.
Tianyuf669f6a2018-10-10 15:34:32 -0700685 if (fence_state_->load(std::memory_order_acquire) & consumer_state_mask) {
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700686 epoll_event event;
687 event.events = EPOLLIN;
Tianyuf669f6a2018-10-10 15:34:32 -0700688 event.data.u64 = consumer_state_mask;
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700689 if (epoll_ctl(release_fence_fd_.Get(), EPOLL_CTL_MOD,
690 dummy_fence_fd_.Get(), &event) < 0) {
691 ALOGE(
Tianyuf669f6a2018-10-10 15:34:32 -0700692 "%s: Failed to modify the shared release fence to include the "
693 "dummy fence: %s",
694 __FUNCTION__, strerror(errno));
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700695 return;
696 }
Tianyuf669f6a2018-10-10 15:34:32 -0700697 ALOGW("%s: signal dummy release fence buffer_id=%d", __FUNCTION__,
698 buffer_id());
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700699 eventfd_write(dummy_fence_fd_.Get(), 1);
700 }
701 }
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800702}
703
Tianyuf669f6a2018-10-10 15:34:32 -0700704// Returns true if the given parameters match the underlying buffer
705// parameters.
Corey Tabakacd52dd92017-04-07 18:03:57 -0700706bool ProducerChannel::CheckParameters(uint32_t width, uint32_t height,
Hendrik Wagenaar108e84f2017-05-07 22:19:17 -0700707 uint32_t layer_count, uint32_t format,
Corey Tabaka52ea25c2017-09-13 18:02:48 -0700708 uint64_t usage,
709 size_t user_metadata_size) {
710 return user_metadata_size == user_metadata_size_ &&
711 buffer_.width() == width && buffer_.height() == height &&
712 buffer_.layer_count() == layer_count && buffer_.format() == format &&
713 buffer_.usage() == usage;
Alex Vakulenkoe4eec202017-01-27 14:41:04 -0800714}
715
716} // namespace dvr
717} // namespace android