def __init__(self):
# Diffie-Hellman core
self.DHE_core = DiffieHellman()
# Peer Public Key
self.public_key = self.DHE_core.publicKey
class ComSecCore:
def __init__(self):
# Diffie-Hellman core
self.DHE_core = DiffieHellman()
# Peer Public Key
self.public_key = self.DHE_core.publicKey
def get_public_key(self):
return self.public_key
def generate_shared_key(self, received_public_key):
self.DHE_core.genKey(received_public_key)
return self.DHE_core.getKey()
