jNDDUddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z mZddlmZmZddl mZ ddlmZgd Zej0d j2ZGd d ej6Zej:Zej<Zej>Zej@Z ejBZ!ejDZ"eeefZ#e e!e"fZ$Gd dej6Z%e%jLde%jNde%jPde%jRde%jTde%jVde%jXde%jZde%j\de%j^de%j`de%jbde%jdde%jfde%jhdiZ5de6d<e%jLe%jNe%jPe%jRe%jXe%jZe%j\e%j^e%j`e%jbe%jde%jfh Z7e%jLe%jNe%jThZ8ejrGd d!Z:ejrGd"d#Z; d& d'd$ZZ>y#e$r ddlmZYUwxYw)() annotationsN) GeneratorSequence)CallableSelf) PayloadTooBig ProtocolError) BytesLike) apply_mask)Opcode CloseCodeFrameCloses[\x09-\x0D\x20-\x7E]*c(eZdZdZd\ZZZd\ZZZ y)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONG^/home/oem/CyberBrain26-unified/dev-env/.venv/lib/python3.12/site-packages/websockets/frames.pyr r !s-)D$(E4r!r cLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZy)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"rr3sT7NJNNLONOOKMr!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 jjd dZ eded ediZddZddZeddd ddZdd ddZddZy)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_SIZE75c|jj|j}|j\}}||k(rdn|rdnd}t |j dt |j dk(rdnd}|j rdnd}djtd |||g}|jjd |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%0Bvh  N#5s499~/B(LM((B 99VD9fi*HIJ++""#1YKr(1==r!cd}d}|js'd|jj|jfS|jtur- t t j|j}d}n|jtvrlt|jr1tt|jjd}d}ndjd |jD}d}nt!|jd |j"zkDrt|jdd |j"zd z}t|jd |j"zd zd}t%||jt&k7xrt%||j(}|rzt||zjd}n[t|j}t%||jt&k7|j(}|rt|jd}|sr|j}t!||j"d zkDr3|j"d zdz d z}dj|dd|zd|| dg}djd|D}t!||j"kDr$|j"d zdz }|dd|zdz|| dz}||fS#ttf$r'djd|jD}d}YuwxYw)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. r@Nz''TrEc3$K|]}|d yw02xNr .0bytes r" z#Frame._data_repr..$ItSz]$IFasciic3$K|]}|d ywrYr r[s r"r^z#Frame._data_repr..r_r`r)must_start_clean)must_end_cleanreplace)errors)rfrgrr!rsc3$K|]}|d ywrYr r[s r"r^z#Frame._data_repr..s$FtSz]$Fr`z...)r7rGrHr6rstrrparser UnicodeDecodeErrorrK CTRL_OPCODESis_space_or_printable_asciireprbytesdecoderJ MAX_LOG_SIZEis_utf8_fragmentrr9)rNrPrQ data_startdata_endr7rBcuts r"rIzFrame._data_reprs #yy--11$++>> > ;;%   DII 67 [[L (*4995 tyy!1!8!8!AB HH$Ityy$II 499~D$5$5 55#499-Iq43D3D/D/I#JK  20A0A+AQ+F+H!IJ*%)[[D%8'#'88 $j8&;%C%C9%C%U VITYY'*%)[[D%8#'88  $T[[ [%B CI v;!2!2a!77,,1A5!;C XXviC'8+vsde}&UVFHH$Fv$FF  y>D-- -##q(1,C!)AG,u4y#7GGI'!!A"#56 HH$Ityy$II  s *K3LLN)max_size extensionsc#K|dEd{}tjd|\}}|dzrdnd}|dzrdnd} |dzrdnd} |d zrdnd} t|d z} |dzrdnd|k7r t d |d z}|dk(r)|dEd{}tjd|\}n-|d k(r(|dEd{}tjd|\}|||kDr t |||r|dEd{}||Ed{}|r t |}|| ||| | | }|g}t|D]}|j||}|j|S7Q#t$r} t d | d} ~ wwxYw7777~w)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!Qrc)rx) structunpackr ValueErrorr r r reversedrrcheck)cls read_exactmaskrxryr7head1head2r9r:r;r<r6excrS mask_bytesframe extensions r"rlz Frame.parses:%Q-'}}UD1 uj(dez)tuz)tuz)tu ;EJ./FJ&DEd : 34 4# S=(m+D dD1IV s](m+D dD1IV  FX$51 1 $.qM1J%V,, dJ/DFD#tT48  J!*- ?I$$UX$>E ?   W( ; 01s : ;,, 2-sw FEAFE",F E? -F9F:B C e^ LLVUECKH I LLVUECKH I  ,,Q/J LL $ dii4D99D T  r!c|js|js |jr td|jt vr;t |jdkDr td|js tdyy)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 r6rnrJr7r9rNs r"rz Frame.checksh 99 TYY 9: : ;;, &499~##$<==88#$>?? 'r!returnrk)rztuple[str, bool | None]) rz9Callable[[int], Generator[None, None, bytes | bytearray]]rr8rxz int | Nonery%Sequence[extensions.Extension] | NonerzGenerator[None, None, Frame])rr8ryrrrqrNone)rrrr__annotations__r9r:r;r<intosenvironrHrsrrrrGrVrI classmethodrlrrr r!r"rrus N OCD$D$D$rzz~~&?FGLT65%>O>.X"t $<@ GMG G  G : G &GGZ=A :!:!: :!  :!x@r!rcPeZdZUdZded<ded<d dZed dZd dZdd Z y )rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. zCloseCode | intcoderkreasoncd|jcxkrdkrnnd}n;d|jcxkrdkrnnd}n tj|jd}|jd|d}|jr|d |j}|S) zU Return a human-readable representation of a close code and reason. i registeredz private useunknownz ()rE)rr4rHr)rN explanationresults r"rVz Close.__str__s| 499 #t #&K TYY % %'K155diiKKII;b Q/ ;;xq .F r!cFt|tr tdt|dk\rIt j d|dd\}|ddj }|||}|j|St|dk(r|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. z.only compressed outgoing frames use memoryviewrrNrr@zclose frame too short) isinstance memoryviewAssertionErrorrJrrrrrrr(r )rr7rrcloses r"rlz Close.parses dJ ' !QR R t9>mmD$r(3GT!"X__&Ff%E KKML Y!^y//4 4 78 8r!c|jtjd|j|jj zS)z: Serialize the payload of a close frame. r)rrrrrrrs r"rzClose.serializes4 {{4+dkk.@.@.BBBr!c|jtvs-d|jcxkrdkstdtdyy)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.checksH 11TTYY5M5M 56 66N 56 66N1r!Nr)r7r rr)rrqr) rrrrrrVrrlrrr r!r"rrs8  K$992C 7r!rcdt|}}|s.tdt|}||kr||}|dzdk(r|dz } |s|dz}tt|dz |}||k\r^||}|dzdk(r|dz}|dzdk(rd}n#|dzdk(rd}n|d zdk(rd}n |d zd k(rd}nd}|t||z kr t|} |||j} d t | vS#t $rYy wxYw) z*Guess if data is a fragment of UTF-8 text.rrdr|rrcrz\xF)rJminmaxrrrprm) r7rfrgstartend max_startr]min_endseq_lenrAs r"rtrtsICI3E 3t9% i;Dj J.    qc$i!mU+Wn9Dj J.qj J. "j0 "j0 "j0#d)c/)$i 'E#%%' DJ&& sC!! C-,C-r)FF)r7rqrfr8rgr8rr8)? __future__r dataclassesenumrrrerrcollections.abcrrtypingrr exceptionsr r r speedupsr ImportErrorutils__all__compile fullmatchroIntEnumr rrrrrr DATA_OPCODESrnrr$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r4rrOK_CLOSE_CODES dataclassrrrtr@ryr r!r"rs"  /!4"$  )bjj)BCMM)T\\){{ {{   {{ {{T6! dD  .d , .  2 =  = 8  2 0 !!#8 . 0 0 = ?+,       !!     "   c@c@c@L  J7J7J7^# E' E'E'E' E'RA"!!"sJ JJ