xref: /base/inputmethod/imf/test/fuzztest/inputmethodcontroller_fuzzer/inputmethodcontroller_fuzzer.cpp

/*
 * Copyright (c) 2022-2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#define private public
#define protected public
#include "input_method_controller.h"
#include "input_method_system_ability_proxy.h"
#undef private

#include "inputmethodcontroller_fuzzer.h"

#include <cstddef>
#include <cstdint>

#include "global.h"
#include "input_attribute.h"
#include "key_event.h"
#include "message_parcel.h"
#include "text_listener.h"

using namespace OHOS::MiscServices;
namespace OHOS {
void TestListInputMethod(sptr<InputMethodController> imc)
{
    std::vector<Property> properties = {};
    imc->ListInputMethod(properties);
    imc->ListInputMethod(false, properties);
    imc->ListInputMethod(true, properties);
    imc->DisplayOptionalInputMethod();
}

void TestListInputMethodSubtype(sptr<InputMethodController> imc, const std::string &fuzzedString, uint32_t fuzzedUint32)
{
    std::vector<SubProperty> subProperties = {};
    Property property;
    property.name = fuzzedString;
    property.id = fuzzedString;
    property.label = fuzzedString;
    property.icon = fuzzedString;
    property.iconId = fuzzedUint32;
    imc->ListInputMethodSubtype(property, subProperties);
}

void TestDispatchKeyEvent(sptr<InputMethodController> imc, int32_t fuzzedInt32)
{
    sptr<OnTextChangedListener> textListener = new TextListener();
    imc->Attach(textListener);

    std::shared_ptr<MMI::KeyEvent> keyEvent = MMI::KeyEvent::Create();
    keyEvent->SetKeyAction(fuzzedInt32);
    keyEvent->SetKeyCode(fuzzedInt32);
    imc->DispatchKeyEvent(keyEvent, [](std::shared_ptr<MMI::KeyEvent> &keyEvent, bool isConsumed) {});
}

void TestOnSelectionChange(sptr<InputMethodController> imc, std::u16string fuzzedU16String, int fuzzedInt,
    double fuzzedDouble)
{
    sptr<OnTextChangedListener> textListener = new TextListener();
    imc->Attach(textListener);

    CursorInfo cursorInfo;
    cursorInfo.height = fuzzedDouble;
    cursorInfo.left = fuzzedDouble;
    cursorInfo.top = fuzzedDouble;
    cursorInfo.width = fuzzedDouble;
    imc->OnCursorUpdate(cursorInfo);

    imc->OnSelectionChange(fuzzedU16String, fuzzedInt, fuzzedInt);
}

void TestOnConfigurationChange(sptr<InputMethodController> imc)
{
    sptr<OnTextChangedListener> textListener = new TextListener();
    imc->Attach(textListener);

    Configuration info;
    EnterKeyType keyType = EnterKeyType::DONE;
    info.SetEnterKeyType(keyType);
    TextInputType textInputType = TextInputType::DATETIME;
    info.SetTextInputType(textInputType);
    imc->OnConfigurationChange(info);
    int32_t enterKeyType;
    int32_t inputPattern;
    imc->GetEnterKeyType(enterKeyType);
    imc->GetInputPattern(inputPattern);
}

void TestSwitchInputMethod(SwitchTrigger fuzzedTrigger, sptr<InputMethodController> imc,
    const std::string &fuzzedString)
{
    imc->SwitchInputMethod(fuzzedTrigger, fuzzedString, fuzzedString);
    imc->ShowOptionalInputMethod();
}

void TestSetCallingWindow(sptr<InputMethodController> imc, uint32_t fuzzedUInt32)
{
    sptr<OnTextChangedListener> textListener = new TextListener();
    imc->Attach(textListener);

    imc->SetCallingWindow(fuzzedUInt32);
    imc->ShowSoftKeyboard();
    imc->HideSoftKeyboard();
}

void TestShowSomething(sptr<InputMethodController> imc)
{
    sptr<OnTextChangedListener> textListener = new TextListener();
    imc->Attach(textListener);
    imc->ShowCurrentInput();
    imc->HideCurrentInput();

    imc->ShowTextInput();
    imc->HideTextInput();

    imc->GetCurrentInputMethod();
    imc->GetCurrentInputMethodSubtype();

    imc->StopInputSession();
    imc->Close();
}

void TestUpdateListenEventFlag(sptr<InputMethodController> imc, uint32_t fuzzedUint32)
{
    imc->UpdateListenEventFlag(static_cast<uint32_t>(fuzzedUint32), static_cast<uint32_t>(fuzzedUint32), true);
    imc->UpdateListenEventFlag(static_cast<uint32_t>(fuzzedUint32), static_cast<uint32_t>(fuzzedUint32), false);
}

void TestAttach(sptr<InputMethodController> imc, int32_t fuzzedInt32)
{
    sptr<OnTextChangedListener> textListener = new TextListener();
    InputAttribute inputAttribute;
    inputAttribute.inputPattern = fuzzedInt32;
    inputAttribute.enterKeyType = fuzzedInt32;
    inputAttribute.inputOption = fuzzedInt32;
    imc->Attach(textListener, true, inputAttribute);
    imc->Attach(textListener, false, inputAttribute);
}

void FUZZHideInput(sptr<InputMethodController> imc)
{
    sptr<IInputClient> client = new (std::nothrow) InputClientStub();
    imc->HideInput(client);
    imc->RequestHideInput();
}

void FUZZShowInput(sptr<InputMethodController> imc)
{
    sptr<IInputClient> client = new (std::nothrow) InputClientStub();
    imc->ShowInput(client);
    imc->RequestShowInput();
}

void FUZZRestore(sptr<InputMethodController> imc)
{
    imc->RestoreListenEventFlag();
    imc->RestoreListenInfoInSaDied();
    imc->RestoreAttachInfoInSaDied();
}

void InputType(sptr<InputMethodController> imc)
{
    imc->IsInputTypeSupported(InputType::CAMERA_INPUT);
    imc->IsInputTypeSupported(InputType::SECURITY_INPUT);
    imc->StartInputType(InputType::CAMERA_INPUT);
    imc->StartInputType(InputType::SECURITY_INPUT);
}

void FUZZIsPanelShown(sptr<InputMethodController> imc, const uint8_t *data)
{
    PanelInfo panelInfo = { .panelType = SOFT_KEYBOARD, .panelFlag = FLG_FIXED };
    bool flag = static_cast<bool>(data[0] % 2);
    imc->IsPanelShown(panelInfo, flag);
}

void FUZZPrintLogIfAceTimeout(sptr<InputMethodController> imc, int64_t start)
{
    imc->PrintLogIfAceTimeout(start);
}
} // namespace OHOS

/* Fuzzer entry point */
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
{
    /* Run your code on data */
    std::string fuzzedString(data, data + size);
    std::u16string fuzzedU16String = u"insert text";

    auto fuzzedInt = static_cast<int>(size);
    auto fuzzedInt32 = static_cast<int32_t>(size);
    auto fuzzedUint32 = static_cast<uint32_t>(size);
    auto fuzzedint64 = static_cast<int64_t>(size);
    auto fuzzedDouble = static_cast<double>(size);
    auto fuzzedTrigger = static_cast<SwitchTrigger>(size);

    OHOS::sptr<InputMethodController> imc = InputMethodController::GetInstance();

    OHOS::TestListInputMethod(imc);
    OHOS::TestListInputMethodSubtype(imc, fuzzedString, fuzzedUint32);
    OHOS::TestOnSelectionChange(imc, fuzzedU16String, fuzzedInt, fuzzedDouble);
    OHOS::TestOnConfigurationChange(imc);
    OHOS::TestSwitchInputMethod(fuzzedTrigger, imc, fuzzedString);
    OHOS::TestSetCallingWindow(imc, fuzzedUint32);
    OHOS::TestDispatchKeyEvent(imc, fuzzedInt32);
    OHOS::TestShowSomething(imc);
    OHOS::FUZZHideInput(imc);
    OHOS::FUZZShowInput(imc);
    OHOS::FUZZRestore(imc);
    OHOS::InputType(imc);
    OHOS::FUZZIsPanelShown(imc, data);
    OHOS::FUZZPrintLogIfAceTimeout(imc, fuzzedint64);
    OHOS::TestUpdateListenEventFlag(imc, fuzzedUint32);
    return 0;
}