1# It's intended that this script be run by hand.  It runs speed tests on
2# hashlib functions; it does not test for correctness.
3
4import sys, time
5import hashlib
6
7
8def creatorFunc():
9    raise RuntimeError, "eek, creatorFunc not overridden"
10
11def test_scaled_msg(scale, name):
12    iterations = 106201/scale * 20
13    longStr = 'Z'*scale
14
15    localCF = creatorFunc
16    start = time.time()
17    for f in xrange(iterations):
18        x = localCF(longStr).digest()
19    end = time.time()
20
21    print ('%2.2f' % (end-start)), "seconds", iterations, "x", len(longStr), "bytes", name
22
23def test_create():
24    start = time.time()
25    for f in xrange(20000):
26        d = creatorFunc()
27    end = time.time()
28
29    print ('%2.2f' % (end-start)), "seconds", '[20000 creations]'
30
31def test_zero():
32    start = time.time()
33    for f in xrange(20000):
34        x = creatorFunc().digest()
35    end = time.time()
36
37    print ('%2.2f' % (end-start)), "seconds", '[20000 "" digests]'
38
39
40
41hName = sys.argv[1]
42
43#
44# setup our creatorFunc to test the requested hash
45#
46if hName in ('_md5', '_sha'):
47    exec 'import '+hName
48    exec 'creatorFunc = '+hName+'.new'
49    print "testing speed of old", hName, "legacy interface"
50elif hName == '_hashlib' and len(sys.argv) > 3:
51    import _hashlib
52    exec 'creatorFunc = _hashlib.%s' % sys.argv[2]
53    print "testing speed of _hashlib.%s" % sys.argv[2], getattr(_hashlib, sys.argv[2])
54elif hName == '_hashlib' and len(sys.argv) == 3:
55    import _hashlib
56    exec 'creatorFunc = lambda x=_hashlib.new : x(%r)' % sys.argv[2]
57    print "testing speed of _hashlib.new(%r)" % sys.argv[2]
58elif hasattr(hashlib, hName) and callable(getattr(hashlib, hName)):
59    creatorFunc = getattr(hashlib, hName)
60    print "testing speed of hashlib."+hName, getattr(hashlib, hName)
61else:
62    exec "creatorFunc = lambda x=hashlib.new : x(%r)" % hName
63    print "testing speed of hashlib.new(%r)" % hName
64
65try:
66    test_create()
67except ValueError:
68    print
69    print "pass argument(s) naming the hash to run a speed test on:"
70    print " '_md5' and '_sha' test the legacy builtin md5 and sha"
71    print " '_hashlib' 'openssl_hName' 'fast' tests the builtin _hashlib"
72    print " '_hashlib' 'hName' tests builtin _hashlib.new(shaFOO)"
73    print " 'hName' tests the hashlib.hName() implementation if it exists"
74    print "         otherwise it uses hashlib.new(hName)."
75    print
76    raise
77
78test_zero()
79test_scaled_msg(scale=106201, name='[huge data]')
80test_scaled_msg(scale=10620, name='[large data]')
81test_scaled_msg(scale=1062, name='[medium data]')
82test_scaled_msg(scale=424, name='[4*small data]')
83test_scaled_msg(scale=336, name='[3*small data]')
84test_scaled_msg(scale=212, name='[2*small data]')
85test_scaled_msg(scale=106, name='[small data]')
86test_scaled_msg(scale=creatorFunc().digest_size, name='[digest_size data]')
87test_scaled_msg(scale=10, name='[tiny data]')
88