xjFUddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z mZddlmZmZddl mZ ddlmZn#e$r ddlmZYnwxYwgd Zejd jZGd d ejZejZejZejZej Z ej!Z!ej"Z"eeefZ#e e!e"fZ$Gd dejZ%e%j&de%j'de%j(de%j)de%j*de%j+de%j,de%j-de%j.de%j/de%j0de%j1de%j2de%j3de%j4diZ5de6d<e%j&e%j'e%j(e%j)e%j,e%j-e%j.e%j/e%j0e%j1e%j2e%j3h Z7e%j&e%j'e%j*hZ8ej9Gd d!Z:ej9Gd"d#Z; d-d.d+ZZ>dS)/) annotationsN) GeneratorSequence)CallableSelf) PayloadTooBig ProtocolError) BytesLike) apply_mask)Opcode CloseCodeFrameCloses[\x09-\x0D\x20-\x7E]*c*eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGBC:\Projects\Historical\venv\Lib\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)rrrrNORMAL_CLOSURE GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATANO_STATUS_RCVDABNORMAL_CLOSURE INVALID_DATAPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATER BAD_GATEWAY TLS_HANDSHAKEr r!r"rr3sc77NJNNLONOOKMMMr!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) a 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 in data frames, even in binary frames, because those frequently contain UTF-8 data. textbinaryz byters continuedz, N z []) DEFAULT_IS_TEXTgetr6 _data_reprlenr7r9joinfiltername)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]c4d}d}|js!d|j|jfS|jturo t t |j}d}ni#ttf$r*d d|jD}d}Yn/wxYw|jtvrtt|jr8tt|jd}d}nd d |jD}d}nt!|jd |jzkrt|jdd |jzd z}t|jd |jzd zd}t%||jt&kot%||j}|r&t||zd}n^t|j}t%||jt&k|j}|r#t|d}|s}|j}t!||jd zkr<|jd zdz d z}d |dd|zd|| dg}d d|D}t!||jkr)|jd zdz }|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. rBNz''TrGc3K|]}|dV dS02xNr .0bytes r" z#Frame._data_repr..&$I$It]]$I$I$I$I$I$Ir!Fasciic3K|]}|dV dSr\r r^s r"raz#Frame._data_repr..rbr!r)must_start_clean)must_end_cleanreplace)errors)rhrirr!rsc3K|]}|dV dSr\r r^s r"raz#Frame._data_repr..s&$F$Ft]]$F$F$F$F$F$Fr!z...)r7rIrJr6rr@rparser UnicodeDecodeErrorrM CTRL_OPCODESis_space_or_printable_asciireprbytesdecoderL MAX_LOG_SIZEis_utf8_fragmentrr9)rPrRrS data_startdata_endr7rDcuts r"rKzFrame._data_reprsW #y ?-11$+>>> > ;%    DI 6 677 !#56   HH$I$Ity$I$I$III  [L ( (*4955  ty!1!1!8!8!A!ABB HH$I$Ity$I$I$III 49~~D$5 555#49-Iq43D/D/I-I#JKK  20A+AQ+F+H+H!IJJ*%)[D%8'#'8W $j8&;%C%C9%C%U%U V VITY''*%)[D%8#'8 D $T[[ [%B%B C CI  Gv;;!2a!777,1A5!;C XXviCi'8+vsdee}&UVVFHH$F$Fv$F$F$FFF  y>>D- - -#q(1,C!)AG),u4y#7GGI'!!s.A,,7B'&B'N)max_size extensions read_exact9Callable[[int], Generator[None, None, bytes | bytearray]]maskry int | Nonerz%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!Qre)ry) structunpackr ValueErrorr r r reversedrscheck)clsr{r}ryrzr7head1head2r9r:r;r<r6excrU mask_bytesframe extensions r"rmz 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>>Brzrrc<||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!BBQrre)rencodeioBytesIOr9r:r;r<r6rLr7writerpacksecrets token_bytesr getvalue) rPr}rzroutputrrrUrr7s r" serializezFrame.serializeNs$  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) r:r;r<r r6rorLr7r9rPs r"rz Frame.checks 9 ;  ;TY ; 9:: : ;, & &49~~###$<===8 @#$>??? ' & @ @r!r?r@)r?rY) r{r|r}r8ryr~rzrr?r)r}r8rzrr?rrr?r)rrrr__annotations__r9r:r;r<intosenvironrJrtrrrrIrXrK classmethodrmrrr r!r"rrusQ NNNOOOCDDD3rz~~&?FFGGLT65%>O>>>>.X"X"X"X"t $<@ 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 | intcoder@reasonr?cd|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 ()rG)rr4rJr)rP explanationresults r"rXz Close.__str__s 49 # # # #t # # # # #&KK TY % % % % % % % % %'KK155diKKKI///// ; /....F r!r7r rct|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 memoryviewrrNrrBzclose frame too short) isinstance memoryviewAssertionErrorrLrrrsrrr(r )rr7rrcloses r"rmz Close.parses dJ ' ' S !QRR R t99>>mD$rr(33GT!""X__&&FCf%%E KKMMML YY!^^3y/44 4 788 8r!rrc|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)r7r r?r)r?rrr) rrrrrrXrrmrrr r!r"rrsKKK$999[92CCCC 7 7 7 7 7 7r!rFr7rrrhr8rir?cTdt|}}|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.rrfrrrerz\xF)rLminmaxrsrqrn) r7rhristartend max_startr`min_endseq_lenrCs r"rurusCII3E  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)r7rrrhr8rir8r?r8)? __future__r dataclassesenumrrrerrcollections.abcrrtypingrr exceptionsr r r speedupsr ImportErrorutils__all__compile fullmatchrpIntEnumr rrrrrr DATA_OPCODESrorr$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r4rrOK_CLOSE_CODES dataclassrrrurBrzr r!r"rs"""""""  ////////!!!!!!!!44444444"$$$$$$$"""!!!!!!!!"   )bj)BCCM)))))T\))){ {   { {T6! dD  .d , .  2 =  = 8  2 0 !#8 . 0 0 = ?+,       !     "   c@c@c@c@c@c@c@c@L  J7J7J7J7J7J7J7J7^# E'E'E'E'E'RsA AA