This is an automated email from the ASF dual-hosted git repository. markt pushed a commit to branch 10.1.x in repository https://gitbox.apache.org/repos/asf/tomcat.git
commit 063a326f79908a58d2aed1fbe4629865075240ab Author: Mark Thomas <ma...@apache.org> AuthorDate: Wed Jan 25 16:28:34 2023 +0000 Fix BZ 66442 - h2 responses without bodies should not use DATA frames https://bz.apache.org/bugzilla/show_bug.cgi?id=66442 --- java/org/apache/coyote/http2/Stream.java | 9 +++- java/org/apache/coyote/http2/StreamProcessor.java | 4 ++ test/org/apache/coyote/http2/Http2TestBase.java | 12 ++++++ .../apache/coyote/http2/TestStreamProcessor.java | 49 ++++++++++++++++++++++ webapps/docs/changelog.xml | 5 +++ 5 files changed, 77 insertions(+), 2 deletions(-) diff --git a/java/org/apache/coyote/http2/Stream.java b/java/org/apache/coyote/http2/Stream.java index 26eaf1b7f4..36aebe44d4 100644 --- a/java/org/apache/coyote/http2/Stream.java +++ b/java/org/apache/coyote/http2/Stream.java @@ -334,8 +334,7 @@ class Stream extends AbstractNonZeroStream implements HeaderEmitter { if (coyoteRequest.method().isNull()) { coyoteRequest.method().setString(value); if ("HEAD".equals(value)) { - addOutputFilter(new VoidOutputFilter()); - streamOutputBuffer.closed = true; + configureVoidOutputFilter(); } } else { throw new HpackException( @@ -437,6 +436,12 @@ class Stream extends AbstractNonZeroStream implements HeaderEmitter { } + void configureVoidOutputFilter() { + addOutputFilter(new VoidOutputFilter()); + // Prevent further writes by the application + streamOutputBuffer.closed = true; + } + private void parseAuthority(String value, boolean host) throws HpackException { int i; try { diff --git a/java/org/apache/coyote/http2/StreamProcessor.java b/java/org/apache/coyote/http2/StreamProcessor.java index fef400d4d8..1a991a05ab 100644 --- a/java/org/apache/coyote/http2/StreamProcessor.java +++ b/java/org/apache/coyote/http2/StreamProcessor.java @@ -205,6 +205,10 @@ class StreamProcessor extends AbstractProcessor { headers.addValue("content-length").setLong(contentLength); } } else { + // Disable response body + if (stream != null) { + stream.configureVoidOutputFilter(); + } if (statusCode == 205) { // RFC 7231 requires the server to explicitly signal an empty // response in this case diff --git a/test/org/apache/coyote/http2/Http2TestBase.java b/test/org/apache/coyote/http2/Http2TestBase.java index 301450011c..f9763e3fe9 100644 --- a/test/org/apache/coyote/http2/Http2TestBase.java +++ b/test/org/apache/coyote/http2/Http2TestBase.java @@ -1306,6 +1306,18 @@ public abstract class Http2TestBase extends TomcatBaseTest { } + public static class NoContentServlet extends HttpServlet { + + private static final long serialVersionUID = 1L; + + @Override + protected void doGet(HttpServletRequest req, HttpServletResponse resp) + throws ServletException, IOException { + resp.setStatus(HttpServletResponse.SC_NO_CONTENT); + } + } + + public static class SimpleServlet extends HttpServlet { private static final long serialVersionUID = 1L; diff --git a/test/org/apache/coyote/http2/TestStreamProcessor.java b/test/org/apache/coyote/http2/TestStreamProcessor.java index 8516d98f22..16fa106567 100644 --- a/test/org/apache/coyote/http2/TestStreamProcessor.java +++ b/test/org/apache/coyote/http2/TestStreamProcessor.java @@ -164,6 +164,55 @@ public class TestStreamProcessor extends Http2TestBase { } expected.append("3-Header-[date]-[Wed, 11 Nov 2015 19:18:42 GMT]\n"); expected.append("3-HeadersEnd\n"); + expected.append("3-EndOfStream\n"); + + Assert.assertEquals(expected.toString(), output.getTrace()); + } + + + @Test + public void testPrepareHeadersNoContent() throws Exception { + enableHttp2(); + + Tomcat tomcat = getTomcatInstance(); + + File appDir = new File("test/webapp"); + Context ctxt = tomcat.addWebapp(null, "", appDir.getAbsolutePath()); + + Tomcat.addServlet(ctxt, "simple", new SimpleServlet()); + ctxt.addServletMappingDecoded("/simple", "simple"); + Tomcat.addServlet(ctxt, "noContent", new NoContentServlet()); + ctxt.addServletMappingDecoded("/noContent", "noContent"); + + + tomcat.start(); + + openClientConnection(); + doHttpUpgrade(); + sendClientPreface(); + validateHttp2InitialResponse(); + + byte[] frameHeader = new byte[9]; + ByteBuffer headersPayload = ByteBuffer.allocate(128); + + List<Header> headers = new ArrayList<>(3); + headers.add(new Header(":method", "GET")); + headers.add(new Header(":scheme", "http")); + headers.add(new Header(":path", "/noContent")); + headers.add(new Header(":authority", "localhost:" + getPort())); + + buildGetRequest(frameHeader, headersPayload, null, headers, 3); + + writeFrame(frameHeader, headersPayload); + + parser.readFrame(); + + StringBuilder expected = new StringBuilder(); + expected.append("3-HeadersStart\n"); + expected.append("3-Header-[:status]-[204]\n"); + expected.append("3-Header-[date]-[Wed, 11 Nov 2015 19:18:42 GMT]\n"); + expected.append("3-HeadersEnd\n"); + expected.append("3-EndOfStream\n"); Assert.assertEquals(expected.toString(), output.getTrace()); } diff --git a/webapps/docs/changelog.xml b/webapps/docs/changelog.xml index 886dd1521a..82c6c566a4 100644 --- a/webapps/docs/changelog.xml +++ b/webapps/docs/changelog.xml @@ -127,6 +127,11 @@ Log basic information for each configured TLS certificate when Tomcat starts. (markt) </add> + <fix> + <bug>66442</bug>: When an HTTP/2 response must not include a body, + ensure that the end of stream flag is set on the headers frame and that + no data frame is sent. (markt) + </fix> </changelog> </subsection> <subsection name="Jasper"> --------------------------------------------------------------------- To unsubscribe, e-mail: dev-unsubscr...@tomcat.apache.org For additional commands, e-mail: dev-h...@tomcat.apache.org