Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

websockets: replace masker with more optimized one

Browse Source
This commit is contained in:
Mathieu Borderé 2019-07-06 14:28:30 +02:00
parent b5404f82da
commit 102e64bda5
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mitmproxy/net/websockets/masker.py
View File

@ -1,3 +1,6 @@
import sys
class Masker:
"""
Data sent from the server must be masked to prevent malicious clients
@ -12,12 +15,13 @@ class Masker:
self.offset = 0
def mask(self, offset, data):
result = bytearray(data)
for i in range(len(data)):
result[i] ^= self.key[offset % 4]
offset += 1
result = bytes(result)
return result
datalen = len(data)
offset_mod = offset % 4
data = int.from_bytes(data, sys.byteorder)
num_keys = (datalen + offset_mod + 3) // 4
mask = int.from_bytes((self.key * num_keys)[offset_mod:datalen +
offset_mod], sys.byteorder)
return (data ^ mask).to_bytes(datalen, sys.byteorder)
def __call__(self, data):
ret = self.mask(self.offset, data)