added a convenience method to create a summary query from a call site

Author: StevenArzt

soot-infoflow-summaries/src/soot/jimple/infoflow/methodSummary/data/sourceSink/AbstractFlowSinkSource.java

@@ -292,4 +292,5 @@ public Local getBaseLocal(Stmt stmt) {
 		}
 		return null;
 	}
+
 }

soot-infoflow-summaries/src/soot/jimple/infoflow/methodSummary/taintWrappers/resolvers/SummaryQuery.java

@@ -1,6 +1,12 @@
 package soot.jimple.infoflow.methodSummary.taintWrappers.resolvers;
 
+import soot.RefType;
 import soot.SootClass;
+import soot.SootMethod;
+import soot.Type;
+import soot.jimple.InstanceInvokeExpr;
+import soot.jimple.InvokeExpr;
+import soot.jimple.Stmt;
 
 /**
  * Query that retrieves summaries for a given class and method.
@@ -20,6 +26,30 @@ public SummaryQuery(SootClass calleeClass, SootClass declaredClass, String subsi
 		this.subsignature = subsignature;
 	}
 
+	/**
+	 * Creates a new summary query from a given call site. This method can be used
+	 * to query summaries for all potential callees of the given call site.
+	 * 
+	 * @param stmt The call site
+	 * @return The summary query that corresponds to the given call site
+	 */
+	public static SummaryQuery fromStmt(Stmt stmt) {
+		if (stmt != null && stmt.containsInvokeExpr()) {
+			InvokeExpr iexpr = stmt.getInvokeExpr();
+			SootClass declaredClass = null;
+			if (iexpr instanceof InstanceInvokeExpr) {
+				InstanceInvokeExpr iiexpr = (InstanceInvokeExpr) iexpr;
+				Type baseType = iiexpr.getBase().getType();
+				if (baseType instanceof RefType)
+					declaredClass = ((RefType) baseType).getSootClass();
+			}
+			SootMethod callee = iexpr.getMethod();
+			SootClass calleeClass = callee.getDeclaringClass();
+			return new SummaryQuery(calleeClass, declaredClass, callee.getSubSignature());
+		}
+		return null;
+	}
+
 	@Override
 	public String toString() {
 		StringBuilder sb = new StringBuilder();