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
16 #include <cstddef>
17 #include <cstdint>
18 #include "avcodec_common.h"
19 #include "avcodec_audio_common.h"
20 #include "native_avcodec_audioencoder.h"
21 #include "common/native_mfmagic.h"
22 #include "native_avcodec_audiocodec.h"
23 #include "avcodec_audio_encoder.h"
24 #include "audioencoderdemo.h"
25 #define FUZZ_PROJECT_NAME "audioencoderpreapi_fuzzer"
26
27 using namespace OHOS::MediaAVCodec::AudioEncDemoAuto;
28
29 namespace OHOS {
AudioEncoderAACFuzzTest(const uint8_t * data,size_t size)30 bool AudioEncoderAACFuzzTest(const uint8_t *data, size_t size)
31 {
32 if (size < sizeof(int64_t)) {
33 return false;
34 }
35 // FUZZ aac
36 AEncDemoAuto* aDecBufferDemo = new AEncDemoAuto();
37 aDecBufferDemo->InitFile("aac");
38 bool ret = aDecBufferDemo->RunCase(data, size);
39 delete aDecBufferDemo;
40 return ret;
41 }
42
AudioEncoderOPUSFuzzTest(const uint8_t * data,size_t size)43 bool AudioEncoderOPUSFuzzTest(const uint8_t *data, size_t size)
44 {
45 if (size < sizeof(int64_t)) {
46 return false;
47 }
48 // FUZZ opus
49 AEncDemoAuto* aDecBufferDemo = new AEncDemoAuto();
50 aDecBufferDemo->InitFile("opus");
51 bool ret = aDecBufferDemo->RunCase(data, size);
52 delete aDecBufferDemo;
53 return ret;
54 }
55
AudioEncoderG711FuzzTest(const uint8_t * data,size_t size)56 bool AudioEncoderG711FuzzTest(const uint8_t *data, size_t size)
57 {
58 if (size < sizeof(int64_t)) {
59 return false;
60 }
61 // FUZZ g711
62 AEncDemoAuto* aDecBufferDemo = new AEncDemoAuto();
63 aDecBufferDemo->InitFile("g711");
64 bool ret = aDecBufferDemo->RunCase(data, size);
65 delete aDecBufferDemo;
66 return ret;
67 }
68
AudioEncoderFLACFuzzTest(const uint8_t * data,size_t size)69 bool AudioEncoderFLACFuzzTest(const uint8_t *data, size_t size)
70 {
71 if (size < sizeof(int64_t)) {
72 return false;
73 }
74 // FUZZ flac
75 AEncDemoAuto* aDecBufferDemo = new AEncDemoAuto();
76 aDecBufferDemo->InitFile("flac");
77 bool ret = aDecBufferDemo->RunCase(data, size);
78 delete aDecBufferDemo;
79 return ret;
80 }
81
82 } // namespace OHOS
83
84 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)85 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
86 {
87 /* Run your code on data */
88 OHOS::AudioEncoderAACFuzzTest(data, size);
89 OHOS::AudioEncoderG711FuzzTest(data, size);
90 OHOS::AudioEncoderOPUSFuzzTest(data, size);
91 OHOS::AudioEncoderFLACFuzzTest(data, size);
92 return 0;
93 }
94