org/objectweb/proactive/core/runtime/ProActiveRuntimeAdapterImpl.java

/* 
 * ################################################################
 * 
 * ProActive: The Java(TM) library for Parallel, Distributed, 
 *            Concurrent computing with Security and Mobility
 * 
 * Copyright (C) 1997-2007 INRIA/University of Nice-Sophia Antipolis
 * Contact: proactive@objectweb.org
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or any later version.
 *  
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 * USA
 *  
 *  Initial developer(s):               The ProActive Team
 *                        http://www.inria.fr/oasis/ProActive/contacts.html
 *  Contributor(s): 
 * 
 * ################################################################
 */ 
package org.objectweb.proactive.core.runtime;

import java.io.IOException;
import java.io.ObjectStreamException;
import java.io.Serializable;
import java.lang.reflect.InvocationTargetException;
import java.rmi.AlreadyBoundException;
import java.rmi.UnmarshalException;
import java.security.PublicKey;
import java.security.cert.X509Certificate;
import java.util.ArrayList;

import org.objectweb.proactive.Body;
import org.objectweb.proactive.core.ProActiveException;
import org.objectweb.proactive.core.UniqueRuntimeID;
import org.objectweb.proactive.core.body.UniversalBody;
import org.objectweb.proactive.core.body.ft.checkpointing.Checkpoint;
import org.objectweb.proactive.core.config.ProActiveConfiguration;
import org.objectweb.proactive.core.descriptor.data.ProActiveDescriptor;
import org.objectweb.proactive.core.descriptor.data.VirtualNode;
import org.objectweb.proactive.core.mop.ConstructorCall;
import org.objectweb.proactive.core.mop.ConstructorCallExecutionFailedException;
import org.objectweb.proactive.core.node.NodeException;
import org.objectweb.proactive.core.process.ExternalProcess;
import org.objectweb.proactive.core.process.UniversalProcess;
import org.objectweb.proactive.ext.security.Communication;
import org.objectweb.proactive.ext.security.ProActiveSecurityManager;
import org.objectweb.proactive.ext.security.SecurityContext;
import org.objectweb.proactive.ext.security.crypto.KeyExchangeException;
import org.objectweb.proactive.ext.security.exceptions.RenegotiateSessionException;
import org.objectweb.proactive.ext.security.exceptions.SecurityNotAvailableException;
import org.objectweb.proactive.ext.security.securityentity.Entity;


