| /* |
| * Copyright (c) 2002, 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 4636628 |
| * @summary HttpURLConnection duplicates HTTP GET requests when used with multiple threads |
| */ |
| |
| /* |
| * This tests keep-alive behavior using chunkedinputstreams |
| * It checks that keep-alive connections are used and also |
| * that requests are not being repeated (due to errors) |
| * |
| * It also checks that the keepalive connections are closed eventually |
| * because the test will not terminate if the connections |
| * are not closed by the keep-alive timer. |
| */ |
| |
| import java.net.*; |
| import java.io.*; |
| |
| public class MultiThreadTest extends Thread { |
| |
| /* |
| * Is debugging enabled - start with -d to enable. |
| */ |
| static boolean debug = false; |
| |
| static Object threadlock = new Object (); |
| static int threadCounter = 0; |
| |
| static Object getLock() { return threadlock; } |
| |
| static void debug(String msg) { |
| if (debug) |
| System.out.println(msg); |
| } |
| |
| static int reqnum = 0; |
| |
| void doRequest(String uri) throws Exception { |
| URL url = new URL(uri + "?foo="+reqnum); |
| reqnum ++; |
| HttpURLConnection http = (HttpURLConnection)url.openConnection(); |
| |
| InputStream in = http.getInputStream(); |
| byte b[] = new byte[100]; |
| int total = 0; |
| int n; |
| do { |
| n = in.read(b); |
| if (n > 0) total += n; |
| } while (n > 0); |
| debug ("client: read " + total + " bytes"); |
| in.close(); |
| http.disconnect(); |
| } |
| |
| String uri; |
| byte[] b; |
| int requests; |
| |
| MultiThreadTest(int port, int requests) throws Exception { |
| uri = "http://localhost:" + |
| port + "/foo.html"; |
| |
| b = new byte [256]; |
| this.requests = requests; |
| |
| synchronized (threadlock) { |
| threadCounter ++; |
| } |
| } |
| |
| public void run () { |
| try { |
| for (int i=0; i<requests; i++) { |
| doRequest (uri); |
| } |
| } catch (Exception e) { |
| throw new RuntimeException (e.getMessage()); |
| } |
| synchronized (threadlock) { |
| threadCounter --; |
| if (threadCounter == 0) { |
| threadlock.notifyAll(); |
| } |
| } |
| } |
| |
| static int threads=5; |
| |
| public static void main(String args[]) throws Exception { |
| |
| int x = 0, arg_len = args.length; |
| int requests = 20; |
| |
| if (arg_len > 0 && args[0].equals("-d")) { |
| debug = true; |
| x = 1; |
| arg_len --; |
| } |
| if (arg_len > 0) { |
| threads = Integer.parseInt (args[x]); |
| requests = Integer.parseInt (args[x+1]); |
| } |
| |
| /* start the server */ |
| ServerSocket ss = new ServerSocket(0); |
| Server svr = new Server(ss); |
| svr.start(); |
| |
| Object lock = MultiThreadTest.getLock(); |
| synchronized (lock) { |
| for (int i=0; i<threads; i++) { |
| MultiThreadTest t = new MultiThreadTest(ss.getLocalPort(), requests); |
| t.start (); |
| } |
| try { |
| lock.wait(); |
| } catch (InterruptedException e) {} |
| } |
| |
| // shutdown server - we're done. |
| svr.shutdown(); |
| |
| int cnt = svr.connectionCount(); |
| MultiThreadTest.debug("Connections = " + cnt); |
| int reqs = Worker.getRequests (); |
| MultiThreadTest.debug("Requests = " + reqs); |
| System.out.println ("Connection count = " + cnt + " Request count = " + reqs); |
| if (cnt > threads) { // could be less |
| throw new RuntimeException ("Expected "+threads + " connections: used " +cnt); |
| } |
| if (reqs != threads*requests) { |
| throw new RuntimeException ("Expected "+ threads*requests+ " requests: got " +reqs); |
| } |
| } |
| } |
| |
| /* |
| * Server thread to accept connection and create worker threads |
| * to service each connection. |
| */ |
| class Server extends Thread { |
| ServerSocket ss; |
| int connectionCount; |
| boolean shutdown = false; |
| |
| Server(ServerSocket ss) { |
| this.ss = ss; |
| } |
| |
| public synchronized int connectionCount() { |
| return connectionCount; |
| } |
| |
| public synchronized void shutdown() { |
| shutdown = true; |
| } |
| |
| public void run() { |
| try { |
| ss.setSoTimeout(2000); |
| |
| for (;;) { |
| Socket s; |
| try { |
| MultiThreadTest.debug("server: calling accept."); |
| s = ss.accept(); |
| MultiThreadTest.debug("server: return accept."); |
| } catch (SocketTimeoutException te) { |
| MultiThreadTest.debug("server: STE"); |
| synchronized (this) { |
| if (shutdown) { |
| MultiThreadTest.debug("server: Shuting down."); |
| return; |
| } |
| } |
| continue; |
| } |
| |
| int id; |
| synchronized (this) { |
| id = connectionCount++; |
| } |
| |
| Worker w = new Worker(s, id); |
| w.start(); |
| MultiThreadTest.debug("server: Started worker " + id); |
| } |
| |
| } catch (Exception e) { |
| e.printStackTrace(); |
| } finally { |
| try { |
| ss.close(); |
| } catch (Exception e) { } |
| } |
| } |
| } |
| |
| /* |
| * Worker thread to service single connection - can service |
| * multiple http requests on same connection. |
| */ |
| class Worker extends Thread { |
| Socket s; |
| int id; |
| |
| Worker(Socket s, int id) { |
| this.s = s; |
| this.id = id; |
| } |
| |
| static int requests = 0; |
| static Object rlock = new Object(); |
| |
| public static int getRequests () { |
| synchronized (rlock) { |
| return requests; |
| } |
| } |
| public static void incRequests () { |
| synchronized (rlock) { |
| requests++; |
| } |
| } |
| |
| int readUntil (InputStream in, char[] seq) throws IOException { |
| int i=0, count=0; |
| while (true) { |
| int c = in.read(); |
| if (c == -1) |
| return -1; |
| count++; |
| if (c == seq[i]) { |
| i++; |
| if (i == seq.length) |
| return count; |
| continue; |
| } else { |
| i = 0; |
| } |
| } |
| } |
| |
| public void run() { |
| try { |
| int max = 400; |
| byte b[] = new byte[1000]; |
| InputStream in = new BufferedInputStream (s.getInputStream()); |
| // response to client |
| PrintStream out = new PrintStream( |
| new BufferedOutputStream( |
| s.getOutputStream() )); |
| |
| for (;;) { |
| |
| // read entire request from client |
| int n=0; |
| |
| n = readUntil (in, new char[] {'\r','\n', '\r','\n'}); |
| |
| if (n <= 0) { |
| MultiThreadTest.debug("worker: " + id + ": Shutdown"); |
| s.close(); |
| return; |
| } |
| |
| MultiThreadTest.debug("worker " + id + |
| ": Read request from client " + |
| "(" + n + " bytes)."); |
| |
| incRequests(); |
| out.print("HTTP/1.1 200 OK\r\n"); |
| out.print("Transfer-Encoding: chunked\r\n"); |
| out.print("Content-Type: text/html\r\n"); |
| out.print("Connection: Keep-Alive\r\n"); |
| out.print ("Keep-Alive: timeout=15, max="+max+"\r\n"); |
| out.print("\r\n"); |
| out.print ("6\r\nHello \r\n"); |
| out.print ("5\r\nWorld\r\n"); |
| out.print ("0\r\n\r\n"); |
| out.flush(); |
| |
| if (--max == 0) { |
| s.close(); |
| return; |
| } |
| } |
| |
| } catch (Exception e) { |
| e.printStackTrace(); |
| } finally { |
| try { |
| s.close(); |
| } catch (Exception e) { } |
| } |
| } |
| } |