o hs@sTdZddlmZddlmZmZmZmZddlm Z m Z m Z dZ Gddde Zd S) zy h2/frame_buffer ~~~~~~~~~~~~~~~ A data structure that provides a way to iterate over a byte buffer in terms of frames. )InvalidFrameError)Frame HeadersFrameContinuationFramePushPromiseFrame) ProtocolErrorFrameTooLargeErrorFrameDataMissingError@c@sReZdZdZdddZddZddZd d Zd d Zd dZ ddZ ddZ 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)selfserverrW/var/www/html/magazine_api/magazine_env/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.)rrminrrr)rrZdata_lenZof_which_preamblerrradd_data)s  zFrameBuffer.add_datac CsJzt|dd\}}W||fSty$}ztdt|d}~ww)z Parses the frame header from the data. Either returns a tuple of (frame, length), or throws an exception. The returned frame may be None if the frame is of unknown type. N z!Invalid frame header received: %s)rZparse_frame_header ValueErrorrstr)rrframelengtherrr_parse_frame_header<szFrameBuffer._parse_frame_headercCs ||jkrtd||jfdS)zB Confirm that the frame is an appropriate length. z*Received overlong frame: length %d, max %dN)rr )rrrrr_validate_frame_lengthJs 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.Z END_HEADERSr css|]}|jVqdSN)r).0xrrr tsz4FrameBuffer._update_header_buffer..)r stream_id isinstancerrappendrCONTINUATION_BACKLOGflagsaddjoinrrr)rfr'Z valid_framerrr_update_header_bufferTs2        z!FrameBuffer._update_header_buffercCs|Sr#rrrrr__iter__szFrameBuffer.__iter__cCst|jdkr tz ||j\}}Wn tytdwt|j|dkr,t|||durQz|t|jdd|Wn tyPt dw|jd|d|_| |}|durf|S| S)Nrz)Received frame with invalid frame header.zFrame data missing or invalid) rr StopIterationr!rrr"Z parse_body memoryviewr r/next)rr.rrrrr4s&     zFrameBuffer.nextcCs|Sr#)r4r0rrr__next__szFrameBuffer.__next__N)F) __name__ __module__ __qualname____doc__rrr!r"r/r1r4r5rrrrr s  . )r N)r9Zhyperframe.exceptionsrZhyperframe.framerrrr exceptionsrr r r*objectr rrrrs