Fix remainingStreamingBytes set before requestHandler for GET requests

Co-authored-by: uNetworkingAB <110806833+uNetworkingAB@users.noreply.github.com>

Author: Copilot

src/HttpParser.h

@@ -533,15 +533,6 @@ struct HttpParser {
             const char *querySeparatorPtr = (const char *) memchr(req->headers->value.data(), '?', req->headers->value.length());
             req->querySeparator = (unsigned int) ((querySeparatorPtr ? querySeparatorPtr : req->headers->value.data() + req->headers->value.length()) - req->headers->value.data());
 
-            /* If returned socket is not what we put in we need
-             * to break here as we either have upgraded to
-             * WebSockets or otherwise closed the socket. */
-            void *returnedUser = requestHandler(user, req);
-            if (returnedUser != user) {
-                /* We are upgraded to WebSocket or otherwise broken */
-                return {consumedTotal, returnedUser};
-            }
-
             /* The rules at play here according to RFC 9112 for requests are essentially:
              * If both content-length and transfer-encoding then invalid message; must break.
              * If has transfer-encoding then must be chunked regardless of value.
@@ -566,6 +557,28 @@ struct HttpParser {
                  * This could be made stricter but makes no difference either way, unless forwarding the identical message as a proxy. */
 
                 remainingStreamingBytes = STATE_IS_CHUNKED;
+            } else if (contentLengthString.length()) {
+                remainingStreamingBytes = toUnsignedInteger(contentLengthString);
+                if (remainingStreamingBytes == UINT64_MAX) {
+                    /* Parser error */
+                    return {HTTP_ERROR_400_BAD_REQUEST, FULLPTR};
+                }
+            } else {
+                /* No body (e.g. GET requests); set to 0 to match this assumption */
+                remainingStreamingBytes = 0;
+            }
+
+            /* If returned socket is not what we put in we need
+             * to break here as we either have upgraded to
+             * WebSockets or otherwise closed the socket. */
+            void *returnedUser = requestHandler(user, req);
+            if (returnedUser != user) {
+                /* We are upgraded to WebSocket or otherwise broken */
+                return {consumedTotal, returnedUser};
+            }
+
+            /* Consume body data */
+            if (transferEncodingString.length()) {
                 /* If consume minimally, we do not want to consume anything but we want to mark this as being chunked */
                 if (!CONSUME_MINIMALLY) {
                     /* Go ahead and parse it (todo: better heuristics for emitting FIN to the app level) */
@@ -582,12 +595,6 @@ struct HttpParser {
                     consumedTotal += consumed;
                 }
             } else if (contentLengthString.length()) {
-                remainingStreamingBytes = toUnsignedInteger(contentLengthString);
-                if (remainingStreamingBytes == UINT64_MAX) {
-                    /* Parser error */
-                    return {HTTP_ERROR_400_BAD_REQUEST, FULLPTR};
-                }
-
                 if (!CONSUME_MINIMALLY) {
                     unsigned int emittable = (unsigned int) std::min<uint64_t>(remainingStreamingBytes, length);
                     dataHandler(user, std::string_view(data, emittable), emittable == remainingStreamingBytes);

tests/BloomFilter

@@ -12,7 +12,7 @@ int main() {
 
     uWS::HttpParser httpParser;
 
-    auto [err, returnedUser] = httpParser.consumePostPadded((char *) data, size, user, reserved, [reserved](void *s, uWS::HttpRequest *httpRequest) -> void * {
+    auto [err, returnedUser] = httpParser.consumePostPadded((char *) data, size, user, reserved, [reserved, &httpParser](void *s, uWS::HttpRequest *httpRequest) -> void * {
 
         std::cout << httpRequest->getMethod() << std::endl;
 
@@ -23,6 +23,9 @@ int main() {
         /* Since we did proper whitespace trimming this thing is there, but empty */
         assert(httpRequest->getHeader("utf8").data());
 
+        /* maxRemainingBodyLength() must be 0 for GET requests (no body) when called inside requestHandler */
+        assert(httpParser.maxRemainingBodyLength() == 0);
+
         /* Return ok */
         return s;
 
@@ -35,4 +38,24 @@ int main() {
 
     std::cout << "HTTP DONE" << std::endl;
 
+    /* Test that maxRemainingBodyLength() is set correctly before requestHandler for a POST with content-length */
+    {
+        /* POST / HTTP/1.1\r\nHost: localhost\r\nContent-Length: 5\r\n\r\nhello + 8 bytes padding */
+        const char postData[] = "POST / HTTP/1.1\r\nHost: localhost\r\nContent-Length: 5\r\n\r\nhelloEEEEEEEE";
+        int postSize = (int)(sizeof(postData) - 1 - 8);
+        uWS::HttpParser postParser;
+        uint64_t bodyLengthInHandler = UINT64_MAX;
+
+        postParser.consumePostPadded((char *) postData, postSize, user, reserved, [&postParser, &bodyLengthInHandler](void *s, uWS::HttpRequest *httpRequest) -> void * {
+            bodyLengthInHandler = postParser.maxRemainingBodyLength();
+            return s;
+        }, [](void *user, std::string_view data, bool fin) -> void * {
+            return user;
+        });
+
+        /* maxRemainingBodyLength() must return content-length (5) when called inside requestHandler */
+        assert(bodyLengthInHandler == 5);
+        std::cout << "POST content-length test PASS" << std::endl;
+    }
+
 }