--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/sun/net/www/http/ChunkedOutputStream/Test.java Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,420 @@
+/*
+ * Copyright 2004-2007 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.
+ */
+
+/**
+ * @test
+ * @bug 5026745
+ * @run main/othervm/timeout=500 Test
+ * @summary Cannot flush output stream when writing to an HttpUrlConnection
+ */
+
+import java.io.*;
+import java.net.*;
+import com.sun.net.httpserver.*;
+
+public class Test implements HttpHandler {
+
+ static int count = 0;
+
+ static final String str1 = "Helloworld1234567890abcdefghijklmnopqrstuvwxyz"+
+ "1234567890abcdefkjsdlkjflkjsldkfjlsdkjflkj"+
+ "1434567890abcdefkjsdlkjflkjsldkfjlsdkjflkj";
+
+ static final String str2 = "Helloworld1234567890abcdefghijklmnopqrstuvwxyz"+
+ "1234567890";
+
+ public void handle(HttpExchange exchange) {
+ String reqbody;
+ try {
+ switch (count) {
+ case 0: /* test1 -- keeps conn alive */
+ case 1: /* test2 -- closes conn */
+ printRequestURI(exchange);
+ reqbody = read(exchange.getRequestBody());
+ if (!reqbody.equals(str1)) {
+ exchange.sendResponseHeaders(500, 0);
+ break;
+ }
+
+ Headers headers = exchange.getRequestHeaders();
+ String chunk = headers.getFirst("Transfer-encoding");
+
+ if (!"chunked".equals (chunk)) {
+ exchange.sendResponseHeaders(501, 0);
+ break;
+ }
+
+ exchange.sendResponseHeaders(200, reqbody.length());
+ write(exchange.getResponseBody(), reqbody);
+
+ if (count == 1) {
+ Headers resHeaders = exchange.getResponseHeaders() ;
+ resHeaders.set("Connection", "close");
+ }
+ break;
+ case 2: /* test 3 */
+ printRequestURI(exchange);
+ reqbody = read(exchange.getRequestBody());
+
+ if (!reqbody.equals(str2)) {
+ exchange.sendResponseHeaders(500, 0);
+ break;
+ }
+ headers = exchange.getRequestHeaders();
+ int clen = Integer.parseInt( headers.getFirst("Content-length"));
+
+ if (clen != str2.length()) {
+ exchange.sendResponseHeaders(501, 0);
+ break;
+ }
+ Headers resHeaders = exchange.getResponseHeaders() ;
+ resHeaders.set("Connection", "close");
+
+ exchange.sendResponseHeaders(200, reqbody.length());
+ write(exchange.getResponseBody(), reqbody);
+ break;
+ case 3: /* test 4 */
+ case 4: /* test 5 */
+ printRequestURI(exchange);
+ break;
+ case 5: /* test 6 */
+ printRequestURI(exchange);
+ resHeaders = exchange.getResponseHeaders() ;
+ resHeaders.set("Location", "http://foo.bar/");
+ resHeaders.set("Connection", "close");
+ exchange.sendResponseHeaders(307, 0);
+ break;
+ case 6: /* test 7 */
+ case 7: /* test 8 */
+ printRequestURI(exchange);
+ reqbody = read(exchange.getRequestBody());
+ if (reqbody != null && !"".equals(reqbody)) {
+ exchange.sendResponseHeaders(501, 0);
+ break;
+ }
+ resHeaders = exchange.getResponseHeaders() ;
+ resHeaders.set("Connection", "close");
+ exchange.sendResponseHeaders(200, 0);
+ break;
+ case 8: /* test 9 */
+ printRequestURI(exchange);
+ reqbody = read(exchange.getRequestBody());
+ if (!reqbody.equals(str1)) {
+ exchange.sendResponseHeaders(500, 0);
+ break;
+ }
+
+ headers = exchange.getRequestHeaders();
+ chunk = headers.getFirst("Transfer-encoding");
+ if (!"chunked".equals(chunk)) {
+ exchange.sendResponseHeaders(501, 0);
+ break;
+ }
+
+ exchange.sendResponseHeaders(200, reqbody.length());
+ write(exchange.getResponseBody(), reqbody);
+ break;
+ case 9: /* test10 */
+ printRequestURI(exchange);
+ InputStream is = exchange.getRequestBody();
+ String s = read (is, str1.length());
+
+ boolean error = false;
+ for (int i=10; i< 200 * 1024; i++) {
+ byte c = (byte)is.read();
+
+ if (c != (byte)i) {
+ error = true;
+ System.out.println ("error at position " + i);
+ }
+ }
+ if (!s.equals(str1) ) {
+ System.out.println ("received string : " + s);
+ exchange.sendResponseHeaders(500, 0);
+ } else if (error) {
+ System.out.println ("error");
+ exchange.sendResponseHeaders(500, 0);
+ } else {
+ exchange.sendResponseHeaders(200, 0);
+ }
+ break;
+ }
+ exchange.close();
+ count ++;
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+ }
+
+ static void printRequestURI(HttpExchange exchange) {
+ URI uri = exchange.getRequestURI();
+ System.out.println("HttpServer: handle " + uri);
+ }
+
+
+ static String read (InputStream is, int len) {
+ try {
+ byte[] ba = new byte [len];
+ int c;
+ int l = 0;
+ while ((c= is.read(ba, l, ba.length-l)) != -1 && l<len) {
+ l += c;
+ }
+ return new String (ba, 0, l, "ISO8859-1");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ return null;
+ }
+
+ static String read(InputStream is) {
+ try {
+ byte[] ba = new byte [8096];
+ int off = 0, c;
+ while ((c= is.read(ba, off, ba.length)) != -1) {
+ off += c;
+ }
+ return new String(ba, 0, off, "ISO8859-1");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ return null;
+ }
+
+ static void write(OutputStream os, String str) {
+ try {
+ byte[] ba = str.getBytes("ISO8859-1");
+ os.write(ba);
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
+ static void readAndCompare(InputStream is, String cmp) throws IOException {
+ int c;
+ byte buf[] = new byte [1024];
+ int off = 0;
+ int len = 1024;
+ while ((c=is.read(buf, off, len)) != -1) {
+ off += c;
+ len -= c;
+ }
+ String s1 = new String(buf, 0, off, "ISO8859_1");
+ if (!cmp.equals(s1)) {
+ throw new IOException("strings not same");
+ }
+ }
+
+ /* basic chunked test (runs twice) */
+
+ static void test1 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setChunkedStreamingMode (20);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ os.write (str1.getBytes());
+ os.close();
+ InputStream is = urlc.getInputStream();
+ readAndCompare (is, str1);
+ is.close();
+ }
+
+ /* basic fixed length test */
+
+ static void test3 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setFixedLengthStreamingMode (str2.length());
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ os.write (str2.getBytes());
+ os.close();
+ InputStream is = urlc.getInputStream();
+ readAndCompare (is, str2);
+ is.close();
+ }
+
+ /* write too few bytes */
+
+ static void test4 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setFixedLengthStreamingMode (str2.length()+1);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ os.write (str2.getBytes());
+ try {
+ os.close();
+ throw new Exception ("should have thrown IOException");
+ } catch (IOException e) {}
+ }
+
+ /* write too many bytes */
+
+ static void test5 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setFixedLengthStreamingMode (str2.length()-1);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ try {
+ os.write (str2.getBytes());
+ throw new Exception ("should have thrown IOException");
+ } catch (IOException e) {}
+ }
+
+ /* check for HttpRetryException on redirection */
+
+ static void test6 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setChunkedStreamingMode (20);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ os.write (str1.getBytes());
+ os.close();
+ try {
+ InputStream is = urlc.getInputStream();
+ throw new Exception ("should have gotten HttpRetryException");
+ } catch (HttpRetryException e) {
+ if (e.responseCode() != 307) {
+ throw new Exception ("Wrong response code " + e.responseCode());
+ }
+ if (!e.getLocation().equals ("http://foo.bar/")) {
+ throw new Exception ("Wrong location " + e.getLocation());
+ }
+ }
+ }
+
+ /* next two tests send zero length posts */
+
+ static void test7 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setChunkedStreamingMode (20);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ os.close();
+ int ret = urlc.getResponseCode();
+ if (ret != 200) {
+ throw new Exception ("Expected 200: got " + ret);
+ }
+ }
+
+ static void test8 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setFixedLengthStreamingMode (0);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ os.close();
+ int ret = urlc.getResponseCode();
+ if (ret != 200) {
+ throw new Exception ("Expected 200: got " + ret);
+ }
+ }
+
+ /* calling setChunkedStreamingMode with -1 should entail using
+ the default chunk size */
+ static void test9 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setChunkedStreamingMode (-1);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ os.write (str1.getBytes());
+ os.close();
+ InputStream is = urlc.getInputStream();
+ readAndCompare (is, str1);
+ is.close();
+ }
+
+ static void test10 (String u) throws Exception {
+ URL url = new URL (u);
+ System.out.println ("client opening connection to: " + u);
+ HttpURLConnection urlc = (HttpURLConnection)url.openConnection ();
+ urlc.setChunkedStreamingMode (4 * 1024);
+ urlc.setDoOutput(true);
+ urlc.setRequestMethod ("POST");
+ OutputStream os = urlc.getOutputStream ();
+ byte[] buf = new byte [200 * 1024];
+ for (int i=0; i< 200 * 1024; i++) {
+ buf[i] = (byte) i;
+ }
+ /* write a small bit first, and then the large buffer */
+ os.write (str1.getBytes());
+ os.write (buf, 10, buf.length - 10); /* skip 10 bytes to test offset */
+ os.close();
+ InputStream is = urlc.getInputStream();
+ is.close();
+ int ret = urlc.getResponseCode();
+ if (ret != 200) {
+ throw new Exception ("Expected 200: got " + ret);
+ }
+ }
+
+ static com.sun.net.httpserver.HttpServer httpserver;
+
+ public static void main (String[] args) throws Exception {
+ try {
+ httpserver = com.sun.net.httpserver.HttpServer.create(new InetSocketAddress(0), 0);
+ HttpContext ctx = httpserver.createContext("/test/", new Test() );
+ httpserver.start();
+
+ int port = httpserver.getAddress().getPort();
+
+ System.out.println ("Server started: listening on port: " + port);
+ test1("http://localhost:"+ port + "/test/test1");
+ test1("http://localhost:"+ port + "/test/test2");
+ test3("http://localhost:"+ port + "/test/test3");
+ test4("http://localhost:"+ port + "/test/test4");
+ test5("http://localhost:"+ port + "/test/test5");
+ test6("http://localhost:"+ port + "/test/test6");
+ test7("http://localhost:"+ port + "/test/test7");
+ test8("http://localhost:"+ port + "/test/test8");
+ test9("http://localhost:"+ port + "/test/test9");
+ test10("http://localhost:"+ port + "/test/test10");
+ } finally {
+ if (httpserver != null)
+ httpserver.stop(0);
+ }
+ }
+
+}