UnpicklingError, not UnpickleError

[invirt/packages/invirt-vnc-server.git] / python / vnc / extauth.py

diff --git a/python/vnc/extauth.py b/python/vnc/extauth.py
index a2cd570..c986771 100644 (file)
--- a/python/vnc/extauth.py
+++ b/python/vnc/extauth.py
@@ -20,7 +20,7 @@ import socket
 import time
 
 def getTokenKey():
 import time
 
 def getTokenKey():

-    return file('/etc/invirt/secrets/vnc-key').read().strip()
+    return file('/etc/invirt/vnc/token-key').read().strip()

 
 def getPort(name, auth_data):
     import get_port
 
 def getPort(name, auth_data):
     import get_port

@@ -65,11 +65,10 @@ class VNCAuth(protocol.Protocol):
     def validateToken(self, token):
         self.auth_error = "Invalid token"
         try:
     def validateToken(self, token):
         self.auth_error = "Invalid token"
         try:

-            token = base64.urlsafe_b64decode(token)
-            token = cPickle.loads(token)
+            (pickled_data, digest) = map(base64.urlsafe_b64decode, token.split("."))

             m = hmac.new(getTokenKey(), digestmod=sha)
             m = hmac.new(getTokenKey(), digestmod=sha)

-            m.update(token['data'])
-            if (m.digest() == token['digest']):
+            m.update(pickled_data)
+            if (m.digest() == digest):

                 data = cPickle.loads(token['data'])
                 expires = data["expires"]
                 if (time.time() < expires):
                 data = cPickle.loads(token['data'])
                 expires = data["expires"]
                 if (time.time() < expires):

@@ -79,7 +78,7 @@ class VNCAuth(protocol.Protocol):
                     self.auth_data = data
                 else:
                     self.auth_error = "Token has expired; please try logging in again"
                     self.auth_data = data
                 else:
                     self.auth_error = "Token has expired; please try logging in again"

-        except (TypeError, cPickle.UnpicklingError):
+        except (TypeError, ValueError, cPickle.UnpicklingError):

             self.auth = None            
             print sys.exc_info()
 
             self.auth = None            
             print sys.exc_info()