o '= j@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)selfrrYC:\Users\psant\Downloads\Anu Raj Backend\school\venv\Lib\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_lenZof_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.Z END_HEADERSrcss|]}|jVqdSN)r).0xrrr fsz4FrameBuffer._update_header_buffer..)r stream_id isinstancerr appendrCONTINUATION_BACKLOGflagsaddjoinrrr)rr(r.Z valid_framer#rrr_update_header_bufferCs:       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 StopIterationrZparse_frame_header memoryviewrrr r'Z parse_bodyr r5__next__)rr(r%errr#Z new_framerrrr:ws4        zFrameBuffer.__next__N)F)rrrr)rr rr)r%r&rr)r(r)rr))rr)rr) __name__ __module__ __qualname____doc__rr$r'r5r6r:rrrrrs     1rN)r? __future__rZhyperframe.exceptionsrrZhyperframe.framerrrr exceptionsr r r r1rrrrrs