1 /*
2  * Copyright (c) 2023 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 "sinkstubonremoterequest_fuzzer.h"
17 
18 #include <cstddef>
19 #include <cstdint>
20 #include <random>
21 
22 #include "daudio_sink_stub.h"
23 #include "daudio_sink_service.h"
24 
25 #include "daudio_ipc_interface_code.h"
26 #include "if_system_ability_manager.h"
27 #include "iservice_registry.h"
28 
29 namespace OHOS {
30 namespace DistributedHardware {
31 const uint32_t RANGE = 8;
SinkStubOnRemoteRequestFuzzTest(const uint8_t * data,size_t size)32 void SinkStubOnRemoteRequestFuzzTest(const uint8_t* data, size_t size)
33 {
34     if ((data == nullptr) || (size < (sizeof(int32_t)))) {
35         return;
36     }
37 
38     int32_t saId = *(reinterpret_cast<const int32_t*>(data));
39     bool runOnCreate = *(reinterpret_cast<const bool*>(data));
40     auto dAudioSinkService = std::make_shared<DAudioSinkService>(saId, runOnCreate);
41     std::random_device rd;
42     MessageParcel pdata;
43     MessageParcel reply;
44     MessageOption option;
45     std::string devId = "1";
46     std::string dhId = "2";
47     std::string reqId = "3";
48     int32_t status = *(reinterpret_cast<const int32_t*>(data));
49     std::string resultData(reinterpret_cast<const char*>(data), size);
50     pdata.WriteString(devId);
51     pdata.WriteString(dhId);
52     pdata.WriteString(reqId);
53     pdata.WriteInt32(status);
54     pdata.WriteString(resultData);
55     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::INIT_SINK)] =
56         &DAudioSinkStub::InitSinkInner;
57     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::RELEASE_SINK)] =
58         &DAudioSinkStub::ReleaseSinkInner;
59     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::SUBSCRIBE_LOCAL_HARDWARE)] =
60         &DAudioSinkStub::SubscribeLocalHardwareInner;
61     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::UNSUBSCRIBE_LOCAL_HARDWARE)] =
62         &DAudioSinkStub::UnsubscribeLocalHardwareInner;
63     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::DAUDIO_NOTIFY)] =
64         &DAudioSinkStub::DAudioNotifyInner;
65     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::PAUSE_DISTRIBUTED_HARDWARE)] =
66         &DAudioSinkStub::PauseDistributedHardwareInner;
67     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::RESUME_DISTRIBUTED_HARDWARE)] =
68         &DAudioSinkStub::ResumeDistributedHardwareInner;
69     dAudioSinkService->memberFuncMap_[static_cast<uint32_t>(IDAudioSinkInterfaceCode::STOP_DISTRIBUTED_HARDWARE)] =
70         &DAudioSinkStub::StopDistributedHardwareInner;
71     const uint32_t code = rd() % RANGE;
72     dAudioSinkService->OnRemoteRequest(code, pdata, reply, option);
73 }
74 }
75 }
76 
77 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)78 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
79 {
80     /* Run your code on data */
81     OHOS::DistributedHardware::SinkStubOnRemoteRequestFuzzTest(data, size);
82     return 0;
83 }
84 
85