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 "inputmethodability_fuzzer.h"
17
18 #include <utility>
19
20 #include "input_method_ability.h"
21 #include "input_method_engine_listener_impl.h"
22
23 using namespace OHOS::MiscServices;
24 namespace OHOS {
25 class KeyboardListenerImpl : public KeyboardListener {
OnKeyEvent(int32_t keyCode,int32_t keyStatus,sptr<KeyEventConsumerProxy> & consumer)26 bool OnKeyEvent(int32_t keyCode, int32_t keyStatus, sptr<KeyEventConsumerProxy> &consumer)
27 {
28 if (consumer != nullptr) {
29 consumer->OnKeyCodeConsumeResult(true);
30 }
31 return true;
32 }
OnKeyEvent(const std::shared_ptr<MMI::KeyEvent> & keyEvent,sptr<KeyEventConsumerProxy> & consumer)33 bool OnKeyEvent(const std::shared_ptr<MMI::KeyEvent> &keyEvent, sptr<KeyEventConsumerProxy> &consumer)
34 {
35 if (consumer != nullptr) {
36 consumer->OnKeyEventConsumeResult(true);
37 }
38 return true;
39 }
OnDealKeyEvent(const std::shared_ptr<MMI::KeyEvent> & keyEvent,sptr<KeyEventConsumerProxy> & consumer)40 bool OnDealKeyEvent(const std::shared_ptr<MMI::KeyEvent> &keyEvent, sptr<KeyEventConsumerProxy> &consumer)
41 {
42 OnKeyEvent(keyEvent->GetKeyCode(), keyEvent->GetKeyAction(), consumer);
43 OnKeyEvent(keyEvent, consumer);
44 return true;
45 }
OnCursorUpdate(int32_t positionX,int32_t positionY,int32_t height)46 void OnCursorUpdate(int32_t positionX, int32_t positionY, int32_t height)
47 {
48 }
OnSelectionChange(int32_t oldBegin,int32_t oldEnd,int32_t newBegin,int32_t newEnd)49 void OnSelectionChange(int32_t oldBegin, int32_t oldEnd, int32_t newBegin, int32_t newEnd)
50 {
51 }
OnTextChange(const std::string & text)52 void OnTextChange(const std::string &text)
53 {
54 }
OnEditorAttributeChange(const InputAttribute & inputAttribute)55 void OnEditorAttributeChange(const InputAttribute &inputAttribute)
56 {
57 }
58 };
59
TestInsertText(std::string fuzzedString)60 void TestInsertText(std::string fuzzedString)
61 {
62 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
63 ability->InsertText(std::move(fuzzedString));
64 }
65
TestSetImeListener()66 void TestSetImeListener()
67 {
68 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
69 auto engineListener = std::make_shared<InputMethodEngineListenerImpl>();
70 ability->SetImeListener(engineListener);
71 }
72
TestSetKdListener()73 void TestSetKdListener()
74 {
75 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
76 auto keyBoardListener = std::make_shared<KeyboardListenerImpl>();
77 ability->SetKdListener(keyBoardListener);
78 }
79
TestDeleteForward(int32_t fuzzedInt32)80 void TestDeleteForward(int32_t fuzzedInt32)
81 {
82 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
83 ability->DeleteForward(fuzzedInt32);
84 }
85
TestDeleteBackward(int32_t fuzzedInt32)86 void TestDeleteBackward(int32_t fuzzedInt32)
87 {
88 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
89 ability->DeleteBackward(fuzzedInt32);
90 }
91
TestSendExtendAction(int32_t fuzzedInt32)92 void TestSendExtendAction(int32_t fuzzedInt32)
93 {
94 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
95 ability->SendExtendAction(fuzzedInt32);
96 }
97
TestHideKeyboardSelf()98 void TestHideKeyboardSelf()
99 {
100 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
101 ability->HideKeyboardSelf();
102 }
103
TestGetTextBeforeCursor(int32_t fuzzedInt32)104 void TestGetTextBeforeCursor(int32_t fuzzedInt32)
105 {
106 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
107 std::u16string text;
108 ability->GetTextBeforeCursor(fuzzedInt32, text);
109 }
110
TestGetTextAfterCursor(int32_t fuzzedInt32)111 void TestGetTextAfterCursor(int32_t fuzzedInt32)
112 {
113 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
114 std::u16string text;
115 ability->GetTextAfterCursor(fuzzedInt32, text);
116 }
117
TestSendFunctionKey(int32_t fuzzedInt32)118 void TestSendFunctionKey(int32_t fuzzedInt32)
119 {
120 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
121 ability->SendFunctionKey(fuzzedInt32);
122 }
123
TestMoveCursor(int32_t fuzzedInt32)124 void TestMoveCursor(int32_t fuzzedInt32)
125 {
126 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
127 ability->MoveCursor(fuzzedInt32);
128 }
129
TestDispatchKeyEvent(int32_t fuzzedInt32)130 void TestDispatchKeyEvent(int32_t fuzzedInt32)
131 {
132 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
133 std::shared_ptr<MMI::KeyEvent> keyEvent = MMI::KeyEvent::Create();
134 keyEvent->SetKeyCode(fuzzedInt32);
135 keyEvent->SetKeyAction(fuzzedInt32);
136 sptr<KeyEventConsumerProxy> consumer = new (std::nothrow) KeyEventConsumerProxy(nullptr);
137 ability->DispatchKeyEvent(keyEvent, consumer);
138 }
139
TestSetCallingWindow(int32_t fuzzedInt32)140 void TestSetCallingWindow(int32_t fuzzedInt32)
141 {
142 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
143 ability->SetCallingWindow(fuzzedInt32);
144 }
145
TestGetEnterKeyType()146 void TestGetEnterKeyType()
147 {
148 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
149 int32_t keyType;
150 ability->GetEnterKeyType(keyType);
151 }
152
TestGetInputPattern()153 void TestGetInputPattern()
154 {
155 sptr<InputMethodAbility> ability = InputMethodAbility::GetInstance();
156 int32_t inputPattern;
157 ability->GetInputPattern(inputPattern);
158 }
159 } // namespace OHOS
160
161 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)162 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
163 {
164 /* Run your code on data */
165 std::string fuzzedString(reinterpret_cast<const char *>(data), size);
166 auto fuzzedInt32 = static_cast<int32_t>(size);
167
168 OHOS::TestInsertText(fuzzedString);
169
170 OHOS::TestSetImeListener();
171
172 OHOS::TestSetKdListener();
173
174 OHOS::TestDeleteForward(fuzzedInt32);
175 OHOS::TestDeleteBackward(fuzzedInt32);
176 OHOS::TestSendExtendAction(fuzzedInt32);
177
178 OHOS::TestHideKeyboardSelf();
179
180 OHOS::TestGetTextBeforeCursor(fuzzedInt32);
181 OHOS::TestGetTextAfterCursor(fuzzedInt32);
182
183 OHOS::TestSendFunctionKey(fuzzedInt32);
184 OHOS::TestMoveCursor(fuzzedInt32);
185
186 OHOS::TestDispatchKeyEvent(fuzzedInt32);
187
188 OHOS::TestSetCallingWindow(fuzzedInt32);
189
190 OHOS::TestGetEnterKeyType();
191 OHOS::TestGetInputPattern();
192 return 0;
193 }