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)))