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 "startlight_fuzzer.h"
17 #include "securec.h"
18 #include "light_agent.h"
19 
20 namespace OHOS {
21 namespace Sensors {
22 template<class T>
GetObject(T & object,const uint8_t * data,size_t size)23 size_t GetObject(T &object, const uint8_t *data, size_t size)
24 {
25     size_t objectSize = sizeof(object);
26     if (objectSize > size) {
27         return 0;
28     }
29     errno_t ret = memcpy_s(&object, objectSize, data, objectSize);
30     if (ret != EOK) {
31         return 0;
32     }
33     return objectSize;
34 }
35 
StartLightFuzzTest(const uint8_t * data,size_t size)36 void StartLightFuzzTest(const uint8_t *data, size_t size)
37 {
38     size_t startPos = 0;
39     int32_t lightId;
40     startPos += GetObject<int32_t>(lightId, data + startPos, size - startPos);
41     LightColor color;
42     startPos += GetObject<LightColor>(color, data + startPos, size - startPos);
43     LightAnimation animation;
44     GetObject<LightAnimation>(animation, data + startPos, size - startPos);
45     OHOS::Sensors::TurnOn(lightId, color, animation);
46 }
47 } // namespace Sensors
48 } // namespace OHOS
49 
50 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)51 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
52 {
53     /* Run your code on data */
54     OHOS::Sensors::StartLightFuzzTest(data, size);
55     return 0;
56 }
57