90 lines
2.1 KiB
Python
90 lines
2.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from __future__ import absolute_import, division, print_function
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import sys
|
||
|
|
import timeit
|
||
|
|
|
||
|
|
import six
|
||
|
|
|
||
|
|
from . import (
|
||
|
|
DEFAULT_HASH_LENGTH,
|
||
|
|
DEFAULT_MEMORY_COST,
|
||
|
|
DEFAULT_PARALLELISM,
|
||
|
|
DEFAULT_TIME_COST,
|
||
|
|
PasswordHasher,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def main(argv):
|
||
|
|
parser = argparse.ArgumentParser(description="Benchmark Argon2.")
|
||
|
|
parser.add_argument(
|
||
|
|
"-n", type=int, default=100, help="Number of iterations to measure."
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"-t", type=int, help="`time_cost`", default=DEFAULT_TIME_COST
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"-m", type=int, help="`memory_cost`", default=DEFAULT_MEMORY_COST
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"-p", type=int, help="`parallellism`", default=DEFAULT_PARALLELISM
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"-l", type=int, help="`hash_length`", default=DEFAULT_HASH_LENGTH
|
||
|
|
)
|
||
|
|
|
||
|
|
args = parser.parse_args(argv[1:])
|
||
|
|
|
||
|
|
password = b"secret"
|
||
|
|
ph = PasswordHasher(
|
||
|
|
time_cost=args.t,
|
||
|
|
memory_cost=args.m,
|
||
|
|
parallelism=args.p,
|
||
|
|
hash_len=args.l,
|
||
|
|
)
|
||
|
|
hash = ph.hash(password)
|
||
|
|
|
||
|
|
params = {
|
||
|
|
"time_cost": (args.t, "iterations"),
|
||
|
|
"memory_cost": (args.m, "KiB"),
|
||
|
|
"parallelism": (args.p, "threads"),
|
||
|
|
"hash_len": (args.l, "bytes"),
|
||
|
|
}
|
||
|
|
|
||
|
|
print("Running Argon2id %d times with:" % (args.n,))
|
||
|
|
|
||
|
|
for k, v in sorted(six.iteritems(params)):
|
||
|
|
print("%s: %d %s" % (k, v[0], v[1]))
|
||
|
|
|
||
|
|
print("\nMeasuring...")
|
||
|
|
duration = timeit.timeit(
|
||
|
|
"ph.verify({hash!r}, {password!r})".format(
|
||
|
|
hash=hash, password=password
|
||
|
|
),
|
||
|
|
setup="""\
|
||
|
|
from argon2 import PasswordHasher, Type
|
||
|
|
|
||
|
|
ph = PasswordHasher(
|
||
|
|
time_cost={time_cost!r},
|
||
|
|
memory_cost={memory_cost!r},
|
||
|
|
parallelism={parallelism!r},
|
||
|
|
hash_len={hash_len!r},
|
||
|
|
)
|
||
|
|
gc.enable()""".format(
|
||
|
|
time_cost=args.t,
|
||
|
|
memory_cost=args.m,
|
||
|
|
parallelism=args.p,
|
||
|
|
hash_len=args.l,
|
||
|
|
),
|
||
|
|
number=args.n,
|
||
|
|
)
|
||
|
|
print(
|
||
|
|
"\n{0:.3}ms per password verification".format(duration / args.n * 1000)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__": # pragma: nocover
|
||
|
|
main(sys.argv)
|