public class ProActiveRuntimeAdapterImpl extends ProActiveRuntimeAdapter
    implements Serializable {
    protected RemoteProActiveRuntime proActiveRuntime;
    protected String proActiveRuntimeURL;
    private UniqueRuntimeID urid; // Cached for speed issue

    //this boolean is used when killing the runtime. Indeed in case of co-allocation, we avoid a second call to the runtime
    // which is already dead
    protected boolean alreadykilled = false;

    //
    // -- Constructors -----------------------------------------------
    //
    public ProActiveRuntimeAdapterImpl() {
    }

    public ProActiveRuntimeAdapterImpl(RemoteProActiveRuntime proActiveRuntime)
        throws ProActiveException {
        this.proActiveRuntime = proActiveRuntime;

        try {
            this.vmInformation = proActiveRuntime.getVMInformation();
            this.proActiveRuntimeURL = proActiveRuntime.getURL();
            this.urid = new UniqueRuntimeID(proActiveRuntime.getVMInformation()
                                                            .getName(),
                    proActiveRuntime.getVMInformation().getVMID());
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    protected Object readResolve() throws ObjectStreamException {
        if (ProActiveConfiguration.isForwarder()) {
            ProActiveRuntimeForwarderImpl partf = (ProActiveRuntimeForwarderImpl) ProActiveRuntimeImpl.getProActiveRuntime();

            if (!partf.registeredRuntimes.containsKey(urid)) {
                partf.registeredRuntimes.put(urid, this);
            }

            try {
                ProActiveRuntime part = RuntimeFactory.getDefaultRuntime();
                ProActiveRuntimeAdapterForwarderImpl partAdapter = (ProActiveRuntimeAdapterForwarderImpl) part;

                return new ProActiveRuntimeAdapterForwarderImpl(partAdapter,
                    this);
            } catch (ProActiveException e) {
                runtimeLogger.warn(e.getMessage());

                return this;
            }
        }

        return this;
    }

    //
    // -- PUBLIC METHODS -----------------------------------------------
    //
    public boolean equals(Object o) {
        if (!(o instanceof ProActiveRuntimeAdapterImpl)) {
            return false;
        }

        ProActiveRuntimeAdapterImpl runtime = (ProActiveRuntimeAdapterImpl) o;

        return proActiveRuntime.equals(runtime.proActiveRuntime);
    }

    public int hashCode() {
        return proActiveRuntime.hashCode();
    }

    //
    // -- Implements ProActiveRuntime -----------------------------------------------
    //

    public String createLocalNode(String nodeName,
        boolean replacePreviousBinding, ProActiveSecurityManager policyServer,
        String vnName, String jobId)
        throws NodeException, AlreadyBoundException {
        try {
            return proActiveRuntime.createLocalNode(nodeName,
                replacePreviousBinding, policyServer, vnName, jobId);
        } catch (IOException e) {
            throw new NodeException(e);
        }
    }

    public void killAllNodes() throws ProActiveException {
        try {
            proActiveRuntime.killAllNodes();
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void killNode(String nodeName) throws ProActiveException {
        try {
            proActiveRuntime.killNode(nodeName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void createVM(UniversalProcess remoteProcess)
        throws IOException, ProActiveException {
        proActiveRuntime.createVM(remoteProcess);
    }

    public String[] getLocalNodeNames() throws ProActiveException {
        try {
            return proActiveRuntime.getLocalNodeNames();
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public VMInformation getVMInformation() {
        return vmInformation;
    }

    public void register(ProActiveRuntime proActiveRuntimeDist,
        String proActiveRuntimeUrl, String creatorID, String creationProtocol,
        String vmName) throws ProActiveException {
        try {
            proActiveRuntime.register(proActiveRuntimeDist,
                proActiveRuntimeUrl, creatorID, creationProtocol, vmName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void unregister(ProActiveRuntime proActiveRuntimeDist,
        String proActiveRuntimeUrl, String creatorID, String creationProtocol,
        String vmName) throws ProActiveException {
        try {
            this.proActiveRuntime.unregister(proActiveRuntimeDist,
                proActiveRuntimeUrl, creatorID, creationProtocol, vmName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public ProActiveRuntime[] getProActiveRuntimes() throws ProActiveException {
        try {
            return proActiveRuntime.getProActiveRuntimes();
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public ProActiveRuntime getProActiveRuntime(String proActiveRuntimeName)
        throws ProActiveException {
        try {
            return proActiveRuntime.getProActiveRuntime(proActiveRuntimeName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void addAcquaintance(String proActiveRuntimeName)
        throws ProActiveException {
        try {
            proActiveRuntime.addAcquaintance(proActiveRuntimeName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public String[] getAcquaintances() throws ProActiveException {
        try {
            return proActiveRuntime.getAcquaintances();
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void rmAcquaintance(String proActiveRuntimeName)
        throws ProActiveException {
        try {
            proActiveRuntime.rmAcquaintance(proActiveRuntimeName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void killRT(boolean softly) throws Exception {
        try {
            if (!alreadykilled) {
                proActiveRuntime.killRT(softly);
            }

            alreadykilled = true;
        } catch (UnmarshalException e) {
            //here should be caught the exception from System.exit
            alreadykilled = true;
            throw e;
        }
    }

    public String getURL() {
        return proActiveRuntimeURL;
    }

    public ArrayList getActiveObjects(String nodeName)
        throws ProActiveException {
        try {
            return proActiveRuntime.getActiveObjects(nodeName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public ArrayList getActiveObjects(String nodeName, String className)
        throws ProActiveException {
        try {
            return proActiveRuntime.getActiveObjects(nodeName, className);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public VirtualNode getVirtualNode(String virtualNodeName)
        throws ProActiveException {
        try {
            return proActiveRuntime.getVirtualNode(virtualNodeName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void registerVirtualNode(String virtualNodeName,
        boolean replacePreviousBinding)
        throws ProActiveException, AlreadyBoundException {
        try {
            proActiveRuntime.registerVirtualNode(virtualNodeName,
                replacePreviousBinding);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void unregisterVirtualNode(String virtualNodeName)
        throws ProActiveException {
        try {
            proActiveRuntime.unregisterVirtualNode(virtualNodeName);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void unregisterAllVirtualNodes() throws ProActiveException {
        try {
            proActiveRuntime.unregisterAllVirtualNodes();
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public String getJobID(String nodeUrl) throws ProActiveException {
        try {
            return proActiveRuntime.getJobID(nodeUrl);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public UniversalBody createBody(String nodeName,
        ConstructorCall bodyConstructorCall, boolean isNodeLocal)
        throws ConstructorCallExecutionFailedException,
            InvocationTargetException, ProActiveException {
        try {
            return proActiveRuntime.createBody(nodeName, bodyConstructorCall,
                isNodeLocal);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public UniversalBody receiveBody(String nodeName, Body body)
        throws ProActiveException {
        try {
            return proActiveRuntime.receiveBody(nodeName, body);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public UniversalBody receiveCheckpoint(String nodeName, Checkpoint ckpt,
        int inc) throws ProActiveException {
        try {
            return proActiveRuntime.receiveCheckpoint(nodeName, ckpt, inc);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public String getVNName(String Nodename) throws ProActiveException {
        try {
            return proActiveRuntime.getVNName(Nodename);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public byte[] getClassDataFromThisRuntime(String className)
        throws ProActiveException {
        try {
            return proActiveRuntime.getClassDataFromThisRuntime(className);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public byte[] getClassDataFromParentRuntime(String className)
        throws ProActiveException {
        try {
            return proActiveRuntime.getClassDataFromParentRuntime(className);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public ExternalProcess getProcessToDeploy(
        ProActiveRuntime proActiveRuntimeDist, String creatorID, String vmName,
        String padURL) throws ProActiveException {
        try {
            return proActiveRuntime.getProcessToDeploy(proActiveRuntimeDist,
                creatorID, vmName, padURL);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    //
    // Implements Job Interface
    //

    public String getJobID() {
        return vmInformation.getJobID();
    }

    //
    // Implements SecurityEntity Interface
    //
    public X509Certificate getCertificate()
        throws SecurityNotAvailableException {
        try {
            return proActiveRuntime.getCertificate();
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public long startNewSession(Communication policy)
        throws SecurityNotAvailableException, RenegotiateSessionException {
        try {
            return proActiveRuntime.startNewSession(policy);
        } catch (IOException e) {
            e.printStackTrace();

            return 0;
        }
    }

    public PublicKey getPublicKey() throws SecurityNotAvailableException {
        try {
            return proActiveRuntime.getPublicKey();
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public byte[] randomValue(long sessionID, byte[] clientRandomValue)
        throws SecurityNotAvailableException, RenegotiateSessionException {
        try {
            return proActiveRuntime.randomValue(sessionID, clientRandomValue);
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public byte[][] publicKeyExchange(long sessionID, byte[] myPublicKey,
        byte[] myCertificate, byte[] signature)
        throws SecurityNotAvailableException, RenegotiateSessionException,
            KeyExchangeException {
        try {
            return proActiveRuntime.publicKeyExchange(sessionID, myPublicKey,
                myCertificate, signature);
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public byte[][] secretKeyExchange(long sessionID, byte[] encodedAESKey,
        byte[] encodedIVParameters, byte[] encodedClientMacKey,
        byte[] encodedLockData, byte[] parametersSignature)
        throws SecurityNotAvailableException, RenegotiateSessionException {
        try {
            return proActiveRuntime.secretKeyExchange(sessionID, encodedAESKey,
                encodedIVParameters, encodedClientMacKey, encodedLockData,
                parametersSignature);
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public SecurityContext getPolicy(SecurityContext securityContext)
        throws SecurityNotAvailableException {
        try {
            return proActiveRuntime.getPolicy(securityContext);
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public byte[] getCertificateEncoded() throws SecurityNotAvailableException {
        try {
            return proActiveRuntime.getCertificateEncoded();
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public ArrayList<Entity> getEntities() throws SecurityNotAvailableException {
        try {
            return proActiveRuntime.getEntities();
        } catch (IOException e) {
            e.printStackTrace();

            return null;
        }
    }

    public void terminateSession(long sessionID)
        throws SecurityNotAvailableException {
        try {
            proActiveRuntime.terminateSession(sessionID);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void launchMain(String className, String[] parameters)
        throws ClassNotFoundException, NoSuchMethodException, ProActiveException {
        try {
            proActiveRuntime.launchMain(className, parameters);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public void newRemote(String className)
        throws ClassNotFoundException, ProActiveException {
        try {
            proActiveRuntime.newRemote(className);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public ProActiveDescriptor getDescriptor(String url,
        boolean isHierarchicalSearch) throws IOException, ProActiveException {
        return proActiveRuntime.getDescriptor(url, isHierarchicalSearch);
    }

    public Object setLocalNodeProperty(String nodeName, String key, String value)
        throws ProActiveException {
        try {
            return this.proActiveRuntime.setLocalNodeProperty(nodeName, key,
                value);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }

    public String getLocalNodeProperty(String nodeName, String key)
        throws ProActiveException {
        try {
            return this.proActiveRuntime.getLocalNodeProperty(nodeName, key);
        } catch (IOException e) {
            throw new ProActiveException(e);
        }
    }
}

---