Path: blob/master/ invest-robot-contest_TinkoffBotTwitch-main/venv/lib/python3.8/site-packages/aiohttp/_websocket.pyx
7762 views
from cpython cimport PyBytes_AsString123#from cpython cimport PyByteArray_AsString # cython still not exports that4cdef extern from "Python.h":5char* PyByteArray_AsString(bytearray ba) except NULL67from libc.stdint cimport uint32_t, uint64_t, uintmax_t8910def _websocket_mask_cython(object mask, object data):11"""Note, this function mutates its `data` argument12"""13cdef:14Py_ssize_t data_len, i15# bit operations on signed integers are implementation-specific16unsigned char * in_buf17const unsigned char * mask_buf18uint32_t uint32_msk19uint64_t uint64_msk2021assert len(mask) == 42223if not isinstance(mask, bytes):24mask = bytes(mask)2526if isinstance(data, bytearray):27data = <bytearray>data28else:29data = bytearray(data)3031data_len = len(data)32in_buf = <unsigned char*>PyByteArray_AsString(data)33mask_buf = <const unsigned char*>PyBytes_AsString(mask)34uint32_msk = (<uint32_t*>mask_buf)[0]3536# TODO: align in_data ptr to achieve even faster speeds37# does it need in python ?! malloc() always aligns to sizeof(long) bytes3839if sizeof(size_t) >= 8:40uint64_msk = uint32_msk41uint64_msk = (uint64_msk << 32) | uint32_msk4243while data_len >= 8:44(<uint64_t*>in_buf)[0] ^= uint64_msk45in_buf += 846data_len -= 8474849while data_len >= 4:50(<uint32_t*>in_buf)[0] ^= uint32_msk51in_buf += 452data_len -= 45354for i in range(0, data_len):55in_buf[i] ^= mask_buf[i]565758