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
16 #include "codec_share_buffer.h"
17 #include <hdf_base.h>
18 #include <securec.h>
19 #include <unistd.h>
20 #include "codec_log_wrapper.h"
21 using namespace OHOS::HDI::Codec::V1_0;
22 namespace OHOS {
23 namespace Codec {
24 namespace Omx {
CodecShareBuffer(struct OmxCodecBuffer & codecBuffer)25 CodecShareBuffer::CodecShareBuffer(struct OmxCodecBuffer &codecBuffer) : ICodecBuffer(codecBuffer)
26 {}
27
~CodecShareBuffer()28 CodecShareBuffer::~CodecShareBuffer()
29 {
30 if (shMem_ != nullptr) {
31 shMem_->UnmapAshmem();
32 shMem_->CloseAshmem();
33 shMem_ = nullptr;
34 }
35 }
36
SetAshMem(std::shared_ptr<OHOS::Ashmem> shMem)37 void CodecShareBuffer::SetAshMem(std::shared_ptr<OHOS::Ashmem> shMem)
38 {
39 shMem_ = shMem;
40 }
41
Create(struct OmxCodecBuffer & codecBuffer)42 OHOS::sptr<ICodecBuffer> CodecShareBuffer::Create(struct OmxCodecBuffer &codecBuffer)
43 {
44 if (codecBuffer.fd < 0) {
45 CODEC_LOGE("codecBuffer.fd is invalid");
46 return nullptr;
47 }
48 int size = OHOS::AshmemGetSize(codecBuffer.fd);
49 std::shared_ptr<OHOS::Ashmem> sharedMem = std::make_shared<OHOS::Ashmem>(codecBuffer.fd, size);
50 bool mapd = false;
51 if (codecBuffer.type == READ_WRITE_TYPE) {
52 mapd = sharedMem->MapReadAndWriteAshmem();
53 } else {
54 mapd = sharedMem->MapReadOnlyAshmem();
55 }
56 if (!mapd) {
57 CODEC_LOGE("MapReadAndWriteAshmem or MapReadOnlyAshmem return false");
58 return nullptr;
59 }
60 codecBuffer.fd = -1;
61 CodecShareBuffer *buffer = new CodecShareBuffer(codecBuffer);
62 buffer->SetAshMem(sharedMem);
63 return OHOS::sptr<ICodecBuffer>(buffer);
64 }
65
Allocate(struct OmxCodecBuffer & codecBuffer)66 OHOS::sptr<ICodecBuffer> CodecShareBuffer::Allocate(struct OmxCodecBuffer &codecBuffer)
67 {
68 codecBuffer.bufferType = CODEC_BUFFER_TYPE_AVSHARE_MEM_FD;
69 // create shared memory
70 int sharedFD = AshmemCreate(nullptr, codecBuffer.allocLen);
71
72 std::shared_ptr<Ashmem> sharedMemory = std::make_shared<Ashmem>(sharedFD, codecBuffer.allocLen);
73 bool mapd = false;
74 if (codecBuffer.type == READ_WRITE_TYPE) {
75 mapd = sharedMemory->MapReadAndWriteAshmem();
76 } else {
77 mapd = sharedMemory->MapReadOnlyAshmem();
78 }
79 if (!mapd) {
80 CODEC_LOGE("MapReadAndWriteAshmem or MapReadOnlyAshmem return false");
81 return nullptr;
82 }
83 codecBuffer.offset = 0;
84 codecBuffer.filledLen = 0;
85 CodecShareBuffer *buffer = new CodecShareBuffer(codecBuffer);
86 codecBuffer.fd = sharedFD;
87 buffer->SetAshMem(sharedMemory);
88 return OHOS::sptr<ICodecBuffer>(buffer);
89 }
90
FillOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)91 int32_t CodecShareBuffer::FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
92 {
93 if (!CheckInvalid(codecBuffer) || codecBuffer_.type != READ_WRITE_TYPE) {
94 CODEC_LOGE("CheckInvalid return false or mem has no right to write ");
95 return HDF_ERR_INVALID_PARAM;
96 }
97 ReleaseFd(codecBuffer);
98 return ICodecBuffer::FillOmxBuffer(codecBuffer, omxBuffer);
99 }
100
EmptyOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)101 int32_t CodecShareBuffer::EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
102 {
103 if (!CheckInvalid(codecBuffer)) {
104 CODEC_LOGE("shMem_ is null or CheckInvalid return false");
105 return HDF_ERR_INVALID_PARAM;
106 }
107
108 ReleaseFd(codecBuffer);
109
110 void *sharedPtr = const_cast<void *>(shMem_->ReadFromAshmem(codecBuffer.filledLen, codecBuffer.offset));
111 if (!sharedPtr) {
112 CODEC_LOGE("omxBuffer.length [%{public}d omxBuffer.offset[%{public}d]", codecBuffer.filledLen,
113 codecBuffer.offset);
114 return HDF_ERR_INVALID_PARAM;
115 }
116 auto ret = memcpy_s(omxBuffer.pBuffer + codecBuffer.offset, codecBuffer.allocLen - codecBuffer.offset, sharedPtr,
117 codecBuffer.filledLen);
118 if (ret != EOK) {
119 CODEC_LOGE("memcpy_s ret [%{public}d]", ret);
120 return HDF_ERR_INVALID_PARAM;
121 }
122 return ICodecBuffer::EmptyOmxBuffer(codecBuffer, omxBuffer);
123 }
124
FreeBuffer(struct OmxCodecBuffer & codecBuffer)125 int32_t CodecShareBuffer::FreeBuffer(struct OmxCodecBuffer &codecBuffer)
126 {
127 if (!CheckInvalid(codecBuffer)) {
128 CODEC_LOGE("shMem_ is null or CheckInvalid return false");
129 return HDF_ERR_INVALID_PARAM;
130 }
131
132 ReleaseFd(codecBuffer);
133
134 shMem_->UnmapAshmem();
135 shMem_->CloseAshmem();
136 shMem_ = nullptr;
137 return HDF_SUCCESS;
138 }
139
GetBuffer()140 uint8_t *CodecShareBuffer::GetBuffer()
141 {
142 return nullptr;
143 }
144
EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)145 int32_t CodecShareBuffer::EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
146 {
147 return ICodecBuffer::EmptyOmxBufferDone(omxBuffer);
148 }
149
FillOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)150 int32_t CodecShareBuffer::FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
151 {
152 if (shMem_ == nullptr || !shMem_->WriteToAshmem(omxBuffer.pBuffer, omxBuffer.nFilledLen, omxBuffer.nOffset)) {
153 CODEC_LOGE("write to ashmem fail");
154 return HDF_ERR_INVALID_PARAM;
155 }
156
157 return ICodecBuffer::FillOmxBufferDone(omxBuffer);
158 }
159
CheckInvalid(struct OmxCodecBuffer & codecBuffer)160 bool CodecShareBuffer::CheckInvalid(struct OmxCodecBuffer &codecBuffer)
161 {
162 if (!ICodecBuffer::CheckInvalid(codecBuffer) || shMem_ == nullptr) {
163 CODEC_LOGE("shMem_ is null or CheckInvalid return false");
164 return false;
165 }
166 return true;
167 }
168
ReleaseFd(struct OmxCodecBuffer & codecBuffer)169 void CodecShareBuffer::ReleaseFd(struct OmxCodecBuffer &codecBuffer)
170 {
171 // close the fd, if fd is sent by codecBuffer
172 if (codecBuffer.fd >= 0) {
173 close(codecBuffer.fd);
174 codecBuffer.fd = -1;
175 }
176 }
177 } // namespace Omx
178 } // namespace Codec
179 } // namespace OHOS