B ]b]0@sddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZddlmZyddlmZWn ek rddlmZYnXdd d d d d dddddddg ZGdddejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%ddddddddddd d!d"d#d$d%Z&d&d'd(d)d*d+d,d-d.d/d0d1h Z'd&d'hZ(e)e*e+fZ,ej-Gd2ddZ.d3d4d5d6dZ/d3d7d5d8dZ0ej-Gd9ddZ1dS):) annotationsN)Callable GeneratorOptionalSequenceTuple) exceptions extensions)Data) apply_maskOpcodeOP_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(0lib/python3.7/site-packages/websockets/frames.pyr %s ZOKz going awayzprotocol errorzunsupported typezno status code [internal]z'connection closed abnormally [internal]z invalid datazpolicy violationzmessage too bigzextension requiredzunexpected errorzservice restartztry again laterz bad gatewayzTLS failure [internal])iiiiiiiiiiiiiiiiiiiiiiiiiiic@seZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <d dddZ e dddddddddddZ ddddddddZ dddd 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 opcodebytesdataTboolfinFrsv1rsv2rsv3str)returnc Csd}t|jdt|jdkr"dnd}|jr4dnd}|jtkrRt|j}n|jtkr|j}t|dkrd|dd d |d dg}d d d|D}n|jt krt t |j}n|jrHyt|j}d}Wnft tfk rD|j}t|dkr(d|dd d |d dg}d dd|D}d}YnXnd}t|dkrv|ddd|dd}dtd|||g}|jjd |d|dS)zC Return a human-readable represention of a frame. Nz byters continuedsi css|]}|dVqdS)02xNr().0byter(r(r) sz Frame.__str__..textcss|]}|dVqdS)r;Nr()r<r=r(r(r)r>sbinaryz''K0z...iz, z [])lenr,r.r*rreprdecoderjoinrr2rparseUnicodeDecodeErrorAttributeErrorfiltername)selfZcodinglengthZ non_finalr,r@Zmetadatar(r(r)__str__|s6&       z Frame.__str__N)max_sizer z-Callable[[int], Generator[None, None, bytes]]z Optional[int]z(Optional[Sequence[extensions.Extension]]zGenerator[None, None, Frame]) read_exactmaskrPr r3c cs|dEdH}td|\}}|d@r*dnd}|d@r:dnd} |d@rJdnd} |d @rZdnd} yt|d @} Wn.tk r} ztd | Wdd} ~ XYnX|d@rdnd|krtd |d @}|dkr|dEdH}td|\}n&|d kr|dEdH}td|\}|dk r<||kr z bytes))rP) structunpackr ValueErrorr ProtocolErrorZ PayloadTooBigr reversedrFcheck)clsrQrRrPr r,head1head2r.r/r0r1r*excrN mask_bytesframe extensionr(r(r)rHsD    z Frame.parse)r )rRr r3c Cs.||dkrg}x|D]}||}qWt}|jr>dnd|jrJdndB|jrXdndB|jrfdndB|jB}|rzdnd}t |j }|dkr| t d|||Bn>|d kr| t d ||dB|n| t d ||d B||rtd }| ||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. NrSrrTrUr9rXz!BBiz!BBHz!BBQrWrY)r_encodeioBytesIOr.r/r0r1r*rDr,writerZpacksecretsZ token_bytesr getvalue) rMrRr rfoutputrarbrNrdr,r(r(r) serializes, >     zFrame.serializeNonecCsR|js|js|jrtd|jtkrNt|jdkr>td|j sNtddS)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/r0r1r r]r*rrDr,r.)rMr(r(r)r_1s   z Frame.check)rrr r!__annotations__r.r/r0r1rO classmethodrHror_r(r(r(r)rcs     /H7r zTuple[int, bytes])r,r3cCs6t|trt|dfSt|tr*t|fStddS)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. zutf-8zdata must be str or bytes-likeN) isinstancer2rrg BytesLiker TypeError)r,r(r(r)rCs   r+cCs2t|tr|dSt|tr&t|StddS)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. zutf-8zdata must be str or bytes-likeN)rtr2rgrur+rv)r,r(r(r)r[s    c@s`eZdZUdZded<ded<ddddZed dd d d Zd dd dZddddZ dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. intcoder2reason)r3cCszd|jkrdkr nnd}n.d|jkr6dkr@nnd}nt|jd}|jd|d}|jrv|d |j}|S) zS Return a human-readable represention of a close code and reason. i iZ registerediz private useunknownz ()r:)rx CLOSE_CODESgetry)rMZ explanationresultr(r(r)rOsz Close.__str__r+)r,r3cCsnt|dkrJtd|dd\}|ddd}|||}||St|dkr`|ddStddS) 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. rz!HNzutf-8rir4zclose frame too short)rDrZr[rFr_r r])r`r,rxrycloser(r(r)rHs    z Close.parsecCs"|td|j|jdS)z: Serialize the payload of a close frame. z!Hzutf-8)r_rZrkrxryrg)rMr(r(r)roszClose.serializerpcCs0|jtks,d|jkr dks,ntddS)z Check that the close code has a valid value for a close frame. Raises: ProtocolError: if the close code is invalid. i izinvalid status codeN)rxEXTERNAL_CLOSE_CODESr r])rMr(r(r)r_s"z Close.checkN) rrr r!rrrOrsrHror_r(r(r(r)rrs )2Z __future__rZ dataclassesenumrhrlrZtypingrrrrrr4r r r Zspeedupsr ImportErrorZutils__all__IntEnumr r"rr#rr$rr%rr&rr'rrrr|rZOK_CLOSE_CODESr+ bytearray memoryviewruZ dataclassrrrrr(r(r(r)s     a