Merge pull request #1914 from DaveLak/add-diff-fuzz-target

Add `Diff` Fuzz Target

Author: Byron

fuzzing/fuzz-targets/fuzz_diff.py

@@ -0,0 +1,83 @@
+importsys
+importos
+importio
+importtempfile
+frombinasciiimportErrorasBinasciiError
+
+importatheris
+
+ifgetattr(sys, "frozen", False) andhasattr(sys, "_MEIPASS"):
+path_to_bundled_git_binary=os.path.abspath(os.path.join(os.path.dirname(__file__), "git"))
+os.environ["GIT_PYTHON_GIT_EXECUTABLE"] =path_to_bundled_git_binary
+
+withatheris.instrument_imports():
+fromgitimportRepo, Diff
+
+
+classBytesProcessAdapter:
+"""Allows bytes to be used as process objects returned by subprocess.Popen."""
+
+def__init__(self, input_string):
+self.stdout=io.BytesIO(input_string)
+self.stderr=io.BytesIO()
+
+defwait(self):
+return0
+
+poll=wait
+
+
+defTestOneInput(data):
+fdp=atheris.FuzzedDataProvider(data)
+
+withtempfile.TemporaryDirectory() astemp_dir:
+repo=Repo.init(path=temp_dir)
+try:
+diff=Diff(
+repo,
+a_rawpath=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())),
+b_rawpath=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())),
+a_blob_id=fdp.ConsumeBytes(20),
+b_blob_id=fdp.ConsumeBytes(20),
+a_mode=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())),
+b_mode=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())),
+new_file=fdp.ConsumeBool(),
+deleted_file=fdp.ConsumeBool(),
+copied_file=fdp.ConsumeBool(),
+raw_rename_from=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())),
+raw_rename_to=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())),
+diff=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())),
+change_type=fdp.PickValueInList(["A", "D", "C", "M", "R", "T", "U"]),
+score=fdp.ConsumeIntInRange(0, fdp.remaining_bytes()),
+ )
+exceptBinasciiError:
+return-1
+exceptAssertionErrorase:
+if"Require 20 byte binary sha, got"instr(e):
+return-1
+else:
+raisee
+
+_=diff.__str__()
+_=diff.a_path
+_=diff.b_path
+_=diff.rename_from
+_=diff.rename_to
+_=diff.renamed_file
+
+diff_index=diff._index_from_patch_format(
+repo, proc=BytesProcessAdapter(fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())))
+ )
+
+diff._handle_diff_line(
+lines_bytes=fdp.ConsumeBytes(fdp.ConsumeIntInRange(0, fdp.remaining_bytes())), repo=repo, index=diff_index
+ )
+
+
+defmain():
+atheris.Setup(sys.argv, TestOneInput)
+atheris.Fuzz()
+
+
+if__name__=="__main__":
+main()