14 lines
277 B
Python
14 lines
277 B
Python
import itertools
|
|
|
|
|
|
__all__ = ['apply_mask']
|
|
|
|
|
|
def apply_mask(data, mask):
|
|
"""
|
|
Apply masking to websocket message.
|
|
|
|
"""
|
|
if len(mask) != 4:
|
|
raise ValueError("mask must contain 4 bytes")
|
|
return bytes(b ^ m for b, m in zip(data, itertools.cycle(mask)))
|