novnc redis backed token_plyugins.py

2 files changed, 21 insertions, 1 deletions

diff --git a/websockify/auth_plugins.py b/websockify/auth_plugins.py
index ed3a169..2d636c2 100644
--- a/websockify/auth_plugins.py
+++ b/websockify/auth_plugins.py
@@ -87,7 +87,7 @@ class ExpectOrigin(object):
         origin = headers.get('Origin', None)
         if origin is None or origin not in self.source:
             raise InvalidOriginError(expected=self.source, actual=origin)
-            
+
 class ClientCertCNAuth(object):
     """Verifies client by SSL certificate. Specify src as whitespace separated list of common names."""
 
diff --git a/websockify/token_plugins.py b/websockify/token_plugins.py
index 972907b..92aec98 100644
--- a/websockify/token_plugins.py
+++ b/websockify/token_plugins.py
@@ -134,3 +134,23 @@ class JWTTokenApi(BasePlugin):
         except ImportError as e:
             print("package jwcrypto not found, are you sure you've installed it correctly?", file=sys.stderr)
             return None
+
+import sys
+
+if sys.version_info >= (2, 7):
+    import redis
+    import simplejson
+
+    class TokenRedis(object):
+        def __init__(self, src):
+            self._server, self._port = src.split(":")
+
+        def lookup(self, token):
+            client = redis.Redis(host=self._server,port=self._port)
+            stuff = client.get(token)
+            if stuff is None:
+                return None
+            else:
+                combo = simplejson.loads(stuff.decode("utf-8"))
+                pair = combo["host"]
+                return pair.split(':')