Merge pull request #1915 from DaveLak/fix-blockers-in-fuzz-diff

Byron · web-flow · commit adf4bf0a81c3 · 2024-05-15T21:22:07.000+02:00
Instrument test utility functions to increase fuzzer efficiency
diff --git a/fuzzing/fuzz-targets/fuzz_diff.py b/fuzzing/fuzz-targets/fuzz_diff.py
@@ -17,16 +17,19 @@
 class BytesProcessAdapter:
     """Allows bytes to be used as process objects returned by subprocess.Popen."""
 
+    @atheris.instrument_func
     def __init__(self, input_string):
         self.stdout = io.BytesIO(input_string)
         self.stderr = io.BytesIO()
 
+    @atheris.instrument_func
     def wait(self):
         return 0
 
     poll = wait
 
 
+@atheris.instrument_func
 def TestOneInput(data):
     fdp = atheris.FuzzedDataProvider(data)
 
