Moving 'AST Graph' from scripting to an action in the decompiler window so that options are persistent. Also renamed it to 'Graph Data Flow'. Renamed existing graph action to 'Graph Control Flow'.

Ghidra/Features/Decompiler/ghidra_scripts/GraphASTScript.java

@@ -0,0 +1,77 @@
+/* ###
+ * IP: GHIDRA
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+//Decompile the function at the cursor, then build data-flow graph (AST)
+//@category PCode
+
+import ghidra.app.decompiler.*;
+import ghidra.app.plugin.core.decompile.actions.PCodeDfgGraphTask;
+import ghidra.app.script.GhidraScript;
+import ghidra.app.services.GraphDisplayBroker;
+import ghidra.framework.plugintool.PluginTool;
+import ghidra.program.model.listing.Function;
+import ghidra.program.model.pcode.HighFunction;
+import ghidra.util.Msg;
+
+public class GraphASTScript extends GhidraScript {
+
+	private Function func;
+	protected HighFunction high;
+
+	@Override
+	public void run() throws Exception {
+		PluginTool tool = state.getTool();
+		if (tool == null) {
+			println("Script is not running in GUI");
+		}
+		GraphDisplayBroker graphDisplayBroker = tool.getService(GraphDisplayBroker.class);
+		if (graphDisplayBroker == null) {
+			Msg.showError(this, tool.getToolFrame(), "GraphAST Error",
+				"No graph display providers found: Please add a graph display provider to your tool");
+			return;
+		}
+
+		func = this.getFunctionContaining(this.currentAddress);
+		if (func == null) {
+			Msg.showWarn(this, state.getTool().getToolFrame(), "GraphAST Error",
+				"No Function at current location");
+			return;
+		}
+
+		buildAST();
+		PCodeDfgGraphTask astGraphTask = createTask(graphDisplayBroker);
+		astGraphTask.monitoredRun(monitor);
+	}
+
+	protected PCodeDfgGraphTask createTask(GraphDisplayBroker graphDisplayBroker) {
+		return new PCodeDfgGraphTask(state.getTool(), graphDisplayBroker, high);
+	}
+
+	private void buildAST() throws DecompileException {
+		DecompileOptions options = new DecompileOptions();
+		DecompInterface ifc = new DecompInterface();
+		ifc.setOptions(options);
+
+		if (!ifc.openProgram(this.currentProgram)) {
+			throw new DecompileException("Decompiler",
+				"Unable to initialize: " + ifc.getLastMessage());
+		}
+		ifc.setSimplificationStyle("normalize");
+		DecompileResults res = ifc.decompileFunction(func, 30, null);
+		high = res.getHighFunction();
+
+	}
+
+}