main_bench.py

import timeit

#Testa DH generando simulando lo scambio di chiavi tra due utenti, e controllando che entrambi generino la stessa chiave z
def key_test(dh):
    try:
        d1,Q1=dh.key_pair_generator()
        d2,Q2=dh.key_pair_generator()
        z1=dh.shared_key_generator(d1,Q2)
        z2=dh.shared_key_generator(d2,Q1)
    except:
        print("*** An error has occurred, and a shared key could not be generated ***")
        raise ValueError("Error during the generation")
    
    if z1==z2:
        #print("-Key["+hex(z1)+"] successfully generated")
        pass
    else:
        print("*** Two different secret key were generated ***")
        raise ValueError("Two different secret key were generated")

#Genera un certo numero di chiavi su di una curva
#prima con double-and-add, poi con wNAF
#e misura il tempo di esecuzione con timeit
def bench(curve,times):
    print("Type of curve: "+curve)
    SETUP='''
from __main__ import key_test
from dh import ECC_CDH
dh=ECC_CDH("'''+curve+'''")'''
    code='''key_test(dh)'''
    d_a_a=timeit.timeit(setup=SETUP,stmt=code,number=times)
    print("*** All keys correctley generated ***")
    print(str(times)+" keys generated using Double-and-Add in "+str(d_a_a)+" seconds")

    SETUP='''
from __main__ import key_test
from dh import ECC_CDH
dh=ECC_CDH("'''+curve+'''",wNAF=1)'''
    code='''key_test(dh)'''
    wNAF=timeit.timeit(setup=SETUP, stmt=code,number=times)
    print("*** All keys correctley generated ***")
    print(str(times)+" keys generated using wNAF in "+str(wNAF)+" seconds")

    if(wNAF<d_a_a):
        print("*** WNAF IS FASTER***")
    else:
        print("*** DOUBLE AND ADD IS FASTER***")
    print("----------------------------------------------------")

def multiple_bench(curve,keyTimes,nBench):
    for i in range(0,nBench):
        bench(curve,keyTimes)
    print("----------------------------------------------------")

multiple_bench("curves/p192.xml",1000,10)
multiple_bench("curves/p224.xml",800,10)
multiple_bench("curves/p256.xml",600,10)
multiple_bench("curves/p384.xml",400,10)
multiple_bench("curves/p521.xml",200,10)