--- a/jaxws/src/java.xml.soap/share/classes/com/sun/xml/internal/messaging/saaj/soap/AttachmentPartImpl.java Thu Jan 15 11:30:13 2015 +0100
+++ /dev/null Thu Jan 01 00:00:00 1970 +0000
@@ -1,612 +0,0 @@
-/*
- * Copyright (c) 1997, 2014, 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.messaging.saaj.soap;
-
-import com.sun.xml.internal.messaging.saaj.SOAPExceptionImpl;
-
-import com.sun.xml.internal.messaging.saaj.packaging.mime.util.ASCIIUtility;
-
-import com.sun.xml.internal.messaging.saaj.packaging.mime.Header;
-import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimePartDataSource;
-import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.InternetHeaders;
-import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimeBodyPart;
-import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimeUtility;
-import com.sun.xml.internal.messaging.saaj.util.ByteOutputStream;
-import com.sun.xml.internal.messaging.saaj.util.LogDomainConstants;
-
-import java.io.IOException;
-import java.io.InputStream;
-import java.io.ByteArrayOutputStream;
-import java.io.ByteArrayInputStream;
-import java.io.OutputStream;
-import java.util.Iterator;
-import java.util.List;
-import java.util.logging.Level;
-import java.util.logging.Logger;
-
-import javax.activation.*;
-import javax.xml.soap.*;
-import com.sun.xml.internal.org.jvnet.mimepull.MIMEPart;
-
-/**
- * Implementation of attachments.
- *
- * @author Anil Vijendran (akv@eng.sun.com)
- */
-public class AttachmentPartImpl extends AttachmentPart {
-
- protected static final Logger log =
- Logger.getLogger(LogDomainConstants.SOAP_DOMAIN,
- "com.sun.xml.internal.messaging.saaj.soap.LocalStrings");
-
- private final MimeHeaders headers;
- private MimeBodyPart rawContent = null;
- private DataHandler dataHandler = null;
-
- //alternate impl that uses a MIMEPart
- private MIMEPart mimePart = null;
-
- public AttachmentPartImpl() {
- headers = new MimeHeaders();
-
- // initialization from here should cover most of cases;
- // if not, it would be necessary to call
- // AttachmentPartImpl.initializeJavaActivationHandlers()
- // explicitly by programmer
- initializeJavaActivationHandlers();
- }
-
- public AttachmentPartImpl(MIMEPart part) {
- headers = new MimeHeaders();
- mimePart = part;
- List<? extends com.sun.xml.internal.org.jvnet.mimepull.Header> hdrs = part.getAllHeaders();
- for (com.sun.xml.internal.org.jvnet.mimepull.Header hd : hdrs) {
- headers.addHeader(hd.getName(), hd.getValue());
- }
- }
-
- public int getSize() throws SOAPException {
- if (mimePart != null) {
- try {
- return mimePart.read().available();
- } catch (IOException e) {
- return -1;
- }
- }
- if ((rawContent == null) && (dataHandler == null))
- return 0;
-
- if (rawContent != null) {
- try {
- return rawContent.getSize();
- } catch (Exception ex) {
- log.log(
- Level.SEVERE,
- "SAAJ0573.soap.attachment.getrawbytes.ioexception",
- new String[] { ex.getLocalizedMessage()});
- throw new SOAPExceptionImpl("Raw InputStream Error: " + ex);
- }
- } else {
- ByteOutputStream bout = new ByteOutputStream();
- try {
- dataHandler.writeTo(bout);
- } catch (IOException ex) {
- log.log(
- Level.SEVERE,
- "SAAJ0501.soap.data.handler.err",
- new String[] { ex.getLocalizedMessage()});
- throw new SOAPExceptionImpl("Data handler error: " + ex);
- }
- return bout.size();
- }
- }
-
- public void clearContent() {
- if (mimePart != null) {
- mimePart.close();
- mimePart = null;
- }
- dataHandler = null;
- rawContent = null;
- }
-
- public Object getContent() throws SOAPException {
- try {
- if (mimePart != null) {
- //return an inputstream
- return mimePart.read();
- }
- if (dataHandler != null) {
- return getDataHandler().getContent();
- } else if (rawContent != null) {
- return rawContent.getContent();
- } else {
- log.severe("SAAJ0572.soap.no.content.for.attachment");
- throw new SOAPExceptionImpl("No data handler/content associated with this attachment");
- }
- } catch (Exception ex) {
- log.log(Level.SEVERE, "SAAJ0575.soap.attachment.getcontent.exception", ex);
- throw new SOAPExceptionImpl(ex.getLocalizedMessage());
- }
- }
-
- public void setContent(Object object, String contentType)
- throws IllegalArgumentException {
- if (mimePart != null) {
- mimePart.close();
- mimePart = null;
- }
- DataHandler dh = new DataHandler(object, contentType);
-
- setDataHandler(dh);
- }
-
-
- public DataHandler getDataHandler() throws SOAPException {
- if (mimePart != null) {
- //return an inputstream
- return new DataHandler(new DataSource() {
-
- public InputStream getInputStream() throws IOException {
- return mimePart.read();
- }
-
- public OutputStream getOutputStream() throws IOException {
- throw new UnsupportedOperationException("getOutputStream cannot be supported : You have enabled LazyAttachments Option");
- }
-
- public String getContentType() {
- return mimePart.getContentType();
- }
-
- public String getName() {
- return "MIMEPart Wrapper DataSource";
- }
- });
- }
- if (dataHandler == null) {
- if (rawContent != null) {
- return new DataHandler(new MimePartDataSource(rawContent));
- }
- log.severe("SAAJ0502.soap.no.handler.for.attachment");
- throw new SOAPExceptionImpl("No data handler associated with this attachment");
- }
- return dataHandler;
- }
-
- public void setDataHandler(DataHandler dataHandler)
- throws IllegalArgumentException {
- if (mimePart != null) {
- mimePart.close();
- mimePart = null;
- }
- if (dataHandler == null) {
- log.severe("SAAJ0503.soap.no.null.to.dataHandler");
- throw new IllegalArgumentException("Null dataHandler argument to setDataHandler");
- }
- this.dataHandler = dataHandler;
- rawContent = null;
-
- if (log.isLoggable(Level.FINE))
- log.log(Level.FINE, "SAAJ0580.soap.set.Content-Type",
- new String[] { dataHandler.getContentType() });
- setMimeHeader("Content-Type", dataHandler.getContentType());
- }
-
- public void removeAllMimeHeaders() {
- headers.removeAllHeaders();
- }
-
- public void removeMimeHeader(String header) {
- headers.removeHeader(header);
- }
-
- public String[] getMimeHeader(String name) {
- return headers.getHeader(name);
- }
-
- public void setMimeHeader(String name, String value) {
- headers.setHeader(name, value);
- }
-
- public void addMimeHeader(String name, String value) {
- headers.addHeader(name, value);
- }
-
- public Iterator getAllMimeHeaders() {
- return headers.getAllHeaders();
- }
-
- public Iterator getMatchingMimeHeaders(String[] names) {
- return headers.getMatchingHeaders(names);
- }
-
- public Iterator getNonMatchingMimeHeaders(String[] names) {
- return headers.getNonMatchingHeaders(names);
- }
-
- boolean hasAllHeaders(MimeHeaders hdrs) {
- if (hdrs != null) {
- Iterator i = hdrs.getAllHeaders();
- while (i.hasNext()) {
- MimeHeader hdr = (MimeHeader) i.next();
- String[] values = headers.getHeader(hdr.getName());
- boolean found = false;
-
- if (values != null) {
- for (int j = 0; j < values.length; j++)
- if (hdr.getValue().equalsIgnoreCase(values[j])) {
- found = true;
- break;
- }
- }
-
- if (!found) {
- return false;
- }
- }
- }
- return true;
- }
-
- MimeBodyPart getMimePart() throws SOAPException {
- try {
- if (this.mimePart != null) {
- return new MimeBodyPart(mimePart);
- }
- if (rawContent != null) {
- copyMimeHeaders(headers, rawContent);
- return rawContent;
- }
-
- MimeBodyPart envelope = new MimeBodyPart();
-
- envelope.setDataHandler(dataHandler);
- copyMimeHeaders(headers, envelope);
-
- return envelope;
- } catch (Exception ex) {
- log.severe("SAAJ0504.soap.cannot.externalize.attachment");
- throw new SOAPExceptionImpl("Unable to externalize attachment", ex);
- }
- }
-
- public static void copyMimeHeaders(MimeHeaders headers, MimeBodyPart mbp)
- throws SOAPException {
-
- Iterator i = headers.getAllHeaders();
-
- while (i.hasNext())
- try {
- MimeHeader mh = (MimeHeader) i.next();
-
- mbp.setHeader(mh.getName(), mh.getValue());
- } catch (Exception ex) {
- log.severe("SAAJ0505.soap.cannot.copy.mime.hdr");
- throw new SOAPExceptionImpl("Unable to copy MIME header", ex);
- }
- }
-
- public static void copyMimeHeaders(MimeBodyPart mbp, AttachmentPartImpl ap)
- throws SOAPException {
- try {
- List hdr = mbp.getAllHeaders();
- int sz = hdr.size();
- for( int i=0; i<sz; i++ ) {
- Header h = (Header)hdr.get(i);
- if(h.getName().equalsIgnoreCase("Content-Type"))
- continue; // skip
- ap.addMimeHeader(h.getName(), h.getValue());
- }
- } catch (Exception ex) {
- log.severe("SAAJ0506.soap.cannot.copy.mime.hdrs.into.attachment");
- throw new SOAPExceptionImpl(
- "Unable to copy MIME headers into attachment",
- ex);
- }
- }
-
- public void setBase64Content(InputStream content, String contentType)
- throws SOAPException {
-
- if (mimePart != null) {
- mimePart.close();
- mimePart = null;
- }
- dataHandler = null;
- InputStream decoded = null;
- ByteOutputStream bos = null;
- try {
- decoded = MimeUtility.decode(content, "base64");
- InternetHeaders hdrs = new InternetHeaders();
- hdrs.setHeader("Content-Type", contentType);
- //TODO: reading the entire attachment here is ineffcient. Somehow the MimeBodyPart
- // Ctor with inputStream causes problems based on the InputStream
- // has markSupported()==true
- bos = new ByteOutputStream();
- bos.write(decoded);
- rawContent = new MimeBodyPart(hdrs, bos.getBytes(), bos.getCount());
- setMimeHeader("Content-Type", contentType);
- } catch (Exception e) {
- log.log(Level.SEVERE, "SAAJ0578.soap.attachment.setbase64content.exception", e);
- throw new SOAPExceptionImpl(e.getLocalizedMessage());
- } finally {
- if (bos != null)
- bos.close();
- try {
- if (decoded != null)
- decoded.close();
- } catch (IOException ex) {
- throw new SOAPException(ex);
- }
- }
- }
-
- public InputStream getBase64Content() throws SOAPException {
- InputStream stream;
- if (mimePart != null) {
- stream = mimePart.read();
- } else if (rawContent != null) {
- try {
- stream = rawContent.getInputStream();
- } catch (Exception e) {
- log.log(Level.SEVERE,"SAAJ0579.soap.attachment.getbase64content.exception", e);
- throw new SOAPExceptionImpl(e.getLocalizedMessage());
- }
- } else if (dataHandler != null) {
- try {
- stream = dataHandler.getInputStream();
- } catch (IOException e) {
- log.severe("SAAJ0574.soap.attachment.datahandler.ioexception");
- throw new SOAPExceptionImpl("DataHandler error" + e);
- }
- } else {
- log.severe("SAAJ0572.soap.no.content.for.attachment");
- throw new SOAPExceptionImpl("No data handler/content associated with this attachment");
- }
-
- //TODO: Write a BASE64EncoderInputStream instead,
- // this code below is inefficient
- // where we are trying to read the whole attachment first
- int len;
- int size = 1024;
- byte [] buf;
- if (stream != null) {
- try {
- ByteArrayOutputStream bos = new ByteArrayOutputStream(size);
- //TODO: try and optimize this on the same lines as
- // ByteOutputStream : to eliminate the temp buffer here
- OutputStream ret = MimeUtility.encode(bos, "base64");
- buf = new byte[size];
- while ((len = stream.read(buf, 0, size)) != -1) {
- ret.write(buf, 0, len);
- }
- ret.flush();
- buf = bos.toByteArray();
- return new ByteArrayInputStream(buf);
- } catch (Exception e) {
- // throw new SOAPException
- log.log(Level.SEVERE,"SAAJ0579.soap.attachment.getbase64content.exception", e);
- throw new SOAPExceptionImpl(e.getLocalizedMessage());
- } finally {
- try {
- stream.close();
- } catch (IOException ex) {
- //close the stream
- }
- }
- } else {
- //throw new SOAPException
- log.log(Level.SEVERE,"SAAJ0572.soap.no.content.for.attachment");
- throw new SOAPExceptionImpl("No data handler/content associated with this attachment");
- }
- }
-
- public void setRawContent(InputStream content, String contentType)
- throws SOAPException {
- if (mimePart != null) {
- mimePart.close();
- mimePart = null;
- }
- dataHandler = null;
- ByteOutputStream bos = null;
- try {
- InternetHeaders hdrs = new InternetHeaders();
- hdrs.setHeader("Content-Type", contentType);
- //TODO: reading the entire attachment here is ineffcient. Somehow the MimeBodyPart
- // Ctor with inputStream causes problems based on whether the InputStream has
- // markSupported()==true or false
- bos = new ByteOutputStream();
- bos.write(content);
- rawContent = new MimeBodyPart(hdrs, bos.getBytes(), bos.getCount());
- setMimeHeader("Content-Type", contentType);
- } catch (Exception e) {
- log.log(Level.SEVERE, "SAAJ0576.soap.attachment.setrawcontent.exception", e);
- throw new SOAPExceptionImpl(e.getLocalizedMessage());
- } finally {
- if (bos != null)
- bos.close();
- try {
- content.close();
- } catch (IOException ex) {
- throw new SOAPException(ex);
- }
- }
- }
-
- /*
- public void setRawContentBytes(byte[] content, String contentType)
- throws SOAPException {
- if (content == null) {
- throw new SOAPExceptionImpl("Null content passed to setRawContentBytes");
- }
- dataHandler = null;
- try {
- InternetHeaders hdrs = new InternetHeaders();
- hdrs.setHeader("Content-Type", contentType);
- rawContent = new MimeBodyPart(hdrs, content, content.length);
- setMimeHeader("Content-Type", contentType);
- } catch (Exception e) {
- log.log(Level.SEVERE, "SAAJ0576.soap.attachment.setrawcontent.exception", e);
- throw new SOAPExceptionImpl(e.getLocalizedMessage());
- }
- } */
-
- public void setRawContentBytes(
- byte[] content, int off, int len, String contentType)
- throws SOAPException {
- if (mimePart != null) {
- mimePart.close();
- mimePart = null;
- }
- if (content == null) {
- throw new SOAPExceptionImpl("Null content passed to setRawContentBytes");
- }
- dataHandler = null;
- try {
- InternetHeaders hdrs = new InternetHeaders();
- hdrs.setHeader("Content-Type", contentType);
- rawContent = new MimeBodyPart(hdrs, content, off, len);
- setMimeHeader("Content-Type", contentType);
- } catch (Exception e) {
- log.log(Level.SEVERE,
- "SAAJ0576.soap.attachment.setrawcontent.exception", e);
- throw new SOAPExceptionImpl(e.getLocalizedMessage());
- }
- }
-
- public InputStream getRawContent() throws SOAPException {
- if (mimePart != null) {
- return mimePart.read();
- }
- if (rawContent != null) {
- try {
- return rawContent.getInputStream();
- } catch (Exception e) {
- log.log(Level.SEVERE,"SAAJ0577.soap.attachment.getrawcontent.exception", e);
- throw new SOAPExceptionImpl(e.getLocalizedMessage());
- }
- } else if (dataHandler != null) {
- try {
- return dataHandler.getInputStream();
- } catch (IOException e) {
- log.severe("SAAJ0574.soap.attachment.datahandler.ioexception");
- throw new SOAPExceptionImpl("DataHandler error" + e);
- }
- } else {
- log.severe("SAAJ0572.soap.no.content.for.attachment");
- throw new SOAPExceptionImpl("No data handler/content associated with this attachment");
- }
- }
-
- public byte[] getRawContentBytes() throws SOAPException {
- InputStream ret;
- if (mimePart != null) {
- try {
- ret = mimePart.read();
- return ASCIIUtility.getBytes(ret);
- } catch (IOException ex) {
- log.log(Level.SEVERE,"SAAJ0577.soap.attachment.getrawcontent.exception", ex);
- throw new SOAPExceptionImpl(ex);
- }
- }
- if (rawContent != null) {
- try {
- ret = rawContent.getInputStream();
- return ASCIIUtility.getBytes(ret);
- } catch (Exception e) {
- log.log(Level.SEVERE,"SAAJ0577.soap.attachment.getrawcontent.exception", e);
- throw new SOAPExceptionImpl(e);
- }
- } else if (dataHandler != null) {
- try {
- ret = dataHandler.getInputStream();
- return ASCIIUtility.getBytes(ret);
- } catch (IOException e) {
- log.severe("SAAJ0574.soap.attachment.datahandler.ioexception");
- throw new SOAPExceptionImpl("DataHandler error" + e);
- }
- } else {
- log.severe("SAAJ0572.soap.no.content.for.attachment");
- throw new SOAPExceptionImpl("No data handler/content associated with this attachment");
- }
- }
-
- // attachments are equal if they are the same reference
- public boolean equals(Object o) {
- return (this == o);
- }
-
- // In JDK 8 we get a warning if we implement equals() but not hashCode().
- // There is no intuitive value for this, the default one in Object is fine.
- public int hashCode() {
- return super.hashCode();
- }
-
- public MimeHeaders getMimeHeaders() {
- return headers;
- }
-
- public static void initializeJavaActivationHandlers() {
- // DataHandler.writeTo() may search for DCH. So adding some default ones.
- try {
- CommandMap map = CommandMap.getDefaultCommandMap();
- if (map instanceof MailcapCommandMap) {
- MailcapCommandMap mailMap = (MailcapCommandMap) map;
-
- // registering our DCH since javamail's DCH doesn't handle
- if (!cmdMapInitialized(mailMap)) {
- mailMap.addMailcap("text/xml;;x-java-content-handler=com.sun.xml.internal.messaging.saaj.soap.XmlDataContentHandler");
- mailMap.addMailcap("application/xml;;x-java-content-handler=com.sun.xml.internal.messaging.saaj.soap.XmlDataContentHandler");
- mailMap.addMailcap("application/fastinfoset;;x-java-content-handler=com.sun.xml.internal.messaging.saaj.soap.FastInfosetDataContentHandler");
- //mailMap.addMailcap("multipart/*;;x-java-content-handler=com.sun.xml.internal.messaging.saaj.soap.MultipartDataContentHandler");
- mailMap.addMailcap("image/*;;x-java-content-handler=com.sun.xml.internal.messaging.saaj.soap.ImageDataContentHandler");
- mailMap.addMailcap("text/plain;;x-java-content-handler=com.sun.xml.internal.messaging.saaj.soap.StringDataContentHandler");
- }
- }
- } catch (Throwable t) {
- // ignore the exception.
- }
- }
-
- private static boolean cmdMapInitialized(MailcapCommandMap mailMap) {
-
- // checking fastinfoset handler, since this one is specific to SAAJ
- CommandInfo[] commands = mailMap.getAllCommands("application/fastinfoset");
- if (commands == null || commands.length == 0) {
- return false;
- }
-
- String saajClassName = "com.sun.xml.internal.ws.binding.FastInfosetDataContentHandler";
- for (CommandInfo command : commands) {
- String commandClass = command.getCommandClass();
- if (saajClassName.equals(commandClass)) {
- return true;
- }
- }
- return false;
- }
-}