Key generation timing.

Initial version, very basic, RSA-only.  Gussy up later.

1 files changed, 37 insertions, 0 deletions

diff --git a/tests/time-keygen.py b/tests/time-keygen.py
new file mode 100644
index 0000000..fcd47c6
--- /dev/null
+++ b/tests/time-keygen.py
@@ -0,0 +1,37 @@
+#!/usr/bin/env python
+
+"""
+Time libhal RSA key generation
+"""
+
+from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
+from datetime import datetime, timedelta
+
+from cryptech.libhal import *
+
+parser = ArgumentParser(description = __doc__, formatter_class = ArgumentDefaultsHelpFormatter)
+parser.add_argument("-i", "--iterations", default = 100, type = int,    help = "iterations")
+parser.add_argument("-p", "--pin",        default = "fnord",            help = "user PIN")
+parser.add_argument("-t", "--token",      action  = "store_true",       help = "store key on token")
+parser.add_argument("-k", "--keylen",     default = 2048, type = int,   help = "key length")
+args = parser.parse_args()
+
+hsm = HSM()
+hsm.login(HAL_USER_NORMAL, args.pin)
+
+flags = HAL_KEY_FLAG_USAGE_DIGITALSIGNATURE | (HAL_KEY_FLAG_TOKEN if args.token else 0)
+sum   = timedelta()
+
+for n in xrange(1, args.iterations):
+
+    t0 = datetime.now()
+
+    k  = hsm.pkey_generate_rsa(args.keylen, flags)
+
+    t1 = datetime.now()
+
+    k.delete()
+
+    sum += t1 - t0
+
+    print "{:4d} this {} mean {}".format(n, t1 - t0, sum / n)