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
16 #include "codec_handle_buffer.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 {
CodecHandleBuffer(struct OmxCodecBuffer & codecBuffer)24 CodecHandleBuffer::CodecHandleBuffer(struct OmxCodecBuffer &codecBuffer) : ICodecBuffer(codecBuffer)
25 {}
26
~CodecHandleBuffer()27 CodecHandleBuffer::~CodecHandleBuffer()
28 {
29 if (bufferHandle_ != nullptr) {
30 FreeBufferHandle(bufferHandle_);
31 bufferHandle_ = nullptr;
32 }
33 }
34
Create(struct OmxCodecBuffer & codecBuffer)35 sptr<ICodecBuffer> CodecHandleBuffer::Create(struct OmxCodecBuffer &codecBuffer)
36 {
37 auto bufferHandle = reinterpret_cast<BufferHandle *>(codecBuffer.buffer);
38 if (bufferHandle == nullptr) {
39 HDF_LOGE("%{public}s error, bufferHandle is null", __func__);
40 return nullptr;
41 }
42 codecBuffer.buffer = nullptr;
43 codecBuffer.bufferLen = 0;
44
45 CodecHandleBuffer *buffer = new CodecHandleBuffer(codecBuffer);
46 buffer->bufferHandle_ = bufferHandle;
47 return sptr<ICodecBuffer>(buffer);
48 }
49
FillOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)50 int32_t CodecHandleBuffer::FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
51 {
52 if (!CheckInvalid(codecBuffer)) {
53 HDF_LOGE("%{public}s :CheckInvalid return false or mem has no right to write ", __func__);
54 return HDF_ERR_INVALID_PARAM;
55 }
56 ResetBuffer(codecBuffer, omxBuffer);
57
58 int fenceFd = codecBuffer.fenceFd;
59 if (fenceFd >= 0) {
60 auto ret = SyncWait(fenceFd, TIME_WAIT_MS);
61 if (ret != EOK) {
62 HDF_LOGW("%{public}s : SyncWait ret err", __func__);
63 }
64 close(codecBuffer.fenceFd);
65 codecBuffer.fenceFd = -1;
66 }
67 return ICodecBuffer::FillOmxBuffer(codecBuffer, omxBuffer);
68 }
69
EmptyOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)70 int32_t CodecHandleBuffer::EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
71 {
72 HDF_LOGE("%{public}s : bufferHandle is not support in EmptyThisBuffer", __func__);
73 (void)codecBuffer;
74 (void)omxBuffer;
75 return HDF_ERR_INVALID_PARAM;
76 }
77
FreeBuffer(struct OmxCodecBuffer & codecBuffer)78 int32_t CodecHandleBuffer::FreeBuffer(struct OmxCodecBuffer &codecBuffer)
79 {
80 if (!CheckInvalid(codecBuffer)) {
81 HDF_LOGE("%{public}s :shMem_ is null or CheckInvalid return false", __func__);
82 return HDF_ERR_INVALID_PARAM;
83 }
84
85 if (codecBuffer.buffer != nullptr) {
86 auto bufferHandle = reinterpret_cast<BufferHandle *>(codecBuffer.buffer);
87 FreeBufferHandle(bufferHandle);
88 codecBuffer.buffer = 0;
89 codecBuffer.bufferLen = 0;
90 }
91
92 if (bufferHandle_ != nullptr) {
93 FreeBufferHandle(bufferHandle_);
94 bufferHandle_ = nullptr;
95 }
96
97 return HDF_SUCCESS;
98 }
99
EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)100 int32_t CodecHandleBuffer::EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
101 {
102 HDF_LOGE("%{public}s : bufferHandle is not support in EmptyThisBuffer", __func__);
103 (void)omxBuffer;
104 return HDF_ERR_INVALID_PARAM;
105 }
106
FillOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)107 int32_t CodecHandleBuffer::FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
108 {
109 return ICodecBuffer::FillOmxBufferDone(omxBuffer);
110 }
111
GetBuffer()112 uint8_t *CodecHandleBuffer::GetBuffer()
113 {
114 return reinterpret_cast<uint8_t *>(bufferHandle_);
115 }
116
CheckInvalid(struct OmxCodecBuffer & codecBuffer)117 bool CodecHandleBuffer::CheckInvalid(struct OmxCodecBuffer &codecBuffer)
118 {
119 if (!ICodecBuffer::CheckInvalid(codecBuffer) || bufferHandle_ == nullptr) {
120 HDF_LOGE("%{public}s :bufferHandle_ is null or CheckInvalid return false", __func__);
121 return false;
122 }
123 return true;
124 }
125
ResetBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)126 void CodecHandleBuffer::ResetBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
127 {
128 if (codecBuffer.buffer != nullptr) {
129 auto bufferHandle = reinterpret_cast<BufferHandle *>(codecBuffer.buffer);
130 // if recv new BufferHandle, save it, and save the new bufferhandle to omxbuffer
131 if (bufferHandle_ != nullptr) {
132 FreeBufferHandle(bufferHandle_);
133 }
134 bufferHandle_ = bufferHandle;
135
136 omxBuffer.pBuffer = reinterpret_cast<uint8_t *>(bufferHandle_);
137 codecBuffer.buffer = 0;
138 codecBuffer.bufferLen = 0;
139 }
140 }
141 } // namespace Omx
142 } // namespace Codec
143 } // namespace OHOS