/*

 * Copyright (c) 2018, 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.

 */

import java.io.InputStream;

import java.io.OutputStream;

import java.net.InetSocketAddress;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.URI;

import java.net.SocketTimeoutException;

import java.time.Duration;

import javax.net.ssl.SSLServerSocketFactory;

import javax.net.ServerSocketFactory;

import javax.net.ssl.SSLContext;

import javax.net.ssl.SSLParameters;

import java.net.http.HttpClient;

import java.net.http.HttpRequest;

import java.net.http.HttpResponse;

import jdk.testlibrary.SimpleSSLContext;

/**

 * @test

 * @bug 8207966

 * @library /lib/testlibrary

 * @build jdk.testlibrary.SimpleSSLContext

 * @run main/othervm -Djdk.tls.acknowledgeCloseNotify=true UnknownBodyLengthTest plain false

 * @run main/othervm -Djdk.tls.acknowledgeCloseNotify=true UnknownBodyLengthTest SSL false

 * @run main/othervm -Djdk.tls.acknowledgeCloseNotify=true UnknownBodyLengthTest plain true

 * @run main/othervm -Djdk.tls.acknowledgeCloseNotify=true UnknownBodyLengthTest SSL true

 */

public class UnknownBodyLengthTest {

    static final byte[] BUF = new byte[32 * 10234 + 2];

    volatile SSLContext ctx;

    volatile ServerSocketFactory factory;

    volatile String clientURL;

    volatile int port;

    final ServerSocket ss;

    UnknownBodyLengthTest(boolean useSSL) throws Exception {

        ctx = new SimpleSSLContext().get();

        SSLContext.setDefault(ctx);

        factory = useSSL ? SSLServerSocketFactory.getDefault()

                         : ServerSocketFactory.getDefault();

        ss = factory.createServerSocket();

        ss.setReuseAddress(true);

        ss.bind(new InetSocketAddress("127.0.0.1", 0));

        System.out.println("ServerSocket = " + ss.getClass() + " " + ss);

        port = ss.getLocalPort();

        clientURL = (useSSL ? "https" : "http") + "://localhost:"

            + Integer.toString(port) + "/test";

    }

    static void fillBuf(byte[] buf) {

        for (int i=0; i<buf.length; i++)

            buf[i] = (byte)i;

    }

    static void checkBuf(byte[] buf) {

        if (buf.length != BUF.length)

            throw new RuntimeException("buffer lengths not the same");

        for (int i=0; i<buf.length; i++)

            if (buf[i] != BUF[i])

                throw new RuntimeException("error at position " + i);

    }

    void server(final boolean withContentLength) {

        fillBuf(BUF);

        try {

            Socket s = ss.accept();

            s.setTcpNoDelay(true);

            s.setSoLinger(true, 1);

            System.out.println("Accepted: "+s.getRemoteSocketAddress());

            System.out.println("Accepted: "+s);

            OutputStream os = s.getOutputStream();

            InputStream is = s.getInputStream();

            boolean done = false;

            byte[] buf = new byte[1024];

            String rsp = "";

            while (!done) {

                int c = is.read(buf);

                if (c < 0) break;

                String s1 = new String(buf, 0, c, "ISO-8859-1");

                rsp += s1;

                done = rsp.endsWith("!#!#");

            }

            String r = "HTTP/1.0 200 OK\r\nConnection: close\r\nContent-Type:" +

                       " text/xml; charset=UTF-8\r\n";

            os.write(r.getBytes());

            String chdr = "Content-Length: " + Integer.toString(BUF.length) +

                     "\r\n";

            System.out.println(chdr);

            if(withContentLength)

                os.write(chdr.getBytes());

            os.write("\r\n".getBytes());

            os.write(BUF);

            if (is.available() > 0)

                is.read(buf);

            os.flush();

            os.close();

            s.shutdownOutput();

            s.close();

        } catch(final Throwable t) {

          t.printStackTrace();

        } finally {

            try {ss.close(); } catch (Exception e) {}

        }

    }

    void client(boolean useSSL) throws Exception {

        SSLContext ctx = SSLContext.getDefault();

        HttpClient.Builder clientB = HttpClient.newBuilder()

            .version(HttpClient.Version.HTTP_2);

        if (useSSL) {

            clientB = clientB.sslContext(ctx)

                .sslParameters(ctx.getSupportedSSLParameters());

        }

        final HttpClient client = clientB.build();

        System.out.println("URL: " + clientURL);

        final HttpResponse<byte[]> response = client

            .send(HttpRequest

                .newBuilder(new URI(clientURL))

                .timeout(Duration.ofMillis(120_000))

                .POST(HttpRequest.BodyPublishers.ofString("body!#!#"))

                .build(), HttpResponse.BodyHandlers.ofByteArray());

        System.out.println("Received reply: " + response.statusCode());

        byte[] bb = response.body();

        checkBuf(bb);

    }

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

        boolean ssl = args[0].equals("SSL");

        boolean fixedlen = args[1].equals("true");

        UnknownBodyLengthTest test = new UnknownBodyLengthTest(ssl);

        test.run(ssl, fixedlen);

    }

    public void run(boolean ssl, boolean fixedlen) throws Exception {

        new Thread(()->server(fixedlen)).start();

        client(ssl);

    }

}