/* Soot - a J*va Optimization Framework * Copyright (C) 1999 Patrick Lam * * This library 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 library 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. */ /* * Modified by the Sable Research Group and others 1997-1999. * See the 'credits' file distributed with Soot for the complete list of * contributors. (Soot is distributed at http://www.sable.mcgill.ca/soot) */ package org.kit.furia.fragment.soot.representation; import soot.options.*; import soot.*; import soot.jimple.*; import soot.jimple.internal.*; import soot.shimple.ShimpleBody; import soot.tagkit.CodeAttribute; import soot.tagkit.Tag; import java.util.*; /** Implementation of the Body class for the Grimp IR. */ public class FrimpBody extends StmtBody { /** * */ private static final long serialVersionUID = -559420704371577505L; /** Construct an empty GrimpBody **/ FrimpBody(SootMethod m) { super(m); } public Object clone() { Body b = Frimp.v().newBody(getMethod()); b.importBodyContentsFrom(this); return b; } /** Constructs a GrimpBody from the given Body. */ public FrimpBody(Body body) { super(body.getMethod()); if(Options.v().verbose()) G.v().out.println("[" + getMethod().getName() + "] Constructing FrimpBody..."); if(! (body instanceof ShimpleBody || body instanceof JimpleBody)) throw new RuntimeException("Can only construct FrimpBody's from ShimpleBody (for now)"); // NOPMD by amuller on 11/16/06 4:07 PM Body jBody = body; Iterator it = jBody.getLocals().iterator(); while (it.hasNext()) getLocals().add(((Local)(it.next()))); // getLocals().add(((Local)(it.next())).clone()); it = jBody.getUnits().iterator(); final HashMap oldToNew = new HashMap(getUnits().size() * 2 + 1, 0.7f); LinkedList updates = new LinkedList(); /* we should Frimpify (* x *) the Stmt's here... */ while (it.hasNext()) { Stmt oldStmt = (Stmt)(it.next()); final StmtBox newStmtBox = (StmtBox) Frimp.v().newStmtBox(null); final StmtBox updateStmtBox = (StmtBox) Frimp.v().newStmtBox(null); /* we can't have a general StmtSwapper on Grimp.v() */ /* because we need to collect a list of updates */ oldStmt.apply(new AbstractStmtSwitch() { public void caseAssignStmt(AssignStmt s) { newStmtBox.setUnit(Frimp.v().newAssignStmt(s)); } public void caseIdentityStmt(IdentityStmt s) { newStmtBox.setUnit(Frimp.v().newIdentityStmt(s)); } public void caseBreakpointStmt(BreakpointStmt s) { newStmtBox.setUnit(Frimp.v().newBreakpointStmt(s)); } public void caseInvokeStmt(InvokeStmt s) { newStmtBox.setUnit(Frimp.v().newInvokeStmt(s)); } public void caseEnterMonitorStmt(EnterMonitorStmt s) { newStmtBox.setUnit(Frimp.v().newEnterMonitorStmt(s)); } public void caseExitMonitorStmt(ExitMonitorStmt s) { newStmtBox.setUnit(Frimp.v().newExitMonitorStmt(s)); } public void caseGotoStmt(GotoStmt s) { newStmtBox.setUnit(Frimp.v().newGotoStmt(s)); updateStmtBox.setUnit(s); } public void caseIfStmt(IfStmt s) { newStmtBox.setUnit(Frimp.v().newIfStmt(s)); updateStmtBox.setUnit(s); } public void caseLookupSwitchStmt(LookupSwitchStmt s) { newStmtBox.setUnit(Frimp.v().newLookupSwitchStmt(s)); updateStmtBox.setUnit(s); } public void caseNopStmt(NopStmt s) { newStmtBox.setUnit(Frimp.v().newNopStmt(s)); } public void caseReturnStmt(ReturnStmt s) { newStmtBox.setUnit(Frimp.v().newReturnStmt(s)); } public void caseReturnVoidStmt(ReturnVoidStmt s) { newStmtBox.setUnit(Frimp.v().newReturnVoidStmt(s)); } public void caseTableSwitchStmt(TableSwitchStmt s) { newStmtBox.setUnit(Frimp.v().newTableSwitchStmt(s)); updateStmtBox.setUnit(s); } public void caseThrowStmt(ThrowStmt s) { newStmtBox.setUnit(Frimp.v().newThrowStmt(s)); } }); /* map old Expr's to new Expr's. */ Stmt newStmt = (Stmt)(newStmtBox.getUnit()); Iterator useBoxesIt; useBoxesIt = newStmt.getUseBoxes().iterator(); while(useBoxesIt.hasNext()) { ValueBox b = (ValueBox) (useBoxesIt.next()); b.setValue(Frimp.v().newExpr(b.getValue())); } useBoxesIt = newStmt.getDefBoxes().iterator(); while(useBoxesIt.hasNext()) { ValueBox b = (ValueBox) (useBoxesIt.next()); b.setValue(Frimp.v().newExpr(b.getValue())); } getUnits().add(newStmt); oldToNew.put(oldStmt, newStmt); if (updateStmtBox.getUnit() != null) updates.add(updateStmtBox.getUnit()); } /* fixup stmt's which have had moved targets */ it = updates.iterator(); while (it.hasNext()) { Stmt stmt = (Stmt)(it.next()); stmt.apply(new AbstractStmtSwitch() { public void caseGotoStmt(GotoStmt s) { GotoStmt newStmt = (GotoStmt)(oldToNew.get(s)); newStmt.setTarget((Stmt)oldToNew.get(newStmt.getTarget())); } public void caseIfStmt(IfStmt s) { IfStmt newStmt = (IfStmt)(oldToNew.get(s)); newStmt.setTarget((Stmt)oldToNew.get(newStmt.getTarget())); } public void caseLookupSwitchStmt(LookupSwitchStmt s) { LookupSwitchStmt newStmt = (LookupSwitchStmt)(oldToNew.get(s)); newStmt.setDefaultTarget ((Unit)(oldToNew.get(newStmt.getDefaultTarget()))); Unit[] newTargList = new Unit[newStmt.getTargetCount()]; for (int i = 0; i < newStmt.getTargetCount(); i++) newTargList[i] = (Unit)(oldToNew.get (newStmt.getTarget(i))); newStmt.setTargets(newTargList); } public void caseTableSwitchStmt(TableSwitchStmt s) { TableSwitchStmt newStmt = (TableSwitchStmt)(oldToNew.get(s)); newStmt.setDefaultTarget ((Unit)(oldToNew.get(newStmt.getDefaultTarget()))); int tc = newStmt.getHighIndex() - newStmt.getLowIndex()+1; LinkedList newTargList = new LinkedList(); for (int i = 0; i < tc; i++) newTargList.add(oldToNew.get (newStmt.getTarget(i))); newStmt.setTargets(newTargList); } }); } it = jBody.getTraps().iterator(); while (it.hasNext()) { Trap oldTrap = (Trap)(it.next()); getTraps().add(Frimp.v().newTrap (oldTrap.getException(), (Unit)(oldToNew.get(oldTrap.getBeginUnit())), (Unit)(oldToNew.get(oldTrap.getEndUnit())), (Unit)(oldToNew.get(oldTrap.getHandlerUnit())))); } } /** * Returns the result of iterating through all Units in this body * and querying them for their UnitBoxes. All UnitBoxes thus * found are returned. Branching Units and statements which use * PhiExpr will have UnitBoxes; a UnitBox contains a Unit that is * either a target of a branch or is being used as a pointer to * the end of a CFG block. * *

This method is typically used for pointer patching, eg when * the unit chain is cloned. * * @return A list of all the UnitBoxes held by this body's units. * @see UnitBox * @see #getUnitBoxes(boolean) * @see Unit#getUnitBoxes() * @see soot.shimple.PhiExpr#getUnitBoxes() **/ public List getAllUnitBoxes() { ArrayList unitBoxList = new ArrayList(); { Iterator it = unitChain.iterator(); while(it.hasNext()) { Unit item = (Unit) it.next(); assert item.getUnitBoxes() !=null: "This item returns null unitboxes: " + item + "\n class: " + item.getClass().getName(); unitBoxList.addAll(item.getUnitBoxes()); } } { Iterator it = trapChain.iterator(); while(it.hasNext()) { Trap item = (Trap) it.next(); unitBoxList.addAll(item.getUnitBoxes()); } } { Iterator it = getTags().iterator(); while(it.hasNext()) { Tag t = (Tag) it.next(); if( t instanceof CodeAttribute) unitBoxList.addAll(((CodeAttribute) t).getUnitBoxes()); } } return unitBoxList; } }