jdk/src/share/classes/javax/management/remote/rmi/RMIJRMPServerImpl.java - platform/libcore - Git at Google

 /*
  * Copyright (c) 2002, 2016, 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.  Oracle designates this
  * particular file as subject to the "Classpath" exception as provided
  * by Oracle in the LICENSE file that accompanied this code.
  *
  * 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.
  */

 package javax.management.remote.rmi;

 import java.io.IOException;
 import java.rmi.NoSuchObjectException;
 import java.rmi.Remote;
 import java.rmi.RemoteException;
 import java.rmi.server.RMIClientSocketFactory;
 import java.rmi.server.RMIServerSocketFactory;
 import java.rmi.server.UnicastRemoteObject;
 import java.rmi.server.RemoteObject;
 import java.util.Map;
 import java.util.Collections;
 import javax.security.auth.Subject;

 import com.sun.jmx.remote.internal.RMIExporter;
 import com.sun.jmx.remote.util.EnvHelp;
 import java.io.ObjectStreamClass;
 import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.List;
 import sun.reflect.misc.ReflectUtil;
 import sun.rmi.server.DeserializationChecker;
 import sun.rmi.server.UnicastServerRef;
 import sun.rmi.server.UnicastServerRef2;

 /**
  * <p>An {@link RMIServer} object that is exported through JRMP and that
  * creates client connections as RMI objects exported through JRMP.
  * User code does not usually reference this class directly.</p>
  *
  * @see RMIServerImpl
  *
  * @since 1.5
  */
 public class RMIJRMPServerImpl extends RMIServerImpl {

     private final ExportedWrapper exportedWrapper;

     /**
      * <p>Creates a new {@link RMIServer} object that will be exported
      * on the given port using the given socket factories.</p>
      *
      * @param port the port on which this object and the {@link
      * RMIConnectionImpl} objects it creates will be exported.  Can be
      * zero, to indicate any available port.
      *
      * @param csf the client socket factory for the created RMI
      * objects.  Can be null.
      *
      * @param ssf the server socket factory for the created RMI
      * objects.  Can be null.
      *
      * @param env the environment map.  Can be null.
      *
      * @exception IOException if the {@link RMIServer} object
      * cannot be created.
      *
      * @exception IllegalArgumentException if <code>port</code> is
      * negative.
      */
     public RMIJRMPServerImpl(int port,
                              RMIClientSocketFactory csf,
                              RMIServerSocketFactory ssf,
                              Map<String,?> env)
             throws IOException {

         super(env);

         if (port < 0)
             throw new IllegalArgumentException("Negative port: " + port);

         this.port = port;
         this.csf = csf;
         this.ssf = ssf;
         this.env = (env == null) ? Collections.<String, Object>emptyMap() : env;

         String[] credentialsTypes
                 = (String[]) this.env.get(EnvHelp.CREDENTIAL_TYPES);
         List<String> types = null;
         if (credentialsTypes != null) {
             types = new ArrayList<>();
             for (String type : credentialsTypes) {
                 if (type == null) {
                     throw new IllegalArgumentException("A credential type is null.");
                 }
                 ReflectUtil.checkPackageAccess(type);
                 types.add(type);
             }
         }
         exportedWrapper = types != null ?
                 new ExportedWrapper(this, types) :
                 null;
     }

     protected void export() throws IOException {
         if (exportedWrapper != null) {
             export(exportedWrapper);
         } else {
             export(this);
         }
     }

     private void export(Remote obj) throws RemoteException {
         final RMIExporter exporter =
             (RMIExporter) env.get(RMIExporter.EXPORTER_ATTRIBUTE);
         final boolean daemon = EnvHelp.isServerDaemon(env);

         if (daemon && exporter != null) {
             throw new IllegalArgumentException("If "+EnvHelp.JMX_SERVER_DAEMON+
                     " is specified as true, "+RMIExporter.EXPORTER_ATTRIBUTE+
                     " cannot be used to specify an exporter!");
         }

         if (daemon) {
             if (csf == null && ssf == null) {
                 new UnicastServerRef(port).exportObject(obj, null, true);
             } else {
                 new UnicastServerRef2(port, csf, ssf).exportObject(obj, null, true);
             }
         } else if (exporter != null) {
             exporter.exportObject(obj, port, csf, ssf);
         } else {
             UnicastRemoteObject.exportObject(obj, port, csf, ssf);
         }
     }

     private void unexport(Remote obj, boolean force)
             throws NoSuchObjectException {
         RMIExporter exporter =
             (RMIExporter) env.get(RMIExporter.EXPORTER_ATTRIBUTE);
         if (exporter == null)
             UnicastRemoteObject.unexportObject(obj, force);
         else
             exporter.unexportObject(obj, force);
     }

     protected String getProtocol() {
         return "rmi";
     }

     /**
      * <p>Returns a serializable stub for this {@link RMIServer} object.</p>
      *
      * @return a serializable stub.
      *
      * @exception IOException if the stub cannot be obtained - e.g the
      *            RMIJRMPServerImpl has not been exported yet.
      */
     public Remote toStub() throws IOException {
         if (exportedWrapper != null) {
             return RemoteObject.toStub(exportedWrapper);
         } else {
             return RemoteObject.toStub(this);
         }
     }

     /**
      * <p>Creates a new client connection as an RMI object exported
      * through JRMP. The port and socket factories for the new
      * {@link RMIConnection} object are the ones supplied
      * to the <code>RMIJRMPServerImpl</code> constructor.</p>
      *
      * @param connectionId the ID of the new connection. Every
      * connection opened by this connector server will have a
      * different id.  The behavior is unspecified if this parameter is
      * null.
      *
      * @param subject the authenticated subject.  Can be null.
      *
      * @return the newly-created <code>RMIConnection</code>.
      *
      * @exception IOException if the new {@link RMIConnection}
      * object cannot be created or exported.
      */
     protected RMIConnection makeClient(String connectionId, Subject subject)
             throws IOException {

         if (connectionId == null)
             throw new NullPointerException("Null connectionId");

         RMIConnection client =
             new RMIConnectionImpl(this, connectionId, getDefaultClassLoader(),
                                   subject, env);
         export(client);
         return client;
     }

     protected void closeClient(RMIConnection client) throws IOException {
         unexport(client, true);
     }

     /**
      * <p>Called by {@link #close()} to close the connector server by
      * unexporting this object.  After returning from this method, the
      * connector server must not accept any new connections.</p>
      *
      * @exception IOException if the attempt to close the connector
      * server failed.
      */
     protected void closeServer() throws IOException {
         if (exportedWrapper != null) {
             unexport(exportedWrapper, true);
         } else {
             unexport(this, true);
         }
     }

     private final int port;
     private final RMIClientSocketFactory csf;
     private final RMIServerSocketFactory ssf;
     private final Map<String, ?> env;

     private static class ExportedWrapper implements RMIServer, DeserializationChecker {
         private final RMIServer impl;
         private final List<String> allowedTypes;
         private ExportedWrapper(RMIServer impl, List<String> credentialsTypes) {
             this.impl = impl;
             allowedTypes = credentialsTypes;
         }

         @Override
         public String getVersion() throws RemoteException {
             return impl.getVersion();
         }

         @Override
         public RMIConnection newClient(Object credentials) throws IOException {
             return impl.newClient(credentials);
         }

         @Override
         public void check(Method method, ObjectStreamClass descriptor,
                 int paramIndex, int callID) {

             String type = descriptor.getName();
             if (!allowedTypes.contains(type)) {
                 throw new ClassCastException("Unsupported type: " + type);
             }
         }

         @Override
         public void checkProxyClass(Method method, String[] ifaces,
                 int paramIndex, int callID) {
             if (ifaces != null && ifaces.length > 0) {
                 for (String iface : ifaces) {
                     if (!allowedTypes.contains(iface)) {
                         throw new ClassCastException("Unsupported type: " + iface);
                     }
                 }
             }
         }
     }
 }
	/*
	* Copyright (c) 2002, 2016, 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. Oracle designates this
	* particular file as subject to the "Classpath" exception as provided
	* by Oracle in the LICENSE file that accompanied this code.
	*
	* 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.
	*/

	package javax.management.remote.rmi;

	import java.io.IOException;
	import java.rmi.NoSuchObjectException;
	import java.rmi.Remote;
	import java.rmi.RemoteException;
	import java.rmi.server.RMIClientSocketFactory;
	import java.rmi.server.RMIServerSocketFactory;
	import java.rmi.server.UnicastRemoteObject;
	import java.rmi.server.RemoteObject;
	import java.util.Map;
	import java.util.Collections;
	import javax.security.auth.Subject;

	import com.sun.jmx.remote.internal.RMIExporter;
	import com.sun.jmx.remote.util.EnvHelp;
	import java.io.ObjectStreamClass;
	import java.lang.reflect.Method;
	import java.util.ArrayList;
	import java.util.List;
	import sun.reflect.misc.ReflectUtil;
	import sun.rmi.server.DeserializationChecker;
	import sun.rmi.server.UnicastServerRef;
	import sun.rmi.server.UnicastServerRef2;

	/**
	* <p>An {@link RMIServer} object that is exported through JRMP and that
	* creates client connections as RMI objects exported through JRMP.
	* User code does not usually reference this class directly.</p>
	*
	* @see RMIServerImpl
	*
	* @since 1.5
	*/
	public class RMIJRMPServerImpl extends RMIServerImpl {

	private final ExportedWrapper exportedWrapper;

	/**
	* <p>Creates a new {@link RMIServer} object that will be exported
	* on the given port using the given socket factories.</p>
	*
	* @param port the port on which this object and the {@link
	* RMIConnectionImpl} objects it creates will be exported. Can be
	* zero, to indicate any available port.
	*
	* @param csf the client socket factory for the created RMI
	* objects. Can be null.
	*
	* @param ssf the server socket factory for the created RMI
	* objects. Can be null.
	*
	* @param env the environment map. Can be null.
	*
	* @exception IOException if the {@link RMIServer} object
	* cannot be created.
	*
	* @exception IllegalArgumentException if <code>port</code> is
	* negative.
	*/
	public RMIJRMPServerImpl(int port,
	RMIClientSocketFactory csf,
	RMIServerSocketFactory ssf,
	Map<String,?> env)
	throws IOException {

	super(env);

	if (port < 0)
	throw new IllegalArgumentException("Negative port: " + port);

	this.port = port;
	this.csf = csf;
	this.ssf = ssf;
	this.env = (env == null) ? Collections.<String, Object>emptyMap() : env;

	String[] credentialsTypes
	= (String[]) this.env.get(EnvHelp.CREDENTIAL_TYPES);
	List<String> types = null;
	if (credentialsTypes != null) {
	types = new ArrayList<>();
	for (String type : credentialsTypes) {
	if (type == null) {
	throw new IllegalArgumentException("A credential type is null.");
	}
	ReflectUtil.checkPackageAccess(type);
	types.add(type);
	}
	}
	exportedWrapper = types != null ?
	new ExportedWrapper(this, types) :
	null;
	}

	protected void export() throws IOException {
	if (exportedWrapper != null) {
	export(exportedWrapper);
	} else {
	export(this);
	}
	}

	private void export(Remote obj) throws RemoteException {
	final RMIExporter exporter =
	(RMIExporter) env.get(RMIExporter.EXPORTER_ATTRIBUTE);
	final boolean daemon = EnvHelp.isServerDaemon(env);

	if (daemon && exporter != null) {
	throw new IllegalArgumentException("If "+EnvHelp.JMX_SERVER_DAEMON+
	" is specified as true, "+RMIExporter.EXPORTER_ATTRIBUTE+
	" cannot be used to specify an exporter!");
	}

	if (daemon) {
	if (csf == null && ssf == null) {
	new UnicastServerRef(port).exportObject(obj, null, true);
	} else {
	new UnicastServerRef2(port, csf, ssf).exportObject(obj, null, true);
	}
	} else if (exporter != null) {
	exporter.exportObject(obj, port, csf, ssf);
	} else {
	UnicastRemoteObject.exportObject(obj, port, csf, ssf);
	}
	}

	private void unexport(Remote obj, boolean force)
	throws NoSuchObjectException {
	RMIExporter exporter =
	(RMIExporter) env.get(RMIExporter.EXPORTER_ATTRIBUTE);
	if (exporter == null)
	UnicastRemoteObject.unexportObject(obj, force);
	else
	exporter.unexportObject(obj, force);
	}

	protected String getProtocol() {
	return "rmi";
	}

	/**
	* <p>Returns a serializable stub for this {@link RMIServer} object.</p>
	*
	* @return a serializable stub.
	*
	* @exception IOException if the stub cannot be obtained - e.g the
	* RMIJRMPServerImpl has not been exported yet.
	*/
	public Remote toStub() throws IOException {
	if (exportedWrapper != null) {
	return RemoteObject.toStub(exportedWrapper);
	} else {
	return RemoteObject.toStub(this);
	}
	}

	/**
	* <p>Creates a new client connection as an RMI object exported
	* through JRMP. The port and socket factories for the new
	* {@link RMIConnection} object are the ones supplied
	* to the <code>RMIJRMPServerImpl</code> constructor.</p>
	*
	* @param connectionId the ID of the new connection. Every
	* connection opened by this connector server will have a
	* different id. The behavior is unspecified if this parameter is
	* null.
	*
	* @param subject the authenticated subject. Can be null.
	*
	* @return the newly-created <code>RMIConnection</code>.
	*
	* @exception IOException if the new {@link RMIConnection}
	* object cannot be created or exported.
	*/
	protected RMIConnection makeClient(String connectionId, Subject subject)
	throws IOException {

	if (connectionId == null)
	throw new NullPointerException("Null connectionId");

	RMIConnection client =
	new RMIConnectionImpl(this, connectionId, getDefaultClassLoader(),
	subject, env);
	export(client);
	return client;
	}

	protected void closeClient(RMIConnection client) throws IOException {
	unexport(client, true);
	}

	/**
	* <p>Called by {@link #close()} to close the connector server by
	* unexporting this object. After returning from this method, the
	* connector server must not accept any new connections.</p>
	*
	* @exception IOException if the attempt to close the connector
	* server failed.
	*/
	protected void closeServer() throws IOException {
	if (exportedWrapper != null) {
	unexport(exportedWrapper, true);
	} else {
	unexport(this, true);
	}
	}

	private final int port;
	private final RMIClientSocketFactory csf;
	private final RMIServerSocketFactory ssf;
	private final Map<String, ?> env;

	private static class ExportedWrapper implements RMIServer, DeserializationChecker {
	private final RMIServer impl;
	private final List<String> allowedTypes;
	private ExportedWrapper(RMIServer impl, List<String> credentialsTypes) {
	this.impl = impl;
	allowedTypes = credentialsTypes;
	}

	@Override
	public String getVersion() throws RemoteException {
	return impl.getVersion();
	}

	@Override
	public RMIConnection newClient(Object credentials) throws IOException {
	return impl.newClient(credentials);
	}

	@Override
	public void check(Method method, ObjectStreamClass descriptor,
	int paramIndex, int callID) {

	String type = descriptor.getName();
	if (!allowedTypes.contains(type)) {
	throw new ClassCastException("Unsupported type: " + type);
	}
	}

	@Override
	public void checkProxyClass(Method method, String[] ifaces,
	int paramIndex, int callID) {
	if (ifaces != null && ifaces.length > 0) {
	for (String iface : ifaces) {
	if (!allowedTypes.contains(iface)) {
	throw new ClassCastException("Unsupported type: " + iface);
	}
	}
	}
	}
	}
	}