1 /* 2 * Copyright (c) 1997, 2014, 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 26 package com.sun.xml.internal.messaging.saaj.packaging.mime.internet; 27 28 import com.sun.xml.internal.messaging.saaj.packaging.mime.MessagingException; 29 import com.sun.xml.internal.messaging.saaj.soap.AttachmentPartImpl; 30 import java.io.IOException; 31 import java.io.InputStream; 32 import java.util.List; 33 import javax.activation.DataSource; 34 import javax.xml.soap.AttachmentPart; 35 import com.sun.xml.internal.org.jvnet.mimepull.MIMEConfig; 36 import com.sun.xml.internal.org.jvnet.mimepull.MIMEMessage; 37 import com.sun.xml.internal.org.jvnet.mimepull.MIMEPart; 38 39 /** 40 * 41 * @author Kumar 42 */ 43 public class MimePullMultipart extends MimeMultipart { 44 45 private InputStream in = null; 46 private String boundary = null; 47 private MIMEMessage mm = null; 48 private DataSource dataSource = null; 49 private ContentType contType = null; 50 private String startParam = null; 51 private MIMEPart soapPart = null; 52 53 public MimePullMultipart(DataSource ds, ContentType ct) 54 throws MessagingException { 55 parsed = false; 56 if (ct==null) 57 contType = new ContentType(ds.getContentType()); 58 else 59 contType = ct; 60 61 dataSource = ds; 62 boundary = contType.getParameter("boundary"); 63 } 64 65 public MIMEPart readAndReturnSOAPPart() throws MessagingException { 66 if (soapPart != null) { 67 throw new MessagingException("Inputstream from datasource was already consumed"); 68 } 69 readSOAPPart(); 70 return soapPart; 71 72 } 73 74 protected void readSOAPPart() throws MessagingException { 75 try { 76 if (soapPart != null) { 77 return; 78 } 79 in = dataSource.getInputStream(); 80 MIMEConfig config = new MIMEConfig(); //use defaults 81 mm = new MIMEMessage(in, boundary, config); 82 String st = contType.getParameter("start"); 83 if(startParam == null) { 84 soapPart = mm.getPart(0); 85 } else { 86 // Strip <...> from root part's Content-I 87 if (st != null && st.length() > 2 && st.charAt(0) == '<' && st.charAt(st.length()-1) == '>') { 88 st = st.substring(1, st.length()-1); 89 } 90 startParam = st; 91 soapPart = mm.getPart(startParam); 92 93 } 94 } catch (IOException ex) { 95 throw new MessagingException("No inputstream from datasource", ex); 96 } 97 } 98 99 public void parseAll() throws MessagingException { 100 if (parsed) { 101 return; 102 } 103 if (soapPart == null) { 104 readSOAPPart(); 105 } 106 107 List<MIMEPart> prts = mm.getAttachments(); 108 for(MIMEPart part : prts) { 109 if (part != soapPart) { 110 new AttachmentPartImpl(part); 111 this.addBodyPart(new MimeBodyPart(part)); 112 } 113 } 114 parsed = true; 115 } 116 117 protected void parse() throws MessagingException { 118 parseAll(); 119 } 120 121 }