don't save ghidra proj in bin dir

purseclab · Aug 3, 2024 · 8833fdc · 8833fdc
1 parent 84d6271
commit 8833fdc
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/patcherex2/components/binary_analyzers/ghidra.py b/src/patcherex2/components/binary_analyzers/ghidra.py
@@ -1,4 +1,5 @@
 import logging
+import tempfile
 
 from .binary_analyzer import BinaryAnalyzer
 
@@ -9,8 +10,11 @@ class Ghidra(BinaryAnalyzer):
     def __init__(self, binary_path: str, **kwargs):
         import pyhidra
 
-        self.ctx = pyhidra.open_program(binary_path)
-        self.flatapi = self.ctx.__enter__()
+        self.temp_proj_dir_ctx = tempfile.TemporaryDirectory()
+        self.temp_proj_dir = self.temp_proj_dir_ctx.__enter__()
+
+        self.pyhidra_ctx = pyhidra.open_program(binary_path, self.temp_proj_dir)
+        self.flatapi = self.pyhidra_ctx.__enter__()
         self.currentProgram = self.flatapi.getCurrentProgram()
 
         import ghidra
@@ -20,7 +24,8 @@ def __init__(self, binary_path: str, **kwargs):
         self.bbm = self.ghidra.program.model.block.BasicBlockModel(self.currentProgram)
 
     def shutdown(self):
-        self.ctx.__exit__(None, None, None)
+        self.pyhidra_ctx.__exit__(None, None, None)
+        self.temp_proj_dir_ctx.__exit__(None, None, None)
 
     def normalize_addr(self, addr):
         addr = addr.getOffset()