3 files changed, 73 insertions, 0 deletions
diff --git a/tools/ghidra/ExportAiVarSetLoader.java b/tools/ghidra/ExportAiVarSetLoader.java
new file mode 100644
index 0000000..70c6fa4
--- /dev/null
+++ b/tools/ghidra/ExportAiVarSetLoader.java
@@ -0,0 +1,23 @@
+// Emits the text `.var` loader called after the AI script loader selects a
+// bundle-local or shared varset. Run headless; the original PE remains read only.
+import ghidra.app.decompiler.DecompInterface;
+import ghidra.app.script.GhidraScript;
+import ghidra.program.model.address.Address;
+import ghidra.program.model.listing.Function;
+
+public class ExportAiVarSetLoader extends GhidraScript {
+    private static final long ADDRESS = 0x10011ea0L;
+
+    @Override
+    public void run() throws Exception {
+        Address address = currentProgram.getAddressFactory().getDefaultAddressSpace()
+            .getAddress(ADDRESS);
+        Function function = currentProgram.getFunctionManager().getFunctionAt(address);
+        println("===== AI varset loader =====");
+        if (function == null) { println("missing"); return; }
+        DecompInterface decompiler = new DecompInterface();
+        decompiler.openProgram(currentProgram);
+        println(decompiler.decompileFunction(function, 60, monitor).getDecompiledFunction().getC());
+        decompiler.dispose();
+    }
+}
diff --git a/tools/ghidra/ExportAiVmHandler30.java b/tools/ghidra/ExportAiVmHandler30.java
new file mode 100644
index 0000000..81918ff
--- /dev/null
+++ b/tools/ghidra/ExportAiVmHandler30.java
@@ -0,0 +1,23 @@
+// Emits Handler(30), the most frequent non-sentinel selector in the GOG
+// compiled-script corpus. Run headless; the original PE remains read only.
+import ghidra.app.decompiler.DecompInterface;
+import ghidra.app.script.GhidraScript;
+import ghidra.program.model.address.Address;
+import ghidra.program.model.listing.Function;
+
+public class ExportAiVmHandler30 extends GhidraScript {
+    private static final long ADDRESS = 0x1000c266L;
+
+    @Override
+    public void run() throws Exception {
+        Address address = currentProgram.getAddressFactory().getDefaultAddressSpace()
+            .getAddress(ADDRESS);
+        Function function = currentProgram.getFunctionManager().getFunctionAt(address);
+        println("===== AI VM Handler(30) =====");
+        if (function == null) { println("missing"); return; }
+        DecompInterface decompiler = new DecompInterface();
+        decompiler.openProgram(currentProgram);
+        println(decompiler.decompileFunction(function, 60, monitor).getDecompiledFunction().getC());
+        decompiler.dispose();
+    }
+}
diff --git a/tools/ghidra/FindAiVmHandler30Callback.java b/tools/ghidra/FindAiVmHandler30Callback.java
new file mode 100644
index 0000000..a6da0ad
--- /dev/null
+++ b/tools/ghidra/FindAiVmHandler30Callback.java
@@ -0,0 +1,27 @@
+// Finds writers and readers of Handler(30)'s callback pointer, then decompiles
+// their containing functions. Run headless; the original PE remains read only.
+import ghidra.app.decompiler.DecompInterface;
+import ghidra.app.script.GhidraScript;
+import ghidra.program.model.address.Address;
+import ghidra.program.model.listing.Function;
+import ghidra.program.model.symbol.Reference;
+
+public class FindAiVmHandler30Callback extends GhidraScript {
+    private static final long ADDRESS = 0x100555e4L;
+
+    @Override
+    public void run() throws Exception {
+        Address address = currentProgram.getAddressFactory().getDefaultAddressSpace()
+            .getAddress(ADDRESS);
+        DecompInterface decompiler = new DecompInterface();
+        decompiler.openProgram(currentProgram);
+        for (Reference reference : currentProgram.getReferenceManager().getReferencesTo(address)) {
+            Function function = currentProgram.getFunctionManager()
+                .getFunctionContaining(reference.getFromAddress());
+            println("===== callback reference " + reference.getFromAddress() + " =====");
+            if (function == null) { println("no containing function"); continue; }
+            println(decompiler.decompileFunction(function, 60, monitor).getDecompiledFunction().getC());
+        }
+        decompiler.dispose();
+    }
+}