ReportMissingSummaryWrapper: Add pretty printing and option to exclude application classes

Author: MarcMil

soot-infoflow-summaries/src/soot/jimple/infoflow/methodSummary/taintWrappers/ReportMissingSummaryWrapper.java

@@ -12,6 +12,7 @@
 import javax.xml.parsers.DocumentBuilder;
 import javax.xml.parsers.DocumentBuilderFactory;
 import javax.xml.parsers.ParserConfigurationException;
+import javax.xml.transform.OutputKeys;
 import javax.xml.transform.Transformer;
 import javax.xml.transform.TransformerException;
 import javax.xml.transform.TransformerFactory;
@@ -31,11 +32,16 @@ public ReportMissingSummaryWrapper(IMethodSummaryProvider flows) {
 		super(flows);
 	}
 
-	ConcurrentHashMap<SootClass, AtomicInteger> classSummariesMissing = new ConcurrentHashMap<>();
+	private ConcurrentHashMap<SootClass, AtomicInteger> classSummariesMissing = new ConcurrentHashMap<>();
+	private boolean prettyPrint = false;
+	private boolean showAppClasses = false;
 
 	@Override
 	protected void reportMissingMethod(SootMethod method) {
-		count(method.getDeclaringClass(), classSummariesMissing);
+		SootClass decl = method.getDeclaringClass();
+		if (!showAppClasses && decl.isApplicationClass())
+			return;
+		count(decl, classSummariesMissing);
 	}
 
 	private static <T> void count(T item, Map<T, AtomicInteger> map) {
@@ -49,6 +55,22 @@ private static <T> void count(T item, Map<T, AtomicInteger> map) {
 		ai.incrementAndGet();
 	}
 
+	public void setPrettyPrinting(boolean prettyPrint) {
+		this.prettyPrint = prettyPrint;
+	}
+
+	public boolean isPrettyPrinting() {
+		return prettyPrint;
+	}
+
+	public void setShowApplicationClasses(boolean showAppClasses) {
+		this.showAppClasses = showAppClasses;
+	}
+
+	public boolean isShowingApplicationClasses() {
+		return showAppClasses;
+	}
+
 	public void writeResults(File file) throws IOException, ParserConfigurationException, TransformerException {
 		Map<SootClass, Integer> sortedClassSummariesMissing = sortMap(classSummariesMissing);
 		DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance();
@@ -71,6 +93,11 @@ public void writeResults(File file) throws IOException, ParserConfigurationExcep
 
 		TransformerFactory transformerFactory = TransformerFactory.newInstance();
 		Transformer transformer = transformerFactory.newTransformer();
+
+		if (prettyPrint) {
+			transformer.setOutputProperty(OutputKeys.INDENT, "yes");
+			transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
+		}
 		DOMSource source = new DOMSource(doc);
 		StreamResult result = new StreamResult(file);