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

websockets: (very) slightly nicer is_valid constraints

Browse Source
This commit is contained in:
Aldo Cortesi 2015-04-21 23:49:27 +12:00
parent 176e29fc09
commit 4fb49c8e55
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
netlib/websockets.py
View File

@ -129,14 +129,12 @@ class Frame(object):
1 <= self.opcode <= 4,
0 <= self.mask_bit <= 1,
1 <= self.payload_length_code <= 127,
self.actual_payload_length == len(self.payload)
self.actual_payload_length == len(self.payload),
1 <= len(self.masking_key) <= 4 if self.mask_bit else True,
self.masking_key is not None if self.mask_bit else True
]
if not all(constraints):
return False
elif self.mask_bit == 1 and not 1 <= len(self.masking_key) <= 4:
return False
elif self.mask_bit == 0 and self.masking_key is not None:
return False
elif self.payload and self.masking_key:
decoded = apply_mask(self.payload, self.masking_key)
if decoded != self.decoded_payload: