1 /* 2 * Copyright (c) 2006, 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 5021246 27 * @summary Check that class downloading is supported by RMI connector 28 * @author Eamonn McManus 29 * @modules java.management 30 * @run main RMIDownloadTest receive without 31 * @run main RMIDownloadTest send without 32 * @run main RMIDownloadTest receive with 33 * @run main RMIDownloadTest send with 34 */ 35 36 /* 37 * This test checks that class downloading is supported by the RMI connector. 38 * We copy a precompiled class file into the temporary directory (which we 39 * assume is not in the classpath). We also create an instance of that 40 * class using a hardcoded ClassLoader. Then we try to get a remote attribute 41 * that returns that instance, and we try to set the remote attribute to the 42 * instance. In both cases, this will only work if the class can be downloaded 43 * based on the codebase that we have set to the temporary directory. We also 44 * test that it does *not* work when the codebase is not set, in case the test 45 * is succeeding for some other reason. 46 * 47 * We run the test four times, for each combination of (send, receive) x 48 * (with-codebase, without-codebase). Doing all four tests within the same 49 * run doesn't work, probably because RMI remembers the codebase property 50 * setting at some point. 51 */ 52 53 import java.io.File; 54 import java.io.FileOutputStream; 55 import java.io.OutputStream; 56 import java.lang.management.ManagementFactory; 57 import java.net.URL; 58 import java.security.Permission; 59 import java.util.Arrays; 60 import javax.management.Attribute; 61 import javax.management.MBeanServer; 62 import javax.management.MBeanServerConnection; 63 import javax.management.ObjectName; 64 import javax.management.remote.JMXConnector; 65 import javax.management.remote.JMXConnectorFactory; 66 import javax.management.remote.JMXConnectorServer; 67 import javax.management.remote.JMXConnectorServerFactory; 68 import javax.management.remote.JMXServiceURL; 69 70 public class RMIDownloadTest { 71 /* Following byte array was produced from this class: 72 * 73 * public class Zooby implements java.io.Serializable {} 74 * 75 * by this program: 76 * 77 * public class MakeZooby { 78 * public static void main(String[] args) throws Exception { 79 * int b; 80 * for (int offset = 0; (b = System.in.read()) >= 0; offset++) { 81 * System.out.print((byte) b + "," + 82 * ((offset % 16) == 15 ? '\n' : ' ')); 83 * } 84 * System.out.println(); 85 * } 86 * } 87 */ 88 private static final byte[] zoobyClassBytes = { 89 -54, -2, -70, -66, 0, 0, 0, 49, 0, 12, 10, 0, 3, 0, 8, 7, 90 0, 9, 7, 0, 10, 7, 0, 11, 1, 0, 6, 60, 105, 110, 105, 116, 91 62, 1, 0, 3, 40, 41, 86, 1, 0, 4, 67, 111, 100, 101, 12, 0, 92 5, 0, 6, 1, 0, 5, 90, 111, 111, 98, 121, 1, 0, 16, 106, 97, 93 118, 97, 47, 108, 97, 110, 103, 47, 79, 98, 106, 101, 99, 116, 1, 0, 94 20, 106, 97, 118, 97, 47, 105, 111, 47, 83, 101, 114, 105, 97, 108, 105, 95 122, 97, 98, 108, 101, 0, 33, 0, 2, 0, 3, 0, 1, 0, 4, 0, 96 0, 0, 1, 0, 1, 0, 5, 0, 6, 0, 1, 0, 7, 0, 0, 0, 97 17, 0, 1, 0, 1, 0, 0, 0, 5, 42, -73, 0, 1, -79, 0, 0, 98 0, 0, 0, 0, 99 }; 100 101 private static class ZoobyClassLoader extends ClassLoader { 102 protected Class<?> findClass(String name) throws ClassNotFoundException { 103 if (name.equals("Zooby")) { 104 return super.defineClass(name, zoobyClassBytes, 105 0, zoobyClassBytes.length); 106 } else 107 throw new ClassNotFoundException(name); 108 } 109 } 110 111 112 private static MBeanServer pmbs; 113 private static ObjectName getSetName; 114 private static GetSet getSetInstance; 115 116 public static void main(String[] args) throws Exception { 117 int sendIndex = -1; 118 int withIndex = -1; 119 if (args.length == 2) { 120 sendIndex = 121 Arrays.asList("send", "receive").indexOf(args[0]); 122 withIndex = 123 Arrays.asList("with", "without").indexOf(args[1]); 124 } 125 if (sendIndex < 0 || withIndex < 0) 126 throw new Exception("Usage: RMIDownloadTest (send|receive) (with|without)"); 127 final boolean send = (sendIndex == 0); 128 final boolean with = (withIndex == 0); 129 130 pmbs = ManagementFactory.getPlatformMBeanServer(); 131 getSetName = new ObjectName(":type=GetSet"); 132 getSetInstance = new GetSet(); 133 pmbs.registerMBean(getSetInstance, getSetName); 134 135 System.setSecurityManager(new LaidBackSecurityManager()); 136 137 // System.setProperty("sun.rmi.loader.logLevel", "VERBOSE"); 138 139 String tmpdir = System.getProperty("java.io.tmpdir"); 140 String classfile = tmpdir + File.separator + "Zooby.class"; 141 File zoobyFile = new File(classfile); 142 zoobyFile.deleteOnExit(); 143 OutputStream os = new FileOutputStream(zoobyFile); 144 for (byte b : zoobyClassBytes) 145 os.write(b); 146 os.close(); 147 148 // Check that we can't load the Zooby class from the classpath 149 try { 150 Class.forName("Zooby"); 151 throw new Exception("Class \"Zooby\" is in the classpath!"); 152 } catch (ClassNotFoundException e) { 153 // OK: expected 154 } 155 156 if (send) 157 System.out.println("Testing we can send an object from client to server"); 158 else 159 System.out.println("Testing we can receive an object from server to client"); 160 161 if (with) { 162 // Test with the codebase property. Downloading should work. 163 URL zoobyURL = zoobyFile.getParentFile().toURI().toURL(); 164 System.setProperty("java.rmi.server.codebase", zoobyURL.toString()); 165 System.out.println("Testing with codebase, should work"); 166 System.out.println("Codebase is " + 167 System.getProperty("java.rmi.server.codebase")); 168 test(send, true); 169 } else { 170 // Test without setting the codebase property. 171 // This should not work; if it does it probably means java.io.tmpdir 172 // is in the classpath. 173 System.out.println("Testing without codebase, should fail"); 174 test(send, false); 175 } 176 177 } 178 179 private static void test(boolean send, boolean shouldWork) throws Exception { 180 try { 181 testWithException(send); 182 } catch (Exception e) { 183 if (shouldWork) 184 throw e; 185 System.out.println("Got exception as expected: " + e); 186 return; 187 } 188 if (!shouldWork) 189 throw new Exception("Test passed without codebase but should not"); 190 } 191 192 private static void testWithException(boolean send) 193 throws Exception { 194 ClassLoader zoobyCL = new ZoobyClassLoader(); 195 Class<?> zoobyClass = Class.forName("Zooby", false, zoobyCL); 196 Object zooby = zoobyClass.newInstance(); 197 198 JMXServiceURL url = new JMXServiceURL("service:jmx:rmi:///"); 199 JMXConnectorServer cs = 200 JMXConnectorServerFactory.newJMXConnectorServer(url, null, pmbs); 201 cs.start(); 202 JMXServiceURL addr = cs.getAddress(); 203 JMXConnector cc = JMXConnectorFactory.connect(addr); 204 MBeanServerConnection mbsc = cc.getMBeanServerConnection(); 205 206 Object rzooby; 207 if (send) { 208 System.out.println("Sending object..."); 209 mbsc.setAttribute(getSetName, new Attribute("It", zooby)); 210 rzooby = getSetInstance.getIt(); 211 } else { 212 System.out.println("Receiving object..."); 213 getSetInstance.setIt(zooby); 214 rzooby = mbsc.getAttribute(getSetName, "It"); 215 } 216 217 if (!rzooby.getClass().getName().equals("Zooby")) { 218 throw new Exception("FAILED: remote object is not a Zooby"); 219 } 220 if (rzooby.getClass().getClassLoader() == 221 zooby.getClass().getClassLoader()) { 222 throw new Exception("FAILED: same class loader: " + 223 zooby.getClass().getClassLoader()); 224 } 225 226 cc.close(); 227 cs.stop(); 228 } 229 230 public static interface GetSetMBean { 231 public Object getIt(); 232 public void setIt(Object x); 233 } 234 235 public static class GetSet implements GetSetMBean { 236 public GetSet() { 237 } 238 239 public Object getIt() { 240 return what; 241 } 242 243 public void setIt(Object x) { 244 this.what = x; 245 } 246 247 private Object what; 248 } 249 250 public static class LaidBackSecurityManager extends SecurityManager { 251 public void checkPermission(Permission perm) { 252 // OK, dude 253 } 254 } 255 }