summaryrefslogtreecommitdiffstats
path: root/src/console/zmq/auth/ioloop.py
diff options
context:
space:
mode:
authorDan Klein <danklei@cisco.com>2015-08-24 09:28:43 +0300
committerDan Klein <danklei@cisco.com>2015-08-24 09:28:43 +0300
commit20eb7d362f9bce1951bd61ad3f78cf8f4267d1d5 (patch)
treeb2c3fe1c721d5f6898b4bb8e81f7c68502e313e9 /src/console/zmq/auth/ioloop.py
parent49f6b00b58c3ec734218fcd69259771a42c157bd (diff)
parent651a7d779551e193bd9dbadbe8b2a02bdab231b4 (diff)
Merge branch 'master' into dan_stateless
Diffstat (limited to 'src/console/zmq/auth/ioloop.py')
-rwxr-xr-xsrc/console/zmq/auth/ioloop.py34
1 files changed, 34 insertions, 0 deletions
diff --git a/src/console/zmq/auth/ioloop.py b/src/console/zmq/auth/ioloop.py
new file mode 100755
index 00000000..1f448b47
--- /dev/null
+++ b/src/console/zmq/auth/ioloop.py
@@ -0,0 +1,34 @@
+"""ZAP Authenticator integrated with the tornado IOLoop.
+
+.. versionadded:: 14.1
+"""
+
+# Copyright (C) PyZMQ Developers
+# Distributed under the terms of the Modified BSD License.
+
+from zmq.eventloop import ioloop, zmqstream
+from .base import Authenticator
+
+
+class IOLoopAuthenticator(Authenticator):
+ """ZAP authentication for use in the tornado IOLoop"""
+
+ def __init__(self, context=None, encoding='utf-8', log=None, io_loop=None):
+ super(IOLoopAuthenticator, self).__init__(context)
+ self.zap_stream = None
+ self.io_loop = io_loop or ioloop.IOLoop.instance()
+
+ def start(self):
+ """Start ZAP authentication"""
+ super(IOLoopAuthenticator, self).start()
+ self.zap_stream = zmqstream.ZMQStream(self.zap_socket, self.io_loop)
+ self.zap_stream.on_recv(self.handle_zap_message)
+
+ def stop(self):
+ """Stop ZAP authentication"""
+ if self.zap_stream:
+ self.zap_stream.close()
+ self.zap_stream = None
+ super(IOLoopAuthenticator, self).stop()
+
+__all__ = ['IOLoopAuthenticator']