o _c0@slddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZddlmZzddlmZWneyMddlmZYnwgdZGd d d ejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%d d d dddddddddddddZ&hdZ'ddhZ(e)e*e+fZ,ej-GdddZ.d+d$d%Z/d,d'd(Z0ej-Gd)d*d*Z1dS)-) annotationsN)Callable GeneratorOptionalSequenceTuple) exceptions extensions)Data) apply_mask) OpcodeOP_CONTOP_TEXT OP_BINARYOP_CLOSEOP_PINGOP_PONG DATA_OPCODES CTRL_OPCODESFrame prepare_data prepare_ctrlClosec@s$eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGr(r( r+r,r-r.r0r1r2r3r4r5r6r7r+r,c@seZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <d#ddZ e dddd$ddZ ddd%ddZ d&d!d"Z dS)'raN WebSocket frame. Attributes: opcode: Opcode. data: Payload data. fin: FIN bit. rsv1: RSV1 bit. rsv2: RSV2 bit. rsv3: RSV3 bit. Only these fields are needed. The MASK bit, payload length and masking-key are handled on the fly when parsing and serializing frames. r opcodebytesdataTboolfinFrsv1rsv2rsv3returnstrc Csd}t|jdt|jdkrdnd}|jrdnd}|jtur)t|j}nx|jturR|j}t|dkrGd|dd d |d dg}d d d|D}nO|jt ur`t t |j}nA|jrz t|j}d}Wn2t tfy|j}t|dkrd|dd d |d dg}d dd|D}d}Ynwd}t|dkr|ddd|dd}dtd|||g}|jjd |d|dS)zC Return a human-readable represention of a frame. Nz byters continuedsi cs|]}|dVqdS02xNr(.0byter(r(r) z Frame.__str__..textcsrIrJr(rLr(r(r)rOrPbinaryz''K0z...iz, z [])lenr:r<r8rreprdecoderjoinrrArparseUnicodeDecodeErrorAttributeErrorfiltername)selfZcodinglengthZ non_finalr:rRmetadatar(r(r)__str__|s8&        z Frame.__str__N)max_sizer read_exact-Callable[[int], Generator[None, None, bytes]]maskrc Optional[int]r (Optional[Sequence[extensions.Extension]]Generator[None, None, Frame]c cs|dEdH}td|\}}|d@rdnd}|d@rdnd} |d@r&dnd} |d @r.dnd} zt|d @} WntyJ} ztd | d} ~ ww|d@rQdnd|krZtd |d @}|dkrq|dEdH}td|\}n|d kr|dEdH}td|\}|dur||krtd|d|d|r|dEdH}||EdH}|rt||}|| ||| | | }|durg}t|D] }|j ||d}q| |S)a& Parse a WebSocket frame. This is a generator-based coroutine. Args: read_exact: generator-based coroutine that reads the requested bytes or raises an exception if there isn't enough data. mask: whether the frame should be masked i.e. whether the read happens on the server side. max_size: maximum payload size in bytes. extensions: list of extensions, applied in reverse order. Raises: EOFError: if the connection is closed without a full WebSocket frame. UnicodeDecodeError: if the frame contains invalid UTF-8. PayloadTooBig: if the frame's payload size exceeds ``max_size``. ProtocolError: if the frame contains incorrect values. rN!BBTF@ rGzinvalid opcodezincorrect masking~!Hrz!Qzover size limit (z > z bytes))rc) structunpackr ValueErrorr ProtocolErrorZ PayloadTooBigr reversedrXcheck)clsrdrfrcr r:head1head2r<r=r>r?r8excr` mask_bytesframe extensionr(r(r)rZsL    z Frame.parse)r c Cs&||dur g}|D]}||}q t}|jrdnd|jr#dndB|jr*dndB|jr1dndB|jB}|r;dnd}t |j }|dkrS| t d|||Bn|d kre| t d ||dB|n | t d ||d B||r~td }| ||rt|j |} n|j } | | |S)aH Serialize a WebSocket frame. Args: mask: whether the frame should be masked i.e. whether the write happens on the client side. extensions: list of extensions, applied in order. Raises: ProtocolError: if the frame contains incorrect values. NrkrrlrmrGrprjiz!BBHz!BBQrorr)rxencodeioBytesIOr<r=r>r?r8rVr:writerspacksecretsZ token_bytesr getvalue) r_rfr routputrzr{r`r}r:r(r(r) serializes>          zFrame.serializeNonecCsV|js |js |jrtd|jtvr't|jdkrtd|j s)tddSdS)z Check that reserved bits and opcode have acceptable values. Raises: ProtocolError: if a reserved bit or the opcode is invalid. zreserved bits must be 0}zcontrol frame too longzfragmented control frameN) r=r>r?r rvr8rrVr:r<r_r(r(r)rx3s    z Frame.checkr@rA) rdrerfr;rcrgr rhr@ri)rfr;r rhr@r9r@r)rrr r!__annotations__r<r=r>r?rb classmethodrZrrxr(r(r(r)rcs      /P;rr:r r@Tuple[int, bytes]cCs2t|tr t|dfSt|trt|fStd)a Convert a string or byte-like object to an opcode and a bytes-like object. This function is designed for data frames. If ``data`` is a :class:`str`, return ``OP_TEXT`` and a :class:`bytes` object encoding ``data`` in UTF-8. If ``data`` is a bytes-like object, return ``OP_BINARY`` and a bytes-like object. Raises: TypeError: if ``data`` doesn't have a supported type. utf-8data must be str or bytes-like) isinstancerArr BytesLiker TypeErrorr:r(r(r)rEs  rr9cCs.t|tr |dSt|trt|Std)ai Convert a string or byte-like object to bytes. This function is designed for ping and pong frames. If ``data`` is a :class:`str`, return a :class:`bytes` object encoding ``data`` in UTF-8. If ``data`` is a bytes-like object, return a :class:`bytes` object. Raises: TypeError: if ``data`` doesn't have a supported type. rr)rrArrr9rrr(r(r)r]s   rc@sNeZdZUdZded<ded<dddZedd d Zdd dZdddZ dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. intcoderAreasonr@cCszd|jkr dkrnnd}nd|jkrdkr nnd}nt|jd}|jd|d}|jr;|d |j}|S) zS Return a human-readable represention of a close code and reason. i registeredz private useunknownz ()rH)r CLOSE_CODESgetr)r_Z explanationresultr(r(r)rbsz Close.__str__r:r9cCsjt|dkr%td|dd\}|ddd}|||}||St|dkr0|ddStd) z Parse the payload of a close frame. Args: data: payload of the close frame. Raises: ProtocolError: if data is ill-formed. UnicodeDecodeError: if the reason isn't valid UTF-8. rrqNrrr/rBzclose frame too short)rVrsrtrXrxr rv)ryr:rrcloser(r(r)rZs     z Close.parsecCs"|td|j|jdS)z: Serialize the payload of a close frame. rqr)rxrsrrrrrr(r(r)rszClose.serializercCs4|jtvsd|jkrdksntddSdS)z Check that the close code has a valid value for a close frame. Raises: ProtocolError: if the close code is invalid. rrzinvalid status codeN)rEXTERNAL_CLOSE_CODESr rvrr(r(r)rxs" z Close.checkNr)r:r9r@r)r@r9r) rrr r!rrbrrZrrxr(r(r(r)rts     r)r:r r@r)r:r r@r9)2 __future__r dataclassesenumrrrstypingrrrrrrBr r r Zspeedupsr ImportErrorutils__all__IntEnumr r"rr#rr$rr%rr&rr'rrrrrZOK_CLOSE_CODESr9 bytearray memoryviewr dataclassrrrrr(r(r(r)sd       b