fix AsyncPageTransportServlet issues

prestodb · Nov 19, 2024 · d6b4b9e · d6b4b9e
1 parent 150c7eb
commit d6b4b9e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/server/AsyncPageTransportServlet.java b/presto-main/src/main/java/com/facebook/presto/server/AsyncPageTransportServlet.java
@@ -128,12 +128,14 @@ protected void parseURI(String requestURI, HttpServletRequest request, HttpServl
         OutputBufferId bufferId = null;
         long token = 0;
 
+        String sanitizedRequestURI = requestURI.replaceAll("[\\r\\n]", "");
+
         int previousIndex = -1;
         for (int part = 0; part < 8; part++) {
             int nextIndex = requestURI.indexOf('/', previousIndex + 1);
 
             if (nextIndex == -1 && part != 7 || nextIndex != -1 && part == 7) {
-                reportFailure(response, format("Unexpected URI for task result request in async mode: %s", requestURI));
+                reportFailure(response, format("Unexpected URI for task result request in async mode: %s", sanitizedRequestURI));
                 return;
             }
 
@@ -152,7 +154,7 @@ protected void parseURI(String requestURI, HttpServletRequest request, HttpServl
             previousIndex = nextIndex;
         }
         // This is sent forward instead of returned to avoid allocations
-        processRequest(requestURI, taskId, bufferId, token, request, response);
+        processRequest(sanitizedRequestURI, taskId, bufferId, token, request, response);
     }
 
     protected void processRequest(