From 8b50e8f07b3d6487f07e91f244b1bfbd2afd9270 Mon Sep 17 00:00:00 2001
From: DavidKorczynski <david@adalogics.com>
Date: Fri, 19 Aug 2022 14:36:30 +0100
Subject: [PATCH] flask-restx: initial integration (#8229)

* flask-restx: initial integration

* nit
---
 projects/flask-restx/Dockerfile       | 22 ++++++++++
 projects/flask-restx/build.sh         | 24 +++++++++++
 projects/flask-restx/fuzz_reqparse.py | 58 +++++++++++++++++++++++++++
 projects/flask-restx/project.yaml     | 12 ++++++
 4 files changed, 116 insertions(+)
 create mode 100644 projects/flask-restx/Dockerfile
 create mode 100755 projects/flask-restx/build.sh
 create mode 100644 projects/flask-restx/fuzz_reqparse.py
 create mode 100644 projects/flask-restx/project.yaml

diff --git a/projects/flask-restx/Dockerfile b/projects/flask-restx/Dockerfile
new file mode 100644
index 000000000..47707c052
--- /dev/null
+++ b/projects/flask-restx/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 make autoconf automake libtool
+RUN pip3 install --upgrade pip && pip3 install werkzeug==2.1.2
+RUN git clone --depth 1 https://github.com/python-restx/flask-restx flask-restx
+WORKDIR flask-restx
+COPY build.sh *.py $SRC/
diff --git a/projects/flask-restx/build.sh b/projects/flask-restx/build.sh
new file mode 100755
index 000000000..6c299d28c
--- /dev/null
+++ b/projects/flask-restx/build.sh
@@ -0,0 +1,24 @@
+#!/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.
+#
+################################################################################
+
+sed -i "s/werkzeug !=2.0.0/werkzeug ==2.1.2/g" ./requirements/install.pip
+cd $SRC/flask-restx
+
+pip3 install .
+for fuzzer in $(find $SRC -name 'fuzz_*.py'); do
+  compile_python_fuzzer $fuzzer
+done
diff --git a/projects/flask-restx/fuzz_reqparse.py b/projects/flask-restx/fuzz_reqparse.py
new file mode 100644
index 000000000..7bd126c26
--- /dev/null
+++ b/projects/flask-restx/fuzz_reqparse.py
@@ -0,0 +1,58 @@
+#!/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 os
+import sys
+import atheris
+import traceback
+
+from werkzeug.wrappers import Request
+import flask
+from flask_restx.reqparse import RequestParser
+
+
+def TestOneInput(data):
+    fdp = atheris.FuzzedDataProvider(data)
+    try:
+        req = Request.from_values(fdp.ConsumeUnicodeNoSurrogates(1024))
+    except:
+        # We don't care about errors in werkzeug Request.
+        return
+
+    app = flask.Flask("fuzz")
+    try:
+        with app.app_context():
+            parser = RequestParser()
+            parser.add_argument("ex1")
+            parser.add_argument("ex2")
+            # Parse arbitrary req
+            parser.parse_args(req)
+    except Exception as e2:
+        tb = traceback.format_exc()
+        # We don't care about werkzeug errors
+        if "werkzeug" in str(tb):
+            pass
+        else:
+            raise e2
+
+
+def main():
+    atheris.instrument_all()
+    atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
+    atheris.Fuzz()
+
+
+if __name__ == "__main__":
+    main()
diff --git a/projects/flask-restx/project.yaml b/projects/flask-restx/project.yaml
new file mode 100644
index 000000000..d82885656
--- /dev/null
+++ b/projects/flask-restx/project.yaml
@@ -0,0 +1,12 @@
+homepage: "https://github.com/python-restx/flask-restx"
+language: python
+main_repo: "https://github.com/python-restx/flask-restx"
+fuzzing_engines:
+- libfuzzer
+sanitizers:
+- address
+- undefined
+vendor_ccs:
+- david@adalogics.com
+- adam@adalogics.com
+- arthur.chan@adalogics.com