o /h5@sUddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZddlmZzddlmZWneyNddlmZYnwgdZGd d d ejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%Gd d d ejZ&e&j'd e&j(de&j)de&j*de&j+de&j,de&j-de&j.de&j/de&j0de&j1de&j2de&j3de&j4de&j5diZ6de7d<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;eGdddZ?d+d$d%Z@d,d'd(ZAej>Gd)d*d*ZBdS)-) 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(J/var/www/html/govbot/env/lib/python3.10/site-packages/websockets/frames.pyr %s r 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#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)zE Return a human-readable representation of a frame. Nz byters continuedsi cs|]}|dVqdS02xNr(.0byter(r(r) z Frame.__str__..textcsrMrNr(rPr(r(r)rSrTbinaryz''K0z...iz, z [])lenr>r@r<rreprdecoderjoinrrErparseUnicodeDecodeErrorAttributeErrorfiltername)selfcodinglength non_finalr>rVmetadatar(r(r)__str__s8&        z Frame.__str__N)max_sizer read_exact-Callable[[int], Generator[None, None, bytes]]maskri 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@ rKzinvalid opcodezincorrect masking~!Hrz!Qzover size limit (z > z bytes))ri) structunpackr ValueErrorr ProtocolError PayloadTooBigr reversedr\check)clsrjrlrir r>head1head2r@rArBrCr<excre mask_bytesframe extensionr(r(r)r^sL    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. NrqrrrrsrKrvrpiz!BBHz!BBQrurx)rencodeioBytesIOr@rArBrCr<rZr>writerypacksecrets token_bytesr getvalue) rcrlr routputrrrerr>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) rArBrCr r|r<rrZr>r@rcr(r(r)rLs    z Frame.checkrDrE) rjrkrlr?rirmr rnrDro)rlr?r rnrDr=rDr)rrr r!__annotations__r@rArBrCrh classmethodr^rrr(r(r(r)r|s      /P;rr>r rDTuple[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) isinstancerErr BytesLiker TypeErrorr>r(r(r)r^s  rr=cCs.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)rrErrr=rrr(r(r)rvs   rc@sNeZdZUdZded<ded<dddZedd d Zdd dZdddZ dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. intcoderEreasonrDcCszd|jkr dkrnnd}nd|jkrdkr nnd}nt|jd}|jd|d}|jr;|d |j}|S) zU Return a human-readable representation of a close code and reason. i registeredz private useunknownz ()rL)rr;getr)rc explanationresultr(r(r)rhsz Close.__str__r>r=cCslt|dkr%td|dd\}|ddd}|||}||St|dkr1|tjdStd)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. rrwNrrrFzclose frame too short) rZryrzr\rr*r/r r|)rr>rrcloser(r(r)r^s     z Close.parsecCs"|td|j|jdS)z: Serialize the payload of a close frame. rwr)rryrrrrrr(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 r|rr(r(r)rs" z Close.checkNr)r>r=rDr)rDr=r) rrr r!rrhrr^rrr(r(r(r)rs     r)r>r rDr)r>r rDr=)C __future__r dataclassesenumrrrytypingrrrrrrFr r r speedupsr ImportErrorutils__all__IntEnumr r"rr#rr$rr%rr&rr'rrrr*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r;rrOK_CLOSE_CODESr= bytearray memoryviewr dataclassrrrrr(r(r(r)s       b