java-decompiler: post-import cleanup (common fixes and optimizations)
This commit is contained in:
@@ -45,9 +45,9 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
|
||||
private List<ExceptionRangeCFG> exceptions;
|
||||
|
||||
private HashMap<BasicBlock, BasicBlock> subroutines;
|
||||
private Map<BasicBlock, BasicBlock> subroutines;
|
||||
|
||||
private HashSet<BasicBlock> finallyExits = new HashSet<BasicBlock>();
|
||||
private Set<BasicBlock> finallyExits = new HashSet<BasicBlock>();
|
||||
|
||||
// *****************************************************************************
|
||||
// constructors
|
||||
@@ -85,16 +85,16 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
|
||||
String new_line_separator = DecompilerContext.getNewLineSeparator();
|
||||
|
||||
StringBuffer buf = new StringBuffer();
|
||||
StringBuilder buf = new StringBuilder();
|
||||
|
||||
for (BasicBlock block : blocks) {
|
||||
buf.append("----- Block " + block.id + " -----" + new_line_separator);
|
||||
buf.append("----- Block ").append(block.id).append(" -----").append(new_line_separator);
|
||||
buf.append(block.toString());
|
||||
buf.append("----- Edges -----" + new_line_separator);
|
||||
buf.append("----- Edges -----").append(new_line_separator);
|
||||
|
||||
List<BasicBlock> suc = block.getSuccs();
|
||||
for (int j = 0; j < suc.size(); j++) {
|
||||
buf.append(">>>>>>>>(regular) Block " + suc.get(j).id + new_line_separator);
|
||||
buf.append(">>>>>>>>(regular) Block ").append(suc.get(j).id).append(new_line_separator);
|
||||
}
|
||||
suc = block.getSuccExceptions();
|
||||
for (int j = 0; j < suc.size(); j++) {
|
||||
@@ -102,21 +102,22 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
ExceptionRangeCFG range = getExceptionRange(handler, block);
|
||||
|
||||
if (range == null) {
|
||||
buf.append(">>>>>>>>(exception) Block " + handler.id + "\t" + "ERROR: range not found!" + new_line_separator);
|
||||
buf.append(">>>>>>>>(exception) Block ").append(handler.id).append("\t").append("ERROR: range not found!")
|
||||
.append(new_line_separator);
|
||||
}
|
||||
else {
|
||||
List<String> exceptionTypes = range.getExceptionTypes();
|
||||
if (exceptionTypes == null) {
|
||||
buf.append(">>>>>>>>(exception) Block " + handler.id + "\t" + "NULL" + new_line_separator);
|
||||
buf.append(">>>>>>>>(exception) Block ").append(handler.id).append("\t").append("NULL").append(new_line_separator);
|
||||
}
|
||||
else {
|
||||
for (String exceptionType : exceptionTypes) {
|
||||
buf.append(">>>>>>>>(exception) Block " + handler.id + "\t" + exceptionType + new_line_separator);
|
||||
buf.append(">>>>>>>>(exception) Block ").append(handler.id).append("\t").append(exceptionType).append(new_line_separator);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
buf.append("----- ----- -----" + new_line_separator);
|
||||
buf.append("----- ----- -----").append(new_line_separator);
|
||||
}
|
||||
|
||||
return buf.toString();
|
||||
@@ -223,7 +224,7 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
|
||||
short[] states = findStartInstructions(instrseq);
|
||||
|
||||
HashMap<Integer, BasicBlock> mapInstrBlocks = new HashMap<Integer, BasicBlock>();
|
||||
Map<Integer, BasicBlock> mapInstrBlocks = new HashMap<Integer, BasicBlock>();
|
||||
VBStyleCollection<BasicBlock, Integer> colBlocks = createBasicBlocks(states, instrseq, mapInstrBlocks);
|
||||
|
||||
blocks = colBlocks;
|
||||
@@ -237,12 +238,12 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
setFirstAndLastBlocks();
|
||||
}
|
||||
|
||||
private short[] findStartInstructions(InstructionSequence seq) {
|
||||
private static short[] findStartInstructions(InstructionSequence seq) {
|
||||
|
||||
int len = seq.length();
|
||||
short[] inststates = new short[len];
|
||||
|
||||
HashSet<Integer> excSet = new HashSet<Integer>();
|
||||
Set<Integer> excSet = new HashSet<Integer>();
|
||||
|
||||
for (ExceptionHandler handler : seq.getExceptionTable().getHandlers()) {
|
||||
excSet.add(handler.from_instr);
|
||||
@@ -287,8 +288,9 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
}
|
||||
|
||||
|
||||
private VBStyleCollection<BasicBlock, Integer> createBasicBlocks(short[] startblock, InstructionSequence instrseq,
|
||||
HashMap<Integer, BasicBlock> mapInstrBlocks) {
|
||||
private VBStyleCollection<BasicBlock, Integer> createBasicBlocks(short[] startblock,
|
||||
InstructionSequence instrseq,
|
||||
Map<Integer, BasicBlock> mapInstrBlocks) {
|
||||
|
||||
VBStyleCollection<BasicBlock, Integer> col = new VBStyleCollection<BasicBlock, Integer>();
|
||||
|
||||
@@ -329,7 +331,7 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
}
|
||||
|
||||
|
||||
private void connectBlocks(List<BasicBlock> lstbb, HashMap<Integer, BasicBlock> mapInstrBlocks) {
|
||||
private static void connectBlocks(List<BasicBlock> lstbb, Map<Integer, BasicBlock> mapInstrBlocks) {
|
||||
|
||||
for (int i = 0; i < lstbb.size(); i++) {
|
||||
|
||||
@@ -365,7 +367,7 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
}
|
||||
}
|
||||
|
||||
private void setExceptionEdges(InstructionSequence instrseq, HashMap<Integer, BasicBlock> instrBlocks) {
|
||||
private void setExceptionEdges(InstructionSequence instrseq, Map<Integer, BasicBlock> instrBlocks) {
|
||||
|
||||
exceptions = new ArrayList<ExceptionRangeCFG>();
|
||||
|
||||
@@ -404,7 +406,7 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
|
||||
private void setSubroutineEdges() {
|
||||
|
||||
final HashMap<BasicBlock, BasicBlock> subroutines = new HashMap<BasicBlock, BasicBlock>();
|
||||
final Map<BasicBlock, BasicBlock> subroutines = new HashMap<BasicBlock, BasicBlock>();
|
||||
|
||||
for (BasicBlock block : blocks) {
|
||||
|
||||
@@ -413,7 +415,7 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
LinkedList<BasicBlock> stack = new LinkedList<BasicBlock>();
|
||||
LinkedList<LinkedList<BasicBlock>> stackJsrStacks = new LinkedList<LinkedList<BasicBlock>>();
|
||||
|
||||
HashSet<BasicBlock> setVisited = new HashSet<BasicBlock>();
|
||||
Set<BasicBlock> setVisited = new HashSet<BasicBlock>();
|
||||
|
||||
stack.add(block);
|
||||
stackJsrStacks.add(new LinkedList<BasicBlock>());
|
||||
@@ -461,53 +463,64 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
}
|
||||
|
||||
private void processJsr() {
|
||||
while (true) {
|
||||
if (processJsrRanges() == 0) break;
|
||||
}
|
||||
}
|
||||
|
||||
while (processJsrRanges() != 0) ;
|
||||
private static class JsrRecord {
|
||||
private final BasicBlock jsr;
|
||||
private final Set<BasicBlock> range;
|
||||
private final BasicBlock ret;
|
||||
|
||||
private JsrRecord(BasicBlock jsr, Set<BasicBlock> range, BasicBlock ret) {
|
||||
this.jsr = jsr;
|
||||
this.range = range;
|
||||
this.ret = ret;
|
||||
}
|
||||
}
|
||||
|
||||
private int processJsrRanges() {
|
||||
|
||||
List<Object[]> lstJsrAll = new ArrayList<Object[]>();
|
||||
List<JsrRecord> lstJsrAll = new ArrayList<JsrRecord>();
|
||||
|
||||
// get all jsr ranges
|
||||
for (Entry<BasicBlock, BasicBlock> ent : subroutines.entrySet()) {
|
||||
BasicBlock jsr = ent.getKey();
|
||||
BasicBlock ret = ent.getValue();
|
||||
|
||||
lstJsrAll.add(new Object[]{jsr, getJsrRange(jsr, ret), ret});
|
||||
lstJsrAll.add(new JsrRecord(jsr, getJsrRange(jsr, ret), ret));
|
||||
}
|
||||
|
||||
// sort ranges
|
||||
// FIXME: better sort order
|
||||
List<Object[]> lstJsr = new ArrayList<Object[]>();
|
||||
for (Object[] arr : lstJsrAll) {
|
||||
List<JsrRecord> lstJsr = new ArrayList<JsrRecord>();
|
||||
for (JsrRecord arr : lstJsrAll) {
|
||||
int i = 0;
|
||||
for (; i < lstJsr.size(); i++) {
|
||||
Object[] arrJsr = lstJsr.get(i);
|
||||
|
||||
if (((HashSet<BasicBlock>)arrJsr[1]).contains(arr[0])) {
|
||||
JsrRecord arrJsr = lstJsr.get(i);
|
||||
if (arrJsr.range.contains(arr.jsr)) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
lstJsr.add(i, arr);
|
||||
}
|
||||
|
||||
// find the first intersection
|
||||
for (int i = 0; i < lstJsr.size(); i++) {
|
||||
Object[] arr = lstJsr.get(i);
|
||||
HashSet<BasicBlock> set = (HashSet<BasicBlock>)arr[1];
|
||||
JsrRecord arr = lstJsr.get(i);
|
||||
Set<BasicBlock> set = arr.range;
|
||||
|
||||
for (int j = i + 1; j < lstJsr.size(); j++) {
|
||||
Object[] arr1 = lstJsr.get(j);
|
||||
HashSet<BasicBlock> set1 = (HashSet<BasicBlock>)arr1[1];
|
||||
JsrRecord arr1 = lstJsr.get(j);
|
||||
Set<BasicBlock> set1 = arr1.range;
|
||||
|
||||
if (!set.contains(arr1[0]) && !set1.contains(arr[0])) { // rang 0 doesn't contain entry 1 and vice versa
|
||||
HashSet<BasicBlock> setc = new HashSet<BasicBlock>(set);
|
||||
if (!set.contains(arr1.jsr) && !set1.contains(arr.jsr)) { // rang 0 doesn't contain entry 1 and vice versa
|
||||
Set<BasicBlock> setc = new HashSet<BasicBlock>(set);
|
||||
setc.retainAll(set1);
|
||||
|
||||
if (!setc.isEmpty()) {
|
||||
splitJsrRange((BasicBlock)arr[0], (BasicBlock)arr[2], setc);
|
||||
splitJsrRange(arr.jsr, arr.ret, setc);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
@@ -517,11 +530,11 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
return 0;
|
||||
}
|
||||
|
||||
private HashSet<BasicBlock> getJsrRange(BasicBlock jsr, BasicBlock ret) {
|
||||
private Set<BasicBlock> getJsrRange(BasicBlock jsr, BasicBlock ret) {
|
||||
|
||||
HashSet<BasicBlock> blocks = new HashSet<BasicBlock>();
|
||||
Set<BasicBlock> blocks = new HashSet<BasicBlock>();
|
||||
|
||||
LinkedList<BasicBlock> lstNodes = new LinkedList<BasicBlock>();
|
||||
List<BasicBlock> lstNodes = new LinkedList<BasicBlock>();
|
||||
lstNodes.add(jsr);
|
||||
|
||||
BasicBlock dom = jsr.getSuccs().get(0);
|
||||
@@ -581,10 +594,10 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
return blocks;
|
||||
}
|
||||
|
||||
private void splitJsrRange(BasicBlock jsr, BasicBlock ret, HashSet<BasicBlock> common_blocks) {
|
||||
private void splitJsrRange(BasicBlock jsr, BasicBlock ret, Set<BasicBlock> common_blocks) {
|
||||
|
||||
LinkedList<BasicBlock> lstNodes = new LinkedList<BasicBlock>();
|
||||
HashMap<Integer, BasicBlock> mapNewNodes = new HashMap<Integer, BasicBlock>();
|
||||
List<BasicBlock> lstNodes = new LinkedList<BasicBlock>();
|
||||
Map<Integer, BasicBlock> mapNewNodes = new HashMap<Integer, BasicBlock>();
|
||||
|
||||
lstNodes.add(jsr);
|
||||
mapNewNodes.put(jsr.id, jsr);
|
||||
@@ -662,7 +675,7 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
splitJsrExceptionRanges(common_blocks, mapNewNodes);
|
||||
}
|
||||
|
||||
private void splitJsrExceptionRanges(HashSet<BasicBlock> common_blocks, HashMap<Integer, BasicBlock> mapNewNodes) {
|
||||
private void splitJsrExceptionRanges(Set<BasicBlock> common_blocks, Map<Integer, BasicBlock> mapNewNodes) {
|
||||
|
||||
for (int i = exceptions.size() - 1; i >= 0; i--) {
|
||||
|
||||
@@ -696,7 +709,7 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
removeJsrInstructions(mt.getClassStruct().getPool(), first, DataPoint.getInitialDataPoint(mt));
|
||||
}
|
||||
|
||||
private void removeJsrInstructions(ConstantPool pool, BasicBlock block, DataPoint data) {
|
||||
private static void removeJsrInstructions(ConstantPool pool, BasicBlock block, DataPoint data) {
|
||||
|
||||
ListStack<VarType> stack = data.getStack();
|
||||
|
||||
@@ -765,18 +778,18 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
|
||||
public List<BasicBlock> getReversePostOrder() {
|
||||
|
||||
LinkedList<BasicBlock> res = new LinkedList<BasicBlock>();
|
||||
List<BasicBlock> res = new LinkedList<BasicBlock>();
|
||||
addToReversePostOrderListIterative(first, res);
|
||||
|
||||
return res;
|
||||
}
|
||||
|
||||
private void addToReversePostOrderListIterative(BasicBlock root, List<BasicBlock> lst) {
|
||||
private static void addToReversePostOrderListIterative(BasicBlock root, List<BasicBlock> lst) {
|
||||
|
||||
LinkedList<BasicBlock> stackNode = new LinkedList<BasicBlock>();
|
||||
LinkedList<Integer> stackIndex = new LinkedList<Integer>();
|
||||
|
||||
HashSet<BasicBlock> setVisited = new HashSet<BasicBlock>();
|
||||
Set<BasicBlock> setVisited = new HashSet<BasicBlock>();
|
||||
|
||||
stackNode.add(root);
|
||||
stackIndex.add(0);
|
||||
@@ -845,32 +858,26 @@ public class ControlFlowGraph implements CodeConstants {
|
||||
this.exceptions = exceptions;
|
||||
}
|
||||
|
||||
|
||||
public BasicBlock getLast() {
|
||||
return last;
|
||||
}
|
||||
|
||||
|
||||
public void setLast(BasicBlock last) {
|
||||
this.last = last;
|
||||
}
|
||||
|
||||
|
||||
public HashMap<BasicBlock, BasicBlock> getSubroutines() {
|
||||
public Map<BasicBlock, BasicBlock> getSubroutines() {
|
||||
return subroutines;
|
||||
}
|
||||
|
||||
|
||||
public void setSubroutines(HashMap<BasicBlock, BasicBlock> subroutines) {
|
||||
public void setSubroutines(Map<BasicBlock, BasicBlock> subroutines) {
|
||||
this.subroutines = subroutines;
|
||||
}
|
||||
|
||||
|
||||
public HashSet<BasicBlock> getFinallyExits() {
|
||||
public Set<BasicBlock> getFinallyExits() {
|
||||
return finallyExits;
|
||||
}
|
||||
|
||||
|
||||
public void setFinallyExits(HashSet<BasicBlock> finallyExits) {
|
||||
this.finallyExits = finallyExits;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user