All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.eclipse.jetty.server.HttpConnection Maven / Gradle / Ivy

There is a newer version: 12.1.0.alpha0
Show newest version
//
// ========================================================================
// Copyright (c) 1995 Mort Bay Consulting Pty Ltd and others.
//
// This program and the accompanying materials are made available under the
// terms of the Eclipse Public License v. 2.0 which is available at
// https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
// which is available at https://www.apache.org/licenses/LICENSE-2.0.
//
// SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
// ========================================================================
//

package org.eclipse.jetty.server;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.WritePendingException;
import java.util.concurrent.RejectedExecutionException;
import java.util.concurrent.atomic.LongAdder;

import org.eclipse.jetty.http.BadMessageException;
import org.eclipse.jetty.http.HttpCompliance;
import org.eclipse.jetty.http.HttpField;
import org.eclipse.jetty.http.HttpGenerator;
import org.eclipse.jetty.http.HttpHeader;
import org.eclipse.jetty.http.HttpHeaderValue;
import org.eclipse.jetty.http.HttpMethod;
import org.eclipse.jetty.http.HttpParser;
import org.eclipse.jetty.http.MetaData;
import org.eclipse.jetty.http.PreEncodedHttpField;
import org.eclipse.jetty.io.AbstractConnection;
import org.eclipse.jetty.io.ByteBufferPool;
import org.eclipse.jetty.io.Connection;
import org.eclipse.jetty.io.EndPoint;
import org.eclipse.jetty.io.EofException;
import org.eclipse.jetty.io.RetainableByteBuffer;
import org.eclipse.jetty.io.RetainableByteBufferPool;
import org.eclipse.jetty.io.WriteFlusher;
import org.eclipse.jetty.util.BufferUtil;
import org.eclipse.jetty.util.Callback;
import org.eclipse.jetty.util.IteratingCallback;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import static org.eclipse.jetty.http.HttpStatus.INTERNAL_SERVER_ERROR_500;

