add benchmark tool (just python RSA for now)

Author: tomato42

scripts/speed.py

@@ -0,0 +1,73 @@
+import timeit
+
+from tlslite.utils.cryptomath import gmpyLoaded
+
+print("Acceleration backends loaded:")
+print("gmpy: {0}".format(gmpyLoaded))
+print("")
+
+def do(setup_statements, statement):
+    # extracted from timeit.py
+    t = timeit.Timer(stmt=statement, setup="\n".join(setup_statements))
+    # determine number so that 0.2 <= total time < 2.0
+    for i in range(1, 10):
+        number = 10 ** i
+        x = t.timeit(number)
+        if x >= 0.2:
+            break
+    return x / number
+
+
+prnt_form = (
+    "{name:>16}{sep:1} {keygen:>9{form}}{unit:1} "
+    "{keygen_inv:>9{form_inv}} {sign:>9{form}}{unit:1} "
+    "{sign_inv:>9{form_inv}} {verify:>9{form}}{unit:1} "
+    "{verify_inv:>9{form_inv}}"
+)
+
+print(
+    prnt_form.format(
+        keygen="keygen",
+        keygen_inv="keygen/s",
+        sign="sign",
+        sign_inv="sign/s",
+        verify="verify",
+        verify_inv="verify/s",
+        name="",
+        sep="",
+        unit="",
+        form="",
+        form_inv="",
+    )
+)
+
+for size in [1024, 2048, 3072, 4096]:
+    S1 = "from tlslite.utils.python_rsakey import Python_RSAKey"
+    S2 = "from tlslite.utils.cryptomath import secureHash"
+    S3 = "key = Python_RSAKey.generate(%s)" % size
+    S4 = "msg = b'msg'"
+    S5 = "msg_hash = secureHash(msg, 'sha1')"
+    S6 = "sig = key.sign(msg_hash)"
+    S7 = "key.verify(sig, msg_hash)"
+    keygen = do([S1, S2], S3)
+    sign = do([S1, S2, S3, S4, S5], S6)
+    verf = do([S1, S2, S3, S4, S5, S6], S7)
+
+    print(
+        prnt_form.format(
+            name="RSA {0} bits".format(size),
+            sep=":",
+            unit="s",
+            keygen=keygen,
+            keygen_inv=1.0 / keygen,
+            sign=sign,
+            sign_inv=1.0 / sign,
+            verify=verf,
+            verify_inv=1.0 / verf,
+            form=".5f",
+            form_inv=".2f",
+        )
+    )
+
+print("")
+