1/* 2 * Copyright (c) 1997, 2017, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26package com.sun.xml.internal.messaging.saaj.packaging.mime.internet; 27 28import com.sun.xml.internal.messaging.saaj.packaging.mime.MessagingException; 29import com.sun.xml.internal.messaging.saaj.soap.AttachmentPartImpl; 30import java.io.IOException; 31import java.io.InputStream; 32import java.util.List; 33import javax.activation.DataSource; 34import com.sun.xml.internal.org.jvnet.mimepull.MIMEConfig; 35import com.sun.xml.internal.org.jvnet.mimepull.MIMEMessage; 36import com.sun.xml.internal.org.jvnet.mimepull.MIMEPart; 37 38/** 39 * 40 * @author Kumar 41 */ 42public class MimePullMultipart extends MimeMultipart { 43 44 private InputStream in = null; 45 private String boundary = null; 46 private MIMEMessage mm = null; 47 private DataSource dataSource = null; 48 private ContentType contType = null; 49 private String startParam = null; 50 private MIMEPart soapPart = null; 51 52 public MimePullMultipart(DataSource ds, ContentType ct) 53 throws MessagingException { 54 parsed = false; 55 if (ct==null) 56 contType = new ContentType(ds.getContentType()); 57 else 58 contType = ct; 59 60 dataSource = ds; 61 boundary = contType.getParameter("boundary"); 62 } 63 64 public MIMEPart readAndReturnSOAPPart() throws MessagingException { 65 if (soapPart != null) { 66 throw new MessagingException("Inputstream from datasource was already consumed"); 67 } 68 readSOAPPart(); 69 return soapPart; 70 71 } 72 73 protected void readSOAPPart() throws MessagingException { 74 try { 75 if (soapPart != null) { 76 return; 77 } 78 in = dataSource.getInputStream(); 79 MIMEConfig config = new MIMEConfig(); //use defaults 80 mm = new MIMEMessage(in, boundary, config); 81 String st = contType.getParameter("start"); 82 if(startParam == null) { 83 soapPart = mm.getPart(0); 84 } else { 85 // Strip <...> from root part's Content-I 86 if (st != null && st.length() > 2 && st.charAt(0) == '<' && st.charAt(st.length()-1) == '>') { 87 st = st.substring(1, st.length()-1); 88 } 89 startParam = st; 90 soapPart = mm.getPart(startParam); 91 92 } 93 } catch (IOException ex) { 94 throw new MessagingException("No inputstream from datasource", ex); 95 } 96 } 97 98 public void parseAll() throws MessagingException { 99 if (parsed) { 100 return; 101 } 102 if (soapPart == null) { 103 readSOAPPart(); 104 } 105 106 List<MIMEPart> prts = mm.getAttachments(); 107 for(MIMEPart part : prts) { 108 if (part != soapPart) { 109 new AttachmentPartImpl(part); 110 this.addBodyPart(new MimeBodyPart(part)); 111 } 112 } 113 parsed = true; 114 } 115 116 @Override 117 protected void parse() throws MessagingException { 118 parseAll(); 119 } 120 121} 122