// © 2023 and later: Unicode, Inc. and others.

// License & terms of use: http://www.unicode.org/copyright.html

// Fuzzer for ICU PluralRules.

#include <cstring>

#include "fuzzer_utils.h"

#include "unicode/plurrule.h"

#include "unicode/locid.h"

void TestPluralRules(icu::PluralRules* pp, int32_t number, double dbl,  UErrorCode& status) {

    if (U_FAILURE(status)) {

        return;

    }

    pp->select(number);

    pp->select(dbl);

}

void TestPluralRulesWithLocale(

    const icu::Locale& locale, int32_t number, double dbl,  UPluralType type, UErrorCode& status) {

    if (U_FAILURE(status)) {

        return;

    }

    std::unique_ptr<icu::PluralRules> pp(icu::PluralRules::forLocale(locale, status));

    TestPluralRules(pp.get(), number, dbl, status);

    status = U_ZERO_ERROR;

    pp.reset(icu::PluralRules::forLocale(locale, type, status));

    TestPluralRules(pp.get(), number, dbl, status);

    type = static_cast<UPluralType>(

        static_cast<int>(type) % (static_cast<int>(UPLURAL_TYPE_COUNT)));

    status = U_ZERO_ERROR;

    pp.reset(icu::PluralRules::forLocale(locale, type, status));

    TestPluralRules(pp.get(), number, dbl, status);

}

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

    uint16_t rnd;

    int32_t number;

    double dbl;

    UPluralType  type;

    if (size > 5000) {

        size = 5000;

    }

    if (size < sizeof(rnd) + sizeof(number) + sizeof(dbl) + sizeof(type)) return 0;

    icu::StringPiece fuzzData(reinterpret_cast<const char *>(data), size);

    std::memcpy(&rnd, fuzzData.data(), sizeof(rnd));

    icu::Locale locale = GetRandomLocale(rnd);

    fuzzData.remove_prefix(sizeof(rnd));

    std::memcpy(&number, fuzzData.data(), sizeof(number));

    fuzzData.remove_prefix(sizeof(number));

    std::memcpy(&dbl, fuzzData.data(), sizeof(dbl));

    fuzzData.remove_prefix(sizeof(dbl));

    std::memcpy(&type, fuzzData.data(), sizeof(type));

    fuzzData.remove_prefix(sizeof(type));

    size_t len = fuzzData.size() / sizeof(char16_t);

    icu::UnicodeString text(false, reinterpret_cast<const char16_t*>(fuzzData.data()), len);

    UErrorCode status = U_ZERO_ERROR;

    std::unique_ptr<icu::PluralRules> pp(

        icu::PluralRules::createRules(text, status));

    TestPluralRules(pp.get(), number, dbl, status);

    status = U_ZERO_ERROR;

    TestPluralRulesWithLocale(locale, number, dbl, type, status);

    std::string str(fuzzData.data(), fuzzData.size()); // ensure null-terminate

                                                       // by std::string c_str

    icu::Locale locale2(str.c_str());

    status = U_ZERO_ERROR;

    TestPluralRulesWithLocale(locale2, number, dbl, type, status);

    return EXIT_SUCCESS;

}