summaryrefslogtreecommitdiff
path: root/icu4c/source/test/fuzzer/plurrule_fuzzer.cpp
blob: 634d505aedeb35ab1649c869816511895f9cde20 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// © 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;
}