This is an automated email from the ASF dual-hosted git repository.

markt pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tomcat.git

commit 519403fb8aace2c64e1ee3999511719418dc6968
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 349e2f5942..a0f997e13a 100644
--- a/java/org/apache/coyote/http2/Stream.java
+++ b/java/org/apache/coyote/http2/Stream.java
@@ -335,8 +335,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(
@@ -448,6 +447,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 635949fa34..7398ce0de5 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 4dc566fb9a..a81a450f6b 100644
--- a/test/org/apache/coyote/http2/Http2TestBase.java
+++ b/test/org/apache/coyote/http2/Http2TestBase.java
@@ -1339,6 +1339,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 ba24331f65..ede9540a77 100644
--- a/webapps/docs/changelog.xml
+++ b/webapps/docs/changelog.xml
@@ -151,6 +151,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

Reply via email to