1 /*
2 * Copyright (c) 2023-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 "dinputinitsource_fuzzer.h"
17
18 #include <cstddef>
19 #include <cstdint>
20 #include <functional>
21 #include <iostream>
22 #include <thread>
23
24 #include <refbase.h>
25
26 #include "constants_dinput.h"
27 #include "distributed_input_handler.h"
28 #include "distributed_input_kit.h"
29 #include "distributed_input_source_handler.h"
30 #include "i_distributed_source_input.h"
31
32 namespace OHOS {
33 namespace DistributedHardware {
InitSourceFuzzTest(const uint8_t * data,size_t size)34 void InitSourceFuzzTest(const uint8_t *data, size_t size)
35 {
36 if ((data == nullptr) || (size == 0)) {
37 return;
38 }
39 std::string params(reinterpret_cast<const char*>(data), size);
40 DistributedInput::DistributedInputSourceHandler::GetInstance().InitSource(params);
41 }
42
OnLoadSystemAbilitySuccessFuzzTest(const uint8_t * data,size_t size)43 void OnLoadSystemAbilitySuccessFuzzTest(const uint8_t *data, size_t size)
44 {
45 if ((data == nullptr) || (size < sizeof(int32_t))) {
46 return;
47 }
48 int32_t saId = *(reinterpret_cast<const int32_t*>(data));
49 DistributedInput::DistributedInputSourceHandler::SALoadSourceCb saLoadSourceCb;
50 saLoadSourceCb.OnLoadSystemAbilitySuccess(saId, nullptr);
51 }
52
OnLoadSystemAbilityFailFuzzTest(const uint8_t * data,size_t size)53 void OnLoadSystemAbilityFailFuzzTest(const uint8_t *data, size_t size)
54 {
55 if ((data == nullptr) || (size < sizeof(int32_t))) {
56 return;
57 }
58 int32_t saId = *(reinterpret_cast<const int32_t*>(data));
59 DistributedInput::DistributedInputSourceHandler::SALoadSourceCb saLoadSourceCb;
60 saLoadSourceCb.OnLoadSystemAbilityFail(saId);
61 }
62
OnRemoteSourceSvrDiedFuzzTest(const uint8_t * data,size_t size)63 void OnRemoteSourceSvrDiedFuzzTest(const uint8_t *data, size_t size)
64 {
65 (void)data;
66 (void)size;
67 DistributedInput::DistributedInputSourceHandler::GetInstance().OnRemoteSourceSvrDied(nullptr);
68 DistributedInput::DistributedInputSourceHandler::GetInstance().RegisterDistributedHardwareStateListener(nullptr);
69 DistributedInput::DistributedInputSourceHandler::GetInstance().UnregisterDistributedHardwareStateListener();
70 DistributedInput::DistributedInputSourceHandler::GetInstance().RegisterDataSyncTriggerListener(nullptr);
71 DistributedInput::DistributedInputSourceHandler::GetInstance().UnregisterDataSyncTriggerListener();
72 }
73 } // namespace DistributedHardware
74 } // namespace OHOS
75
76 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)77 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
78 {
79 /* Run your code on data */
80 OHOS::DistributedHardware::InitSourceFuzzTest(data, size);
81 OHOS::DistributedHardware::OnLoadSystemAbilitySuccessFuzzTest(data, size);
82 OHOS::DistributedHardware::OnLoadSystemAbilityFailFuzzTest(data, size);
83 OHOS::DistributedHardware::OnRemoteSourceSvrDiedFuzzTest(data, size);
84 return 0;
85 }