/*

 * Copyright (c) 2008, 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 6756771 6755625

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

    void test(String[] args) throws IOException {

        HttpServer httpServer = startHttpServer();

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

        try {

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

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

            }

            os.close();

            InputStream is = uc.getInputStream();

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

            is.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");}

}