oss-fuzz/projects/log4j2/Log4jFuzzer.java

83 lines
3.4 KiB
Java

// Copyright 2021 Google LLC
//
// 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.
//
////////////////////////////////////////////////////////////////////////////////
import com.code_intelligence.jazzer.api.FuzzedDataProvider;
import org.apache.logging.log4j.Level;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.core.Appender;
import org.apache.logging.log4j.core.Core;
import org.apache.logging.log4j.core.LogEvent;
import org.apache.logging.log4j.core.appender.AbstractAppender;
import org.apache.logging.log4j.core.config.Configurator;
import org.apache.logging.log4j.core.config.builder.api.AppenderComponentBuilder;
import org.apache.logging.log4j.core.config.builder.api.RootLoggerComponentBuilder;
import org.apache.logging.log4j.core.config.builder.impl.DefaultConfigurationBuilder;
import org.apache.logging.log4j.core.config.plugins.Plugin;
import org.apache.logging.log4j.core.config.plugins.PluginAttribute;
import org.apache.logging.log4j.core.config.plugins.PluginFactory;
import org.apache.logging.log4j.core.layout.PatternLayout;
import org.apache.logging.log4j.status.StatusLogger;
// This fuzzer reproduces the log4j RCE vulnerability CVE-2021-44228.
public class Log4jFuzzer {
private final static Logger log = LogManager.getLogger(Log4jFuzzer.class.getName());
public static void fuzzerTestOneInput(FuzzedDataProvider data) {
log.error(data.consumeRemainingAsString());
}
public static void fuzzerInitialize() {
// Install a logger that constructs the log message, but never prints it.
// This noticeably increases the fuzzing performance
DefaultConfigurationBuilder configBuilder = new DefaultConfigurationBuilder();
configBuilder.setPackages(FuzzingAppender.class.getPackage().getName());
AppenderComponentBuilder fuzzingAppender =
configBuilder.newAppender("nullAppender", "FuzzingAppender");
configBuilder.add(fuzzingAppender);
RootLoggerComponentBuilder rootLogger = configBuilder.newRootLogger();
rootLogger.add(configBuilder.newAppenderRef("nullAppender"));
configBuilder.add(rootLogger);
Configurator.reconfigure(configBuilder.build());
// Disable logging of exceptions caught in log4j itself.
StatusLogger.getLogger().reset();
StatusLogger.getLogger().setLevel(Level.OFF);
}
@Plugin(
name = "FuzzingAppender", category = Core.CATEGORY_NAME, elementType = Appender.ELEMENT_TYPE)
public static class FuzzingAppender extends AbstractAppender {
protected FuzzingAppender(String name) {
super(name, null, PatternLayout.createDefaultLayout(), true);
}
@PluginFactory
public static FuzzingAppender createAppender(@PluginAttribute("name") String name) {
return new FuzzingAppender(name);
}
@Override
public void append(LogEvent event) {
try {
getLayout().toByteArray(event);
} catch (Exception ignored) {
// Prevent exceptions from being logged to stderr.
}
}
}
}