Improve the fuzzer wrt. the current atheris version

python-pillow · Aug 25, 2021 · 9d48143 · 9d48143
1 parent d50052a
commit 9d48143
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/Tests/oss-fuzz/fuzz_font.py b/Tests/oss-fuzz/fuzz_font.py
@@ -14,10 +14,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import sys
 
-import atheris_no_libfuzzer as atheris
-import fuzzers
+import atheris
+with atheris.instrument_imports():
+    import sys
+    import fuzzers
 
 
 def TestOneInput(data):
@@ -26,13 +27,12 @@ def TestOneInput(data):
     except Exception:
         # We're catching all exceptions because Pillow's exceptions are
         # directly inheriting from Exception.
-        return
-    return
+        pass
 
 
 def main():
     fuzzers.enable_decompressionbomb_error()
-    atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
+    atheris.Setup(sys.argv, TestOneInput)
     atheris.Fuzz()
     fuzzers.disable_decompressionbomb_error()
 

diff --git a/Tests/oss-fuzz/fuzz_pillow.py b/Tests/oss-fuzz/fuzz_pillow.py
@@ -14,10 +14,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import sys
 
-import atheris_no_libfuzzer as atheris
-import fuzzers
+import atheris
+with atheris.instrument_imports():
+    import sys
+    import fuzzers
 
 
 def TestOneInput(data):
@@ -26,13 +27,12 @@ def TestOneInput(data):
     except Exception:
         # We're catching all exceptions because Pillow's exceptions are
         # directly inheriting from Exception.
-        return
-    return
+        pass
 
 
 def main():
     fuzzers.enable_decompressionbomb_error()
-    atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
+    atheris.Setup(sys.argv, TestOneInput)
     atheris.Fuzz()
     fuzzers.disable_decompressionbomb_error()