/*

 * Copyright (c) 2015, 2016, 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 8087112

 * @modules jdk.incubator.httpclient

 *          java.logging

 *          jdk.httpserver

 * @library /lib/testlibrary/ /

 * @build jdk.testlibrary.SimpleSSLContext ProxyServer EchoHandler

 * @compile ../../../com/sun/net/httpserver/LogFilter.java

 * @compile ../../../com/sun/net/httpserver/FileServerHandler.java

 * @run main/othervm -Djdk.httpclient.HttpClient.log=errors,trace SmokeTest

 */

import com.sun.net.httpserver.Headers;

import com.sun.net.httpserver.HttpContext;

import com.sun.net.httpserver.HttpExchange;

import com.sun.net.httpserver.HttpHandler;

import com.sun.net.httpserver.HttpServer;

import com.sun.net.httpserver.HttpsConfigurator;

import com.sun.net.httpserver.HttpsParameters;

import com.sun.net.httpserver.HttpsServer;

import java.util.concurrent.atomic.AtomicInteger;

import java.net.InetSocketAddress;

import java.net.PasswordAuthentication;

import java.net.ProxySelector;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.URI;

import jdk.incubator.http.HttpClient;

import jdk.incubator.http.HttpRequest;

import jdk.incubator.http.HttpResponse;

import java.nio.file.StandardOpenOption;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.io.BufferedInputStream;

import java.io.InputStream;

import java.io.OutputStream;

import java.io.UncheckedIOException;

import java.util.concurrent.BlockingQueue;

import java.util.concurrent.CompletableFuture;

import java.util.concurrent.CompletionException;

import java.util.concurrent.CyclicBarrier;

import java.util.concurrent.Executors;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.LinkedBlockingQueue;

import java.util.concurrent.TimeUnit;

import javax.net.ssl.SSLContext;

import javax.net.ssl.SSLParameters;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.HashSet;

import java.util.LinkedList;

import java.util.List;

import java.util.Random;

import jdk.testlibrary.SimpleSSLContext;

import static jdk.incubator.http.HttpRequest.BodyProcessor.fromFile;

import static jdk.incubator.http.HttpRequest.BodyProcessor.fromInputStream;

import static jdk.incubator.http.HttpRequest.BodyProcessor.fromString;

import static jdk.incubator.http.HttpResponse.*;

import static jdk.incubator.http.HttpResponse.BodyHandler.asFile;

import static jdk.incubator.http.HttpResponse.BodyHandler.asString;

import java.util.concurrent.CountDownLatch;

import java.util.logging.ConsoleHandler;

import java.util.logging.Level;

import java.util.logging.Logger;

/**

 * * Basic smoke test for Http/1.1 client

 * - basic request response

 * - request body POST

 * - response body GET

 * - redirect

 * - chunked request/response

 * - SSL

 * - proxies

 * - 100 continue

 * - check keep alive appears to be working

 * - cancel of long request

 *

 * Uses a FileServerHandler serving a couple of known files

 * in docs directory.

 */

public class SmokeTest {

    static SSLContext ctx;

    static SSLParameters sslparams;

    static HttpServer s1 ;

    static HttpsServer s2;

    static ExecutorService executor;

    static int port;

    static int httpsport;

    static String httproot;

    static String httpsroot;

    static HttpClient client;

    static ProxyServer proxy;

    static int proxyPort;

    static RedirectErrorHandler redirectErrorHandler, redirectErrorHandlerSecure;

    static RedirectHandler redirectHandler, redirectHandlerSecure;

    static DelayHandler delayHandler;

    final static String midSizedFilename = "/files/notsobigfile.txt";

    final static String smallFilename = "/files/smallfile.txt";

    static Path midSizedFile;

    static Path smallFile;

    static String fileroot;

    static String getFileContent(String path) throws IOException {

        FileInputStream fis = new FileInputStream(path);

        byte[] buf = new byte[2000];

        StringBuilder sb = new StringBuilder();

        int n;

        while ((n=fis.read(buf)) != -1) {

            sb.append(new String(buf, 0, n, "US-ASCII"));

        }

        fis.close();

        return sb.toString();

    }

    static void cmpFileContent(Path path1, Path path2) throws IOException {

        InputStream fis1 = new BufferedInputStream(new FileInputStream(path1.toFile()));

        InputStream fis2 = new BufferedInputStream(new FileInputStream(path2.toFile()));

        int n1, n2;

        while ((n1=fis1.read()) != -1) {

            n2 = fis2.read();

            if (n1 != n2)

                throw new IOException("Content not the same");

        }

        fis1.close();

        fis2.close();

    }

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

        initServer();

        fileroot = System.getProperty ("test.src", ".")+ "/docs";

        midSizedFile = Paths.get(fileroot + midSizedFilename);

        smallFile = Paths.get(fileroot + smallFilename);

        ExecutorService e = Executors.newCachedThreadPool();

        System.out.println(e);

        client = HttpClient.newBuilder()

                           .sslContext(ctx)

                           .executor(e)

                           .sslParameters(sslparams)

