Merge pull request #2 from keraattin/development

File & content digests extracting added

email-analyzer.py

@@ -1,9 +1,11 @@
 #!/usr/bin/env python3
 
+from email import message_from_string
 from email.parser import HeaderParser
 import pyfiglet
 from argparse import ArgumentParser
 import sys
+import hashlib
 
 SUPPORTED_FILE_TYPES = ["eml"]
 
@@ -19,6 +21,31 @@ def get_headers(mail_data : str):
         print(val)
         print("_"*70)
 
+def get_digests(mail_data : str, filename : str):
+    '''Get & Print Hash value of mail'''
+    with open(filename, 'rb') as f:
+        file        = f.read()
+        file_md5    = hashlib.md5(file).hexdigest()
+        file_sha1   = hashlib.sha1(file).hexdigest()
+        file_sha256 = hashlib.sha256(file).hexdigest()
+
+    digests = {
+        "File MD5":file_md5,
+        "File SHA1":file_sha1,
+        "File SHA256":file_sha256,
+        "Content MD5":hashlib.md5(mail_data.encode("utf-8")).hexdigest(),
+        "Content SHA1":hashlib.sha1(mail_data.encode("utf-8")).hexdigest(),
+        "Content SHA256":hashlib.sha256(mail_data.encode("utf-8")).hexdigest()
+    }
+
+    print(pyfiglet.figlet_format("Digests")) # Print Banner
+    # Print digests
+    for key,val in digests.items():
+        print("_"*70)
+        print(key+":")
+        print(val)
+        print("_"*70)
+
 # Main
 if __name__ == '__main__':
     parser = ArgumentParser()
@@ -36,6 +63,13 @@ def get_headers(mail_data : str):
         required=False,
         action="store_true"
     )
+    parser.add_argument(
+        "-d",
+        "--digests",
+        help="Digests of the eml file", 
+        required=False,
+        action="store_true"
+    )
     args = parser.parse_args()
 
     # Filename
@@ -55,3 +89,8 @@ def get_headers(mail_data : str):
     if args.headers:
         # Get & Print Headers
         get_headers(data)
+
+    # Digests
+    if args.digests:
+        # Get & Print Digests
+        get_digests(data,filename)