1 /*
2  * Copyright (c) 2022 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 "dcameraconfigurestreams_fuzzer.h"
17 
18 #include <cstddef>
19 #include <cstdint>
20 
21 #include "dcamera_provider.h"
22 #include "v1_1/dcamera_types.h"
23 
24 namespace OHOS {
25 namespace DistributedHardware {
26 namespace {
27 const uint32_t DC_ENCODE_SIZE = 4;
28 const DCEncodeType encodeType[DC_ENCODE_SIZE] = {
29     DCEncodeType::ENCODE_TYPE_NULL, DCEncodeType::ENCODE_TYPE_H264, DCEncodeType::ENCODE_TYPE_H265,
30     DCEncodeType::ENCODE_TYPE_JPEG
31 };
32 const uint32_t DC_STREAM_SIZE = 2;
33 const DCStreamType streamType[DC_STREAM_SIZE] = {
34     DCStreamType::CONTINUOUS_FRAME, DCStreamType::SNAPSHOT_FRAME
35 };
36 }
DcameraConfigureStreamsFuzzTest(const uint8_t * data,size_t size)37 void DcameraConfigureStreamsFuzzTest(const uint8_t* data, size_t size)
38 {
39     if ((data == nullptr) || (size < sizeof(int))) {
40         return;
41     }
42 
43     std::string deviceId(reinterpret_cast<const char*>(data), size);
44     std::string dhId(reinterpret_cast<const char*>(data), size);
45     DHBase dhBase;
46     dhBase.deviceId_ = deviceId;
47     dhBase.dhId_ = dhId;
48     std::vector<DCStreamInfo> streamInfos;
49     DCStreamInfo streamInfo;
50     streamInfo.streamId_ = *(reinterpret_cast<const int*>(data));
51     streamInfo.width_ = *(reinterpret_cast<const int*>(data));
52     streamInfo.height_ = *(reinterpret_cast<const int*>(data));
53     streamInfo.stride_ = *(reinterpret_cast<const int*>(data));
54     streamInfo.format_ = *(reinterpret_cast<const int*>(data));
55     streamInfo.dataspace_ = *(reinterpret_cast<const int*>(data));
56     streamInfo.encodeType_ = encodeType[data[0] % DC_ENCODE_SIZE];
57     streamInfo.type_ = streamType[data[0] % DC_STREAM_SIZE];
58     streamInfos.push_back(streamInfo);
59     DCameraProvider::GetInstance()->ConfigureStreams(dhBase, streamInfos);
60 }
61 }
62 }
63 
64 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)65 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
66 {
67     /* Run your code on data */
68     OHOS::DistributedHardware::DcameraConfigureStreamsFuzzTest(data, size);
69     return 0;
70 }
71 
72