From 33127a01cb5bdf03d797ca18112e5dfb32d19423 Mon Sep 17 00:00:00 2001
From: DavidKorczynski <david@adalogics.com>
Date: Fri, 22 Jul 2022 00:44:58 +0100
Subject: [PATCH] pytz: initial integration (#7730)

* pytz: initial integration

* Catch highest pytz exception
---
 projects/pytz/Dockerfile   | 22 +++++++++++++++
 projects/pytz/build.sh     | 29 ++++++++++++++++++++
 projects/pytz/fuzz_pytz.py | 55 ++++++++++++++++++++++++++++++++++++++
 projects/pytz/project.yaml | 12 +++++++++
 4 files changed, 118 insertions(+)
 create mode 100644 projects/pytz/Dockerfile
 create mode 100644 projects/pytz/build.sh
 create mode 100644 projects/pytz/fuzz_pytz.py
 create mode 100644 projects/pytz/project.yaml

diff --git a/projects/pytz/Dockerfile b/projects/pytz/Dockerfile
new file mode 100644
index 000000000..de5c3f86d
--- /dev/null
+++ b/projects/pytz/Dockerfile
@@ -0,0 +1,22 @@
+# Copyright 2022 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.
+#
+################################################################################
+
+FROM gcr.io/oss-fuzz-base/base-builder-python
+RUN apt-get update && apt-get install -y python3-distutils libpq-dev python3-wheel
+RUN git clone https://github.com/stub42/pytz
+RUN pip3 install wheel
+COPY build.sh *.py $SRC/
+WORKDIR pytz
diff --git a/projects/pytz/build.sh b/projects/pytz/build.sh
new file mode 100644
index 000000000..9c25cf006
--- /dev/null
+++ b/projects/pytz/build.sh
@@ -0,0 +1,29 @@
+#!/bin/bash -eu
+# Copyright 2022 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.
+#
+################################################################################
+
+# Ensure usage of OSS-Fuzz python
+sed -i "s/\/usr\/bin\/python/\/usr\/local\/bin\/python/g" Makefile
+
+#pip3 install wheel
+make
+cd build/dist
+python3 setup.py install
+
+# Build flask fuzzers
+for fuzzer in $(find $SRC -name 'fuzz_*.py'); do
+  compile_python_fuzzer $fuzzer
+done
diff --git a/projects/pytz/fuzz_pytz.py b/projects/pytz/fuzz_pytz.py
new file mode 100644
index 000000000..5804d5d3f
--- /dev/null
+++ b/projects/pytz/fuzz_pytz.py
@@ -0,0 +1,55 @@
+#!/usr/bin/python3
+# Copyright 2022 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.
+
+import atheris
+import sys
+
+with atheris.instrument_imports():
+  import pytz
+  from datetime import datetime
+  from pytz.exceptions import Error
+
+
+def TestOneInput(data):
+  fdp = atheris.FuzzedDataProvider(data)
+
+  try:
+    d1 = datetime(
+      fdp.ConsumeInt(4),
+      fdp.ConsumeInt(4),
+      fdp.ConsumeInt(4),
+      fdp.ConsumeInt(4),
+      fdp.ConsumeInt(4),
+      fdp.ConsumeInt(4)
+    )
+  except: # Catch everything
+    return
+
+  s1 = fdp.ConsumeString(sys.maxsize)
+  try:
+    east2 = pytz.timezone(s1)
+    east2.localize(d1)
+  except Error as e:
+    pass
+
+
+def main():
+  atheris.instrument_all()
+  atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
+  atheris.Fuzz()
+
+
+if __name__ == "__main__":
+  main()
diff --git a/projects/pytz/project.yaml b/projects/pytz/project.yaml
new file mode 100644
index 000000000..0754d2861
--- /dev/null
+++ b/projects/pytz/project.yaml
@@ -0,0 +1,12 @@
+fuzzing_engines:
+- libfuzzer
+homepage: https://github.com/stub42/pytz
+language: python
+main_repo: https://github.com/stub42/pytz
+sanitizers:
+- address
+- undefined
+vendor_ccs:
+- david@adalogics.com
+- adam@adalogics.com
+- athur.chan@adalogics.com