Merge pull request #772 from DavidKorczynski/main

Added fuzzer to be run with OSS-Fuzz

1 files changed, 47 insertions, 0 deletions

diff --git a/jsonschema/tests/fuzz_validate.py b/jsonschema/tests/fuzz_validate.py
new file mode 100644
index 0000000..52675d1
--- /dev/null
+++ b/jsonschema/tests/fuzz_validate.py
@@ -0,0 +1,47 @@
+"""
+Fuzzing setup for OSS-Fuzz.
+
+See https://github.com/google/oss-fuzz/tree/master/projects/jsonschema for the
+other half of the setup here.
+"""
+import sys
+
+from hypothesis import given, strategies
+
+import jsonschema
+
+PRIM = strategies.one_of(
+    strategies.booleans(),
+    strategies.integers(),
+    strategies.floats(allow_nan=False, allow_infinity=False),
+    strategies.text(),
+)
+DICT = strategies.recursive(
+    base=(
+        strategies.booleans()
+        | strategies.dictionaries(strategies.text(), PRIM),
+    ),
+    extend=lambda inner: strategies.dictionaries(strategies.text(), inner),
+)
+
+
+@given(obj1=DICT, obj2=DICT)
+def test_schemas(obj1, obj2):
+    try:
+        jsonschema.validate(instance=obj1, schema=obj2)
+    except jsonschema.exceptions.ValidationError:
+        pass
+    except jsonschema.exceptions.SchemaError:
+        pass
+
+
+def main():
+    atheris.Setup(sys.argv,
+                  test_schemas.hypothesis.fuzz_one_input,
+                  enable_python_coverage=True)
+    atheris.Fuzz()
+
+
+if __name__ == "__main__":
+    import atheris
+    main()