soot.jimple.infoflow.solver.gcSolver.AbstractReferenceProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of soot-infoflow Show documentation

Soot extending data flow tracking components for Java

There is a newer version: 2.9.0

package soot.jimple.infoflow.solver.gcSolver;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import soot.SootMethod;
import soot.jimple.toolkits.ide.icfg.BiDiInterproceduralCFG;

/**
 * Abstract base class for reference providers
 * 
 * @author Steven Arzt
 *
 */
public abstract class AbstractReferenceProvider implements IGCReferenceProvider {

	protected final BiDiInterproceduralCFG icfg;

	public AbstractReferenceProvider(BiDiInterproceduralCFG icfg) {
		this.icfg = icfg;
	}

	/**
	 * Computes the set of transitive callees of the given method
	 * 
	 * @param method The method for which to compute callees
	 * @return The set of transitive callees of the given method
	 */
	protected Set getTransitiveCallees(SootMethod method) {
		Set callees = new HashSet<>();
		List workList = new ArrayList<>();
		workList.add(method);

		while (!workList.isEmpty()) {
			SootMethod sm = workList.remove(0);
			if (sm.isConcrete()) {
				// We can only look for callees if we have a body
				if (!sm.hasActiveBody())
					sm.retrieveActiveBody();

				// Schedule the callees
				for (N callSite : icfg.getCallsFromWithin(sm)) {
					for (SootMethod callee : icfg.getCalleesOfCallAt(callSite)) {
						if (callees.add(callee))
							workList.add(callee);
					}
				}
			}
		}

		return callees;
	}

}