Rooba
/
mitmproxy
mirror of https://github.com/mitmproxy/mitmproxy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

websockets: more flexible masking interface.

This commit is contained in:
Aldo Cortesi 2015-05-01 10:31:20 +12:00
parent 7d9e38ffb1
commit 08b2e2a6a9
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
netlib/websockets.py
View File

@ -48,13 +48,18 @@ class Masker:
self.masks = [utils.bytes_to_int(byte) for byte in key] self.masks = [utils.bytes_to_int(byte) for byte in key]
self.offset = 0 self.offset = 0
def __call__(self, data): def mask(self, offset, data):
result = "" result = ""
for c in data: for c in data:
result += chr(ord(c) ^ self.masks[self.offset % 4]) result += chr(ord(c) ^ self.masks[offset % 4])
self.offset += 1 offset += 1
return result return result
def __call__(self, data):
ret = self.mask(self.offset, data)
self.offset += len(ret)
return ret
def client_handshake_headers(key=None, version=VERSION): def client_handshake_headers(key=None, version=VERSION):
""" """