Support Ghidra processor modules in the icicle python module directory

icicle-emu · Aug 11, 2023 · 621572f · 621572f
1 parent 050670e
commit 621572f
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -21,4 +21,4 @@ tracing = "*"
 tracing-subscriber = "0.3.17"
 
 [build-dependencies]
-pyo3-build-config = "0.18.3"
+pyo3-build-config = "0.18.3"
diff --git a/src/lib.rs b/src/lib.rs
@@ -515,6 +515,18 @@ class MemoryError(Exception):
         self.code = code
     def __str__(self):
         return f"{super().__str__()}: {self.code}"
+
+def __ghidra_init():
+    import os
+    for path in __path__ + [os.getenv("GHIDRA_SRC")]:
+        processors_dir = os.path.join(path, "Ghidra/Processors")
+        if os.path.isdir(processors_dir):
+            os.putenv("GHIDRA_SRC", path)
+            os.environ["GHIDRA_SRC"] = path
+            return
+    raise FileNotFoundError("Ghidra processors not found")
+
+__ghidra_init()
 "#, "icicle_exceptions.py", "icicle")?;
     Ok(())
 }