org.fabric3.binding.rmi.wire.InterfacePreProcessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fabric3-binding-rmi Show documentation
Fabric3 RMI Binding Implementation.
The newest version!
/*
 * Fabric3
 * Copyright ? 2008 Metaform Systems Limited
 *
 * This proprietary software may be used only connection with the Fabric3 license
 * (the ?License?), a copy of which is included in the software or may be
 * obtained at: http://www.metaformsystems.com/licenses/license.html.

 * Software distributed under the License is distributed on an ?as is? basis,
 * without warranties or conditions of any kind.  See the License for the
 * specific language governing permissions and limitations of use of the software.
 * This software is distributed in conjunction with other software licensed under
 * different terms.  See the separate licenses for those programs included in the
 * distribution for the permitted and restricted uses of such software.
 *
 */
package org.fabric3.binding.rmi.wire;

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.TreeSet;

import org.objectweb.asm.ClassAdapter;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.ClassWriter;
import org.objectweb.asm.MethodVisitor;
import org.osoa.sca.annotations.Remotable;

public class InterfacePreProcessor {

    private static final int VERSION_15 = 49;
    private static final Class REMOTABLE = Remotable.class;

    public static Class generateRemoteInterface(String name, InputStream in, CodeGenClassLoader cl)
            throws IOException, ClassNotFoundException {
        ClassReader cr = new ClassReader(new BufferedInputStream(in));
        ClassWriter cw = new ClassWriter(cr, 0);
        Local2RemoteInterfaceTransformer transformer = new Local2RemoteInterfaceTransformer(cw);
        cr.accept(transformer, 0);
        try {
            return cl.defineClass(name, cw.toByteArray());
        } catch (LinkageError e) {
            // the class was already defined, just return it. 
            return cl.loadClass(name);
        }
    }


    private static class Local2RemoteInterfaceTransformer extends ClassAdapter {

        private static final String REMOTE = "java/rmi/Remote";
        private static final String REMOTE_EXCEPTION = "java/rmi/RemoteException";
        private static final String[] EXCEPTIONS = new String[]{REMOTE_EXCEPTION};

        private final ClassVisitor classvisitor;

        public Local2RemoteInterfaceTransformer(ClassVisitor classVisitor) {
            super(classVisitor);
            this.classvisitor = classVisitor;
        }

        @Override
        public void visit(int version, int access, String name, String signature,
                          String superName, String[] interfaces) {
            ArrayList list = new ArrayList(interfaces.length + 1);
            for (String intf : interfaces) {
                list.add(intf);
            }
            list.add(REMOTE);
            String[] result = new String[list.size()];
            result = list.toArray(result);
            classvisitor.visit(VERSION_15, access, name, signature, superName, result);
        }

        @Override
        public MethodVisitor visitMethod(int access, String name,
                                         String desc, String signature,
                                         String[] exceptions) {
            return classvisitor.visitMethod(access, name, desc, signature,
                                            getExceptions(exceptions));
        }


        private static String[] getExceptions(String[] existing) {
            if (existing != null && existing.length > 0) {
                TreeSet set = new TreeSet();
                for (String exception : existing) {
                    set.add(exception);
                }
                set.add(REMOTE_EXCEPTION);
                String[] newExceptions = new String[set.size()];
                set.toArray(newExceptions);
                return newExceptions;
            }
            return EXCEPTIONS;
        }

    }

}