diff options
Diffstat (limited to 'src/leap/bonafide/zmq_service.py')
| -rw-r--r-- | src/leap/bonafide/zmq_service.py | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/src/leap/bonafide/zmq_service.py b/src/leap/bonafide/zmq_service.py index df8e234..38c0305 100644 --- a/src/leap/bonafide/zmq_service.py +++ b/src/leap/bonafide/zmq_service.py @@ -14,11 +14,13 @@ # # You should have received a copy of the GNU General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. + """ Bonafide ZMQ Service. """ + from leap.bonafide import config -from leap.bonafide.protocol import BonafideProtocol, COMMANDS +from leap.bonafide._protocol import BonafideProtocol, COMMANDS from txzmq import ZmqEndpoint, ZmqFactory, ZmqREPConnection @@ -27,6 +29,8 @@ from twisted.internet import reactor from twisted.python import log +# TODO [] should shutdown all the ongoing connections when stopping the service + class BonafideZMQService(service.Service): def __init__(self): @@ -36,9 +40,14 @@ class BonafideZMQService(service.Service): def startService(self): zf = ZmqFactory() e = ZmqEndpoint("bind", config.ENDPOINT) + self._conn = _BonafideZmqREPConnection(zf, e, self._bonafide) reactor.callWhenRunning(self._conn.do_greet) + #def stopService(self): + # pass + + class _BonafideZmqREPConnection(ZmqREPConnection): @@ -50,7 +59,7 @@ class _BonafideZmqREPConnection(ZmqREPConnection): print "Bonafide service running..." def do_bye(self): - print "[+] Bonafide service stopped. Have a nice day." + print "Bonafide service stopped. Have a nice day." reactor.stop() def gotMessage(self, msgId, *parts): @@ -59,7 +68,6 @@ class _BonafideZmqREPConnection(ZmqREPConnection): def log_err(failure): log.err(failure) - print "FAILURE", failure defer_reply("ERROR: %r" % failure) cmd = parts[0] |