/*
* JBoss, Home of Professional Open Source.
* Copyright 2012, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This 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 (at your option) any later version.
*
* This software 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 software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.ejb3.iiop.stub;
import org.jboss.as.ejb3.EjbLogger;
import org.jboss.as.ejb3.EjbMessages;
import org.jboss.classfilewriter.ClassFile;
import org.jboss.com.sun.corba.se.impl.presentation.rmi.StubFactoryBase;
import org.jboss.com.sun.corba.se.impl.presentation.rmi.StubFactoryFactoryDynamicBase;
import org.jboss.com.sun.corba.se.spi.presentation.rmi.PresentationManager;
/**
* @author Stuart Douglas
*/
public class DynamicStubFactoryFactory extends StubFactoryFactoryDynamicBase {
@Override
public PresentationManager.StubFactory makeDynamicStubFactory(final PresentationManager pm, final PresentationManager.ClassData classData, final ClassLoader classLoader) {
final Class<?> myClass = classData.getMyClass();
Class<?> theClass = makeStubClass(myClass);
return new StubFactory(classData, theClass);
}
/**
* Makes a dynamic stub class, if it does not already exist.
* @param myClass The class to create a stub for
* @return The dynamic stub class
*/
public static Class<?> makeStubClass(final Class<?> myClass) {
final String stubClassName = myClass + "_Stub";
ClassLoader cl = SecurityActions.getContextClassLoader();
if (cl == null) {
cl = myClass.getClassLoader();
}
if (cl == null) {
throw EjbMessages.MESSAGES.couldNotFindClassLoaderForStub(stubClassName);
}
Class<?> theClass;
try {
theClass = cl.loadClass(stubClassName);
} catch (ClassNotFoundException e) {
try {
final ClassFile clazz = IIOPStubCompiler.compile(myClass, stubClassName);
theClass = clazz.define(cl);
} catch (RuntimeException ex) {
//there is a possibility that another thread may have defined the same class in the meantime
try {
theClass = cl.loadClass(stubClassName);
} catch (ClassNotFoundException e1) {
EjbLogger.EJB3_LOGGER.dynamicStubCreationFailed(stubClassName, ex);
throw ex;
}
}
}
return theClass;
}
private static final class StubFactory extends StubFactoryBase {
private final Class<?> clazz;
protected StubFactory(PresentationManager.ClassData classData, final Class<?> clazz) {
super(classData);
this.clazz = clazz;
}
@Override
public org.omg.CORBA.Object makeStub() {
try {
return (org.omg.CORBA.Object) clazz.newInstance();
} catch (InstantiationException e) {
throw new RuntimeException(e);
} catch (IllegalAccessException e) {
throw new RuntimeException(e);
}
}
}
}