1 /*
2 * Copyright (c) 2021-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 <string>
19
20 #include <algorithm>
21 #include <chrono>
22 #include <cstddef>
23 #include <cstdint>
24 #include <string>
25 #include <fuzzer/FuzzedDataProvider.h>
26 #include "ipc_client_manager.h"
27 #include "ipc_set_useroperation_req.h"
28 #include "ipc_rsp.h"
29 #include "ipc_def.h"
30 #include "device_manager_notify.h"
31 #include "dm_device_info.h"
32 #include "ipc_client_stub.h"
33 #include "ipc_register_listener_req.h"
34 #include "ipc_remote_broker.h"
35 #include "iremote_object.h"
36 #include "iservice_registry.h"
37 #include "dm_constants.h"
38 #include "system_ability_definition.h"
39 #include "device_manager_notify_fuzzer.h"
40
41 namespace OHOS {
42 namespace DistributedHardware {
DeviceManagerNotifyUnRegisterFuzzTest(const uint8_t * data,size_t size)43 void DeviceManagerNotifyUnRegisterFuzzTest(const uint8_t* data, size_t size)
44 {
45 if ((data == nullptr) || (size == 0)) {
46 return;
47 }
48
49 std::string pkgName(reinterpret_cast<const char*>(data), size);
50 std::string deviceId(reinterpret_cast<const char*>(data), size);
51 uint16_t subscribeId = 33;
52 int32_t publishId = 123;
53
54 DeviceManagerNotify::GetInstance().UnRegisterDeathRecipientCallback(pkgName);
55 DeviceManagerNotify::GetInstance().UnRegisterDeviceStateCallback(pkgName);
56 DeviceManagerNotify::GetInstance().UnRegisterPackageCallback(pkgName);
57 DeviceManagerNotify::GetInstance().UnRegisterDeviceManagerFaCallback(pkgName);
58 DeviceManagerNotify::GetInstance().UnRegisterDiscoveryCallback(pkgName, subscribeId);
59 DeviceManagerNotify::GetInstance().UnRegisterPublishCallback(pkgName, publishId);
60 DeviceManagerNotify::GetInstance().UnRegisterAuthenticateCallback(pkgName, deviceId);
61 DeviceManagerNotify::GetInstance().OnUiCall(pkgName, deviceId);
62 }
63
DeviceManagerNotifyDeviceStatusFuzzTest(const uint8_t * data,size_t size)64 void DeviceManagerNotifyDeviceStatusFuzzTest(const uint8_t* data, size_t size)
65 {
66 if ((data == nullptr) || (size == 0)) {
67 return;
68 }
69
70 std::string pkgName(reinterpret_cast<const char*>(data), size);
71 std::string deviceId(reinterpret_cast<const char*>(data), size);
72 std::string token(reinterpret_cast<const char*>(data), size);
73 uint16_t subscribeId = 12;
74 int32_t failedReason = 231;
75 uint32_t status = 3;
76 uint32_t reason = 14;
77 DmDeviceInfo deviceInfo;
78 DmDeviceBasicInfo deviceBasicInfo;
79
80 DeviceManagerNotify::GetInstance().OnDeviceOnline(pkgName, deviceInfo);
81 DeviceManagerNotify::GetInstance().OnDeviceOnline(pkgName, deviceBasicInfo);
82 DeviceManagerNotify::GetInstance().OnDeviceOffline(pkgName, deviceInfo);
83 DeviceManagerNotify::GetInstance().OnDeviceOffline(pkgName, deviceBasicInfo);
84 DeviceManagerNotify::GetInstance().OnDeviceChanged(pkgName, deviceInfo);
85 DeviceManagerNotify::GetInstance().OnDeviceReady(pkgName, deviceInfo);
86 DeviceManagerNotify::GetInstance().OnDeviceReady(pkgName, deviceBasicInfo);
87 DeviceManagerNotify::GetInstance().OnDeviceFound(pkgName, subscribeId, deviceInfo);
88 DeviceManagerNotify::GetInstance().OnDeviceFound(pkgName, subscribeId, deviceBasicInfo);
89 DeviceManagerNotify::GetInstance().OnDiscoveryFailed(pkgName, subscribeId, failedReason);
90 DeviceManagerNotify::GetInstance().OnDiscoverySuccess(pkgName, subscribeId);
91 DeviceManagerNotify::GetInstance().OnAuthResult(pkgName, deviceId, token, status, reason);
92 }
93
DeviceManagerNotifyOnPublishResultFuzzTest(const uint8_t * data,size_t size)94 void DeviceManagerNotifyOnPublishResultFuzzTest(const uint8_t* data, size_t size)
95 {
96 if ((data == nullptr) || (size < sizeof(int32_t) + sizeof(int32_t))) {
97 return;
98 }
99 FuzzedDataProvider fdp(data, size);
100 std::string pkgName(reinterpret_cast<const char*>(data), size);
101 int32_t publishId = fdp.ConsumeIntegral<int32_t>();
102 int32_t publishResult = fdp.ConsumeIntegral<int32_t>();
103
104 DeviceManagerNotify::GetInstance().OnPublishResult(pkgName, publishId, publishResult);
105 }
106 }
107 }
108
109 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)110 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
111 {
112 /* Run your code on data */
113 OHOS::DistributedHardware::DeviceManagerNotifyUnRegisterFuzzTest(data, size);
114 OHOS::DistributedHardware::DeviceManagerNotifyDeviceStatusFuzzTest(data, size);
115 OHOS::DistributedHardware::DeviceManagerNotifyOnPublishResultFuzzTest(data, size);
116
117 return 0;
118 }
119