Examples with HttpFields org.eclipse.jetty.http.HttpFields used on opensource projects

Example 16 with HttpFields

use of org.eclipse.jetty.http.HttpFields in project jetty.project by eclipse.

the class BufferingResponseListener method onHeaders.

@Override
public void onHeaders(Response response) {
    super.onHeaders(response);
    HttpFields headers = response.getHeaders();
    long length = headers.getLongField(HttpHeader.CONTENT_LENGTH.asString());
    if (length > maxLength) {
        response.abort(new IllegalArgumentException("Buffering capacity exceeded"));
        return;
    }
    buffer = BufferUtil.allocate(length > 0 ? (int) length : 1024);
    String contentType = headers.get(HttpHeader.CONTENT_TYPE);
    if (contentType != null) {
        String media = contentType;
        String charset = "charset=";
        int index = contentType.toLowerCase(Locale.ENGLISH).indexOf(charset);
        if (index > 0) {
            media = contentType.substring(0, index);
            String encoding = contentType.substring(index + charset.length());
            // Sometimes charsets arrive with an ending semicolon
            int semicolon = encoding.indexOf(';');
            if (semicolon > 0)
                encoding = encoding.substring(0, semicolon).trim();
            this.encoding = encoding;
        }
        int semicolon = media.indexOf(';');
        if (semicolon > 0)
            media = media.substring(0, semicolon).trim();
        this.mediaType = media;
    }
}

Example 17 with HttpFields

use of org.eclipse.jetty.http.HttpFields in project jetty.project by eclipse.

the class HttpChannelOverHTTP method exchangeTerminated.

@Override
public void exchangeTerminated(HttpExchange exchange, Result result) {
    super.exchangeTerminated(exchange, result);
    Response response = result.getResponse();
    HttpFields responseHeaders = response.getHeaders();
    String closeReason = null;
    if (result.isFailed())
        closeReason = "failure";
    else if (receiver.isShutdown())
        closeReason = "server close";
    else if (sender.isShutdown())
        closeReason = "client close";
    if (closeReason == null) {
        if (response.getVersion().compareTo(HttpVersion.HTTP_1_1) < 0) {
            // HTTP 1.0 must close the connection unless it has
            // an explicit keep alive or it's a CONNECT method.
            boolean keepAlive = responseHeaders.contains(HttpHeader.CONNECTION, HttpHeaderValue.KEEP_ALIVE.asString());
            boolean connect = HttpMethod.CONNECT.is(exchange.getRequest().getMethod());
            if (!keepAlive && !connect)
                closeReason = "http/1.0";
        } else {
            // HTTP 1.1 closes only if it has an explicit close.
            if (responseHeaders.contains(HttpHeader.CONNECTION, HttpHeaderValue.CLOSE.asString()))
                closeReason = "http/1.1";
        }
    }
    if (closeReason != null) {
        if (LOG.isDebugEnabled())
            LOG.debug("Closing, reason: {} - {}", closeReason, connection);
        connection.close();
    } else {
        if (response.getStatus() == HttpStatus.SWITCHING_PROTOCOLS_101)
            connection.remove();
        else
            release();
    }
}

Example 18 with HttpFields

use of org.eclipse.jetty.http.HttpFields in project jetty.project by eclipse.

the class FlowControlStrategyTest method testFlowControlWhenServerResetsStream.

