a `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;GdddZrALLOWED_CLOSE_CODES)Tr)Frc@seZdZdZeeejddddZe dddZ e e dd d Z efd ej dd d dZddddZddddZddddddZedddZedddZdS)WebSocketDataQueuezmWebSocketDataQueue resumes and pauses an underlying stream. It is a destination for WebSocket data. N)protocollimitloopreturncCsNd|_||_|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__r'cCs|jSN)r,r4rrris_eofEszWebSocketDataQueue.is_eofcCs|jSr7)r.r8rrr exceptionHszWebSocketDataQueue.exception BaseException)exc exc_causer'cCs0d|_||_|j}dur,d|_t|||dSNT)r,r.r-r)r4r<r=waiterrrrrKs z WebSocketDataQueue.set_exceptioncCs.|j}durdSd|_|s*|ddSr7)r-done set_result)r4r?rrr_release_waiterVs z"WebSocketDataQueue._release_waitercCsd|_|dSr>)r,rBr8rrrfeed_eof]szWebSocketDataQueue.feed_eofrint_)datasizer'cCsF|j|7_|||f||j|jkrB|jjsB|jdSr7)r(r3rBr*r)_reading_paused pause_readingr4rErFrrr feed_dataas zWebSocketDataQueue.feed_datac s^|jsV|jsV|jrJ|j|_z|jIdHWn"tjtjfyTd|_Yn0|Sr7) r/r,r-r+ create_futureasyncioCancelledError TimeoutError_read_from_bufferr8rrrreadhs   zWebSocketDataQueue.readcCsZ|jrB|\}}|j|8_|j|jkr>|jjr>|j|S|jdurR|jtdSr7) r/r1r(r*r)rGresume_readingr.rrIrrrrOss   z$WebSocketDataQueue._read_from_buffer)__name__ __module__ __qualname____doc__r rrLAbstractEventLoopr5boolr9rr;r:r builtinsrrBrCrJrrPrOrrrrr#1s"   r#c@seZdZdeeeddddZddddZee e e fe ee fd d d Z e dd d d Ze ee eeeee e feefdddZdS)WebSocketReaderTN)queue max_msg_sizecompressr'cCsr||_||_d|_t|_t|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_d|_||_dS)NFrr)rZ _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)r4rZr[r\rrrr5s$zWebSocketReader.__init__r6cCs|jdSr7)rZrCr8rrrrCszWebSocketReader.feed_eof)rEr'c Cstt|turt|}|jdur&d|fSz||Wn:tyn}z"||_t|j|tWYd}~Sd}~00tSr>) typebytesr^ _feed_data ExceptionrrZEMPTY_FRAME_ERROR EMPTY_FRAME)r4rEr<rrrrJs   zWebSocketReader.feed_datac Cs||D]}|d}|d}|d}|d}|tk}|tksN|tksN|rD|s|s\||_|j|7_|jr t|j|jkr tt j d t|j|jq t |j}|r|jdurtt j d|j}d|_n|rtt j d ||r|j|} |jn|} |jr8t| |jkr8tt j d t| |j|r|jsRtdd |_|j| t|j} |jjrt|jj} tt j d |j| |jnt| tur| } nt| } |tkr$z| d } Wn4ty} ztt jd | WYd} ~ n d} ~ 00|jttt| d ft| n|jttt| d ft| q |tkr>t|dkrt |ddd}|dkr|t!vrtt j d|z|ddd }Wn4ty} ztt jd | WYd} ~ n d} ~ 00ttt"j#||f}n8|rtt j d|d|d|nttt"j#dd f}|j|dq |t$krnttt"j%|d f}|j|t|q |t&krttt"j'|d f}|j|t|q tt j d|q dS)Nrrr r!z Message size {} exceeds limit {}z*Continuation frame for non started messagezttjd t ||_| |_ t | |_ | |_ t|_|jtkr|j }|dkr||dkrqz||}||d}|d7}|d>|B|_nJ|dkr||dkrڐqz|||d}|d7}t|d|_n||_|j rtnt|_|jtkrX||dkr8qz|||d|_|d7}t|_|jtkr(||}|j|kr|}|j|8_n||j}d|_|jrt|jturt|j|_|j|||7_n||||_|j||7_|}|jdkr qz|j rL|jdus"Jt|jtur>t|j|_t|j|j||j|j |j|j fd|_d|_t|_q(||kr||dnd|_|S)z&Return the next frame from the socket.rrr rr"z*Received frame with non-zero reserved bitsz!Received fragmented control frame}z5Control frame payload cannot be larger than 125 bytesNTF~)rhr|rbrarorrr~rdrmrWrerirlREAD_PAYLOAD_LENGTHrkrREAD_PAYLOAD_MASK READ_PAYLOADrjrgrprfr_rr2)r4rframes start_posZ buf_lengthZ first_byteZ second_byterZrsv1Zrsv2Zrsv3rZhas_masklengthZ length_flagrEZ chunk_lenend_posrrrrx>s                      zWebSocketReader.parse_frame)T)rRrSrTr#rrWr5rCr rqr_ memoryviewrrJrrrrrxrrrrrYs   "rY)=rUrLrX collectionsrtypingrrrrrrr Z base_protocolr Zcompression_utilsr helpersr rstreamsrrrrmodelsrrrrrr r__annotations__rarrrBINARYrZTEXTrZ CONTINUATIONvalueryrzr{rrrrrrrtrutuple__new__rrDr#rYrrrrs: $   N