fix: Properly handle plain text attachments

pull/31/head
Raphaël Vinot 2018-08-02 17:02:17 +02:00
parent 517bb19786
commit 45fa0c7323
3 changed files with 10 additions and 2 deletions

View File

@ -83,7 +83,7 @@ class Mail2MISP():
forwarded_emails.append(self.forwarded_email(pseudofile=BytesIO(attachment_content.as_bytes())))
else:
if isinstance(attachment_content, str):
attachment_content = BytesIO(attachment_content.encode())
attachment_content = attachment_content.encode()
filename = attachment.get_filename()
if not filename:
filename = 'missing_filename'

@ -1 +1 @@
Subproject commit d57c26e25f6da499a806de2b60f1861f3ed882af
Subproject commit 58d3cfbe1355aa184e0d68fcda88ead371974574

View File

@ -62,5 +62,13 @@ class TestMailToMISP(unittest.TestCase):
self.assertTrue('attachment' in [a.type for a in self.mail2misp.misp_event.attributes])
self.assertTrue(self.mail2misp.misp_event.publish)
def test_textfile(self):
config = importlib.import_module('tests.config_forward')
self.mail2misp = Mail2MISP('', '', '', config=config, offline=True)
with open('tests/mails/test_textattachment.eml', 'rb') as f:
self.mail2misp.load_email(BytesIO(f.read()))
self.mail2misp.process_email_body()
if __name__ == '__main__':
unittest.main()