1 /*
2 * Copyright (C) 2024 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 #include <atomic>
16 #include <iostream>
17 #include <fstream>
18 #include <queue>
19 #include <string>
20 #include <thread>
21 #include "audiodecoderdemo.h"
22 #define FUZZ_PROJECT_NAME "audiodecoderpreapi_fuzzer"
23
24 using namespace std;
25 using namespace OHOS::MediaAVCodec;
26 using namespace OHOS;
27 using namespace OHOS::MediaAVCodec::AudioDemoAuto;
28
29 namespace OHOS {
AudioDecoderAACFuzzTest(const uint8_t * data,size_t size)30 bool AudioDecoderAACFuzzTest(const uint8_t *data, size_t size)
31 {
32 if (size < sizeof(int64_t)) {
33 return false;
34 }
35 // FUZZ aac
36 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
37 aDecBufferDemo->InitFile("aac");
38 auto res = aDecBufferDemo->RunCase(data, size);
39 delete aDecBufferDemo;
40 return res;
41 }
42
AudioDecoderFlacFuzzTest(const uint8_t * data,size_t size)43 bool AudioDecoderFlacFuzzTest(const uint8_t *data, size_t size)
44 {
45 if (size < sizeof(int64_t)) {
46 return false;
47 }
48 // FUZZ flac
49 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
50 aDecBufferDemo->InitFile("flac");
51 auto res = aDecBufferDemo->RunCase(data, size);
52 delete aDecBufferDemo;
53 return res;
54 }
55
AudioDecoderVORBISFuzzTest(const uint8_t * data,size_t size)56 bool AudioDecoderVORBISFuzzTest(const uint8_t *data, size_t size)
57 {
58 if (size < sizeof(int64_t)) {
59 return false;
60 }
61 // FUZZ vorbis
62 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
63 aDecBufferDemo->InitFile("vorbis");
64 auto res = aDecBufferDemo->RunCase(data, size);
65 delete aDecBufferDemo;
66 return res;
67 }
68
AudioDecoderAMRNBFuzzTest(const uint8_t * data,size_t size)69 bool AudioDecoderAMRNBFuzzTest(const uint8_t *data, size_t size)
70 {
71 if (size < sizeof(int64_t)) {
72 return false;
73 }
74 // FUZZ amrnb
75 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
76 aDecBufferDemo->InitFile("amrnb");
77 auto res = aDecBufferDemo->RunCase(data, size);
78 delete aDecBufferDemo;
79 return res;
80 }
81
AudioDecoderAMRWBFuzzTest(const uint8_t * data,size_t size)82 bool AudioDecoderAMRWBFuzzTest(const uint8_t *data, size_t size)
83 {
84 if (size < sizeof(int64_t)) {
85 return false;
86 }
87 // FUZZ amrwb
88 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
89 aDecBufferDemo->InitFile("amrwb");
90 auto res = aDecBufferDemo->RunCase(data, size);
91 delete aDecBufferDemo;
92 return res;
93 }
94
AudioDecoderOPUSFuzzTest(const uint8_t * data,size_t size)95 bool AudioDecoderOPUSFuzzTest(const uint8_t *data, size_t size)
96 {
97 if (size < sizeof(int64_t)) {
98 return false;
99 }
100 // FUZZ opus
101 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
102 aDecBufferDemo->InitFile("opus");
103 auto res = aDecBufferDemo->RunCase(data, size);
104 delete aDecBufferDemo;
105 return res;
106 }
107
AudioDecoderG711FuzzTest(const uint8_t * data,size_t size)108 bool AudioDecoderG711FuzzTest(const uint8_t *data, size_t size)
109 {
110 if (size < sizeof(int64_t)) {
111 return false;
112 }
113 // FUZZ g711
114 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
115 aDecBufferDemo->InitFile("g711mu");
116 auto res = aDecBufferDemo->RunCase(data, size);
117 delete aDecBufferDemo;
118 return res;
119 }
120
AudioDecoderMP3FuzzTest(const uint8_t * data,size_t size)121 bool AudioDecoderMP3FuzzTest(const uint8_t *data, size_t size)
122 {
123 if (size < sizeof(int64_t)) {
124 return false;
125 }
126 // FUZZ mp3
127 ADecDemoAuto* aDecBufferDemo = new ADecDemoAuto();
128 aDecBufferDemo->InitFile("mp3");
129 auto res = aDecBufferDemo->RunCase(data, size);
130 delete aDecBufferDemo;
131 return res;
132 }
133
134 }
135
136 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)137 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
138 {
139 /* Run your code on data */
140 OHOS::AudioDecoderAACFuzzTest(data, size);
141 OHOS::AudioDecoderFlacFuzzTest(data, size);
142 OHOS::AudioDecoderG711FuzzTest(data, size);
143 OHOS::AudioDecoderVORBISFuzzTest(data, size);
144 OHOS::AudioDecoderAMRNBFuzzTest(data, size);
145 OHOS::AudioDecoderAMRWBFuzzTest(data, size);
146 OHOS::AudioDecoderMP3FuzzTest(data, size);
147 return 0;
148 }