org.graylog2.inputs.transports.HttpTransport Maven / Gradle / Ivy
/**
* This file is part of Graylog.
*
* Graylog is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Graylog is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Graylog. If not, see > getBaseChannelHandlers(MessageInput input) {
final LinkedHashMap> baseChannelHandlers =
super.getBaseChannelHandlers(input);
baseChannelHandlers.put("decoder", new Callable() {
@Override
public ChannelHandler call() throws Exception {
return new HttpRequestDecoder(DEFAULT_MAX_INITIAL_LINE_LENGTH, DEFAULT_MAX_HEADER_SIZE, maxChunkSize);
}
});
baseChannelHandlers.put("encoder", new Callable() {
@Override
public ChannelHandler call() throws Exception {
return new HttpResponseEncoder();
}
});
baseChannelHandlers.put("decompressor", new Callable() {
@Override
public ChannelHandler call() throws Exception {
return new HttpContentDecompressor();
}
});
return baseChannelHandlers;
}
@Override
protected LinkedHashMap> getFinalChannelHandlers(MessageInput input) {
final LinkedHashMap> handlers = Maps.newLinkedHashMap();
handlers.put("http-handler", new Callable() {
@Override
public ChannelHandler call() throws Exception {
return new Handler(enableCors);
}
});
handlers.putAll(super.getFinalChannelHandlers(input));
return handlers;
}
@FactoryClass
public interface Factory extends Transport.Factory {
@Override
HttpTransport create(Configuration configuration);
@Override
Config getConfig();
}
@ConfigClass
public static class Config extends AbstractTcpTransport.Config {
@Override
public ConfigurationRequest getRequestedConfiguration() {
final ConfigurationRequest r = super.getRequestedConfiguration();
r.addField(new BooleanField(CK_ENABLE_CORS,
"Enable CORS",
true,
"Input sends CORS headers to satisfy browser security policies"));
r.addField(new NumberField(CK_MAX_CHUNK_SIZE,
"Max. HTTP chunk size",
DEFAULT_MAX_CHUNK_SIZE,
"The maximum HTTP chunk size in bytes (e. g. length of HTTP request body)",
ConfigurationField.Optional.OPTIONAL));
return r;
}
}
public static class Handler extends SimpleChannelHandler {
private final boolean enableCors;
public Handler(boolean enableCors) {
this.enableCors = enableCors;
}
@Override
public void messageReceived(final ChannelHandlerContext ctx, final MessageEvent e) throws Exception {
final Channel channel = e.getChannel();
final HttpRequest request = (HttpRequest) e.getMessage();
final boolean keepAlive = isKeepAlive(request);
final HttpVersion httpRequestVersion = request.getProtocolVersion();
final String origin = request.headers().get(Names.ORIGIN);
// to allow for future changes, let's be at least a little strict in what we accept here.
if (request.getMethod() != HttpMethod.POST) {
writeResponse(channel, keepAlive, httpRequestVersion, METHOD_NOT_ALLOWED, origin);
return;
}
final ChannelBuffer buffer = request.getContent();
if ("/gelf".equals(request.getUri())) {
// send on to raw message handler
writeResponse(channel, keepAlive, httpRequestVersion, ACCEPTED, origin);
fireMessageReceived(ctx, buffer);
} else {
writeResponse(channel, keepAlive, httpRequestVersion, NOT_FOUND, origin);
}
}
private void writeResponse(Channel channel,
boolean keepAlive,
HttpVersion httpRequestVersion,
HttpResponseStatus status,
String origin) {
final HttpResponse response =
new DefaultHttpResponse(httpRequestVersion, status);
response.headers().set(Names.CONTENT_LENGTH, 0);
response.headers().set(Names.CONNECTION,
keepAlive ? Values.KEEP_ALIVE : Values.CLOSE);
if (enableCors) {
if (origin != null && !origin.isEmpty()) {
response.headers().set(Names.ACCESS_CONTROL_ALLOW_ORIGIN, origin);
response.headers().set(Names.ACCESS_CONTROL_ALLOW_CREDENTIALS, true);
response.headers().set(Names.ACCESS_CONTROL_ALLOW_HEADERS, "Authorization");
}
}
final ChannelFuture channelFuture = channel.write(response);
if (!keepAlive) {
channelFuture.addListener(ChannelFutureListener.CLOSE);
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy