Example 6 with ResetFrame
use of org.eclipse.jetty.http2.frames.ResetFrame 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));
}
Also used :
ServerSessionListener(org.eclipse.jetty.http2.api.server.ServerSessionListener)
FuturePromise(org.eclipse.jetty.util.FuturePromise)
InvocationType(org.eclipse.jetty.util.thread.Invocable.InvocationType)
DataFrame(org.eclipse.jetty.http2.frames.DataFrame)
CountDownLatch(java.util.concurrent.CountDownLatch)
HeadersFrame(org.eclipse.jetty.http2.frames.HeadersFrame)
ByteBuffer(java.nio.ByteBuffer)
Callback(org.eclipse.jetty.util.Callback)
FutureCallback(org.eclipse.jetty.util.FutureCallback)
MetaData(org.eclipse.jetty.http.MetaData)
HttpFields(org.eclipse.jetty.http.HttpFields)
HTTP2Stream(org.eclipse.jetty.http2.HTTP2Stream)
Stream(org.eclipse.jetty.http2.api.Stream)
ResetFrame(org.eclipse.jetty.http2.frames.ResetFrame)
ServerSessionListener(org.eclipse.jetty.http2.api.server.ServerSessionListener)
HTTP2Session(org.eclipse.jetty.http2.HTTP2Session)
Session(org.eclipse.jetty.http2.api.Session)
ISession(org.eclipse.jetty.http2.ISession)
Test(org.junit.Test)
Example 7 with ResetFrame
use of org.eclipse.jetty.http2.frames.ResetFrame in project jetty.project by eclipse.
the class PushCacheFilterTest method testPushIsReset.
@Test
public void testPushIsReset() throws Exception {
final String primaryResource = "/primary.html";
final String secondaryResource = "/secondary.png";
final byte[] secondaryData = "SECONDARY".getBytes("UTF-8");
start(new HttpServlet() {
@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
String requestURI = req.getRequestURI();
ServletOutputStream output = resp.getOutputStream();
if (requestURI.endsWith(primaryResource))
output.print("<html><head></head><body>PRIMARY</body></html>");
else if (requestURI.endsWith(secondaryResource))
output.write(secondaryData);
}
});
final Session session = newClient(new Session.Listener.Adapter());
// Request for the primary and secondary resource to build the cache.
final String primaryURI = newURI(primaryResource);
HttpFields primaryFields = new HttpFields();
MetaData.Request primaryRequest = newRequest("GET", primaryResource, primaryFields);
final CountDownLatch warmupLatch = new CountDownLatch(1);
session.newStream(new HeadersFrame(primaryRequest, null, true), new Promise.Adapter<>(), new Stream.Listener.Adapter() {
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
if (frame.isEndStream()) {
// Request for the secondary resource.
HttpFields secondaryFields = new HttpFields();
secondaryFields.put(HttpHeader.REFERER, primaryURI);
MetaData.Request secondaryRequest = newRequest("GET", secondaryResource, secondaryFields);
session.newStream(new HeadersFrame(secondaryRequest, null, true), new Promise.Adapter<>(), new Stream.Listener.Adapter() {
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
warmupLatch.countDown();
}
});
}
}
});
Assert.assertTrue(warmupLatch.await(5, TimeUnit.SECONDS));
// Request again the primary resource, we should get the secondary resource pushed.
primaryRequest = newRequest("GET", primaryResource, primaryFields);
final CountDownLatch primaryResponseLatch = new CountDownLatch(1);
final CountDownLatch pushLatch = new CountDownLatch(1);
session.newStream(new HeadersFrame(primaryRequest, null, true), new Promise.Adapter<>(), new Stream.Listener.Adapter() {
@Override
public Stream.Listener onPush(Stream stream, PushPromiseFrame frame) {
// Reset the stream as soon as we see the push.
ResetFrame resetFrame = new ResetFrame(stream.getId(), ErrorCode.REFUSED_STREAM_ERROR.code);
stream.reset(resetFrame, Callback.NOOP);
return new Adapter() {
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
pushLatch.countDown();
}
};
}
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
if (frame.isEndStream())
primaryResponseLatch.countDown();
}
});
// We should not receive pushed data that we reset.
Assert.assertFalse(pushLatch.await(1, TimeUnit.SECONDS));
Assert.assertTrue(primaryResponseLatch.await(5, TimeUnit.SECONDS));
// Make sure the session is sane by requesting the secondary resource.
HttpFields secondaryFields = new HttpFields();
secondaryFields.put(HttpHeader.REFERER, primaryURI);
MetaData.Request secondaryRequest = newRequest("GET", secondaryResource, secondaryFields);
final CountDownLatch secondaryResponseLatch = new CountDownLatch(1);
session.newStream(new HeadersFrame(secondaryRequest, null, true), new Promise.Adapter<>(), new Stream.Listener.Adapter() {
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
if (frame.isEndStream())
secondaryResponseLatch.countDown();
}
});
Assert.assertTrue(secondaryResponseLatch.await(5, TimeUnit.SECONDS));
}
Also used :
ServletOutputStream(javax.servlet.ServletOutputStream)
HeadersFrame(org.eclipse.jetty.http2.frames.HeadersFrame)
HttpServletRequest(javax.servlet.http.HttpServletRequest)
ServletException(javax.servlet.ServletException)
MetaData(org.eclipse.jetty.http.MetaData)
HttpFields(org.eclipse.jetty.http.HttpFields)
Stream(org.eclipse.jetty.http2.api.Stream)
ServletOutputStream(javax.servlet.ServletOutputStream)
HttpServlet(javax.servlet.http.HttpServlet)
HttpServletRequest(javax.servlet.http.HttpServletRequest)
HttpServletResponse(javax.servlet.http.HttpServletResponse)
IOException(java.io.IOException)
DataFrame(org.eclipse.jetty.http2.frames.DataFrame)
CountDownLatch(java.util.concurrent.CountDownLatch)
PushPromiseFrame(org.eclipse.jetty.http2.frames.PushPromiseFrame)
Promise(org.eclipse.jetty.util.Promise)
Callback(org.eclipse.jetty.util.Callback)
ResetFrame(org.eclipse.jetty.http2.frames.ResetFrame)
Session(org.eclipse.jetty.http2.api.Session)
Test(org.junit.Test)
Example 8 with ResetFrame
use of org.eclipse.jetty.http2.frames.ResetFrame in project jetty.project by eclipse.
the class StreamCountTest method testServerAllowsOneStreamEnforcedByServer.
@Test
public void testServerAllowsOneStreamEnforcedByServer() throws Exception {
final CountDownLatch resetLatch = new CountDownLatch(1);
start(new ServerSessionListener.Adapter() {
@Override
public Stream.Listener onNewStream(Stream stream, HeadersFrame frame) {
HTTP2Session session = (HTTP2Session) stream.getSession();
session.setMaxRemoteStreams(1);
return new Stream.Listener.Adapter() {
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
if (frame.isEndStream()) {
HttpFields fields = new HttpFields();
MetaData.Response metaData = new MetaData.Response(HttpVersion.HTTP_2, 200, fields);
stream.headers(new HeadersFrame(stream.getId(), metaData, null, true), callback);
} else {
callback.succeeded();
}
}
};
}
});
Session session = newClient(new Session.Listener.Adapter());
HttpFields fields = new HttpFields();
MetaData.Request metaData = newRequest("GET", fields);
HeadersFrame frame1 = new HeadersFrame(metaData, null, false);
FuturePromise<Stream> streamPromise1 = new FuturePromise<>();
final CountDownLatch responseLatch = new CountDownLatch(1);
session.newStream(frame1, streamPromise1, new Stream.Listener.Adapter() {
@Override
public void onHeaders(Stream stream, HeadersFrame frame) {
if (frame.isEndStream())
responseLatch.countDown();
}
});
Stream stream1 = streamPromise1.get(5, TimeUnit.SECONDS);
HeadersFrame frame2 = new HeadersFrame(metaData, null, false);
FuturePromise<Stream> streamPromise2 = new FuturePromise<>();
session.newStream(frame2, streamPromise2, new Stream.Listener.Adapter() {
@Override
public void onReset(Stream stream, ResetFrame frame) {
resetLatch.countDown();
}
});
streamPromise2.get(5, TimeUnit.SECONDS);
Assert.assertTrue(resetLatch.await(5, TimeUnit.SECONDS));
stream1.data(new DataFrame(stream1.getId(), BufferUtil.EMPTY_BUFFER, true), Callback.NOOP);
Assert.assertTrue(responseLatch.await(5, TimeUnit.SECONDS));
}
Also used :
ServerSessionListener(org.eclipse.jetty.http2.api.server.ServerSessionListener)
FuturePromise(org.eclipse.jetty.util.FuturePromise)
DataFrame(org.eclipse.jetty.http2.frames.DataFrame)
CountDownLatch(java.util.concurrent.CountDownLatch)
HTTP2Session(org.eclipse.jetty.http2.HTTP2Session)
HeadersFrame(org.eclipse.jetty.http2.frames.HeadersFrame)
Callback(org.eclipse.jetty.util.Callback)
MetaData(org.eclipse.jetty.http.MetaData)
HttpFields(org.eclipse.jetty.http.HttpFields)
Stream(org.eclipse.jetty.http2.api.Stream)
ResetFrame(org.eclipse.jetty.http2.frames.ResetFrame)
ServerSessionListener(org.eclipse.jetty.http2.api.server.ServerSessionListener)
HTTP2Session(org.eclipse.jetty.http2.HTTP2Session)
Session(org.eclipse.jetty.http2.api.Session)
Test(org.junit.Test)
Example 9 with ResetFrame
use of org.eclipse.jetty.http2.frames.ResetFrame in project jetty.project by eclipse.
the class StreamResetTest method testClientResetConsumesQueuedData.
@Test
public void testClientResetConsumesQueuedData() throws Exception {
start(new EmptyHttpServlet());
Session client = newClient(new Session.Listener.Adapter());
MetaData.Request request = newRequest("GET", new HttpFields());
HeadersFrame frame = new HeadersFrame(request, null, false);
FuturePromise<Stream> promise = new FuturePromise<>();
client.newStream(frame, promise, new Stream.Listener.Adapter());
Stream stream = promise.get(5, TimeUnit.SECONDS);
ByteBuffer data = ByteBuffer.allocate(FlowControlStrategy.DEFAULT_WINDOW_SIZE);
CountDownLatch dataLatch = new CountDownLatch(1);
stream.data(new DataFrame(stream.getId(), data, false), new Callback() {
@Override
public void succeeded() {
dataLatch.countDown();
}
});
// The server does not read the data, so the flow control window should be zero.
Assert.assertTrue(dataLatch.await(5, TimeUnit.SECONDS));
Assert.assertEquals(0, ((ISession) client).updateSendWindow(0));
// Now reset the stream.
stream.reset(new ResetFrame(stream.getId(), ErrorCode.CANCEL_STREAM_ERROR.code), Callback.NOOP);
// Wait for the server to receive the reset and process
// it, and for the client to process the window updates.
Thread.sleep(1000);
Assert.assertThat(((ISession) client).updateSendWindow(0), Matchers.greaterThan(0));
}
Also used :
ServerSessionListener(org.eclipse.jetty.http2.api.server.ServerSessionListener)
WriteListener(javax.servlet.WriteListener)
FuturePromise(org.eclipse.jetty.util.FuturePromise)
DataFrame(org.eclipse.jetty.http2.frames.DataFrame)
CountDownLatch(java.util.concurrent.CountDownLatch)
HeadersFrame(org.eclipse.jetty.http2.frames.HeadersFrame)
ByteBuffer(java.nio.ByteBuffer)
Callback(org.eclipse.jetty.util.Callback)
FutureCallback(org.eclipse.jetty.util.FutureCallback)
MetaData(org.eclipse.jetty.http.MetaData)
HttpFields(org.eclipse.jetty.http.HttpFields)
Stream(org.eclipse.jetty.http2.api.Stream)
ServletOutputStream(javax.servlet.ServletOutputStream)
IStream(org.eclipse.jetty.http2.IStream)
ResetFrame(org.eclipse.jetty.http2.frames.ResetFrame)
Session(org.eclipse.jetty.http2.api.Session)
ISession(org.eclipse.jetty.http2.ISession)
Test(org.junit.Test)
Example 10 with ResetFrame
use of org.eclipse.jetty.http2.frames.ResetFrame in project jetty.project by eclipse.
the class StreamResetTest method testStreamResetDoesNotCloseConnection.
@Test
public void testStreamResetDoesNotCloseConnection() throws Exception {
final CountDownLatch serverResetLatch = new CountDownLatch(1);
final CountDownLatch serverDataLatch = new CountDownLatch(1);
start(new ServerSessionListener.Adapter() {
@Override
public Stream.Listener onNewStream(Stream stream, HeadersFrame requestFrame) {
MetaData.Response response = new MetaData.Response(HttpVersion.HTTP_2, 200, new HttpFields());
HeadersFrame responseFrame = new HeadersFrame(stream.getId(), response, null, false);
Callback.Completable completable = new Callback.Completable();
stream.headers(responseFrame, completable);
return new Stream.Listener.Adapter() {
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
completable.thenRun(() -> stream.data(new DataFrame(stream.getId(), ByteBuffer.allocate(16), true), new Callback() {
@Override
public void succeeded() {
serverDataLatch.countDown();
}
}));
}
@Override
public void onReset(Stream s, ResetFrame frame) {
// Simulate that there is pending data to send.
IStream stream = (IStream) s;
stream.getSession().frames(stream, new Callback() {
@Override
public void failed(Throwable x) {
serverResetLatch.countDown();
}
}, new DataFrame(s.getId(), ByteBuffer.allocate(16), true));
}
};
}
});
Session client = newClient(new Session.Listener.Adapter());
MetaData.Request request1 = newRequest("GET", new HttpFields());
HeadersFrame requestFrame1 = new HeadersFrame(request1, null, false);
FuturePromise<Stream> promise1 = new FuturePromise<>();
final CountDownLatch stream1HeadersLatch = new CountDownLatch(1);
final CountDownLatch stream1DataLatch = new CountDownLatch(1);
client.newStream(requestFrame1, promise1, new Stream.Listener.Adapter() {
@Override
public void onHeaders(Stream stream, HeadersFrame frame) {
stream1HeadersLatch.countDown();
}
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
stream1DataLatch.countDown();
}
});
Stream stream1 = promise1.get(5, TimeUnit.SECONDS);
Assert.assertTrue(stream1HeadersLatch.await(5, TimeUnit.SECONDS));
MetaData.Request request2 = newRequest("GET", new HttpFields());
HeadersFrame requestFrame2 = new HeadersFrame(request2, null, false);
FuturePromise<Stream> promise2 = new FuturePromise<>();
final CountDownLatch stream2DataLatch = new CountDownLatch(1);
client.newStream(requestFrame2, promise2, new Stream.Listener.Adapter() {
@Override
public void onData(Stream stream, DataFrame frame, Callback callback) {
callback.succeeded();
stream2DataLatch.countDown();
}
});
Stream stream2 = promise2.get(5, TimeUnit.SECONDS);
ResetFrame resetFrame = new ResetFrame(stream1.getId(), ErrorCode.CANCEL_STREAM_ERROR.code);
stream1.reset(resetFrame, Callback.NOOP);
Assert.assertTrue(serverResetLatch.await(5, TimeUnit.SECONDS));
// Stream MUST NOT receive data sent by server after reset.
Assert.assertFalse(stream1DataLatch.await(1, TimeUnit.SECONDS));
// The other stream should still be working.
stream2.data(new DataFrame(stream2.getId(), ByteBuffer.allocate(16), true), Callback.NOOP);
Assert.assertTrue(serverDataLatch.await(5, TimeUnit.SECONDS));
Assert.assertTrue(stream2DataLatch.await(5, TimeUnit.SECONDS));
}
Also used :
ServerSessionListener(org.eclipse.jetty.http2.api.server.ServerSessionListener)
WriteListener(javax.servlet.WriteListener)
IStream(org.eclipse.jetty.http2.IStream)
FuturePromise(org.eclipse.jetty.util.FuturePromise)
DataFrame(org.eclipse.jetty.http2.frames.DataFrame)
CountDownLatch(java.util.concurrent.CountDownLatch)
HeadersFrame(org.eclipse.jetty.http2.frames.HeadersFrame)
HttpServletResponse(javax.servlet.http.HttpServletResponse)
Callback(org.eclipse.jetty.util.Callback)
FutureCallback(org.eclipse.jetty.util.FutureCallback)
MetaData(org.eclipse.jetty.http.MetaData)
HttpFields(org.eclipse.jetty.http.HttpFields)
Stream(org.eclipse.jetty.http2.api.Stream)
ServletOutputStream(javax.servlet.ServletOutputStream)
IStream(org.eclipse.jetty.http2.IStream)
ResetFrame(org.eclipse.jetty.http2.frames.ResetFrame)
ServerSessionListener(org.eclipse.jetty.http2.api.server.ServerSessionListener)
Session(org.eclipse.jetty.http2.api.Session)
ISession(org.eclipse.jetty.http2.ISession)
Test(org.junit.Test)
Aggregations
ResetFrame (org.eclipse.jetty.http2.frames.ResetFrame)24
Test (org.junit.Test)21
Stream (org.eclipse.jetty.http2.api.Stream)20
HeadersFrame (org.eclipse.jetty.http2.frames.HeadersFrame)20
CountDownLatch (java.util.concurrent.CountDownLatch)19
MetaData (org.eclipse.jetty.http.MetaData)19
HttpFields (org.eclipse.jetty.http.HttpFields)18
ServerSessionListener (org.eclipse.jetty.http2.api.server.ServerSessionListener)18
Session (org.eclipse.jetty.http2.api.Session)16
Callback (org.eclipse.jetty.util.Callback)13
FuturePromise (org.eclipse.jetty.util.FuturePromise)12
DataFrame (org.eclipse.jetty.http2.frames.DataFrame)11
IOException (java.io.IOException)10
HttpServletResponse (javax.servlet.http.HttpServletResponse)10
ByteBuffer (java.nio.ByteBuffer)9
ServletOutputStream (javax.servlet.ServletOutputStream)9
HttpServletRequest (javax.servlet.http.HttpServletRequest)9
ISession (org.eclipse.jetty.http2.ISession)9
ServletException (javax.servlet.ServletException)8
WriteListener (javax.servlet.WriteListener)8
