JPolicyContextHandler.java

package org.objectweb.jonas.security.jacc;

import javax.security.jacc.PolicyContextException;
import javax.security.jacc.PolicyContextHandler;

import org.objectweb.jonas.common.Log;

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

public class JPolicyContextHandler implements PolicyContextHandler {

    private static final String CONTAINER_SUBJECT = "javax.security.auth.Subject.container";

    private static final String SOAP_MESSAGE = "javax.xml.soap.SOAPMessage";

    private static final String HTTP_SERVLET_REQUEST = "javax.servlet.http.HttpServletRequest";

    private static final String ENTERPRISE_BEAN = "javax.ejb.EnterpriseBean";

    private static final String EJB_ARGUMENTS = "javax.ejb.arguments";

    private static final String[] SUPPORTED_KEYS = new String[] {CONTAINER_SUBJECT, SOAP_MESSAGE, HTTP_SERVLET_REQUEST,
            ENTERPRISE_BEAN, EJB_ARGUMENTS};

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

    public JPolicyContextHandler() {
        super();

    }

    public boolean supports(String key) throws PolicyContextException {
        for (int k = 0; k < SUPPORTED_KEYS.length; k++) {
            if (key.equals(SUPPORTED_KEYS[k])) {
                return true;
            }
        }
        return false;
    }

    public String[] getKeys() throws PolicyContextException {
        return SUPPORTED_KEYS;
    }

    public Object getContext(String key, Object data) throws PolicyContextException {

        if (logger.isLoggable(BasicLevel.DEBUG)) {
            logger.log(BasicLevel.DEBUG, "Asking key" + key);
        }

        // Handle only JOnAS HandleData
        if (data == null || !(data instanceof JPolicyContextHandlerData)) {
            logger.log(BasicLevel.WARN, "No data object or Data object not instance of JPolicyContextHandlerData");
            return null;
        }

        Object contextObject = null;
        JPolicyContextHandlerData jPolicyContextHandlerData = (JPolicyContextHandlerData) data;

        if (key.equals(HTTP_SERVLET_REQUEST)) {
            if (logger.isLoggable(BasicLevel.DEBUG)) {
                logger.log(BasicLevel.DEBUG, "Key == '" + HTTP_SERVLET_REQUEST + "'");
            }
            contextObject = jPolicyContextHandlerData.getHttpServletRequest();
        } else if (key.equals(EJB_ARGUMENTS)) {
            if (logger.isLoggable(BasicLevel.DEBUG)) {
                logger.log(BasicLevel.DEBUG, "Key == '" + EJB_ARGUMENTS + "'");
            }
            contextObject = jPolicyContextHandlerData.getEjbArguments();
        } else if (key.equals(CONTAINER_SUBJECT)) {
            if (logger.isLoggable(BasicLevel.DEBUG)) {
                logger.log(BasicLevel.DEBUG, "Key == '" + CONTAINER_SUBJECT + "'");
            }
            contextObject = jPolicyContextHandlerData.getContainerSubject();

        }

        if (logger.isLoggable(BasicLevel.DEBUG)) {
            logger.log(BasicLevel.DEBUG, "Returning object '" + contextObject + "' for key '" + key + "'");
        }
        return contextObject;

    }

}

---