main/java/MSMultipartDataSource.java

/*
 * Copyright (c) 1997-2010 Oracle and/or its affiliates. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *   - Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *
 *   - Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in the
 *     documentation and/or other materials provided with the distribution.
 *
 *   - Neither the name of Oracle nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

import java.util.*;
import javax.mail.*;
import javax.mail.internet.*;

/**
 * A special MultipartDataSource used with MSMessage.
 *
 * @author John Mani
 * @author Bill Shannon
 */
public class MSMultipartDataSource extends MimePartDataSource
                implements MultipartDataSource {
    //private List<MSBodyPart> parts;
    private List parts;

    public MSMultipartDataSource(MimePart part, byte[] content)
                throws MessagingException {
    super(part);
    //parts = new ArrayList<MSBodyPart>();
    parts = new ArrayList();

    /*
     * Parse the text of the message to find the attachments.
     *
     * Currently we just look for the lines that mark the
     * begin and end of uuencoded data, but this can be
     * fooled by similar text in the message body.  Instead,
     * we could use the Encoding header, which indicates how
     * many lines are in each body part.  For example:
     *
     * Encoding: 41 TEXT, 38 UUENCODE, 3155 UUENCODE, 1096 UUENCODE
     *
     * Similarly, we could get the filenames of the attachments
     * from the X-MS-Attachment headers.  For example:
     *
     * X-MS-Attachment: ATT00000.htx 0 00-00-1980 00:00
     * X-MS-Attachment: Serengeti 2GG.mpp 0 00-00-1980 00:00
     * X-MS-Attachment: project team update 031298.doc 0 00-00-1980 00:00
     *
     * (Note that there might be unquoted spaces in the filename.)
     */
    int pos = startsWith(content, 0, "begin");
    if (pos == -1)
        throw new MessagingException("invalid multipart");

    if (pos > 0)    // we have an unencoded main body part
        parts.add(new MSBodyPart(content, 0, pos, "inline", "7bit"));
    else        // no main body part
        pos = 0;

    // now collect all the uuencoded individual body parts
    int start;
    for (;;) {
        start = startsWith(content, pos, "begin");
        if (start == -1)
        break;
        pos = startsWith(content, start, "end");
        if (pos == -1)
        break;
        pos += 3;   // skip to the end of "end"
        parts.add(new MSBodyPart(content, start, pos,
                    "attachment", "uuencode"));
    }
    }

    public int getCount() {
    return parts.size();
    }

    public BodyPart getBodyPart(int index) throws MessagingException {
    return (BodyPart)parts.get(index);
    }

    /**
     * This method scans the given byte[], beginning at "start", for
     * lines that begin with the sequence "seq".  If found, the start
     * position of the sequence within the byte[] is returned.
     */
    private int startsWith(byte[] content, int start, String seq) {
    int slen = seq.length();
    boolean bol = true;
    for (int i = start; i < content.length; i++) {
        if (bol) {
        if ((i + slen) < content.length) {
            String s = MSMessage.toString(content, i, i + slen);
            if (s.equalsIgnoreCase(seq))
            return i;
        }
        }
        int b = content[i] & 0xff;
        bol = b == '\r' || b == '\n';
    }
    return -1;
    }
}