o }jh@sbdZddlmZddlmZmZddlmZmZm Z m Z ddl m Z m Z mZdZGdd d Zd S) zy h2/frame_buffer ~~~~~~~~~~~~~~~ A data structure that provides a way to iterate over a byte buffer in terms of frames. ) annotations)InvalidDataErrorInvalidFrameError)ContinuationFrameFrame HeadersFramePushPromiseFrame)FrameDataMissingErrorFrameTooLargeError ProtocolError@c@sNeZdZdZddddZdd d ZdddZdddZd ddZd!ddZ dS)" FrameBufferza A buffer data structure for HTTP/2 data that allows iteraton in terms of H2 frames. FserverboolreturnNonecCs2t|_d|_|r dnd|_t|j|_g|_dS)NrsPRI * HTTP/2.0 SM ) bytearray_datamax_frame_size _preamblelen _preamble_len_headers_buffer)selfrrE/var/www/html/bot/env/lib/python3.10/site-packages/h2/frame_buffer.py__init__ s   zFrameBuffer.__init__databytescCs||jr5t|}t|j|}|jd||d|kr d}t|||d}|j|8_|j|d|_|j|7_dS)zs Add more data to the frame buffer. :param data: A bytestring containing the byte buffer. NzInvalid HTTP/2 preamble.)rrminrr r)rrdata_lenof_which_preamblemsgrrradd_data's  zFrameBuffer.add_datalengthintcCs(||jkrd|d|j}t|dS)zB Confirm that the frame is an appropriate length. z Received overlong frame: length z, max N)rr )rr&r$rrr_validate_frame_length;s z"FrameBuffer._validate_frame_lengthf Frame | NonecCs|jr^|jdj}|duot|to|j|k}|sd}t|t|ts&J|j|t|jtkr9d}t|d|jvrZ|jd}|j dd dd|jD|_ g|_|Sd}|St|t t frrd|jvrr|j|d}|S) z Updates the internal header buffer. Returns a frame that should replace the current one. May throw exceptions if this frame is invalid. rNz"Invalid frame during header block.z&Too many continuation frames received. END_HEADERSrcss|]}|jVqdSN)r).0xrrr fsz4FrameBuffer._update_header_buffer..)r stream_id isinstancerr appendrCONTINUATION_BACKLOGflagsaddjoinrrr)rr)r0 valid_framer$rrr_update_header_bufferCs8        z!FrameBuffer._update_header_buffercCs|Sr,r)rrrr__iter__tszFrameBuffer.__iter__rc Cs"t|jdkr tztt|jdd\}}Wnttfy1}z d|}t||d}~wwt|j|dkr=t| |z| t|jdd|Wn%tye}zd}t||d}~wtyw}zd}t ||d}~ww|jd|d|_| |}|dur|S| S)N z$Received frame with invalid header: z&Received frame with non-compliant datazFrame data missing or invalid)rr StopIterationrparse_frame_header memoryviewrrr r( parse_bodyr r8__next__)rr)r&errr$ new_framerrrr?ws4        zFrameBuffer.__next__N)F)rrrr)rr rr)r&r'rr)r)r*rr*)rr)rr) __name__ __module__ __qualname____doc__rr%r(r8r9r?rrrrrs     1rN)rE __future__rhyperframe.exceptionsrrhyperframe.framerrrr exceptionsr r r r3rrrrrs