/*
* Copyright (c) 2001, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.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<len; i++) {
buf[i] = (byte)('a' + rand.nextInt(26));
cs = (cs + buf[i]) % 65536;
}
/*
* Stream the chunks to the client
*/
int remaining = len;
int pos = 0;
while (remaining > 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<nread; i++) {
cs = (cs + b[i]) % 65536;
}
}
} while (nread > 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();
}
}