benchmark_elgamal_schnorr.py

import os
import pickle

from charm.schemes.pkenc.pkenc_elgamal85 import ElGamal
from charm.toolbox.integergroup import IntegerGroupQ
from charm.schemes.pksig.pksig_schnorr91 import SchnorrSig
from charm.toolbox.conversion import Conversion

trials = 20
group = IntegerGroupQ()
elgamal = ElGamal(group)
schnorr = SchnorrSig()

assert group.InitBenchmark()
group.StartBenchmark(["Mul", "Div", "Exp", "RealTime", "CpuTime", "Add", "Sub"])
for i in range(trials):
    schnorr.params(bits=128)
    pk_sig, sk_sig = schnorr.keygen()
    pk_enc, sk_enc = elgamal.keygen(secparam=2048)

    msg = os.urandom(16)
    original_len = len(msg)

    S = schnorr.sign(pk=pk_sig, x=sk_sig, M=msg)
    signature_bytes = Conversion.IP2OS(S['e']) + Conversion.IP2OS(S['s'])
    ciphertext = elgamal.encrypt(pk_enc, msg)

    ciphertext_dict = {'c1': int(ciphertext['c1']), 'c2': int(ciphertext['c2'])}
    message_len = len(pickle.dumps(ciphertext_dict))
    unencrypted = elgamal.decrypt(pk_enc, sk_enc, ciphertext)

    print(msg)
    print(unencrypted)

    assert msg == unencrypted
    assert schnorr.verify(pk_sig, S, unencrypted)
    print(f'Message overhead iteration {i} is: {((message_len - original_len) / original_len) * 100}%')
group.EndBenchmark()
msmtDict = group.GetGeneralBenchmarks()
print(msmtDict)