mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
376 B
Python
19 lines
376 B
Python
4 years ago
|
import itertools
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ["apply_mask"]
|
||
|
|
|
||
|
|
|
||
|
|
def apply_mask(data: bytes, mask: bytes) -> bytes:
|
||
|
|
"""
|
||
|
|
Apply masking to the data of a WebSocket message.
|
||
|
|
|
||
|
|
:param data: Data to mask
|
||
|
|
:param mask: 4-bytes mask
|
||
|
|
|
||
|
|
"""
|
||
|
|
if len(mask) != 4:
|
||
|
|
raise ValueError("mask must contain 4 bytes")
|
||
|
|
|
||
|
|
return bytes(b ^ m for b, m in zip(data, itertools.cycle(mask)))
|