fjiNUdZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z dd lmZd d l mZmZmZd d lmZmZmZmZmZmZd eDZeeeed<d ZdZdZ dZ!ej"Z#ej$Z%dZ&ej'j(Z)ej$j(Z*ej"j(Z+ej,j(Z-ej.j(Z/ej0j(Z1dZ2dZ3dZ4dZ5d Z6e7j8Z9eZ:GddZ;GddZrs'D'D'D1A'D'D'DALLOWED_CLOSE_CODES)Tr)FrceZdZdZdededejddfdZde fdZ de dzfd Z e fd e d ej ddfd Zdd ZddZ ddZdeezfdZdeezfdZdS)WebSocketDataQueuezmWebSocketDataQueue resumes and pauses an underlying stream. It is a destination for WebSocket data. protocollimitloopreturnNcd|_||_|dz|_||_d|_d|_d|_t|_|jj |_ |jj |_ dS)NrrF) _size _protocol_limit_loop_eof_waiter _exceptionr_bufferpopleft _get_bufferappend _put_buffer)selfr#r$r%s r__init__zWebSocketDataQueue.__init__=sd !ai   48 04JO'' </<.rc|jSN)r,r4s ris_eofzWebSocketDataQueue.is_eofKs yrc|jSr7)r.r8s r exceptionzWebSocketDataQueue.exceptionNs rexc exc_causechd|_||_|jx}d|_t|||dSdSNT)r,r.r-r )r4r<r=waiters rr z WebSocketDataQueue.set_exceptionQsF  l "F /DL &#y 1 1 1 1 1 0 /rc|jx}dSd|_|s|ddSdSr7)r-done set_result)r4r@s r_release_waiterz"WebSocketDataQueue._release_waiter\sOl "F + F {{}} $   d # # # # # $ $rcJd|_|d|_dSr?)r,rDr.r8s rfeed_eofzWebSocketDataQueue.feed_eofcs&  rdataWSMessage | WSMessageTextBytessize cython_intc|xj|z c_|||f||j|jkr'|jjs|jdSdSdSr7)r(r3rDr*r)_reading_paused pause_readingr4rGrIs r feed_datazWebSocketDataQueue.feed_datahs d  $&&&  : # #DN,J # N ( ( * * * * * $ # # #rcK|jsd|js]|jrJ|j|_ |jd{Vn'#t jt jf$r d|_wxYw|Sr7) r/r,r-r+ create_futureasyncioCancelledError TimeoutError_read_from_bufferr8s rreadzWebSocketDataQueue.readqs| DI | # # #:3355DL l""""""""*G,@A   #  %%'''s A$A+c|jr^|\}}|xj|zc_|j|jkr%|jjr|j|S|j|jtr7) r/r1r(r*r)rLresume_readingr.r rNs rrUz$WebSocketDataQueue._read_from_buffer|sx < ))++JD$ JJ$ JJzDK''DN,J'--///K ? &/ !rr&N)rGrHrIrJr&N)__name__ __module__ __qualname____doc__rrrRAbstractEventLoopr5boolr9 BaseExceptionr;rbuiltinsr rDrFrOrrrVrUrrrr"r"7sc /$ /-0 /;B;T /  / / / /=4/ -: 2 2  2) 2  2 2 2 2$$$$ +4+kr/t!|d krtA|dd d } | d kr&| tBvrt t jd |  |d dd} n-#t,$r } t t jd| d} ~ wwxYwt5t6tDj#| | f} nG|r#t t jd|d|d|t5t6tDj#d d f} |j| d dS|tHkrLt5t6tDj%|d f} |j| t!|dS|tLkrLt5t6tDj'|d f} |j| t!|dSt t jd|)Nz*Continuation frame for non started messagez8The opcode in non-fin frame is expected to be zero, got T)suppress_deflate_headerr z(Decompressed message exceeds size limit zutf-8zInvalid UTF-8 text messagerri zInvalid close code: zInvalid close frame:  Unexpected opcode=)( OP_CODE_TEXTOP_CODE_BINARYOP_CODE_CONTINUATIONrqrprrPROTOCOL_ERRORrmr_clearr}rdecompress_syncrrilenMESSAGE_TOO_BIGrrrjdecodeUnicodeDecodeError INVALID_TEXTrdrO TUPLE_NEWrWS_MSG_TYPE_TEXTrWS_MSG_TYPE_BINARY OP_CODE_CLOSEr rrCLOSE OP_CODE_PINGPING OP_CODE_PONGPONG) r4rrrr has_partialassembled_payloadpayload_mergedtextr< close_code close_messagemsgs r _handle_framezWebSocketReader._handle_framesH lN4HI I I---$,/2Q2Q$.@  111#)DL ( t}--K---.  $.2'-22 ,$(MG$;! ##%%%%$+! :*Y*:SW*X*X*XD' "&!4!D!D%(;; -0*Q..!/ ""%#n*=*=@R*R*R(#3W4CUWW'((E11!2!&'8!9!9%%$#-44W==-###,'46R"##J((!).>b-IJJN++ J((!.1A>SU0VN++  $$i* $$;N)N)N(#2;z;;$+ABBK$6$6w$?$?MM)(#02N IOZ+WXX E$.FCFF&FF7FF  IOQ+CDD J a ( ( ( ( ( | # #I 'DEEC J c'll 3 3 3 3 3 | # #I 'DEEC J c'll 3 3 3 3 3 *,K,K,K s0G H!G<<H<M N$M??Nc& |jr|j|zdc}|_d}t|}|} |jtkr||z dkrn||}||dz}|dz }|dz dz}|dz dz}|dz dz} |d z dz} |d z} | s | s |r!|jst t jd | tttttthvrt t jd | | dkr |dkrt t jd |dz dz} |dz} | dkr | dkrt t jd|js|jt kr|rt"nt$|_n|rt t jd t'||_| |_t'| |_| |_t.|_|jt.kr|j}|dkr1||z dkrn||}||dz}|dz }|dz|z|_n9|dkr,||z dkrnt3||d|_|dz }n||_|jrg|jttthvrL|jt|jz}||jkr%t t jd|d|j|jrt:nt<|_|jt:kr.||z d krn|||d z|_|d z }t<|_|jt<kr||z }|j|kr|}|xj|zc_n||jz}d|_|j }|xj ||z z c_ |}|}|jdkr$|j!"|||nD|r|j!"||||jrH|jJtGd$|j!}tK|j||}nd$|j!}|j!&na|jrP|jJ|||}tO|tFurtG|}tK|j||}n |||}|(|j|j||jd|_ t|_||kr |||nd|_dS)z&Return the next frame from the socket.rrTrr rz*Received frame with non-zero reserved bitsrz!Received fragmented control frame}z5Control frame payload cannot be larger than 125 bytes~z Message size z exceeds limit N))rvrrornr~rrrrrrrrrrrr|r{COMPRESSED_TRUECOMPRESSED_FALSEr_rsrwrzREAD_PAYLOAD_LENGTHryr rirmrREAD_PAYLOAD_MASK READ_PAYLOADrxrurtr2rljoinrrrr)r4rG start_posdata_len data_cstr first_byte second_byterrsv1rsv2rsv3rhas_masklengthlen_flagprojected_size chunk_len f_end_pos had_fragments f_start_pospayload_bytearrayrs rrzWebSocketReader._feed_dataFs  : 6#zD0# D$* t99 k *{k))i'!++&y1 ' A 6 Q !Q!+"a1,"a1,"a1,#c)4D(#2D ( "!  ")#27V77 C<'TDTD$$(#2D #'s))%+"!%h)/&1 {1111s??)+a//!*9!5J"+IM":KNI2</K2OD//^^)+a//2=dI2N2Nq2QD/NII2:D/ % $*< "(A++ &*%@3t}CUCU%UN%);;;,'7BNBB-1-?BB 48>S//| {///i'!++#,YQ-F#G Q * {l**$y0 .);; (I//9</// )D,G GI23D/ $ 7 ''9y+@@''' % .!33+229[=R3STTT!?+229[=R3STTT~D#/;;;,5chht?V6W6W,X,X)&t'79JKKK"3"%((4+B"C"C+113333^ ?+777(1+i2G(H%-..i??-66G,H,H)"4#35FGGG/GG' I(=>G""OT%7$BR+,') Wk *\7@(6J6JYy122PS r)TTrY)rZr[r\r"rr_r5rFrrl memoryviewtuplerOrJrrrrrrcrcs=  ""!"" "  "  """"< ei/*<tU{AS G Gj G" G *$ G  GGGGRwTuwTwTwTwTwTwTwTrrc)=r]rRra collectionsrtypingr base_protocolrcompression_utilsrhelpersrr streamsr r r rmodelsrrrrrrrsetr__annotations__rnrrrBINARYrTEXTrrp CONTINUATIONvaluerrrrrrrrrrrr{rrr__new__rrJr"rcrrrrs8666((((((00000022222222CCCCCCCCCC(E'D 'D'D'DU3s8_DDD  %> -3~# !'% ~# ~#   M  NNNNNNNNbuTuTuTuTuTuTuTuTuTuTr