o 4jKO@sBUdZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z dd lmZd d l mZmZmZd d lmZmZmZmZmZmZd deDZeeeed<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:GdddZ;GdddZsrALLOWED_CLOSE_CODES)T)Frc@seZdZdZdededejddfddZde fd d Z de dBfd d Z e fd e dej ddfddZdddZdddZ      d ddZdeeBfddZdeeBfddZdS)!WebSocketDataQueuezmWebSocketDataQueue resumes and pauses an underlying stream. It is a destination for WebSocket data. protocollimitloopreturnNcCsNd|_||_|d|_||_d|_d|_d|_t|_|jj |_ |jj |_ dS)NrrF) _size _protocol_limit_loop_eof_waiter _exceptionr_bufferpopleft _get_bufferappend _put_buffer)selfr r!r"rrr__init__=s  zWebSocketDataQueue.__init__cC|jSN)r(r0rrris_eofKzWebSocketDataQueue.is_eofcCr2r3)r*r4rrr exceptionNr6zWebSocketDataQueue.exceptionexc exc_causecCs4d|_||_|j}durd|_t|||dSdSNT)r(r*r)r)r0r8r9waiterrrrrQs z WebSocketDataQueue.set_exceptioncCs2|j}dur dSd|_|s|ddSdSr3)r)done set_result)r0r;rrr_release_waiter\s z"WebSocketDataQueue._release_waitercCsd|_|d|_dSr:)r(r>r*r4rrrfeed_eofcs zWebSocketDataQueue.feed_eofdataWSMessage | WSMessageTextBytessize cython_intcCsN|j|7_|||f||j|jkr#|jjs%|jdSdSdSr3)r$r/r>r&r%_reading_paused pause_readingr0r@rBrrr feed_datahs zWebSocketDataQueue.feed_datac sb|js-|js-|jr J|j|_z |jIdHW|Stjtjfy,d|_w|Sr3) r+r(r)r' create_futureasyncioCancelledError TimeoutError_read_from_bufferr4rrrreadqs   zWebSocketDataQueue.readcCsV|jr!|\}}|j|8_|j|jkr|jjr|j|S|jdur)|jtr3) r+r-r$r&r%rDresume_readingr*r rFrrrrL|s   z$WebSocketDataQueue._read_from_bufferr#N)r@rArBrCr#N)__name__ __module__ __qualname____doc__rrrIAbstractEventLoopr1boolr5 BaseExceptionr7rbuiltinsrr>r?rGrrrMrLrrrrr7s@     rc @seZdZdededededdf ddZdd d Zd ee Be Bde eeffd d Z dedee Bdee Bdee Bddf ddZd eddfddZdS)WebSocketReaderqueue max_msg_sizecompress decode_textr#NcCsx||_||_||_d|_t|_t|_t|_ d|_ t|_ g|_ d|_ d|_d|_d|_d|_d|_t|_d|_||_dS)NFrr)rY _max_msg_size _decode_text_exc bytearray_partial READ_HEADER_stateOP_CODE_NOT_SET_opcode _frame_fin _frame_opcode_payload_fragments_frame_payload_len_tail _has_mask _frame_mask_payload_bytes_to_read_payload_len_flagCOMPRESSED_NOT_SET _compressed_decompressobj _compress)r0rYrZr[r\rrrr1s& zWebSocketReader.__init__cCs|jdSr3)rYr?r4rrrr?szWebSocketReader.feed_eofr@c Csrt|tur t|}|jdurd|fSz||WtSty8}z||_t|j|tWYd}~Sd}~wwr:) typebytesr_ _feed_data ExceptionrrYEMPTY_FRAME_ERROR EMPTY_FRAME)r0r@r8rrrrGs    zWebSocketReader.feed_datafinopcodepayload compressedc Cs0|ttthvr|tkr|jtkrttjd|s(|tkr||_|j|7_dSt |j}|tkr8|j}t|_n |rCttjd||rP|j|}|j n|}|r|j s]t dd|_ |j |t|jrk|jdn|j}|jrt||jkrttjd|jn t|tur|}nt|}|tkr|jrz|d}Wnty} zttjd| d} ~ ww|jttt|d ft|dS|jttt|d ft|dS|jttt|d ft|dS|tkr]t|d kr9t |dd d } | d kr| t!vrttjd | z |d dd} Wnty.} zttjd| d} ~ wwttt"j#| | f} n|rKttjd|d|d|ttt"j#d d f} |j| d dS|t$krvttt"j%|d f} |j| t|dS|t&krttt"j'|d f} |j| t|dSttjd|)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_CONTINUATIONrerdrrPROTOCOL_ERRORrarUclearrqrdecompress_syncrr]lenMESSAGE_TOO_BIGrsrtr^decodeUnicodeDecodeError INVALID_TEXTrYrG TUPLE_NEWrWS_MSG_TYPE_TEXTrWS_MSG_TYPE_BINARY OP_CODE_CLOSEr rrCLOSE OP_CODE_PINGPING OP_CODE_PONGPONG) r0ryrzr{r| has_partialassembled_payloadpayload_mergedtextr8 close_code close_messagemsgrrr _handle_frames             zWebSocketReader._handle_framecCsH|jr |j|d}|_d}t|}|} |jtkr||dkr"n||}||d}|d7}|d?d@}|d?d@}|d?d@} |d ?d@} |d @} | sU| sU|r[|js[ttjd | tt t t t t hvrnttjd | | dkr||dkr|ttjd |d?d@} |d@} | dkr| dkrttjd|js|jtkr|rtnt|_n|rttjd | dkrt||_| |_t| |_| |_t|_|jtkr<|j}|dkr||dkrؐn9||}||d}|d7}|d>|B|_n|dkr||dkrnt||d|_|d7}n||_|jr3|jt t thvr3|jt|j}||jkr3ttjd|d|j|jr9tnt|_|jtkrZ||d krJn|||d |_|d 7}t|_|jtkr||}|j|krt|}|j|8_n||j}d|_|j }|j ||7_ |}|}|jdkr|j!"|||nt|r|j!"||||jr|jdusJt#d$|j!}t%|j||}nd$|j!}|j!&n,|jr|jdusJ|||}t'|t#urt#|}t%|j||}n|||}|(|j|j||jd|_ t|_q||kr||||_dSd|_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))rjrrcrbrrrrrrrrrrrrfrproCOMPRESSED_TRUECOMPRESSED_FALSErUrgrkrnREAD_PAYLOAD_LENGTHrmr r]rarREAD_PAYLOAD_MASK READ_PAYLOADrlrirhr.r`joinr rrsr)r0r@ start_posdata_len data_cstr first_byte second_byteryrsv1rsv2rsv3rzhas_masklengthlen_flagprojected_size chunk_len f_end_pos had_fragments f_start_pospayload_bytearrayr{rrrruFs                               &3zWebSocketReader._feed_datarO)rPrQrRrrrUr1r?rtr` memoryviewtuplerGrCrrurrrrrXs6 "  rX)=rSrIrW collectionsrtypingr base_protocolrcompression_utilsrhelpersrrstreamsr r r r modelsrrrrrrrsetr__annotations__rbrrrBINARYrTEXTrrd CONTINUATIONvaluerrrrrrrrrrwrxrorrr__new__rrCrrXrrrrsD       Q