From f2222868f5fc4c3d962048ff3f4bc39fa2d9b64c Mon Sep 17 00:00:00 2001
From: Magnus Ahltorp <map@kth.se>
Date: Mon, 9 Feb 2015 17:23:13 +0100
Subject: submitcert.py: handle .zip files

---
 tools/certtools.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'tools/certtools.py')

diff --git a/tools/certtools.py b/tools/certtools.py
index f6c1cd9..428d623 100644
--- a/tools/certtools.py
+++ b/tools/certtools.py
@@ -11,6 +11,7 @@ import sys
 import hashlib
 import ecdsa
 import datetime
+import cStringIO
 
 publickeys = {
     "https://ct.googleapis.com/pilot/":
@@ -44,11 +45,14 @@ def get_cert_info(s):
 
 
 def get_pemlike(filename, marker):
+    return get_pemlike_from_file(open(filename), marker)
+
+def get_pemlike_from_file(f, marker):
     entries = []
     entry = ""
     inentry = False
 
-    for line in open(filename):
+    for line in f:
         line = line.strip()
         if line == "-----BEGIN " + marker + "-----":
             entry = ""
@@ -63,6 +67,10 @@ def get_pemlike(filename, marker):
 def get_certs_from_file(certfile):
     return get_pemlike(certfile, "CERTIFICATE")
 
+def get_certs_from_string(s):
+    f = cStringIO.StringIO(s)
+    return get_pemlike_from_file(f, "CERTIFICATE")
+
 def get_eckey_from_file(keyfile):
     keys = get_pemlike(keyfile, "EC PRIVATE KEY")
     assert len(keys) == 1
-- 
cgit v1.1