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 #define private public
17 #define protected public
18 #include "event_handler.h"
19 #undef private
20 #undef protected
21 
22 #include "eventhandler_fuzzer.h"
23 #include "securec.h"
24 #include "deamon_io_waiter.h"
25 
26 namespace OHOS {
27 namespace {
28     constexpr size_t U32_AT_SIZE = 4;
29 }
30 
31 class DumperTest : public AppExecFwk::Dumper {
32 public:
33     DumperTest() = default;
~DumperTest()34     virtual ~DumperTest()
35     {};
Dump(const std::string & message)36     void Dump(const std::string &message) override
37     {}
GetTag()38     std::string GetTag() override
39     {
40         return {};
41     }
42 };
43 
44 class MyFileDescriptorListener : public AppExecFwk::FileDescriptorListener {
45 public:
MyFileDescriptorListener()46     MyFileDescriptorListener()
47     {}
~MyFileDescriptorListener()48     ~MyFileDescriptorListener()
49     {}
50 
51     /* @param int32_t fileDescriptor */
OnReadable(int32_t)52     void OnReadable(int32_t)
53     {}
54 
55     /* @param int32_t fileDescriptor */
OnWritable(int32_t)56     void OnWritable(int32_t)
57     {}
58 
59     /* @param int32_t fileDescriptor */
OnException(int32_t)60     void OnException(int32_t)
61     {}
62 
63     MyFileDescriptorListener(const MyFileDescriptorListener &) = delete;
64     MyFileDescriptorListener &operator=(const MyFileDescriptorListener &) = delete;
65     MyFileDescriptorListener(MyFileDescriptorListener &&) = delete;
66     MyFileDescriptorListener &operator=(MyFileDescriptorListener &&) = delete;
67 };
68 
DoSomethingInterestingWithMyAPI(const char * data,size_t size)69 bool DoSomethingInterestingWithMyAPI(const char* data, size_t size)
70 {
71     std::shared_ptr<AppExecFwk::EventRunner> runner = nullptr;
72     AppExecFwk::EventHandler eventHandler(runner);
73     uint32_t innerEventId = *data;
74     std::list<AppExecFwk::InnerEvent::Pointer> events;
75     AppExecFwk::InnerEvent::Pointer event = std::move(events.front());
76     int64_t taskTime = U32_AT(reinterpret_cast<const uint8_t*>(data));
77     AppExecFwk::EventQueue::Priority priority = AppExecFwk::EventQueue::Priority::LOW;
78     int32_t fileDescriptor = U32_AT(reinterpret_cast<const uint8_t*>(data));
79     DumperTest dumper;
80     eventHandler.Dump(dumper);
81     eventHandler.GetEventName(event);
82     eventHandler.ProcessEvent(event);
83     eventHandler.RemoveAllFileDescriptorListeners();
84     eventHandler.SendTimingEvent(event, taskTime, priority);
85     eventHandler.RemoveFileDescriptorListener(fileDescriptor);
86     eventHandler.SendEvent(event, taskTime, priority);
87     eventHandler.SendSyncEvent(event, priority);
88     eventHandler.RemoveAllEvents();
89     eventHandler.RemoveEvent(innerEventId, taskTime);
90     eventHandler.RemoveEvent(innerEventId);
91     std::string stringData(data);
92     eventHandler.RemoveTask(stringData);
93     auto fileDescriptorListener = std::make_shared<MyFileDescriptorListener>();
94     eventHandler.AddFileDescriptorListener(fileDescriptor, innerEventId, fileDescriptorListener,
95         "DoSomethingInterestingWithMyAPI");
96     eventHandler.SetEventRunner(runner);
97     eventHandler.GetEventRunner();
98     eventHandler.DistributeEvent(event);
99     eventHandler.HasInnerEvent(taskTime);
100     AppExecFwk::DeamonIoWaiter::GetInstance().Init();
101     AppExecFwk::DeamonIoWaiter::GetInstance().StartEpollIoWaiter();
102     AppExecFwk::DeamonIoWaiter::GetInstance().StopEpollIoWaiter();
103     AppExecFwk::DeamonIoWaiter::GetInstance().NotifyOne();
104     AppExecFwk::DeamonIoWaiter::GetInstance().NotifyAll();
105     AppExecFwk::DeamonIoWaiter::GetInstance().SupportListeningFileDescriptor();
106     return eventHandler.HasInnerEvent(innerEventId);
107 }
108 }
109 
110 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)111 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
112 {
113     /* Run your code on data */
114     if (data == nullptr) {
115         return 0;
116     }
117 
118     if (size < OHOS::U32_AT_SIZE) {
119         return 0;
120     }
121 
122     char* ch = reinterpret_cast<char *>(malloc(size + 1));
123     if (ch == nullptr) {
124         return 0;
125     }
126 
127     (void)memset_s(ch, size + 1, 0x00, size + 1);
128     if (memcpy_s(ch, size, data, size) != EOK) {
129         free(ch);
130         ch = nullptr;
131         return 0;
132     }
133 
134     OHOS::DoSomethingInterestingWithMyAPI(ch, size);
135     free(ch);
136     ch = nullptr;
137     return 0;
138 }
139