test/java/net/PortUnreachableException/Test.java

/*
 * Copyright (c) 2001, 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 4413768
 * @summary Checking that PortUnreachableException is thrown when
 *          ICMP Port Unreachable is received.
 */
import java.net.*;
import java.util.Properties;

public class Test {

    /*
     * Return an available port
     */
    int getPort() throws Exception {
        DatagramSocket s = new DatagramSocket(0);
        int port = s.getLocalPort();
        s.close();
        return port;
    }

    /*
     * Perform test by sending to remote_host:port
     * sendOnly => send datagram to host and expect PUE on subsequent
     *             send
     * !sendOnly => send datagram to host and expect PUE on subsequent
     *              send or receive.
     */
    void doTest(String remote_host, int port, boolean sendOnly) throws Exception {

        System.out.println("***");
        System.out.println("Test Description:");
        System.out.println("    DatagramSocket.connect");
        System.out.println("    Loop: DatagramSocket.send");
        if (!sendOnly) {
            System.out.println("          DatagramSocket.receive");
        }
        System.out.println("");
        System.out.println("Test Run:");

        InetAddress ia = InetAddress.getByName(remote_host);
        DatagramSocket s = new DatagramSocket(0);
        s.setSoTimeout(1000);
        s.connect(ia, port);

        byte[] b = "Hello".getBytes();
        DatagramPacket p1 = new DatagramPacket(b, b.length, ia, port);

        DatagramPacket p2 = new DatagramPacket(b, b.length);

        int i = 0;
        boolean gotPUE = false;

        do {

            System.out.println("Sending datagram to unreachable port...");
            try {
                s.send(p1);
            } catch (PortUnreachableException e) {
                System.out.println("DatagramSocket.send threw PUE");
                gotPUE = true;
            }

            if (!gotPUE) {
                Thread.currentThread().sleep(1000);
            }

            if (!sendOnly && !gotPUE) {
                System.out.println("DatagramSocket.receive...");
                try {
                    s.receive(p2);
                } catch (PortUnreachableException e) {
                    System.out.println("DatagramSocket.receive threw PUE");
                    gotPUE = true;
                } catch (SocketTimeoutException e) {
                    System.out.println("DatagramSocket.receive timed out - no PUE");
                }
            }

            i++;
        } while (i < 10 && !gotPUE);

        if (!gotPUE) {
            System.out.println("DatagramSocket.{send,receive} didn't throw " +
                "PortUnreachableException - passing anyway!");
        } else {
            System.out.println("    Test passed.");
        }
        System.out.println("");
    }

    /*
     * Perform tests via remote_host.
     */
    Test(String remote_host) throws Exception {

        int port = getPort();

        doTest(remote_host, port, true);
        doTest(remote_host, port, false);
    }

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

        String remote_host = "localhost";
        if (args.length > 0) {
            remote_host = args[0];
        }

        new Test(remote_host);
    }
}