All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.mockserver.netty.MockServerHandler Maven / Gradle / Ivy
package org.mockserver.netty;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.base64.Base64;
import io.netty.util.AttributeKey;
import org.apache.commons.text.StringEscapeUtils;
import org.mockserver.configuration.ConfigurationProperties;
import org.mockserver.dashboard.DashboardHandler;
import org.mockserver.lifecycle.LifeCycle;
import org.mockserver.log.model.LogEntry;
import org.mockserver.logging.MockServerLogger;
import org.mockserver.mock.HttpStateHandler;
import org.mockserver.mock.action.ActionHandler;
import org.mockserver.model.HttpRequest;
import org.mockserver.model.MediaType;
import org.mockserver.model.PortBinding;
import org.mockserver.netty.proxy.connect.HttpConnectHandler;
import org.mockserver.netty.responsewriter.NettyResponseWriter;
import org.mockserver.responsewriter.ResponseWriter;
import org.mockserver.serialization.PortBindingSerializer;
import org.slf4j.event.Level;
import java.net.BindException;
import java.nio.charset.StandardCharsets;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import static io.netty.handler.codec.http.HttpHeaderNames.*;
import static io.netty.handler.codec.http.HttpResponseStatus.*;
import static org.apache.commons.lang3.StringUtils.isNotBlank;
import static org.mockserver.configuration.ConfigurationProperties.addSubjectAlternativeName;
import static org.mockserver.exception.ExceptionHandling.closeOnFlush;
import static org.mockserver.exception.ExceptionHandling.connectionClosedException;
import static org.mockserver.mock.HttpStateHandler.PATH_PREFIX;
import static org.mockserver.model.HttpResponse.response;
import static org.mockserver.model.PortBinding.portBinding;
import static org.mockserver.netty.unification.PortUnificationHandler.enableSslUpstreamAndDownstream;
/**
* @author jamesdbloom
*/
@ChannelHandler.Sharable
public class MockServerHandler extends SimpleChannelInboundHandler {
public static final AttributeKey PROXYING = AttributeKey.valueOf("PROXYING");
public static final AttributeKey> LOCAL_HOST_HEADERS = AttributeKey.valueOf("LOCAL_HOST_HEADERS");
private MockServerLogger mockServerLogger;
private HttpStateHandler httpStateHandler;
private PortBindingSerializer portBindingSerializer;
private LifeCycle server;
private ActionHandler actionHandler;
private DashboardHandler dashboardHandler = new DashboardHandler();
public MockServerHandler(LifeCycle server, HttpStateHandler httpStateHandler, ActionHandler actionHandler) {
super(false);
this.server = server;
this.httpStateHandler = httpStateHandler;
this.mockServerLogger = httpStateHandler.getMockServerLogger();
this.portBindingSerializer = new PortBindingSerializer(mockServerLogger);
this.actionHandler = actionHandler;
}
private static boolean isProxyingRequest(ChannelHandlerContext ctx) {
if (ctx != null && ctx.channel().attr(PROXYING).get() != null) {
return ctx.channel().attr(PROXYING).get();
}
return false;
}
private static Set getLocalAddresses(ChannelHandlerContext ctx) {
if (ctx != null &&
ctx.channel().attr(LOCAL_HOST_HEADERS) != null &&
ctx.channel().attr(LOCAL_HOST_HEADERS).get() != null) {
return ctx.channel().attr(LOCAL_HOST_HEADERS).get();
}
return new HashSet<>();
}
@Override
protected void channelRead0(final ChannelHandlerContext ctx, final HttpRequest request) {
ResponseWriter responseWriter = new NettyResponseWriter(mockServerLogger, ctx, httpStateHandler.getScheduler());
try {
addSubjectAlternativeName(request.getFirstHeader(HOST.toString()));
if (!httpStateHandler.handle(request, responseWriter, false)) {
if (request.matches("PUT", PATH_PREFIX + "/status", "/status") ||
isNotBlank(ConfigurationProperties.livenessHttpGetPath()) && request.matches("GET", ConfigurationProperties.livenessHttpGetPath())) {
responseWriter.writeResponse(request, OK, portBindingSerializer.serialize(portBinding(server.getLocalPorts())), "application/json");
} else if (request.matches("PUT", PATH_PREFIX + "/bind", "/bind")) {
PortBinding requestedPortBindings = portBindingSerializer.deserialize(request.getBodyAsString());
if (requestedPortBindings != null) {
try {
List actualPortBindings = server.bindServerPorts(requestedPortBindings.getPorts());
responseWriter.writeResponse(request, OK, portBindingSerializer.serialize(portBinding(actualPortBindings)), "application/json");
} catch (RuntimeException e) {
if (e.getCause() instanceof BindException) {
responseWriter.writeResponse(request, BAD_REQUEST, e.getMessage() + " port already in use", MediaType.create("text", "plain").toString());
} else {
throw e;
}
}
}
} else if (request.matches("PUT", PATH_PREFIX + "/stop", "/stop")) {
ctx.writeAndFlush(response().withStatusCode(OK.code()));
new Thread(() -> server.stop()).start();
} else if (request.getMethod().getValue().equals("GET") && request.getPath().getValue().startsWith(PATH_PREFIX + "/dashboard")) {
dashboardHandler.renderDashboard(ctx, request);
} else if (request.getMethod().getValue().equals("CONNECT")) {
String username = ConfigurationProperties.proxyAuthenticationUsername();
String password = ConfigurationProperties.proxyAuthenticationPassword();
if (isNotBlank(username) && isNotBlank(password) &&
!request.containsHeader(PROXY_AUTHORIZATION.toString(), "Basic " + Base64.encode(Unpooled.copiedBuffer(username + ':' + password, StandardCharsets.UTF_8), false).toString(StandardCharsets.US_ASCII))) {
ctx.writeAndFlush(response()
.withStatusCode(PROXY_AUTHENTICATION_REQUIRED.code())
.withHeader(PROXY_AUTHENTICATE.toString(), "Basic realm=\"" + StringEscapeUtils.escapeJava(ConfigurationProperties.proxyAuthenticationRealm()) + "\", charset=\"UTF-8\""));
} else {
ctx.channel().attr(PROXYING).set(Boolean.TRUE);
// assume SSL for CONNECT request
enableSslUpstreamAndDownstream(ctx.channel());
// add Subject Alternative Name for SSL certificate
if (isNotBlank(request.getPath().getValue())) {
server.getScheduler().submit(() -> addSubjectAlternativeName(request.getPath().getValue()));
}
ctx.pipeline().addLast(new HttpConnectHandler(server, mockServerLogger, request.getPath().getValue(), -1));
ctx.pipeline().remove(this);
ctx.fireChannelRead(request);
}
} else {
try {
actionHandler.processAction(request, responseWriter, ctx, getLocalAddresses(ctx), isProxyingRequest(ctx), false);
} catch (Throwable throwable) {
mockServerLogger.logEvent(
new LogEntry()
.setType(LogEntry.LogMessageType.EXCEPTION)
.setLogLevel(Level.ERROR)
.setHttpRequest(request)
.setMessageFormat("exception processing:{}")
.setArguments(request)
.setThrowable(throwable)
);
}
}
}
} catch (IllegalArgumentException iae) {
mockServerLogger.logEvent(
new LogEntry()
.setType(LogEntry.LogMessageType.EXCEPTION)
.setLogLevel(Level.ERROR)
.setHttpRequest(request)
.setMessageFormat("exception processing:{}error:{}")
.setArguments(request, iae.getMessage())
);
// send request without API CORS headers
responseWriter.writeResponse(request, BAD_REQUEST, iae.getMessage(), MediaType.create("text", "plain").toString());
} catch (Exception ex) {
mockServerLogger.logEvent(
new LogEntry()
.setLogLevel(Level.ERROR)
.setType(LogEntry.LogMessageType.EXCEPTION)
.setHttpRequest(request)
.setMessageFormat("exception processing " + request)
.setThrowable(ex)
);
responseWriter.writeResponse(request, response().withStatusCode(BAD_REQUEST.code()).withBody(ex.getMessage()), true);
}
}
@Override
public void channelReadComplete(ChannelHandlerContext ctx) {
ctx.flush();
}
@Override
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
if (connectionClosedException(cause)) {
mockServerLogger.logEvent(
new LogEntry()
.setType(LogEntry.LogMessageType.EXCEPTION)
.setLogLevel(Level.ERROR)
.setMessageFormat("exception caught by " + server.getClass() + " handler -> closing pipeline " + ctx.channel())
.setThrowable(cause)
);
}
closeOnFlush(ctx.channel());
}
}
