/*

 * 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 6518816

 * @run main/othervm B6518816

 */

import java.net.*;

import java.util.*;

import java.io.*;

import com.sun.net.httpserver.*;

import java.util.concurrent.Executors;

import java.util.concurrent.ExecutorService;

public class B6518816

{

    com.sun.net.httpserver.HttpServer httpServer;

    ExecutorService executorService;

    public static void main(String[] args)

    {

        new B6518816();

    }

    public B6518816()

    {

        try {

            startHttpServer();

            doClient();

        } catch (IOException ioe) {

            System.err.println(ioe);

        }

    }

    final static int MAX_CONNS = 10;

    final static int TEN_MB = 10 * 1024 * 1024;

    static boolean stopped = false;

    /*

     * we send approx 100MB and if more than 90MB is retained

     * then the bug is still there

     */

    void doClient() {

        try {

            InetSocketAddress address = httpServer.getAddress();

            List<HttpURLConnection> conns = new LinkedList<HttpURLConnection>();

            long totalmem1=0, totalmem2=0;

            System.gc();

            Runtime runtime = Runtime.getRuntime();

            totalmem1 = runtime.totalMemory() - runtime.freeMemory();

            System.out.println ("Sending " + (TEN_MB*MAX_CONNS/1024) + " KB");

            System.out.println ("At start: " + totalmem1/1024 + " KB");

            byte [] buf = new byte [TEN_MB];

            for (int j=0; j<MAX_CONNS; j++) {

                URL url = new URL("http://localhost:"+address.getPort()+"/test/");

                HttpURLConnection uc = (HttpURLConnection)url.openConnection();

                uc.setDoOutput (true);

                OutputStream os = uc.getOutputStream ();

                os.write (buf);

                InputStream is = uc.getInputStream();

                int resp = uc.getResponseCode();

                if (resp != 200) {

                    throw new RuntimeException("Failed: Part 1, Response code is not 200");

                }

                while  (is.read() != -1) ;

                is.close();

                conns.add (uc);

            }

            httpServer.stop(1);

            httpServer = null;

            executorService.shutdown();

            executorService = null;

            stopped = true;

            buf = null;

            System.runFinalization();

            try {Thread.sleep (1000); } catch (InterruptedException e){}

            System.gc();

            try {Thread.sleep (1000); } catch (InterruptedException e){}

            System.gc();

            totalmem2 = runtime.totalMemory() - runtime.freeMemory();

            System.out.println ("At end: " + totalmem2/1024 + " KB");

            long diff = (totalmem2 - totalmem1) ;;

            System.out.println ("Diff " + diff/1024 + " kbytes ");

            if (diff > (TEN_MB*MAX_CONNS*0.9)) {

                /* if >= 90 MB retained, then problem still there */

                throw new RuntimeException ("Excessive memory retained");

            }

        } catch (IOException e) {

            e.printStackTrace();

            throw new RuntimeException ("IOException");

        } finally {

            if (!stopped) {

                httpServer.stop(1);

                executorService.shutdown();

            }

        }

    }

    /**

     * Http Server

     */

    public void startHttpServer() throws IOException {

        httpServer = com.sun.net.httpserver.HttpServer.create(new InetSocketAddress(0), 0);

        // create HttpServer context for Part 1.

        HttpContext ctx = httpServer.createContext("/test/", new MyHandler());

        executorService = Executors.newCachedThreadPool();

        httpServer.setExecutor(executorService);

        httpServer.start();

    }

    class MyHandler implements HttpHandler {

        public void handle(HttpExchange t) throws IOException {

            InputStream is = t.getRequestBody();

            Headers reqHeaders = t.getRequestHeaders();

            Headers resHeaders = t.getResponseHeaders();

            byte[] buf = new byte [16 * 1024];

            while (is.read (buf) != -1) ;

            t.sendResponseHeaders(200, -1);

            t.close();

        }

    }

}