/*
* Copyright (c) 2006, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.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
}
}
}