diff options
Diffstat (limited to 'src/main/java/com/code_intelligence/jazzer/junit/AgentConfigurator.java')
-rw-r--r-- | src/main/java/com/code_intelligence/jazzer/junit/AgentConfigurator.java | 72 |
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.**")); + } +} |