1 /*
2 * Copyright (c) 2023-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 <fuzzer/FuzzedDataProvider.h>
17 #include "pinholder_fuzzer.h"
18
19 #include "device_manager.h"
20 #include "device_manager_callback.h"
21 #include "dm_ability_manager.h"
22 #include "dm_constants.h"
23 #include "dm_log.h"
24 #include "pin_holder.h"
25 #include "json_object.h"
26 #include "parameter.h"
27 #include "device_manager_service_listener.h"
28
29 namespace OHOS {
30 namespace DistributedHardware {
31 class PinHolderCallbackFuzzTest : public PinHolderCallback {
32 public:
~PinHolderCallbackFuzzTest()33 virtual ~PinHolderCallbackFuzzTest() {}
34
OnPinHolderCreate(const std::string & deviceId,DmPinType pinType,const std::string & payload)35 void OnPinHolderCreate(const std::string &deviceId, DmPinType pinType, const std::string &payload) override {}
OnPinHolderDestroy(DmPinType pinType,const std::string & payload)36 void OnPinHolderDestroy(DmPinType pinType, const std::string &payload) override {}
OnCreateResult(int32_t result)37 void OnCreateResult(int32_t result) override {}
OnDestroyResult(int32_t result)38 void OnDestroyResult(int32_t result) override {}
OnPinHolderEvent(DmPinHolderEvent event,int32_t result,const std::string & content)39 void OnPinHolderEvent(DmPinHolderEvent event, int32_t result, const std::string &content) override {}
40 };
41
PinHolderFuzzTest(const uint8_t * data,size_t size)42 void PinHolderFuzzTest(const uint8_t* data, size_t size)
43 {
44 if ((data == nullptr) || (size < sizeof(uint16_t) + sizeof(uint16_t))) {
45 return;
46 }
47
48 FuzzedDataProvider fdp(data, size);
49 std::string pkgName(reinterpret_cast<const char*>(data), size);
50 std::string payload(reinterpret_cast<const char*>(data), size);
51 PeerTargetId peerTargetId;
52 peerTargetId.deviceId = pkgName;
53 peerTargetId.brMac = pkgName;
54 peerTargetId.bleMac = pkgName;
55 peerTargetId.wifiIp = pkgName;
56 peerTargetId.wifiPort = fdp.ConsumeIntegral<uint16_t>();
57 uint16_t tmp = fdp.ConsumeIntegral<uint16_t>();
58 DmPinType pinType = static_cast<DmPinType>(tmp);
59
60 std::shared_ptr<PinHolderCallback> pinHolderCallback = std::make_shared<PinHolderCallbackFuzzTest>();
61
62 DeviceManager::GetInstance().RegisterPinHolderCallback(pkgName, pinHolderCallback);
63 DeviceManager::GetInstance().CreatePinHolder(pkgName, peerTargetId, pinType, payload);
64 DeviceManager::GetInstance().DestroyPinHolder(pkgName, peerTargetId, pinType, payload);
65 }
66 }
67 }
68
69 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)70 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
71 {
72 /* Run your code on data */
73 OHOS::DistributedHardware::PinHolderFuzzTest(data, size);
74
75 return 0;
76 }
77