/*

 * Copyright (c) 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 8134577

 * @summary Test the internal NameService implementation which is enabled via

 *          the system property jdk.net.hosts.file. This property specifies

 *          a file name that contains address host mappings, similar to those in

 *          /etc/hosts file.

 * @run main/othervm  InternalNameServiceTest

 */

import java.io.BufferedWriter;

import java.io.FileWriter;

import java.io.PrintWriter;

import java.net.InetAddress;

import java.net.UnknownHostException;

import java.util.Arrays;

public class InternalNameServiceTest {

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

        String hostsFileName = System.getProperty("test.src", ".") + "/TestHosts";

        System.setProperty("jdk.net.hosts.file", hostsFileName);

        System.setProperty("sun.net.inetaddr.ttl", "0");

        testHostToIPAddressMappings(hostsFileName);

        testIpAddressToHostNameMappings(hostsFileName);

    }

    private static void testHostToIPAddressMappings(String hostsFileName)

            throws Exception, UnknownHostException {

        System.out.println(" TEST HOST TO  IP ADDRESS MAPPINGS ");

        InetAddress testAddress;

        byte[] retrievedIpAddr;

        byte[] expectedIpAddr1 = { 1, 2, 3, 4 };

        byte[] expectedIpAddr2 = { 5, 6, 7, 8 };

        byte[] expectedIpAddrIpv6_1 = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1};

        // hosts file with

        // # test hosts file for internal NameService

        // 1.2.3.4 host.sample-domain

        // 5.6.7.8 host1.sample-domain

        // 1.2.3.4 host2.sample-domain  # this is a comment

        // host3.sample-domain # no ip address

        //  host4.sample-domain # space as ip address

        //   host5.sample-domain # double space as ip address

        // add comment to hosts file

        addMappingToHostsFile("test hosts file for internal NameService ", "#", hostsFileName,

                false);

        addMappingToHostsFile("host.sample-domain", "1.2.3.4", hostsFileName,

                true);

        testAddress = InetAddress.getByName("host.sample-domain");

        retrievedIpAddr = testAddress.getAddress();

        if (!Arrays.equals(retrievedIpAddr, expectedIpAddr1)) {

            throw new RuntimeException(

                    "retrievedIpAddr not equal to expectedipAddr");

        }

        addMappingToHostsFile("host1.sample-domain", "5.6.7.8", hostsFileName,

                true);

        addMappingToHostsFile("host2.sample-domain", "1.2.3.4", hostsFileName,

                true);

        testAddress = InetAddress.getByName("host1.sample-domain");

        retrievedIpAddr = testAddress.getAddress();

        if (!Arrays.equals(retrievedIpAddr, expectedIpAddr2)) {

            throw new RuntimeException(

                    "retrievedIpAddr not equal to expectedIpAddr");

        }

        testAddress = InetAddress.getByName("host2.sample-domain");

        retrievedIpAddr = testAddress.getAddress();

        if (!Arrays.equals(retrievedIpAddr, expectedIpAddr1)) {

            throw new RuntimeException(

                    "retrievedIpAddr not equal to expectedIpAddr");

        }

        try {

            addMappingToHostsFile("host3.sample-domain", "", hostsFileName,

                    true);

            testAddress = InetAddress.getByName("host3.sample-domain");

            throw new RuntimeException(

                    "Expected UnknownHostException not thrown");

        } catch (UnknownHostException uhEx) {

            System.out.println("UnknownHostException as expected for host host3.sample-domain");

        }

        try {

            addMappingToHostsFile("host4.sample-domain", " ", hostsFileName,

                    true);

            testAddress = InetAddress.getByName("host4.sample-domain");

            throw new RuntimeException(

                    "Expected UnknownHostException not thrown");

        } catch (UnknownHostException uhEx) {

            System.out.println("UnknownHostException as expected for host host4.sample-domain");

        }

        try {

            addMappingToHostsFile("host5.sample-domain", "  ", hostsFileName,

                    true);

            testAddress = InetAddress.getByName("host4.sample-domain");

            throw new RuntimeException(

                    "Expected UnknownHostException not thrown");

        } catch (UnknownHostException uhEx) {

            System.out.println("UnknownHostException as expected for host host5.sample-domain");

        }

        // IPV6 tests

        // IPV6 tests

        addMappingToHostsFile("host-ipv6.sample-domain", "::1", hostsFileName,

                true);

        testAddress = InetAddress.getByName("host-ipv6.sample-domain");

        retrievedIpAddr = testAddress.getAddress();

        if (!Arrays.equals(retrievedIpAddr, expectedIpAddrIpv6_1)) {

            System.out.println("retrieved ipv6 addr == " + Arrays.toString(retrievedIpAddr));

            System.out.println("expected ipv6 addr == " + Arrays.toString(expectedIpAddrIpv6_1));

            throw new RuntimeException(

                    "retrieved IPV6 Addr not equal to expected IPV6 Addr");

        }

    }

    private static void testIpAddressToHostNameMappings(String hostsFileName)

            throws Exception {

        System.out.println(" TEST IP ADDRESS TO HOST MAPPINGS ");

        InetAddress testAddress;

        String retrievedHost;

        String expectedHost = "testHost.testDomain";

        byte[] testHostIpAddr = { 10, 2, 3, 4 };

        byte[] testHostIpAddr2 = { 10, 5, 6, 7 };

        byte[] testHostIpAddr3 = { 10, 8, 9, 10 };

        byte[] testHostIpAddr4 = { 10, 8, 9, 11 };

        // add comment to hosts file

        addMappingToHostsFile("test hosts file for internal NameService ", "#", hostsFileName,

                false);

        addMappingToHostsFile("testHost.testDomain", "10.2.3.4", hostsFileName,

                true);

        testAddress = InetAddress.getByAddress(testHostIpAddr);

        System.out.println("*******   testAddress == " + testAddress);

        retrievedHost = testAddress.getHostName();

        if (!expectedHost.equals(retrievedHost)) {

            throw new RuntimeException(

                    "retrieved host name not equal to expected host name");

        }

        addMappingToHostsFile("testHost.testDomain", "10.5.6.7", hostsFileName,

                true);

        testAddress = InetAddress.getByAddress(testHostIpAddr2);

        System.out.println("*******   testAddress == " + testAddress);

        retrievedHost = testAddress.getHostName();

        System.out.println("*******   retrievedHost == " + retrievedHost);

        if (!expectedHost.equals(retrievedHost)) {

            throw new RuntimeException("retrieved host name " + retrievedHost

                    + " not equal to expected host name" + expectedHost);

        }

        testAddress = InetAddress.getByAddress(testHostIpAddr4);

        System.out.println("*******   testAddress == " + testAddress);

        if ("10.8.9.11".equalsIgnoreCase(testAddress.getCanonicalHostName())) {

            System.out.println("addr = " + addrToString(testHostIpAddr4)

                    + "  resolve to a host address as expected");

        } else {

            System.out.println("addr = " + addrToString(testHostIpAddr4)

                    + " does not resolve as expected, testAddress == " + testAddress.getCanonicalHostName());

            throw new RuntimeException("problem with resolving "

                    + addrToString(testHostIpAddr4));

        }

        try {

            addMappingToHostsFile("", "10.8.9.10", hostsFileName, true);

            testAddress = InetAddress.getByAddress(testHostIpAddr3);

            System.out.println("*******   testAddress == " + testAddress);

            retrievedHost = testAddress.getCanonicalHostName();

        } catch (Throwable t) {

            throw new RuntimeException("problem with resolving "

                    + addrToString(testHostIpAddr3));

        }

    }

    private static String addrToString(byte addr[]) {

        return Byte.toString(addr[0]) + "." + Byte.toString(addr[1]) + "."

                + Byte.toString(addr[2]) + "." + Byte.toString(addr[3]);

    }

    private static void addMappingToHostsFile( String host,

                                               String addr,

                                               String hostsFileName,

                                               boolean append)

                                               throws Exception {

        String mapping = addr + " " + host;

        try (PrintWriter hfPWriter = new PrintWriter(new BufferedWriter(

                new FileWriter(hostsFileName, append)))) {

            hfPWriter.println(mapping);

        }

    }

}