| /* |
| * Copyright 2005-2006 Sun Microsystems, Inc. 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. Sun designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, |
| * CA 95054 USA or visit www.sun.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.InvalidMapException; |
| 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 InvalidMapException(new QName(version.nsUri,rootLocalName),AddressingVersion.fault_duplicateAddressInEpr); |
| address = xsr.getElementText(); |
| } 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 InvalidMapException(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; |
| public @Nullable QName getServiceName(){ |
| return serviceName; |
| } |
| public @Nullable QName getPortName(){ |
| return portName; |
| } |
| public @Nullable QName getPortTypeName(){ |
| return portTypeName; |
| } |
| public @Nullable Source getWsdlSource(){ |
| return wsdlSource; |
| } |
| |
| 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())) { |
| 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); |
| serviceName = getElementTextAsQName(xsr); |
| if (serviceName != null) |
| portName = new QName(serviceName.getNamespaceURI(), portStr); |
| } else if (localName.equals(version.eprType.portTypeName)) { |
| 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) |
| 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(); |
| 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; |
| } |
| } |
| } |