1 /* 2 * Copyright (c) 2006, 2016, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* 25 * @test 26 * @bug 6403794 27 * @summary Test that all the platform MXBeans are wrapped in StandardMBean so 28 * an MBeanServer which does not have support for MXBeans can be used. 29 * @author Luis-Miguel Alventosa 30 * @modules jdk.management 31 * @run clean MBeanServerMXBeanUnsupportedTest 32 * @run build MBeanServerMXBeanUnsupportedTest 33 * @run main/othervm MBeanServerMXBeanUnsupportedTest 34 */ 35 36 import java.lang.management.ManagementFactory; 37 import java.lang.reflect.InvocationHandler; 38 import java.lang.reflect.Method; 39 import java.lang.reflect.Proxy; 40 import java.util.Arrays; 41 import java.util.HashSet; 42 import javax.management.MBeanServer; 43 import javax.management.MBeanServerBuilder; 44 import javax.management.MBeanServerDelegate; 45 import javax.management.ObjectName; 46 import javax.management.StandardMBean; 47 import javax.management.remote.MBeanServerForwarder; 48 49 public class MBeanServerMXBeanUnsupportedTest { 50 51 /** 52 * An MBeanServerBuilder that returns an MBeanServer which throws a 53 * RuntimeException if MXBeans are not converted into StandardMBean. 54 */ 55 public static class MBeanServerBuilderImpl extends MBeanServerBuilder { 56 57 private final MBeanServerBuilder inner; 58 59 public MBeanServerBuilderImpl() { 60 inner = new MBeanServerBuilder(); 61 } 62 63 public MBeanServer newMBeanServer( 64 String defaultDomain, 65 MBeanServer outer, 66 MBeanServerDelegate delegate) { 67 final MBeanServerForwarder mbsf = 68 MBeanServerForwarderInvocationHandler.newProxyInstance(); 69 70 final MBeanServer innerMBeanServer = 71 inner.newMBeanServer(defaultDomain, 72 (outer == null ? mbsf : outer), 73 delegate); 74 75 mbsf.setMBeanServer(innerMBeanServer); 76 return mbsf; 77 } 78 } 79 80 /** 81 * An MBeanServerForwarderInvocationHandler that throws a 82 * RuntimeException if we try to register a non StandardMBean. 83 */ 84 public static class MBeanServerForwarderInvocationHandler 85 implements InvocationHandler { 86 87 public static final HashSet<String> excludeList = new HashSet<String>( 88 Arrays.asList("com.sun.management:type=DiagnosticCommand")); 89 90 public static MBeanServerForwarder newProxyInstance() { 91 92 final InvocationHandler handler = 93 new MBeanServerForwarderInvocationHandler(); 94 95 final Class[] interfaces = 96 new Class[] {MBeanServerForwarder.class}; 97 98 Object proxy = Proxy.newProxyInstance( 99 MBeanServerForwarder.class.getClassLoader(), 100 interfaces, 101 handler); 102 103 return MBeanServerForwarder.class.cast(proxy); 104 } 105 106 public Object invoke(Object proxy, Method method, Object[] args) 107 throws Throwable { 108 109 final String methodName = method.getName(); 110 111 if (methodName.equals("getMBeanServer")) { 112 return mbs; 113 } 114 115 if (methodName.equals("setMBeanServer")) { 116 if (args[0] == null) 117 throw new IllegalArgumentException("Null MBeanServer"); 118 if (mbs != null) 119 throw new IllegalArgumentException("MBeanServer object " + 120 "already initialized"); 121 mbs = (MBeanServer) args[0]; 122 return null; 123 } 124 125 if (methodName.equals("registerMBean")) { 126 Object mbean = args[0]; 127 ObjectName name = (ObjectName) args[1]; 128 String domain = name.getDomain(); 129 System.out.println("registerMBean: class=" + 130 mbean.getClass().getName() + "\tname=" + name); 131 Object result = method.invoke(mbs, args); 132 if (domain.equals("java.lang") || 133 domain.equals("java.util.logging") || 134 domain.equals("com.sun.management")) { 135 if(!excludeList.contains(name.getCanonicalName())) { 136 String mxbean = (String) 137 mbs.getMBeanInfo(name).getDescriptor().getFieldValue("mxbean"); 138 if (mxbean == null || !mxbean.equals("true")) { 139 throw new RuntimeException( 140 "Platform MBeans must be MXBeans!"); 141 } 142 if (!(mbean instanceof StandardMBean)) { 143 throw new RuntimeException( 144 "MXBeans must be wrapped in StandardMBean!"); 145 } 146 } 147 } 148 return result; 149 } 150 151 return method.invoke(mbs, args); 152 } 153 154 private MBeanServer mbs; 155 } 156 157 /* 158 * Standalone entry point. 159 * 160 * Run the test and report to stdout. 161 */ 162 public static void main(String args[]) throws Exception { 163 System.setProperty("javax.management.builder.initial", 164 MBeanServerBuilderImpl.class.getName()); 165 try { 166 ManagementFactory.getPlatformMBeanServer(); 167 } catch (RuntimeException e) { 168 System.out.println(">>> Unhappy Bye, Bye!"); 169 throw e; 170 } 171 System.out.println(">>> Happy Bye, Bye!"); 172 } 173 }