JonasHandler.java

package org.objectweb.jonas.ws.axis;


import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;

import org.apache.axis.AxisFault;
import org.apache.axis.Handler;
import org.apache.axis.MessageContext;
import org.apache.axis.handlers.BasicHandler;

import org.objectweb.jonas_ejb.container.JServiceEndpointHome;

import org.objectweb.jonas.common.Log;
import org.objectweb.jonas.security.ws.SecurityContextHelper;

import org.objectweb.util.monolog.api.BasicLevel;
import org.objectweb.util.monolog.api.Logger;


public class JonasHandler extends BasicHandler {

    private static Logger logger = Log.getLogger(Log.JONAS_WS_EJBPROVIDER_PREFIX);

    private static InitialContext cachedContext;

    private Context savedCtx = null;

    private JServiceEndpointHome sehome = null;

    public void onFault(MessageContext msgContext) {
        logger.log(BasicLevel.DEBUG, "*------* Fault");
        if (sehome != null) {
            logger.log(BasicLevel.DEBUG, "*--* Fault");
            sehome.resetCompCtx(savedCtx);
            if (savedCtx != null) {
                logger.log(BasicLevel.ERROR, "Setting a non null context:" + savedCtx);
            }
            sehome = null;
        }
    }

    public void invoke(MessageContext msgContext) throws AxisFault {

        if (msgContext.getPastPivot()) {
            logger.log(BasicLevel.DEBUG, "*------* Response");
            // Response
            if (sehome != null) {
                logger.log(BasicLevel.DEBUG, "*--* Response");
                sehome.resetCompCtx(savedCtx);
                if (savedCtx != null) {
                    logger.log(BasicLevel.ERROR, "Setting a non null context:" + savedCtx);
                }
                sehome = null;
            }
        } else {
            // Request
            logger.log(BasicLevel.DEBUG, "*------* Request");

            // Get the service class name
            Handler service = msgContext.getService();
            String clsName = (String) service.getOption(JOnASEJBProvider.OPTION_SEINTERFACENAME);
            if (clsName == null) {
                // We are not in the case of jonas ejb call: do nothing.
                return;
            }
            logger.log(BasicLevel.DEBUG, "*--* Request");

            // Get ServiceEndpointHome in JNDI
            String jndiName = (String) service.getOption(JOnASEJBProvider.OPTION_SEJNDINAME);
            if (jndiName == null) {
                logger.log(BasicLevel.ERROR, "Service Endpoint JNDI name is null");
                throw new AxisFault("Missing parameter OPTION_SEJNDINAME in service");
            }
            try {
                InitialContext ic = getCachedContext();
                sehome = (JServiceEndpointHome) ic.lookup(jndiName);
            } catch (NamingException ne) {
                logger.log(BasicLevel.ERROR, "Cannot lookup ServiceEndpointHome");
                throw new AxisFault("Cannot lookup ServiceEndpointHome: " + jndiName);
            }
            
            // add the security context
            String username = msgContext.getUsername();
            if (username != null) {
                // Do not forget to initialize the security context
                SecurityContextHelper.getInstance().login(username, msgContext.getPassword());
                // Check the security and throw exception if user is not authorized
                // to access the EJB before lauching handlers
                sehome.checkSecurity(msgContext);
            }
            
            // Switch Context on the context of the target bean (java:comp/env)
            savedCtx = sehome.setCompCtx();
            if (savedCtx != null) {
                logger.log(BasicLevel.ERROR, "Saving a non null context:" + savedCtx);
            }

        }
    }

    private InitialContext getCachedContext() throws javax.naming.NamingException {
        if (cachedContext == null) {
            cachedContext = new InitialContext();
        }
        return cachedContext;
    }
}

---