@Test
public void testFlowControlWhenServerResetsStream() throws Exception {
    // On server, don't consume the data and immediately reset.
    start(new ServerSessionListener.Adapter() {

        @Override
        public Stream.Listener onNewStream(Stream stream, HeadersFrame frame) {
            MetaData.Request request = (MetaData.Request) frame.getMetaData();
            if (HttpMethod.GET.is(request.getMethod()))
                return new Stream.Listener.Adapter();
            return new Stream.Listener.Adapter() {

                @Override
                public void onData(Stream stream, DataFrame frame, Callback callback) {
                    // Fail the callback to enlarge the session window.
                    // More data frames will be discarded because the
                    // stream is reset, and automatically consumed to
                    // keep the session window large for other streams.
                    callback.failed(new Throwable());
                    stream.reset(new ResetFrame(stream.getId(), ErrorCode.CANCEL_STREAM_ERROR.code), Callback.NOOP);
                }
            };
        }
    });
    Session session = newClient(new Session.Listener.Adapter());
    MetaData.Request metaData = newRequest("POST", new HttpFields());
    HeadersFrame frame = new HeadersFrame(metaData, null, false);
    FuturePromise<Stream> streamPromise = new FuturePromise<>();
    final CountDownLatch resetLatch = new CountDownLatch(1);
    session.newStream(frame, streamPromise, new Stream.Listener.Adapter() {

        @Override
        public void onReset(Stream stream, ResetFrame frame) {
            resetLatch.countDown();
        }
    });
    Stream stream = streamPromise.get(5, TimeUnit.SECONDS);
    // Perform a big upload that will stall the flow control windows.
    ByteBuffer data = ByteBuffer.allocate(5 * FlowControlStrategy.DEFAULT_WINDOW_SIZE);
    final CountDownLatch dataLatch = new CountDownLatch(1);
    stream.data(new DataFrame(stream.getId(), data, true), new Callback() {

        @Override
        public InvocationType getInvocationType() {
            return InvocationType.NON_BLOCKING;
        }

        @Override
        public void failed(Throwable x) {
            dataLatch.countDown();
        }
    });
    Assert.assertTrue(resetLatch.await(5, TimeUnit.SECONDS));
    Assert.assertTrue(dataLatch.await(5, TimeUnit.SECONDS));
}

Example 19 with HttpFields

use of org.eclipse.jetty.http.HttpFields in project jetty.project by eclipse.

the class FlowControlStrategyTest method testWindowSizeUpdates.

@Test
public void testWindowSizeUpdates() throws Exception {
    final CountDownLatch prefaceLatch = new CountDownLatch(1);
    final CountDownLatch stream1Latch = new CountDownLatch(1);
    final CountDownLatch stream2Latch = new CountDownLatch(1);
    final CountDownLatch settingsLatch = new CountDownLatch(1);
    start(new ServerSessionListener.Adapter() {

        @Override
        public Map<Integer, Integer> onPreface(Session session) {
            HTTP2Session serverSession = (HTTP2Session) session;
            Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, serverSession.getSendWindow());
            Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, serverSession.getRecvWindow());
            prefaceLatch.countDown();
            return null;
        }

        @Override
        public void onSettings(Session session, SettingsFrame frame) {
            for (Stream stream : session.getStreams()) {
                HTTP2Stream serverStream = (HTTP2Stream) stream;
                Assert.assertEquals(0, serverStream.getSendWindow());
                Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, serverStream.getRecvWindow());
            }
            settingsLatch.countDown();
        }

        @Override
        public Stream.Listener onNewStream(Stream stream, HeadersFrame frame) {
            HTTP2Stream serverStream = (HTTP2Stream) stream;
            MetaData.Request request = (MetaData.Request) frame.getMetaData();
            if ("GET".equalsIgnoreCase(request.getMethod())) {
                Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, serverStream.getSendWindow());
                Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, serverStream.getRecvWindow());
                stream1Latch.countDown();
            } else {
                Assert.assertEquals(0, serverStream.getSendWindow());
                Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, serverStream.getRecvWindow());
                stream2Latch.countDown();
            }
            return null;
        }
    });
    HTTP2Session clientSession = (HTTP2Session) newClient(new Session.Listener.Adapter());
    Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, clientSession.getSendWindow());
    Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, clientSession.getRecvWindow());
    Assert.assertTrue(prefaceLatch.await(5, TimeUnit.SECONDS));
    MetaData.Request request1 = newRequest("GET", new HttpFields());
    FuturePromise<Stream> promise1 = new FuturePromise<>();
    clientSession.newStream(new HeadersFrame(request1, null, true), promise1, new Stream.Listener.Adapter());
    HTTP2Stream clientStream1 = (HTTP2Stream) promise1.get(5, TimeUnit.SECONDS);
    Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, clientStream1.getSendWindow());
    Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, clientStream1.getRecvWindow());
    Assert.assertTrue(stream1Latch.await(5, TimeUnit.SECONDS));
    // Send a SETTINGS frame that changes the window size.
    // This tells the server that its stream send window must be updated,
    // so on the client it's the receive window that must be updated.
    Map<Integer, Integer> settings = new HashMap<>();
    settings.put(SettingsFrame.INITIAL_WINDOW_SIZE, 0);
    SettingsFrame frame = new SettingsFrame(settings, false);
    FutureCallback callback = new FutureCallback();
    clientSession.settings(frame, callback);
    callback.get(5, TimeUnit.SECONDS);
    Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, clientStream1.getSendWindow());
    Assert.assertEquals(0, clientStream1.getRecvWindow());
    settingsLatch.await(5, TimeUnit.SECONDS);
    // Now create a new stream, it must pick up the new value.
    MetaData.Request request2 = newRequest("POST", new HttpFields());
    FuturePromise<Stream> promise2 = new FuturePromise<>();
    clientSession.newStream(new HeadersFrame(request2, null, true), promise2, new Stream.Listener.Adapter());
    HTTP2Stream clientStream2 = (HTTP2Stream) promise2.get(5, TimeUnit.SECONDS);
    Assert.assertEquals(FlowControlStrategy.DEFAULT_WINDOW_SIZE, clientStream2.getSendWindow());
    Assert.assertEquals(0, clientStream2.getRecvWindow());
    Assert.assertTrue(stream2Latch.await(5, TimeUnit.SECONDS));
}

