Merge 4ac567670e into 6b1d42e643

README.md

@@ -1,3 +1,54 @@
+## Edits
+
+Added support for Tokens in Cookies in `websocketproxy.py` on `get_target`
+
+
+```python
+    def get_target(self, target_plugin, path):
+        """
+        Parses the path, extracts a token, and looks up a target
+        for that token using the token plugin. Sets
+        target_host and target_port if successful
+        """
+        # The files in targets contain the lines
+        # in the form of token: host:port
+		
+        token = ''
+		
+		# Extract the token parameter from cookie
+        if 'Cookie' in self.headers:
+            cookiestr = self.headers['Cookie']
+            cookiestr.replace( 'Cookie: ', '', 1 )
+            cookies = cookiestr.split('; ')
+
+            for cookie in cookies:
+                cookie = cookie.split('=')
+                if 'token' in cookie[0]:
+                    token = cookie[1]
+                    break
+
+        # Extract the token parameter from url
+        if token == '':
+            args = parse_qs(urlparse(path)[4]) # 4 is the query from url
+
+            if not 'token' in args or not len(args['token']):
+                raise self.server.EClose("Token not present")
+
+            token = args['token'][0].rstrip('\n')
+
+        if token == '':
+            raise self.server.EClose( "Empty Token defined" )
+
+        # Search for the token
+        result_pair = target_plugin.lookup( token )
+
+        if result_pair is not None:
+            return result_pair
+        else:
+            raise self.server.EClose( "Token '%s' not found" % token )
+```
+
+
 ## websockify: WebSockets support for any application/server
 
 websockify was formerly named wsproxy and was part of the

websockify/websocketproxy.py

@@ -140,7 +140,22 @@ Traffic Legend:
         # The files in targets contain the lines
         # in the form of token: host:port
 
+        token = ''
+
+		# Extract the token parameter from cookie
+        if 'Cookie' in self.headers:
+            cookiestr = self.headers['Cookie']
+            cookiestr.replace( 'Cookie: ', '', 1 )
+            cookies = cookiestr.split('; ')
+
+            for cookie in cookies:
+                cookie = cookie.split('=')
+                if 'token' in cookie[0]:
+                    token = cookie[1]
+                    break
+
         # Extract the token parameter from url
+        if token == '':
             args = parse_qs(urlparse(path)[4]) # 4 is the query from url
 
             if not 'token' in args or not len(args['token']):
@@ -148,12 +163,16 @@ Traffic Legend:
 
             token = args['token'][0].rstrip('\n')
 
-        result_pair = target_plugin.lookup(token)
+        if token == '':
+            raise self.server.EClose( "Empty Token defined" )
+
+        # Search for the token
+        result_pair = target_plugin.lookup( token )
 
         if result_pair is not None:
             return result_pair
         else:
-            raise self.server.EClose("Token '%s' not found" % token)
+            raise self.server.EClose( "Token '%s' not found" % token )
 
     def do_proxy(self, target):
         """