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()