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

Instrument test utility functions to increase fuzzer efficiency
gitpython-developers · May 15, 2024 · adf4bf0 · adf4bf0
2 parents 4e212c6 + 315a2fd
commit adf4bf0
Showing 1 changed file with 3 additions and 0 deletions.
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)