/*
* Copyright (c) 2005, 2006, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code 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 General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package com.sun.xml.internal.ws.api.addressing;
import com.sun.istack.internal.NotNull;
import com.sun.istack.internal.Nullable;
import com.sun.xml.internal.stream.buffer.MutableXMLStreamBuffer;
import com.sun.xml.internal.stream.buffer.XMLStreamBuffer;
import com.sun.xml.internal.stream.buffer.XMLStreamBufferResult;
import com.sun.xml.internal.stream.buffer.XMLStreamBufferSource;
import com.sun.xml.internal.stream.buffer.sax.SAXBufferProcessor;
import com.sun.xml.internal.stream.buffer.stax.StreamReaderBufferProcessor;
import com.sun.xml.internal.stream.buffer.stax.StreamWriterBufferCreator;
import com.sun.xml.internal.ws.addressing.EndpointReferenceUtil;
import com.sun.xml.internal.ws.addressing.W3CAddressingConstants;
import com.sun.xml.internal.ws.addressing.model.InvalidAddressingHeaderException;
import com.sun.xml.internal.ws.addressing.v200408.MemberSubmissionAddressingConstants;
import com.sun.xml.internal.ws.api.message.Header;
import com.sun.xml.internal.ws.api.message.HeaderList;
import com.sun.xml.internal.ws.api.message.Message;
import com.sun.xml.internal.ws.api.streaming.XMLStreamReaderFactory;
import com.sun.xml.internal.ws.resources.AddressingMessages;
import com.sun.xml.internal.ws.resources.ClientMessages;
import com.sun.xml.internal.ws.spi.ProviderImpl;
import com.sun.xml.internal.ws.streaming.XMLStreamReaderUtil;
import com.sun.xml.internal.ws.util.DOMUtil;
import com.sun.xml.internal.ws.util.xml.XMLStreamWriterFilter;
import com.sun.xml.internal.ws.util.xml.XmlUtil;
import com.sun.xml.internal.ws.wsdl.parser.WSDLConstants;
import org.w3c.dom.Element;
import org.xml.sax.Attributes;
import org.xml.sax.ContentHandler;
import org.xml.sax.ErrorHandler;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.XMLFilterImpl;
import javax.xml.bind.JAXBContext;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;
import javax.xml.transform.Source;
import javax.xml.transform.TransformerException;
import javax.xml.transform.sax.SAXSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.ws.Dispatch;
import javax.xml.ws.EndpointReference;
import javax.xml.ws.Service;
import javax.xml.ws.WebServiceException;
import javax.xml.ws.WebServiceFeature;
import java.io.InputStream;
import java.io.StringWriter;
import java.net.URI;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
/**
* Internal representation of the EPR.
*
* <p>
* Instances of this class are immutable and thread-safe.
*
* @author Kohsuke Kawaguchi
* @see AddressingVersion#anonymousEpr
*/
public final class WSEndpointReference {
private final XMLStreamBuffer infoset;
/**
* Version of the addressing spec.
*/
private final AddressingVersion version;
/**
* Marked Reference parameters inside this EPR.
*
* Parsed when the object is created. can be empty but never null.
* @see #parse()
*/
private @NotNull Header[] referenceParameters;
private @NotNull String address;
/**
* Creates from the spec version of {@link EndpointReference}.
*
* <p>
* This method performs the data conversion, so it's slow.
* Do not use this method in a performance critical path.
*/
public WSEndpointReference(EndpointReference epr, AddressingVersion version) {
try {
MutableXMLStreamBuffer xsb = new MutableXMLStreamBuffer();
epr.writeTo(new XMLStreamBufferResult(xsb));
this.infoset = xsb;
this.version = version;
parse();
} catch (XMLStreamException e) {
throw new WebServiceException(ClientMessages.FAILED_TO_PARSE_EPR(epr),e);
}
}
/**
* Creates from the spec version of {@link EndpointReference}.
*
* <p>
* This method performs the data conversion, so it's slow.
* Do not use this method in a performance critical path.
*/
public WSEndpointReference(EndpointReference epr) {
this(epr,AddressingVersion.fromSpecClass(epr.getClass()));
}
/**
* Creates a {@link WSEndpointReference} that wraps a given infoset.
*/
public WSEndpointReference(XMLStreamBuffer infoset, AddressingVersion version) {
try {
this.infoset = infoset;
this.version = version;
parse();
} catch (XMLStreamException e) {
// this can never happen because XMLStreamBuffer never has underlying I/O error.
throw new AssertionError(e);
}
}
/**
* Creates a {@link WSEndpointReference} by parsing an infoset.
*/
public WSEndpointReference(InputStream infoset, AddressingVersion version) throws XMLStreamException {
this(XMLStreamReaderFactory.create(null,infoset,false),version);
}
/**
* Creates a {@link WSEndpointReference} from the given infoset.
* The {@link XMLStreamReader} must point to either a document or an element.
*/
public WSEndpointReference(XMLStreamReader in, AddressingVersion version) throws XMLStreamException {
this(XMLStreamBuffer.createNewBufferFromXMLStreamReader(in), version);
}
/**
* @see #WSEndpointReference(String, AddressingVersion)
*/
public WSEndpointReference(URL address, AddressingVersion version) {
this(address.toExternalForm(), version);
}
/**
* @see #WSEndpointReference(String, AddressingVersion)
*/
public WSEndpointReference(URI address, AddressingVersion version) {
this(address.toString(), version);
}
/**
* Creates a {@link WSEndpointReference} that only has an address.
*/
public WSEndpointReference(String address, AddressingVersion version) {
this.infoset = createBufferFromAddress(address,version);
this.version = version;
this.address = address;
this.referenceParameters = EMPTY_ARRAY;
}
private static XMLStreamBuffer createBufferFromAddress(String address, AddressingVersion version) {
try {
MutableXMLStreamBuffer xsb = new MutableXMLStreamBuffer();
StreamWriterBufferCreator w = new StreamWriterBufferCreator(xsb);
w.writeStartDocument();
w.writeStartElement(version.getPrefix(),
"EndpointReference", version.nsUri);
w.writeNamespace(version.getPrefix(),
version.nsUri);
w.writeStartElement(version.getPrefix(),
W3CAddressingConstants.WSA_ADDRESS_NAME, version.nsUri);
w.writeCharacters(address);
w.writeEndElement();
w.writeEndElement();
w.writeEndDocument();
w.close();
return xsb;
} catch (XMLStreamException e) {
// can never happen because we are writing to XSB
throw new AssertionError(e);
}
}
/**
* Creates an EPR from individual components.
*
* <p>
* This version takes various information about metadata, and creates an EPR that has
* the necessary embedded WSDL.
*/
public WSEndpointReference(@NotNull AddressingVersion version,
@NotNull String address,
@Nullable QName service,
@Nullable QName port,
@Nullable QName portType,
@Nullable List<Element> metadata,
@Nullable String wsdlAddress,
@Nullable List<Element> referenceParameters) {
this(
createBufferFromData(version, address, referenceParameters, service, port, portType, metadata, wsdlAddress),
version );
}
private static XMLStreamBuffer createBufferFromData(AddressingVersion version, String address, List<Element> referenceParameters, QName service, QName port, QName portType, List<Element> metadata, String wsdlAddress) {
StreamWriterBufferCreator writer = new StreamWriterBufferCreator();
try {
writer.writeStartDocument();
writer.writeStartElement(version.getPrefix(),"EndpointReference", version.nsUri);
writer.writeNamespace(version.getPrefix(),version.nsUri);
writer.writeStartElement(version.getPrefix(),"Address", version.nsUri);
writer.writeCharacters(address);
writer.writeEndElement();
if(referenceParameters != null) {
writer.writeStartElement(version.getPrefix(),"ReferenceParameters", version.nsUri);
for (Element e : referenceParameters)
DOMUtil.serializeNode(e, writer);
writer.writeEndElement();
}
switch(version) {
case W3C:
writeW3CMetaData(writer, service, port, portType, metadata, wsdlAddress);
break;
case MEMBER:
writeMSMetaData(writer, service, port, portType, metadata);
if (wsdlAddress != null) {
//Inline the wsdl as extensibility element
//Write mex:Metadata wrapper
writer.writeStartElement(MemberSubmissionAddressingConstants.MEX_METADATA.getPrefix(),
MemberSubmissionAddressingConstants.MEX_METADATA.getLocalPart(),
MemberSubmissionAddressingConstants.MEX_METADATA.getNamespaceURI());
writer.writeStartElement(MemberSubmissionAddressingConstants.MEX_METADATA_SECTION.getPrefix(),
MemberSubmissionAddressingConstants.MEX_METADATA_SECTION.getLocalPart(),
MemberSubmissionAddressingConstants.MEX_METADATA_SECTION.getNamespaceURI());
writer.writeAttribute(MemberSubmissionAddressingConstants.MEX_METADATA_DIALECT_ATTRIBUTE,
MemberSubmissionAddressingConstants.MEX_METADATA_DIALECT_VALUE);
writeWsdl(writer, service, wsdlAddress);
writer.writeEndElement();
writer.writeEndElement();
}
break;
}
writer.writeEndElement();
writer.writeEndDocument();
writer.flush();
return writer.getXMLStreamBuffer();
} catch (XMLStreamException e) {
throw new WebServiceException(e);
}
}
private static void writeW3CMetaData(StreamWriterBufferCreator writer,
QName service,
QName port,
QName portType, List<Element> metadata,
String wsdlAddress) throws XMLStreamException {
writer.writeStartElement(AddressingVersion.W3C.getPrefix(),
W3CAddressingConstants.WSA_METADATA_NAME, AddressingVersion.W3C.nsUri);
writer.writeNamespace(AddressingVersion.W3C.getWsdlPrefix(),
AddressingVersion.W3C.wsdlNsUri);
//Write Interface info
if (portType != null) {
writer.writeStartElement(AddressingVersion.W3C.getWsdlPrefix(),
W3CAddressingConstants.WSAW_INTERFACENAME_NAME,
AddressingVersion.W3C.wsdlNsUri);
String portTypePrefix = portType.getPrefix();
if (portTypePrefix == null || portTypePrefix.equals("")) {
//TODO check prefix again
portTypePrefix = "wsns";
}
writer.writeNamespace(portTypePrefix, portType.getNamespaceURI());
writer.writeCharacters(portTypePrefix + ":" + portType.getLocalPart());
writer.writeEndElement();
}
if (service != null) {
//Write service and Port info
if (!(service.getNamespaceURI().equals("") || service.getLocalPart().equals(""))) {
writer.writeStartElement(AddressingVersion.W3C.getWsdlPrefix(),
W3CAddressingConstants.WSAW_SERVICENAME_NAME,
AddressingVersion.W3C.wsdlNsUri);
String servicePrefix = service.getPrefix();
if (servicePrefix == null || servicePrefix.equals("")) {
//TODO check prefix again
servicePrefix = "wsns";
}
writer.writeNamespace(servicePrefix, service.getNamespaceURI());
if (port != null) {
writer.writeAttribute(W3CAddressingConstants.WSAW_ENDPOINTNAME_NAME, port.getLocalPart());
}
writer.writeCharacters(servicePrefix + ":" + service.getLocalPart());
writer.writeEndElement();
}
}
//Inline the wsdl
if (wsdlAddress != null) {
writeWsdl(writer, service, wsdlAddress);
}
//Add the extra metadata Elements
if (metadata != null)
for (Element e : metadata) {
DOMUtil.serializeNode(e, writer);
}
writer.writeEndElement();
}
private static void writeMSMetaData(StreamWriterBufferCreator writer,
QName service,
QName port,
QName portType, List<Element> metadata) throws XMLStreamException {
// TODO: write ReferenceProperties
//TODO: write ReferenceParameters
if (portType != null) {
//Write Interface info
writer.writeStartElement(AddressingVersion.MEMBER.getPrefix(),
MemberSubmissionAddressingConstants.WSA_PORTTYPE_NAME,
AddressingVersion.MEMBER.nsUri);
String portTypePrefix = portType.getPrefix();
if (portTypePrefix == null || portTypePrefix.equals("")) {
//TODO check prefix again
portTypePrefix = "wsns";
}
writer.writeNamespace(portTypePrefix, portType.getNamespaceURI());
writer.writeCharacters(portTypePrefix + ":" + portType.getLocalPart());
writer.writeEndElement();
}
//Write service and Port info
if (service != null) {
if (!(service.getNamespaceURI().equals("") || service.getLocalPart().equals(""))) {
writer.writeStartElement(AddressingVersion.MEMBER.getPrefix(),
MemberSubmissionAddressingConstants.WSA_SERVICENAME_NAME,
AddressingVersion.MEMBER.nsUri);
String servicePrefix = service.getPrefix();
if (servicePrefix == null || servicePrefix.equals("")) {
//TODO check prefix again
servicePrefix = "wsns";
}
writer.writeNamespace(servicePrefix, service.getNamespaceURI());
if (port != null) {
writer.writeAttribute(MemberSubmissionAddressingConstants.WSA_PORTNAME_NAME,
port.getLocalPart());
}
writer.writeCharacters(servicePrefix + ":" + service.getLocalPart());
writer.writeEndElement();
}
}
}
private static void writeWsdl(StreamWriterBufferCreator writer, QName service, String wsdlAddress) throws XMLStreamException {
// Inline-wsdl
writer.writeStartElement(WSDLConstants.PREFIX_NS_WSDL,
WSDLConstants.QNAME_DEFINITIONS.getLocalPart(),
WSDLConstants.NS_WSDL);
writer.writeNamespace(WSDLConstants.PREFIX_NS_WSDL, WSDLConstants.NS_WSDL);
writer.writeStartElement(WSDLConstants.PREFIX_NS_WSDL,
WSDLConstants.QNAME_IMPORT.getLocalPart(),
WSDLConstants.NS_WSDL);
writer.writeAttribute("namespace", service.getNamespaceURI());
writer.writeAttribute("location", wsdlAddress);
writer.writeEndElement();
writer.writeEndElement();
}
/**
* Converts from {@link EndpointReference}.
*
* This handles null {@link EndpointReference} correctly.
* Call {@link #WSEndpointReference(EndpointReference)} directly
* if you know it's not null.
*/
public static @Nullable
WSEndpointReference create(@Nullable EndpointReference epr) {
if (epr != null)
return new WSEndpointReference(epr);
else
return null;
}
/**
* @see #createWithAddress(String)
*/
public @NotNull WSEndpointReference createWithAddress(@NotNull URI newAddress) {
return createWithAddress(newAddress.toString());
}
/**
* @see #createWithAddress(String)
*/
public @NotNull WSEndpointReference createWithAddress(@NotNull URL newAddress) {
return createWithAddress(newAddress.toString());
}
/**
* Creates a new {@link WSEndpointReference} by replacing the address of this EPR
* to the new one.
*
* <p>
* The following example shows how you can use this to force an HTTPS EPR,
* when the endpoint can serve both HTTP and HTTPS requests.
* <pre>
* if(epr.getAddress().startsWith("http:"))
* epr = epr.createWithAddress("https:"+epr.getAddress().substring(5));
* </pre>
*
* @param newAddress
* This is a complete URL to be written inside <Adress> element of the EPR,
* such as "http://foo.bar/abc/def"
*/
public @NotNull WSEndpointReference createWithAddress(@NotNull final String newAddress) {
MutableXMLStreamBuffer xsb = new MutableXMLStreamBuffer();
XMLFilterImpl filter = new XMLFilterImpl() {
private boolean inAddress = false;
public void startElement(String uri, String localName, String qName, Attributes atts) throws SAXException {
if(localName.equals("Address") && uri.equals(version.nsUri))
inAddress = true;
super.startElement(uri,localName,qName,atts);
}
public void characters(char ch[], int start, int length) throws SAXException {
if(!inAddress)
super.characters(ch, start, length);
}
public void endElement(String uri, String localName, String qName) throws SAXException {
if(inAddress)
super.characters(newAddress.toCharArray(),0,newAddress.length());
inAddress = false;
super.endElement(uri, localName, qName);
}
};
filter.setContentHandler(xsb.createFromSAXBufferCreator());
try {
infoset.writeTo(filter,false);
} catch (SAXException e) {
throw new AssertionError(e); // impossible since we are writing from XSB to XSB.
}
return new WSEndpointReference(xsb,version);
}
/**
* Convert the EPR to the spec version. The actual type of
* {@link EndpointReference} to be returned depends on which version
* of the addressing spec this EPR conforms to.
*
* @throws WebServiceException
* if the conversion fails, which can happen if the EPR contains
* invalid infoset (wrong namespace URI, etc.)
*/
public @NotNull EndpointReference toSpec() {
return ProviderImpl.INSTANCE.readEndpointReference(asSource("EndpointReference"));
}
/**
* Converts the EPR to the specified spec version.
*
* If the {@link #getVersion() the addressing version in use} and
* the given class is different, then this may involve version conversion.
*/
public @NotNull <T extends EndpointReference> T toSpec(Class<T> clazz) {
return EndpointReferenceUtil.transform(clazz,toSpec());
}
/**
* Creates a proxy that can be used to talk to this EPR.
*
* <p>
* All the normal WS-Addressing processing happens automatically,
* such as setting the endpoint address to {@link #getAddress() the address},
* and sending the reference parameters associated with this EPR as
* headers, etc.
*/
public @NotNull <T> T getPort(@NotNull Service jaxwsService,
@NotNull Class<T> serviceEndpointInterface,
WebServiceFeature... features) {
// TODO: implement it in a better way
return jaxwsService.getPort(toSpec(),serviceEndpointInterface,features);
}
/**
* Creates a {@link Dispatch} that can be used to talk to this EPR.
*
* <p>
* All the normal WS-Addressing processing happens automatically,
* such as setting the endpoint address to {@link #getAddress() the address},
* and sending the reference parameters associated with this EPR as
* headers, etc.
*/
public @NotNull <T> Dispatch<T> createDispatch(
@NotNull Service jaxwsService,
@NotNull Class<T> type,
@NotNull Service.Mode mode,
WebServiceFeature... features) {
// TODO: implement it in a better way
return jaxwsService.createDispatch(toSpec(),type,mode,features);
}
/**
* Creates a {@link Dispatch} that can be used to talk to this EPR.
*
* <p>
* All the normal WS-Addressing processing happens automatically,
* such as setting the endpoint address to {@link #getAddress() the address},
* and sending the reference parameters associated with this EPR as
* headers, etc.
*/
public @NotNull Dispatch<Object> createDispatch(
@NotNull Service jaxwsService,
@NotNull JAXBContext context,
@NotNull Service.Mode mode,
WebServiceFeature... features) {
// TODO: implement it in a better way
return jaxwsService.createDispatch(toSpec(),context,mode,features);
}
/**
* Gets the addressing version of this EPR.
*/
public @NotNull AddressingVersion getVersion() {
return version;
}
/**
* The value of the <wsa:address> header.
*/
public @NotNull String getAddress() {
return address;
}
/**
* Returns true if this has anonymous URI as the {@link #getAddress() address}.
*/
public boolean isAnonymous() {
return address.equals(version.anonymousUri);
}
/**
* Returns true if this has {@link AddressingVersion#noneUri none URI}
* as the {@link #getAddress() address}.
*/
public boolean isNone() {
return address.equals(version.noneUri);
}
/**
* Parses inside EPR and mark all reference parameters.
*/
private void parse() throws XMLStreamException {
// TODO: validate the EPR structure.
// check for non-existent Address, that sort of things.
StreamReaderBufferProcessor xsr = infoset.readAsXMLStreamReader();
// parser should be either at the start element or the start document
if(xsr.getEventType()==XMLStreamReader.START_DOCUMENT)
xsr.nextTag();
assert xsr.getEventType()==XMLStreamReader.START_ELEMENT;
String rootLocalName = xsr.getLocalName();
if(!xsr.getNamespaceURI().equals(version.nsUri))
throw new WebServiceException(AddressingMessages.WRONG_ADDRESSING_VERSION(
version.nsUri, xsr.getNamespaceURI()));
// since often EPR doesn't have a reference parameter, create array lazily
List<Header> marks=null;
while(xsr.nextTag()==XMLStreamReader.START_ELEMENT) {
String localName = xsr.getLocalName();
if(version.isReferenceParameter(localName)) {
XMLStreamBuffer mark;
while((mark = xsr.nextTagAndMark())!=null) {
if(marks==null)
marks = new ArrayList<Header>();
// TODO: need a different header for member submission version
marks.add(version.createReferenceParameterHeader(
mark, xsr.getNamespaceURI(), xsr.getLocalName()));
XMLStreamReaderUtil.skipElement(xsr);
}
} else
if(localName.equals("Address")) {
if(address!=null) // double <Address>. That's an error.
throw new InvalidAddressingHeaderException(new QName(version.nsUri,rootLocalName),AddressingVersion.fault_duplicateAddressInEpr);
address = xsr.getElementText().trim();
} else {
XMLStreamReaderUtil.skipElement(xsr);
}
}
// hit to </EndpointReference> by now
if(marks==null) {
this.referenceParameters = EMPTY_ARRAY;
} else {
this.referenceParameters = marks.toArray(new Header[marks.size()]);
}
if(address==null)
throw new InvalidAddressingHeaderException(new QName(version.nsUri,rootLocalName),version.fault_missingAddressInEpr);
}
/**
* Reads this EPR as {@link XMLStreamReader}.
*
* @param localName
* EPR uses a different root tag name depending on the context.
* The returned {@link XMLStreamReader} will use the given local name
* for the root element name.
*/
public XMLStreamReader read(final @NotNull String localName) throws XMLStreamException {
return new StreamReaderBufferProcessor(infoset) {
protected void processElement(String prefix, String uri, String _localName) {
if (_depth == 0)
_localName = localName;
super.processElement(prefix, uri, _localName);
}
};
}
/**
* Returns a {@link Source} that represents this EPR.
*
* @param localName
* EPR uses a different root tag name depending on the context.
* The returned {@link Source} will use the given local name
* for the root element name.
*/
public Source asSource(@NotNull String localName) {
return new SAXSource(new SAXBufferProcessorImpl(localName),new InputSource());
}
/**
* Writes this EPR to the given {@link ContentHandler}.
*
* @param localName
* EPR uses a different root tag name depending on the context.
* The returned {@link Source} will use the given local name
* for the root element name.
* @param fragment
* If true, generate a fragment SAX events without start/endDocument callbacks.
* If false, generate a full XML document event.
*/
public void writeTo(@NotNull String localName, ContentHandler contentHandler, ErrorHandler errorHandler, boolean fragment) throws SAXException {
SAXBufferProcessorImpl p = new SAXBufferProcessorImpl(localName);
p.setContentHandler(contentHandler);
p.setErrorHandler(errorHandler);
p.process(infoset,fragment);
}
/**
* Writes this EPR into the given writer.
*
* @param localName
* EPR uses a different root tag name depending on the context.
* The returned {@link Source} will use the given local name
*/
public void writeTo(final @NotNull String localName, @NotNull XMLStreamWriter w) throws XMLStreamException {
infoset.writeToXMLStreamWriter(new XMLStreamWriterFilter(w) {
private boolean root=true;
@Override
public void writeStartDocument() throws XMLStreamException {
}
@Override
public void writeStartDocument(String encoding, String version) throws XMLStreamException {
}
@Override
public void writeStartDocument(String version) throws XMLStreamException {
}
@Override
public void writeEndDocument() throws XMLStreamException {
}
private String override(String ln) {
if(root) {
root = false;
return localName;
}
return ln;
}
public void writeStartElement(String localName) throws XMLStreamException {
super.writeStartElement(override(localName));
}
public void writeStartElement(String namespaceURI, String localName) throws XMLStreamException {
super.writeStartElement(namespaceURI, override(localName));
}
public void writeStartElement(String prefix, String localName, String namespaceURI) throws XMLStreamException {
super.writeStartElement(prefix, override(localName), namespaceURI);
}
},true/*write as fragment*/);
}
/**
* Returns a {@link Header} that wraps this {@link WSEndpointReference}.
*
* <p>
* The returned header is immutable too, and can be reused with
* many {@link Message}s.
*
* @param rootTagName
* The header tag name to be used, such as <ReplyTo> or <FaultTo>.
* (It's bit ugly that this method takes {@link QName} and not just local name,
* unlike other methods. If it's making the caller's life miserable, then
* we can talk.)
*/
public Header createHeader(QName rootTagName) {
return new EPRHeader(rootTagName,this);
}
/**
* Copies all the reference parameters in this EPR as headers
* to the given {@link HeaderList}.
*/
public void addReferenceParameters(HeaderList outbound) {
for (Header header : referenceParameters) {
outbound.add(header);
}
}
/**
* Dumps the EPR infoset in a human-readable string.
*/
@Override
public String toString() {
try {
// debug convenience
StringWriter sw = new StringWriter();
XmlUtil.newTransformer().transform(asSource("EndpointReference"),new StreamResult(sw));
return sw.toString();
} catch (TransformerException e) {
return e.toString();
}
}
/**
* Filtering {@link SAXBufferProcessor} that replaces the root tag name.
*/
class SAXBufferProcessorImpl extends SAXBufferProcessor {
private final String rootLocalName;
private boolean root=true;
public SAXBufferProcessorImpl(String rootLocalName) {
super(infoset,false);
this.rootLocalName = rootLocalName;
}
protected void processElement(String uri, String localName, String qName) throws SAXException {
if(root) {
root = false;
if(qName.equals(localName)) {
qName = localName = rootLocalName;
} else {
localName = rootLocalName;
int idx = qName.indexOf(':');
qName = qName.substring(0,idx+1)+rootLocalName;
}
}
super.processElement(uri, localName, qName);
}
}
private static final OutboundReferenceParameterHeader[] EMPTY_ARRAY = new OutboundReferenceParameterHeader[0];
/**
* Parses the metadata inside this EPR and obtains it in a easy-to-process form.
*
* <p>
* See {@link Metadata} class for what's avaliable as "metadata".
*/
public @NotNull Metadata getMetaData() {
return new Metadata();
}
/**
* Parses the Metadata in an EPR and provides convenience methods to access
* the metadata.
*
*/
public class Metadata {
private @Nullable QName serviceName;
private @Nullable QName portName;
private @Nullable QName portTypeName; //interfaceName
private @Nullable Source wsdlSource;
private @Nullable String wsdliLocation;
public @Nullable QName getServiceName(){
return serviceName;
}
public @Nullable QName getPortName(){
return portName;
}
public @Nullable QName getPortTypeName(){
return portTypeName;
}
public @Nullable Source getWsdlSource(){
return wsdlSource;
}
public @Nullable String getWsdliLocation(){
return wsdliLocation;
}
private Metadata() {
try {
parseMetaData();
} catch (XMLStreamException e) {
throw new WebServiceException(e);
}
}
/**
* Parses the Metadata section of the EPR.
*/
private void parseMetaData() throws XMLStreamException {
StreamReaderBufferProcessor xsr = infoset.readAsXMLStreamReader();
// parser should be either at the start element or the start document
if (xsr.getEventType() == XMLStreamReader.START_DOCUMENT)
xsr.nextTag();
assert xsr.getEventType() == XMLStreamReader.START_ELEMENT;
String rootElement = xsr.getLocalName();
if (!xsr.getNamespaceURI().equals(version.nsUri))
throw new WebServiceException(AddressingMessages.WRONG_ADDRESSING_VERSION(
version.nsUri, xsr.getNamespaceURI()));
String localName;
String ns;
if (version == AddressingVersion.W3C) {
do {
//If the current element is metadata enclosure, look inside
if (xsr.getLocalName().equals(version.eprType.wsdlMetadata.getLocalPart())) {
String wsdlLoc = xsr.getAttributeValue("http://www.w3.org/ns/wsdl-instance","wsdlLocation");
if (wsdlLoc != null)
wsdliLocation = wsdlLoc.trim();
XMLStreamBuffer mark;
while ((mark = xsr.nextTagAndMark()) != null) {
localName = xsr.getLocalName();
ns = xsr.getNamespaceURI();
if (localName.equals(version.eprType.serviceName)) {
String portStr = xsr.getAttributeValue(null, version.eprType.portName);
if(serviceName != null)
throw new RuntimeException("More than one "+ version.eprType.serviceName +" element in EPR Metadata");
serviceName = getElementTextAsQName(xsr);
if (serviceName != null && portStr != null)
portName = new QName(serviceName.getNamespaceURI(), portStr);
} else if (localName.equals(version.eprType.portTypeName)) {
if(portTypeName != null)
throw new RuntimeException("More than one "+ version.eprType.portTypeName +" element in EPR Metadata");
portTypeName = getElementTextAsQName(xsr);
} else if (ns.equals(WSDLConstants.NS_WSDL)
&& localName.equals(WSDLConstants.QNAME_DEFINITIONS.getLocalPart())) {
wsdlSource = new XMLStreamBufferSource(mark);
} else {
XMLStreamReaderUtil.skipElement(xsr);
}
}
} else {
//Skip is it is not root element
if (!xsr.getLocalName().equals(rootElement))
XMLStreamReaderUtil.skipElement(xsr);
}
} while (XMLStreamReaderUtil.nextElementContent(xsr) == XMLStreamReader.START_ELEMENT);
} else if (version == AddressingVersion.MEMBER) {
do {
localName = xsr.getLocalName();
ns = xsr.getNamespaceURI();
//If the current element is metadata enclosure, look inside
if (localName.equals(version.eprType.wsdlMetadata.getLocalPart()) &&
ns.equals(version.eprType.wsdlMetadata.getNamespaceURI())) {
while (xsr.nextTag() == XMLStreamReader.START_ELEMENT) {
XMLStreamBuffer mark;
while ((mark = xsr.nextTagAndMark()) != null) {
localName = xsr.getLocalName();
ns = xsr.getNamespaceURI();
if (ns.equals(WSDLConstants.NS_WSDL)
&& localName.equals(WSDLConstants.QNAME_DEFINITIONS.getLocalPart())) {
wsdlSource = new XMLStreamBufferSource(mark);
} else {
XMLStreamReaderUtil.skipElement(xsr);
}
}
}
} else if (localName.equals(version.eprType.serviceName)) {
String portStr = xsr.getAttributeValue(null, version.eprType.portName);
serviceName = getElementTextAsQName(xsr);
if (serviceName != null && portStr != null)
portName = new QName(serviceName.getNamespaceURI(), portStr);
} else if (localName.equals(version.eprType.portTypeName)) {
portTypeName = getElementTextAsQName(xsr);
} else {
//Skip is it is not root element
if (!xsr.getLocalName().equals(rootElement))
XMLStreamReaderUtil.skipElement(xsr);
}
} while (XMLStreamReaderUtil.nextElementContent(xsr) == XMLStreamReader.START_ELEMENT);
}
}
private QName getElementTextAsQName(StreamReaderBufferProcessor xsr) throws XMLStreamException {
String text = xsr.getElementText().trim();
String prefix = XmlUtil.getPrefix(text);
String name = XmlUtil.getLocalPart(text);
if (name != null) {
if (prefix != null) {
String ns = xsr.getNamespaceURI(prefix);
if (ns != null)
return new QName(ns, name, prefix);
} else {
return new QName(null, name);
}
}
return null;
}
}
}