/*
 * Copyright 2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

/*
 * @test
 * @bug 6505016
 * @summary Socket spec should clarify what getInetAddress/getPort/etc return after the Socket is closed
 */

import java.net.*;
import java.io.*;

public class TestAfterClose
{
    static int failCount;

    public static void main(String[] args) {
        try {
            DatagramSocket socket = new DatagramSocket();
            socket.connect(InetAddress.getLoopbackAddress(), 5001);
            test(socket, true);

            socket = new DatagramSocket();
            test(socket, false);

        } catch (IOException ioe) {
            ioe.printStackTrace();
        }

        if (failCount > 0)
            throw new RuntimeException("Failed: failcount = " + failCount);

    }

    static void test(DatagramSocket socket, boolean connected) throws IOException {
        //Before Close
        int socketPort = socket.getPort();
        InetAddress socketInetAddress = socket.getInetAddress();
        SocketAddress socketRemoteSocketAddress = socket.getRemoteSocketAddress();

        //After Close
        socket.close();

        if (connected ? !(socket.getPort() == socketPort)
                      : !(socket.getPort() == -1)) {
            System.out.println("Socket.getPort failed");
            failCount++;
        }

        if (connected ? !socket.getInetAddress().equals(socketInetAddress)
                      : !(socket.getInetAddress() == null)) {
            System.out.println("Socket.getInetAddress failed");
            failCount++;
        }

        if (connected ? !socket.getRemoteSocketAddress().equals(socketRemoteSocketAddress)
                      : !(socket.getRemoteSocketAddress() == null)) {
            System.out.println("Socket.getRemoteSocketAddresss failed");
            failCount++;
        }

        if (socket.getLocalPort() != -1) {
            System.out.println("Socket.getLocalPort failed");
            failCount++;
        }

        if (socket.getLocalAddress() != null) {
            System.out.println("Socket.getLocalAddress failed");
            failCount++;
        }

        if (socket.getLocalSocketAddress() != null) {
            System.out.println("Socket.getLocalSocketAddress failed");
            failCount++;
        }

        if (connected && !socket.isConnected()) {
            System.out.println("Socket.isConnected failed");
            failCount++;
        }

        if (!socket.isBound()) {
            System.out.println("Socket.isBound failed");
            failCount++;
        }

    }
}
