for (SootMethod method : xmlReturners) {
Body body = method.retrieveActiveBody();
for (Unit unit : body.getUnits()) {
if (unit instanceof ReturnStmt) {
ReturnStmt stmt = (ReturnStmt) unit;
final ValueBox box = stmt.getOpBox();
returnLocations.put(stmt, new MethodStatementContainer(
method, stmt));
stmt2box.put(stmt, box);
}
}
}
// add the plug statements
for (Plugging plugging : pluggings) {
final AssignStmt stmt = plugging.getStmt();
final ValueBox box = stmt.getInvokeExprBox();
stmt2box.put(stmt, box);
}
return stmt2box;
}