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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
import sys
from typing import List
import atheris
from fuzz_helpers import EnhancedFuzzedDataProvider
with atheris.instrument_imports():
import dateparser
import re
import pytz
import dateparser.data
import dateparser.parser
language_codes = dateparser.data.languages_info.language_order
directives = [
"%a",
"%A",
"%w",
"%d",
"%b",
"%B",
"%m",
"%y",
"%Y",
"%H",
"%I",
"%p",
"%M",
"%S",
"%f",
"%z",
"%Z",
"%j",
"%U",
"%W",
"%c",
"%x",
"%X",
"%%",
"%G",
"%u",
"%V",
"%:Z",
]
locale_codes = ["fr-PF", "qu-EC", "af-NA"]
date_order = list(dateparser.parser.date_order_chart.keys())
timezone = list(pytz.all_timezones)
preferred_date = ["last", "first", "current"]
preferred_dates_from = ["past", "future", "current_period"]
parsers = [
"timestamp",
"negative-timestamp",
"relative-time",
"custom-formats",
"absolute-time",
"no-spaces-time",
]
def _get_format_strings(fdp: EnhancedFuzzedDataProvider) -> List[str]:
format_strings = []
for _ in range(fdp.ConsumeIntInRange(0, 5)):
format_strings.append(fdp.ConsumeString(1).join(fdp.ConsumeSublist(directives)))
return format_strings
def TestOneInput(data):
fdp = EnhancedFuzzedDataProvider(data)
settings = {
"DATE_ORDER": fdp.PickValueInList(date_order),
"PREFER_LOCALE_DATE_ORDER": fdp.ConsumeBool(),
"TIMEZONE": fdp.PickValueInList(timezone),
"TO_TIMEZONE": fdp.PickValueInList(timezone),
"RETURN_AS_TIMEZONE_AWARE": fdp.ConsumeBool(),
"PREFER_MONTH_OF_YEAR": fdp.PickValueInList(preferred_date),
"PREFER_DAY_OF_MONTH": fdp.PickValueInList(preferred_date),
"PREFER_DATES_FROM": fdp.PickValueInList(preferred_dates_from),
"RELATIVE_BASE": fdp.ConsumeDate(),
"STRICT_PARSING": fdp.ConsumeBool(),
"REQUIRE_PARTS": [],
"SKIP_TOKENS": [
fdp.ConsumeRandomString() for _ in range(fdp.ConsumeIntInRange(0, 3))
],
"NORMALIZE": fdp.ConsumeBool(),
"RETURN_TIME_AS_PERIOD": fdp.ConsumeBool(),
"PARSERS": fdp.ConsumeSublist(parsers),
"DEFAULT_LANGUAGES": fdp.ConsumeSublist(language_codes),
"LANGUAGE_DETECTION_CONFIDENCE_THRESHOLD": fdp.ConsumeProbability(),
}
try:
dateparser.parse(
fdp.ConsumeRandomString(),
date_formats=_get_format_strings(fdp),
languages=fdp.ConsumeSublist(language_codes),
locales=fdp.ConsumeSublist(locale_codes),
region=fdp.ConsumeString(2),
settings=settings,
)
except re.error:
return -1
def main():
atheris.Setup(sys.argv, TestOneInput)
atheris.Fuzz()
if __name__ == "__main__":
main()
|
