o iX@sDdZddlmZddlmZGdddZGdddZddgZdS) zV HTTP/2 request wrapper. Provides a Request-compatible interface for HTTP/2 streams. )BytesIO)split_request_uric@sNeZdZdZddZdddZdddZdd d Zd d Zd dZ ddZ dS) HTTP2BodyzBody wrapper for HTTP/2 request data. Provides a file-like interface to the request body, compatible with gunicorn's Body class expectations. cCst||_t||_dS)zfInitialize with body data. Args: data: bytes containing the request body N)r_datalen_len)selfdatar J/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/http2/request.py__init__s zHTTP2Body.__init__NcC|dur |jS|j|S)zRead data from the body. Args: size: Number of bytes to read, or None for all remaining Returns: bytes: The requested data N)rreadrsizer r r r!  zHTTP2Body.readcCr )zRead a line from the body. Args: size: Maximum bytes to read Returns: bytes: A line of data N)rreadlinerr r r r.rzHTTP2Body.readlinecCs |j|S)zRead all lines from the body. Args: hint: Approximate byte count hint Returns: list: List of lines )r readlines)rhintr r r r;s zHTTP2Body.readlinescCs t|jS)zIterate over lines in the body.)iterrrr r r __iter__F zHTTP2Body.__iter__cCs|jS)zReturn the content length.)rrr r r __len__JszHTTP2Body.__len__cCs|jdS)zClose the body stream.N)rcloserr r r rNszHTTP2Body.close)N) __name__ __module__ __qualname____doc__r rrrrrrr r r r rs   rc@sPeZdZdZddZddZddZdd Zed d Z ed d Z ddZ dS) HTTP2RequestzHTTP/2 request wrapper compatible with gunicorn Request interface. Wraps an HTTP2Stream to provide the same interface as the HTTP/1.x Request class, allowing workers to handle HTTP/2 requests using existing code paths. c Csl||_||_||_||_d|_|}|dd|_|dd|_|dd}|dd }||_ zt |}|j p9d|_ |j p?d|_ |j pEd|_ WntyZ||_ d|_ d|_ Ynw||_g|_|D]\}} |j|| fqe|rd d |jD|_|jd |fg|_|jrd d |jD|_|} t| |_d|_d|_|j|_d|_|j|_|j|_dS)zInitialize from an HTTP/2 stream. Args: stream: HTTP2Stream instance with received headers/body cfg: Gunicorn configuration object peer_addr: Client address tuple (host, port) )rz:methodGETz:schemehttpsz :authorityz:path/cSs g|] \}}|dkr||fqS)HOSTr ).0nvr r r s z)HTTP2Request.__init__..r%cSsg|] \}}||fqSr )upper)r&namevaluer r r r)s FN)streamcfg peer_addr remote_addrversionget_pseudo_headersgetmethodschemeurirpathqueryfragment ValueError _authorityheadersget_regular_headersappendr*trailersget_request_bodyrbody must_close_expected_100_continue stream_id req_numberproxy_protocol_infopriority_weightpriority_depends_on) rr-r.r/pseudo authorityr7partsr+r, body_datar r r r [sR        zHTTP2Request.__init__cCs d|_dS)z1Force the connection to close after this request.TNrBrr r r force_closerzHTTP2Request.force_closecCs|jrdSdS)zCheck if connection should close after this request. HTTP/2 connections are persistent by design, but we may still need to close if explicitly requested. Returns: bool: True if connection should close TFrMrr r r should_closes zHTTP2Request.should_closecCs,|}|jD] \}}||kr|SqdS)zGet a header value by name. Args: name: Header name (case-insensitive) Returns: str: Header value, or None if not found N)r*r<)rr+h_nameh_valuer r r get_headers  zHTTP2Request.get_headercCs6|d}|durzt|WStyYdSwdS)zsGet the Content-Length header value. Returns: int: Content length, or None if not set zCONTENT-LENGTHN)rRintr:)rclr r r content_lengths   zHTTP2Request.content_lengthcCs |dS)zoGet the Content-Type header value. Returns: str: Content type, or None if not set z CONTENT-TYPE)rRrr r r content_types zHTTP2Request.content_typecCs d|jd|jd|jjdS)Nz)r4r7r-rDrr r r __repr__szHTTP2Request.__repr__N) rrrrr rNrOrRpropertyrUrVrXr r r r rSsM   rN)rior gunicorn.utilrrr__all__r r r r s  B