/**
 * 

A {@link Connection} that handles the HTTP protocol.

*/ public class HttpConnection extends AbstractConnection implements Runnable, HttpTransport, WriteFlusher.Listener, Connection.UpgradeFrom, Connection.UpgradeTo { private static final Logger LOG = LoggerFactory.getLogger(HttpConnection.class); public static final HttpField CONNECTION_CLOSE = new PreEncodedHttpField(HttpHeader.CONNECTION, HttpHeaderValue.CLOSE.asString()); private static final ThreadLocal __currentConnection = new ThreadLocal<>(); private final HttpConfiguration _config; private final Connector _connector; private final ByteBufferPool _bufferPool; private final RetainableByteBufferPool _retainableByteBufferPool; private final HttpInput _input; private final HttpGenerator _generator; private final HttpChannelOverHttp _channel; private final HttpParser _parser; private volatile RetainableByteBuffer _retainableByteBuffer; private final AsyncReadCallback _asyncReadCallback = new AsyncReadCallback(); private final SendCallback _sendCallback = new SendCallback(); private final boolean _recordHttpComplianceViolations; private final LongAdder bytesIn = new LongAdder(); private final LongAdder bytesOut = new LongAdder(); private boolean _useInputDirectByteBuffers; private boolean _useOutputDirectByteBuffers; /** * Get the current connection that this thread is dispatched to. * Note that a thread may be processing a request asynchronously and * thus not be dispatched to the connection. * * @return the current HttpConnection or null * @see Request#getAttribute(String) for a more general way to access the HttpConnection */ public static HttpConnection getCurrentConnection() { return __currentConnection.get(); } protected static HttpConnection setCurrentConnection(HttpConnection connection) { HttpConnection last = __currentConnection.get(); __currentConnection.set(connection); return last; } public HttpConnection(HttpConfiguration config, Connector connector, EndPoint endPoint, boolean recordComplianceViolations) { super(endPoint, connector.getExecutor()); _config = config; _connector = connector; _bufferPool = _connector.getByteBufferPool(); _retainableByteBufferPool = _bufferPool.asRetainableByteBufferPool(); _generator = newHttpGenerator(); _channel = newHttpChannel(); _input = _channel.getRequest().getHttpInput(); _parser = newHttpParser(config.getHttpCompliance()); _recordHttpComplianceViolations = recordComplianceViolations; if (LOG.isDebugEnabled()) LOG.debug("New HTTP Connection {}", this); } public long getBeginNanoTime() { return _parser.getBeginNanoTime(); } public HttpConfiguration getHttpConfiguration() { return _config; } public boolean isRecordHttpComplianceViolations() { return _recordHttpComplianceViolations; } protected HttpGenerator newHttpGenerator() { return new HttpGenerator(_config.getSendServerVersion(), _config.getSendXPoweredBy()); } protected HttpChannelOverHttp newHttpChannel() { return new HttpChannelOverHttp(this, _connector, _config, getEndPoint(), this); } protected HttpParser newHttpParser(HttpCompliance compliance) { HttpParser parser = new HttpParser(newRequestHandler(), getHttpConfiguration().getRequestHeaderSize(), compliance); parser.setHeaderCacheSize(getHttpConfiguration().getHeaderCacheSize()); parser.setHeaderCacheCaseSensitive(getHttpConfiguration().isHeaderCacheCaseSensitive()); return parser; } protected HttpParser.RequestHandler newRequestHandler() { return _channel; } public Server getServer() { return _connector.getServer(); } public Connector getConnector() { return _connector; } public HttpChannel getHttpChannel() { return _channel; } public HttpParser getParser() { return _parser; } public HttpGenerator getGenerator() { return _generator; } @Override public long getMessagesIn() { return getHttpChannel().getRequests(); } @Override public long getMessagesOut() { return getHttpChannel().getRequests(); } public boolean isUseInputDirectByteBuffers() { return _useInputDirectByteBuffers; } public void setUseInputDirectByteBuffers(boolean useInputDirectByteBuffers) { _useInputDirectByteBuffers = useInputDirectByteBuffers; } public boolean isUseOutputDirectByteBuffers() { return _useOutputDirectByteBuffers; } public void setUseOutputDirectByteBuffers(boolean useOutputDirectByteBuffers) { _useOutputDirectByteBuffers = useOutputDirectByteBuffers; } @Override public ByteBuffer onUpgradeFrom() { if (!isRequestBufferEmpty()) { ByteBuffer unconsumed = ByteBuffer.allocateDirect(_retainableByteBuffer.remaining()); unconsumed.put(_retainableByteBuffer.getBuffer()); unconsumed.flip(); releaseRequestBuffer(); return unconsumed; } return null; } @Override public void onUpgradeTo(ByteBuffer buffer) { BufferUtil.append(getRequestBuffer(), buffer); } @Override public void onFlushed(long bytes) throws IOException { // Unfortunately cannot distinguish between header and content // bytes, and for content bytes whether they are chunked or not. _channel.getResponse().getHttpOutput().onFlushed(bytes); } void releaseRequestBuffer() { if (_retainableByteBuffer != null && !_retainableByteBuffer.hasRemaining()) { if (LOG.isDebugEnabled()) LOG.debug("releaseRequestBuffer {}", this); if (_retainableByteBuffer.release()) _retainableByteBuffer = null; else throw new IllegalStateException("unreleased buffer " + _retainableByteBuffer); } } private ByteBuffer getRequestBuffer() { if (_retainableByteBuffer == null) _retainableByteBuffer = _retainableByteBufferPool.acquire(getInputBufferSize(), isUseInputDirectByteBuffers()); return _retainableByteBuffer.getBuffer(); } public boolean isRequestBufferEmpty() { return _retainableByteBuffer == null || _retainableByteBuffer.isEmpty(); } @Override public void onFillable() { if (LOG.isDebugEnabled()) LOG.debug("{} onFillable enter {} {}", this, _channel.getState(), _retainableByteBuffer); HttpConnection last = setCurrentConnection(this); try { while (getEndPoint().isOpen()) { // Fill the request buffer (if needed). int filled = fillRequestBuffer(); if (filled < 0 && getEndPoint().isOutputShutdown()) close(); // Parse the request buffer. boolean handle = parseRequestBuffer(); // There could be a connection upgrade before handling // the HTTP/1.1 request, for example PRI * HTTP/2. // If there was a connection upgrade, the other // connection took over, nothing more to do here. if (getEndPoint().getConnection() != this) break; // Handle channel event if (handle) { boolean suspended = !_channel.handle(); // We should break iteration if we have suspended or upgraded the connection. if (suspended || getEndPoint().getConnection() != this) break; } else if (filled == 0) { fillInterested(); break; } else if (filled < 0) { if (_channel.getState().isIdle()) getEndPoint().shutdownOutput(); break; } } } catch (Throwable x) { try { if (LOG.isDebugEnabled()) LOG.debug("{} caught exception {}", this, _channel.getState(), x); if (_retainableByteBuffer != null) { _retainableByteBuffer.clear(); releaseRequestBuffer(); } } finally { getEndPoint().close(x); } } finally { setCurrentConnection(last); if (LOG.isDebugEnabled()) LOG.debug("{} onFillable exit {} {}", this, _channel.getState(), _retainableByteBuffer); } } /** * Parse and fill data, looking for content. * We do parse first, and only fill if we're out of bytes to avoid unnecessary system calls. */ void parseAndFillForContent() { // Defensive check to avoid an infinite select/wakeup/fillAndParseForContent/wait loop // in case the parser was mistakenly closed and the connection was not aborted. if (_parser.isTerminated()) throw new IllegalStateException("Parser is terminated: " + _parser); // When fillRequestBuffer() is called, it must always be followed by a parseRequestBuffer() call otherwise this method // doesn't trigger EOF/earlyEOF which breaks AsyncRequestReadTest.testPartialReadThenShutdown(). // This loop was designed by a committee and voted by a majority. while (_parser.inContentState()) { if (parseRequestBuffer()) break; // Re-check the parser state after parsing to avoid filling, // otherwise fillRequestBuffer() would acquire a ByteBuffer // that may be leaked. if (_parser.inContentState() && fillRequestBuffer() <= 0) break; } } private int fillRequestBuffer() { if (_retainableByteBuffer != null && _retainableByteBuffer.isRetained()) throw new IllegalStateException("fill with unconsumed content on " + this); if (isRequestBufferEmpty()) { // Get a buffer // We are not in a race here for the request buffer as we have not yet received a request, // so there are not an possible legal threads calling #parseContent or #completed. ByteBuffer requestBuffer = getRequestBuffer(); // fill try { int filled = getEndPoint().fill(requestBuffer); if (filled == 0) // Do a retry on fill 0 (optimization for SSL connections) filled = getEndPoint().fill(requestBuffer); if (filled > 0) { bytesIn.add(filled); } else { if (filled < 0) _parser.atEOF(); releaseRequestBuffer(); } if (LOG.isDebugEnabled()) LOG.debug("{} filled {} {}", this, filled, _retainableByteBuffer); return filled; } catch (Throwable x) { if (LOG.isDebugEnabled()) LOG.debug("Unable to fill from endpoint {}", getEndPoint(), x); _parser.atEOF(); if (_retainableByteBuffer != null) { _retainableByteBuffer.clear(); releaseRequestBuffer(); } return -1; } } return 0; } private boolean parseRequestBuffer() { if (LOG.isDebugEnabled()) LOG.debug("{} parse {}", this, _retainableByteBuffer); boolean handle = _parser.parseNext(_retainableByteBuffer == null ? BufferUtil.EMPTY_BUFFER : _retainableByteBuffer.getBuffer()); if (LOG.isDebugEnabled()) LOG.debug("{} parsed {} {}", this, handle, _parser); // recycle buffer ? if (_retainableByteBuffer != null && !_retainableByteBuffer.isRetained()) releaseRequestBuffer(); return handle; } private boolean upgrade() { Connection connection = (Connection)_channel.getRequest().getAttribute(UPGRADE_CONNECTION_ATTRIBUTE); if (connection == null) return false; if (LOG.isDebugEnabled()) LOG.debug("Upgrade from {} to {}", this, connection); _channel.getState().upgrade(); getEndPoint().upgrade(connection); _channel.recycle(); _parser.reset(); _generator.reset(); if (_retainableByteBuffer != null) { if (!_retainableByteBuffer.isRetained()) { releaseRequestBuffer(); } else { LOG.warn("{} lingering content references?!?!", this); _retainableByteBuffer = null; // Not returned to pool! } } return true; } @Override public void onCompleted() { // If we are fill interested, then a read is pending and we must abort if (isFillInterested()) { LOG.warn("Pending read in onCompleted {} {}", this, getEndPoint()); _channel.abort(new IOException("Pending read in onCompleted")); } else { // Handle connection upgrades. if (upgrade()) return; } // Drive to EOF, EarlyEOF or Error boolean complete = _input.consumeAll(); // Finish consuming the request // If we are still expecting if (_channel.isExpecting100Continue()) { // close to seek EOF _parser.close(); } // else abort if we can't consume all else if (_generator.isPersistent() && !complete) { if (LOG.isDebugEnabled()) LOG.debug("unconsumed input {} {}", this, _parser); _channel.abort(new IOException("unconsumed input")); } // Reset the channel, parsers and generator _channel.recycle(); if (!_parser.isClosed()) { if (_generator.isPersistent()) _parser.reset(); else _parser.close(); } _generator.reset(); // if we are not called from the onfillable thread, schedule completion if (getCurrentConnection() != this) { // If we are looking for the next request if (_parser.isStart()) { // if the buffer is empty if (isRequestBufferEmpty()) { // look for more data fillInterested(); } // else if we are still running else if (getConnector().isRunning()) { // Dispatched to handle a pipelined request try { getExecutor().execute(this); } catch (RejectedExecutionException e) { if (getConnector().isRunning()) LOG.warn("Failed dispatch of {}", this, e); else LOG.trace("IGNORED", e); getEndPoint().close(); } } else { getEndPoint().close(); } } // else the parser must be closed, so seek the EOF if we are still open else if (getEndPoint().isOpen()) fillInterested(); } } @Override protected boolean onReadTimeout(Throwable timeout) { return _channel.onIdleTimeout(timeout); } @Override protected void onFillInterestedFailed(Throwable cause) { _parser.close(); super.onFillInterestedFailed(cause); } @Override public void onOpen() { super.onOpen(); if (isRequestBufferEmpty()) fillInterested(); else getExecutor().execute(this); } @Override public void onClose(Throwable cause) { if (cause == null) _sendCallback.close(); else _sendCallback.failed(cause); super.onClose(cause); } @Override public void run() { onFillable(); } @Override public void send(MetaData.Request request, MetaData.Response response, ByteBuffer content, boolean lastContent, Callback callback) { if (response == null) { if (!lastContent && BufferUtil.isEmpty(content)) { callback.succeeded(); return; } } else { // If we are still expecting a 100 continues when we commit if (_channel.isExpecting100Continue()) // then we can't be persistent _generator.setPersistent(false); } if (_sendCallback.reset(request, response, content, lastContent, callback)) { _sendCallback.iterate(); } } HttpInput.Content newContent(ByteBuffer c) { return new Content(c); } @Override public void abort(Throwable failure) { if (LOG.isDebugEnabled()) LOG.debug("abort {} {}", this, failure); // Do a direct close of the output, as this may indicate to a client that the // response is bad either with RST or by abnormal completion of chunked response. getEndPoint().close(); } @Override public boolean isPushSupported() { return false; } @Override public void push(org.eclipse.jetty.http.MetaData.Request request) { LOG.debug("ignore push in {}", this); } public void asyncReadFillInterested() { getEndPoint().tryFillInterested(_asyncReadCallback); } @Override public long getBytesIn() { return bytesIn.longValue(); } @Override public long getBytesOut() { return bytesOut.longValue(); } @Override public String toConnectionString() { return String.format("%s@%x[p=%s,g=%s]=>%s", getClass().getSimpleName(), hashCode(), _parser, _generator, _channel); } private class Content extends HttpInput.Content { public Content(ByteBuffer content) { super(content); _retainableByteBuffer.retain(); } @Override public void succeeded() { _retainableByteBuffer.release(); } @Override public void failed(Throwable x) { succeeded(); } } private class AsyncReadCallback implements Callback { @Override public void succeeded() { if (_channel.getRequest().getHttpInput().onContentProducible()) _channel.handle(); } @Override public void failed(Throwable x) { if (_channel.failed(x)) _channel.handle(); } @Override public InvocationType getInvocationType() { // This callback does not block when the HttpInput is in blocking mode, // rather it wakes up the thread that is blocked waiting on the read; // but it can if it is in async mode, hence the varying InvocationType. return _channel.getRequest().getHttpInput().isAsync() ? InvocationType.BLOCKING : InvocationType.NON_BLOCKING; } } private class SendCallback extends IteratingCallback { private MetaData.Response _info; private boolean _head; private ByteBuffer _content; private boolean _lastContent; private Callback _callback; private ByteBuffer _header; private ByteBuffer _chunk; private boolean _shutdownOut; private SendCallback() { super(true); } @Override public InvocationType getInvocationType() { return _callback.getInvocationType(); } private boolean reset(MetaData.Request request, MetaData.Response info, ByteBuffer content, boolean last, Callback callback) { if (reset()) { _info = info; _head = request != null && HttpMethod.HEAD.is(request.getMethod()); _content = content; _lastContent = last; _callback = callback; _header = null; _shutdownOut = false; if (getConnector().isShutdown()) _generator.setPersistent(false); return true; } if (isClosed()) callback.failed(new EofException()); else callback.failed(new WritePendingException()); return false; } @Override public Action process() throws Exception { if (_callback == null) throw new IllegalStateException(); boolean useDirectByteBuffers = isUseOutputDirectByteBuffers(); while (true) { HttpGenerator.Result result = _generator.generateResponse(_info, _head, _header, _chunk, _content, _lastContent); if (LOG.isDebugEnabled()) LOG.debug("generate: {} for {} ({},{},{})@{}", result, this, BufferUtil.toSummaryString(_header), BufferUtil.toSummaryString(_content), _lastContent, _generator.getState()); switch (result) { case NEED_INFO: throw new EofException("request lifecycle violation"); case NEED_HEADER: { _header = _bufferPool.acquire(Math.min(_config.getResponseHeaderSize(), _config.getOutputBufferSize()), useDirectByteBuffers); continue; } case HEADER_OVERFLOW: { if (_header.capacity() >= _config.getResponseHeaderSize()) throw new BadMessageException(INTERNAL_SERVER_ERROR_500, "Response header too large"); releaseHeader(); _header = _bufferPool.acquire(_config.getResponseHeaderSize(), useDirectByteBuffers); continue; } case NEED_CHUNK: { _chunk = _bufferPool.acquire(HttpGenerator.CHUNK_SIZE, useDirectByteBuffers); continue; } case NEED_CHUNK_TRAILER: { releaseChunk(); _chunk = _bufferPool.acquire(_config.getResponseHeaderSize(), useDirectByteBuffers); continue; } case FLUSH: { // Don't write the chunk or the content if this is a HEAD response, or any other type of response that should have no content if (_head || _generator.isNoContent()) { BufferUtil.clear(_chunk); BufferUtil.clear(_content); } byte gatherWrite = 0; long bytes = 0; if (BufferUtil.hasContent(_header)) { gatherWrite += 4; bytes += _header.remaining(); } if (BufferUtil.hasContent(_chunk)) { gatherWrite += 2; bytes += _chunk.remaining(); } if (BufferUtil.hasContent(_content)) { gatherWrite += 1; bytes += _content.remaining(); } HttpConnection.this.bytesOut.add(bytes); switch (gatherWrite) { case 7: getEndPoint().write(this, _header, _chunk, _content); break; case 6: getEndPoint().write(this, _header, _chunk); break; case 5: getEndPoint().write(this, _header, _content); break; case 4: getEndPoint().write(this, _header); break; case 3: getEndPoint().write(this, _chunk, _content); break; case 2: getEndPoint().write(this, _chunk); break; case 1: getEndPoint().write(this, _content); break; default: succeeded(); } return Action.SCHEDULED; } case SHUTDOWN_OUT: { _shutdownOut = true; continue; } case DONE: { // If this is the end of the response and the connector was shutdown after response was committed, // we can't add the Connection:close header, but we are still allowed to close the connection // by shutting down the output. if (getConnector().isShutdown() && _generator.isEnd() && _generator.isPersistent()) _shutdownOut = true; return Action.SUCCEEDED; } case CONTINUE: { break; } default: { throw new IllegalStateException("generateResponse=" + result); } } } } private Callback release() { Callback complete = _callback; _callback = null; _info = null; _content = null; releaseHeader(); releaseChunk(); return complete; } private void releaseHeader() { if (_header != null) _bufferPool.release(_header); _header = null; } private void releaseChunk() { if (_chunk != null) _bufferPool.release(_chunk); _chunk = null; } @Override protected void onCompleteSuccess() { boolean upgrading = _channel.getRequest().getAttribute(UPGRADE_CONNECTION_ATTRIBUTE) != null; release().succeeded(); // If successfully upgraded it is responsibility of the next protocol to close the connection. if (_shutdownOut && !upgrading) getEndPoint().shutdownOutput(); } @Override public void onCompleteFailure(final Throwable x) { failedCallback(release(), x); if (_shutdownOut) getEndPoint().shutdownOutput(); } @Override public String toString() { return String.format("%s[i=%s,cb=%s]", super.toString(), _info, _callback); } } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy