1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
from Pyro5.api import behavior, expose, locate_ns, serve, config
from diffiehellman import DiffieHellman
config.SERVERTYPE = "multiplex"
ns = locate_ns()
@behavior(instance_mode="session")
class KeyExchange(object):
def __init__(self):
print("New KeyExchange, initializing Diffie-Hellman")
self.dh = DiffieHellman(group=14)
@expose
def exchange_key(self, other_public_key):
print("received a public key, calculating shared secret...")
self.dh.make_shared_secret_and_key(other_public_key)
print("shared secret key = ", self.dh.key.hex())
return self.dh.public_key
serve({
KeyExchange: "example.dh.keyexchange"
}, use_ns=True)
|
