/*
* JBoss, Home of Professional Open Source.
* Copyright 2006, Red Hat Middleware LLC, 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.ws.extensions.security.jaxrpc;
import java.io.IOException;
import javax.xml.namespace.QName;
import javax.xml.rpc.Stub;
import javax.xml.rpc.handler.GenericHandler;
import javax.xml.rpc.handler.MessageContext;
import javax.xml.rpc.soap.SOAPFaultException;
import javax.xml.soap.SOAPException;
import org.jboss.logging.Logger;
import org.jboss.ws.WSException;
import org.jboss.ws.core.CommonMessageContext;
import org.jboss.ws.core.soap.SOAPMessageImpl;
import org.jboss.ws.extensions.security.Constants;
import org.jboss.ws.extensions.security.WSSecurityDispatcher;
import org.jboss.ws.metadata.umdm.EndpointMetaData;
import org.jboss.ws.metadata.umdm.ServiceMetaData;
import org.jboss.ws.metadata.wsse.WSSecurityConfigFactory;
import org.jboss.ws.metadata.wsse.WSSecurityConfiguration;
import org.jboss.wsf.spi.deployment.UnifiedVirtualFile;
/**
* An abstract JAXRPC handler that delegates to the WSSecurityDispatcher
*
* @author Thomas.Diesler@jboss.org
* @since 12-Nov-2005
*/
public abstract class WSSecurityHandler extends GenericHandler
{
// provide logging
private static Logger log = Logger.getLogger(WSSecurityHandler.class);
protected static String FAULT_THROWN = "org.jboss.ws.wsse.faultThrown";
public QName[] getHeaders()
{
return new QName[] {Constants.WSSE_HEADER_QNAME};
}
protected boolean thrownByMe(MessageContext msgContext)
{
Boolean bool = (Boolean)msgContext.getProperty(FAULT_THROWN);
return bool != null && bool.booleanValue();
}
protected boolean handleInboundSecurity(MessageContext msgContext)
{
Exception exception = null;
try
{
WSSecurityConfiguration configuration = getSecurityConfiguration(msgContext);
if (configuration != null)
{
CommonMessageContext ctx = (CommonMessageContext)msgContext;
SOAPMessageImpl soapMessage = (SOAPMessageImpl)ctx.getSOAPMessage();
new WSSecurityDispatcher().decodeMessage(configuration, soapMessage, null);
}
}
catch (SOAPException ex)
{
exception = ex;
}
if (exception != null)
{
msgContext.setProperty(FAULT_THROWN, true);
if (exception instanceof SOAPFaultException)
throw (SOAPFaultException) exception;
// Unexpected exception, log it
log.error("Cannot handle inbound ws-security", exception);
return false;
}
return true;
}
protected boolean handleOutboundSecurity(MessageContext msgContext)
{
Exception exception = null;
try
{
WSSecurityConfiguration configuration = getSecurityConfiguration(msgContext);
if (configuration != null)
{
CommonMessageContext ctx = (CommonMessageContext)msgContext;
SOAPMessageImpl soapMessage = (SOAPMessageImpl)ctx.getSOAPMessage();
String user = (String)ctx.get(Stub.USERNAME_PROPERTY);
String pass = (String)ctx.get(Stub.PASSWORD_PROPERTY);
new WSSecurityDispatcher().encodeMessage(configuration, soapMessage, null, user, pass);
}
}
catch (SOAPException ex)
{
exception = ex;
}
if (exception != null)
{
msgContext.setProperty(FAULT_THROWN, true);
if (exception instanceof SOAPFaultException)
throw (SOAPFaultException) exception;
// Unexpected exception, log it
log.error("Cannot handle outbound ws-security", exception);
return false;
}
return true;
}
/**
* Load security config from vfsRoot
* @param msgContext
*/
private WSSecurityConfiguration getSecurityConfiguration(MessageContext msgContext)
{
EndpointMetaData epMetaData = ((CommonMessageContext)msgContext).getEndpointMetaData();
ServiceMetaData serviceMetaData = epMetaData.getServiceMetaData();
WSSecurityConfiguration config = serviceMetaData.getSecurityConfiguration();
if (config == null) // might be set through ServiceObjectFactory
{
try
{
WSSecurityConfigFactory wsseConfFactory = WSSecurityConfigFactory.newInstance();
UnifiedVirtualFile vfsRoot = serviceMetaData.getUnifiedMetaData().getRootFile();
config = wsseConfFactory.createConfiguration(vfsRoot, getConfigResourceName());
}
catch (IOException e)
{
throw new WSException("Cannot obtain security config: " + getConfigResourceName());
}
// it's required further down the processing chain
serviceMetaData.setSecurityConfiguration(config);
}
return config;
}
protected abstract String getConfigResourceName();
}