o `g7F@sFUdZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZmZdd lmZd d lmZmZmZd d lmZmZmZmZmZd deDZee e e!d<d Z"dZ#dZ$dZ%ej&Z'ej(Z)ej*j+Z,ej(j+Z-ej&j+Z.ej/j+Z0ej1j+Z2ej3j+Z4dZ5dZ6e7j8Z9e Z:GdddZ;GdddZsrALLOWED_CLOSE_CODES)T)Fr"c@seZdZdZdededejddfddZde fd d Z de e fd d Z efd ddej ddfddZd ddZd ddZd!ddZdefddZdefddZdS)"WebSocketDataQueuezmWebSocketDataQueue resumes and pauses an underlying stream. It is a destination for WebSocket data. protocollimitloopreturnNcCsNd|_||_|d|_||_d|_d|_d|_t|_|jj |_ |jj |_ dS)Nrr F) _size _protocol_limit_loop_eof_waiter _exceptionr_bufferpopleft _get_bufferappend _put_buffer)selfr$r%r&rrr__init__7s  zWebSocketDataQueue.__init__cC|jSN)r,r4rrris_eofEzWebSocketDataQueue.is_eofcCr6r7)r.r8rrr exceptionHr:zWebSocketDataQueue.exceptionexc BaseException exc_causecCs4d|_||_|j}durd|_t|||dSdSNT)r,r.r-r)r4r<r>waiterrrrrKs z WebSocketDataQueue.set_exceptioncCs2|j}dur dSd|_|s|ddSdSr7)r-done set_result)r4r@rrr_release_waiterVs z"WebSocketDataQueue._release_waitercCsd|_|dSr?)r,rCr8rrrfeed_eof]s zWebSocketDataQueue.feed_eofdatarsizeint_cCsN|j|7_|||f||j|jkr#|jjs%|jdSdSdSr7)r(r3rCr*r)_reading_paused pause_readingr4rErFrrr feed_dataas zWebSocketDataQueue.feed_datac sb|js-|js-|jr J|j|_z |jIdHW|Stjtjfy,d|_w|Sr7) r/r,r-r+ create_futureasyncioCancelledError TimeoutError_read_from_bufferr8rrrreadhs   zWebSocketDataQueue.readcCsV|jr!|\}}|j|8_|j|jkr|jjr|j|S|jdur)|jtr7) r/r1r(r*r)rHresume_readingr.rrJrrrrPss   z$WebSocketDataQueue._read_from_bufferr'N)rErrFrGr'N)__name__ __module__ __qualname____doc__r rrMAbstractEventLoopr5boolr9rr=r;r builtinsrrCrDrKrrQrPrrrrr#1s4    r#c @seZdZ ddedededdfddZdd d Zd ee e e fde ee ffd d Z d e ddfddZde dee eeeee e feeffddZdS)WebSocketReaderTqueue max_msg_sizecompressr'NcCsr||_||_d|_t|_t|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_d|_||_dS)NFr"r)r\ _max_msg_size_exc bytearray_partial READ_HEADER_state_opcode _frame_fin _frame_opcode_frame_payload_frame_payload_len_tail _has_mask _frame_mask_payload_length_payload_length_flag _compressed_decompressobj _compress)r4r\r]r^rrrr5s$ zWebSocketReader.__init__cCs|jdSr7)r\rDr8rrrrDszWebSocketReader.feed_eofrEc Csrt|tur t|}|jdurd|fSz||WtSty8}z||_t|j|tWYd}~Sd}~wwr?) typebytesr` _feed_data Exceptionrr\EMPTY_FRAME_ERROR EMPTY_FRAME)r4rEr<rrrrKs    zWebSocketReader.feed_datac Cs||D]}|d}|d}|d}|d}|tk}|tks'|tks'|r|sO|s.||_|j|7_|jrNt|j|jkrNtt j d t|j|jqt |j}|rh|jduratt j d|j}d|_n |rstt j d ||r|j|} |jn|} |jrt| |jkrtt j d t| |j|r|jstdd |_|j| t|j} |jjrt|jj} tt j d |j| |jn t| tur| } nt| } |tkrz| d } Wnty} ztt jd | d} ~ ww|jttt| d ft| q|jttt| d ft| q|tkrt|dkrgt |ddd}|dkr=|t!vr=tt j d|z |ddd }Wnty\} ztt jd | d} ~ wwttt"j#||f}n|rytt j d|d|d|ttt"j#dd f}|j|dq|t$krttt"j%|d f}|j|t|q|t&krttt"j'|d f}|j|t|qtt j d|dS)Nrrr r z Message size {} exceeds limit {}z*Continuation frame for non started messagez|B|_n"|dkr||dkrn|||d}|d7}t|d|_n||_|j rtnt|_|jtkr||d kr n|||d |_|d 7}t|_|jtkr||}|j|kr7|}|j|8_n||j}d|_|jr]t|jturQt|j|_|j|||7_n||||_|j||7_|}|jdkrvn8|j r|jdusJt|jturt|j|_t|j|j||j|j |j|j fd|_d|_t|_q||kr||d|_|Sd|_|S)z&Return the next frame from the socket.r"rTr rr!z*Received frame with non-zero reserved bitsz!Received fragmented control frame}z5Control frame payload cannot be larger than 125 bytesNF~)rjrrdrcrqrrrrfrorYrgrkrnREAD_PAYLOAD_LENGTHrmrREAD_PAYLOAD_MASK READ_PAYLOADrlrirrrhrarr2)r4rframes start_pos buf_length first_byte second_byterrsv1rsv2rsv3rhas_masklength length_flagrE chunk_lenend_posrrrr{>s                      zWebSocketReader.parse_frame)TrS)rTrUrVr#rrYr5rDr rsra memoryviewrrKrtrrr{rrrrr[s0    "r[)=rWrMrZ collectionsrtypingrrrrrrr base_protocolr compression_utilsr helpersr rstreamsrrrrmodelsrrrrrrr__annotations__rcrrrBINARYrTEXTr CONTINUATIONvaluer|r}r~rrrrrrrvrwtuple__new__rrGr#r[rrrrs< $   N