/*

 * Copyright 2004 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 5012634

 * @summary Test that JMX classes use fully-qualified class names

 * in MBeanNotificationInfo

 * @author Eamonn McManus

 * @run clean NotificationInfoTest

 * @run build NotificationInfoTest

 * @run main NotificationInfoTest

 */

import java.io.*;

import java.lang.reflect.*;

import java.net.*;

import java.security.CodeSource;

import java.util.*;

import java.util.jar.*;

import javax.management.*;

import javax.management.modelmbean.*;

import javax.management.relation.*;

/*

 * This test finds all classes in the same code-base as the JMX

 * classes that look like Standard MBeans, and checks that if they are

 * NotificationBroadcasters they declare existent notification types.

 * A class looks like a Standard MBean if both Thing and ThingMBean

 * classes exist.  So for example javax.management.timer.Timer looks

 * like a Standard MBean because javax.management.timer.TimerMBean

 * exists.  Timer is instanceof NotificationBroadcaster, so we expect

 * that ((NotificationBroadcaster) timer).getNotificationInfo() will

 * return an array of MBeanNotificationInfo where each entry has a

 * getName() that names an existent Java class that is a Notification.

 *

 * An MBean is "suspicious" if it is a NotificationBroadcaster but its

 * MBeanNotificationInfo[] is empty.  This is legal, but surprising.

 *

 * In order to call getNotificationInfo(), we need an instance of the

 * class.  We attempt to make one by calling a public no-arg

 * constructor.  But the "construct" method below can be extended to

 * construct specific MBean classes for which the no-arg constructor

 * doesn't exist.

 *

 * The test is obviously not exhaustive, but does catch the cases that

 * failed in 5012634.

 */

public class NotificationInfoTest {

    // class or object names where the test failed

    private static final Set/*<String>*/ failed = new TreeSet();

    // class or object names where there were no MBeanNotificationInfo entries

    private static final Set/*<String>*/ suspicious = new TreeSet();

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

        System.out.println("Checking that all known MBeans that are " +

                           "NotificationBroadcasters have sane " +

                           "MBeanInfo.getNotifications()");

        System.out.println("Checking platform MBeans...");

        checkPlatformMBeans();

        CodeSource cs =

            javax.management.MBeanServer.class.getProtectionDomain()

            .getCodeSource();

        URL codeBase;

        if (cs == null) {

            codeBase = new URL("file:" + System.getProperty("java.home") +

                               "/lib/rt.jar");

        } else

            codeBase = cs.getLocation();

        System.out.println();

        System.out.println("Looking for standard MBeans...");

        String[] classes = findStandardMBeans(codeBase);

        System.out.println("Testing standard MBeans...");

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

            String name = classes[i];

            Class c;

            try {

                c = Class.forName(name);

            } catch (Throwable e) {

                System.out.println(name + ": cannot load (not public?): " + e);

                continue;

            }

            if (!NotificationBroadcaster.class.isAssignableFrom(c)) {

                System.out.println(name + ": not a NotificationBroadcaster");

                continue;

            }

            NotificationBroadcaster mbean;

            Constructor constr;

            try {

                constr = c.getConstructor(null);

            } catch (Exception e) {

                System.out.println(name + ": no public no-arg constructor: "

                                   + e);

                continue;

            }

            try {

                mbean = (NotificationBroadcaster) constr.newInstance(null);

            } catch (Exception e) {

                System.out.println(name + ": no-arg constructor failed: " + e);

                continue;

            }

