FilesExpand file tree

 master

/

Generate.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

113 lines (95 loc) · 4.47 KB

 master

/

Generate.java

Copy path

Top

File metadata and controls

• Code
• Blame

113 lines (95 loc) · 4.47 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

package soot;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import soot.Body;

import soot.BodyTransformer;

import soot.G;

import soot.PackManager;

import soot.PhaseOptions;

import soot.SootMethod;

import soot.Transform;

import soot.Unit;

import soot.jimple.JimpleBody;

import soot.options.Options;

import soot.toolkits.graph.DirectedGraph;

import soot.toolkits.graph.UnitGraph;

import soot.util.cfgcmd.AltClassLoader;

import soot.util.cfgcmd.CFGGraphType;

import soot.util.cfgcmd.CFGIntermediateRep;

import soot.util.cfgcmd.CFGToDotGraph;

import soot.util.dot.DotGraph;

public class Generate extends BodyTransformer {

private static final String altClassPathOptionName = "alt-class-path";

private static final String graphTypeOptionName = "graph-type";

private static final String defaultGraph = "BriefUnitGraph";

private static final String irOptionName = "ir";

private static final String defaultIR = "jimple";

private static final String multipageOptionName = "multipages";

private static final String briefLabelOptionName = "brief";

private CFGGraphType graphtype;

private CFGIntermediateRep ir;

private CFGToDotGraph drawer;

protected void internalTransform(Body b, String phaseName, Map<String, String> options) {

initialize(options);

System.out.println(options);

System.out.println(b);

Body body = ir.getBody((JimpleBody) b);

//System.out.println(body);

print_cfg(body);

}

public static void main(String[] args) {

Generate viewer = new Generate();

Transform printTransform = new Transform("jtp.printcfg", viewer);

printTransform.setDeclaredOptions("enabled " + altClassPathOptionName + ' ' + graphTypeOptionName + ' '

+ irOptionName + ' ' + multipageOptionName + ' ' + briefLabelOptionName + ' ');

printTransform.setDefaultOptions("enabled " + altClassPathOptionName + ": " + graphTypeOptionName + ':'

+ defaultGraph + ' ' + irOptionName + ':' + defaultIR + ' ' + multipageOptionName + ":false " + ' '

+ briefLabelOptionName + ":false ");

PackManager.v().getPack("jtp").add(printTransform);

//args = viewer.parse_options(args);

String[] soot_args = new String[3];

//soot_args[0] = "-cp";

soot_args[0] = "--soot-classpath";

soot_args[1] = ".;E:/JavaJdk/JDK1.7/jre/lib/rt.jar";

soot_args[2] = "TriangleClass.Triangle";

soot.Main.main(soot_args);

}

private void initialize(Map<String, String> options) {

if (drawer == null) {

drawer = new CFGToDotGraph();

drawer.setBriefLabels(PhaseOptions.getBoolean(options, briefLabelOptionName));

drawer.setOnePage(!PhaseOptions.getBoolean(options, multipageOptionName));

drawer.setUnexceptionalControlFlowAttr("color", "black");

drawer.setExceptionalControlFlowAttr("color", "red");

drawer.setExceptionEdgeAttr("color", "lightgray");

drawer.setShowExceptions(Options.v().show_exception_dests());

ir = CFGIntermediateRep.getIR(PhaseOptions.getString(options, irOptionName));

graphtype = CFGGraphType.getGraphType(PhaseOptions.getString(options, graphTypeOptionName));

AltClassLoader.v().setAltClassPath(PhaseOptions.getString(options, altClassPathOptionName));

AltClassLoader.v().setAltClasses(

new String[] { "soot.toolkits.graph.ArrayRefBlockGraph", "soot.toolkits.graph.Block",

"soot.toolkits.graph.Block$AllMapTo", "soot.toolkits.graph.BlockGraph",

"soot.toolkits.graph.BriefBlockGraph", "soot.toolkits.graph.BriefUnitGraph",

"soot.toolkits.graph.CompleteBlockGraph", "soot.toolkits.graph.CompleteUnitGraph",

"soot.toolkits.graph.TrapUnitGraph", "soot.toolkits.graph.UnitGraph",

"soot.toolkits.graph.ZonedBlockGraph", });

}

}

protected void print_cfg(Body body) {

DirectedGraph<Unit> graph = graphtype.buildGraph(body);

//System.out.println(graph);

DotGraph canvas = graphtype.drawGraph(drawer, graph, body);

//GenerateFlow gen = new GenerateFlow((UnitGraph) graph);

String methodname = body.getMethod().getSubSignature();

String classname = body.getMethod().getDeclaringClass().getName().replaceAll("\\$", "\\.");

String filename = soot.SourceLocator.v().getOutputDir();

if (filename.length() > 0) {

filename = filename + java.io.File.separator;

}

filename = filename + classname + " " + methodname.replace(java.io.File.separatorChar, '.') + DotGraph.DOT_EXTENSION;

G.v().out.println("Generate dot file in " + filename);

canvas.plot(filename);

}

}