159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/* 259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Copyright (c) 2009-2010 jMonkeyEngine 359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * All rights reserved. 459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Redistribution and use in source and binary forms, with or without 659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * modification, are permitted provided that the following conditions are 759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * met: 859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions of source code must retain the above copyright 1059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer. 1159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions in binary form must reproduce the above copyright 1359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer in the 1459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * documentation and/or other materials provided with the distribution. 1559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Neither the name of 'jMonkeyEngine' nor the names of its contributors 1759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * may be used to endorse or promote products derived from this software 1859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * without specific prior written permission. 1959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 2059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 2259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 2359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 2459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 2559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 2659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 2759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 2859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 2959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 3059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 3259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapackage com.jme3.network.rmi; 3459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.ClientStateListener.DisconnectInfo; 3659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.*; 3759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.serializing.Serializer; 3859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.util.IntMap; 3959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.util.IntMap.Entry; 4059b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.IOException; 4159b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.lang.reflect.InvocationTargetException; 4259b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.lang.reflect.Method; 4359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.lang.reflect.Proxy; 4459b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.ArrayList; 4559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.HashMap; 4659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.logging.Level; 4759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.logging.Logger; 4859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapublic class ObjectStore { 5059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static final Logger logger = Logger.getLogger(ObjectStore.class.getName()); 5259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static final class Invocation { 5459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object retVal; 5659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean available = false; 5759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta @Override 5959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public String toString(){ 6059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return "Invocation[" + retVal + "]"; 6159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 6259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 6359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private Client client; 6559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private Server server; 6659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private ClientEventHandler clientEventHandler = new ClientEventHandler(); 6859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private ServerEventHandler serverEventHandler = new ServerEventHandler(); 6959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Local object ID counter 7159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private volatile short objectIdCounter = 0; 7259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Local invocation ID counter 7459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private volatile short invocationIdCounter = 0; 7559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Invocations waiting .. 7759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private IntMap<Invocation> pendingInvocations = new IntMap<Invocation>(); 7859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Objects I share with other people 8059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private IntMap<LocalObject> localObjects = new IntMap<LocalObject>(); 8159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Objects others share with me 8359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private HashMap<String, RemoteObject> remoteObjects = new HashMap<String, RemoteObject>(); 8459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private IntMap<RemoteObject> remoteObjectsById = new IntMap<RemoteObject>(); 8559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private final Object receiveObjectLock = new Object(); 8759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public class ServerEventHandler implements MessageListener<HostedConnection>, 8959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ConnectionListener { 9059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void messageReceived(HostedConnection source, Message m) { 9259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta onMessage(source, m); 9359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 9459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void connectionAdded(Server server, HostedConnection conn) { 9659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta onConnection(conn); 9759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 9859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void connectionRemoved(Server server, HostedConnection conn) { 10059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public class ClientEventHandler implements MessageListener, 10559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ClientStateListener { 10659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void messageReceived(Object source, Message m) { 10859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta onMessage(null, m); 10959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void clientConnected(Client c) { 11259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta onConnection(null); 11359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void clientDisconnected(Client c, DisconnectInfo info) { 11659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta static { 12159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer s = new RmiSerializer(); 12259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.registerClass(RemoteObjectDefMessage.class, s); 12359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.registerClass(RemoteMethodCallMessage.class, s); 12459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.registerClass(RemoteMethodReturnMessage.class, s); 12559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public ObjectStore(Client client) { 12859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.client = client; 12959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta client.addMessageListener(clientEventHandler, 13059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteObjectDefMessage.class, 13159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodCallMessage.class, 13259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodReturnMessage.class); 13359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta client.addClientStateListener(clientEventHandler); 13459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public ObjectStore(Server server) { 13759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.server = server; 13859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta server.addMessageListener(serverEventHandler, 13959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteObjectDefMessage.class, 14059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodCallMessage.class, 14159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodReturnMessage.class); 14259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta server.addConnectionListener(serverEventHandler); 14359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 14459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private ObjectDef makeObjectDef(LocalObject localObj){ 14659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ObjectDef def = new ObjectDef(); 14759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta def.objectName = localObj.objectName; 14859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta def.objectId = localObj.objectId; 14959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta def.methods = localObj.methods; 15059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return def; 15159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void exposeObject(String name, Object obj) throws IOException{ 15459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Create a local object 15559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta LocalObject localObj = new LocalObject(); 15659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta localObj.objectName = name; 15759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta localObj.objectId = objectIdCounter++; 15859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta localObj.theObject = obj; 15959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta //localObj.methods = obj.getClass().getMethods(); 16059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ArrayList<Method> methodList = new ArrayList<Method>(); 16259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (Method method : obj.getClass().getMethods()){ 16359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (method.getDeclaringClass() == obj.getClass()){ 16459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta methodList.add(method); 16559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta localObj.methods = methodList.toArray(new Method[methodList.size()]); 16859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Put it in the store 17059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta localObjects.put(localObj.objectId, localObj); 17159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 17259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Inform the others of its existence 17359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteObjectDefMessage defMsg = new RemoteObjectDefMessage(); 17459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defMsg.objects = new ObjectDef[]{ makeObjectDef(localObj) }; 17559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 17659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (client != null) { 17759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta client.send(defMsg); 17859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Client: Sending {0}", defMsg); 17959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } else { 18059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta server.broadcast(defMsg); 18159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Server: Sending {0}", defMsg); 18259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 18559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public <T> T getExposedObject(String name, Class<T> type, boolean waitFor) throws InterruptedException{ 18659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteObject ro = remoteObjects.get(name); 18759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (ro == null){ 18859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!waitFor) 18959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new RuntimeException("Cannot find remote object named: " + name); 19059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta else{ 19159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta do { 19259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta synchronized (receiveObjectLock){ 19359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta receiveObjectLock.wait(); 19459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 19559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } while ( (ro = remoteObjects.get(name)) == null ); 19659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 19759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 19859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 19959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object proxy = Proxy.newProxyInstance(getClass().getClassLoader(), new Class[]{ type }, ro); 20059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ro.loadMethods(type); 20159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return (T) proxy; 20259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 20359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 20459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object invokeRemoteMethod(RemoteObject remoteObj, Method method, Object[] args){ 20559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Integer methodIdInt = remoteObj.methodMap.get(method); 20659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (methodIdInt == null) 20759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new RuntimeException("Method not implemented by remote object owner: "+method); 20859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 20959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean needReturn = method.getReturnType() != void.class; 21059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta short objectId = remoteObj.objectId; 21159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta short methodId = methodIdInt.shortValue(); 21259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodCallMessage call = new RemoteMethodCallMessage(); 21359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta call.methodId = methodId; 21459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta call.objectId = objectId; 21559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta call.args = args; 21659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 21759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Invocation invoke = null; 21859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (needReturn){ 21959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta call.invocationId = invocationIdCounter++; 22059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta invoke = new Invocation(); 22159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Note: could cause threading issues if used from multiple threads 22259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta pendingInvocations.put(call.invocationId, invoke); 22359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 22459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 22559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (server != null){ 22659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta remoteObj.client.send(call); 22759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Server: Sending {0}", call); 22859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 22959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta client.send(call); 23059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Client: Sending {0}", call); 23159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 23259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 23359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (invoke != null){ 23459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta synchronized(invoke){ 23559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while (!invoke.available){ 23659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 23759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta invoke.wait(); 23859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch (InterruptedException ex){ 23959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ex.printStackTrace(); 24059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 24159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 24259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 24359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Note: could cause threading issues if used from multiple threads 24459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta pendingInvocations.remove(call.invocationId); 24559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return invoke.retVal; 24659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 24759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return null; 24859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 24959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 25059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 25159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private void onMessage(HostedConnection source, Message message) { 25259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Might want to do more strict validation of the data 25359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // in the message to prevent crashes 25459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 25559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (message instanceof RemoteObjectDefMessage){ 25659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteObjectDefMessage defMsg = (RemoteObjectDefMessage) message; 25759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 25859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ObjectDef[] defs = defMsg.objects; 25959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (ObjectDef def : defs){ 26059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteObject remoteObject = new RemoteObject(this, source); 26159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta remoteObject.objectId = (short)def.objectId; 26259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta remoteObject.methodDefs = def.methodDefs; 26359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta remoteObjects.put(def.objectName, remoteObject); 26459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta remoteObjectsById.put(def.objectId, remoteObject); 26559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 26659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 26759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta synchronized (receiveObjectLock){ 26859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta receiveObjectLock.notifyAll(); 26959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 27059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else if (message instanceof RemoteMethodCallMessage){ 27159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodCallMessage call = (RemoteMethodCallMessage) message; 27259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta LocalObject localObj = localObjects.get(call.objectId); 27359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (localObj == null) 27459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return; 27559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 27659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (call.methodId < 0 || call.methodId >= localObj.methods.length) 27759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return; 27859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 27959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object obj = localObj.theObject; 28059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Method method = localObj.methods[call.methodId]; 28159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object[] args = call.args; 28259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object ret = null; 28359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 28459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ret = method.invoke(obj, args); 28559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch (IllegalAccessException ex){ 28659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.WARNING, "RMI: Error accessing method", ex); 28759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch (IllegalArgumentException ex){ 28859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.WARNING, "RMI: Invalid arguments", ex); 28959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch (InvocationTargetException ex){ 29059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.WARNING, "RMI: Invocation exception", ex); 29159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 29259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 29359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (method.getReturnType() != void.class){ 29459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // send return value back 29559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodReturnMessage retMsg = new RemoteMethodReturnMessage(); 29659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta retMsg.invocationID = call.invocationId; 29759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta retMsg.retVal = ret; 29859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (server != null){ 29959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta source.send(retMsg); 30059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Server: Sending {0}", retMsg); 30159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } else{ 30259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta client.send(retMsg); 30359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Client: Sending {0}", retMsg); 30459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 30559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 30659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else if (message instanceof RemoteMethodReturnMessage){ 30759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteMethodReturnMessage retMsg = (RemoteMethodReturnMessage) message; 30859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Invocation invoke = pendingInvocations.get(retMsg.invocationID); 30959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (invoke == null){ 31059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.WARNING, "Cannot find invocation ID: {0}", retMsg.invocationID); 31159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return; 31259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 31359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 31459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta synchronized (invoke){ 31559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta invoke.retVal = retMsg.retVal; 31659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta invoke.available = true; 31759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta invoke.notifyAll(); 31859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 31959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 32059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 32159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 32259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private void onConnection(HostedConnection conn) { 32359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (localObjects.size() > 0){ 32459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // send a object definition message 32559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ObjectDef[] defs = new ObjectDef[localObjects.size()]; 32659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int i = 0; 32759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (Entry<LocalObject> entry : localObjects){ 32859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defs[i] = makeObjectDef(entry.getValue()); 32959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta i++; 33059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 33159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 33259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta RemoteObjectDefMessage defMsg = new RemoteObjectDefMessage(); 33359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defMsg.objects = defs; 33459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (this.client != null){ 33559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.client.send(defMsg); 33659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Client: Sending {0}", defMsg); 33759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } else{ 33859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta conn.send(defMsg); 33959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.INFO, "Server: Sending {0}", defMsg); 34059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 34159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 34259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 34359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 34459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta} 345