Crypto zmq.rpc

rpctest.py

import sys

from Crypto.Cipher import Blowfish

from zmq.rpc.simplerpc import (
    RPCService,
    RPCServiceProxy,
    rpc_method,
    Serializer
)
from zmq.utils import jsonapi


class EncryptedJSONSerializer(Serializer):
    def __init__(self, key):
        self.crypt = Blowfish.new(key)

    def dumps(self, msg):
        jmsg = jsonapi.dumps(msg)
        jmsg += " " * (8 - len(jmsg) % 8)
        return self.crypt.encrypt(jmsg)

    def loads(self, msg):
        return jsonapi.loads(self.crypt.decrypt(msg))


class MehService(RPCService):

    @rpc_method
    def add(self, a, b):
        return sum([a, b])


if __name__ == '__main__':
    if sys.argv[1] == 'client':
        client = RPCServiceProxy(serializer=EncryptedJSONSerializer('hello'))
        client.connect('tcp://127.0.0.1:8888')
        print client.add(1, 2)
    else:
        server = MehService(serializer=EncryptedJSONSerializer('hello'))
        server.bind('tcp://127.0.0.1:8888')
        server.start()