forked from soot-oss/soot
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathLabeledUnitPrinter.java
More file actions
128 lines (109 loc) · 3.51 KB
/
LabeledUnitPrinter.java
File metadata and controls
128 lines (109 loc) · 3.51 KB
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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
package soot;
/*-
* #%L
* Soot - a J*va Optimization Framework
* %%
* Copyright (C) 2003 - 2004 Ondrej Lhotak
* %%
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation, either version 2.1 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Lesser Public License for more details.
*
* You should have received a copy of the GNU General Lesser Public
* License along with this program. If not, see
* <http://www.gnu.org/licenses/lgpl-2.1.html>.
* #L%
*/
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import soot.util.Chain;
/**
* UnitPrinter implementation for representations that have labelled stmts, such as Jimple, Grimp, and Baf
*/
public abstract class LabeledUnitPrinter extends AbstractUnitPrinter {
/**
* branch targets
*/
protected Map<Unit, String> labels;
/**
* for unit references in Phi nodes
*/
protected Map<Unit, String> references;
protected String labelIndent = "\u0020\u0020\u0020\u0020\u0020";
public LabeledUnitPrinter(Body b) {
createLabelMaps(b);
}
public Map<Unit, String> labels() {
return labels;
}
public Map<Unit, String> references() {
return references;
}
@Override
public void unitRef(Unit u, boolean branchTarget) {
String oldIndent = getIndent();
// normal case, ie labels
if (branchTarget) {
setIndent(labelIndent);
handleIndent();
setIndent(oldIndent);
String label = labels.get(u);
if (label == null || "<unnamed>".equals(label)) {
label = "[?= " + u + "]";
}
output.append(label);
} else {
// refs to control flow predecessors (for Shimple)
String ref = references.get(u);
if (startOfLine) {
String newIndent = "(" + ref + ")" + indent.substring(ref.length() + 2);
setIndent(newIndent);
handleIndent();
setIndent(oldIndent);
} else {
output.append(ref);
}
}
}
private void createLabelMaps(Body body) {
Chain<Unit> units = body.getUnits();
labels = new HashMap<Unit, String>(units.size() * 2 + 1, 0.7f);
references = new HashMap<Unit, String>(units.size() * 2 + 1, 0.7f);
// Create statement name table
Set<Unit> labelStmts = new HashSet<Unit>();
Set<Unit> refStmts = new HashSet<Unit>();
// Build labelStmts and refStmts
for (UnitBox box : body.getAllUnitBoxes()) {
Unit stmt = box.getUnit();
if (box.isBranchTarget()) {
labelStmts.add(stmt);
} else {
refStmts.add(stmt);
}
}
// left side zero padding for all labels
// this simplifies debugging the jimple code in simple editors, as it
// avoids the situation where a label is the prefix of another label
final int maxDigits = 1 + (int) Math.log10(labelStmts.size());
final String formatString = "label%0" + maxDigits + "d";
int labelCount = 0;
int refCount = 0;
// Traverse the stmts and assign a label if necessary
for (Unit s : units) {
if (labelStmts.contains(s)) {
labels.put(s, String.format(formatString, ++labelCount));
}
if (refStmts.contains(s)) {
references.put(s, Integer.toString(refCount++));
}
}
}
}