From 832dba2e1cd4e6e6ff9e8ea7afa98eebb44595e5 Mon Sep 17 00:00:00 2001
From: Magnus Ahltorp <map@kth.se>
Date: Wed, 28 Jan 2015 13:42:06 +0100
Subject: Move hardcoded merge parameters to command line

---
 tools/certtools.py | 37 +++++++++++++++++++++++--------------
 1 file changed, 23 insertions(+), 14 deletions(-)

(limited to 'tools/certtools.py')

diff --git a/tools/certtools.py b/tools/certtools.py
index cbb4ff7..af94fb8 100644
--- a/tools/certtools.py
+++ b/tools/certtools.py
@@ -42,22 +42,31 @@ def get_cert_info(s):
             result[key] = value
     return result
 
-def get_certs_from_file(certfile):
-    certs = []
-    cert = ""
-    incert = False
 
-    for line in open(certfile):
+def get_pemlike(filename, marker):
+    entries = []
+    entry = ""
+    inentry = False
+
+    for line in open(filename):
         line = line.strip()
-        if line == "-----BEGIN CERTIFICATE-----":
-            cert = ""
-            incert = True
-        elif line == "-----END CERTIFICATE-----":
-            certs.append(base64.decodestring(cert))
-            incert = False
-        elif incert:
-            cert += line
-    return certs
+        if line == "-----BEGIN " + marker + "-----":
+            entry = ""
+            inentry = True
+        elif line == "-----END " + marker + "-----":
+            entries.append(base64.decodestring(entry))
+            inentry = False
+        elif inentry:
+            entry += line
+    return entries
+
+def get_certs_from_file(certfile):
+    return get_pemlike(certfile, "CERTIFICATE")
+
+def get_eckey_from_file(keyfile):
+    keys = get_pemlike(keyfile, "EC PRIVATE KEY")
+    assert len(keys) == 1
+    return keys[0]
 
 def get_root_cert(issuer):
     accepted_certs = \
-- 
cgit v1.1