1 /*
2 * Copyright (c) 2022-2023 Shenzhen Kaihong DID Co., Ltd..
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15 #include "icodec_buffer.h"
16 #include <hdf_base.h>
17 #include <poll.h>
18 #include <securec.h>
19 #include "codec_dyna_buffer.h"
20 #include "codec_handle_buffer.h"
21 #include "codec_dma_buffer.h"
22 #include "codec_log_wrapper.h"
23 #include "codec_share_buffer.h"
24 #include "v4_0/codec_types.h"
25 using namespace OHOS::HDI::Codec::V4_0;
26 namespace OHOS {
27 namespace Codec {
28 namespace Omx {
29
UseBuffer(OMX_HANDLETYPE comp,uint32_t portIndex,OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE * & header,bool doCopy)30 sptr<ICodecBuffer> ICodecBuffer::UseBuffer(OMX_HANDLETYPE comp, uint32_t portIndex,
31 OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE *&header, bool doCopy)
32 {
33 switch (codecBuffer.bufferType) {
34 case CODEC_BUFFER_TYPE_AVSHARE_MEM_FD:
35 return CodecShareBuffer::UseBuffer(comp, portIndex, codecBuffer, header, doCopy);
36 case CODEC_BUFFER_TYPE_HANDLE:
37 return CodecHandleBuffer::UseBuffer(comp, portIndex, codecBuffer, header);
38 case CODEC_BUFFER_TYPE_DYNAMIC_HANDLE:
39 return CodecDynaBuffer::UseBuffer(comp, portIndex, codecBuffer, header);
40 case CODEC_BUFFER_TYPE_DMA_MEM_FD:
41 return CodecDMABuffer::UseBuffer(comp, portIndex, codecBuffer, header);
42 default:
43 CODEC_LOGE("bufferType[%{public}d] is unexpected", codecBuffer.bufferType);
44 return nullptr;
45 }
46 }
47
AllocateBuffer(OMX_HANDLETYPE comp,uint32_t portIndex,OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE * & header)48 sptr<ICodecBuffer> ICodecBuffer::AllocateBuffer(OMX_HANDLETYPE comp, uint32_t portIndex,
49 OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE *&header)
50 {
51 switch (codecBuffer.bufferType) {
52 case CODEC_BUFFER_TYPE_AVSHARE_MEM_FD:
53 return CodecShareBuffer::AllocateBuffer(comp, portIndex, codecBuffer, header);
54 case CODEC_BUFFER_TYPE_DMA_MEM_FD:
55 return CodecDMABuffer::AllocateBuffer(comp, portIndex, codecBuffer, header);
56 default:
57 CODEC_LOGE("bufferType[%{public}d] is unexpected", codecBuffer.bufferType);
58 return nullptr;
59 }
60 }
61
EmptyThisBuffer(OmxCodecBuffer & codecBuffer)62 int32_t ICodecBuffer::EmptyThisBuffer(OmxCodecBuffer &codecBuffer)
63 {
64 CHECK_AND_RETURN_RET_LOG(comp_ != nullptr, OMX_ErrorInvalidComponent, "null component");
65 CHECK_AND_RETURN_RET_LOG(omxBufHeader_ != nullptr, OMX_ErrorBadParameter, "null header");
66 omxBufHeader_->nOffset = codecBuffer.offset;
67 omxBufHeader_->nFilledLen = codecBuffer.filledLen;
68 omxBufHeader_->nFlags = codecBuffer.flag;
69 omxBufHeader_->nTimeStamp = codecBuffer.pts;
70
71 codecBuffer_.alongParam = std::move(codecBuffer.alongParam);
72 omxBufHeader_->pAppPrivate = nullptr;
73 OMXBufferAppPrivateData privateData{};
74 if (codecBuffer_.bufferType == CODEC_BUFFER_TYPE_DYNAMIC_HANDLE && !codecBuffer_.alongParam.empty()) {
75 privateData.sizeOfParam = static_cast<uint32_t>(codecBuffer_.alongParam.size());
76 privateData.param = static_cast<void *>(codecBuffer_.alongParam.data());
77 omxBufHeader_->pAppPrivate = static_cast<void *>(&privateData);
78 }
79 int32_t ret = OMX_EmptyThisBuffer(comp_, omxBufHeader_);
80 omxBufHeader_->pAppPrivate = nullptr;
81 return ret;
82 }
83
FillThisBuffer(OmxCodecBuffer & codecBuffer)84 int32_t ICodecBuffer::FillThisBuffer(OmxCodecBuffer &codecBuffer)
85 {
86 CHECK_AND_RETURN_RET_LOG(comp_ != nullptr, OMX_ErrorInvalidComponent, "null component");
87 CHECK_AND_RETURN_RET_LOG(omxBufHeader_ != nullptr, OMX_ErrorBadParameter, "null header");
88 omxBufHeader_->nOffset = codecBuffer.offset;
89 omxBufHeader_->nFilledLen = codecBuffer.filledLen;
90 omxBufHeader_->nFlags = codecBuffer.flag;
91 return OMX_FillThisBuffer(comp_, omxBufHeader_);
92 }
93
EmptyBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer,OmxCodecBuffer & codecBuffer)94 int32_t ICodecBuffer::EmptyBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer, OmxCodecBuffer& codecBuffer)
95 {
96 (void)omxBuffer;
97 codecBuffer = codecBuffer_;
98 codecBuffer.bufferhandle = nullptr;
99 codecBuffer.fd = nullptr;
100 codecBuffer.fenceFd = nullptr;
101 codecBuffer.alongParam.clear();
102 return 0;
103 }
104
FillBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer,OmxCodecBuffer & codecBuffer)105 int32_t ICodecBuffer::FillBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer, OmxCodecBuffer& codecBuffer)
106 {
107 codecBuffer = codecBuffer_;
108 codecBuffer.bufferhandle = nullptr;
109 codecBuffer.fd = nullptr;
110 codecBuffer.fenceFd = nullptr;
111 codecBuffer.offset = omxBuffer.nOffset;
112 codecBuffer.filledLen = omxBuffer.nFilledLen;
113 codecBuffer.flag = omxBuffer.nFlags;
114 codecBuffer.pts = omxBuffer.nTimeStamp;
115 auto appPrivate = static_cast<OMXBufferAppPrivateData *>(omxBuffer.pAppPrivate);
116 if (appPrivate != nullptr && appPrivate->param != nullptr &&
117 appPrivate->sizeOfParam < 1024) { // 1024: to protect from taint data
118 codecBuffer.alongParam.resize(appPrivate->sizeOfParam);
119 std::copy(static_cast<uint8_t*>(appPrivate->param),
120 static_cast<uint8_t*>(appPrivate->param) + appPrivate->sizeOfParam,
121 codecBuffer.alongParam.begin());
122 } else {
123 codecBuffer.alongParam.clear();
124 }
125 return 0;
126 }
127
SyncWait(int fd,uint32_t timeout)128 int32_t ICodecBuffer::SyncWait(int fd, uint32_t timeout)
129 {
130 int retCode = -EPERM;
131 if (fd < 0) {
132 CODEC_LOGE("The fence id is invalid.");
133 return retCode;
134 }
135
136 struct pollfd pollfds = {0};
137 pollfds.fd = fd;
138 pollfds.events = POLLIN;
139
140 do {
141 retCode = poll(&pollfds, 1, timeout);
142 } while (retCode == -EPERM && (errno == EINTR || errno == EAGAIN));
143
144 if (retCode == 0) {
145 retCode = -EPERM;
146 errno = ETIME;
147 } else if (retCode > 0) {
148 if (static_cast<uint32_t>(pollfds.revents) & (POLLERR | POLLNVAL)) {
149 retCode = -EPERM;
150 errno = EINVAL;
151 }
152 }
153 return retCode < 0 ? -errno : EOK;
154 }
155
~ICodecBuffer()156 ICodecBuffer::~ICodecBuffer()
157 {
158 FreeBuffer();
159 }
160
FreeBuffer()161 void ICodecBuffer::FreeBuffer()
162 {
163 if (comp_ && omxBufHeader_) {
164 OMX_FreeBuffer(comp_, portIndex_, omxBufHeader_);
165 }
166 comp_ = nullptr;
167 omxBufHeader_ = nullptr;
168 }
169
170 } // namespace Omx
171 } // namespace Codec
172 } // namespace OHOS