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 <iostream>
17 #include <cstddef>
18 #include <cstdint>
19 
20 #include "message_parcel.h"
21 
22 #include "i_audio_renderer_sink.h"
23 #include "audio_manager_base.h"
24 #include "audio_server.h"
25 using namespace std;
26 
27 namespace OHOS {
28 namespace AudioStandard {
29 const int32_t SYSTEM_ABILITY_ID = 3001;
30 const bool RUN_ON_CREATE = false;
31 const int32_t LIMITSIZE = 4;
32 
Convert2Float(const uint8_t * ptr)33 float Convert2Float(const uint8_t *ptr)
34 {
35     float floatValue = static_cast<float>(*ptr);
36     return floatValue / 128.0f - 1.0f;
37 }
38 
AudioServerBalanceFuzzer(const uint8_t * rawData,size_t size,std::shared_ptr<AudioServer> AudioServerPtr)39 void AudioServerBalanceFuzzer(const uint8_t *rawData, size_t size, std::shared_ptr<AudioServer> AudioServerPtr)
40 {
41     float balanceValue = Convert2Float(rawData);
42     MessageParcel data;
43     data.WriteFloat(balanceValue);
44     MessageParcel reply;
45     MessageOption option;
46     AudioServerPtr->OnRemoteRequest(
47         static_cast<uint32_t>(AudioServerInterfaceCode::SET_AUDIO_BALANCE_VALUE), data, reply, option);
48 }
49 
Convert2Bool(const uint8_t * ptr)50 bool Convert2Bool(const uint8_t *ptr)
51 {
52     return (ptr[0] & 1) ? true : false;
53 }
54 
AudioServerMonoFuzzer(const uint8_t * rawData,size_t size,std::shared_ptr<AudioServer> AudioServerPtr)55 void AudioServerMonoFuzzer(const uint8_t *rawData, size_t size, std::shared_ptr<AudioServer> AudioServerPtr)
56 {
57     bool monoState = Convert2Bool(rawData);
58     MessageParcel data;
59     data.WriteBool(monoState);
60     MessageParcel reply;
61     MessageOption option;
62     AudioServerPtr->OnRemoteRequest(
63         static_cast<uint32_t>(AudioServerInterfaceCode::SET_AUDIO_MONO_STATE), data, reply, option);
64 }
65 
AudioServerBalanceFuzzTest(const uint8_t * rawData,size_t size)66 void AudioServerBalanceFuzzTest(const uint8_t *rawData, size_t size)
67 {
68     if (rawData == nullptr || size < LIMITSIZE) {
69         return;
70     }
71     std::shared_ptr<AudioServer> AudioServerPtr =
72         std::make_shared<AudioServer>(SYSTEM_ABILITY_ID, RUN_ON_CREATE);
73     AudioServerBalanceFuzzer(rawData, size, AudioServerPtr);
74     AudioServerMonoFuzzer(rawData, size, AudioServerPtr);
75 }
76 } // namespace AudioStandard
77 } // namesapce OHOS
78 
79 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)80 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
81 {
82     /* Run your code on data */
83     OHOS::AudioStandard::AudioServerBalanceFuzzTest(data, size);
84     return 0;
85 }
86