Загрузить вложения с помощью Java Mail

96

Теперь, когда я загрузил все сообщения и сохранил их в

Message[] temp;

Как получить список вложений для каждого из этих сообщений в

List<File> attachments;

Примечание: никаких сторонних библиотек, только JavaMail.

фолоне
источник

Ответы:

110

Без обработки исключений, но здесь:

List<File> attachments = new ArrayList<File>();
for (Message message : temp) {
    Multipart multipart = (Multipart) message.getContent();

    for (int i = 0; i < multipart.getCount(); i++) {
        BodyPart bodyPart = multipart.getBodyPart(i);
        if(!Part.ATTACHMENT.equalsIgnoreCase(bodyPart.getDisposition()) &&
               StringUtils.isBlank(bodyPart.getFileName())) {
            continue; // dealing with attachments only
        } 
        InputStream is = bodyPart.getInputStream();
        // -- EDIT -- SECURITY ISSUE --
        // do not do this in production code -- a malicious email can easily contain this filename: "../etc/passwd", or any other path: They can overwrite _ANY_ file on the system that this code has write access to!
//      File f = new File("/tmp/" + bodyPart.getFileName());
        FileOutputStream fos = new FileOutputStream(f);
        byte[] buf = new byte[4096];
        int bytesRead;
        while((bytesRead = is.read(buf))!=-1) {
            fos.write(buf, 0, bytesRead);
        }
        fos.close();
        attachments.add(f);
    }
}
Дэвид Рабиновиц
источник
2
Но подождите минутку, разве мы не должны проверять if (bodyPart.getDisposition () == Part.ATTACHMENT) {} перед сохранением файла, чтобы он не сохранял тело письма?
folone
8
Разве StringUtils.isBlank () не было бы более естественным для чтения, чем использование! StringUtils.isNotBlank?
Kuchi
В этом ответе не рассматриваются вложенные составные вложения (обычно используемые, например, Thunderbird). Чтобы найти вложенные многокомпонентные вложения, см. Ответ @mefi.
Руслан Стельмаченко
3
Фрагмент кода - это ожидаемое нарушение безопасности. Я отредактировал фрагмент, чтобы выделить это.
rzwitserloot 05
1
Конечно, и спасибо за это. Этот фрагмент предназначен только для демонстрации того, как это можно сделать, естественно, это не производственный код
Дэвид Рабинович
33

Вопрос очень старый, но может кому-то поможет. Я хотел бы расширить ответ Дэвида Рабиновица.

if(!Part.ATTACHMENT.equalsIgnoreCase(bodyPart.getDisposition()))

не должны возвращать все вложения, как вы ожидаете, потому что вы можете получать почту, в которой смешанная часть находится без определенного расположения.

   ----boundary_328630_1e15ac03-e817-4763-af99-d4b23cfdb600
Content-Type: application/octet-stream;
    name="00000000009661222736_236225959_20130731-7.txt"
Content-Transfer-Encoding: base64

поэтому в этом случае вы также можете проверить имя файла. Как это:

if (!Part.ATTACHMENT.equalsIgnoreCase(part.getDisposition()) && StringUtils.isBlank(part.getFileName())) {...}

РЕДАКТИРОВАТЬ

существует весь рабочий код с использованием условия, описанного выше. Поскольку каждая часть может инкапсулировать другие части, а вложение должно быть вложенным, для обхода всех частей используется рекурсия.

public List<InputStream> getAttachments(Message message) throws Exception {
    Object content = message.getContent();
    if (content instanceof String)
        return null;        

    if (content instanceof Multipart) {
        Multipart multipart = (Multipart) content;
        List<InputStream> result = new ArrayList<InputStream>();

        for (int i = 0; i < multipart.getCount(); i++) {
            result.addAll(getAttachments(multipart.getBodyPart(i)));
        }
        return result;

    }
    return null;
}

private List<InputStream> getAttachments(BodyPart part) throws Exception {
    List<InputStream> result = new ArrayList<InputStream>();
    Object content = part.getContent();
    if (content instanceof InputStream || content instanceof String) {
        if (Part.ATTACHMENT.equalsIgnoreCase(part.getDisposition()) || StringUtils.isNotBlank(part.getFileName())) {
            result.add(part.getInputStream());
            return result;
        } else {
            return new ArrayList<InputStream>();
        }
    }

    if (content instanceof Multipart) {
            Multipart multipart = (Multipart) content;
            for (int i = 0; i < multipart.getCount(); i++) {
                BodyPart bodyPart = multipart.getBodyPart(i);
                result.addAll(getAttachments(bodyPart));
            }
    }
    return result;
}
мефи
источник
выражение проверяет наличие пустого имени файла или null. Это правильно?
Кеэртхивасан,
Осьминог: Да. Проверяет, не является ли CharSequence пустым (""), не пустым и не только пробелом.
mefi
Эй, вы можете сказать, как преобразовать List <InputStream> в List <File>?
kumuda
kumunda: привет, вы должны повторить этот список и использовать org.apache.commons.io.FileUtils.copyInputStreamToFile (источник InputStream, место назначения файла) и каждый добавить файл в другую коллекцию. Если вы используете Guava, проверьте Lists.transform (...), который вы можете использовать вместо итерации (зависит от того, как вам нужно инициализировать каждый экземпляр файла)
mefi
9

Некоторая экономия времени для кода, в котором вы сохраняете файл вложения:

с javax mail версии 1.4 и более поздних вы можете сказать

// SECURITY LEAK - do not do this! Do not trust the 'getFileName' input. Imagine it is: "../etc/passwd", for example.
// bodyPart.saveFile("/tmp/" + bodyPart.getFileName());

вместо того

    InputStream is = bodyPart.getInputStream();
    File f = new File("/tmp/" + bodyPart.getFileName());
    FileOutputStream fos = new FileOutputStream(f);
    byte[] buf = new byte[4096];
    int bytesRead;
    while((bytesRead = is.read(buf))!=-1) {
        fos.write(buf, 0, bytesRead);
    }
    fos.close();
Коммрад
источник
3
Видимо bodyPart сначала нужно преобразовать в MimeBodyPart, например:((MimeBodyPart) bodyPart).saveFile("/tmp/" + bodyPart.getFileName());
yair 09
5

Вы можете просто использовать Apache Commons Mail API MimeMessageParser - getAttachmentList () вместе с Commons IO и Commons Lang.

MimeMessageParser parser = ....
parser.parse();
for(DataSource dataSource : parser.getAttachmentList()) {

    if (StringUtils.isNotBlank(dataSource.getName())) {}

        //use apache commons IOUtils to save attachments
        IOUtils.copy(dataSource.getInputStream(), ..dataSource.getName()...)
    } else {
        //handle how you would want attachments without file names
        //ex. mails within emails have no file name
    }
}
Stackee007
источник