Changeset 52 for branches/3.0/turbomail/extensions/smime.py

Timestamp:

10/26/07 05:48:32 (1 year ago)

Author:

amcgregor

Message:

TurboMail 3.0 is now usable: the smtp and debug providers are working, immediate and demand managers are good-to-go, and S/MIME signatures work, too\!

Files:

• branches/3.0/turbomail/extensions/smime.py (modified) (1 diff)

branches/3.0/turbomail/extensions/smime.py

@@ +1 @@
+# encoding: utf-8
+
+"""TurboMail extension API."""
+
+import logging
+log = logging.getLogger("turbomail.utf8qp")
+
+import turbomail, os
+from email.mime.multipart import MIMEMultipart
+from email.mime.application import MIMEApplication
+
+__all__ = ['start']
+
+
+
+class Message(turbomail.Message):
+        def __init__(self, **kw):
+                super(Message, self).__init__(**kw)
+                
+                self.sign = kw.get("sign", turbomail.config.get("mail.message.sign", False))
+                self.encrypt = kw.get("encrypt", turbomail.config.get("mail.message.encrypt", False))
+                self.force = kw.get("force", turbomail.config.get("mail.smime.force", False))
+        
+        def process_mime_message(self, message):
+                newmessage = message
+                
+                if self.sign:
+                        from M2Crypto import BIO, Rand, SMIME
+                        
+                        # Make a MemoryBuffer of the message.
+                        buf = BIO.MemoryBuffer(str(message))
+                        
+                        # Seed the PRNG.
+                        Rand.load_file('randpool.dat', -1)
+                        
+                        # Instantiate an SMIME object; set it up; sign the buffer.
+                        s = SMIME.SMIME()
+                        s.load_key(
+                                        os.path.join(turbomail.config.get("mail.smime.keystore", "./"), str(self.author.addresses[0]) + '.key'),
+                                        os.path.join(turbomail.config.get("mail.smime.keystore", "./"), str(self.author.addresses[0]) + '.x509'),
+                                )
+                        p7 = s.sign(buf)
+                        
+                        # I'm assuming -a-lot- here.  There's got to be a better way to do this, but the M2Crypto lib sucks ass from this PoV.
+                        newmessage = MIMEMultipart("signed", protocol='application/x-pkcs7-signature', micalg="sha1")
+                        newmessage.attach(message)
+                        
+                        buf = BIO.MemoryBuffer()
+                        p7.write_der(buf)
+                        signature = MIMEApplication(buf.read(), "x-pkcs7-signature")
+                        signature.set_param("name", "smime.p7s")
+                        del signature['Content-Disposition']
+                        signature.add_header("Content-Disposition", "attachment", filename="smime.p7s")
+                        
+                        newmessage.attach(signature)
+                        
+                        # Save the PRNG's state.
+                        Rand.save_file('randpool.dat')
+                
+                return newmessage