colton: developing ghidra backend tests

capa/main.py

@@ -1333,6 +1333,7 @@ def ghidra_main():
     
     globl_features: List[Tuple[Feature, Address]] = []
     globl_features.extend(capa.features.extractors.ghidra.global_.extract_os())
+    globl_features.extend(capa.features.extractors.ghidra.global_.extract_arch())
     print(globl_features)
 
 

tests/test_ghidra_features.py

@@ -0,0 +1,25 @@
+import logging
+
+import fixtures
+from fixtures import *
+
+import capa.main
+
+logger = logging.getLogger(__file__)
+
+# We need to skip the ghidra test if we cannot import ghidra modules, e.g., in GitHub CI.
+ghidra_present: bool = False
+try:
+    import ghidra.program.flatapi as flatapi
+    ghidraapi = flatapi.FlatProgramAPI(currentProgram) 
+
+    try:
+        current_program_test = ghidraapi.getCurrentProgram()
+    except RuntimeError as e:
+        logger.warning("Ghidra runtime not detected")
+    else:
+        ghidra_present = True
+except ImportError:
+    pass
+
+