p = 11
g = 2
d1 = DiffieHellman(generator=g, prime=p, key_length=3)
# d2 = DiffieHellman(generator=g, prime=p, key_length=3)
# print(d1.public_key)
# print(d2.public_key)
# d1.gen_key(d2.public_key)
# d2.gen_key(d1.public_key)
d1.gen_key(2)
print('Public key: ' + str(d1.public_key))
print('Private key: ' + str(d1.private_key))
print(d1.get_key())
# d1.show_params()
# d1.show_results()
# d2.show_params()
# d2.show_results()
# if d1.get_key() == d2.get_key():
# print('Shared keys match')
# print('Key: ', hexlify(bytes(d1.key, 'ascii')))
# print(d1.get_key())
# else:
# print("Shared secrets didn't match!")
# print("Shared secret A: ", d1.gen_secret(d2.public_key))
# print("Shared secret B: ", d2.gen_secret(d1.public_key))
