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_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 codecBuffer.type = READ_WRITE_TYPE;
74 bool mapd = false;
75 if (codecBuffer.type == READ_WRITE_TYPE) {
76 mapd = sharedMemory->MapReadAndWriteAshmem();
77 } else {
78 mapd = sharedMemory->MapReadOnlyAshmem();
79 }
80 if (!mapd) {
81 CODEC_LOGE("MapReadAndWriteAshmem or MapReadOnlyAshmem return false");
82 return nullptr;
83 }
84 codecBuffer.offset = 0;
85 codecBuffer.filledLen = 0;
86 CodecShareBuffer *buffer = new CodecShareBuffer(codecBuffer);
87 codecBuffer.fd = sharedFD;
88 buffer->SetAshMem(sharedMemory);
89 return OHOS::sptr<ICodecBuffer>(buffer);
90 }
91
FillOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)92 int32_t CodecShareBuffer::FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
93 {
94 if (!CheckInvalid(codecBuffer) || codecBuffer_.type != READ_WRITE_TYPE) {
95 CODEC_LOGE("CheckInvalid return false or mem has no right to write ");
96 return HDF_ERR_INVALID_PARAM;
97 }
98 ReleaseFd(codecBuffer);
99 return ICodecBuffer::FillOmxBuffer(codecBuffer, omxBuffer);
100 }
101
EmptyOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)102 int32_t CodecShareBuffer::EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
103 {
104 if (!CheckInvalid(codecBuffer)) {
105 CODEC_LOGE("shMem_ is null or CheckInvalid return false");
106 return HDF_ERR_INVALID_PARAM;
107 }
108
109 ReleaseFd(codecBuffer);
110
111 void *sharedPtr = const_cast<void *>(shMem_->ReadFromAshmem(codecBuffer.filledLen, codecBuffer.offset));
112 if (!sharedPtr) {
113 CODEC_LOGE("omxBuffer.length [%{public}d omxBuffer.offset[%{public}d]", codecBuffer.filledLen,
114 codecBuffer.offset);
115 return HDF_ERR_INVALID_PARAM;
116 }
117 auto ret = memcpy_s(omxBuffer.pBuffer + codecBuffer.offset, codecBuffer.allocLen - codecBuffer.offset, sharedPtr,
118 codecBuffer.filledLen);
119 if (ret != EOK) {
120 CODEC_LOGE("memcpy_s ret [%{public}d]", ret);
121 return HDF_ERR_INVALID_PARAM;
122 }
123 return ICodecBuffer::EmptyOmxBuffer(codecBuffer, omxBuffer);
124 }
125
FreeBuffer(struct OmxCodecBuffer & codecBuffer)126 int32_t CodecShareBuffer::FreeBuffer(struct OmxCodecBuffer &codecBuffer)
127 {
128 if (!CheckInvalid(codecBuffer)) {
129 CODEC_LOGE("shMem_ is null or CheckInvalid return false");
130 return HDF_ERR_INVALID_PARAM;
131 }
132
133 ReleaseFd(codecBuffer);
134
135 shMem_->UnmapAshmem();
136 shMem_->CloseAshmem();
137 shMem_ = nullptr;
138 return HDF_SUCCESS;
139 }
140
GetBuffer()141 uint8_t *CodecShareBuffer::GetBuffer()
142 {
143 return nullptr;
144 }
145
EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)146 int32_t CodecShareBuffer::EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
147 {
148 return ICodecBuffer::EmptyOmxBufferDone(omxBuffer);
149 }
150
FillOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)151 int32_t CodecShareBuffer::FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
152 {
153 if (shMem_ == nullptr || !shMem_->WriteToAshmem(omxBuffer.pBuffer, omxBuffer.nFilledLen, omxBuffer.nOffset)) {
154 CODEC_LOGE("write to ashmem[%{public}p] fail", shMem_.get());
155 return HDF_ERR_INVALID_PARAM;
156 }
157
158 return ICodecBuffer::FillOmxBufferDone(omxBuffer);
159 }
160
CheckInvalid(struct OmxCodecBuffer & codecBuffer)161 bool CodecShareBuffer::CheckInvalid(struct OmxCodecBuffer &codecBuffer)
162 {
163 if (!ICodecBuffer::CheckInvalid(codecBuffer) || shMem_ == nullptr) {
164 CODEC_LOGE("shMem_ is null or CheckInvalid return false");
165 return false;
166 }
167 return true;
168 }
169
ReleaseFd(struct OmxCodecBuffer & codecBuffer)170 void CodecShareBuffer::ReleaseFd(struct OmxCodecBuffer &codecBuffer)
171 {
172 // close the fd, if fd is sent by codecBuffer
173 if (codecBuffer.fd >= 0) {
174 close(codecBuffer.fd);
175 codecBuffer.fd = -1;
176 }
177 }
178 } // namespace Omx
179 } // namespace Codec
180 } // namespace OHOS