U Mf5@sUddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZddlmZzddlmZWn ek rddlmZYnXdd d d d d dddddddg ZGdddejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%GdddejZ&e&j'de&j(de&j)de&j*de&j+de&j,de&j-de&j.de&j/d e&j0d!e&j1d"e&j2d#e&j3d$e&j4d%e&j5d&iZ6d'e7d(<e&j'e&j(e&j)e&j*e&j-e&j.e&j/e&j0e&j1e&j2e&j3e&j4h Z8e&j'e&j(e&j+hZ9e:e;eGd)ddZ?d*d+d,d-dZ@d*d.d,d/dZAej>Gd0ddZBdS)1) 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(5/tmp/pip-unpacked-wheel-yzabpfcc/websockets/frames.pyr %s c@sLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS) CloseCodez-Close code values for WebSocket close frames.iiiiiiiiiiiiiiiN)rrr r!NORMAL_CLOSURE GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATANO_STATUS_RCVDABNORMAL_CLOSURE INVALID_DATAPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATER BAD_GATEWAY TLS_HANDSHAKEr(r(r(r)r*7s r*OKz going awayzprotocol errorzunsupported datazno status received [internal]zabnormal closure [internal]zinvalid frame payload datazpolicy violationzmessage too bigzmandatory extensionzinternal errorzservice restartztry again laterz bad gatewayz TLS handshake failure [internal]zdict[int, str]CLOSE_CODE_EXPLANATIONSc@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 opcodebytesdataTboolfinFrsv1rsv2rsv3strreturnc 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|jrHzt|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)zE Return a human-readable representation of a frame. Nz byters continuedsi css|]}|dVqdSZ02xNr(.0byter(r(r) sz Frame.__str__..textcss|]}|dVqdSrNr(rOr(r(r)rRsbinaryz''K0z...iz, z [])lenr>r@r<rreprdecoderjoinrrDrparseUnicodeDecodeErrorAttributeErrorfiltername)selfZcodinglengthZ non_finalr>rTmetadatar(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_exactmaskrer rFc cs|dEdH}td|\}}|d@r*dnd}|d@r:dnd} |d@rJdnd} |d @rZdnd} zt|d @} Wn.tk r} ztd | W5d} ~ XYnX|d@rdnd|krtd |d @}|dkr|dEdH}td|\}n&|d kr|dEdH}td|\}|dk r<||kr z bytes))re) structunpackr ValueErrorr ProtocolErrorZ PayloadTooBigr reversedrZcheck)clsrfrgrer r>head1head2r@rArBrCr<excrb mask_bytesframe extensionr(r(r)r\sF     z Frame.parse)r )rgr rFc Cs(||dkrg}|D]}||}qt}|jr:dnd|jrFdndB|jrTdndB|jrbdndB|jB}|rvdnd}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. NrirrjrkrLrnrhiz!BBHz!BBQrmrp)rvencodeioBytesIOr@rArBrCr<rXr>writerqpacksecretsZ token_bytesr getvalue) rargr r}outputrxryrbr{r>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) rArBrCr rtr<rrXr>r@rar(r(r)rvLs   z Frame.check)rrr r!__annotations__r@rArBrCrd classmethodr\rrvr(r(r(r)r|s     /P;r zTuple[int, bytes]r>rFcCs6t|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. utf-8data must be str or bytes-likeN) isinstancerDrr~ BytesLiker TypeErrorr>r(r(r)r^s   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. rrN)rrDr~rr=rrr(r(r)rvs    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. intcoderDreasonrEcCszd|jkrdkr nnd}n.d|jkr6dkr@nnd}nt|jd}|jd|d}|jrv|d |j}|S) zU Return a human-readable representation of a close code and reason. iZ registeredz private useunknownz ()rM)rr;getr)raZ explanationresultr(r(r)rdsz Close.__str__r=rcCspt|dkrJtd|dd\}|ddd}|||}||St|dkrb|tjdStddS)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. rroNrrrGzclose frame too short) rXrqrrrZrvr*r/r rt)rwr>rrcloser(r(r)r\s    z Close.parsecCs"|td|j|jdS)z: Serialize the payload of a close frame. ror)rvrqrrrr~rr(r(r)rszClose.serializercCs0|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. rrzinvalid status codeN)rEXTERNAL_CLOSE_CODESr rtrr(r(r)rvs"z Close.checkN) rrr r!rrdrr\rrvr(r(r(r)rs  )C __future__rZ dataclassesenumrrrqtypingrrrrrrGr r r Zspeedupsr ImportErrorutils__all__IntEnumr r"rr#rr$rr%rr&rr'rrrr*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r;rrZOK_CLOSE_CODESr= bytearray memoryviewrZ dataclassrrrrr(r(r(r)s      b