1/*
2 * Copyright (c) 2023-2023 Huawei Device 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 "gtest/gtest.h"
17#include "plugin/common/plugin_meta.h"
18#include "pipeline/filters/common/plugin_utils.h"
19#include "plugin/common/plugin_tags.h"
20#include "plugin/common/plugin_types.h"
21#include "plugin/plugins/codec_adapter/codec_utils.h"
22#include "pipeline/filters/common/plugin_settings.h"
23#include "pipeline/core/filter_type.h"
24#include "plugin/convert/ffmpeg_convert.h"
25#include "hdf_base.h"
26#include "OMX_Core.h"
27
28using namespace testing::ext;
29using namespace OHOS::Media::Plugin;
30using namespace OHOS::Media::Plugin::CodecAdapter;
31using namespace OHOS::Media::Pipeline;
32
33namespace OHOS {
34namespace Media {
35namespace Test {
36HWTEST(TestMeta, find_unavailable_plugins, TestSize.Level1)
37{
38    Meta meta;
39    std::string artist("abcd");
40    meta.Set<Tag::MEDIA_TITLE>(artist);
41    auto cap = MetaToCapability(meta);
42    auto plugins = FindAvailablePlugins(*cap, Plugin::PluginType::AUDIO_DECODER);
43    ASSERT_TRUE(plugins.size() == 0);
44}
45
46HWTEST(TestMeta, set_para_to_sink_plugins, TestSize.Level1)
47{
48    std::shared_ptr<Ffmpeg::ResamplePara> resamplePara = std::make_shared<Ffmpeg::ResamplePara>();
49    resamplePara->channels = 2;
50    resamplePara->sampleRate = 96000;
51    resamplePara->bitsPerSample = 16;
52    resamplePara->channelLayout = 3;
53    resamplePara->srcFfFmt = AVSampleFormat::AV_SAMPLE_FMT_FLTP;
54    resamplePara->destSamplesPerFrame = 2048;
55    resamplePara->destFmt = AVSampleFormat::AV_SAMPLE_FMT_S16;
56    std::shared_ptr<Ffmpeg::Resample> resample = std::make_shared<Ffmpeg::Resample>();
57    auto src = new uint8_t;
58    size_t srcLength = 16384;
59    auto des = new uint8_t;
60    size_t desLength = 16384;
61    ASSERT_TRUE(resample->Init(*resamplePara) == Status::OK);
62    ASSERT_TRUE(resample->Convert(src, srcLength, des, desLength) == Status::OK);
63    resamplePara->destFmt = AVSampleFormat::AV_SAMPLE_FMT_U8P;
64    ASSERT_TRUE(resample->Init(*resamplePara) == Status::OK);
65    ASSERT_TRUE(resample->Convert(src, srcLength, des, desLength) == Status::OK);
66
67    resamplePara->bitsPerSample = 8;
68    ASSERT_TRUE(resample->Init(*resamplePara) == Status::OK);
69    ASSERT_FALSE(resample->Convert(src, srcLength, des, desLength) == Status::OK);
70
71    resamplePara->bitsPerSample = 24;
72    ASSERT_TRUE(resample->Init(*resamplePara) == Status::OK);
73    ASSERT_FALSE(resample->Convert(src, srcLength, des, desLength) == Status::OK);
74}
75
76HWTEST(TestMeta, hdf_status_to_string, TestSize.Level1)
77{
78    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_SUCCESS) != "null");
79    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_FAILURE) != "null");
80    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_NOT_SUPPORT) != "null");
81    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_INVALID_PARAM) != "null");
82    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_INVALID_OBJECT) != "null");
83    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_MALLOC_FAIL) != "null");
84    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_TIMEOUT) != "null");
85    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_THREAD_CREATE_FAIL) != "null");
86    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_QUEUE_FULL) != "null");
87    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_DEVICE_BUSY) != "null");
88    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_IO) != "null");
89    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_BAD_FD) != "null");
90    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_NOPERM) != "null");
91    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_BSP_ERR_OP) != "null");
92    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_ERR_BSP_PLT_API_ERR) != "null");
93    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_PAL_ERR_DEV_CREATE) != "null");
94    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_PAL_ERR_INNER) != "null");
95    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_NO_MEMORY) != "null");
96    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_NO_DEVICE) != "null");
97    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_NO_DEVICE_SERVICE) != "null");
98    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_DEV_INIT_FAIL) != "null");
99    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_PUBLISH_FAIL) != "null");
100    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_ATTACHDEV_FAIL) != "null");
101    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_NODATA) != "null");
102    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_NORANGE) != "null");
103    ASSERT_TRUE(HdfStatus2String(HDF_STATUS::HDF_DEV_ERR_OP) != "null");
104    ASSERT_TRUE(HdfStatus2String(100) == "null");
105}
106
107HWTEST(TestMeta, omx_error_type_to_string, TestSize.Level1)
108{
109    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorNone) == "OMX_ErrorNone");
110    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorInsufficientResources) == "OMX_ErrorInsufficientResources");
111    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorUndefined) == "OMX_ErrorUndefined");
112    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorInvalidComponentName) == "OMX_ErrorInvalidComponentName");
113    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorComponentNotFound) == "OMX_ErrorComponentNotFound");
114    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorInvalidComponent) == "OMX_ErrorInvalidComponent");
115    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorBadParameter) == "OMX_ErrorBadParameter");
116    ASSERT_TRUE(OmxErrorType2String(OMX_ErrorNotImplemented) == "OMX_ErrorNotImplemented");
117    ASSERT_TRUE(OmxErrorType2String(100) == "OMX_ErrorNone");
118}
119
120HWTEST(TestMeta, transfor_hdi_ret_value_to_status, TestSize.Level1)
121{
122    ASSERT_TRUE(TransHdiRetVal2Status(HDF_SUCCESS) == Status::OK);
123    ASSERT_TRUE(TransHdiRetVal2Status(HDF_FAILURE) == Status::ERROR_UNKNOWN);
124    ASSERT_TRUE(TransHdiRetVal2Status(HDF_ERR_NOT_SUPPORT) == Status::ERROR_INVALID_OPERATION);
125    ASSERT_TRUE(TransHdiRetVal2Status(HDF_ERR_INVALID_PARAM) == Status::ERROR_INVALID_PARAMETER);
126    ASSERT_TRUE(TransHdiRetVal2Status(HDF_ERR_MALLOC_FAIL) == Status::ERROR_NO_MEMORY);
127    ASSERT_TRUE(TransHdiRetVal2Status(100) == Status::ERROR_UNKNOWN);
128}
129
130HWTEST(TestMeta, translate_to_omx_flag_set, TestSize.Level1)
131{
132    ASSERT_TRUE(Translate2omxFlagSet(BUFFER_FLAG_EOS) == OMX_BUFFERFLAG_EOS);
133    ASSERT_TRUE(Translate2omxFlagSet(100) == 0);
134    ASSERT_TRUE(Translate2PluginFlagSet(OMX_BUFFERFLAG_EOS) == BUFFER_FLAG_EOS);
135    ASSERT_TRUE(Translate2PluginFlagSet(100) == 0);
136}
137
138HWTEST(TestMeta, coding_type_hst_to_hdi, TestSize.Level1)
139{
140    ASSERT_TRUE(CodingTypeHstToHdi(MEDIA_MIME_VIDEO_H264) == OMX_VIDEO_CodingAVC);
141    ASSERT_TRUE(CodingTypeHstToHdi(MEDIA_MIME_VIDEO_H265)
142                == static_cast<OMX_VIDEO_CODINGTYPE>(CODEC_OMX_VIDEO_CodingHEVC));
143    ASSERT_TRUE(CodingTypeHstToHdi("NULL") == OMX_VIDEO_CodingUnused);
144}
145
146HWTEST(TestMeta, format_hst_to_omx, TestSize.Level1)
147{
148    ASSERT_TRUE(FormatHstToOmx(VideoPixelFormat::NV12) == OMX_COLOR_FormatYUV420SemiPlanar);
149    ASSERT_TRUE(FormatHstToOmx(VideoPixelFormat::NV21) == OMX_COLOR_FormatYUV420SemiPlanar);
150    ASSERT_TRUE(FormatHstToOmx(VideoPixelFormat::BGRA) == OMX_COLOR_Format32bitBGRA8888);
151    ASSERT_TRUE(FormatHstToOmx(VideoPixelFormat::RGBA) == OMX_COLOR_Format32bitARGB8888);
152    ASSERT_TRUE(FormatHstToOmx(VideoPixelFormat::YUV420P) == OMX_COLOR_FormatYUV420Planar);
153    ASSERT_TRUE(FormatHstToOmx(VideoPixelFormat::RGB24) == OMX_COLOR_FormatUnused);
154}
155
156HWTEST(TestMeta, omx_state_to_string, TestSize.Level1)
157{
158    ASSERT_TRUE(OmxStateToString(OMX_StateInvalid) == "OMX_StateInvalid");
159    ASSERT_TRUE(OmxStateToString(OMX_StateLoaded) == "OMX_StateLoaded");
160    ASSERT_TRUE(OmxStateToString(OMX_StateLoaded) == "OMX_StateLoaded");
161    ASSERT_TRUE(OmxStateToString(OMX_StateIdle) == "OMX_StateIdle");
162    ASSERT_TRUE(OmxStateToString(OMX_StateExecuting) == "OMX_StateExecuting");
163    ASSERT_TRUE(OmxStateToString(OMX_StatePause) == "OMX_StatePause");
164    ASSERT_TRUE(OmxStateToString(OMX_StateWaitForResources) == "OMX_StateWaitForResources");
165    ASSERT_TRUE(OmxStateToString(OMX_StateKhronosExtensions) == "OMX_StateKhronosExtensions");
166    ASSERT_TRUE(OmxStateToString(OMX_StateVendorStartUnused) == "OMX_StateVendorStartUnused");
167    ASSERT_TRUE(OmxStateToString(OMX_StateMax) == "OMX_StateMax");
168}
169
170HWTEST(TestMeta, get_omx_buffer_type, TestSize.Level1)
171{
172    ASSERT_TRUE(GetOmxBufferType(static_cast<const Plugin::MemoryType>(MemoryType::SHARE_MEMORY), true)
173                == CODEC_BUFFER_TYPE_AVSHARE_MEM_FD);
174    ASSERT_TRUE(GetOmxBufferType(static_cast<const Plugin::MemoryType>(MemoryType::SURFACE_BUFFER), true)
175                == CODEC_BUFFER_TYPE_DYNAMIC_HANDLE);
176    ASSERT_TRUE(GetOmxBufferType(static_cast<const Plugin::MemoryType>(MemoryType::SURFACE_BUFFER), false)
177                == CODEC_BUFFER_TYPE_HANDLE);
178    ASSERT_TRUE(GetOmxBufferType(static_cast<const Plugin::MemoryType>(MemoryType::VIRTUAL_ADDR), true)
179                == CODEC_BUFFER_TYPE_INVALID);
180}
181
182HWTEST(TestMeta, codec_buffer_pool, TestSize.Level1)
183{
184    PluginParaAllowedMap map = PluginParameterTable::FindAllowedParameterMap(FilterType::MEDIA_SOURCE);
185    ASSERT_TRUE(map.size() == 0);
186    map = PluginParameterTable::FindAllowedParameterMap(FilterType::CAPTURE_SOURCE);
187    ASSERT_TRUE(map.size() == 0);
188    map = PluginParameterTable::FindAllowedParameterMap(FilterType::DEMUXER);
189    ASSERT_TRUE(map.size() == 0);
190    map = PluginParameterTable::FindAllowedParameterMap(FilterType::MUXER);
191    ASSERT_TRUE(map.size() > 0);
192    map = PluginParameterTable::FindAllowedParameterMap(FilterType::AUDIO_DECODER);
193    ASSERT_TRUE(map.size() > 0);
194    map = PluginParameterTable::FindAllowedParameterMap(FilterType::VIDEO_DECODER);
195    ASSERT_TRUE(map.size() > 0);
196    map = PluginParameterTable::FindAllowedParameterMap(FilterType::AUDIO_ENCODER);
197    ASSERT_TRUE(map.size() > 0);
198    map = PluginParameterTable::FindAllowedParameterMap(FilterType::VIDEO_ENCODER);
199    ASSERT_TRUE(map.size() > 0);
200    map = PluginParameterTable::FindAllowedParameterMap(FilterType::AUDIO_SINK);
201    ASSERT_TRUE(map.size() > 0);
202    map = PluginParameterTable::FindAllowedParameterMap(FilterType::VIDEO_SINK);
203    ASSERT_TRUE(map.size() > 0);
204    map = PluginParameterTable::FindAllowedParameterMap(FilterType::OUTPUT_SINK);
205    ASSERT_TRUE(map.size() == 0);
206    map = PluginParameterTable::FindAllowedParameterMap(FilterType::NONE);
207    ASSERT_TRUE(map.size() == 0);
208}
209
210HWTEST(TestMeta, assign_parameter_if_match, TestSize.Level1)
211{
212    uint32_t value = 0;
213    uint32_t ret;
214    ASSERT_FALSE(AssignParameterIfMatch(Tag::SECTION_REGULAR_START, ret, value));
215    ASSERT_FALSE(AssignParameterIfMatch(Tag::MIME, value, 0));
216    ASSERT_TRUE(AssignParameterIfMatch(Tag::TRACK_ID, ret, value));
217}
218
219HWTEST(TestMeta, translate_plugin_status, TestSize.Level1)
220{
221    auto status = TranslatePluginStatus(Plugin::Status::END_OF_STREAM);
222    ASSERT_TRUE(status == ErrorCode::END_OF_STREAM);
223    status = TranslatePluginStatus(Plugin::Status::OK);
224    ASSERT_TRUE(status == ErrorCode::SUCCESS);
225    status = TranslatePluginStatus(Plugin::Status::NO_ERROR);
226    ASSERT_TRUE(status == ErrorCode::SUCCESS);
227    status = TranslatePluginStatus(Plugin::Status::ERROR_UNKNOWN);
228    ASSERT_TRUE(status == ErrorCode::ERROR_UNKNOWN);
229    status = TranslatePluginStatus(Plugin::Status::ERROR_CLIENT);
230    ASSERT_TRUE(status == ErrorCode::ERROR_UNKNOWN);
231}
232
233HWTEST(TestMeta, translate_into_parameter, TestSize.Level1)
234{
235    Tag tag = Tag::SECTION_REGULAR_START;
236    auto ret = TranslateIntoParameter(-1, tag);
237    ASSERT_FALSE(ret);
238    ASSERT_TRUE(tag == Tag::SECTION_REGULAR_START);
239    ret = TranslateIntoParameter(static_cast<const int>(Tag::SECTION_REGULAR_START), tag);
240    ASSERT_TRUE(ret);
241    ASSERT_TRUE(tag == Tag::SECTION_REGULAR_START);
242}
243
244HWTEST(TestMeta, capability_to_string, TestSize.Level1)
245{
246    std::shared_ptr<Capability> capability = std::make_shared<Capability>();
247    std::string string = Capability2String(*capability);
248    ASSERT_TRUE(string == "Capability{mime:}");
249    capability->SetMime("video/avc");
250    capability->AppendFixedKey(Capability::Key::MEDIA_BITRATE, nullptr);
251    capability->AppendFixedKey(Capability::Key::VIDEO_BIT_STREAM_FORMAT, nullptr);
252    string = Capability2String(*capability);
253    ASSERT_TRUE(string == "Capability{mime:video/avc, ");
254}
255
256HWTEST(TestMeta, find_allowed_parameter_map, TestSize.Level1)
257{
258    FilterType filterType = FilterType::NONE;
259    PluginParaAllowedMap map = PluginParameterTable::FindAllowedParameterMap(filterType);
260    ASSERT_TRUE(map.size() == 0);
261    filterType = FilterType::AUDIO_DECODER;
262    map = PluginParameterTable::FindAllowedParameterMap(filterType);
263    ASSERT_TRUE(map.size() != 0);
264}
265} // namespace Test
266} // namespace Media
267} // namespace OHOS
268