bjAUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZmZddl mZ ddlmZn#e$r ddlmZYnwxYwgd ZGd d ejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%Gd d ejZ&e&j'de&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+hZ9ej:Gdd Z;ej:Gd!d"Z< d,d-d*Z=dd+l>m?Z?dS).) annotationsN) GeneratorSequence)Callable) PayloadTooBig ProtocolError) BytesLike) apply_mask) OpcodeOP_CONTOP_TEXT OP_BINARYOP_CLOSEOP_PINGOP_PONG DATA_OPCODES CTRL_OPCODES CloseCodeFrameClosec*eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGE/home/drrone/.local/lib/python3.11/site-packages/websockets/frames.pyr r &s,--)D$(E4r(r cNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS)rz-Close code values for WebSocket close frames.iiiiiiiiiiiiiiiN)rrrr 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)rr8sc77NJNNLONOOKMMMr(rOKz 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_EXPLANATIONSceZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <e e j d dZ eded ediZd$dZd%dZedddd&dZddd'd!Zd(d#ZdS))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 opcoder dataTboolfinFrsv1rsv2rsv3WEBSOCKETS_MAX_LOG_SIZE75returnstrc||j|j}|\}}||krdn|rdnd}t |jdt |jdkrdnd}|jrdnd}dtd |||g}|jj d |d |d S) ax Return a human-readable representation of a frame. This function is intended for logging and debugging. It doesn't aim to support round-tripping because payloads can be too long for displaying conveniently. Instead, it shows the beginning and the end. It's robust to incorrect data. It attempts to decode UTF-8 payloads whenever possible, even for binary frames and control frames, because those frequently contain UTF-8 data. It applies the same logic to continuation frames, because we don't know if they continue a text frame or a binary frame. textbinaryz byters continuedz, N z []) DEFAULT_IS_TEXTgetr= _data_reprlenr>r@joinfiltername)self expect_text data_repris_text data_typelength non_finalmetadatas r)__str__z Frame.__str__s*..t{;; !__.. 7%00BB7Vvvh  NNMMs49~~/B/BMM(3BB 99VD9fi*HIIJJ+"==Y==(====r(tuple[str, bool | None]c|js!d|j|jfS|jturF t t |jdfS#ttf$rYnwxYwt|jd|j zkrt|jdd|j zdz}t|jd|j zdzd}t||jtkot||j }|r&t!||zd }n^t|j}t||jtk|j }|r#t!|d }|s}|j}t||j dzkr<|j dzd z dz}d|dd|zd|| dg}dd|D}t||j kr)|j dzd z }|dd|zdz|| dz}||fS)z Return a human-readable representation of the payload. Also returns whether the payload is text. The representation is elided to fit ``MAX_LOG_SIZE``. This is a helper for the __str__ method. z''TNr)must_start_clean)must_end_cleanreplace)errors)rerfrr(rsrNc3K|]}|dV dS)02xNr').0bytes r) z#Frame._data_repr..s& B B4D B B B B B Br(z...)r>rPrQr=rrGrparser UnicodeDecodeErrorrS MAX_LOG_SIZEbytesis_utf8_fragmentr r@reprdecoderT)rW data_startdata_endrZrYr>rKcuts r)rRzFrame._data_reprsy ?-11$+>>> > ;( " " 5;;ty1122D88!#56     ty>>A 11 1 1ty)E1t/@+@A+E)EFGGJTYrD,='='B'D'DEFFH&!%!7##x  S *x"7!?!?y!?!Q!QRR ##D&!%!7#xG  @ I!>!>??  CYF6{{T.!333(A-1a76)AG)#4k63$%%="QRR B B6 B B BBBI y>>D- - -#q(1,C!)AG),u4y#7GGI'!!s-A&&A:9A:N)max_size extensions read_exact9Callable[[int], Generator[None, None, bytes | bytearray]]maskrx int | Nonery%Sequence[extensions.Extension] | NoneGenerator[None, None, Frame]c#JK|dEd{V}tjd|\}}|dzrdnd}|dzrdnd} |dzrdnd} |d zrdnd} t|d z} n"#t$r} t d | d} ~ wwxYw|dzrdnd|krt d |d z}|dkr)|dEd{V}tjd|\}n.|d kr(|dEd{V}tjd|\}|||krt |||r|dEd{V}||Ed{V}|rt ||}|| ||| | | }|g}t|D]}|||}| |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. PayloadTooBig: If the frame's payload size exceeds ``max_size``. ProtocolError: If the frame contains incorrect values. rN!BBTF@ zinvalid opcodezincorrect masking~!Hrz!Qrb)rx) structunpackr ValueErrorr rr reversedrtcheck)clsrzr|rxryr>head1head2r@rArBrCr=excr\ mask_bytesframe extensions r)rnz Frame.parsesg:%*Q--''''''}UD11 uj(3ddez)4ttuz)4ttuz)4ttu ;EJ.//FF ; ; ; 011s : ;J& 1DDEd : : 344 4# S==(jmm++++++D dD11IVV s]](jmm++++++D dD11IV  FX$5$511 1  2$.JqMM111111J%*V,,,,,,,,  0dJ//DFD#tT488  J!*-- ? ?I$$UX$>>EE  sA$$ B.A>>Bryrqc<||g}|D]}||}tj}|jrdnd|jrdndz|jrdndz|jrdndz|jz}|rdnd}t|j }|dkr-| tj d|||zna|d kr.| tj d ||dz|n-| tj d ||d z||r)tjd }| ||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. Nrrrrrrriz!BBHz!BBQrrb)rencodeioBytesIOr@rArBrCr=rSr>writerpacksecrets token_bytesr getvalue) rWr|ryroutputrrr\rr>s r) serializezFrame.serializeCs$  J# * *I##D))DD 8 *ZZ!Y-zzA /!Y-zzA /"Y-zzA /k   #) TY C<< LLUE56>BB C C C C e^^ LLVUECKHH I I I I LLVUECKHH I I I  % ,Q//J LL $ $ $  di44DD9D T   r(Nonec|js|js|jrtd|jt vr=t |jdkrtd|jstddSdS)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=rrSr>r@rWs r)rz Frame.checks 9 ;  ;TY ; 9:: : ;, & &49~~###$<===8 @#$>??? ' & @ @r(rFrG)rFr`) rzr{r|r?rxr}ryr~rFr)r|r?ryr~rFrqrFr)rrrr __annotations__r@rArBrCintosenvironrQrprrrrPr_rR classmethodrnrrr'r(r)rrzsQ NNNOOOCDDD3rz~~&?FFGGLi$GO>>>>2F"F"F"F"P $<@ GGGGG[GZ=A :!:!:!:!:!:!x@@@@@@r(rcXeZdZUdZded<ded<ddZedd Zdd ZddZ dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. zCloseCode | intcoderGreasonrFcd|jcxkrdkrnnd}n8d|jcxkrdkrnnd}n t|jd}|jd|d}|jr |d |j}|S) zU Return a human-readable representation of a close code and reason. i registeredz private useunknownz ()rN)rr;rQr)rW explanationresults r)r_z Close.__str__s 49 # # # #t # # # # #&KK TY % % % % % % % % %'KK155diKKKI///// ; /....F r(r>r ct|trtdt|dkr]t jd|dd\}|dd}|||}||St|dkr|tj dStd)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. z.only compressed outgoing frames use memoryviewrrNrrIzclose frame too short) isinstance memoryviewAssertionErrorrSrrrtrrr/r )rr>rrcloses r)rnz Close.parses dJ ' ' S !QRR R t99>>mD$rr(33GT!""X__&&FCf%%E KKMMML YY!^^3y/44 4 788 8r(rqc|tjd|j|jzS)z: Serialize the payload of a close frame. r)rrrrrrrs r)rzClose.serializes8 {4++dk.@.@.B.BBBr(rcl|jtvs#d|jcxkrdksntddSdS)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 rs r)rz Close.checksO 111TTY5M5M5M5M5M5M5M5M 566 6215M5Mr(Nr)r>r rFr)rFrqr) rrrr rr_rrnrrr'r(r)rrsKKK$999[92CCCC 7 7 7 7 7 7r(rFr>rqrer?rfrFcTdt|}}|s;tdt|}||kr||}|dzdkr|dz } |s|dz}tt|dz |}||kro||}|dzdkr|dz}|dzdkrd}n&|dzdkrd}n|d zdkrd}n|d zd krd}nd}|t||z krt|} |||} d t | vS#t $rYd SwxYw) z*Guess if data is a fragment of UTF-8 text.rrcrrrbrz\xF)rSminmaxrtrsro) r>rerfstartend max_startrlmin_endseq_lenrJs r)rrrrsCII3E  3t99%% i;Dj J..    qc$ii!mU++Wnn9Dj J..qj J.. "j00 "j00 "j00#d))c/))$ii 'E#I%%'' DJJ&& uus,D D'&D'r)FF)r>rqrer?rfr?rFr?)@ __future__r dataclassesenumrrrrcollections.abcrrtypingr exceptionsrr r speedupsr ImportErrorutils__all__IntEnumr r!r r"rr#rr$rr%rr&rrrrr+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r;rrOK_CLOSE_CODES dataclassrrrrrIryr'r(r)rs{"""""""  ////////44444444"$$$$$$$"""!!!!!!!!"    )))))T\))) + + M < + +* ')  .d , .  2 =  = 8  2 0 !#8 . 0 0 = ?+,       !     "   S@S@S@S@S@S@S@S@l J7J7J7J7J7J7J7J7^# E'E'E'E'E'RsA AA