/*

 * Copyright 2008 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 6756771

 * @summary  com.sun.net.httpserver.HttpServer should handle POSTs larger than 2Gig

 */

import java.io.InputStream;

import java.io.IOException;

import java.io.OutputStream;

import java.io.PrintStream;

import java.net.InetSocketAddress;

import java.net.HttpURLConnection;

import java.net.URL;

import java.net.Socket;

import java.util.logging.*;

import com.sun.net.httpserver.HttpExchange;

import com.sun.net.httpserver.HttpHandler;

import com.sun.net.httpserver.HttpServer;

public class FixedLengthInputStream

{

    static final long POST_SIZE = 4L * 1024L * 1024L * 1024L; // 4Gig

    /* Remove when CR 6755625 is fixed */

    static final String requestHeaders =  ((new StringBuilder())

        .append("POST /flis/ HTTP/1.1\r\n")

        .append("User-Agent: Java/1.7.0\r\n")

        .append("Host: localhost\r\n")

        .append("Accept: text/html, image/gif, image/jpeg,")

        .append(        " *; q=.2, */*; q=.2\r\n")

        .append("Content-Length: 4294967296\r\n\r\n")).toString();

    void test(String[] args) throws IOException {

        HttpServer httpServer = startHttpServer();

        int port = httpServer.getAddress().getPort();

        try {

          /* Uncomment & when CR 6755625 is fixed, remove socket code

            URL url = new URL("http://localhost:" + port + "/flis/");

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

            uc.setDoOutput(true);

            uc.setRequestMethod("POST");

            uc.setFixedLengthStreamingMode(POST_SIZE);

            OutputStream os = uc.getOutputStream();

          */

            Socket socket = new Socket("localhost", port);

            OutputStream os = socket.getOutputStream();

            PrintStream ps = new PrintStream(os);

            debug("Request: " + requestHeaders);

            ps.print(requestHeaders);

            ps.flush();

            /* create a 32K byte array with data to POST */

            int thirtyTwoK = 32 * 1024;

            byte[] ba = new byte[thirtyTwoK];

            for (int i =0; i<thirtyTwoK; i++)

                ba[i] = (byte)i;

            long times = POST_SIZE / thirtyTwoK;

            for (int i=0; i<times; i++) {

                os.write(ba);

            }

          /* Uncomment & when CR 6755625 is fixed, remove socket code

            os.close();

            InputStream is = uc.getInputStream();

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

            is.close();

           */

           InputStream is = socket.getInputStream();

           is.read();

           socket.close();

           pass();

        } finally {

            httpServer.stop(0);

        }

    }

    /**

     * Http Server

     */

    HttpServer startHttpServer() throws IOException {

        if (debug) {

            Logger logger =

            Logger.getLogger("com.sun.net.httpserver");

            Handler outHandler = new StreamHandler(System.out,

                                     new SimpleFormatter());

            outHandler.setLevel(Level.FINEST);

            logger.setLevel(Level.FINEST);

            logger.addHandler(outHandler);

        }

        HttpServer httpServer = HttpServer.create(new InetSocketAddress(0), 0);

        httpServer.createContext("/flis/", new MyHandler(POST_SIZE));

        httpServer.start();

        return httpServer;

    }

    class MyHandler implements HttpHandler {

        static final int BUFFER_SIZE = 32 * 1024;

        long expected;

        MyHandler(long expected){

            this.expected = expected;

        }

        @Override

        public void handle(HttpExchange t) throws IOException {

            InputStream is = t.getRequestBody();

            byte[] ba = new byte[BUFFER_SIZE];

            int read;

            long count = 0L;

            while((read = is.read(ba)) != -1) {

                count += read;

            }

            is.close();

            check(count == expected, "Expected: " + expected + ", received "

                    + count);

            debug("Received " + count + " bytes");

            t.sendResponseHeaders(200, -1);

            t.close();

        }

    }

         //--------------------- Infrastructure ---------------------------

    boolean debug = true;

    volatile int passed = 0, failed = 0;

    void pass() {passed++;}

    void fail() {failed++; Thread.dumpStack();}

    void fail(String msg) {System.err.println(msg); fail();}

    void unexpected(Throwable t) {failed++; t.printStackTrace();}

    void check(boolean cond) {if (cond) pass(); else fail();}

    void check(boolean cond, String failMessage) {if (cond) pass(); else fail(failMessage);}

    void debug(String message) {if(debug) { System.out.println(message); }  }

    public static void main(String[] args) throws Throwable {

        Class<?> k = new Object(){}.getClass().getEnclosingClass();

        try {k.getMethod("instanceMain",String[].class)

                .invoke( k.newInstance(), (Object) args);}

        catch (Throwable e) {throw e.getCause();}}

    public void instanceMain(String[] args) throws Throwable {

        try {test(args);} catch (Throwable t) {unexpected(t);}

        System.out.printf("%nPassed = %d, failed = %d%n%n", passed, failed);

        if (failed > 0) throw new AssertionError("Some tests failed");}

}