mh1pUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZ ddlmZn#e$r ddlmZYnwxYwgdZGdd ejZejZejZejZejZejZej Z!eeefZ"eee!fZ#Gd d ejZ$e$j%d e$j&d e$j'de$j(de$j)de$j*de$j+de$j,de$j-de$j.de$j/de$j0de$j1de$j2de$j3diZ4de5d<e$j%e$j&e$j'e$j(e$j+e$j,e$j-e$j.e$j/e$j0e$j1e$j2h Z6e$j%e$j&e$j)hZ7e8e9e:fZ;ej<GddZ=ej<Gdd Z>dd!l?m@Z@dS)") annotationsN)Callable GeneratorSequence) PayloadTooBig ProtocolError) apply_mask) OpcodeOP_CONTOP_TEXT OP_BINARYOP_CLOSEOP_PINGOP_PONG DATA_OPCODES CTRL_OPCODESFrameClosec*eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGz Frame.__str__..s&==dtMM======r&textc3K|]}|dV dSrOr%rQs r'rTz Frame.__str__..s&AA$4 AAAAAAr&binaryz''z...z, z [])lenr>r@r<r reprdecoder MAX_LOG_SIZEjoinrrGrparseUnicodeDecodeErrorAttributeErrorfiltername)selfcodinglength non_finalr>rWcutmetadatas r'__str__z Frame.__str__s  NNMMs49~~/B/BMM(3BB ;' ! ! ((**++DD [I % %YF6{{T.!333(A-1a76)AG)#4k63$%%="QRR88==f=====DD [H $ $u{{49--..DD Y  "DI,,..//&7 " " "v;;!2a!777,1A5!;C XXviCi'8+vsdee}&UVVFxxAA&AAAAA!  "D t99t( ( (#q(1,C !c' ?U*T3$%%[8D99VD669*EFFGG+"88T88X8888s(E00BHHN)max_size extensions read_exact-Callable[[int], Generator[None, None, bytes]]maskrj int | Nonerk%Sequence[extensions.Extension] | NoneGenerator[None, None, Frame]c#VK|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 d|d|d|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. 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@ zinvalid opcodezincorrect masking~!Hrz!Qzover size limit (z > z bytes))rj) structunpackr ValueErrorr rr reversedr[check)clsrlrnrjrkr>head1head2r@rArBrCr<excre mask_bytesframe extensions r'r^z Frame.parsesy<%*Q--''''''}UD11 uj(3ddez)4ttuz)4ttuz)4ttu ;EJ.//FF ; ; ; 011s : ;J& 1DDEd : : 344 4# S==(jmm++++++D dD11IVV s]](jmm++++++D dD11IV  FX$5$5 PF P Px P P PQQ Q  2$.JqMM111111J%*V,,,,,,,,  0dJ//DFD#tT488  J!*-- ? ?I$$UX$>>EE  sA$$ B.A>>Brkc<||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. Nrtrrurvrwrzrsiz!BBHz!BBQryr|)rencodeioBytesIOr@rArBrCr<rYr>writer}packsecrets token_bytesr getvalue) rcrnrkroutputrrrerr>s r' serializezFrame.serializes$  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<rrYr>r@rcs r'rz Frame.checkNs 9 ;  ;TY ; 9:: : ;, & &49~~###$<===8 @#$>??? ' & @ @r&rFrG) rlrmrnr?rjrorkrprFrq)rnr?rkrprFr=rFr)rrrr__annotations__r@rArBrCintosenvirongetr\ri classmethodr^rrr%r&r'rrzs' NNNKKKCDDD3rz~~&?FFGGL09090909d $<@ HHHHH[H\=A 9!9!9!9!9!9!v@@@@@@r&rcXeZdZUdZded<ded<ddZedd Zdd Zdd Z dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. rcoderGreasonrFcd|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 ()rM)rr:rr)rc explanationresults r'riz Close.__str__ns 49 # # # #t # # # # #&KK TY % % % % % % % % %'KK155diKKKI///// ; /....F r&r>r=cRt|dkr]tjd|dd\}|dd}|||}||St|dkr|t jdStd)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. rr{NrrIzclose frame too short)rYr}r~r[rr)r.r )rr>rrcloses r'r^z Close.parses t99>>mD$rr(33GT!""X__&&FCf%%E KKMMML YY!^^3y/44 4 788 8r&c|tjd|j|jzS)z: Serialize the payload of a close frame. r{)rr}rrrrrs 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)rFr=r) rrrrrrirr^rrr%r&r'rr`sIIIKKK$999[9.CCCC 7 7 7 7 7 7r&rr)A __future__r dataclassesenumrrrr}typingrrr exceptionsrr speedupsr ImportErrorutils__all__IntEnumr rr r r r!rr"rr#rr$rrrr)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r:rrOK_CLOSE_CODESr= bytearray memoryview BytesLike dataclassrrrIrkr%r&r'rsD"""""""  000000000044444444"$$$$$$$"""!!!!!!!!"   )))))T\))) + + M < + +* ')  .d , .  2 =  = 8  2 0 !#8 . 0 0 = ?+,       !     "   9j (  b@b@b@b@b@b@b@b@J H7H7H7H7H7H7H7H7Xs: AA