o Æeº2ã@slddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZddlmZzddlmZWneyMddlmZYnwgd¢ZGd d „d ejƒZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%d d d dddddddddddddœZ&hd£Z'hd£Z(e)e*e+fZ,ej-Gdd„dƒƒZ.d*d#d$„Z/d+d&d'„Z0ej-Gd(d)„d)ƒƒZ1dS),é)Ú annotationsN)ÚCallableÚ GeneratorÚOptionalÚSequenceÚTupleé)Ú exceptionsÚ extensions)ÚData)Ú apply_mask) ÚOpcodeÚOP_CONTÚOP_TEXTÚ OP_BINARYÚOP_CLOSEÚOP_PINGÚOP_PONGÚ DATA_OPCODESÚ CTRL_OPCODESÚFrameÚ prepare_dataÚ prepare_ctrlÚClosec@s$eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rré)éé é N) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚCONTÚTEXTÚBINARYÚCLOSEÚPINGÚPONG©r(r(úAD:\Projects\ConvertPro\env\Lib\site-packages\websockets/frames.pyr %s r ÚOKz 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])éèéééêéëéíiîéïéðéñéòéóéôéõéöi÷> r+r,r-r.r0r1r2r3r4r5r6r7>r+r,r/c@sŒeZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <d#dd„Z e dddœd$dd„ƒZ ddœd%dd„Z 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 ÚopcodeÚbytesÚdataTÚboolÚfinFÚrsv1Úrsv2Úrsv3ÚreturnÚstrc Csšd}t|jƒ›dt|jƒdkrdnd›}|jrdnd}|jtur)t|j ¡ƒ}nx|jturR|j}t|ƒdkrGd |dd …d |d d…g¡}d  d d„|Dƒ¡}nO|jt ur`t t   |j¡ƒ}nA|jrŸz t|j ¡ƒ}d}Wn2t tfyž|j}t|ƒdkrd |dd …d |d d…g¡}d  dd„|Dƒ¡}d}Ynwd}t|ƒdkrµ|dd…d|dd…}d td|||gƒ¡}|jj›d |›d|›dS)zE Return a human-readable representation of a frame. Nz byterÚÚsÚ continuedéóésiøÿÿÿú csó|]}|d›VqdS©Ú02xNr(©Ú.0Úbyter(r(r)Ú “ó€z Frame.__str__..ÚtextcsrIrJr(rLr(r(r)rO£rPÚbinaryz''éKé0z...ièÿÿÿz, z [ú])Úlenr:r<r8rÚreprÚdecoderÚjoinrrArÚparseÚUnicodeDecodeErrorÚAttributeErrorÚfilterÚname)ÚselfZcodingÚlengthZ non_finalr:rRÚmetadatar(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¢|dƒEdH}t d|¡\}}|d@rdnd}|d@rdnd} |d@r&dnd} |d @r.dnd} zt|d @ƒ} WntyJ} zt d ¡| ‚d} ~ ww|d@rQdnd|krZt d ¡‚|d @}|dkrq|dƒEdH}t d|¡\}n|d krƒ|dƒEdH}t d|¡\}|dur—||kr—t d|›d|›d¡‚|r |dƒEdH}||ƒEdH}|r®t||ƒ}|| ||| | | ƒ}|dur½g}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ú!BBé€TFé@é rGézinvalid opcodezincorrect maskingéé~ú!Hrz!Qzover size limit (z > z bytes)é)rc) ÚstructÚunpackr Ú ValueErrorr Ú ProtocolErrorZ PayloadTooBigr ÚreversedrXÚcheck)Úclsrdrfrcr r:Úhead1Úhead2r<r=r>r?r8Úexcr`Ú mask_bytesÚframeÚ extensionr(r(r)rZ¯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|||B¡¡n|d kre|  t   d ||dB|¡¡n |  t   d ||d B|¡¡|r~t d ¡}|  |¡|r‡t|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)rxÚencodeÚioÚBytesIOr<r=r>r?r8rVr:ÚwritersÚpackÚsecretsZ token_bytesr Úgetvalue) r_rfr rÚoutputrzr{r`r}r:r(r(r)Ú serializeüs>   ÿ þ ýüÿ     zFrame.serializeÚNonecCsV|js |js |jrt d¡‚|jtvr't|jƒdkrt d¡‚|j s)t d¡‚dSdS)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)rx7s    üz Frame.check©r@rA) rdrerfr;rcrgr rhr@ri)rfr;r rhr@r9©r@r‰)rrr r!Ú__annotations__r<r=r>r?rbÚ classmethodrZrˆrxr(r(r(r)rgs      /úPü;rr:r r@úTuple[int, bytes]cCs2t|tƒr t| d¡fSt|tƒrt|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-8údata must be str or bytes-like)Ú isinstancerArr€Ú BytesLikerÚ TypeError©r:r(r(r)rIs  rr9cCs.t|tƒr | d¡St|tƒrt|ƒ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. r‘r’)r“rAr€r”r9r•r–r(r(r)ras   rc@sNeZdZUdZded<ded<ddd„Zedd d „ƒZdd d„Zddd„Z dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. ÚintÚcoderAÚreasonr@cCszd|jkr dkrnnd}nd|jkrdkr nnd}nt |jd¡}|j›d|›d}|jr;|›d |j›}|S) zU Return a human-readable representation of a close code and reason. é¸ i Ú registeredéˆz private useÚunknownz (ú)rH)r˜Ú CLOSE_CODESÚgetr™)r_Z explanationÚresultr(r(r)rb†sz Close.__str__r:r9cCsjt|ƒdkr%t d|dd…¡\}|dd… d¡}|||ƒ}| ¡|St|ƒdkr0|ddƒSt d¡‚) 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. rrqNr‘rr/rBzclose frame too short)rVrsrtrXrxr rv)ryr:r˜r™Úcloser(r(r)rZ˜s     z Close.parsecCs"| ¡t d|j¡|j d¡S)z: Serialize the payload of a close frame. rqr‘)rxrsr„r˜r™r€r‹r(r(r)rˆ°szClose.serializer‰cCs4|jtvsd|jkrdksnt d¡‚dSdS)z› Check that the close code has a valid value for a close frame. Raises: ProtocolError: if the close code is invalid. ršrœzinvalid status codeN)r˜ÚEXTERNAL_CLOSE_CODESr rvr‹r(r(r)rx¸s" ÿz Close.checkNrŒ)r:r9r@r)r@r9r) rrr r!rŽrbrrZrˆrxr(r(r(r)rxs     r)r:r r@r)r:r r@r9)2Ú __future__rÚ dataclassesÚenumrr…rsÚtypingrrrrrrBr r r Zspeedupsr Ú ImportErrorÚutilsÚ__all__ÚIntEnumr r"rr#rr$rr%rr&rr'rrrrŸr£ZOK_CLOSE_CODESr9Ú bytearrayÚ memoryviewr”Ú dataclassrrrrr(r(r(r)Úsd   ÿ  ð  b