1 /*
2 * Copyright (c) 2024-2025 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 <fstream>
17 #include <securec.h>
18 #include <cmath>
19 #include <memory>
20 #include "hpae_sink_input_node.h"
21 #include "hpae_sink_output_node.h"
22 #include "hpae_gain_node.h"
23 #include "../fuzz_utils.h"
24
25 using namespace std;
26 using namespace OHOS::AudioStandard::HPAE;
27
28 namespace OHOS {
29 namespace AudioStandard {
30
31 FuzzUtils &g_fuzzUtils = FuzzUtils::GetInstance();
32 const size_t FUZZ_INPUT_SIZE_THRESHOLD = 10;
33 constexpr uint32_t DEFAULT_NODE_ID = 1234;
34 constexpr uint32_t DEFAULT_FRAME_LEN = 960;
35 typedef void (*TestPtr)();
36
GetTestNodeInfo(HpaeNodeInfo & nodeInfo)37 static void GetTestNodeInfo(HpaeNodeInfo &nodeInfo)
38 {
39 nodeInfo.nodeId = DEFAULT_NODE_ID;
40 nodeInfo.frameLen = DEFAULT_FRAME_LEN;
41 nodeInfo.samplingRate = SAMPLE_RATE_48000;
42 nodeInfo.channels = STEREO;
43 nodeInfo.format = SAMPLE_F32LE;
44 nodeInfo.deviceClass = "primary";
45 }
46
HpaeGainNodeSignalProcessFuzzTest()47 void HpaeGainNodeSignalProcessFuzzTest()
48 {
49 HpaeNodeInfo nodeInfo;
50 GetTestNodeInfo(nodeInfo);
51 std::shared_ptr<HpaeGainNode> hpaeGainNode = std::make_shared<HpaeGainNode>(nodeInfo);
52 std::vector<HpaePcmBuffer *> inputs;
53 hpaeGainNode->SignalProcess(inputs);
54 }
55
HpaeGainNodeSetClientVolumeFuzzTest()56 void HpaeGainNodeSetClientVolumeFuzzTest()
57 {
58 HpaeNodeInfo nodeInfo;
59 GetTestNodeInfo(nodeInfo);
60 std::shared_ptr<HpaeGainNode> hpaeGainNode = std::make_shared<HpaeGainNode>(nodeInfo);
61 float gain = g_fuzzUtils.GetData<float>();
62 hpaeGainNode->SetClientVolume(gain);
63 }
64
HpaeGainNodeGetClientVolumeFuzzTest()65 void HpaeGainNodeGetClientVolumeFuzzTest()
66 {
67 HpaeNodeInfo nodeInfo;
68 GetTestNodeInfo(nodeInfo);
69 std::shared_ptr<HpaeGainNode> hpaeGainNode = std::make_shared<HpaeGainNode>(nodeInfo);
70 hpaeGainNode->GetClientVolume();
71 }
72
HpaeGainNodeSetFadeStateFuzzTest()73 void HpaeGainNodeSetFadeStateFuzzTest()
74 {
75 HpaeNodeInfo nodeInfo;
76 GetTestNodeInfo(nodeInfo);
77 std::shared_ptr<HpaeGainNode> hpaeGainNode = std::make_shared<HpaeGainNode>(nodeInfo);
78 IOperation operation = OPERATION_DATA_LINK_CONNECTED;
79 hpaeGainNode->SetFadeState(operation);
80 }
81
HpaeGainNodeSilenceDataFuzzTest()82 void HpaeGainNodeSilenceDataFuzzTest()
83 {
84 HpaeNodeInfo nodeInfo;
85 GetTestNodeInfo(nodeInfo);
86 std::shared_ptr<HpaeGainNode> hpaeGainNode = std::make_shared<HpaeGainNode>(nodeInfo);
87 PcmBufferInfo pcmBufferInfo(MONO, DEFAULT_FRAME_LEN, SAMPLE_RATE_44100);
88 HpaePcmBuffer hpaePcmBuffer(pcmBufferInfo);
89 hpaeGainNode->SilenceData(&hpaePcmBuffer);
90 }
91
HpaeGainNodeIsSilentDataFuzzTest()92 void HpaeGainNodeIsSilentDataFuzzTest()
93 {
94 HpaeNodeInfo nodeInfo;
95 GetTestNodeInfo(nodeInfo);
96 std::shared_ptr<HpaeGainNode> hpaeGainNode = std::make_shared<HpaeGainNode>(nodeInfo);
97 PcmBufferInfo pcmBufferInfo(MONO, DEFAULT_FRAME_LEN, SAMPLE_RATE_44100);
98 HpaePcmBuffer hpaePcmBuffer(pcmBufferInfo);
99 hpaeGainNode->IsSilentData(&hpaePcmBuffer);
100 }
101
102 vector<TestPtr> g_testPtrs = {
103 HpaeGainNodeSignalProcessFuzzTest,
104 HpaeGainNodeSetClientVolumeFuzzTest,
105 HpaeGainNodeGetClientVolumeFuzzTest,
106 HpaeGainNodeSetFadeStateFuzzTest,
107 HpaeGainNodeSilenceDataFuzzTest,
108 HpaeGainNodeIsSilentDataFuzzTest,
109 };
110
111 } // namespace AudioStandard
112 } // namesapce OHOS
113
114 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)115 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
116 {
117 if (size < OHOS::AudioStandard::FUZZ_INPUT_SIZE_THRESHOLD) {
118 return 0;
119 }
120
121 OHOS::AudioStandard::g_fuzzUtils.fuzzTest(data, size, OHOS::AudioStandard::g_testPtrs);
122 return 0;
123 }