dev

main.py

@@ -6,11 +6,11 @@ import tempfile
 
 def new_binary(file):
 
-    with tempfile.TemporaryDirectory() as TEMPDIR, tempfile.TemporaryDirectory() as OUTPUTDIR:
-        shutil.copy2(file.name, TEMPDIR)
+    with tempfile.TemporaryDirectory() as TEMP_DIR, tempfile.TemporaryDirectory() as OUTPUT_DIR:
+        shutil.copy2(file.name, TEMP_DIR)
 
         # python3 generate.py --ghidra PATH_TO_GHIDRA_ANALYZEHEADLESS -t NUM_THREADS -n [NUM_FILES|None] -b BINARIES_DIR -o OUTPUT_DIR
-        output = subprocess.check_output(f"python /DIRTY/dataset-gen-ghidra/generate.py --ghidra /ghidra/support/analyzeHeadless -t 1 -b {TEMPDIR} -o {OUTPUT_DIR}")
+        output = subprocess.check_output(f"python /DIRTY/dataset-gen-ghidra/generate.py --ghidra /ghidra/support/analyzeHeadless -t 1 -b {TEMP_DIR} -o {OUTPUT_DIR}")
         print(output)