o jf@sUdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ddlmZmZdd lmZdd lmZd d lmZmZmZd d lmZmZmZmZmZmZddeDZe e!e"d<d Z#dZ$dZ%dZ&ej'Z(ej)Z*ej+Z,dZ-ej.j/Z0ej+j/Z1ej)j/Z2ej3j/Z4ej5j/Z6ej7j/Z8dZ9dZ:dZ;dZj?Z@dZAee!e"d<dZBeeCe"d<e!ZDGdddZEGdddZFdS)z0Reader for WebSocket protocol versions 13 and 8.N)deque)Final) BaseProtocol)TooManyMembersErrorZLibDecompressor) _EXC_SENTINEL set_exception) ws_logger) EofStream)UNPACK_CLOSE_CODE UNPACK_LEN3websocket_mask)WS_DEFLATE_TRAILINGWebSocketError WSCloseCode WSMessageWSMessageTextBytes WSMsgTypecCsh|]}t|qS)int).0irrO/opt/steambot/venv/lib/python3.10/site-packages/aiohttp/_websocket/reader_py.py srALLOWED_CLOSE_CODES)T)Fr MSG_SIZE_OVERHEADziWebSocketReader was garbage collected while stalled; callers of set_parser() must hold a strong referenceSTALLED_READER_COLLECTEDc@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. protocollimitloopreturnNcCsTd|_||_|d|_||_d|_d|_d|_t|_|jj |_ |jj |_ d|_ dS)NrrF)_size _protocol_limit_loop_eof_waiter _exceptionr_bufferpopleft _get_bufferappend _put_buffer_stalled_reader)selfr%r&r'rrr__init__Os    zWebSocketDataQueue.__init__cC|jSN)r-r6rrris_eof^zWebSocketDataQueue.is_eofcCr8r9)r/r:rrr exceptionar<zWebSocketDataQueue.exceptionexc exc_causecCs4d|_||_|j}durd|_t|||dSdSNT)r-r/r.r )r6r>r?waiterrrrr ds z WebSocketDataQueue.set_exceptioncCs2|j}dur dSd|_|s|ddSdSr9)r.done set_result)r6rArrr_release_waiteros z"WebSocketDataQueue._release_waitercCsd|_|d|_dSr@)r-rDr/r:rrrfeed_eofvs zWebSocketDataQueue.feed_eofdataWSMessage | WSMessageTextBytessize cython_intcCsR|j|t7_|||f||j|jkr%|jjs'|jdSdSdSr9)r)r"r4rDr+r*_reading_paused pause_reading)r6rFrHrrr feed_data{s zWebSocketDataQueue.feed_datac sb|js-|js-|jr J|j|_z |jIdHW|Stjtjfy,d|_w|Sr9) r0r-r.r, create_futureasyncioCancelledError TimeoutError_read_from_bufferr:rrrreads   zWebSocketDataQueue.readcCs|jrW|\}}|j|t8_|jdurA|j|jdkrA|}dur-|dnd|_tt |j durA| t t |jdurU|j|jkrU|j jrU|j |S|j dur_|j t)Nrr )r0r2r)r"r5r+rLr warningr#r/r RuntimeErrorr*rJresume_readingr )r6rFrHreaderrrrrQs&        z$WebSocketDataQueue._read_from_bufferr(N)rFrGrHrIr(N)__name__ __module__ __qualname____doc__rrrNAbstractEventLoopr7boolr; BaseExceptionr=rbuiltinsr rDrErLrrrRrQrrrrr$Is@     r$c @seZdZdededededdf ddZdd d Zd ee Be Bde eeffd d Z dedee Bdee Bdee Bddf ddZd eddfddZdS)WebSocketReaderqueue max_msg_sizecompress decode_textr(NcCs||_||_||_t||_d|_t|_t |_ t |_ d|_ t |_g|_|r.td|dnd|_t|_d|_d|_d|_d|_d|_d|_t|_d|_||_dS)NFirr )ra _max_msg_size _decode_textweakrefref _weak_self_exc bytearray_partial READ_HEADER_stateOP_CODE_NOT_SET_opcode _frame_fin _frame_opcode_payload_fragmentsmax_max_fragments_payload_buffer_frame_payload_len_tail _has_mask _frame_mask_payload_bytes_to_read_payload_len_flagCOMPRESSED_NOT_SET _compressed_decompressobj _compress)r6rarbrcrdrrrr7s,  zWebSocketReader.__init__cCs|jdSr9)rarEr:rrrrEszWebSocketReader.feed_eofrFc Csrt|tur t|}|jdurd|fSz||WtSty8}z||_t|j|tWYd}~Sd}~wwr@) typebytesrk _feed_data Exceptionr raEMPTY_FRAME_ERROR EMPTY_FRAME)r6rFr>rrrrLs    zWebSocketReader.feed_datafinopcodepayload compressedc Csz|ttthvr |tkr|jtkrttjd|s7|tkr.|jtkr+ttjd|||_|j|7_dSt |j}|tkrG|j}t|_n |rRttjd||r_|j|}|j n|}|r|j slt dd|_ z|j |t|jr{|jdn|j}Wnty}zttjd|d}~ww|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|t krt|d kr^t!|dd d } | d kr4| t"vr4ttjd| z |d dd} WntyS}zttjd |d}~wwttt#j$| | f} n|rpttjd|d|d|ttt#j$d d f} |j| d dS|t%krttt#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/Compressed message has too many deflate membersz(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]clearrrdecompress_syncrrfrMESSAGE_TOO_BIGlenrrrgdecodeUnicodeDecodeError INVALID_TEXTrarL TUPLE_NEWrWS_MSG_TYPE_TEXTrWS_MSG_TYPE_BINARY OP_CODE_CLOSEr rrCLOSE OP_CODE_PINGPING OP_CODE_PONGPONG) r6rrrr has_partialassembled_payloadpayload_mergedr>text close_code close_messagemsgrrr _handle_frames              zWebSocketReader._handle_framecCsZd|j_|jr|j|d}|_d}t|}|} ||kr,|jj|jjkr,|j|j_nn|jtkr||dkr9na||}||d}|d7}|d?d@}|d?d@}|d ?d@} |d ?d@} |d @} | sl| sl|rr|j srt t j d | t ttttthvrt t j d | | dkr|dkrt t j d|d?d@} |d@} | dkr| dkrt t j d| dkr|js|jtkr|rtnt|_n|rt t j d t||_| |_t| |_| |_t|_|jtkrx|j}|dkr||dkrn||}||d}|d7}|d>|B|_n<|dkr?||dkrnt||d}|tkr7t t j dt!|t|j"d|j#p3t||_|d7}n||_|j#ro|jttt hvrot|j"}|j|j#|krot t j dt!|j|d|j#|jrut$nt%|_|jt$kr||d krn|||d |_&|d 7}t%|_|jt%kr||}|j|kr|}|j|8_n||j}d|_|j'}|j'||7_'|}|}|jdkr||kr|j()||||j*rt|j(|j*kr|j+d,|j(7_+|j(-n|r\|j()||||j+r3|j+d,|j(7_+|jr+|j&dus$Jt.|j&|j+|j+}t/|_+n#|jrP|j&dus?Jt/d,|j(}t.|j&||}nd,|j(}|j(-n,|jr|j&dushJ|||}t0|t/uryt/|}t.|j&||}n|||}|1|j|j||jd|_'t|_q||kr||||_dSd|_dS)z&Return the next frame from the socket.Nr rTrr 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 )2rar5ryrr)r+rjrornrrrrrrrrrrrrrr~COMPRESSED_TRUECOMPRESSED_FALSEr]rsrzr}READ_PAYLOAD_LENGTHr|rMAX_PAYLOAD_LENrrrmrfREAD_PAYLOAD_MASK READ_PAYLOADr{rxrtr3rvrwjoinrrrlrr)r6rF start_posdata_len data_cstr first_byte second_byterrsv1rsv2rsv3rhas_masklengthlen_flag frame_len partial_len chunk_len f_end_pos had_fragments f_start_posrpayload_bytearrayrrrrs>                                     &WzWebSocketReader._feed_datarW)rXrYrZr$rr]r7rErrl memoryviewtuplerLrIrrrrrrr`s6 &" r`)Gr[rNr_sysrh collectionsrtypingr base_protocolrcompression_utilsrrhelpersrr logr streamsr r rrmodelsrrrrrrrsetr__annotations__rnrrrmaxsizerBINARYrTEXTrrp CONTINUATIONvaluerrrrrrrrrrrr~rrr__new__rr"r#strrIr$r`rrrrsR       k