1#include <private/dvr/detached_buffer.h>
2
3#include <pdx/file_handle.h>
4#include <ui/DetachedBufferHandle.h>
5
6#include <poll.h>
7
8using android::pdx::LocalChannelHandle;
9using android::pdx::LocalHandle;
10using android::pdx::Status;
11
12namespace android {
13namespace dvr {
14
15DetachedBuffer::DetachedBuffer(uint32_t width, uint32_t height,
16                               uint32_t layer_count, uint32_t format,
17                               uint64_t usage, size_t user_metadata_size) {
18  ATRACE_NAME("DetachedBuffer::DetachedBuffer");
19  ALOGD_IF(TRACE,
20           "DetachedBuffer::DetachedBuffer: width=%u height=%u layer_count=%u, "
21           "format=%u usage=%" PRIx64 " user_metadata_size=%zu",
22           width, height, layer_count, format, usage, user_metadata_size);
23
24  auto status = client_.InvokeRemoteMethod<DetachedBufferRPC::Create>(
25      width, height, layer_count, format, usage, user_metadata_size);
26  if (!status) {
27    ALOGE(
28        "DetachedBuffer::DetachedBuffer: Failed to create detached buffer: %s",
29        status.GetErrorMessage().c_str());
30    client_.Close(-status.error());
31  }
32
33  const int ret = ImportGraphicBuffer();
34  if (ret < 0) {
35    ALOGE("DetachedBuffer::DetachedBuffer: Failed to import buffer: %s",
36          strerror(-ret));
37    client_.Close(ret);
38  }
39}
40
41DetachedBuffer::DetachedBuffer(LocalChannelHandle channel_handle)
42    : client_(std::move(channel_handle)) {
43  const int ret = ImportGraphicBuffer();
44  if (ret < 0) {
45    ALOGE("DetachedBuffer::DetachedBuffer: Failed to import buffer: %s",
46          strerror(-ret));
47    client_.Close(ret);
48  }
49}
50
51int DetachedBuffer::ImportGraphicBuffer() {
52  ATRACE_NAME("DetachedBuffer::DetachedBuffer");
53
54  auto status = client_.InvokeRemoteMethod<DetachedBufferRPC::Import>();
55  if (!status) {
56    ALOGE("DetachedBuffer::DetachedBuffer: Failed to import GraphicBuffer: %s",
57          status.GetErrorMessage().c_str());
58    return -status.error();
59  }
60
61  BufferDescription<LocalHandle> buffer_desc = status.take();
62  if (buffer_desc.id() < 0) {
63    ALOGE("DetachedBuffer::DetachedBuffer: Received an invalid id!");
64    return -EIO;
65  }
66
67  // Stash the buffer id to replace the value in id_.
68  const int buffer_id = buffer_desc.id();
69
70  // Import the buffer.
71  IonBuffer ion_buffer;
72  ALOGD_IF(TRACE, "DetachedBuffer::DetachedBuffer: id=%d.", buffer_id);
73
74  if (const int ret = buffer_desc.ImportBuffer(&ion_buffer)) {
75    ALOGE("Failed to import GraphicBuffer, error=%d", ret);
76    return ret;
77  }
78
79  // If all imports succeed, replace the previous buffer and id.
80  id_ = buffer_id;
81  buffer_ = std::move(ion_buffer);
82  return 0;
83}
84
85int DetachedBuffer::Poll(int timeout_ms) {
86  ATRACE_NAME("DetachedBuffer::Poll");
87  pollfd p = {client_.event_fd(), POLLIN, 0};
88  return poll(&p, 1, timeout_ms);
89}
90
91Status<LocalChannelHandle> DetachedBuffer::Promote() {
92  ATRACE_NAME("DetachedBuffer::Promote");
93  ALOGD_IF(TRACE, "DetachedBuffer::Promote: id=%d.", id_);
94
95  auto status_or_handle =
96      client_.InvokeRemoteMethod<DetachedBufferRPC::Promote>();
97  if (status_or_handle.ok()) {
98    // Invalidate the buffer.
99    buffer_ = {};
100  } else {
101    ALOGE("DetachedBuffer::Promote: Failed to promote buffer (id=%d): %s.", id_,
102          status_or_handle.GetErrorMessage().c_str());
103  }
104  return status_or_handle;
105}
106
107sp<GraphicBuffer> DetachedBuffer::TakeGraphicBuffer() {
108  if (!client_.IsValid() || !buffer_.buffer()) {
109    ALOGE("DetachedBuffer::TakeGraphicBuffer: Invalid buffer.");
110    return nullptr;
111  }
112
113  // Technically this should never happen.
114  LOG_FATAL_IF(
115      buffer_.buffer()->isDetachedBuffer(),
116      "DetachedBuffer::TakeGraphicBuffer: GraphicBuffer is already detached.");
117
118  sp<GraphicBuffer> buffer = std::move(buffer_.buffer());
119  buffer->setDetachedBufferHandle(
120      DetachedBufferHandle::Create(client_.TakeChannelHandle()));
121  return buffer;
122}
123
124}  // namespace dvr
125}  // namespace android
126