TOMEE-4306 - Patches from apache/tomcat@d07c821#diff-30a0a7f67d009079…

…f71383913f933b7f16932c75f2854af8dc3a8a1f54e0d4d7
apache · Mar 26, 2024 · e6d2b4d · e6d2b4d
1 parent 37fad09
commit e6d2b4d
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tomee/apache-tomee/src/patch/java/org/apache/coyote/http2/Http2Parser.java b/tomee/apache-tomee/src/patch/java/org/apache/coyote/http2/Http2Parser.java
@@ -289,6 +289,9 @@ protected void readHeadersFrame(int streamId, int flags, int payloadSize, ByteBu
 
         swallowPayload(streamId, FrameType.HEADERS.getId(), padLength, true, buffer);
 
+        // Validate the headers so far
+        hpackDecoder.getHeaderEmitter().validateHeaders();
+
         if (Flags.isEndOfHeaders(flags)) {
             onHeadersComplete(streamId);
         } else {
@@ -467,6 +470,9 @@ protected void readContinuationFrame(int streamId, int flags, int payloadSize, B
 
         readHeaderPayload(streamId, payloadSize, buffer);
 
+        // Validate the headers so far
+        hpackDecoder.getHeaderEmitter().validateHeaders();
+
         if (endOfHeaders) {
             headersCurrentStream = -1;
             onHeadersComplete(streamId);
@@ -633,11 +639,6 @@ protected void onHeadersComplete(int streamId) throws Http2Exception {
                     Http2Error.COMPRESSION_ERROR);
         }
 
-        // Delay validation (and triggering any exception) until this point
-        // since all the headers still have to be read if a StreamException is
-        // going to be thrown.
-        hpackDecoder.getHeaderEmitter().validateHeaders();
-
         synchronized (output) {
             output.headersEnd(streamId);