boomerang.scene.AnalysisScope Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of boomerangPDS Show documentation
There is a newer version: 3.2.2
package boomerang.scene;

import boomerang.Query;
import boomerang.scene.CallGraph.Edge;
import com.google.common.base.Stopwatch;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import java.util.Collection;
import java.util.LinkedList;
import java.util.Set;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class AnalysisScope {

  private static final Logger LOGGER = LoggerFactory.getLogger(AnalysisScope.class);
  private CallGraph cg;
  private boolean scanLibraryClasses = false;

  public AnalysisScope(CallGraph cg) {
    this.cg = cg;
  }

  private final Set seeds = Sets.newHashSet();

  private Collection processed = Sets.newHashSet();
  private int statementCount;

  public void setScanLibraryClasses(boolean enabled) {
    scanLibraryClasses = enabled;
  }

  public Collection computeSeeds() {
    Collection entryPoints = cg.getEntryPoints();
    LOGGER.info("Computing seeds starting at {} entry method(s).", entryPoints.size());

    Stopwatch watch = Stopwatch.createStarted();
    LinkedList worklist = Lists.newLinkedList();
    worklist.addAll(entryPoints);
    while (!worklist.isEmpty()) {
      Method m = worklist.pop();
      if (!processed.add(m)) {
        continue;
      }
      LOGGER.trace("Processing {}", m);
      for (Statement u : m.getStatements()) {
        statementCount++;
        if (u.containsInvokeExpr()) {
          Collection edgesOutOf = cg.edgesOutOf(u);
          for (Edge e : edgesOutOf) {
            Method tgt = e.tgt();
            if (!scanLibraryClasses && !tgt.getDeclaringClass().isApplicationClass()) continue;

            if (!processed.contains(tgt)) {
              worklist.add(tgt);
            }
          }
        }
        for (Statement succ : u.getMethod().getControlFlowGraph().getSuccsOf(u)) {
          seeds.addAll(generate(new ControlFlowGraph.Edge(u, succ)));
        }
      }
    }
    LOGGER.info("Found {} seeds in {} in {} LOC .", seeds.size(), watch, statementCount);

    return seeds;
  }

  protected boolean analyseClassInitializers() {
    return false;
  }

  protected abstract Collection generate(ControlFlowGraph.Edge seed);
}