RPf5UddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZmZddlmZ ddlmZgdZGd d ej,Zej0Zej4Zej8Zej<Zej@Z!ejDZ#eeefZ$ee!e#fZ%Gd d ej,Z&e&jNd e&jPde&jRde&jTde&jVde&jXde&jZde&j\de&j^de&j`de&jbde&jdde&jfde&jhde&jjdiZ6de7d<e&jNe&jPe&jRe&jTe&jZe&j\e&j^e&j`e&jbe&jde&jfe&jhh Z8e&jNe&jPe&jVhZ9e:e;ez Frame.__str__..s=dtCjM=textc3$K|]}|d ywrQr*rSs r,rVz Frame.__str__..sA$4* ArWbinaryz''K0z...iz, z [])lenrCrErArreprdecoderjoinrstrrparseUnicodeDecodeErrorAttributeErrorfiltername)selfcodinglength non_finalrCrZmetadatas r,__str__z Frame.__str__s  N#5s499~/B(LM((B ;;' ! ((*+D [[I %YYF6{R6#2; VBC["IJ88=f==D [[H $u{{499-.D YY  "DII,,./D t9r>9u$tCDz1D99VD669*EFG++""#1TF"XJa88'7 "v;# XXvcr{K&MNFxxA&AA!  "s$%F%%AHHN)max_sizer c#\K|dEd{}tjd|\}}|dzrdnd}|dzrdnd} |dzrdnd} |d zrdnd} t|d z} |dzrdnd|k7rt j d |d z}|dk(r)|dEd{}tjd|\}n-|d k(r(|dEd{}tjd|\}|!||kDrt j d|d|d|r|dEd{}||Ed{}|r t|}|| ||| | | }|g}t|D]}|j||}|j|S7k#t$r} t j d | d} ~ wwxYw7777w)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@ rNzinvalid opcodezincorrect masking~!Hrz!Qzover size limit (z > z bytes))rn) structunpackr ValueErrorr ProtocolError PayloadTooBigr reversedr`check)cls read_exactmaskrnr rChead1head2rErFrGrHrAexcrj mask_bytesframe extensions r,rcz Frame.parses<%Q-'}}UD1 uj(dez)tuz)tuz)tu FEJ./FJ&DEd :**+>? ?# S=(m+D dD1IV s](m+D dD1IV  FX$5**#F83xj@  $.qM1J%V,, dJ/DFD#tT48  J!*- ?I$$UX$>E ?   [( F**+;<# E F,,2-sx F,E9AF,E<6F,F#-F,F&A F,F(F, F*!AF,< F FF  F,&F,(F,*F,)r cF|j|g}|D]}|j|}tj}|jrdnd|j rdndz|j rdndz|jrdndz|jz}|rdnd}t|j}|dkr*|jtjd|||znZ|dkr+|jtjd ||dz|n*|jtjd ||d z||r&tjd }|j||rt!|j} n |j} |j| |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. rqrrrrsrNrvrpiz!BBHz!BBQrurx)rencodeioBytesIOrErFrGrHrAr^rCwriterypacksecrets token_bytesr getvalue) rhrr routputrrrjrrCs r, serializezFrame.serializesm$  J# *I##D)D * 88Z!YYzA /!YYzA /"YYzA /kk   # TYY C< LLUE56>B C e^ LLVUECKH I LLVUECKH I  ,,Q/J LL $ dii4D99D T  r+c8|js|js |jrtjd|j t vrOt|jdkDrtjd|jstjdyy)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) rFrGrHr r|rArr^rCrErhs r,rz Frame.checkLsz 99 TYY**+DE E ;;, &499~# ../GHH88 ../IJJ 'r+returnrb) rz-Callable[[int], Generator[None, None, bytes]]rrDrnz Optional[int]r (Optional[Sequence[extensions.Extension]]rzGenerator[None, None, Frame])rrDr rrrBrNone)r r!r"r#__annotations__rErFrGrHrm classmethodrcrrr*r+r,rr|s N KCD$D$D$-9^ #'?C JAJ J  J = J &JJ`@D 9!9!= 9!  9!vKr+rct|trt|jdfSt|trt |fSt d)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) isinstancerbrr BytesLiker TypeErrorrCs r,rr^sB $ G,,, D) $$899r+ct|tr|jdSt|tr t |St d)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)rrbrrrBrrs r,rrvs=${{7## D) $T{899r+cPeZdZUdZded<ded<d dZed dZd dZdd Z y )rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. intcoderbreasoncd|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 ()rO)rr?getr)rh explanationresults r,rmz Close.__str__s| 499 #t #&K TYY % %'K155diiKKII;b Q/ ;;xq .F r+c&t|dk\rJtjd|dd\}|ddjd}|||}|j |St|dk(r|t j dStjd)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. rrwNrrrJzclose frame too short) r^ryrzr`rr.r3r r|)rrCrrcloses r,rcz Close.parses t9>mmD$r(3GT!"X__W-Ff%E KKML Y!^y//4 4**+BC Cr+c|jtjd|j|jj dzS)z: Serialize the payload of a close frame. rwr)rryrrrrrs r,rzClose.serializes6 {{4+dkk.@.@.IIIr+c|jtvs-d|jcxkrdksntjdyy)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|rs r,rz Close.checks< 11TTYY5M5M**+@A A6N1r+Nr)rCrBrr)rrBr) r r!r"r#rrmrrcrrr*r+r,rrs: I K$DD.J Br+r)rCr rzTuple[int, bytes])rCr rrB)C __future__r dataclassesenumrrrytypingrrrrrrJr r r speedupsr ImportErrorutils__all__IntEnumrr$rr%rr&rr'rr(rr)rrrr.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r?rrOK_CLOSE_CODESrB bytearray memoryviewr dataclassrrrrr*r+r,rsy"  AA$"$  ")T\\) ++ ++ MM << ++ ++* ')  .d , .  2 =  = 8  2 0 !!#8 . 0 0 = ?+,       !!     "   9j (  ^K^K^KB:0:. HBHBHB} "!!"sI## I21I2