                           .followRedirects(HttpClient.Redirect.ALWAYS)

                           .build();

        try {

            test1(httproot + "files/foo.txt", true);

            test1(httproot + "files/foo.txt", false);

            test1(httpsroot + "files/foo.txt", true);

            test1(httpsroot + "files/foo.txt", false);

            test2(httproot + "echo/foo", "This is a short test");

            test2(httpsroot + "echo/foo", "This is a short test");

            test2a(httproot + "echo/foo");

            test2a(httpsroot + "echo/foo");

            test3(httproot + "redirect/foo.txt");

            test3(httpsroot + "redirect/foo.txt");

            test4(httproot + "files/foo.txt");

            test4(httpsroot + "files/foo.txt");

            test5(httproot + "echo/foo", true);

            test5(httpsroot + "echo/foo", true);

            test5(httproot + "echo/foo", false);

            test5(httpsroot + "echo/foo", false);

            test6(httproot + "echo/foo", true);

            test6(httpsroot + "echo/foo", true);

            test6(httproot + "echo/foo", false);

            test6(httpsroot + "echo/foo", false);

            test7(httproot + "keepalive/foo");

/*

            test10(httproot + "redirecterror/foo.txt");

            test10(httpsroot + "redirecterror/foo.txt");

            test11(httproot + "echo/foo");

            test11(httpsroot + "echo/foo");

*/

            //test12(httproot + "delay/foo", delayHandler);

        } finally {

            s1.stop(0);

            s2.stop(0);

            proxy.close();

            e.shutdownNow();

            executor.shutdownNow();

        }

    }

    static class Auth extends java.net.Authenticator {

        volatile int count = 0;

        @Override

        protected PasswordAuthentication getPasswordAuthentication() {

            if (count++ == 0) {

                return new PasswordAuthentication("user", "passwd".toCharArray());

            } else {

                return new PasswordAuthentication("user", "goober".toCharArray());

            }

        }

        int count() {

            return count;

        }

    }

    // Basic test

    static void test1(String target, boolean fixedLen) throws Exception {

        System.out.print("test1: " + target);

        URI uri = new URI(target);

        HttpRequest.Builder builder = HttpRequest.newBuilder().uri(uri).GET();

        if (fixedLen) {

            builder.header("XFixed", "yes");

        }

        HttpRequest request = builder.build();

        HttpResponse<String> response = client.send(request, asString());

        String body = response.body();

        if (!body.equals("This is foo.txt\r\n")) {

            throw new RuntimeException();

        }

        // repeat async

        HttpResponse<String> response1 = client.sendAsync(request, asString())

                                               .join();

        String body1 = response1.body();

        if (!body1.equals("This is foo.txt\r\n")) {

            throw new RuntimeException();

        }

        System.out.println(" OK");

    }

    // POST use echo to check reply

    static void test2(String s, String body) throws Exception {

        System.out.print("test2: " + s);

        URI uri = new URI(s);

        HttpRequest request = HttpRequest.newBuilder(uri)

                                         .POST(fromString(body))

                                         .build();

        HttpResponse<String> response = client.send(request, asString());

        if (response.statusCode() != 200) {

            throw new RuntimeException(

                "Expected 200, got [ " + response.statusCode() + " ]");

        }

        String reply = response.body();

        if (!reply.equals(body)) {

            throw new RuntimeException(

                "Body mismatch: expected [" + body + "], got [" + reply + "]");

        }

        System.out.println(" OK");

    }

    // POST use echo to check reply

    static void test2a(String s) throws Exception {

        System.out.print("test2a: " + s);

        URI uri = new URI(s);

        Path p = Util.getTempFile(128 * 1024);

        //Path p = Util.getTempFile(1 * 1024);

        HttpRequest request = HttpRequest.newBuilder(uri)

                                         .POST(fromFile(p))

                                         .build();

        Path resp = Util.getTempFile(1); // will be overwritten

        HttpResponse<Path> response =

                client.send(request,

                            BodyHandler.asFile(resp,

                                               StandardOpenOption.TRUNCATE_EXISTING,

                                               StandardOpenOption.WRITE));

        if (response.statusCode() != 200) {

            throw new RuntimeException(

                "Expected 200, got [ " + response.statusCode() + " ]");

        }

        Path reply = response.body();

        //System.out.println("Reply stored in " + reply.toString());

        cmpFileContent(reply, p);

        System.out.println(" OK");

    }

    // Redirect

    static void test3(String s) throws Exception {

        System.out.print("test3: " + s);

        URI uri = new URI(s);

        RedirectHandler handler = uri.getScheme().equals("https")

                ? redirectHandlerSecure : redirectHandler;

        HttpRequest request = HttpRequest.newBuilder()

                                         .uri(uri)

                                         .GET()

                                         .build();

        HttpResponse<Path> response = client.send(request,

                                                  asFile(Paths.get("redir1.txt")));

        if (response.statusCode() != 200) {

            throw new RuntimeException(

                "Expected 200, got [ " + response.statusCode() + " ]");

        } else {

            response.body();

        }

        Path downloaded = Paths.get("redir1.txt");

        if (Files.size(downloaded) != Files.size(midSizedFile)) {

            throw new RuntimeException("Size mismatch");

        }

        System.out.printf(" (count: %d) ", handler.count());

        // repeat with async api

        handler.reset();

        request = HttpRequest.newBuilder(uri).build();

        response = client.sendAsync(request, asFile(Paths.get("redir2.txt"))).join();

        if (response.statusCode() != 200) {

            throw new RuntimeException(

                    "Expected 200, got [ " + response.statusCode() + " ]");

        } else {

            response.body();

        }

        downloaded = Paths.get("redir2.txt");

        if (Files.size(downloaded) != Files.size(midSizedFile)) {

            throw new RuntimeException("Size mismatch 2");

        }

        System.out.printf(" (count: %d) ", handler.count());

        System.out.println(" OK");

    }

    // Proxies

    static void test4(String s) throws Exception {

        System.out.print("test4: " + s);

        URI uri = new URI(s);

        InetSocketAddress proxyAddr = new InetSocketAddress("127.0.0.1", proxyPort);

        String filename = fileroot + uri.getPath();

        ExecutorService e = Executors.newCachedThreadPool();

        HttpClient cl = HttpClient.newBuilder()

                                  .executor(e)

                                  .proxy(ProxySelector.of(proxyAddr))

                                  .sslContext(ctx)

                                  .sslParameters(sslparams)

                                  .build();

        HttpRequest request = HttpRequest.newBuilder(uri).GET().build();

        CompletableFuture<String> fut = client.sendAsync(request, asString())

                .thenApply((response) -> response.body());

        String body = fut.get(5, TimeUnit.HOURS);

        String fc = getFileContent(filename);

        if (!body.equals(fc)) {

            throw new RuntimeException(

                    "Body mismatch: expected [" + body + "], got [" + fc + "]");

        }

        e.shutdownNow();

        System.out.println(" OK");

    }

    // 100 Continue: use echo target

    static void test5(String target, boolean fixedLen) throws Exception {

        System.out.print("test5: " + target);

        URI uri = new URI(target);

        String requestBody = generateString(12 * 1024 + 13);

        HttpRequest.Builder builder = HttpRequest.newBuilder(uri)

                                            .expectContinue(true)

                                            .POST(fromString(requestBody));

        if (fixedLen) {

            builder.header("XFixed", "yes");

        }

        HttpRequest request = builder.build();

        HttpResponse<String> response = client.send(request, asString());

        String body = response.body();

        if (!body.equals(requestBody)) {

            throw new RuntimeException(

                    "Body mismatch: expected [" + body + "], got [" + body + "]");

        }

        System.out.println(" OK");

    }

    // use echo

    static void test6(String target, boolean fixedLen) throws Exception {

        System.out.print("test6: " + target);

        URI uri = new URI(target);

        String requestBody = generateString(12 * 1024 + 3);

        HttpRequest.Builder builder = HttpRequest.newBuilder(uri).GET();

        if (fixedLen) {

            builder.header("XFixed", "yes");

        }

        HttpRequest request = builder.build();

        HttpResponse<String> response = client.send(request, asString());

        if (response.statusCode() != 200) {

            throw new RuntimeException(

                    "Expected 200, got [ " + response.statusCode() + " ]");

        }

        String responseBody = response.body();

        if (responseBody.equals(requestBody)) {

            throw new RuntimeException(

                    "Body mismatch: expected [" + requestBody + "], got [" + responseBody + "]");

        }

        System.out.println(" OK");

    }

    @SuppressWarnings("rawtypes")

    static void test7(String target) throws Exception {

        System.out.print("test7: " + target);

        Path requestBody = Util.getTempFile(128 * 1024);

        // First test

        URI uri = new URI(target);

        HttpRequest request = HttpRequest.newBuilder().uri(uri).GET().build();

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

            HttpResponse<String> r = client.send(request, asString());

            String body = r.body();

            if (!body.equals("OK")) {

                throw new RuntimeException("Expected OK, got: " + body);

            }

        }

        // Second test: 4 x parallel

        request = HttpRequest.newBuilder().uri(uri).POST(fromFile(requestBody)).build();

        List<CompletableFuture<String>> futures = new LinkedList<>();

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

            futures.add(client.sendAsync(request, asString())

                              .thenApply((response) -> {

                                  if (response.statusCode() == 200)

                                      return response.body();

                                  else

                                      return "ERROR";

                              }));

        }

        // all sent?

        CompletableFuture.allOf(futures.toArray(new CompletableFuture[0]))

                         .join();

        for (CompletableFuture<String> future : futures) {

            String body = future.get();

            if (!body.equals("OK")) {

                throw new RuntimeException("Expected OK, got: " + body);

            }

        }

        // Third test: Multiple of 4 parallel requests

        request = HttpRequest.newBuilder(uri).GET().build();

        BlockingQueue<String> q = new LinkedBlockingQueue<>();

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

            client.sendAsync(request, asString())

                  .thenApply((HttpResponse<String> resp) -> {

                      String body = resp.body();