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