def test_parse_double_attached_with_images(self): raw_email = read_file_to_string(GET_DOUBLE_ATTACHED_WITH_IMAGES) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(email.account, TEST_MAILBOX_ID) self.assertEqual(email.subject, "Fw: User Test") self.assertEqual(len(email.attached_emails), 2) self.assertEqual(len(email.attached_files), 2) self.assertEqual(email.date_received, "Tue, 13 Aug 2019 20:05:12 +0000") self.assertEqual(len(email.headers), 53) self.assertEqual(email.is_read, False) self.assertEqual(email.recipients, ["*****@*****.**"]) self.assertEqual(email.sender, "*****@*****.**") email1 = email.attached_emails[0] email2 = email.attached_emails[1] self.assertTrue(email1.subject, "Pic and eml attached") self.assertTrue(email2.subject, "Test Email") attached_file = email.attached_files[0] self.assertEqual(attached_file.indicators.md5, "2fca7949ad1004cefe685b81c3889e1c") self.assertEqual(attached_file.indicators.sha1, "b7e20bc9d4eb40adb1c4f103821bd461deab9d3f") self.assertEqual( attached_file.indicators.sha256, "8477aeb65fe7985cc82bc8f231ebfc519b8178d1050a9cee7f1b450e6c370240", )
def test_parse_raw_with_unicode(self): raw_email = read_file_to_string(GET_UNICODE_EML) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(email.subject, "Grüße von Stefan Appel")
def test_parse_from_raw_evil_email(self): raw_email = read_file_to_string(GET_PAYLOAD_EVIL_EMAIL) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(email.account, TEST_MAILBOX_ID) self.assertEqual(email.subject, "FW: [POTENTIAL PHISH] MESINC") self.assertEqual(len(email.attached_emails), 2) self.assertEqual(len(email.attached_files), 4) self.assertEqual(email.date_received, "Thu, 1 Aug 2019 10:36:33 +0000") self.assertEqual(len(email.headers), 69) self.assertEqual(email.is_read, False) self.assertEqual( email.recipients, [ "*****@*****.**", "*****@*****.**", ], ) self.assertEqual(email.sender, "*****@*****.**") self.assertEqual(email.indicators.md5, "1f224c7839c3f7d825864f322a6bb57a") self.assertEqual(email.indicators.sha1, "83dfe966303673168c344c286bb16d2801d1919f") self.assertEqual( email.indicators.sha256, "251ac384891e16088f0be97cf6a872559fd287f1e163e4b83f6b49b8ad6862b8", )
def test_parse_from_raw4(self): raw_email = read_file_to_string(GET_RAW_ATTACHMENT_PAYLOAD4) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(email.account, TEST_MAILBOX_ID) self.assertEqual(email.subject, "Basic Message Attachment") self.assertEqual(len(email.attached_emails), 0) self.assertEqual(len(email.attached_files), 0) expected_body = """<html><head> <meta http-equiv="Content-Type" content="text/html; charset=iso-8859-1"> <style type="text/css" style="display:none;"> P {margin-top:0;margin-bottom:0;} </style> </head> <body dir="ltr"> <div style="font-family: Calibri, Helvetica, sans-serif; font-size: 12pt; color: rgb(0, 0, 0);"> Nothing here, just this text</div> </body> </html> """ self.assertEqual(email.body, expected_body) self.assertEqual(email.indicators.md5, "c537b7db873e8e2cd6b965be6afdd063") self.assertEqual(email.indicators.sha1, "b1f5dd056c0930042fa5549b4c5c8aa6ecb70e3a") self.assertEqual( email.indicators.sha256, "50b779af79f6493165c0b8de3e56c08e0cc3eec3c50e2f9f2ddfe80247b8e886", ) self.assertEqual(email.date_received, "Thu, 8 Aug 2019 21:19:37 +0000") self.assertEqual(len(email.headers), 76) self.assertEqual(email.is_read, False) self.assertEqual(email.recipients, ["*****@*****.**"]) self.assertEqual(email.sender, "*****@*****.**")
def test_parse_from_raw3(self): raw_email = read_file_to_string(GET_RAW_ATTACHMENT_PAYLOAD3) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(email.account, TEST_MAILBOX_ID) self.assertEqual(email.subject, "Text Attachment") self.assertEqual(len(email.attached_emails), 0) self.assertEqual(len(email.attached_files), 1) self.assertTrue("Here is a text attachment" in email.body) self.assertEqual(email.date_received, "Thu, 8 Aug 2019 20:16:38 +0000") self.assertEqual(len(email.headers), 76) self.assertEqual(email.is_read, False) self.assertEqual(email.recipients, ["*****@*****.**"]) self.assertEqual(email.sender, "*****@*****.**") attachment = email.attached_files[0] self.assertEqual(attachment.content_type, "text/plain") expected_content = "VGhpcyBpcyBhIHRlc3QgYXR0YWNobWVudA0KDQpJdCBoYXMgc29tZSB0ZXh0IGluIGl0LiANCg0KYWFkcm9pZC5uZXQNCg==" self.assertEqual(attachment.content, expected_content) self.assertEqual(attachment.name, "test_example.com") self.assertEqual(attachment.indicators.md5, "593aa3b46e3902094303b7ef1349d9ff") self.assertEqual(attachment.indicators.sha1, "d1181c07e87d73803bf5786b37e8f03a176290a2") self.assertEqual( attachment.indicators.sha256, "c08eb82e5383760cad3a4b4863dfb871b4c4252eba039c64a90ffc818907de27", )
def test_parse_attached_eml(self): raw_email = read_file_to_string(GET_EML_WITH_EML_ATTACHED) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(len(email.attached_emails), 4) self.assertEqual(len(email.attached_files), 2)
def test_decode_quoted_printable(self): raw_email = read_file_to_string( f"{CURRENT_DIR}/payloads/quoted_printable.eml") email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) expected = '<p class="MsoNormal"><a href="http://example.com" title="http://proteexample.com.com/s/414KCXDXZofXVRNRZT6ai-n?domain=example.com">http://aexample.com</a><o:p></o:p></p>' self.assertTrue(expected in email.body)
def test_parse_google_link_garbled(self): raw_email = read_file_to_string(GET_GOOGLE_SURVEY) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertTrue( '<a href="http://aexample.com/" rel="nofollow" target="_blank">' in email.body)
def test_parse_from_raw(self): raw_email = read_file_to_string(GET_RAW_ATTACHMENT_PAYLOAD) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(email.account, TEST_MAILBOX_ID) self.assertEqual(email.subject, "level 3") self.assertEqual(len(email.attached_emails), 2) self.assertEqual(len(email.attached_files), 1) self.assertEqual(email.date_received, "Thu, 8 Aug 2019 17:29:14 +0000") self.assertEqual(len(email.headers), 73) self.assertEqual(email.is_read, False) self.assertEqual(email.recipients, ["*****@*****.**"]) self.assertEqual(email.sender, "*****@*****.**") self.assertTrue("level 3" in email.body) self.assertEqual(email.indicators.md5, "a35d0f590b9e2aa1a6afa68880836777") self.assertEqual(email.indicators.sha1, "2b725fe68564fb791c5a33f451c3fbf589e1d2e8") self.assertEqual( email.indicators.sha256, "226a485db0230d4bc95c892f29b7257fee1aefb9ed273ec4cc4abe134a6b8a6b", ) attached_email = email.attached_emails[0] self.assertEqual(attached_email.subject, "Level 2 subject") self.assertEqual(attached_email.sender, "*****@*****.**") self.assertEqual(attached_email.recipients, ["*****@*****.**"]) self.assertTrue("Level 2 body" in attached_email.body) self.assertEqual(attached_email.indicators.md5, "224662a660009095b8eb905a9c0d6964") self.assertEqual(attached_email.indicators.sha1, "8da1890238848f6b8f1c776b0699babb0925f256") self.assertEqual( attached_email.indicators.sha256, "1950205c6953757c7edc8afa12cf26dc87d5e888a49828c997b404fe5fb51b3b", ) attached_email_of_attached_email = attached_email.attached_emails[0] self.assertEqual(attached_email_of_attached_email.subject, "Pic attached") attached_file = email.attached_files[0] self.assertEqual(attached_file.content[:5], "iVBOR") self.assertEqual(attached_file.content_type, "image/png") self.assertEqual(attached_file.name, "example.com") self.assertEqual(attached_file.indicators.md5, "539cefc749ed1d78e3c821307f7c1b0a") self.assertEqual(attached_file.indicators.sha1, "9a95c01f15c461708733dac2b3fbe10901801582") self.assertEqual( attached_file.indicators.sha256, "ac92883cdc6cda0735f9c3f968e6103f8dd93f705cc077dce70924eda523a916", )
def test_hash(self): email_with_nested_attachments_text = read_file_to_string( f"{CURRENT_DIR}/payloads/hash_crash.eml") email_parser = EmailParser(self.logger) icon_email = email_parser.make_email_from_raw( message_from_string(email_with_nested_attachments_text), "fake_account", ) actual = icon_email.__hash__() print(type(actual)) self.assertIsInstance(actual, int)
def test_flatten_real_data(self): email_with_nested_attachments_text = read_file_to_string( f"{CURRENT_DIR}/payloads/2 level deep email attached.eml") email_parser = EmailParser(self.logger) icon_email = email_parser.make_email_from_raw( message_from_string(email_with_nested_attachments_text), "fake_account", ) icon_email.flatten() self.assertEqual(icon_email.subject, "Fw: 2 level deep email attached") self.assertEqual(len(icon_email.flattened_attached_emails), 2) self.assertEqual(len(icon_email.flattened_attached_files), 0) attached_email0 = icon_email.flattened_attached_emails[0] attached_email1 = icon_email.flattened_attached_emails[1] self.assertEqual(attached_email0.subject, "Attachment") self.assertEqual(attached_email1.subject, "Test Message Attachment Subject")
def test_parse_from_raw2(self): raw_email = read_file_to_string(GET_RAW_ATTACHMENT_PAYLOAD2) email_parser = EmailParser(self.log) email = email_parser.make_email_from_raw( message_from_string(raw_email), TEST_MAILBOX_ID) self.assertEqual(email.account, TEST_MAILBOX_ID) self.assertEqual(email.subject, "Attachment") self.assertTrue("Attachment Body" in email.body) self.assertEqual(email.indicators.md5, "267cc9490cec0c3229795e5eb7ef2a25") self.assertEqual(email.indicators.sha1, "e62d99edae16e70bfcf9bd041c08afd183271c56") self.assertEqual( email.indicators.sha256, "aae84980564cc107ea5ef7c3ea393fb18d678fea5956b7b5005f8ae7664bf78c", ) self.assertEqual(len(email.attached_emails), 1) self.assertEqual(len(email.attached_files), 0) self.assertTrue("Attachment Body" in email.body) attached_email = email.attached_emails[0] self.assertTrue("Test Message Attachment Body" in attached_email.body) self.assertEqual(attached_email.indicators.md5, "d8fbc36661e49768c7db3160afdeaf70") self.assertEqual(attached_email.indicators.sha1, "bfccda3b5ade42569e05753fa1ad1f05c4ce2095") self.assertEqual( attached_email.indicators.sha256, "b6ef02bc495b3710bd822a789e2b916c610b6fc78aab1f5a82bb2676fd6ad664", ) self.assertEqual(email.date_received, "Tue, 6 Aug 2019 19:19:40 +0000") self.assertEqual(len(email.headers), 75) self.assertEqual(email.is_read, False) self.assertEqual(email.recipients, ["*****@*****.**"]) self.assertEqual(email.sender, "*****@*****.**")