/*

 * Copyright (c) 2013, 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 7100957

 * @modules jdk.httpserver

 * @summary Java doesn't correctly handle the SOCKS protocol when used over IPv6.

 * @run testng SocksIPv6Test

 */

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.net.Authenticator;

import java.net.InetSocketAddress;

import java.net.URL;

import java.net.Proxy;

import java.lang.Override;

import java.net.InetAddress;

import java.net.Inet6Address;

import java.net.ServerSocket;

import java.net.SocketException;

import java.net.NetworkInterface;

import java.net.UnknownHostException;

import java.util.Collections;

import java.util.List;

import com.sun.net.httpserver.*;

import java.io.BufferedWriter;

import org.testng.annotations.AfterClass;

import org.testng.annotations.BeforeClass;

import org.testng.annotations.Test;

import static org.testng.Assert.*;

public class SocksIPv6Test {

    private HttpServer server;

    private SocksServer socks;

    private String response = "Hello.";

    private static boolean shouldRun = false;

    @BeforeClass

    public void setUp() throws Exception {

        shouldRun = ensureInet6AddressFamily() && ensureIPv6OnLoopback();

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

        server.createContext("/", ex -> {

            ex.sendResponseHeaders(200, response.length());

            try (BufferedWriter writer = new BufferedWriter(

                    new OutputStreamWriter(ex.getResponseBody(), "UTF-8"))) {

                writer.write(response);

            }

            ex.close();

        });

        server.start();

        socks = new SocksServer(0, false);

        socks.addUser("user", "pass");

        socks.start();

        Authenticator.setDefault(new Authenticator() {

            @Override

            protected java.net.PasswordAuthentication getPasswordAuthentication() {

                return new java.net.PasswordAuthentication(

                        "user", "pass".toCharArray());

            }

        });

    }

    private boolean ensureIPv6OnLoopback() throws Exception {

        boolean ipv6 = false;

        List<NetworkInterface> nics = Collections.list(NetworkInterface.getNetworkInterfaces());

        for (NetworkInterface nic : nics) {

            if (!nic.isLoopback()) {

                continue;

            }

            List<InetAddress> addrs = Collections.list(nic.getInetAddresses());

            for (InetAddress addr : addrs) {

                if (addr instanceof Inet6Address) {

                    ipv6 = true;

                    break;

                }

            }

        }

        if (!ipv6)

            System.out.println("IPv6 is not enabled on loopback. Skipping test suite.");

        return ipv6;

    }

    private boolean ensureInet6AddressFamily() throws IOException {

        try (ServerSocket s = new ServerSocket()) {

            s.bind(new InetSocketAddress("::1", 0));

            return true;

        } catch (SocketException e) {

            System.out.println("Inet 6 address family is not available. Skipping test suite.");

        }

        return false;

    }

    @Test(groups = "unit")

    public void testSocksOverIPv6() throws Exception {

        if (!shouldRun) return;

        Proxy proxy = new Proxy(Proxy.Type.SOCKS, new InetSocketAddress("::1",

                socks.getPort()));

        URL url = new URL("http://[::1]:" + server.getAddress().getPort());

        java.net.URLConnection conn = url.openConnection(proxy);

        String actual = "";

        try (BufferedReader reader = new BufferedReader(

                new InputStreamReader(conn.getInputStream()))) {

            actual = reader.readLine();

        }

        assertEquals(actual, response);

    }

    @Test(groups = "unit")

    public void testSocksOverIPv6Hostname() throws Exception {

        if (!shouldRun) return;

        String ipv6Hostname = InetAddress.getByName("::1").getHostName();

        String ipv4Hostname = InetAddress.getByName("127.0.0.1").getHostName();

        if (ipv6Hostname.equals(InetAddress.getByName("::1").getHostAddress())) {

            System.out.println("Unable to get the hostname of the IPv6 loopback "

                    + "address. Skipping test case.");

            return;

        }

        if (ipv6Hostname.equals(ipv4Hostname)) {

            System.out.println("IPv6 and IPv4 loopback addresses map to the"

                    + " same hostname. Skipping test case.");

            return;

        }

        Proxy proxy = new Proxy(Proxy.Type.SOCKS, new InetSocketAddress(ipv6Hostname,

                socks.getPort()));

        URL url = new URL("http://" + ipv6Hostname + ":" + server.getAddress().getPort());

        java.net.URLConnection conn = url.openConnection(proxy);

        String actual = "";

        try (BufferedReader reader = new BufferedReader(

                new InputStreamReader(conn.getInputStream()))) {

            actual = reader.readLine();

        }

        assertEquals(actual, response);

    }

    @AfterClass

    public void tearDown() {

        if (server != null) {

            server.stop(1);

        }

        if (socks != null) {

            socks.terminate();

        }

    }

}