/*

 * 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 4944680

 * @summary Tests that IllegalArgumentException is thrown when

 *          MBeanServerForwrder is null.

 * @author Daniel Fuchs

 * @run clean SetMBeanServerForwarder

 * @run build SetMBeanServerForwarder

 * @run main SetMBeanServerForwarder

 */

import javax.management.*;

import javax.management.remote.*;

import javax.management.remote.rmi.*;

import java.util.Map;

import com.sun.jmx.remote.security.MBeanServerAccessController;

public class SetMBeanServerForwarder {

    static boolean isPresent(String cn) {

        try {

            Class.forName(cn);

            return true;

        } catch (ClassNotFoundException x) {

            return false;

        }

    }

    public static int test(String... urls) {

        int errorCount = 0;

        for (int i=0;i<urls.length;i++) {

            try {

                final MBeanServer mbs = MBeanServerFactory.newMBeanServer();

                final JMXConnectorServer cs1,cs2;

                final JMXServiceURL      url;

                System.out.println("*** -----------------------------------");

                System.out.println("*** JMXConnectorServer("+urls[i]+")");

                System.out.println("*** -----------------------------------");

                try {

                    url = new JMXServiceURL(urls[i]);

                    cs1 = JMXConnectorServerFactory

                        .newJMXConnectorServer(url,(Map)null,mbs);

                    cs2 = JMXConnectorServerFactory

                        .newJMXConnectorServer(url,(Map)null,null);

                } catch (Throwable thr) {

                    System.out.println("Failed to create ConnectorServer "+

                                       "from [" + urls[i] +"]: " + thr);

                    thr.printStackTrace();

                    errorCount++;

                    continue;

                }

                // Test using a JMXConnectorServer already connected to an

                // MBeanServer

                //

                // Set null MBeanServerForwarder - expect exception

                //

                try {

                    cs1.setMBeanServerForwarder(null);

                    errorCount++;

                    System.out.println("Expected IllegalArgumentException "+

                                       " not thrown (null forwarder) for " +

                                       url);

                    System.out.println("\t\t[connected to MBeanServer]");

                } catch (IllegalArgumentException iae) {

                    System.out.println("Received expected exception: " +

                                       iae);

                }

                // Now try with a real MBSF - should not throw exception

                //

                try {

                    final MBeanServerForwarder fwd = new

                        MBeanServerAccessController() {

                            protected void checkRead() {}

                            protected void checkWrite() {}

                        };

                    cs1.setMBeanServerForwarder(fwd);

                    // Verify that the MBSF was correctly set.

                    //

                    if (cs1.getMBeanServer() != fwd) {

                        System.out.println("MBeanServerForwarder not set "+

                                           "for " + url);

                        System.out.println("\t\t[connected to MBeanServer]");

                        throw new AssertionError("cs1.getMBeanServer()!=fwd");

                    }

                    // Verify that the MBS was correctly forwarded to the MBSF

                    //

                    if (fwd.getMBeanServer() != mbs) {

                        System.out.println("MBeanServer not set in Forwarder"+

                                           " for " + url);

                        System.out.println("\t\t[connected to MBeanServer]");

                        throw new AssertionError("fwd.getMBeanServer()!=mbs");

                    }

                    System.out.println("MBeanServerForwarder successfully "+

                                       "set for " + url);

                    System.out.println("\t\t[connected to MBeanServer]");

                } catch (Throwable x) {

                    errorCount++;

                    System.out.println("Failed to set forwarder for " +

                                       url);

                    System.out.println("\t\t[connected to MBeanServer]");

                    System.out.println("Unexpected exception: " +

                                       x);

                    x.printStackTrace();

                }

                // Test using a JMXConnectorServer not connected to any

                // MBeanServer

                //

                // Set null MBeanServerForwarder - expect exception

                //

                try {

                    cs2.setMBeanServerForwarder(null);

                    errorCount++;

                    System.out.println("Expected IllegalArgumentException "+

                                       " not thrown (null forwarder) for " +

                                       url);

                    System.out.println("\t\t[not connected to MBeanServer]");

                } catch (IllegalArgumentException iae) {

                    System.out.println("Received expected exception: " +

                                       iae);

                }

                // Now try with a real MBSF - should not throw exception

                //

                try {

                    final MBeanServerForwarder fwd = new

                        MBeanServerAccessController() {

                            protected void checkRead() {}

                            protected void checkWrite() {}

                        };

                    cs2.setMBeanServerForwarder(fwd);

                    // Verify that the MBSF was correctly set.

                    //

                    if (cs2.getMBeanServer() != fwd) {

                        System.out.println("MBeanServerForwarder not set "+

                                           "for " + url);

                        System.out.println("\t\t[not connected to MBeanServer]");

                        throw new AssertionError("cs2.getMBeanServer()!=fwd");

                    }

                    // Now register the connector

                    //

                    final ObjectName name =

                        new ObjectName(":type="+cs2.getClass().getName()+

                                       ",url="+ObjectName.quote(urls[i]));

                    mbs.registerMBean(cs2,name);

                    try {

                        // Verify that the MBSF was not disconnected.

                        //

                        if (cs2.getMBeanServer() != fwd) {

                            System.out.

                                println("MBeanServerForwarder changed "+

                                        "for " + url);

                            System.out.

                                println("\t\t[registerMBean]");

                            throw new

                                AssertionError("cs2.getMBeanServer()!=fwd");

                        }

                        // Verify that the MBS was not forwarded to the MBSF

                        //

                        if (fwd.getMBeanServer() != null) {

                            System.out.

                                println("MBeanServer changed in Forwarder"+

                                        " for " + url);

                            System.out.println("\t\t[registerMBean]");

                            throw new

                                AssertionError("fwd.getMBeanServer()!=null");

                        }

                    } finally {

                        mbs.unregisterMBean(name);

                    }

                    System.out.println("MBeanServerForwarder successfully "+

                                       "set for " + url);

                    System.out.println("\t\t[not connected to MBeanServer]");

                } catch (Throwable x) {

                    errorCount++;

                    System.out.println("Failed to set forwarder for " +

                                       url);

                    System.out.println("\t\t[not connected to MBeanServer]");

                    System.out.println("Unexpected exception: " +

                                       x);

                    x.printStackTrace();

                }

            } catch (Exception x) {

                System.err.println("Unexpected exception for " +

                                   urls[i] + ": " + x);

                x.printStackTrace();

                errorCount++;

            }

        }

        return errorCount;

    }

    public static void main(String args[]) {

        int errCount = 0;

        // mandatory

        errCount += test("service:jmx:rmi://");

        // optional

        if (isPresent("javax.management.remote.rmi._RMIConnectionImpl_Tie"))

            errCount += test("service:jmx:iiop://");

        if (isPresent("javax.management.remote.generic.GenericConnector"))

            errCount += test("service:jmx:jmxmp://");

        if (errCount > 0) {

            throw new RuntimeException("SetMBeanServerForwarder failed: " +

                                       errCount + " error(s) reported.");

        }

        System.out.println("SetMBeanServerForwarder passed.");

    }

}