Prettyprint application/x-rpki, first cut.

svn path=/scripts/format-application-x-rpki.py; revision=2922

1 files changed, 60 insertions, 0 deletions

diff --git a/scripts/format-application-x-rpki.py b/scripts/format-application-x-rpki.py
new file mode 100644
index 00000000..f3dc4d48
--- /dev/null
+++ b/scripts/format-application-x-rpki.py
@@ -0,0 +1,60 @@
+# $Id$
+#
+# Copyright (C) 2010  Internet Systems Consortium ("ISC")
+# 
+# Permission to use, copy, modify, and distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notice and this permission notice appear in all copies.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
+# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+# AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
+# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
+# OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+# PERFORMANCE OF THIS SOFTWARE.
+
+# Take the basic application/x-rpki messages that rpkid and friends
+# log and translate them into an expanded version that's easier to
+# view without losing any of the original content.
+#
+# Python's mailbox library rocks.
+
+import email.mime, email.mime.application, email.mime.text, email.mime.multipart, email.utils, email.encoders
+import mailbox, POW, lxml.etree
+
+multipart = False
+source_name = "/tmp/deaddrop"
+destination_name = "~/.mh_dir/pretty-x-rpki"
+
+destination = None
+try:
+  destination = mailbox.MH(destination_name, factory = None, create = True)
+
+  for key in destination.iterkeys():
+    destination.discard(key)
+  destination.flush()
+
+  # See http://bugs.python.org/issue7627 for why we only set this lock
+  # -after- discarding old messages in destination mailbox.
+  destination.lock()
+
+  for message in mailbox.Maildir(source_name, factory = None):
+    assert not message.is_multipart() and message.get_content_type() == "application/x-rpki"
+    headers = dict(kv for kv in message.items())
+    payload = message.get_payload(decode = True)
+    cms = POW.derRead(POW.CMS_MESSAGE, payload)
+    txt = cms.verify(POW.X509Store(), None, POW.CMS_NOCRL | POW.CMS_NO_SIGNER_CERT_VERIFY | POW.CMS_NO_ATTR_VERIFY | POW.CMS_NO_CONTENT_VERIFY)
+    msg = email.mime.text.MIMEText(txt)
+    if multipart:
+      msg = email.mime.multipart.MIMEMultipart("related", None, (msg, email.mime.application.MIMEApplication(payload, "x-rpki", email.encoders.encode_7or8bit)))
+    for k, v in headers.iteritems():
+      if k not in msg:
+        msg[k] = v
+    key = destination.add(msg)
+    print "Added", key
+
+finally:
+  if destination:
+    destination.unlock()
+    destination.close()