1 /*
2 * Copyright 2022 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 "codec_dyna_buffer.h"
16 #include <buffer_handle_utils.h>
17 #include <hdf_base.h>
18 #include <hdf_log.h>
19 #include <securec.h>
20 #include <unistd.h>
21 namespace OHOS {
22 namespace Codec {
23 namespace Omx {
CodecDynaBuffer(struct OmxCodecBuffer & codecBuffer)24 CodecDynaBuffer::CodecDynaBuffer(struct OmxCodecBuffer &codecBuffer) : ICodecBuffer(codecBuffer)
25 {}
26
~CodecDynaBuffer()27 CodecDynaBuffer::~CodecDynaBuffer()
28 {
29 dynaBuffer_ = nullptr;
30 }
31
Create(struct OmxCodecBuffer & codecBuffer)32 sptr<ICodecBuffer> CodecDynaBuffer::Create(struct OmxCodecBuffer &codecBuffer)
33 {
34 auto bufferHandle = reinterpret_cast<BufferHandle *>(codecBuffer.buffer);
35 // may be empty for bufferHandle
36 codecBuffer.buffer = nullptr;
37 codecBuffer.bufferLen = 0;
38 codecBuffer.allocLen = sizeof(DynamicBuffer);
39
40 CodecDynaBuffer *buffer = new CodecDynaBuffer(codecBuffer);
41 buffer->dynaBuffer_ = std::make_shared<DynamicBuffer>();
42 buffer->dynaBuffer_->bufferHandle = bufferHandle;
43 return sptr<ICodecBuffer>(buffer);
44 }
45
FillOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)46 int32_t CodecDynaBuffer::FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
47 {
48 HDF_LOGE("%{public}s :dyna buffer handle is not supported in FillThisBuffer ", __func__);
49 (void)codecBuffer;
50 (void)omxBuffer;
51 return HDF_ERR_INVALID_PARAM;
52 }
53
EmptyOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)54 int32_t CodecDynaBuffer::EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
55 {
56 if (!CheckInvalid(codecBuffer)) {
57 HDF_LOGE("%{public}s : CheckInvalid return false", __func__);
58 return HDF_ERR_INVALID_PARAM;
59 }
60 ResetBuffer(codecBuffer, omxBuffer);
61
62 int fence = codecBuffer.fenceFd;
63 if (fence >= 0) {
64 auto ret = SyncWait(fence, TIME_WAIT_MS);
65 if (ret != EOK) {
66 HDF_LOGW("%{public}s : SyncWait ret err", __func__);
67 }
68 close(codecBuffer.fenceFd);
69 codecBuffer.fenceFd = -1;
70 }
71
72 return ICodecBuffer::EmptyOmxBuffer(codecBuffer, omxBuffer);
73 }
74
FreeBuffer(struct OmxCodecBuffer & codecBuffer)75 int32_t CodecDynaBuffer::FreeBuffer(struct OmxCodecBuffer &codecBuffer)
76 {
77 if (!CheckInvalid(codecBuffer)) {
78 HDF_LOGE("%{public}s :shMem_ is null or CheckInvalid return false", __func__);
79 return HDF_ERR_INVALID_PARAM;
80 }
81
82 if (codecBuffer.buffer != nullptr) {
83 auto bufferHandle = reinterpret_cast<BufferHandle *>(codecBuffer.buffer);
84 // if recv new BufferHandle, free it
85 FreeBufferHandle(bufferHandle);
86 codecBuffer.buffer = 0;
87 codecBuffer.bufferLen = 0;
88 }
89
90 dynaBuffer_ = nullptr;
91
92 return HDF_SUCCESS;
93 }
94
EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)95 int32_t CodecDynaBuffer::EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
96 {
97 return ICodecBuffer::EmptyOmxBufferDone(omxBuffer);
98 }
99
FillOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)100 int32_t CodecDynaBuffer::FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
101 {
102 HDF_LOGE("%{public}s :dyna buffer handle is not supported in FillThisBuffer ", __func__);
103 (void)omxBuffer;
104 return HDF_ERR_INVALID_PARAM;
105 }
106
GetBuffer()107 uint8_t *CodecDynaBuffer::GetBuffer()
108 {
109 return reinterpret_cast<uint8_t *>(dynaBuffer_.get());
110 }
111
CheckInvalid(struct OmxCodecBuffer & codecBuffer)112 bool CodecDynaBuffer::CheckInvalid(struct OmxCodecBuffer &codecBuffer)
113 {
114 if (!ICodecBuffer::CheckInvalid(codecBuffer) || dynaBuffer_ == nullptr) {
115 HDF_LOGE("%{public}s :dynaBuffer_ is null or CheckInvalid return false", __func__);
116 return false;
117 }
118 return true;
119 }
120
ResetBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)121 void CodecDynaBuffer::ResetBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
122 {
123 (void)omxBuffer;
124 if (codecBuffer.buffer == nullptr) {
125 return;
126 }
127 auto bufferHandle = reinterpret_cast<BufferHandle *>(codecBuffer.buffer);
128 // if recv new BufferHandle, save it, but do not need to save to omxBuffer
129 if (dynaBuffer_->bufferHandle != nullptr) {
130 FreeBufferHandle(dynaBuffer_->bufferHandle);
131 }
132 dynaBuffer_->bufferHandle = bufferHandle;
133 codecBuffer.buffer = 0;
134 codecBuffer.filledLen = sizeof(DynamicBuffer);
135 codecBuffer.bufferLen = 0;
136 }
137 } // namespace Omx
138 } // namespace Codec
139 } // namespace OHOS