            check(mbean);

        }

        System.out.println();

        System.out.println("Testing some explicit cases...");

        check(new RelationService(false));

        /*

          We can't do this:

            check(new RequiredModelMBean());

          because the Model MBean spec more or less forces us to use the

          names GENERIC and ATTRIBUTE_CHANGE for its standard notifs.

        */

        checkRMIConnectorServer();

        System.out.println();

        if (!suspicious.isEmpty())

            System.out.println("SUSPICIOUS CLASSES: " + suspicious);

        if (failed.isEmpty())

            System.out.println("TEST PASSED");

        else {

            System.out.println("TEST FAILED: " + failed);

            System.exit(1);

        }

    }

    private static void check(NotificationBroadcaster mbean)

            throws Exception {

        System.out.print(mbean.getClass().getName() + ": ");

        check(mbean.getClass().getName(), mbean.getNotificationInfo());

    }

    private static void checkPlatformMBeans() throws Exception {

        Class managementFactory;

        try {

            managementFactory =

                Class.forName("java.lang.management.ManagementFactory");

        } catch (Exception e) {

            System.out.println("...no ManagementFactory, assuming pre-Tiger: "

                               + e);

            return;

        }

        Method getPlatformMBeanServer =

            managementFactory.getMethod("getPlatformMBeanServer", null);

        MBeanServer mbs = (MBeanServer)

            getPlatformMBeanServer.invoke(null, null);

        Set mbeanNames = mbs.queryNames(null, null);

        for (Iterator it = mbeanNames.iterator(); it.hasNext(); ) {

            ObjectName name = (ObjectName) it.next();

            if (!mbs.isInstanceOf(name,

                                  NotificationBroadcaster.class.getName())) {

                System.out.println(name + ": not a NotificationBroadcaster");

            } else {

                MBeanInfo mbi = mbs.getMBeanInfo(name);

                check(name.toString(), mbi.getNotifications());

            }

        }

    }

    private static void checkRMIConnectorServer() throws Exception {

        Class rmiConnectorServer;

        try {

            rmiConnectorServer =

                Class.forName("javax.management.remote.rmi.RMIConnectorServer");

        } catch (Exception e) {

            System.out.println("No RMIConnectorServer class, skipping: " + e);

            return;

        }

        Class jmxServiceURL =

            Class.forName("javax.management.remote.JMXServiceURL");

        Constructor jmxServiceURLConstructor =

            jmxServiceURL.getConstructor(new Class[] {String.class});

        Object url =

            jmxServiceURLConstructor.newInstance(new Object[] {

                "service:jmx:rmi://"

            });

        Constructor rmiConnectorServerConstructor =

            rmiConnectorServer.getConstructor(new Class[] {

                jmxServiceURL, Map.class

            });

        Object connector =

            rmiConnectorServerConstructor.newInstance(new Object[] {

                url, null

            });

        check((NotificationBroadcaster) connector);

    }

    private static void check(String what, MBeanNotificationInfo[] mbnis) {

        System.out.print(what + ": checking notification info: ");

        if (mbnis.length == 0) {

            System.out.println("NONE (suspicious)");

            suspicious.add(what);

            return;

        }

        // Each MBeanNotificationInfo.getName() should be an existent

        // Java class that is Notification or a subclass of it

        for (int j = 0; j < mbnis.length; j++) {

            String notifClassName = mbnis[j].getName();

                Class notifClass;

                try {

                    notifClass = Class.forName(notifClassName);

                } catch (Exception e) {

                    System.out.print("FAILED(" + notifClassName + ": " + e +

                                     ") ");

                    failed.add(what);

                    continue;

                }

                if (!Notification.class.isAssignableFrom(notifClass)) {

                    System.out.print("FAILED(" + notifClassName +

                                     ": not a Notification) ");

                    failed.add(what);

                    continue;

                }

                System.out.print("OK(" + notifClassName + ") ");

        }

        System.out.println();

    }

    private static String[] findStandardMBeans(URL codeBase)

            throws Exception {

        Set names;

        if (codeBase.getProtocol().equalsIgnoreCase("file")

            && codeBase.toString().endsWith("/"))

            names = findStandardMBeansFromDir(codeBase);

        else

            names = findStandardMBeansFromJar(codeBase);

        Set standardMBeanNames = new TreeSet();

        for (Iterator it = names.iterator(); it.hasNext(); ) {

            String name = (String) it.next();

            if (name.endsWith("MBean")) {

                String prefix = name.substring(0, name.length() - 5);

                if (names.contains(prefix))

                    standardMBeanNames.add(prefix);

            }

        }

        return (String[]) standardMBeanNames.toArray(new String[0]);

    }

    private static Set findStandardMBeansFromJar(URL codeBase)

            throws Exception {

        InputStream is = codeBase.openStream();

        JarInputStream jis = new JarInputStream(is);

        Set names = new TreeSet();

        JarEntry entry;

        while ((entry = jis.getNextJarEntry()) != null) {

            String name = entry.getName();

            if (!name.endsWith(".class"))

                continue;

            name = name.substring(0, name.length() - 6);

            name = name.replace('/', '.');

            names.add(name);

        }

        return names;

    }

    private static Set findStandardMBeansFromDir(URL codeBase)

            throws Exception {

        File dir = new File(new URI(codeBase.toString()));

        Set names = new TreeSet();

        scanDir(dir, "", names);

        return names;

    }

    private static void scanDir(File dir, String prefix, Set names)

            throws Exception {

        File[] files = dir.listFiles();

        if (files == null)

            return;

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

            File f = files[i];

            String name = f.getName();

            String p = (prefix.equals("")) ? name : prefix + "." + name;

            if (f.isDirectory())

                scanDir(f, p, names);

            else if (name.endsWith(".class")) {

                p = p.substring(0, p.length() - 6);

                names.add(p);

            }

        }

    }

}