-
Notifications
You must be signed in to change notification settings - Fork 15
Expand file tree
/
Copy pathsetup.py
More file actions
executable file
·78 lines (71 loc) · 2.72 KB
/
setup.py
File metadata and controls
executable file
·78 lines (71 loc) · 2.72 KB
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
#!/usr/bin/env python
import timeit
from setuptools import setup, Command
import versioneer
cmdclass = {}
cmdclass.update(versioneer.get_cmdclass())
class Speed(Command):
description = "run speed benchmarks"
user_options = []
boolean_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
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
def abbrev(t):
if t > 1.0:
return "%.3fs" % t
if t > 1e-3:
return "%.1fms" % (t*1e3)
return "%.1fus" % (t*1e6)
for params in ["ParamsEd25519",
"Params1024", "Params2048", "Params3072"]:
S1 = "from spake2 import SPAKE2_A, SPAKE2_B; from spake2.parameters.all import %s" % params
S2 = "sB = SPAKE2_B(b'password', params=%s)" % params
S3 = "mB = sB.start()"
S4 = "sA = SPAKE2_A(b'password', params=%s)" % params
S5 = "mA = sA.start()"
S8 = "key = sA.finish(mB)"
full = do([S1, S2, S3], ";".join([S4, S5, S8]))
start = do([S1], ";".join([S4, S5]))
# how large is the generated message?
from spake2.parameters import all as all_params
from spake2 import SPAKE2_A
p = getattr(all_params, params)
s = SPAKE2_A(b"pw", params=p)
msglen = len(s.start())
statelen = len(s.serialize())
print("%-13s: msglen=%3d, statelen=%3d, full=%6s, start=%6s"
% (params, msglen, statelen, abbrev(full), abbrev(start)))
cmdclass["speed"] = Speed
setup(name="spake2",
version=versioneer.get_version(),
description="SPAKE2 password-authenticated key exchange (pure python)",
author="Brian Warner",
author_email="warner-pyspake2@lothar.com",
url="https://github.com/warner/python-spake2",
package_dir={"": "src"},
packages=["spake2", "spake2.parameters", "spake2.test"],
license="MIT",
cmdclass=cmdclass,
classifiers=[
"Intended Audience :: Developers",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python",
"Programming Language :: Python :: 3",
"Topic :: Security :: Cryptography",
],
install_requires=["cryptography"],
)