1 files changed, 72 insertions, 0 deletions
diff --git a/src/main/java/com/code_intelligence/jazzer/junit/AgentConfigurator.java b/src/main/java/com/code_intelligence/jazzer/junit/AgentConfigurator.java
new file mode 100644
index 00000000..1f286a31
--- /dev/null
+++ b/src/main/java/com/code_intelligence/jazzer/junit/AgentConfigurator.java
@@ -0,0 +1,72 @@
+// Copyright 2022 Code Intelligence GmbH
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.code_intelligence.jazzer.junit;
+
+import static com.code_intelligence.jazzer.junit.Utils.getClassPathBasedInstrumentationFilter;
+import static com.code_intelligence.jazzer.junit.Utils.getLegacyInstrumentationFilter;
+
+import java.io.File;
+import java.util.concurrent.atomic.AtomicBoolean;
+import org.junit.jupiter.api.extension.ExtensionContext;
+
+class AgentConfigurator {
+  private static final AtomicBoolean hasBeenConfigured = new AtomicBoolean();
+
+  static void forRegressionTest(ExtensionContext extensionContext) {
+    if (!hasBeenConfigured.compareAndSet(false, true)) {
+      return;
+    }
+
+    applyCommonConfiguration();
+
+    // Add logic to the hook instrumentation that allows us to enable and disable hooks at runtime.
+    System.setProperty("jazzer.internal.conditional_hooks", "true");
+    // Apply all hooks, but no coverage or compare instrumentation.
+    System.setProperty("jazzer.instrumentation_excludes", "**");
+    extensionContext.getConfigurationParameter("jazzer.instrument")
+        .ifPresent(s
+            -> System.setProperty(
+                "jazzer.custom_hook_includes", String.join(File.pathSeparator, s.split(","))));
+  }
+
+  static void forFuzzing(ExtensionContext executionRequest) {
+    if (!hasBeenConfigured.compareAndSet(false, true)) {
+      throw new IllegalStateException("Only a single fuzz test should be executed per fuzzing run");
+    }
+
+    applyCommonConfiguration();
+
+    String instrumentationFilter =
+        executionRequest.getConfigurationParameter("jazzer.instrument")
+            .orElseGet(
+                ()
+                    -> getClassPathBasedInstrumentationFilter(System.getProperty("java.class.path"))
+                           .orElseGet(()
+                                          -> getLegacyInstrumentationFilter(
+                                              executionRequest.getRequiredTestClass())));
+    String filter = String.join(File.pathSeparator, instrumentationFilter.split(","));
+    System.setProperty("jazzer.custom_hook_includes", filter);
+    System.setProperty("jazzer.instrumentation_includes", filter);
+  }
+
+  private static void applyCommonConfiguration() {
+    // Do not hook common IDE and JUnit classes and their dependencies.
+    System.setProperty("jazzer.custom_hook_excludes",
+        String.join(File.pathSeparator, "com.google.testing.junit.**", "com.intellij.**",
+            "org.jetbrains.**", "io.github.classgraph.**", "junit.framework.**", "net.bytebuddy.**",
+            "org.apiguardian.**", "org.assertj.core.**", "org.hamcrest.**", "org.junit.**",
+            "org.opentest4j.**", "org.mockito.**", "org.apache.maven.**", "org.gradle.**"));
+  }
+}