1 /*
2  * Copyright (c) 2022-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 "dcamerahost_fuzzer.h"
17 
18 #include <cstddef>
19 #include <cstdint>
20 
21 #include "dcamera_host.h"
22 
23 namespace OHOS {
24 namespace DistributedHardware {
DCameraGetCameraAbilityFromDevFuzzTest(const uint8_t * data,size_t size)25 void DCameraGetCameraAbilityFromDevFuzzTest(const uint8_t* data, size_t size)
26 {
27     if ((data == nullptr) || (size == 0)) {
28         return;
29     }
30     std::string cameraId = "1__2";
31     DHBase dhBase;
32     dhBase.deviceId_ = "1";
33     dhBase.dhId_ = "2";
34     std::string sinkAbilityInfo = "sink";
35     std::string sourceAbilityInfo = "source";
36     OHOS::sptr<DCameraDevice> dcameraDevice(new (std::nothrow) DCameraDevice(dhBase, sinkAbilityInfo,
37         sourceAbilityInfo));
38     if (dcameraDevice == nullptr) {
39         return;
40     }
41     DCameraHost::GetInstance()->dCameraDeviceMap_.emplace(cameraId, dcameraDevice);
42     std::shared_ptr<CameraAbility> cameraAbility = make_shared<CameraAbility>(1, 1);
43     DCameraHost::GetInstance()->GetCameraAbilityFromDev(cameraId, cameraAbility);
44 }
45 
DCameraAddDeviceParamCheckFuzzTest(const uint8_t * data,size_t size)46 void DCameraAddDeviceParamCheckFuzzTest(const uint8_t* data, size_t size)
47 {
48     if ((data == nullptr) || (size == 0)) {
49         return;
50     }
51     std::string deviceId = "1";
52     std::string dhId = "2";
53     std::string sinkAbilityInfo(reinterpret_cast<const char*>(data), size);
54     std::string srcAbilityInfo(reinterpret_cast<const char*>(data), size);
55     DHBase dhBase;
56     dhBase.deviceId_ = deviceId;
57     dhBase.dhId_ = dhId;
58     sptr<IDCameraProviderCallback> callback;
59     DCameraHost::GetInstance()->AddDeviceParamCheck(dhBase, sinkAbilityInfo, srcAbilityInfo, callback);
60 }
61 
DCameraGetCamDevNumFuzzTest(const uint8_t * data,size_t size)62 void DCameraGetCamDevNumFuzzTest(const uint8_t* data, size_t size)
63 {
64     if ((data == nullptr) || (size == 0)) {
65         return;
66     }
67     DCameraHost::GetInstance()->GetCamDevNum();
68 }
69 
DCameraIsCameraIdInvalidFuzzTest(const uint8_t * data,size_t size)70 void DCameraIsCameraIdInvalidFuzzTest(const uint8_t* data, size_t size)
71 {
72     if ((data == nullptr) || (size == 0)) {
73         return;
74     }
75     std::string cameraId = "1__2";
76     DHBase dhBase;
77     dhBase.deviceId_ = "1";
78     dhBase.dhId_ = "2";
79     std::string sinkAbilityInfo = "sink";
80     std::string sourceAbilityInfo = "source";
81     OHOS::sptr<DCameraDevice> dcameraDevice(new (std::nothrow) DCameraDevice(dhBase, sinkAbilityInfo,
82         sourceAbilityInfo));
83     if (dcameraDevice == nullptr) {
84         return;
85     }
86     DCameraHost::GetInstance()->dCameraDeviceMap_.emplace(cameraId, dcameraDevice);
87     DCameraHost::GetInstance()->IsCameraIdInvalid(cameraId);
88 }
89 
DCameraGetCameraIdByDHBaseFuzzTest(const uint8_t * data,size_t size)90 void DCameraGetCameraIdByDHBaseFuzzTest(const uint8_t* data, size_t size)
91 {
92     if ((data == nullptr) || (size == 0)) {
93         return;
94     }
95     DHBase dhBase;
96     dhBase.deviceId_ = "1";
97     dhBase.dhId_ = "2";
98     DCameraHost::GetInstance()->GetCameraIdByDHBase(dhBase);
99 }
100 }
101 }
102 
103 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)104 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
105 {
106     /* Run your code on data */
107     OHOS::DistributedHardware::DCameraGetCameraAbilityFromDevFuzzTest(data, size);
108     OHOS::DistributedHardware::DCameraAddDeviceParamCheckFuzzTest(data, size);
109     OHOS::DistributedHardware::DCameraGetCamDevNumFuzzTest(data, size);
110     OHOS::DistributedHardware::DCameraIsCameraIdInvalidFuzzTest(data, size);
111     OHOS::DistributedHardware::DCameraGetCameraIdByDHBaseFuzzTest(data, size);
112     return 0;
113 }
114