Update Readme.md with edits to the code

This commit is contained in:
Shellrent Srl 2017-11-14 09:38:24 +01:00 committed by GitHub
parent 6b1d42e643
commit 0f66017f6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 51 additions and 0 deletions

View File

@ -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