Example 20 with HttpFields

use of org.eclipse.jetty.http.HttpFields in project jetty.project by eclipse.

the class FlowControlStrategyTest method testServerTwoDataFramesWithStalledStream.

// TODO
// Since we changed the API to disallow consecutive data() calls without waiting
// for the callback, it is now not possible to have DATA1, DATA2 in the queue for
// the same stream. Perhaps this test should just be deleted.
@Ignore
@Test
public void testServerTwoDataFramesWithStalledStream() throws Exception {
    // Frames in queue = DATA1, DATA2.
    // Server writes part of DATA1, then stalls.
    // A window update unstalls the session, verify that the data is correctly sent.
    Random random = new Random();
    final byte[] chunk1 = new byte[1024];
    random.nextBytes(chunk1);
    final byte[] chunk2 = new byte[2048];
    random.nextBytes(chunk2);
    // Two SETTINGS frames: the initial after the preface,
    // and the explicit where we set the stream window size to zero.
    final AtomicReference<CountDownLatch> settingsLatch = new AtomicReference<>(new CountDownLatch(2));
    final CountDownLatch dataLatch = new CountDownLatch(1);
    start(new ServerSessionListener.Adapter() {

        @Override
        public void onSettings(Session session, SettingsFrame frame) {
            settingsLatch.get().countDown();
        }

        @Override
        public Stream.Listener onNewStream(Stream stream, HeadersFrame frame) {
            stream.data(new DataFrame(stream.getId(), ByteBuffer.wrap(chunk1), false), Callback.NOOP);
            stream.data(new DataFrame(stream.getId(), ByteBuffer.wrap(chunk2), true), Callback.NOOP);
            dataLatch.countDown();
            return null;
        }
    });
    Session session = newClient(new Session.Listener.Adapter());
    Map<Integer, Integer> settings = new HashMap<>();
    settings.put(SettingsFrame.INITIAL_WINDOW_SIZE, 0);
    session.settings(new SettingsFrame(settings, false), Callback.NOOP);
    Assert.assertTrue(settingsLatch.get().await(5, TimeUnit.SECONDS));
    byte[] content = new byte[chunk1.length + chunk2.length];
    final ByteBuffer buffer = ByteBuffer.wrap(content);
    MetaData.Request metaData = newRequest("GET", new HttpFields());
    HeadersFrame requestFrame = new HeadersFrame(metaData, null, true);
    final CountDownLatch responseLatch = new CountDownLatch(1);
    session.newStream(requestFrame, new Promise.Adapter<>(), new Stream.Listener.Adapter() {

        @Override
        public void onData(Stream stream, DataFrame frame, Callback callback) {
            buffer.put(frame.getData());
            callback.succeeded();
            if (frame.isEndStream())
                responseLatch.countDown();
        }
    });
    Assert.assertTrue(dataLatch.await(5, TimeUnit.SECONDS));
    // Now we have the 2 DATA frames queued in the server.
    // Unstall the stream window.
    settingsLatch.set(new CountDownLatch(1));
    settings.clear();
    settings.put(SettingsFrame.INITIAL_WINDOW_SIZE, chunk1.length / 2);
    session.settings(new SettingsFrame(settings, false), Callback.NOOP);
    Assert.assertTrue(settingsLatch.get().await(5, TimeUnit.SECONDS));
    Assert.assertTrue(responseLatch.await(5, TimeUnit.SECONDS));
    // Check that the data is sent correctly.
    byte[] expected = new byte[content.length];
    System.arraycopy(chunk1, 0, expected, 0, chunk1.length);
    System.arraycopy(chunk2, 0, expected, chunk1.length, chunk2.length);
    Assert.assertArrayEquals(expected, content);
}