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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
import Pyro4
import Pyro4.errors
from diffiehellman import DiffieHellman
Pyro4.config.SERVERTYPE = "multiplex"
@Pyro4.expose
class SecretStuff(object):
def process(self, message):
if not self._pyroDaemon._pyroHmacKey:
raise Pyro4.errors.PyroError("no hmac key has been set, can't call this method!")
print("Got the message:", message)
return "message was received ok"
ns = Pyro4.locateNS()
daemon = Pyro4.Daemon()
daemon._pyroHmacKey = b"will be set to shared secret key by KeyExchange"
uri = daemon.register(SecretStuff)
ns.register("example.dh.secretstuff", uri)
@Pyro4.behavior(instance_mode="session")
class KeyExchange(object):
def __init__(self):
print("New KeyExchange, initializing Diffie-Hellman")
self.dh = DiffieHellman(group=14)
@Pyro4.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("setting new shared secret key.")
global daemon
daemon._pyroHmacKey = self.dh.key
return self.dh.public_key
# The key exchange service can't be part of the same daemon as the
# other service because it must not have a Hmac key set on the daemon.
# So we create another daemon without hmac key and combine it.
key_daemon = Pyro4.Daemon()
uri = key_daemon.register(KeyExchange)
ns.register("example.dh.keyexchange", uri)
ns._pyroRelease()
key_daemon.combine(daemon)
print("Starting server loop...")
key_daemon.requestLoop()
|
