| /* |
| * Copyright (c) 2001, 2010, 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. |
| */ |
| |
| /** |
| * @test |
| * @bug 4361492 |
| * @summary HTTPUrlConnection does not receive binary data correctly |
| * @run main/timeout=20 ResendPostBody |
| */ |
| |
| import java.io.*; |
| import java.net.*; |
| |
| /* |
| * This test does the following: |
| * 1. client opens HTTP connection to server |
| * 2. client sends POST with a body containing "ZZZ" |
| * 3. server waits for POST and closes connection without replying |
| * 4. client should re-open the connection and re-send the POST |
| * 5. <bug>The client forgets to re-send the body with the POST |
| * The server hangs waiting for the body</bug> |
| * |
| * <bugfixed>The client sends the body. The server reads it and the |
| * test terminates normally </bugfixed> |
| */ |
| |
| public class ResendPostBody { |
| |
| static class Server extends Thread { |
| |
| InputStream in; |
| OutputStream out; |
| Socket sock; |
| StringBuffer response; |
| ServerSocket server; |
| |
| Server (ServerSocket s) throws IOException |
| { |
| server = s; |
| } |
| |
| void waitFor (String s) throws IOException |
| { |
| byte[] w = s.getBytes (); |
| for(int c=0; c<w.length; c++ ) { |
| byte expected = w[c]; |
| int b = in.read(); |
| if (b == -1) { |
| acceptConn (); |
| } |
| if ((byte)b != expected) { |
| c = 0; |
| } |
| } |
| } |
| |
| boolean done = false; |
| |
| public synchronized boolean finished () { |
| return done; |
| } |
| |
| public synchronized void setFinished (boolean b) { |
| done = b; |
| } |
| |
| void acceptConn () throws IOException |
| { |
| sock = server.accept (); |
| in = sock.getInputStream (); |
| out = sock.getOutputStream (); |
| } |
| |
| public void run () { |
| try { |
| response = new StringBuffer (1024); |
| acceptConn (); |
| waitFor ("POST"); |
| waitFor ("ZZZ"); |
| Thread.sleep (500); |
| sock.close (); |
| acceptConn (); |
| waitFor ("POST"); |
| waitFor ("ZZZ"); |
| response.append ("HTTP/1.1 200 OK\r\n"); |
| response.append ("Server: Microsoft-IIS/5.0"); |
| response.append ("Date: Wed, 26 Jul 2000 14:17:04 GMT\r\n\r\n"); |
| out.write (response.toString().getBytes()); |
| while (!finished()) { |
| Thread.sleep (1000); |
| } |
| out.close(); |
| } catch (Exception e) { |
| System.err.println ("Server Exception: " + e); |
| } finally { |
| try { server.close(); } catch (IOException unused) {} |
| } |
| } |
| } |
| |
| ServerSocket ss; |
| Server server; |
| |
| public static void main(String[] args) throws Exception { |
| try { |
| if (args.length == 1 && args[0].equals ("-i")) { |
| System.out.println ("Press return when ready"); |
| System.in.read (); |
| System.out.println ("Done"); |
| } |
| ResendPostBody t = new ResendPostBody (); |
| t. execute (); |
| } catch (IOException e) { |
| System.out.println ("IOException"); |
| } |
| } |
| |
| public void execute () throws Exception { |
| |
| byte b[] = "X=ABCDEFGHZZZ".getBytes(); |
| |
| ss = new ServerSocket (0); |
| server = new Server (ss); |
| server.start (); |
| /* Get the URL */ |
| |
| String s = "http://localhost:"+ss.getLocalPort()+"/test"; |
| URL url = new URL(s); |
| HttpURLConnection conURL = (HttpURLConnection)url.openConnection(); |
| |
| conURL.setDoOutput(true); |
| conURL.setDoInput(true); |
| conURL.setAllowUserInteraction(false); |
| conURL.setUseCaches(false); |
| conURL.setRequestProperty("Content-Type", "application/x-www-form-urlencoded"); |
| conURL.setRequestProperty("Content-Length", ""+b.length); |
| conURL.setRequestProperty("Connection", "Close"); |
| |
| /* POST some data */ |
| |
| DataOutputStream OutStream = new DataOutputStream(conURL.getOutputStream()); |
| OutStream.write(b, 0, b.length); |
| OutStream.flush(); |
| OutStream.close(); |
| |
| /* Read the response */ |
| |
| int resp = conURL.getResponseCode (); |
| server.setFinished (true); |
| |
| if (resp != 200) |
| throw new RuntimeException ("Response code was not 200: " + resp); |
| } |
| } |