Clean the stack traces of findings

1 files changed, 21 insertions, 0 deletions

diff --git a/agent/src/main/java/com/code_intelligence/jazzer/autofuzz/FuzzTarget.java b/agent/src/main/java/com/code_intelligence/jazzer/autofuzz/FuzzTarget.java
index 9806bb3c..677d107e 100644
--- a/agent/src/main/java/com/code_intelligence/jazzer/autofuzz/FuzzTarget.java
+++ b/agent/src/main/java/com/code_intelligence/jazzer/autofuzz/FuzzTarget.java
@@ -145,6 +145,7 @@ public class FuzzTarget {
           return;
         }
       }
+      cleanStackTraces(cause);
       throw cause;
     } catch (Throwable t) {
       System.err.println("Unexpected exception encountered during autofuzz");
@@ -152,4 +153,24 @@ public class FuzzTarget {
       System.exit(1);
     }
   }
+
+  // Removes all stack trace elements that live in the Java standard library, internal JDK classes
+  // or the autofuzz package from the bottom of all stack frames.
+  private static void cleanStackTraces(Throwable t) {
+    Throwable cause = t;
+    while (cause != null) {
+      StackTraceElement[] elements = cause.getStackTrace();
+      int firstInterestingPos;
+      for (firstInterestingPos = elements.length - 1; firstInterestingPos > 0;
+           firstInterestingPos--) {
+        String className = elements[firstInterestingPos].getClassName();
+        if (!className.startsWith("com.code_intelligence.jazzer.autofuzz")
+            && !className.startsWith("java.") && !className.startsWith("jdk.")) {
+          break;
+        }
+      }
+      cause.setStackTrace(Arrays.copyOfRange(elements, 0, firstInterestingPos + 1));
+      cause = cause.getCause();
+    }
+  }
 }