1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/message/saaj/SAAJMessage.java Wed Apr 27 01:27:09 2016 +0800
1.3 @@ -0,0 +1,778 @@
1.4 +/*
1.5 + * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
1.7 + *
1.8 + * This code is free software; you can redistribute it and/or modify it
1.9 + * under the terms of the GNU General Public License version 2 only, as
1.10 + * published by the Free Software Foundation. Oracle designates this
1.11 + * particular file as subject to the "Classpath" exception as provided
1.12 + * by Oracle in the LICENSE file that accompanied this code.
1.13 + *
1.14 + * This code is distributed in the hope that it will be useful, but WITHOUT
1.15 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
1.16 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
1.17 + * version 2 for more details (a copy is included in the LICENSE file that
1.18 + * accompanied this code).
1.19 + *
1.20 + * You should have received a copy of the GNU General Public License version
1.21 + * 2 along with this work; if not, write to the Free Software Foundation,
1.22 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
1.23 + *
1.24 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
1.25 + * or visit www.oracle.com if you need additional information or have any
1.26 + * questions.
1.27 + */
1.28 +
1.29 +package com.sun.xml.internal.ws.message.saaj;
1.30 +
1.31 +import com.sun.istack.internal.FragmentContentHandler;
1.32 +import com.sun.istack.internal.NotNull;
1.33 +import com.sun.istack.internal.Nullable;
1.34 +import com.sun.istack.internal.XMLStreamException2;
1.35 +import com.sun.xml.internal.bind.api.Bridge;
1.36 +import com.sun.xml.internal.bind.unmarshaller.DOMScanner;
1.37 +import com.sun.xml.internal.ws.api.SOAPVersion;
1.38 +import com.sun.xml.internal.ws.api.message.*;
1.39 +import com.sun.xml.internal.ws.message.AttachmentUnmarshallerImpl;
1.40 +import com.sun.xml.internal.ws.spi.db.XMLBridge;
1.41 +import com.sun.xml.internal.ws.streaming.DOMStreamReader;
1.42 +import com.sun.xml.internal.ws.util.ASCIIUtility;
1.43 +import com.sun.xml.internal.ws.util.DOMUtil;
1.44 +import org.w3c.dom.Attr;
1.45 +import org.w3c.dom.Element;
1.46 +import org.w3c.dom.NamedNodeMap;
1.47 +import org.w3c.dom.Node;
1.48 +import org.xml.sax.ContentHandler;
1.49 +import org.xml.sax.ErrorHandler;
1.50 +import org.xml.sax.SAXException;
1.51 +import org.xml.sax.helpers.AttributesImpl;
1.52 +import org.xml.sax.helpers.LocatorImpl;
1.53 +
1.54 +import javax.activation.DataHandler;
1.55 +import javax.xml.bind.JAXBException;
1.56 +import javax.xml.bind.Unmarshaller;
1.57 +import javax.xml.soap.*;
1.58 +import javax.xml.stream.XMLStreamException;
1.59 +import javax.xml.stream.XMLStreamReader;
1.60 +import javax.xml.stream.XMLStreamWriter;
1.61 +import javax.xml.transform.Source;
1.62 +import javax.xml.transform.dom.DOMSource;
1.63 +import javax.xml.transform.stream.StreamSource;
1.64 +import javax.xml.ws.WebServiceException;
1.65 +import java.io.IOException;
1.66 +import java.io.InputStream;
1.67 +import java.io.OutputStream;
1.68 +import java.util.HashMap;
1.69 +import java.util.Iterator;
1.70 +import java.util.List;
1.71 +import java.util.Map;
1.72 +
1.73 +/**
1.74 + * {@link Message} implementation backed by {@link SOAPMessage}.
1.75 + *
1.76 + * @author Vivek Pandey
1.77 + * @author Rama Pulavarthi
1.78 + */
1.79 +public class SAAJMessage extends Message {
1.80 + // flag to switch between representations
1.81 + private boolean parsedMessage;
1.82 + // flag to check if Message API is exercised;
1.83 + private boolean accessedMessage;
1.84 + private final SOAPMessage sm;
1.85 +
1.86 + private MessageHeaders headers;
1.87 + private List<Element> bodyParts;
1.88 + private Element payload;
1.89 +
1.90 + private String payloadLocalName;
1.91 + private String payloadNamespace;
1.92 + private SOAPVersion soapVersion;
1.93 +
1.94 + //Collect the attrbutes on the enclosing elements so that the same message can be reproduced without loss of any
1.95 + // valuable info
1.96 + private NamedNodeMap bodyAttrs, headerAttrs, envelopeAttrs;
1.97 +
1.98 + public SAAJMessage(SOAPMessage sm) {
1.99 + this.sm = sm;
1.100 + }
1.101 +
1.102 + /**
1.103 + * This constructor is a convenience and called by the {@link #copy}
1.104 + *
1.105 + * @param headers
1.106 + * @param sm
1.107 + */
1.108 + private SAAJMessage(MessageHeaders headers, AttachmentSet as, SOAPMessage sm, SOAPVersion version) {
1.109 + this.sm = sm;
1.110 + this.parse();
1.111 + if(headers == null)
1.112 + headers = new HeaderList(version);
1.113 + this.headers = headers;
1.114 + this.attachmentSet = as;
1.115 + }
1.116 +
1.117 + private void parse() {
1.118 + if (!parsedMessage) {
1.119 + try {
1.120 + access();
1.121 + if (headers == null)
1.122 + headers = new HeaderList(getSOAPVersion());
1.123 + SOAPHeader header = sm.getSOAPHeader();
1.124 + if (header != null) {
1.125 + headerAttrs = header.getAttributes();
1.126 + Iterator iter = header.examineAllHeaderElements();
1.127 + while (iter.hasNext()) {
1.128 + headers.add(new SAAJHeader((SOAPHeaderElement) iter.next()));
1.129 + }
1.130 + }
1.131 + attachmentSet = new SAAJAttachmentSet(sm);
1.132 +
1.133 + parsedMessage = true;
1.134 + } catch (SOAPException e) {
1.135 + throw new WebServiceException(e);
1.136 + }
1.137 + }
1.138 + }
1.139 +
1.140 + protected void access() {
1.141 + if (!accessedMessage) {
1.142 + try {
1.143 + envelopeAttrs = sm.getSOAPPart().getEnvelope().getAttributes();
1.144 + Node body = sm.getSOAPBody();
1.145 + bodyAttrs = body.getAttributes();
1.146 + soapVersion = SOAPVersion.fromNsUri(body.getNamespaceURI());
1.147 + //cature all the body elements
1.148 + bodyParts = DOMUtil.getChildElements(body);
1.149 + //we treat payload as the first body part
1.150 + payload = bodyParts.size() > 0 ? bodyParts.get(0) : null;
1.151 + // hope this is correct. Caching the localname and namespace of the payload should be fine
1.152 + // but what about if a Handler replaces the payload with something else? Weel, may be it
1.153 + // will be error condition anyway
1.154 + if (payload != null) {
1.155 + payloadLocalName = payload.getLocalName();
1.156 + payloadNamespace = payload.getNamespaceURI();
1.157 + }
1.158 + accessedMessage = true;
1.159 + } catch (SOAPException e) {
1.160 + throw new WebServiceException(e);
1.161 + }
1.162 + }
1.163 + }
1.164 +
1.165 + public boolean hasHeaders() {
1.166 + parse();
1.167 + return headers.hasHeaders();
1.168 + }
1.169 +
1.170 + public @NotNull MessageHeaders getHeaders() {
1.171 + parse();
1.172 + return headers;
1.173 + }
1.174 +
1.175 + /**
1.176 + * Gets the attachments of this message
1.177 + * (attachments live outside a message.)
1.178 + */
1.179 + @Override
1.180 + public @NotNull AttachmentSet getAttachments() {
1.181 + if (attachmentSet == null) attachmentSet = new SAAJAttachmentSet(sm);
1.182 + return attachmentSet;
1.183 + }
1.184 +
1.185 + /**
1.186 + * Optimization hint for the derived class to check
1.187 + * if we may have some attachments.
1.188 + */
1.189 + @Override
1.190 + protected boolean hasAttachments() {
1.191 + return !getAttachments().isEmpty();
1.192 + }
1.193 +
1.194 + public @Nullable String getPayloadLocalPart() {
1.195 + soapBodyFirstChild();
1.196 + return payloadLocalName;
1.197 + }
1.198 +
1.199 + public String getPayloadNamespaceURI() {
1.200 + soapBodyFirstChild();
1.201 + return payloadNamespace;
1.202 + }
1.203 +
1.204 + public boolean hasPayload() {
1.205 + return soapBodyFirstChild() != null;
1.206 + }
1.207 +
1.208 + private void addAttributes(Element e, NamedNodeMap attrs) {
1.209 + if(attrs == null)
1.210 + return;
1.211 + String elPrefix = e.getPrefix();
1.212 + for(int i=0; i < attrs.getLength();i++) {
1.213 + Attr a = (Attr)attrs.item(i);
1.214 + //check if attr is ns declaration
1.215 + if("xmlns".equals(a.getPrefix()) || "xmlns".equals(a.getLocalName())) {
1.216 + if(elPrefix == null && a.getLocalName().equals("xmlns")) {
1.217 + // the target element has already default ns declaration, dont' override it
1.218 + continue;
1.219 + } else if(elPrefix != null && "xmlns".equals(a.getPrefix()) && elPrefix.equals(a.getLocalName())) {
1.220 + //dont bind the prefix to ns again, its already in the target element.
1.221 + continue;
1.222 + }
1.223 + e.setAttributeNS(a.getNamespaceURI(),a.getName(),a.getValue());
1.224 + continue;
1.225 + }
1.226 + e.setAttributeNS(a.getNamespaceURI(),a.getName(),a.getValue());
1.227 + }
1.228 + }
1.229 +
1.230 + public Source readEnvelopeAsSource() {
1.231 + try {
1.232 + if (!parsedMessage) {
1.233 + SOAPEnvelope se = sm.getSOAPPart().getEnvelope();
1.234 + return new DOMSource(se);
1.235 +
1.236 + } else {
1.237 + SOAPMessage msg = soapVersion.getMessageFactory().createMessage();
1.238 + addAttributes(msg.getSOAPPart().getEnvelope(),envelopeAttrs);
1.239 +
1.240 + SOAPBody newBody = msg.getSOAPPart().getEnvelope().getBody();
1.241 + addAttributes(newBody, bodyAttrs);
1.242 + for (Element part : bodyParts) {
1.243 + Node n = newBody.getOwnerDocument().importNode(part, true);
1.244 + newBody.appendChild(n);
1.245 + }
1.246 + addAttributes(msg.getSOAPHeader(),headerAttrs);
1.247 + for (Header header : headers.asList()) {
1.248 + header.writeTo(msg);
1.249 + }
1.250 + SOAPEnvelope se = msg.getSOAPPart().getEnvelope();
1.251 + return new DOMSource(se);
1.252 + }
1.253 + } catch (SOAPException e) {
1.254 + throw new WebServiceException(e);
1.255 + }
1.256 + }
1.257 +
1.258 + public SOAPMessage readAsSOAPMessage() throws SOAPException {
1.259 + if (!parsedMessage) {
1.260 + return sm;
1.261 + } else {
1.262 + SOAPMessage msg = soapVersion.getMessageFactory().createMessage();
1.263 + addAttributes(msg.getSOAPPart().getEnvelope(),envelopeAttrs);
1.264 + SOAPBody newBody = msg.getSOAPPart().getEnvelope().getBody();
1.265 + addAttributes(newBody, bodyAttrs);
1.266 + for (Element part : bodyParts) {
1.267 + Node n = newBody.getOwnerDocument().importNode(part, true);
1.268 + newBody.appendChild(n);
1.269 + }
1.270 + addAttributes(msg.getSOAPHeader(),headerAttrs);
1.271 + for (Header header : headers.asList()) {
1.272 + header.writeTo(msg);
1.273 + }
1.274 + for (Attachment att : getAttachments()) {
1.275 + AttachmentPart part = msg.createAttachmentPart();
1.276 + part.setDataHandler(att.asDataHandler());
1.277 + part.setContentId('<' + att.getContentId() + '>');
1.278 + addCustomMimeHeaders(att, part);
1.279 + msg.addAttachmentPart(part);
1.280 + }
1.281 + msg.saveChanges();
1.282 + return msg;
1.283 + }
1.284 + }
1.285 +
1.286 + private void addCustomMimeHeaders(Attachment att, AttachmentPart part) {
1.287 + if (att instanceof AttachmentEx) {
1.288 + Iterator<AttachmentEx.MimeHeader> allMimeHeaders = ((AttachmentEx) att).getMimeHeaders();
1.289 + while (allMimeHeaders.hasNext()) {
1.290 + AttachmentEx.MimeHeader mh = allMimeHeaders.next();
1.291 + String name = mh.getName();
1.292 + if (!"Content-Type".equalsIgnoreCase(name)
1.293 + && !"Content-Id".equalsIgnoreCase(name)) {
1.294 + part.addMimeHeader(name, mh.getValue());
1.295 + }
1.296 + }
1.297 + }
1.298 + }
1.299 +
1.300 + public Source readPayloadAsSource() {
1.301 + access();
1.302 + return (payload != null) ? new DOMSource(payload) : null;
1.303 + }
1.304 +
1.305 + public <T> T readPayloadAsJAXB(Unmarshaller unmarshaller) throws JAXBException {
1.306 + access();
1.307 + if (payload != null) {
1.308 + if(hasAttachments())
1.309 + unmarshaller.setAttachmentUnmarshaller(new AttachmentUnmarshallerImpl(getAttachments()));
1.310 + return (T) unmarshaller.unmarshal(payload);
1.311 +
1.312 + }
1.313 + return null;
1.314 + }
1.315 +
1.316 + /** @deprecated */
1.317 + public <T> T readPayloadAsJAXB(Bridge<T> bridge) throws JAXBException {
1.318 + access();
1.319 + if (payload != null)
1.320 + return bridge.unmarshal(payload,hasAttachments()? new AttachmentUnmarshallerImpl(getAttachments()) : null);
1.321 + return null;
1.322 + }
1.323 + public <T> T readPayloadAsJAXB(XMLBridge<T> bridge) throws JAXBException {
1.324 + access();
1.325 + if (payload != null)
1.326 + return bridge.unmarshal(payload,hasAttachments()? new AttachmentUnmarshallerImpl(getAttachments()) : null);
1.327 + return null;
1.328 + }
1.329 +
1.330 + public XMLStreamReader readPayload() throws XMLStreamException {
1.331 + return soapBodyFirstChildReader();
1.332 + }
1.333 +
1.334 + public void writePayloadTo(XMLStreamWriter sw) throws XMLStreamException {
1.335 + access();
1.336 + try {
1.337 + for (Element part : bodyParts)
1.338 + DOMUtil.serializeNode(part, sw);
1.339 + } catch (XMLStreamException e) {
1.340 + throw new WebServiceException(e);
1.341 + }
1.342 + }
1.343 +
1.344 + public void writeTo(XMLStreamWriter writer) throws XMLStreamException {
1.345 + try {
1.346 + writer.writeStartDocument();
1.347 + if (!parsedMessage) {
1.348 + DOMUtil.serializeNode(sm.getSOAPPart().getEnvelope(), writer);
1.349 + } else {
1.350 + SOAPEnvelope env = sm.getSOAPPart().getEnvelope();
1.351 + DOMUtil.writeTagWithAttributes(env, writer);
1.352 + if (hasHeaders()) {
1.353 + if(env.getHeader() != null) {
1.354 + DOMUtil.writeTagWithAttributes(env.getHeader(), writer);
1.355 + } else {
1.356 + writer.writeStartElement(env.getPrefix(), "Header", env.getNamespaceURI());
1.357 + }
1.358 + for (Header h : headers.asList()) {
1.359 + h.writeTo(writer);
1.360 + }
1.361 + writer.writeEndElement();
1.362 + }
1.363 +
1.364 + DOMUtil.serializeNode(sm.getSOAPBody(), writer);
1.365 + writer.writeEndElement();
1.366 + }
1.367 + writer.writeEndDocument();
1.368 + writer.flush();
1.369 + } catch (SOAPException ex) {
1.370 + throw new XMLStreamException2(ex);
1.371 + //for now. ask jaxws team what to do.
1.372 + }
1.373 + }
1.374 +
1.375 + public void writeTo(ContentHandler contentHandler, ErrorHandler errorHandler) throws SAXException {
1.376 + String soapNsUri = soapVersion.nsUri;
1.377 + if (!parsedMessage) {
1.378 + DOMScanner ds = new DOMScanner();
1.379 + ds.setContentHandler(contentHandler);
1.380 + ds.scan(sm.getSOAPPart());
1.381 + } else {
1.382 + contentHandler.setDocumentLocator(NULL_LOCATOR);
1.383 + contentHandler.startDocument();
1.384 + contentHandler.startPrefixMapping("S", soapNsUri);
1.385 + startPrefixMapping(contentHandler, envelopeAttrs,"S");
1.386 + contentHandler.startElement(soapNsUri, "Envelope", "S:Envelope", getAttributes(envelopeAttrs));
1.387 + if (hasHeaders()) {
1.388 + startPrefixMapping(contentHandler, headerAttrs,"S");
1.389 + contentHandler.startElement(soapNsUri, "Header", "S:Header", getAttributes(headerAttrs));
1.390 + MessageHeaders headers = getHeaders();
1.391 + for (Header h : headers.asList()) {
1.392 + h.writeTo(contentHandler, errorHandler);
1.393 + }
1.394 + endPrefixMapping(contentHandler, headerAttrs,"S");
1.395 + contentHandler.endElement(soapNsUri, "Header", "S:Header");
1.396 +
1.397 + }
1.398 + startPrefixMapping(contentHandler, bodyAttrs,"S");
1.399 + // write the body
1.400 + contentHandler.startElement(soapNsUri, "Body", "S:Body", getAttributes(bodyAttrs));
1.401 + writePayloadTo(contentHandler, errorHandler, true);
1.402 + endPrefixMapping(contentHandler, bodyAttrs,"S");
1.403 + contentHandler.endElement(soapNsUri, "Body", "S:Body");
1.404 + endPrefixMapping(contentHandler, envelopeAttrs,"S");
1.405 + contentHandler.endElement(soapNsUri, "Envelope", "S:Envelope");
1.406 + }
1.407 + }
1.408 + /**
1.409 + * Gets the Attributes that are not namesapce declarations
1.410 + * @param attrs
1.411 + * @return
1.412 + */
1.413 + private AttributesImpl getAttributes(NamedNodeMap attrs) {
1.414 + AttributesImpl atts = new AttributesImpl();
1.415 + if(attrs == null)
1.416 + return EMPTY_ATTS;
1.417 + for(int i=0; i < attrs.getLength();i++) {
1.418 + Attr a = (Attr)attrs.item(i);
1.419 + //check if attr is ns declaration
1.420 + if("xmlns".equals(a.getPrefix()) || "xmlns".equals(a.getLocalName())) {
1.421 + continue;
1.422 + }
1.423 + atts.addAttribute(fixNull(a.getNamespaceURI()),a.getLocalName(),a.getName(),a.getSchemaTypeInfo().getTypeName(),a.getValue());
1.424 + }
1.425 + return atts;
1.426 + }
1.427 +
1.428 + /**
1.429 + * Collects the ns declarations and starts the prefix mapping, consequently the associated endPrefixMapping needs to be called.
1.430 + * @param contentHandler
1.431 + * @param attrs
1.432 + * @param excludePrefix , this is to excldue the global prefix mapping "S" used at the start
1.433 + * @throws SAXException
1.434 + */
1.435 + private void startPrefixMapping(ContentHandler contentHandler, NamedNodeMap attrs, String excludePrefix) throws SAXException {
1.436 + if(attrs == null)
1.437 + return;
1.438 + for(int i=0; i < attrs.getLength();i++) {
1.439 + Attr a = (Attr)attrs.item(i);
1.440 + //check if attr is ns declaration
1.441 + if("xmlns".equals(a.getPrefix()) || "xmlns".equals(a.getLocalName())) {
1.442 + if(!fixNull(a.getPrefix()).equals(excludePrefix)) {
1.443 + contentHandler.startPrefixMapping(fixNull(a.getPrefix()), a.getNamespaceURI());
1.444 + }
1.445 + }
1.446 + }
1.447 + }
1.448 +
1.449 + private void endPrefixMapping(ContentHandler contentHandler, NamedNodeMap attrs, String excludePrefix) throws SAXException {
1.450 + if(attrs == null)
1.451 + return;
1.452 + for(int i=0; i < attrs.getLength();i++) {
1.453 + Attr a = (Attr)attrs.item(i);
1.454 + //check if attr is ns declaration
1.455 + if("xmlns".equals(a.getPrefix()) || "xmlns".equals(a.getLocalName())) {
1.456 + if(!fixNull(a.getPrefix()).equals(excludePrefix)) {
1.457 + contentHandler.endPrefixMapping(fixNull(a.getPrefix()));
1.458 + }
1.459 + }
1.460 + }
1.461 + }
1.462 +
1.463 + private static String fixNull(String s) {
1.464 + if(s==null) return "";
1.465 + else return s;
1.466 + }
1.467 +
1.468 + private void writePayloadTo(ContentHandler contentHandler, ErrorHandler errorHandler, boolean fragment) throws SAXException {
1.469 + if(fragment)
1.470 + contentHandler = new FragmentContentHandler(contentHandler);
1.471 + DOMScanner ds = new DOMScanner();
1.472 + ds.setContentHandler(contentHandler);
1.473 + ds.scan(payload);
1.474 + }
1.475 +
1.476 + /**
1.477 + * Creates a copy of a {@link com.sun.xml.internal.ws.api.message.Message}.
1.478 + * <p/>
1.479 + * <p/>
1.480 + * This method creates a new {@link com.sun.xml.internal.ws.api.message.Message} whose header/payload/attachments/properties
1.481 + * are identical to this {@link com.sun.xml.internal.ws.api.message.Message}. Once created, the created {@link com.sun.xml.internal.ws.api.message.Message}
1.482 + * and the original {@link com.sun.xml.internal.ws.api.message.Message} behaves independently --- adding header/
1.483 + * attachment to one {@link com.sun.xml.internal.ws.api.message.Message} doesn't affect another {@link com.sun.xml.internal.ws.api.message.Message}
1.484 + * at all.
1.485 + * <p/>
1.486 + * <h3>Design Rationale</h3>
1.487 + * <p/>
1.488 + * Since a {@link com.sun.xml.internal.ws.api.message.Message} body is read-once, sometimes
1.489 + * (such as when you do fail-over, or WS-RM) you need to
1.490 + * create an idential copy of a {@link com.sun.xml.internal.ws.api.message.Message}.
1.491 + * <p/>
1.492 + * <p/>
1.493 + * The actual copy operation depends on the layout
1.494 + * of the data in memory, hence it's best to be done by
1.495 + * the {@link com.sun.xml.internal.ws.api.message.Message} implementation itself.
1.496 + */
1.497 + public Message copy() {
1.498 + try {
1.499 + if (!parsedMessage) {
1.500 + return new SAAJMessage(readAsSOAPMessage());
1.501 + } else {
1.502 + SOAPMessage msg = soapVersion.getMessageFactory().createMessage();
1.503 + SOAPBody newBody = msg.getSOAPPart().getEnvelope().getBody();
1.504 + for (Element part : bodyParts) {
1.505 + Node n = newBody.getOwnerDocument().importNode(part, true);
1.506 + newBody.appendChild(n);
1.507 + }
1.508 + addAttributes(newBody, bodyAttrs);
1.509 + return new SAAJMessage(getHeaders(), getAttachments(), msg, soapVersion);
1.510 + }
1.511 + } catch (SOAPException e) {
1.512 + throw new WebServiceException(e);
1.513 + }
1.514 + }
1.515 + private static final AttributesImpl EMPTY_ATTS = new AttributesImpl();
1.516 + private static final LocatorImpl NULL_LOCATOR = new LocatorImpl();
1.517 +
1.518 + protected static class SAAJAttachment implements AttachmentEx {
1.519 +
1.520 + final AttachmentPart ap;
1.521 +
1.522 + String contentIdNoAngleBracket;
1.523 +
1.524 + public SAAJAttachment(AttachmentPart part) {
1.525 + this.ap = part;
1.526 + }
1.527 +
1.528 + /**
1.529 + * Content ID of the attachment. Uniquely identifies an attachment.
1.530 + */
1.531 + public String getContentId() {
1.532 + if (contentIdNoAngleBracket == null) {
1.533 + contentIdNoAngleBracket = ap.getContentId();
1.534 + if (contentIdNoAngleBracket != null && contentIdNoAngleBracket.charAt(0) == '<')
1.535 + contentIdNoAngleBracket = contentIdNoAngleBracket.substring(1, contentIdNoAngleBracket.length()-1);
1.536 + }
1.537 + return contentIdNoAngleBracket;
1.538 + }
1.539 +
1.540 + /**
1.541 + * Gets the MIME content-type of this attachment.
1.542 + */
1.543 + public String getContentType() {
1.544 + return ap.getContentType();
1.545 + }
1.546 +
1.547 + /**
1.548 + * Gets the attachment as an exact-length byte array.
1.549 + */
1.550 + public byte[] asByteArray() {
1.551 + try {
1.552 + return ap.getRawContentBytes();
1.553 + } catch (SOAPException e) {
1.554 + throw new WebServiceException(e);
1.555 + }
1.556 + }
1.557 +
1.558 + /**
1.559 + * Gets the attachment as a {@link javax.activation.DataHandler}.
1.560 + */
1.561 + public DataHandler asDataHandler() {
1.562 + try {
1.563 + return ap.getDataHandler();
1.564 + } catch (SOAPException e) {
1.565 + throw new WebServiceException(e);
1.566 + }
1.567 + }
1.568 +
1.569 + /**
1.570 + * Gets the attachment as a {@link javax.xml.transform.Source}.
1.571 + * Note that there's no guarantee that the attachment is actually an XML.
1.572 + */
1.573 + public Source asSource() {
1.574 + try {
1.575 + return new StreamSource(ap.getRawContent());
1.576 + } catch (SOAPException e) {
1.577 + throw new WebServiceException(e);
1.578 + }
1.579 + }
1.580 +
1.581 + /**
1.582 + * Obtains this attachment as an {@link java.io.InputStream}.
1.583 + */
1.584 + public InputStream asInputStream() {
1.585 + try {
1.586 + return ap.getRawContent();
1.587 + } catch (SOAPException e) {
1.588 + throw new WebServiceException(e);
1.589 + }
1.590 + }
1.591 +
1.592 + /**
1.593 + * Writes the contents of the attachment into the given stream.
1.594 + */
1.595 + public void writeTo(OutputStream os) throws IOException {
1.596 + try {
1.597 + ASCIIUtility.copyStream(ap.getRawContent(), os);
1.598 + } catch (SOAPException e) {
1.599 + throw new WebServiceException(e);
1.600 + }
1.601 + }
1.602 +
1.603 + /**
1.604 + * Writes this attachment to the given {@link javax.xml.soap.SOAPMessage}.
1.605 + */
1.606 + public void writeTo(SOAPMessage saaj) {
1.607 + saaj.addAttachmentPart(ap);
1.608 + }
1.609 +
1.610 + AttachmentPart asAttachmentPart(){
1.611 + return ap;
1.612 + }
1.613 +
1.614 + public Iterator<MimeHeader> getMimeHeaders() {
1.615 + final Iterator it = ap.getAllMimeHeaders();
1.616 + return new Iterator<MimeHeader>() {
1.617 + public boolean hasNext() {
1.618 + return it.hasNext();
1.619 + }
1.620 +
1.621 + public MimeHeader next() {
1.622 + final javax.xml.soap.MimeHeader mh = (javax.xml.soap.MimeHeader) it.next();
1.623 + return new MimeHeader() {
1.624 + public String getName() {
1.625 + return mh.getName();
1.626 + }
1.627 +
1.628 + public String getValue() {
1.629 + return mh.getValue();
1.630 + }
1.631 + };
1.632 + }
1.633 +
1.634 + public void remove() {
1.635 + throw new UnsupportedOperationException();
1.636 + }
1.637 + };
1.638 + }
1.639 + }
1.640 +
1.641 + /**
1.642 + * {@link AttachmentSet} for SAAJ.
1.643 + *
1.644 + * SAAJ wants '<' and '>' for the content ID, but {@link AttachmentSet}
1.645 + * doesn't. S this class also does the conversion between them.
1.646 + */
1.647 + protected static class SAAJAttachmentSet implements AttachmentSet {
1.648 +
1.649 + private Map<String, Attachment> attMap;
1.650 + private Iterator attIter;
1.651 +
1.652 + public SAAJAttachmentSet(SOAPMessage sm) {
1.653 + attIter = sm.getAttachments();
1.654 + }
1.655 +
1.656 + /**
1.657 + * Gets the attachment by the content ID.
1.658 + *
1.659 + * @return null
1.660 + * if no such attachment exist.
1.661 + */
1.662 + public Attachment get(String contentId) {
1.663 + // if this is the first time then create the attachment Map
1.664 + if (attMap == null) {
1.665 + if (!attIter.hasNext())
1.666 + return null;
1.667 + attMap = createAttachmentMap();
1.668 + }
1.669 + if(contentId.charAt(0) != '<'){
1.670 + return attMap.get('<'+contentId+'>');
1.671 + }
1.672 + return attMap.get(contentId);
1.673 + }
1.674 +
1.675 + public boolean isEmpty() {
1.676 + if(attMap!=null)
1.677 + return attMap.isEmpty();
1.678 + else
1.679 + return !attIter.hasNext();
1.680 + }
1.681 +
1.682 + /**
1.683 + * Returns an iterator over a set of elements of type T.
1.684 + *
1.685 + * @return an Iterator.
1.686 + */
1.687 + public Iterator<Attachment> iterator() {
1.688 + if (attMap == null) {
1.689 + attMap = createAttachmentMap();
1.690 + }
1.691 + return attMap.values().iterator();
1.692 + }
1.693 +
1.694 + private Map<String, Attachment> createAttachmentMap() {
1.695 + HashMap<String, Attachment> map = new HashMap<String, Attachment>();
1.696 + while (attIter.hasNext()) {
1.697 + AttachmentPart ap = (AttachmentPart) attIter.next();
1.698 + map.put(ap.getContentId(), new SAAJAttachment(ap));
1.699 + }
1.700 + return map;
1.701 + }
1.702 +
1.703 + public void add(Attachment att) {
1.704 + attMap.put('<'+att.getContentId()+'>', att);
1.705 + }
1.706 + }
1.707 +
1.708 + public SOAPVersion getSOAPVersion() {
1.709 + return soapVersion;
1.710 + }
1.711 +
1.712 + private XMLStreamReader soapBodyFirstChildReader;
1.713 +
1.714 + /**
1.715 + * This allow the subclass to retain the XMLStreamReader.
1.716 + */
1.717 + protected XMLStreamReader getXMLStreamReader(SOAPElement soapElement) {
1.718 + return null;
1.719 + }
1.720 +
1.721 + protected XMLStreamReader createXMLStreamReader(SOAPElement soapElement) {
1.722 + DOMStreamReader dss = new DOMStreamReader();
1.723 + dss.setCurrentNode(soapElement);
1.724 + return dss;
1.725 + }
1.726 +
1.727 + protected XMLStreamReader soapBodyFirstChildReader() {
1.728 + if (soapBodyFirstChildReader != null) return soapBodyFirstChildReader;
1.729 + soapBodyFirstChild();
1.730 + if (soapBodyFirstChild != null) {
1.731 + soapBodyFirstChildReader = getXMLStreamReader(soapBodyFirstChild);
1.732 + if (soapBodyFirstChildReader == null) soapBodyFirstChildReader =
1.733 + createXMLStreamReader(soapBodyFirstChild);
1.734 + if (soapBodyFirstChildReader.getEventType() == XMLStreamReader.START_DOCUMENT) {
1.735 + try {
1.736 + while(soapBodyFirstChildReader.getEventType() != XMLStreamReader.START_ELEMENT)
1.737 + soapBodyFirstChildReader.next();
1.738 + } catch (XMLStreamException e) {
1.739 + throw new RuntimeException(e);
1.740 + }
1.741 + }
1.742 + return soapBodyFirstChildReader;
1.743 + } else {
1.744 + payloadLocalName = null;
1.745 + payloadNamespace = null;
1.746 + return null;
1.747 + }
1.748 + }
1.749 +
1.750 + private SOAPElement soapBodyFirstChild;
1.751 +
1.752 + SOAPElement soapBodyFirstChild() {
1.753 + if (soapBodyFirstChild != null) return soapBodyFirstChild;
1.754 + try {
1.755 + boolean foundElement = false;
1.756 + for (Node n = sm.getSOAPBody().getFirstChild(); n != null && !foundElement; n = n.getNextSibling()) {
1.757 + if (n.getNodeType() == Node.ELEMENT_NODE) {
1.758 + foundElement = true;
1.759 + if (n instanceof SOAPElement) {
1.760 + soapBodyFirstChild = (SOAPElement) n;
1.761 + payloadLocalName = soapBodyFirstChild.getLocalName();
1.762 + payloadNamespace = soapBodyFirstChild.getNamespaceURI();
1.763 + return soapBodyFirstChild;
1.764 + }
1.765 + }
1.766 + }
1.767 + if(foundElement) for(Iterator i = sm.getSOAPBody().getChildElements(); i.hasNext();){
1.768 + Object o = i.next();
1.769 + if (o instanceof SOAPElement) {
1.770 + soapBodyFirstChild = (SOAPElement)o;
1.771 + payloadLocalName = soapBodyFirstChild.getLocalName();
1.772 + payloadNamespace = soapBodyFirstChild.getNamespaceURI();
1.773 + return soapBodyFirstChild;
1.774 + }
1.775 + }
1.776 + } catch (SOAPException e) {
1.777 + throw new RuntimeException(e);
1.778 + }
1.779 + return soapBodyFirstChild;
1.780 + }
1.781 +}
