Merge remote-tracking branch 'drebs/feature/4356_add-x-leap-provenance-header-to-incoming-emails' into develop

1 files changed, 24 insertions, 12 deletions

diff --git a/src/leap/mx/mail_receiver.py b/src/leap/mx/mail_receiver.py
index 3890eb2..c06754e 100644
--- a/src/leap/mx/mail_receiver.py
+++ b/src/leap/mx/mail_receiver.py
@@ -105,7 +105,7 @@ class MailReceiver(Service):
         :param pubkey: public key for the owner of the message
         :type pubkey: str
         :param message: message contents
-        :type message: str
+        :type message: email.message.Message
 
         :return: doc to sync with Soledad or None, None if something
                  went wrong.
@@ -116,16 +116,19 @@ class MailReceiver(Service):
                     "I know: %r" % (pubkey,))
             return None
 
-        doc = SoledadDocument(doc_id=str(pyuuid.uuid4()))
-
-        encoding = get_email_charset(message.decode("utf8", "replace"),
-                                     default=None)
+        # find message's encoding
+        message_as_string = message.as_string()
+        encoding = get_email_charset(
+            message_as_string.decode("utf8", "replace"),
+            default=None)
         if encoding is None:
-            result = chardet.detect(message)
+            result = chardet.detect(message_as_string)
             encoding = result["encoding"]
 
-        data = {'incoming': True, 'content': message}
+        doc = SoledadDocument(doc_id=str(pyuuid.uuid4()))
 
+        # store plain text if pubkey is not available
+        data = {'incoming': True, 'content': message_as_string}
         if pubkey is None or len(pubkey) == 0:
             doc.content = {
                 self.INCOMING_KEY: True,
@@ -134,7 +137,7 @@ class MailReceiver(Service):
             }
             return doc
 
-        openpgp_key = None
+        # otherwise, encrypt
         with openpgp.TempGPGWrapper(gpgbinary='/usr/bin/gpg') as gpg:
             gpg.import_keys(pubkey)
             key = gpg.list_keys().pop()
@@ -142,6 +145,15 @@ class MailReceiver(Service):
             # dummy one, we just care about the import of the pubkey
             openpgp_key = openpgp._build_key_from_gpg("dummy@mail.com", key, pubkey)
 
+            # add X-Leap-Provenance header if message is not encrypted
+            if message.get_content_type() != 'multipart/encrypted' and \
+                    '-----BEGIN PGP MESSAGE-----' not in \
+                    message_as_string:
+                message.add_header(
+                    'X-Leap-Provenance',
+                    email.utils.formatdate(),
+                    pubkey=openpgp_key.key_id)
+                data = {'incoming': True, 'content': message.as_string()}
             doc.content = {
                 self.INCOMING_KEY: True,
                 ENC_SCHEME_KEY: EncryptionSchemes.PUBKEY,
@@ -284,8 +296,8 @@ class MailReceiver(Service):
         log.msg("Processing new mail at %r" % (filepath.path,))
         with filepath.open("r") as f:
             mail_data = f.read()
-            mail = message_from_string(mail_data)
-            uuid = self._get_owner(mail)
+            msg = message_from_string(mail_data)
+            uuid = self._get_owner(msg)
             if uuid is None:
                 log.msg("Don't know how to deliver mail %r, skipping..." %
                         (filepath.path,))
@@ -297,12 +309,12 @@ class MailReceiver(Service):
                 defer.returnValue(None)
 
             pubkey = yield self._users_cdb.getPubKey(uuid)
-            if pubkey is None or len(pubkey):
+            if pubkey is None or len(pubkey) == 0:
                 log.msg("No public key, stopping the processing chain")
                 defer.returnValue(None)
 
             log.msg("Encrypting message to %s's pubkey" % (uuid,))
-            doc = yield self._encrypt_message(pubkey, mail_data)
+            doc = yield self._encrypt_message(pubkey, msg)
 
             do_remove = yield self._export_message(uuid, doc)
             yield self._conditional_remove(do_remove, filepath)