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 avmetadata = AVMetadataHelperFactory::CreateAVMetadataHelper();
43 if (avmetadata == nullptr) {
44 avmetadata->Release();
45 return false;
46 }
47
48 const string path = "/data/test/media/H264_AAC.mp4";
49 int32_t retMetadataSetsource = MetaDataSetSource(path);
50 if (retMetadataSetsource != 0) {
51 avmetadata->Release();
52 return false;
53 }
54
55 int32_t avMetadataCodes[AV_METADATA_CODELIST] {
56 AV_KEY_ALBUM,
57 AV_KEY_ALBUM_ARTIST,
58 AV_KEY_ARTIST,
59 AV_KEY_AUTHOR,
60 AV_KEY_DATE_TIME,
61 AV_KEY_COMPOSER,
62 AV_KEY_DURATION,
63 AV_KEY_GENRE,
64 AV_KEY_HAS_AUDIO,
65 AV_KEY_HAS_VIDEO,
66 AV_KEY_MIME_TYPE,
67 AV_KEY_NUM_TRACKS,
68 AV_KEY_SAMPLE_RATE,
69 AV_KEY_TITLE,
70 AV_KEY_VIDEO_HEIGHT,
71 AV_KEY_VIDEO_WIDTH,
72 AV_KEY_VIDEO_ORIENTATION
73 };
74 int32_t keyParameter = avMetadataCodes[*reinterpret_cast<int64_t *>(data) % AV_METADATA_CODELIST];
75 std::string retResolvemetadata = avmetadata->ResolveMetadata(keyParameter);
76 avmetadata->Release();
77 return true;
78 }
79 }
80
FuzzTestAVMetadataResolveMetadata(uint8_t * data,size_t size)81 bool FuzzTestAVMetadataResolveMetadata(uint8_t *data, size_t size)
82 {
83 if (data == nullptr) {
84 return 0;
85 }
86
87 if (size < sizeof(int64_t)) {
88 return 0;
89 }
90 AVMetadataResolveMetadataFuzzer metadata;
91 return metadata.FuzzAVMetadataResolveMetadata(data, size);
92 }
93 }
94
95 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(uint8_t * data,size_t size)96 extern "C" int LLVMFuzzerTestOneInput(uint8_t *data, size_t size)
97 {
98 /* Run your code on data */
99 OHOS::FuzzTestAVMetadataResolveMetadata(data, size);
100 return 0;
101 }
102