package org.jboss.axis.message;
import org.jboss.axis.AxisFault;
import org.jboss.axis.Constants;
import org.jboss.axis.Message;
import org.jboss.axis.MessageContext;
import org.jboss.axis.description.OperationDesc;
import org.jboss.axis.description.ParameterDesc;
import org.jboss.axis.description.ServiceDesc;
import org.jboss.axis.encoding.DeserializationContext;
import org.jboss.axis.encoding.SerializationContext;
import org.jboss.axis.enums.Style;
import org.jboss.axis.enums.Use;
import org.jboss.axis.handlers.soap.SOAPService;
import org.jboss.axis.soap.SOAPConstants;
import org.jboss.axis.utils.JavaUtils;
import org.jboss.axis.utils.Messages;
import org.jboss.axis.wsdl.toJava.Utils;
import org.jboss.logging.Logger;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import javax.xml.namespace.QName;
import javax.xml.soap.SOAPElement;
import javax.xml.soap.SOAPException;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Vector;
public class RPCElement extends SOAPBodyElementAxisImpl
{
private static Logger log = Logger.getLogger(RPCElement.class.getName());
protected Vector params = new Vector();
protected boolean needDeser = false;
OperationDesc[] operations = null;
public RPCElement(String namespace, String localName, String prefix, Attributes attributes, DeserializationContext context, OperationDesc[] operations) throws AxisFault
{
super(namespace, localName, prefix, attributes, context);
needDeser = true;
MessageContext msgContext = context.getMessageContext();
if (operations == null && msgContext != null)
{
SOAPService service = msgContext.getService();
if (service != null)
{
ServiceDesc serviceDesc =
service.getInitializedServiceDesc(msgContext);
String lc = Utils.xmlNameToJava(name);
if (serviceDesc == null)
{
AxisFault.makeFault(new ClassNotFoundException(Messages.getMessage("noClassForService00",
lc)));
}
operations = serviceDesc.getOperationsByName(lc);
}
}
this.operations = operations;
}
public RPCElement(String namespace, String methodName, Object[] args)
{
super(namespace, methodName);
this.setNamespaceURI(namespace);
this.name = methodName;
for (int i = 0; args != null && i < args.length; i++)
{
RPCParam rpcParam = null;
if (args[i] instanceof RPCParam)
{
rpcParam = (RPCParam)args[i];
}
else
{
String name = null;
if (name == null) name = "arg" + i;
rpcParam = new RPCParam(namespace, name, args[i]);
}
addParam(rpcParam);
try
{
addChildElement(new RPCParamElementImpl(rpcParam));
}
catch (SOAPException e)
{
log.error("Cannot ass rpc param child element", e);
}
}
}
public RPCElement(String methodName)
{
super(methodName);
this.name = methodName;
}
public String getMethodName()
{
return name;
}
public void setNeedDeser(boolean needDeser)
{
this.needDeser = needDeser;
}
public void deserialize() throws SAXException
{
needDeser = false;
MessageContext msgContext = context.getMessageContext();
Message msg = msgContext.getCurrentMessage();
SOAPConstants soapConstants = msgContext.getSOAPConstants();
boolean isResponse = ((msg != null) &&
Message.RESPONSE.equals(msg.getMessageType()));
RPCHandler rpcHandler = new RPCHandler(this, isResponse);
if (operations != null)
{
int numParams = getChildren().size();
SAXException savedException = null;
boolean acceptMissingParams = msgContext.isPropertyTrue(MessageContext.ACCEPTMISSINGPARAMS,
true);
for (int i = 0; i < operations.length; i++)
{
OperationDesc operation = operations[i];
boolean needHeaderProcessing =
needHeaderProcessing(operation, isResponse);
if (operation.getStyle() == Style.DOCUMENT ||
operation.getStyle() == Style.WRAPPED ||
operation.getUse() == Use.LITERAL ||
(acceptMissingParams ?
(operation.getNumInParams() >= numParams) :
(operation.getNumInParams() == numParams)))
{
boolean isEncoded = operation.getUse() == Use.ENCODED;
rpcHandler.setOperation(operation);
try
{
if ((operation.getStyle() == Style.DOCUMENT) &&
operation.getNumInParams() > 0)
{
context.pushElementHandler(rpcHandler);
context.setCurElement(null);
}
else
{
context.pushElementHandler(new EnvelopeHandler(rpcHandler));
context.setCurElement(this);
}
publishToHandler((org.xml.sax.ContentHandler)context);
if (needHeaderProcessing)
{
processHeaders(operation, isResponse,
context, rpcHandler);
}
boolean match = true;
for (int j = 0; j < params.size() && match; j++)
{
RPCParam rpcParam = (RPCParam)params.get(j);
Object value = rpcParam.getValue();
ParameterDesc paramDesc = rpcParam.getParamDesc();
if (paramDesc != null && paramDesc.getJavaType() != null)
{
Class sigType = paramDesc.getJavaType();
if (!JavaUtils.isConvertable(value, sigType, isEncoded))
match = false;
}
}
if (!match)
{
params = new Vector();
continue;
}
msgContext.setOperation(operation);
return;
}
catch (SAXException e)
{
savedException = e;
params = new Vector();
continue;
}
catch (AxisFault e)
{
savedException = new SAXException(e);
params = new Vector();
continue;
}
}
}
if (!msgContext.isClient() && soapConstants == SOAPConstants.SOAP12_CONSTANTS)
{
AxisFault fault = new AxisFault(Constants.FAULT_SOAP12_SENDER, "string", null, null);
fault.addFaultSubCode(Constants.FAULT_SUBCODE_BADARGS);
throw new SAXException(fault);
}
if (savedException != null)
{
throw savedException;
}
else if (!msgContext.isClient())
{
QName faultCode = new QName(Constants.FAULT_SERVER_USER);
if (soapConstants == SOAPConstants.SOAP12_CONSTANTS)
faultCode = Constants.FAULT_SOAP12_SENDER;
AxisFault fault = new AxisFault(faultCode,
null, Messages.getMessage("noSuchOperation", name), null, null, null);
throw new SAXException(fault);
}
}
if (operations != null)
{
rpcHandler.setOperation(operations[0]);
}
if (operations != null && operations.length > 0 &&
(operations[0].getStyle() == Style.DOCUMENT))
{
context.pushElementHandler(rpcHandler);
context.setCurElement(null);
}
else
{
context.pushElementHandler(new EnvelopeHandler(rpcHandler));
context.setCurElement(this);
}
publishToHandler((org.xml.sax.ContentHandler)context);
}
public RPCParam getParam(String name) throws SAXException
{
if (needDeser)
{
deserialize();
}
for (int i = 0; i < params.size(); i++)
{
RPCParam param = (RPCParam)params.elementAt(i);
if (param.getName().equals(name))
return param;
}
return null;
}
public Vector getParams() throws SAXException
{
if (needDeser)
{
deserialize();
}
return params;
}
public void addParam(RPCParam param)
{
param.setRPCCall(this);
params.addElement(param);
}
protected void outputImpl(SerializationContext context) throws Exception
{
boolean hasOperationElement = true;
MessageContext msgContext = context.getMessageContext();
if (msgContext != null)
hasOperationElement = msgContext.getOperationStyle() == Style.RPC || msgContext.getOperationStyle() == Style.WRAPPED;
boolean noParams = params.size() == 0;
if (this.needDeser && context.getCurrentMessage() != null)
noParams = getParams().size() == 0;
if (hasOperationElement || noParams)
{
if (encodingStyle != null && encodingStyle.equals(""))
{
context.registerPrefixForURI("", getNamespaceURI());
}
context.startElement(new QName(getNamespaceURI(), name), attributes);
}
for (int i = 0; i < params.size(); i++)
{
RPCParam param = (RPCParam)params.elementAt(i);
if (!hasOperationElement && encodingStyle != null && encodingStyle.equals(""))
{
String uri = param.getQName().getNamespaceURI();
String prefix = context.getPrefixForURI(uri);
context.registerPrefixForURI(prefix, uri);
}
param.serialize(context);
}
if (hasOperationElement || noParams)
{
context.endElement();
}
}
private boolean needHeaderProcessing(OperationDesc operation,
boolean isResponse)
{
ArrayList paramDescs = operation.getParameters();
if (paramDescs != null)
{
for (int j = 0; j < paramDescs.size(); j++)
{
ParameterDesc paramDesc =
(ParameterDesc)paramDescs.get(j);
if ((!isResponse && paramDesc.isInHeader()) ||
(isResponse && paramDesc.isOutHeader()))
{
return true;
}
}
}
if (isResponse &&
operation.getReturnParamDesc() != null &&
operation.getReturnParamDesc().isOutHeader())
{
return true;
}
return false;
}
private void processHeaders(OperationDesc operation,
boolean isResponse,
DeserializationContext context,
RPCHandler handler)
throws AxisFault, SAXException
{
try
{
handler.setHeaderElement(true);
SOAPElement envelope = getParentElement();
while (envelope != null &&
!(envelope instanceof SOAPEnvelopeAxisImpl))
{
envelope = envelope.getParentElement();
}
if (envelope == null)
return;
ArrayList paramDescs = operation.getParameters();
if (paramDescs != null)
{
for (int j = 0; j < paramDescs.size(); j++)
{
ParameterDesc paramDesc =
(ParameterDesc)paramDescs.get(j);
if ((!isResponse && paramDesc.isInHeader()) ||
(isResponse && paramDesc.isOutHeader()))
{
Enumeration headers = ((SOAPEnvelopeAxisImpl)envelope).
getHeadersByName(paramDesc.getQName().getNamespaceURI(),
paramDesc.getQName().getLocalPart(),
true);
while (headers != null &&
headers.hasMoreElements())
{
context.pushElementHandler(handler);
context.setCurElement(null);
((SOAPElementAxisImpl)headers.nextElement()).
publishToHandler((org.xml.sax.ContentHandler)context);
}
}
}
}
if (isResponse &&
operation.getReturnParamDesc() != null &&
operation.getReturnParamDesc().isOutHeader())
{
ParameterDesc paramDesc = operation.getReturnParamDesc();
Enumeration headers =
((SOAPEnvelopeAxisImpl)envelope).
getHeadersByName(paramDesc.getQName().getNamespaceURI(),
paramDesc.getQName().getLocalPart(),
true);
while (headers != null &&
headers.hasMoreElements())
{
context.pushElementHandler(handler);
context.setCurElement(null);
((SOAPElementAxisImpl)headers.nextElement()).
publishToHandler((org.xml.sax.ContentHandler)context);
}
}
}
finally
{
handler.setHeaderElement(false);
}
}
}