yetangitu/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2025-10-05 02:39:44 +02:00
Code Issues Releases Wiki Activity

fixed a few minor issues with AST graphs

Browse source
This commit is contained in:
ghidravore 2021-08-10 14:37:18 -04:00
parent fe6006c300
commit 269ca50d33
3 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

13
Ghidra/Features/Decompiler/ghidra_scripts/GraphAST.java
View file

@ -43,8 +43,8 @@ public class GraphAST extends GhidraScript {
protected static final String ADDRESS_TIED = "Address Tied";
protected static final String OP = "Op";
protected static final String TYPE_OUTPUT = "Output";
protected static final String TYPE_INPUT = "Input";
protected static final String WITHIN_BLOCK = "Within Block";
protected static final String BETWEEN_BLOCK = "Between Block";
private Function func;
private AttributedGraph graph;
protected HighFunction high;
@ -80,8 +80,8 @@ public class GraphAST extends GhidraScript {
.vertexType(ADDRESS_TIED)
.vertexType(OP)
.edgeType(DEFAULT)
.edgeType(TYPE_OUTPUT)
.edgeType(TYPE_INPUT)
.edgeType(WITHIN_BLOCK)
.edgeType(BETWEEN_BLOCK)
.build();
GraphDisplayOptions displayOptions = new GraphDisplayOptionsBuilder(graphType)
@ -92,6 +92,7 @@ public class GraphAST extends GhidraScript {
.defaultVertexShape(VertexShape.ELLIPSE)
.defaultLayoutAlgorithm("Hierarchical MinCross Coffman Graham")
.useIcons(false)
.arrowLength(15)
.labelPosition(GraphLabelPosition.SOUTH)
.shapeOverrideAttribute(SHAPE_ATTRIBUTE)
.vertex(DEFAULT, VertexShape.ELLIPSE, WebColors.RED)
@ -102,8 +103,8 @@ public class GraphAST extends GhidraScript {
.vertex(ADDRESS_TIED, VertexShape.ELLIPSE, WebColors.ORANGE)
.vertex(OP, VertexShape.RECTANGLE, WebColors.RED)
.edge(DEFAULT, WebColors.BLUE)
.edge(TYPE_OUTPUT, WebColors.BLACK)
.edge(TYPE_INPUT, WebColors.RED)
.edge(WITHIN_BLOCK, WebColors.BLACK)
.edge(BETWEEN_BLOCK, WebColors.RED)
.build();
graph = new AttributedGraph("AST Graph", graphType);

4
Ghidra/Features/Decompiler/ghidra_scripts/GraphASTAndFlow.java
View file

@ -77,7 +77,7 @@ public class GraphASTAndFlow extends GraphAST {
}
if (prev != null && map.containsKey(prev) && map.containsKey(next)) {
AttributedEdge edge = createEdge(map.get(prev), map.get(next));
edge.setEdgeType(TYPE_OUTPUT);
edge.setEdgeType(WITHIN_BLOCK);
}
prev = next;
}
@ -92,7 +92,7 @@ public class GraphASTAndFlow extends GraphAST {
PcodeBlock in = block.getIn(i);
if (last.containsKey(in)) {
AttributedEdge edge = createEdge(last.get(in), first.get(block));
edge.setEdgeType(TYPE_INPUT);
edge.setEdgeType(BETWEEN_BLOCK);
}
}
}

3
Ghidra/Framework/Graph/src/main/java/ghidra/service/graph/VertexShape.java
View file

@ -37,7 +37,6 @@ public abstract class VertexShape {
public static VertexShape HEXAGON = new HexagonVertexShape(SIZE);
public static VertexShape OCTAGON = new OctagonVertexShape(SIZE);
private Shape cachedShape;
private String name;
private int size;
@ -169,7 +168,7 @@ public abstract class VertexShape {
}
protected Shape createShape() {
return new Ellipse2D.Double(-1.0, 1.0, 2.0, 2.0);
return new Ellipse2D.Double(-1.0, -1.0, 2.0, 2.0);
}
@Override