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