7186954: Improve connection performance
authorkhazra
Mon, 22 Oct 2012 11:49:37 -0700
changeset 16081 452341da6f0b
parent 16080 0e6266b88242
child 16082 a042f1412a78
7186954: Improve connection performance Reviewed-by: chegar, skoivu
jdk/src/share/classes/sun/net/httpserver/ChunkedInputStream.java
jdk/src/share/classes/sun/net/www/http/ChunkedInputStream.java
--- a/jdk/src/share/classes/sun/net/httpserver/ChunkedInputStream.java	Mon Oct 22 07:28:51 2012 -0700
+++ b/jdk/src/share/classes/sun/net/httpserver/ChunkedInputStream.java	Mon Oct 22 11:49:37 2012 -0700
@@ -41,8 +41,12 @@
 
     private boolean needToReadHeader = true;
 
-    static char CR = '\r';
-    static char LF = '\n';
+    final static char CR = '\r';
+    final static char LF = '\n';
+    /*
+     * Maximum chunk header size of 2KB + 2 bytes for CRLF
+     */
+    private final static int MAX_CHUNK_HEADER_SIZE = 2050;
 
     private int numeric (char[] arr, int nchars) throws IOException {
         assert arr.length >= nchars;
@@ -73,10 +77,14 @@
         char[] len_arr = new char [16];
         int len_size = 0;
         boolean end_of_len = false;
+        int read = 0;
 
         while ((c=in.read())!= -1) {
             char ch = (char) c;
-            if (len_size == len_arr.length -1) {
+            read++;
+            if ((len_size == len_arr.length -1) ||
+                (read > MAX_CHUNK_HEADER_SIZE))
+            {
                 throw new IOException ("invalid chunk header");
             }
             if (gotCR) {
--- a/jdk/src/share/classes/sun/net/www/http/ChunkedInputStream.java	Mon Oct 22 07:28:51 2012 -0700
+++ b/jdk/src/share/classes/sun/net/www/http/ChunkedInputStream.java	Mon Oct 22 11:49:37 2012 -0700
@@ -125,6 +125,11 @@
      */
     private boolean closed;
 
+    /*
+     * Maximum chunk header size of 2KB + 2 bytes for CRLF
+     */
+    private final static int MAX_CHUNK_HEADER_SIZE = 2050;
+
     /**
      * State to indicate that next field should be :-
      *  chunk-size [ chunk-extension ] CRLF
@@ -290,6 +295,10 @@
                             break;
                         }
                         pos++;
+                        if ((pos - rawPos) >= MAX_CHUNK_HEADER_SIZE) {
+                            error = true;
+                            throw new IOException("Chunk header too long");
+                        }
                     }
                     if (pos >= rawCount) {
                         return;