feat: tempfile cleanup for RParser (#3944)

* feat: added tempfile cleanup to RParser

Co-authored-by: Joydeep Tripathy <[email protected]>
Co-authored-by: Terri Oda <[email protected]>

Author: joydeep049

fuzz/fuzz_renv_lock.py

@@ -5,9 +5,10 @@
 This module contains fuzz testing for the RParser's handling of renv.lock files.
 """
 
+import os
+import shutil
 import sys
 import tempfile
-from pathlib import Path
 
 import atheris
 import atheris_libprotobuf_mutator
@@ -25,12 +26,13 @@
 logger = LOGGER.getChild("Fuzz")
 
 
-def RenvLockBuilder(data):
+def RenvLockBuilder(data, file_path):
     """
     This function converts the given data into a renv.lock file.
 
     Args:
         data (protobuf message): The protobuf message to convert and process.
+        file_path: The path of the file to build.
     """
     # Parse the JSON data
     json_data = MessageToDict(
@@ -89,13 +91,16 @@ def RenvLockBuilder(data):
         f.write("}\n")
 
 
-def TestParseData(data):
+def TestParseData(data, cve_db, logger, tmpdir):
     """
-    Fuzz testing function for the RParser's handling of renv.lock files.
-
+    Fuzz test the RustParser's handling of renv.lock files.
     Args:
-        data (protobuf message): The protobuf message to convert and process.
+        data (protobuf message): The protobuf message to convert to a renv.lock file.
+        cve_db: Object for the Database of CVE-BIN-TOOL.
+        logger: Logger object.
+        tmpdir: The temporary direct object.
     """
+    file_path = os.path.join(tmpdir, "renv.lock")
     try:
         RenvLockBuilder(data)
 
@@ -106,7 +111,20 @@ def TestParseData(data):
         return
 
 
-file_path = str(Path(tempfile.mkdtemp(prefix="cve-bin-tool-")) / "renv.lock")
-
-atheris_libprotobuf_mutator.Setup(sys.argv, TestParseData, proto=renv_lock_pb2.RenvLock)
-atheris.Fuzz()
+def main():
+    """Main Function to Run Fuzzing and Facilitate Tempfile cleanup."""
+    tmpdir = tempfile.mkdtemp(prefix="cve-bin-tool-fuzz-renv-")
+    try:
+        atheris_libprotobuf_mutator.Setup(
+            sys.argv,
+            lambda data: TestParseData(data, cve_db, logger, tmpdir),
+            proto=renv_lock_pb2.RenvLock,
+        )
+        atheris.Fuzz()
+    finally:
+        if os.path.exists(tmpdir):
+            shutil.rmtree(tmpdir)
+
+
+if __name__ == "__main__":
+    main()