Add CTR mode benchmark

1 files changed, 8 insertions, 0 deletions

diff --git a/pct-speedtest.py b/pct-speedtest.py
index 132736c..365eca2 100644
--- a/pct-speedtest.py
+++ b/pct-speedtest.py
@@ -33,6 +33,8 @@ from Crypto.Signature import PKCS1_PSS, PKCS1_v1_5 as RSASSA_PKCS1_v1_5
 from Crypto.Cipher import AES, ARC2, ARC4, Blowfish, CAST, DES3, DES, XOR
 from Crypto.Hash import HMAC, MD2, MD4, MD5, SHA224, SHA256, SHA384, SHA512
 from Crypto.Random import get_random_bytes
+import Crypto.Util.Counter
+from Crypto.Util.number import bytes_to_long
 try:
     from Crypto.Hash import SHA1
 except ImportError:
@@ -165,6 +167,11 @@ class Benchmark:
         elif mode == "CTR-LE":
             from Crypto.Util import Counter
             cipher = module.new(key, module.MODE_CTR, counter=Counter.new(module.block_size*8, little_endian=True))
+        elif mode==module.MODE_CTR:
+            ctr = Crypto.Util.Counter.new(module.block_size*8,
+                    initial_value=bytes_to_long(iv),
+                    allow_wraparound=True)
+            cipher = module.new(key, module.MODE_CTR, counter=ctr)
         else:
             cipher = module.new(key, mode, iv)
 
@@ -335,6 +342,7 @@ class Benchmark:
             self.test_encryption("%s-CFB-8" % (cipher_name,), module, key_bytes, module.MODE_CFB)
             self.test_encryption("%s-OFB" % (cipher_name,), module, key_bytes, module.MODE_OFB)
             self.test_encryption("%s-ECB" % (cipher_name,), module, key_bytes, module.MODE_ECB)
+            self.test_encryption("%s-CTR" % (cipher_name,), module, key_bytes, module.MODE_CTR)
             self.test_encryption("%s-OPENPGP" % (cipher_name,), module, key_bytes, module.MODE_OPENPGP)
             self.test_encryption("%s-CTR-BE" % (cipher_name,), module, key_bytes, "CTR-BE")
             self.test_encryption("%s-CTR-LE" % (cipher_name,), module, key_bytes, "CTR-LE")