o `g@@sVdZddlmZmZddlmZmZmZmZddl m Z m Z m Z dZ GdddZd S) zy h2/frame_buffer ~~~~~~~~~~~~~~~ A data structure that provides a way to iterate over a byte buffer in terms of frames. )InvalidFrameErrorInvalidDataError)Frame HeadersFrameContinuationFramePushPromiseFrame) ProtocolErrorFrameTooLargeErrorFrameDataMissingError@c@sBeZdZdZdddZddZddZd d Zd d Zd dZ dS) FrameBufferz This is a data structure that expects to act as a buffer for HTTP/2 data that allows iteraton in terms of H2 frames. FcCs0d|_d|_|r dnd|_t|j|_g|_dS)NrsPRI * HTTP/2.0 SM )datamax_frame_size _preamblelen _preamble_len_headers_buffer)selfserverrJ/var/www/html/propose/venv/lib/python3.10/site-packages/h2/frame_buffer.py__init__"s   zFrameBuffer.__init__cCsx|jr3t|}t|j|}|jd||d|krtd||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_preamblerrradd_data)s  zFrameBuffer.add_datacCs ||jkrtd||jfdS)zB Confirm that the frame is an appropriate length. z*Received overlong frame: length %d, max %dN)rr )rlengthrrr_validate_frame_length<s z"FrameBuffer._validate_frame_lengthcCs|jrS|jdj}|duot|to|j|k}|std|j|t|jtkr.tdd|jvrO|jd}|j dd dd|jD|_ g|_|Sd}|St|t t frgd|jvrg|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)rfr% valid_framerrr_update_header_bufferFs2        z!FrameBuffer._update_header_buffercCs|Sr!r)rrrr__iter__tszFrameBuffer.__iter__c Cst|jdkr tzt|jdd\}}Wnttfy.}ztdt|d}~wwt|j|dkr;t| |z| t |jdd|Wnty[tdtyet dw|jd|d|_| |}|dur{|S|S)N z&Received frame with invalid header: %sz&Received frame with non-compliant datazFrame data missing or invalid)rr StopIterationrparse_frame_headerrrr strr parse_body memoryviewr r.__next__)rr,rerrrr6ws.      zFrameBuffer.__next__N)F) __name__ __module__ __qualname____doc__rrrr.r/r6rrrrr s  . r N)r;hyperframe.exceptionsrrhyperframe.framerrrr exceptionsr r r r(r rrrrs