From b3cf7d7737f1fb32bff222e184ae62a3b44f736f Mon Sep 17 00:00:00 2001
From: Magnus Ahltorp <map@kth.se>
Date: Fri, 22 May 2015 16:23:13 +0200
Subject: Don't require --publickey, but print better error message.

---
 tools/certtools.py | 7 ++++++-
 tools/check-sth.py | 2 +-
 2 files changed, 7 insertions(+), 2 deletions(-)

(limited to 'tools')

diff --git a/tools/certtools.py b/tools/certtools.py
index 405aabd..ccbc76e 100644
--- a/tools/certtools.py
+++ b/tools/certtools.py
@@ -219,7 +219,12 @@ def encode_signature(hash_alg, signature_alg, unpacked_signature):
 
 def check_signature(baseurl, signature, data, publickey=None):
     if publickey == None:
-        publickey = base64.decodestring(publickeys[baseurl])
+        if baseurl in publickeys:
+            publickey = base64.decodestring(publickeys[baseurl])
+        else:
+            print >>sys.stderr, "Public key for", baseurl, \
+                "not found, specify key file with --publickey"
+            sys.exit(1)
     (hash_alg, signature_alg, unpacked_signature) = decode_signature(signature)
     assert hash_alg == 4, \
         "hash_alg is %d, expected 4" % (hash_alg,) # sha256
diff --git a/tools/check-sth.py b/tools/check-sth.py
index 0cdc031..dacd8e6 100755
--- a/tools/check-sth.py
+++ b/tools/check-sth.py
@@ -26,8 +26,8 @@ parser.add_argument('--cur-sth',
                     metavar='file',
                     default=DEFAULT_CUR_FILE,
                     help="File containing current STH (default=%s)" % DEFAULT_CUR_FILE)
-parser.add_argument('publickey', help='File containing the public key for the CT log')
 parser.add_argument('baseurl', help="Base URL for CT log")
+parser.add_argument('--publickey', default=None, metavar='file', help='Public key for the CT log')
 parser.add_argument('--cafile', default=None, metavar='file', help='File containing the CA cert')
 parser.add_argument('--allow-lag', action='store_true', help='Allow node to lag behind previous STH')
 parser.add_argument('--quiet-ok', action='store_true', help="Don't print status if OK")
-- 
cgit v1.1


From 42eebf1c68849b1367a8e87757f8019732058c50 Mon Sep 17 00:00:00 2001
From: Magnus Ahltorp <map@kth.se>
Date: Fri, 22 May 2015 16:27:48 +0200
Subject: Fsync logorder in merge Closes CATLFISH-46

---
 tools/merge.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'tools')

diff --git a/tools/merge.py b/tools/merge.py
index b426039..8766491 100755
--- a/tools/merge.py
+++ b/tools/merge.py
@@ -73,6 +73,11 @@ def add_to_logorder(key):
     f.write(base64.b16encode(key) + "\n")
     f.close()
 
+def fsync_logorder():
+    f = open(logorderfile, "a")
+    os.fsync(f.fileno())
+    f.close()
+
 def get_new_entries(node, baseurl):
     try:
         result = http_request(baseurl + "ct/storage/fetchnewentries", key=own_key, verifynode=node, publickeydir=paths["publickeys"])
@@ -228,6 +233,7 @@ for storagenode in storagenodes:
             logorder.append(hash)
             certsinlog.add(hash)
             added_entries += 1
+fsync_logorder()
 timing_point(timing, "add entries")
 print "added", added_entries, "entries"
 
-- 
cgit v1.1