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