/*

 * 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());

        } finally {

            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) { }

            }

        }

    }