1 /* 2 * Copyright (c) 2003, 2015, 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 4924683 27 * @summary Check RMI/JRMP stubs can be deserialized using user's loader 28 * @author Eamonn McManus 29 * 30 * @run clean DeserializeEncodedURLTest SingleClassLoader 31 * @run build DeserializeEncodedURLTest SingleClassLoader 32 * @run main DeserializeEncodedURLTest 33 */ 34 35 import java.io.*; 36 import java.rmi.*; 37 import java.util.*; 38 import javax.management.*; 39 import javax.management.remote.*; 40 import javax.management.remote.rmi.*; 41 42 /* 43 Test that the RMI connector client can handle a URL of the form 44 where the serialized RMIServer stub is encoded directly in the URL, 45 when the class of that stub is known to the supplied 46 DEFAULT_CLASS_LOADER but not to the calling code's class loader. 47 This is an unusual usage, and is not explicitly specified in the JMX 48 Remote API, but it is potentially useful where client and server 49 agree to a code base for mutant stubs (that e.g. use a different 50 protocol or include debugging or optimization). 51 52 We make an RMI connector server by giving it an instance of an 53 RMIJRMPServerImpl subclass that manufactures mutant stubs. These 54 stubs are known to a special loader (mutantLoader) but not to this 55 test's loader. We set up the client's default loader to 56 mutantLoader, and check that it can deserialize the stub containing 57 the mutant stub. 58 59 This test incidentally creates the connector server as an MBean 60 rather than using the JMXConnectorServerFactory, just because I'm 61 not sure we have coverage of that elsewhere. 62 */ 63 public class DeserializeEncodedURLTest { 64 private static final ClassLoader mutantLoader = 65 new SingleClassLoader("SubMutantRMIServerStub", 66 MutantRMIServerStub.class, 67 MutantRMIServerStub.class.getClassLoader()); 68 private static final Class subMutantRMIServerStubClass; 69 static { 70 try { 71 subMutantRMIServerStubClass = 72 mutantLoader.loadClass("SubMutantRMIServerStub"); 73 } catch (ClassNotFoundException e) { 74 throw new Error(e); 75 } 76 } 77 78 public static void main(String[] args) throws Exception { 79 System.out.println("Check that we can deserialize a mutant stub " + 80 "from an RMI connector URL even when the stub's " + 81 "class is known to the user's default loader " + 82 "but not the caller's loader"); 83 84 System.out.println("Create RMI connector server as an MBean"); 85 86 MBeanServer mbs = MBeanServerFactory.createMBeanServer(); 87 ObjectName csName = new ObjectName("test:type=RMIConnectorServer"); 88 JMXServiceURL url = new JMXServiceURL("rmi", null, 0); 89 RMIServerImpl impl = new MutantRMIServerImpl(); 90 mbs.createMBean("javax.management.remote.rmi.RMIConnectorServer", 91 csName, 92 new Object[] {url, null, impl, null}, 93 new String[] {JMXServiceURL.class.getName(), 94 Map.class.getName(), 95 RMIServerImpl.class.getName(), 96 MBeanServer.class.getName()}); 97 mbs.invoke(csName, "start", new Object[0], new String[0]); 98 99 JMXServiceURL address = 100 (JMXServiceURL) mbs.getAttribute(csName, "Address"); 101 102 System.out.println("Address with mutant stub: " + address); 103 104 Map env = new HashMap(); 105 env.put(JMXConnectorFactory.DEFAULT_CLASS_LOADER, mutantLoader); 106 JMXConnector conn = JMXConnectorFactory.newJMXConnector(address, env); 107 108 System.out.println("Client successfully created with this address"); 109 System.out.println("Try to connect newly-created client"); 110 111 try { 112 conn.connect(); 113 System.out.println("TEST FAILS: Connect worked but should not " + 114 "have"); 115 System.exit(1); 116 } catch (MutantException e) { 117 System.out.println("Caught MutantException as expected"); 118 } catch (Exception e) { 119 System.out.println("TEST FAILS: Caught unexpected exception:"); 120 e.printStackTrace(System.out); 121 System.exit(1); 122 } 123 124 mbs.invoke(csName, "stop", new Object[0], new String[0]); 125 System.out.println("Test passed"); 126 } 127 128 private static class MutantException extends IOException {} 129 130 public static class MutantRMIServerStub 131 implements RMIServer, Serializable { 132 public MutantRMIServerStub() {} 133 134 public String getVersion() { 135 return "1.0 BOGUS"; 136 } 137 138 public RMIConnection newClient(Object credentials) throws IOException { 139 throw new MutantException(); 140 } 141 } 142 143 private static class MutantRMIServerImpl extends RMIJRMPServerImpl { 144 public MutantRMIServerImpl() throws IOException { 145 super(0, null, null, null); 146 } 147 148 public Remote toStub() throws IOException { 149 try { 150 return (Remote) subMutantRMIServerStubClass.newInstance(); 151 } catch (Exception e) { 152 IOException ioe = 153 new IOException("Couldn't make submutant stub"); 154 ioe.initCause(e); 155 throw ioe; 156 } 157 } 158 } 159 }