diff -r fd16c54261b3 -r 90ce3da70b43 jdk/test/java/net/URLConnection/ChunkedEncoding.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/jdk/test/java/net/URLConnection/ChunkedEncoding.java Sat Dec 01 00:00:00 2007 +0000 @@ -0,0 +1,193 @@ +/* + * Copyright 2001 Sun Microsystems, Inc. All Rights Reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, + * CA 95054 USA or visit www.sun.com if you need additional information or + * have any questions. + */ + +/** + * + * @bug 4333920 + * @bug 4394548 + * @summary Check that chunked encoding response doesn't cause + * getInputStream to block until last chunk arrives. + * Also regression against NPE in ChunkedInputStream. + */ +import java.net.*; +import java.io.*; +import java.util.Random; + +public class ChunkedEncoding implements Runnable { + + ServerSocket ss; + + /* + * Our "http" server to return a chunked response + */ + public void run() { + try { + Socket s = ss.accept(); + + PrintStream out = new PrintStream( + new BufferedOutputStream( + s.getOutputStream() )); + + /* send the header */ + out.print("HTTP/1.1 200\r\n"); + out.print("Transfer-Encoding: chunked\r\n"); + out.print("Content-Type: text/html\r\n"); + out.print("\r\n"); + out.flush(); + + /* delay the server before first chunk */ + Thread.sleep(5000); + + /* + * Our response will be of random length + * but > 32k + */ + Random rand = new Random(); + + int len; + do { + len = rand.nextInt(128*1024); + } while (len < 32*1024); + + /* + * Our chunk size will be 2-32k + */ + int chunkSize; + do { + chunkSize = rand.nextInt(len / 3); + } while (chunkSize < 2*1024); + + /* + * Generate random content and check sum it + */ + byte buf[] = new byte[len]; + int cs = 0; + for (int i=0; i 0) { + int size = Math.min(remaining, chunkSize); + out.print( Integer.toHexString(size) ); + out.print("\r\n"); + out.write( buf, pos, size ); + pos += size; + remaining -= size; + out.print("\r\n"); + out.flush(); + } + + /* send EOF chunk */ + out.print("0\r\n"); + out.flush(); + + /* + * Send trailer with checksum + */ + String trailer = "Checksum:" + cs + "\r\n"; + out.print(trailer); + out.print("\r\n"); + out.flush(); + + s.close(); + ss.close(); + } catch (Exception e) { + e.printStackTrace(); + } + } + + ChunkedEncoding() throws Exception { + + /* start the server */ + ss = new ServerSocket(0); + (new Thread(this)).start(); + + /* establish http connection to server */ + String uri = "http://localhost:" + + Integer.toString(ss.getLocalPort()) + + "/foo"; + URL url = new URL(uri); + HttpURLConnection http = (HttpURLConnection)url.openConnection(); + + /* + * Server should only send headers if TE:trailers + * specified - see updated HTTP 1.1 spec. + */ + http.setRequestProperty("TE", "trailers"); + + /* Time how long the getInputStream takes */ + long ts = System.currentTimeMillis(); + InputStream in = http.getInputStream(); + long te = System.currentTimeMillis(); + + /* + * If getInputStream takes >2 seconds it probably means + * that the implementation is waiting for the chunks to + * arrive. + */ + if ( (te-ts) > 2000) { + throw new Exception("getInputStream didn't return immediately"); + } + + /* + * Read the stream and checksum it as it arrives + */ + int nread; + int cs = 0; + byte b[] = new byte[1024]; + do { + nread = in.read(b); + if (nread > 0) { + for (int i=0; i 0); + + /* + * Verify that the checksums match + */ + String trailer = http.getHeaderField("Checksum"); + if (trailer == null) { + throw new Exception("Checksum trailer missing from response"); + } + int rcvd_cs = Integer.parseInt(trailer); + if (rcvd_cs != cs) { + throw new Exception("Trailer checksum doesn't equal calculated checksum"); + } + + http.disconnect(); + + } + + public static void main(String args[]) throws Exception { + new ChunkedEncoding(); + } + +}