com.sun.xml.messaging.saaj.soap.AttachmentPartImpl Maven / Gradle / Ivy
/*
* Copyright (c) 1997, 2020 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Distribution License v. 1.0, which is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
package com.sun.xml.messaging.saaj.soap;
import com.sun.xml.messaging.saaj.SOAPExceptionImpl;
import com.sun.xml.messaging.saaj.packaging.mime.util.ASCIIUtility;
import com.sun.xml.messaging.saaj.packaging.mime.Header;
import com.sun.xml.messaging.saaj.packaging.mime.internet.MimePartDataSource;
import com.sun.xml.messaging.saaj.packaging.mime.internet.InternetHeaders;
import com.sun.xml.messaging.saaj.packaging.mime.internet.MimeBodyPart;
import com.sun.xml.messaging.saaj.packaging.mime.internet.MimeUtility;
import com.sun.xml.messaging.saaj.util.ByteOutputStream;
import com.sun.xml.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 jakarta.activation.*;
import jakarta.xml.soap.*;
import org.jvnet.mimepull.MIMEPart;
/**
* Implementation of attachments.
*
* @author Anil Vijendran ([email protected])
*/
public class AttachmentPartImpl extends AttachmentPart {
protected static final Logger log =
Logger.getLogger(LogDomainConstants.SOAP_DOMAIN,
"com.sun.xml.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 hdrs = part.getAllHeaders();
for (org.jvnet.mimepull.Header hd : hdrs) {
headers.addHeader(hd.getName(), hd.getValue());
}
}
@Override
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();
}
}
@Override
public void clearContent() {
if (mimePart != null) {
mimePart.close();
mimePart = null;
}
dataHandler = null;
rawContent = null;
}
@Override
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());
}
}
@Override
public void setContent(Object object, String contentType)
throws IllegalArgumentException {
if (mimePart != null) {
mimePart.close();
mimePart = null;
}
DataHandler dh = new DataHandler(object, contentType);
setDataHandler(dh);
}
@Override
public DataHandler getDataHandler() throws SOAPException {
if (mimePart != null) {
//return an inputstream
return new DataHandler(new DataSource() {
@Override
public InputStream getInputStream() throws IOException {
return mimePart.read();
}
@Override
public OutputStream getOutputStream() throws IOException {
throw new UnsupportedOperationException("getOutputStream cannot be supported : You have enabled LazyAttachments Option");
}
@Override
public String getContentType() {
return mimePart.getContentType();
}
@Override
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;
}
@Override
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());
}
@Override
public void removeAllMimeHeaders() {
headers.removeAllHeaders();
}
@Override
public void removeMimeHeader(String header) {
headers.removeHeader(header);
}
@Override
public String[] getMimeHeader(String name) {
return headers.getHeader(name);
}
@Override
public void setMimeHeader(String name, String value) {
headers.setHeader(name, value);
}
@Override
public void addMimeHeader(String name, String value) {
headers.addHeader(name, value);
}
@Override
public Iterator getAllMimeHeaders() {
return headers.getAllHeaders();
}
@Override
public Iterator getMatchingMimeHeaders(String[] names) {
return headers.getMatchingHeaders(names);
}
@Override
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