programing

Java Mail을 사용하여 첨부 파일 다운로드

projobs 2022. 10. 21. 21:56
반응형

Java Mail을 사용하여 첨부 파일 다운로드

이제 모든 메시지를 다운로드하여 다음 위치에 저장했습니다.

Message[] temp;

각 메시지에 대한 첨부 파일 목록을 가져오려면 어떻게 해야 합니까?

List<File> attachments;

주의: 서드파티 libs는 사용하지 마시고 Java Mail만 부탁드립니다.

예외 없이 처리할 수 있습니다.

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);
    }
}

질문은 아주 오래되었지만, 아마도 누군가에게 도움이 될 것이다.데이비드 라비노위츠의 답변을 확대해 보겠습니다.

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;
}

첨부 파일을 저장하는 코드의 시간 절약:

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();

Commons IO 및 Commons Lang에서 Apache Commons Mail API MimeMessageParser - getAttachmentList()를 사용할 수 있습니다.

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
    }
}

첨부 파일과 함께 신체 부위 목록을 반환합니다.

@Throws(Exception::class)
    fun getAttachments(message: Message): List<BodyPart>{
        val content = message.content
        if (content is String) return ArrayList<BodyPart>()
        if (content is Multipart) {
            val result: MutableList<BodyPart> = ArrayList<BodyPart>()
            for (i in 0 until content.count) {
                result.addAll(getAttachments(content.getBodyPart(i)))
            }
            return result
        }
        return ArrayList<BodyPart>()
    }

    @Throws(Exception::class)
    private fun getAttachments(part: BodyPart): List<BodyPart> {
        val result: MutableList<BodyPart> = ArrayList<BodyPart>()

        if (Part.ATTACHMENT == part.disposition && !part.fileName.isNullOrBlank()){
            result.add(part)
        }

        val content = part.content
        if (content is Multipart) {
            for (i in 0 until (content ).count) {
                val bodyPart = content.getBodyPart(i)
                result.addAll(getAttachments(bodyPart)!!)
            }
        }
        return result
    }

언급URL : https://stackoverflow.com/questions/1748183/download-attachments-using-java-mail

반응형