icu4c/source/test/fuzzer/dtfmtsym_fuzzer.cpp

// © 2023 and later: Unicode, Inc. and others.
// License & terms of use: http://www.unicode.org/copyright.html

// Fuzzer for ICU DateFormatSymbols.

#include <cstring>

#include "fuzzer_utils.h"

#include "unicode/dtfmtsym.h"
#include "unicode/locid.h"

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
    uint16_t rnd;
    icu::DateFormatSymbols::DtContextType context;
    icu::DateFormatSymbols::DtWidthType width;
    if (size < sizeof(rnd) + sizeof(context) + sizeof(width)) 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(&context, fuzzData.data(), sizeof(context));
    fuzzData.remove_prefix(sizeof(context));
    icu::DateFormatSymbols::DtContextType context_mod =
        static_cast<icu::DateFormatSymbols::DtContextType>(
            context % icu::DateFormatSymbols::DtContextType::DT_CONTEXT_COUNT);
;
    std::memcpy(&width, fuzzData.data(), sizeof(width));
    fuzzData.remove_prefix(sizeof(width));
    icu::DateFormatSymbols::DtWidthType width_mod =
        static_cast<icu::DateFormatSymbols::DtWidthType>(
            width % icu::DateFormatSymbols::DtWidthType::DT_WIDTH_COUNT);

    size_t len = fuzzData.size() / sizeof(char16_t);
    icu::UnicodeString text(false, reinterpret_cast<const char16_t*>(fuzzData.data()), len);
    const icu::UnicodeString items[] = { text, text, text, text };

    UErrorCode status = U_ZERO_ERROR;
    std::unique_ptr<icu::DateFormatSymbols> dfs(
        new icu::DateFormatSymbols(locale, status));
    if (U_FAILURE(status)) {
        return EXIT_SUCCESS;
    }

    int32_t count;
    dfs->getEras(count);
    dfs->getEraNames(count);
    dfs->getNarrowEras(count);
    dfs->getMonths(count);
    dfs->getShortMonths(count);
    dfs->getMonths(count, context, width);
    dfs->getMonths(count, context_mod, width_mod);
    dfs->getWeekdays(count);
    dfs->getShortWeekdays(count);
    dfs->getWeekdays(count, context, width);
    dfs->getWeekdays(count, context_mod, width_mod);
    dfs->getShortWeekdays(count);
    dfs->getQuarters(count, context_mod, width_mod);
    dfs->getAmPmStrings(count);

    icu::UnicodeString output;
    dfs->getTimeSeparatorString(output);
    dfs->getYearNames(count, context, width);
    dfs->getYearNames(count, context_mod, width_mod);
    dfs->getZodiacNames(count, context, width);
    dfs->getZodiacNames(count, context_mod, width_mod);
    dfs->getLeapMonthPatterns(count);
    int32_t count2;
    dfs->getZoneStrings(count, count2);
    dfs->getLocalPatternChars(output);

    return EXIT_SUCCESS;
}