/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.synapse.transport.http.conn;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.nio.channels.ReadableByteChannel;
import java.nio.channels.SelectionKey;
import java.util.concurrent.atomic.AtomicLong;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpException;
import org.apache.http.HttpRequest;
import org.apache.http.HttpRequestFactory;
import org.apache.http.HttpResponse;
import org.apache.http.impl.nio.DefaultNHttpServerConnection;
import org.apache.http.nio.NHttpMessageParser;
import org.apache.http.nio.NHttpMessageWriter;
import org.apache.http.nio.NHttpServerEventHandler;
import org.apache.http.nio.reactor.EventMask;
import org.apache.http.nio.reactor.IOSession;
import org.apache.http.nio.reactor.SessionInputBuffer;
import org.apache.http.nio.reactor.SessionOutputBuffer;
import org.apache.http.nio.util.ByteBufferAllocator;
import org.apache.http.params.HttpParams;
import org.apache.synapse.transport.http.access.AccessHandler;
public class LoggingNHttpServerConnection extends DefaultNHttpServerConnection
implements UpgradableNHttpConnection {
private static final AtomicLong COUNT = new AtomicLong();
private final Log log;
private final Log iolog;
private final Log headerlog;
private final Log accesslog;
private final Log wirelog;
private final String id;
private IOSession original;
public LoggingNHttpServerConnection(
final IOSession session,
final HttpRequestFactory requestFactory,
final ByteBufferAllocator allocator,
final HttpParams params) {
super(session, requestFactory, allocator, params);
this.log = LogFactory.getLog(getClass());
this.iolog = LogFactory.getLog(session.getClass());
this.headerlog = LogFactory.getLog(LoggingUtils.HEADER_LOG_ID);
this.wirelog = LogFactory.getLog(LoggingUtils.WIRE_LOG_ID);
this.accesslog = LogFactory.getLog(LoggingUtils.ACCESS_LOG_ID);
this.id = "http-incoming-" + COUNT.incrementAndGet();
this.original = session;
if (this.iolog.isDebugEnabled() || this.wirelog.isDebugEnabled() || SynapseDebugInfoHolder.getInstance().isDebuggerEnabled()) {
super.bind(new LoggingIOSession(session, this.id, this.iolog, this.wirelog));
}
}
@Override
public void close() throws IOException {
if (this.log.isDebugEnabled()) {
this.log.debug(this.id + ": Close connection");
}
super.close();
}
@Override
public void shutdown() throws IOException {
if (this.log.isDebugEnabled()) {
this.log.debug(this.id + ": Shutdown connection");
}
super.shutdown();
}
@Override
public void submitResponse(final HttpResponse response) throws IOException, HttpException {
if (this.log.isDebugEnabled()) {
this.log.debug(this.id + ": " + response.getStatusLine().toString());
}
super.submitResponse(response);
}
@Override
public void consumeInput(final NHttpServerEventHandler handler) {
if (this.log.isDebugEnabled()) {
this.log.debug(this.id + ": Consume input");
}
if (!SynapseDebugInfoHolder.getInstance().isDebuggerEnabled()) {
//Debugger not enabled, hence going through normal flow
super.consumeInput(handler);
} else {
consumeInputWire(handler);
}
}
@Override
public void produceOutput(final NHttpServerEventHandler handler) {
if (this.log.isDebugEnabled()) {
this.log.debug(this.id + ": Produce output");
}
if (!SynapseDebugInfoHolder.getInstance().isDebuggerEnabled()) {
//Debugger not enabled, hence going through normal flow
super.produceOutput(handler);
} else {
produceOutputWire(handler);
}
}
/**
* Helper method for consume input when synapse debugger is enabled
* Note that this method need to be changed if we upgrade to new httpcore-nio version, this is override of it's
* consumeInput method
*
* @param handler
*/
private void consumeInputWire(final NHttpServerEventHandler handler) {
if (getContext() == null) {
return;
}
SynapseWireLogHolder logHolder = null;
if (getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY) != null) {
logHolder = (SynapseWireLogHolder) getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY);
} else {
logHolder = new SynapseWireLogHolder();
}
synchronized (logHolder) {
logHolder.setPhase(SynapseWireLogHolder.PHASE.SOURCE_REQUEST_READY);
getContext().setAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY, logHolder);
if (this.status != ACTIVE) {
this.session.clearEvent(EventMask.READ);
return;
}
try {
if (this.request == null) {
int bytesRead;
do {
bytesRead = this.requestParser.fillBuffer(this.session.channel());
if (bytesRead > 0) {
this.inTransportMetrics.incrementBytesTransferred(bytesRead);
}
this.request = this.requestParser.parse();
} while (bytesRead > 0 && this.request == null);
if (this.request != null) {
if (this.request instanceof HttpEntityEnclosingRequest) {
// Receive incoming entity
final HttpEntity entity = prepareDecoder(this.request);
((HttpEntityEnclosingRequest) this.request).setEntity(entity);
}
this.connMetrics.incrementRequestCount();
onRequestReceived(this.request);
handler.requestReceived(this);
if (this.contentDecoder == null) {
// No request entity is expected
// Ready to receive a new request
resetInput();
}
}
if (bytesRead == -1) {
handler.endOfInput(this);
}
}
if (this.contentDecoder != null && (this.session.getEventMask() & SelectionKey.OP_READ) > 0) {
handler.inputReady(this, this.contentDecoder);
if (this.contentDecoder.isCompleted()) {
//This is the place where input request read ends
if (getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY) != null) {
logHolder = (SynapseWireLogHolder) getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY);
logHolder.setPhase(SynapseWireLogHolder.PHASE.SOURCE_REQUEST_DONE);
getContext().setAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY, logHolder);
}
// Request entity received
// Ready to receive a new request
resetInput();
}
}
} catch (final HttpException ex) {
resetInput();
handler.exception(this, ex);
} catch (final Exception ex) {
handler.exception(this, ex);
} finally {
// Finally set buffered input flag
if (getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY) != null) {
logHolder = (SynapseWireLogHolder) getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY);
logHolder.setPhase(SynapseWireLogHolder.PHASE.SOURCE_REQUEST_DONE);
getContext().setAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY, logHolder);
}
this.hasBufferedInput = this.inbuf.hasData();
}
}
}
/**
* Helper method for produce output when synapse debugger is enabled
* Note that this method need to be changed if we upgrade to new httpcore-nio version, this is override of it's
* produceOutput method
*
* @param handler
*/
private void produceOutputWire(final NHttpServerEventHandler handler) {
if (getContext() == null) {
return;
}
SynapseWireLogHolder logHolder = null;
if (getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY) != null) {
logHolder = (SynapseWireLogHolder) getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY);
} else {
logHolder = new SynapseWireLogHolder();
}
synchronized (logHolder) {
logHolder.setPhase(SynapseWireLogHolder.PHASE.SOURCE_RESPONSE_READY);
getContext().setAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY, logHolder);
try {
if (this.status == ACTIVE) {
if (this.contentEncoder == null) {
handler.responseReady(this);
}
if (this.contentEncoder != null) {
handler.outputReady(this, this.contentEncoder);
if (this.contentEncoder.isCompleted()) {
resetOutput();
}
}
}
if (this.outbuf.hasData()) {
final int bytesWritten = this.outbuf.flush(this.session.channel());
if (bytesWritten > 0) {
this.outTransportMetrics.incrementBytesTransferred(bytesWritten);
}
}
if (!this.outbuf.hasData()) {
if (this.status == CLOSING) {
this.session.close();
this.status = CLOSED;
resetOutput();
}
if (this.contentEncoder == null && this.status != CLOSED) {
//This is the place where it finishes writing output to wire
if (getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY) != null) {
logHolder = (SynapseWireLogHolder) getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY);
logHolder.setPhase(SynapseWireLogHolder.PHASE.SOURCE_RESPONSE_DONE);
SynapseDebugInfoHolder.getInstance().setWireLogHolder(logHolder); //this is to send final wirlogs to dev studio side
getContext().removeAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY);
}
this.session.clearEvent(EventMask.WRITE);
}
}
} catch (final Exception ex) {
handler.exception(this, ex);
} finally {
if (getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY) != null) {
logHolder = (SynapseWireLogHolder) getContext().getAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY);
logHolder.setPhase(SynapseWireLogHolder.PHASE.SOURCE_RESPONSE_DONE);
getContext().setAttribute(SynapseDebugInfoHolder.SYNAPSE_WIRE_LOG_HOLDER_PROPERTY, logHolder);
}
// Finally set the buffered output flag
this.hasBufferedOutput = this.outbuf.hasData();
}
}
}
@Override
protected NHttpMessageWriter<HttpResponse> createResponseWriter(
final SessionOutputBuffer buffer,
final HttpParams params) {
return new LoggingNHttpMessageWriter(
super.createResponseWriter(buffer, params));
}
@Override
protected NHttpMessageParser<HttpRequest> createRequestParser(
final SessionInputBuffer buffer,
final HttpRequestFactory requestFactory,
final HttpParams params) {
return new LoggingNHttpMessageParser(
super.createRequestParser(buffer, requestFactory, params));
}
@Override
public String toString() {
return this.id;
}
public IOSession getIOSession() {
return this.original;
}
@Override
public void bind(final IOSession session) {
this.original = session;
if (this.iolog.isDebugEnabled() || this.wirelog.isDebugEnabled() || SynapseDebugInfoHolder.getInstance().isDebuggerEnabled()) {
super.bind(new LoggingIOSession(session, this.id, this.iolog, this.wirelog));
} else {
super.bind(session);
}
}
class LoggingNHttpMessageWriter implements NHttpMessageWriter<HttpResponse> {
private final NHttpMessageWriter<HttpResponse> writer;
public LoggingNHttpMessageWriter(final NHttpMessageWriter<HttpResponse> writer) {
super();
this.writer = writer;
}
public void reset() {
this.writer.reset();
}
public void write(final HttpResponse message) throws IOException, HttpException {
if (message != null && headerlog.isDebugEnabled()) {
headerlog.debug(id + " << " + message.getStatusLine().toString());
Header[] headers = message.getAllHeaders();
for (int i = 0; i < headers.length; i++) {
headerlog.debug(id + " << " + headers[i].toString());
}
}
if (message != null && accesslog.isInfoEnabled()) {
HttpResponse response = (HttpResponse) message;
HttpParams params = response.getParams();
final SocketAddress remoteAddress = session.getRemoteAddress();
if (remoteAddress instanceof InetSocketAddress) {
final InetSocketAddress remote = (( InetSocketAddress) remoteAddress);
params.setParameter("http.remote.addr",
remote.getAddress().getHostAddress());
}
AccessHandler.getAccess().addAccessToQueue(message);
}
this.writer.write(message);
}
}
class LoggingNHttpMessageParser implements NHttpMessageParser<HttpRequest> {
private final NHttpMessageParser<HttpRequest> parser;
public LoggingNHttpMessageParser(final NHttpMessageParser<HttpRequest> parser) {
super();
this.parser = parser;
}
public void reset() {
this.parser.reset();
}
public int fillBuffer(final ReadableByteChannel channel) throws IOException {
return this.parser.fillBuffer(channel);
}
public HttpRequest parse() throws IOException, HttpException {
HttpRequest message = this.parser.parse();
if (message != null && headerlog.isDebugEnabled()) {
headerlog.debug(id + " >> " + message.getRequestLine().toString());
Header[] headers = message.getAllHeaders();
for (int i = 0; i < headers.length; i++) {
headerlog.debug(id + " >> " + headers[i].toString());
}
}
if (message != null && accesslog.isInfoEnabled()) {
HttpRequest request = (HttpRequest) message;
HttpParams params = request.getParams();
final SocketAddress remoteAddress = session.getRemoteAddress();
if (remoteAddress instanceof InetSocketAddress) {
final InetSocketAddress remote = (( InetSocketAddress) remoteAddress);
params.setParameter("http.remote.addr",
remote.getAddress().getHostAddress());
}
AccessHandler.getAccess().addAccessToQueue(message);
}
return message;
}
}
}