package edu.umd.cs.findbugs;
import java.util.Iterator;
import org.apache.bcel.classfile.JavaClass;
import org.apache.bcel.classfile.Method;
import org.apache.bcel.generic.ConstantPoolGen;
import org.apache.bcel.generic.MethodGen;
import edu.umd.cs.findbugs.ba.AnalysisContext;
import edu.umd.cs.findbugs.ba.CFG;
import edu.umd.cs.findbugs.ba.CFGBuilderException;
import edu.umd.cs.findbugs.ba.ClassContext;
import edu.umd.cs.findbugs.ba.Dataflow;
import edu.umd.cs.findbugs.ba.DataflowAnalysisException;
import edu.umd.cs.findbugs.ba.DepthFirstSearch;
import edu.umd.cs.findbugs.ba.Location;
import edu.umd.cs.findbugs.ba.ResourceTracker;
import edu.umd.cs.findbugs.ba.ResourceValueAnalysis;
import edu.umd.cs.findbugs.ba.ResourceValueFrame;
import edu.umd.cs.findbugs.ba.SignatureConverter;
import edu.umd.cs.findbugs.classfile.Global;
import edu.umd.cs.findbugs.log.Profiler;
public abstract class ResourceTrackingDetector<Resource, ResourceTrackerType extends ResourceTracker<Resource>> implements
Detector {
private static final boolean DEBUG = SystemProperties.getBoolean("rtd.debug");
private static final String DEBUG_METHOD_NAME = SystemProperties.getProperty("rtd.method");
protected BugAccumulator bugAccumulator;
protected BugReporter bugReporter;
public ResourceTrackingDetector(BugReporter bugReporter) {
this.bugReporter = bugReporter;
this.bugAccumulator = new BugAccumulator(bugReporter);
}
public abstract boolean prescreen(ClassContext classContext, Method method, boolean mightClose);
public abstract ResourceTrackerType getResourceTracker(ClassContext classContext, Method method)
throws DataflowAnalysisException, CFGBuilderException;
public abstract void inspectResult(ClassContext classContext, MethodGen methodGen, CFG cfg,
Dataflow<ResourceValueFrame, ResourceValueAnalysis<Resource>> dataflow, Resource resource);
@Override
public void visitClassContext(ClassContext classContext) {
final JavaClass jclass = classContext.getJavaClass();
Method[] methodList = jclass.getMethods();
for (Method method : methodList) {
if (method.isAbstract() || method.isNative()) {
continue;
}
MethodGen methodGen = classContext.getMethodGen(method);
if (methodGen == null) {
continue;
}
if (DEBUG_METHOD_NAME != null && !DEBUG_METHOD_NAME.equals(method.getName())) {
continue;
}
if (DEBUG) {
System.out.println("----------------------------------------------------------------------");
System.out.println("Analyzing " + SignatureConverter.convertMethodSignature(methodGen));
System.out.println("----------------------------------------------------------------------");
}
try {
ResourceTrackerType resourceTracker = getResourceTracker(classContext, method);
boolean mightClose = mightCloseResource(classContext, method, resourceTracker);
if (!prescreen(classContext, method, mightClose)) {
continue;
}
ResourceCollection<Resource> resourceCollection = buildResourceCollection(classContext, method, resourceTracker);
if (resourceCollection.isEmpty()) {
continue;
}
analyzeMethod(classContext, method, resourceTracker, resourceCollection);
} catch (CFGBuilderException e) {
bugReporter.logError("Error analyzing method " + method.toString(), e);
} catch (DataflowAnalysisException e) {
bugReporter.logError("Error analyzing method " + method.toString(), e);
}
bugAccumulator.reportAccumulatedBugs();
}
}
private ResourceCollection<Resource> buildResourceCollection(ClassContext classContext, Method method,
ResourceTrackerType resourceTracker) throws CFGBuilderException, DataflowAnalysisException {
ResourceCollection<Resource> resourceCollection = new ResourceCollection<Resource>();
CFG cfg = classContext.getCFG(method);
ConstantPoolGen cpg = classContext.getConstantPoolGen();
for (Iterator<Location> i = cfg.locationIterator(); i.hasNext();) {
Location location = i.next();
Resource resource = resourceTracker.isResourceCreation(location.getBasicBlock(), location.getHandle(), cpg);
if (resource != null) {
resourceCollection.addCreatedResource(location, resource);
}
}
return resourceCollection;
}
private boolean mightCloseResource(ClassContext classContext, Method method, ResourceTrackerType resourceTracker)
throws CFGBuilderException, DataflowAnalysisException {
CFG cfg = classContext.getCFG(method);
ConstantPoolGen cpg = classContext.getConstantPoolGen();
for (Iterator<Location> i = cfg.locationIterator(); i.hasNext();) {
Location location = i.next();
if (resourceTracker.mightCloseResource(location.getBasicBlock(), location.getHandle(), cpg)) {
return true;
}
}
return false;
}
public void analyzeMethod(ClassContext classContext, Method method, ResourceTrackerType resourceTracker,
ResourceCollection<Resource> resourceCollection) throws CFGBuilderException, DataflowAnalysisException {
MethodGen methodGen = classContext.getMethodGen(method);
if (methodGen == null) {
return;
}
try {
CFG cfg = classContext.getCFG(method);
DepthFirstSearch dfs = classContext.getDepthFirstSearch(method);
if (DEBUG) {
System.out.println(SignatureConverter.convertMethodSignature(methodGen));
}
for (Iterator<Resource> i = resourceCollection.resourceIterator(); i.hasNext();) {
Resource resource = i.next();
ResourceValueAnalysis<Resource> analysis = new ResourceValueAnalysis<Resource>(methodGen, cfg, dfs,
resourceTracker, resource);
Dataflow<ResourceValueFrame, ResourceValueAnalysis<Resource>> dataflow = new Dataflow<ResourceValueFrame, ResourceValueAnalysis<Resource>>(
cfg, analysis);
Profiler profiler = Global.getAnalysisCache().getProfiler();
profiler.start(resourceTracker.getClass());
try {
dataflow.execute();
} finally {
profiler.end(resourceTracker.getClass());
}
inspectResult(classContext, methodGen, cfg, dataflow, resource);
}
} catch (RuntimeException e) {
AnalysisContext.logError("Exception while analyzing " + methodGen.getClassName() + "." + methodGen.getName() + ":"
+ methodGen.getSignature(), e);
}
}
@Override
public void report() {
}
}