1 /*
2 * Copyright (c) 2022 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 "avmetadataresolvemetadata_fuzzer.h"
17 #include <iostream>
18 #include "string_ex.h"
19 #include "media_errors.h"
20 #include "directory_ex.h"
21 #include "window_option.h"
22 #include "image_type.h"
23
24 using namespace std;
25 using namespace OHOS;
26 using namespace Media;
27
28 namespace OHOS {
29 namespace Media {
AVMetadataResolveMetadataFuzzer()30 AVMetadataResolveMetadataFuzzer::AVMetadataResolveMetadataFuzzer()
31 {
32 }
33
~AVMetadataResolveMetadataFuzzer()34 AVMetadataResolveMetadataFuzzer::~AVMetadataResolveMetadataFuzzer()
35 {
36 }
37
FuzzAVMetadataResolveMetadata(uint8_t * data,size_t size)38 bool AVMetadataResolveMetadataFuzzer::FuzzAVMetadataResolveMetadata(uint8_t *data, size_t size)
39 {
40 constexpr int32_t AV_METADATA_CODELIST = 17;
41
42 std::shared_ptr<AVMetadataHelper> avmetadata = AVMetadataHelperFactory::CreateAVMetadataHelper();
43 if (avmetadata == nullptr) {
44 return false;
45 }
46
47 const string path = "/data/test/media/H264_AAC.mp4";
48 int32_t retMetadataSetsource = MetaDataSetSource(path, avmetadata);
49 if (retMetadataSetsource != 0) {
50 avmetadata->Release();
51 return false;
52 }
53
54 int32_t avMetadataCodes[AV_METADATA_CODELIST] {
55 AV_KEY_ALBUM,
56 AV_KEY_ALBUM_ARTIST,
57 AV_KEY_ARTIST,
58 AV_KEY_AUTHOR,
59 AV_KEY_DATE_TIME,
60 AV_KEY_COMPOSER,
61 AV_KEY_DURATION,
62 AV_KEY_GENRE,
63 AV_KEY_HAS_AUDIO,
64 AV_KEY_HAS_VIDEO,
65 AV_KEY_MIME_TYPE,
66 AV_KEY_NUM_TRACKS,
67 AV_KEY_SAMPLE_RATE,
68 AV_KEY_TITLE,
69 AV_KEY_VIDEO_HEIGHT,
70 AV_KEY_VIDEO_WIDTH,
71 AV_KEY_VIDEO_ORIENTATION
72 };
73 int32_t keyParameter = avMetadataCodes[*reinterpret_cast<int64_t *>(data) % AV_METADATA_CODELIST];
74 std::string retResolvemetadata = avmetadata->ResolveMetadata(keyParameter);
75 avmetadata->Release();
76 return true;
77 }
78 }
79
FuzzTestAVMetadataResolveMetadata(uint8_t * data,size_t size)80 bool FuzzTestAVMetadataResolveMetadata(uint8_t *data, size_t size)
81 {
82 if (data == nullptr) {
83 return 0;
84 }
85
86 if (size < sizeof(int64_t)) {
87 return 0;
88 }
89 AVMetadataResolveMetadataFuzzer metadata;
90 return metadata.FuzzAVMetadataResolveMetadata(data, size);
91 }
92 }
93
94 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(uint8_t * data,size_t size)95 extern "C" int LLVMFuzzerTestOneInput(uint8_t *data, size_t size)
96 {
97 /* Run your code on data */
98 OHOS::FuzzTestAVMetadataResolveMetadata(data, size);
99 return 0;
100 }
101