o f>hgI@sRUdZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZmZdd lmZd d lmZmZmZd d lmZmZmZmZmZd deDZee ee d<d Z!dZ"dZ#dZ$ej%Z&ej'Z(dZ)ej*j+Z,ej'j+Z-ej%j+Z.ej/j+Z0ej1j+Z2ej3j+Z4dZ5dZ6dZ7dZ8d Z9e:j;ZGdddZ?dS)z0Reader for WebSocket protocol versions 13 and 8.N)deque)DequeFinalOptionalSetTupleUnion) BaseProtocol)ZLibDecompressor) _EXC_SENTINEL set_exception) EofStream)UNPACK_CLOSE_CODE UNPACK_LEN3websocket_mask)WS_DEFLATE_TRAILINGWebSocketError WSCloseCode WSMessage WSMsgTypecCsh|]}t|qS)int).0irrO/var/www/html/venv/lib/python3.10/site-packages/aiohttp/_websocket/reader_py.py srALLOWED_CLOSE_CODES)T)Fr"c@seZdZdZdededejddfddZde fd d Z de e fd d Z efd e dej ddfddZdddZdddZd 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__<s  zWebSocketDataQueue.__init__cC|jSN)r,r4rrris_eofJzWebSocketDataQueue.is_eofcCr6r7)r.r8rrr exceptionMr:zWebSocketDataQueue.exceptionexc exc_causecCs4d|_||_|j}durd|_t|||dSdSNT)r,r.r-r )r4r<r=waiterrrrr Ps z WebSocketDataQueue.set_exceptioncCs2|j}dur dSd|_|s|ddSdSr7)r-done set_result)r4r?rrr_release_waiter[s z"WebSocketDataQueue._release_waitercCsd|_|d|_dSr>)r,rBr.r8rrrfeed_eofbs zWebSocketDataQueue.feed_eofdatarsize cython_intcCsN|j|7_|||f||j|jkr#|jjs%|jdSdSdSr7)r(r3rBr*r)_reading_paused pause_readingr4rDrErrr feed_datags zWebSocketDataQueue.feed_datac sb|js-|js-|jr J|j|_z |jIdHW|Stjtjfy,d|_w|Sr7) r/r,r-r+ create_futureasyncioCancelledError TimeoutError_read_from_bufferr8rrrreadns   zWebSocketDataQueue.readcCsV|jr!|\}}|j|8_|j|jkr|jjr|j|S|jdur)|jtr7) r/r1r(r*r)rGresume_readingr.rrIrrrrOys   z$WebSocketDataQueue._read_from_bufferr'N)rDrrErFr'N)__name__ __module__ __qualname____doc__r rrLAbstractEventLoopr5boolr9r BaseExceptionr;r builtinsr rBrCrJrrPrOrrrrr#6s4    r#c @seZdZ ddedededdfddZdd d Zd ee e e fde ee ffd d Z dedeeefdee e fdeeefddf ddZd e ddfddZdS)WebSocketReaderTqueue max_msg_sizecompressr'NcCsr||_||_d|_t|_t|_t|_d|_ t|_ g|_ d|_ d|_ d|_d|_d|_d|_t|_d|_||_dS)NFrr")r\ _max_msg_size_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)r4r\r]r^rrrr5s$ zWebSocketReader.__init__cCs|jdSr7)r\rCr8rrrrCszWebSocketReader.feed_eofrDc Csrt|tur t|}|jdurd|fSz||WtSty8}z||_t|j|tWYd}~Sd}~wwr>) typebytesr` _feed_data Exceptionr r\EMPTY_FRAME_ERROR EMPTY_FRAME)r4rDr<rrrrJs    zWebSocketReader.feed_datafinopcodepayload compressedc Csj|ttthvr|s5|tkr||_|j|7_|jr3t|j|jkr3ttj dt|jd|jdSt |j}|tkrP|jt krIttj d|j}t |_n |r[ttj d||rh|j|}|j n|}|jrt||jkrttj dt|d|j|r|jstdd|_|j|t|jr|jdn|j}|jrt||jkrttj d|jn t|tur|}nt|}|tkrz|d }Wnty} zttjd | d} ~ ww|jttt|d ft|dS|jttt|d ft|dS|tkrzt|d krVt|dd d } | dkr,| tvr,ttj d| z |d dd } WntyK} zttjd | d} ~ wwttt j!| | f} n|rhttj d|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|dSttj d|)Nz Message size z exceeds limit z*Continuation frame for non started messagez8The opcode in non-fin frame is expected to be zero, got T)suppress_deflate_headerrz(Decompressed message exceeds size limit zutf-8zInvalid UTF-8 text messager ri zInvalid close code: zInvalid close frame:  zUnexpected opcode=)& OP_CODE_TEXTOP_CODE_BINARYOP_CODE_CONTINUATIONrfrbr_lenrrMESSAGE_TOO_BIGrXrePROTOCOL_ERRORclearrrr decompress_syncrrtrudecodeUnicodeDecodeError INVALID_TEXTr\rJ TUPLE_NEWrWS_MSG_TYPE_TEXTWS_MSG_TYPE_BINARY OP_CODE_CLOSErrrCLOSE OP_CODE_PINGPING OP_CODE_PONGPONG) r4rzr{r|r} has_partialassembled_payloadpayload_mergedtextr< close_code close_messagemsgrrr _handle_frames              zWebSocketReader._handle_framecCs|jr |j|d}|_d}t|}|} |jtkr||dkr"n||}||d}|d7}|d?d@}|d?d@}|d?d@} |d ?d@} |d @} | sU| sU|r[|js[ttjd | dkri|dkrittjd |d?d@} |d @} | dkr| dkrttjd|js|j t kr|rt nt |_ n|rttjd t ||_| |_t | |_| |_t|_|jtkr|j}|dkr||dkrn||}||d}|d7}|d>|B|_n|dkr||dkrnt||d|_|d7}n||_|jrtnt|_|jtkr||d krn|||d |_|d 7}t|_|jtkr||}|j|kr1|}|j|8_n||j}d|_|j}|j||7_|}|}|jdkrZ|j|||nt|r|j||||jr|jdussJtd|j}t|j||}nd|j}|jn,|jr|jdusJ|||}t|turt|}t|j||}n|||}||j|j||j d|_t|_q||kr||||_dSd|_dS)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 bytes~N) rkrrdrcrsrrrrgrqrpCOMPRESSED_TRUECOMPRESSED_FALSErXrhrlroREAD_PAYLOAD_LENGTHrnrREAD_PAYLOAD_MASK READ_PAYLOADrmrjrir2rajoinrrrtr)r4rD start_posdata_len data_cstr first_byte second_byterzrsv1rsv2rsv3r{has_masklengthlen_flag chunk_len f_end_pos had_fragments f_start_pospayload_bytearrayr|rrrrvBs                           &zWebSocketReader._feed_data)TrR)rSrTrUr#rrXr5rCrrura memoryviewrrJrFrrvrrrrr[s<         r[)@rVrLrZ collectionsrtypingrrrrrr base_protocolr compression_utilsr helpersr r streamsrrrrmodelsrrrrrrr__annotations__rcrrrBINARYrTEXTrre CONTINUATIONvaluerrrrrrrrrrxryrprrtuple__new__rrFr#r[rrrrsD     O