/*

 * Copyright (c) 2022 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.

 */



#include "accessibilitysystemabilityclientability_fuzzer.h"

#include "accessibility_system_ability_client_impl.h"

#include "securec.h"

#include "element_operator_fuzz_test.h"



namespace OHOS {

namespace Accessibility {



bool GetAbilityListFuzzTest(const uint8_t* data, size_t size)

{

    if (data == nullptr || size < DATA_MIN_SIZE) {

        return false;

    }



    auto instance = AccessibilitySystemAbilityClient::GetInstance();

    if (!instance) {

        return false;

    }



    size_t position = 0;

    uint32_t abilityTypes = 0;

    position += GetObject<uint32_t>(abilityTypes, &data[position], size - position);

    AbilityStateType stateType;

    GetObject<AbilityStateType>(stateType, &data[position], size - position);

    std::vector<AccessibilityAbilityInfo> infos;

    instance->GetAbilityList(abilityTypes, stateType, infos);

    return true;

}



bool GetEnabledAbilitiesFuzzTest(const uint8_t* data, size_t size)

{

    if (data == nullptr || size < DATA_MIN_SIZE) {

        return false;

    }



    auto instance = AccessibilitySystemAbilityClient::GetInstance();

    if (!instance) {

        return false;

    }



    size_t position = 0;

    char name[LEN + 1];

    name[LEN] = END_CHAR;

    for (size_t i = 0; i < LEN; i++) {

        position += GetObject<char>(name[i], &data[position], size - position);

    }

    std::string enabledAbility(name);

    std::vector<std::string> enabledAbilities;

    enabledAbilities.emplace_back(enabledAbility);

    instance->GetEnabledAbilities(enabledAbilities);

    return true;

}

} // namespace Accessibility

} // namespace OHOS



/* Fuzzer entry point */

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)

{

    /* Run your code on data */

    OHOS::Accessibility::GetAbilityListFuzzTest(data, size);

    OHOS::Accessibility::GetEnabledAbilitiesFuzzTest(data, size);

    return 0;

}