o /h@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. FserverboolreturnNonecCs0d|_d|_|r dnd|_t|j|_g|_dS)NrsPRI * HTTP/2.0 SM )datamax_frame_size _preamblelen _preamble_len_headers_buffer)selfrrH/var/www/html/govbot/env/lib/python3.10/site-packages/h2/frame_buffer.py__init__ s   zFrameBuffer.__init__rbytescCs||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'r. 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 r6__next__)rr'r$errr" new_framerrrr=ws4        zFrameBuffer.__next__N)F)rrrr)rrrr)r$r%rr)r'r(rr()rr)rr) __name__ __module__ __qualname____doc__rr#r&r6r7r=rrrrrs     1rN)rC __future__rhyperframe.exceptionsrrhyperframe.framerrrr exceptionsr r r r1rrrrrs