--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/javax/management/remote/mandatory/loading/RMIDownloadTest.java Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,254 @@
+/*
+ * Copyright 2006 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 5021246
+ * @summary Check that class downloading is supported by RMI connector
+ * @author Eamonn McManus
+ * @run main RMIDownloadTest receive without
+ * @run main RMIDownloadTest send without
+ * @run main RMIDownloadTest receive with
+ * @run main RMIDownloadTest send with
+ */
+
+/*
+ * This test checks that class downloading is supported by the RMI connector.
+ * We copy a precompiled class file into the temporary directory (which we
+ * assume is not in the classpath). We also create an instance of that
+ * class using a hardcoded ClassLoader. Then we try to get a remote attribute
+ * that returns that instance, and we try to set the remote attribute to the
+ * instance. In both cases, this will only work if the class can be downloaded
+ * based on the codebase that we have set to the temporary directory. We also
+ * test that it does *not* work when the codebase is not set, in case the test
+ * is succeeding for some other reason.
+ *
+ * We run the test four times, for each combination of (send, receive) x
+ * (with-codebase, without-codebase). Doing all four tests within the same
+ * run doesn't work, probably because RMI remembers the codebase property
+ * setting at some point.
+ */
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.OutputStream;
+import java.lang.management.ManagementFactory;
+import java.net.URL;
+import java.security.Permission;
+import java.util.Arrays;
+import javax.management.Attribute;
+import javax.management.MBeanServer;
+import javax.management.MBeanServerConnection;
+import javax.management.ObjectName;
+import javax.management.remote.JMXConnector;
+import javax.management.remote.JMXConnectorFactory;
+import javax.management.remote.JMXConnectorServer;
+import javax.management.remote.JMXConnectorServerFactory;
+import javax.management.remote.JMXServiceURL;
+
+public class RMIDownloadTest {
+ /* Following byte array was produced from this class:
+ *
+ * public class Zooby implements java.io.Serializable {}
+ *
+ * by this program:
+ *
+ * public class MakeZooby {
+ * public static void main(String[] args) throws Exception {
+ * int b;
+ * for (int offset = 0; (b = System.in.read()) >= 0; offset++) {
+ * System.out.print((byte) b + "," +
+ * ((offset % 16) == 15 ? '\n' : ' '));
+ * }
+ * System.out.println();
+ * }
+ * }
+ */
+ private static final byte[] zoobyClassBytes = {
+ -54, -2, -70, -66, 0, 0, 0, 49, 0, 12, 10, 0, 3, 0, 8, 7,
+ 0, 9, 7, 0, 10, 7, 0, 11, 1, 0, 6, 60, 105, 110, 105, 116,
+ 62, 1, 0, 3, 40, 41, 86, 1, 0, 4, 67, 111, 100, 101, 12, 0,
+ 5, 0, 6, 1, 0, 5, 90, 111, 111, 98, 121, 1, 0, 16, 106, 97,
+ 118, 97, 47, 108, 97, 110, 103, 47, 79, 98, 106, 101, 99, 116, 1, 0,
+ 20, 106, 97, 118, 97, 47, 105, 111, 47, 83, 101, 114, 105, 97, 108, 105,
+ 122, 97, 98, 108, 101, 0, 33, 0, 2, 0, 3, 0, 1, 0, 4, 0,
+ 0, 0, 1, 0, 1, 0, 5, 0, 6, 0, 1, 0, 7, 0, 0, 0,
+ 17, 0, 1, 0, 1, 0, 0, 0, 5, 42, -73, 0, 1, -79, 0, 0,
+ 0, 0, 0, 0,
+ };
+
+ private static class ZoobyClassLoader extends ClassLoader {
+ protected Class<?> findClass(String name) throws ClassNotFoundException {
+ if (name.equals("Zooby")) {
+ return super.defineClass(name, zoobyClassBytes,
+ 0, zoobyClassBytes.length);
+ } else
+ throw new ClassNotFoundException(name);
+ }
+ }
+
+
+ private static MBeanServer pmbs;
+ private static ObjectName getSetName;
+ private static GetSet getSetInstance;
+
+ public static void main(String[] args) throws Exception {
+ int sendIndex = -1;
+ int withIndex = -1;
+ if (args.length == 2) {
+ sendIndex =
+ Arrays.asList("send", "receive").indexOf(args[0]);
+ withIndex =
+ Arrays.asList("with", "without").indexOf(args[1]);
+ }
+ if (sendIndex < 0 || withIndex < 0)
+ throw new Exception("Usage: RMIDownloadTest (send|receive) (with|without)");
+ final boolean send = (sendIndex == 0);
+ final boolean with = (withIndex == 0);
+
+ pmbs = ManagementFactory.getPlatformMBeanServer();
+ getSetName = new ObjectName(":type=GetSet");
+ getSetInstance = new GetSet();
+ pmbs.registerMBean(getSetInstance, getSetName);
+
+ System.setSecurityManager(new LaidBackSecurityManager());
+
+// System.setProperty("sun.rmi.loader.logLevel", "VERBOSE");
+
+ String tmpdir = System.getProperty("java.io.tmpdir");
+ String classfile = tmpdir + File.separator + "Zooby.class";
+ File zoobyFile = new File(classfile);
+ zoobyFile.deleteOnExit();
+ OutputStream os = new FileOutputStream(zoobyFile);
+ for (byte b : zoobyClassBytes)
+ os.write(b);
+ os.close();
+
+ // Check that we can't load the Zooby class from the classpath
+ try {
+ Class.forName("Zooby");
+ throw new Exception("Class \"Zooby\" is in the classpath!");
+ } catch (ClassNotFoundException e) {
+ // OK: expected
+ }
+
+ if (send)
+ System.out.println("Testing we can send an object from client to server");
+ else
+ System.out.println("Testing we can receive an object from server to client");
+
+ if (with) {
+ // Test with the codebase property. Downloading should work.
+ URL zoobyURL = zoobyFile.getParentFile().toURI().toURL();
+ System.setProperty("java.rmi.server.codebase", zoobyURL.toString());
+ System.out.println("Testing with codebase, should work");
+ System.out.println("Codebase is " +
+ System.getProperty("java.rmi.server.codebase"));
+ test(send, true);
+ } else {
+ // Test without setting the codebase property.
+ // This should not work; if it does it probably means java.io.tmpdir
+ // is in the classpath.
+ System.out.println("Testing without codebase, should fail");
+ test(send, false);
+ }
+
+ }
+
+ private static void test(boolean send, boolean shouldWork) throws Exception {
+ try {
+ testWithException(send);
+ } catch (Exception e) {
+ if (shouldWork)
+ throw e;
+ System.out.println("Got exception as expected: " + e);
+ return;
+ }
+ if (!shouldWork)
+ throw new Exception("Test passed without codebase but should not");
+ }
+
+ private static void testWithException(boolean send)
+ throws Exception {
+ ClassLoader zoobyCL = new ZoobyClassLoader();
+ Class<?> zoobyClass = Class.forName("Zooby", false, zoobyCL);
+ Object zooby = zoobyClass.newInstance();
+
+ JMXServiceURL url = new JMXServiceURL("service:jmx:rmi:///");
+ JMXConnectorServer cs =
+ JMXConnectorServerFactory.newJMXConnectorServer(url, null, pmbs);
+ cs.start();
+ JMXServiceURL addr = cs.getAddress();
+ JMXConnector cc = JMXConnectorFactory.connect(addr);
+ MBeanServerConnection mbsc = cc.getMBeanServerConnection();
+
+ Object rzooby;
+ if (send) {
+ System.out.println("Sending object...");
+ mbsc.setAttribute(getSetName, new Attribute("It", zooby));
+ rzooby = getSetInstance.getIt();
+ } else {
+ System.out.println("Receiving object...");
+ getSetInstance.setIt(zooby);
+ rzooby = mbsc.getAttribute(getSetName, "It");
+ }
+
+ if (!rzooby.getClass().getName().equals("Zooby")) {
+ throw new Exception("FAILED: remote object is not a Zooby");
+ }
+ if (rzooby.getClass().getClassLoader() ==
+ zooby.getClass().getClassLoader()) {
+ throw new Exception("FAILED: same class loader: " +
+ zooby.getClass().getClassLoader());
+ }
+
+ cc.close();
+ cs.stop();
+ }
+
+ public static interface GetSetMBean {
+ public Object getIt();
+ public void setIt(Object x);
+ }
+
+ public static class GetSet implements GetSetMBean {
+ public GetSet() {
+ }
+
+ public Object getIt() {
+ return what;
+ }
+
+ public void setIt(Object x) {
+ this.what = x;
+ }
+
+ private Object what;
+ }
+
+ public static class LaidBackSecurityManager extends SecurityManager {
+ public void checkPermission(Permission perm) {
+ // OK, dude
+ }
+ }
+}