mirror of https://github.com/google/oss-fuzz.git
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
# Copyright 2023 Google LLC
|
||
|
|
#
|
||
|
|
# 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.
|
||
|
|
"""Targets the pandas to_datetime function."""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import atheris
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
from pandas.errors import (
|
||
|
|
ParserError,
|
||
|
|
OutOfBoundsDatetime
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def TestOneInput(data):
|
||
|
|
fdp = atheris.FuzzedDataProvider(data)
|
||
|
|
|
||
|
|
try:
|
||
|
|
input_str = fdp.ConsumeUnicodeNoSurrogates(100)
|
||
|
|
errors = fdp.PickValueInList(['raise', 'coerce', 'ignore', fdp.ConsumeUnicodeNoSurrogates(20)])
|
||
|
|
unit = fdp.PickValueInList([None, 'ns', 'us', 'ms', 's', 'D', 'M', 'Y', fdp.ConsumeUnicodeNoSurrogates(20)])
|
||
|
|
origin = fdp.PickValueInList(['unix', 'julian', 'epoch', fdp.ConsumeUnicodeNoSurrogates(20)])
|
||
|
|
dayfirst = fdp.ConsumeBool()
|
||
|
|
yearfirst = fdp.ConsumeBool()
|
||
|
|
exact = fdp.ConsumeBool()
|
||
|
|
cache = fdp.ConsumeBool()
|
||
|
|
|
||
|
|
pd.to_datetime(
|
||
|
|
arg=input_str,
|
||
|
|
errors=errors,
|
||
|
|
unit=unit,
|
||
|
|
origin=origin,
|
||
|
|
dayfirst=dayfirst,
|
||
|
|
yearfirst=yearfirst,
|
||
|
|
exact=exact,
|
||
|
|
cache=cache
|
||
|
|
)
|
||
|
|
|
||
|
|
except (
|
||
|
|
ParserError, # When parsing a date from string fails.
|
||
|
|
ValueError, # When another datetime conversion error happens.
|
||
|
|
OutOfBoundsDatetime # catching attempts to create a DatetimeIndex, which may raise from cast()
|
||
|
|
):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
atheris.Setup(sys.argv, TestOneInput)
|
||
|
|
atheris.instrument_all()
|
||
|
|
atheris.Fuzz()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|