o i@s4dZddlmZddlmZmZGdddeZdS)zAsync uWSGI protocol parser for ASGI workers. Reuses the parsing logic from gunicorn/uwsgi/message.py, only async I/O differs. ) UWSGIRequest)InvalidUWSGIHeaderUnsupportedModifierc@sZeZdZdZdddZedddZddZd d Zd d Z dddZ ddZ ddZ dS)AsyncUWSGIRequestaRAsync version of UWSGIRequest. Reuses all parsing logic from the sync version, only async I/O differs. The following methods are reused from the parent class: - _parse_vars() - pure parsing, no I/O - _extract_request_info() - pure transformation - _check_allowed_ip() - no I/O - should_close() - simple logic cCs||_||_||_||_||_d|_d|_d|_d|_d|_ d|_ g|_ g|_ d|_ |jr/dnd|_d|_i|_d|_d|_d|_d|_d|_d|_dS)N)rrhttpshttpFr)cfgunreader peer_addr remote_addr req_numbermethoduripathqueryfragmentversionheaderstrailersbodyis_sslscheme must_close uwsgi_vars modifier1 modifier2proxy_protocol_infocontent_lengthchunked_body_remaining)selfr r r rr#G/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/asgi/uwsgi.py__init__s. zAsyncUWSGIRequest.__init__cs*|||||}||IdH|S)aParse a uWSGI request asynchronously. Args: cfg: gunicorn config object unreader: AsyncUnreader instance peer_addr: client address tuple req_number: request number on this connection (for keepalive) Returns: AsyncUWSGIRequest: Parsed request object Raises: InvalidUWSGIHeader: If the uWSGI header is malformed UnsupportedModifier: If modifier1 is not 0 ForbiddenUWSGIRequest: If source IP is not allowed N)_check_allowed_ip _async_parse)clsr r r rreqr#r#r$parse@s zAsyncUWSGIRequest.parsecs|dIdH}t|dkrtd|d|_t|ddd}|d|_|jdkr1t|j|dkrL||IdH}t||krGtd||| | dS) z=Async version of parse() - reads data then uses sync parsing.Nzincomplete headerrrlittlezincomplete vars block) _async_read_exactlenrrint from_bytesrr _parse_vars_extract_request_info_set_body_reader)r"headerdatasize vars_datar#r#r$r'Ws         zAsyncUWSGIRequest._async_parsecsZt}t||kr)|j|t|IdH}|s t|S||t||ks t|S)z'Read exactly size bytes asynchronously.N) bytearrayr/r readextendbytes)r"sizebufchunkr#r#r$r.os  z#AsyncUWSGIRequest._async_read_exactcCsNd}d|jvrz tt|jdd}Wn tyd}Ynw||_||_dS)z$Set up body state for async reading.rCONTENT_LENGTHN)rmaxr0 ValueErrorrr!)r"rr#r#r$r4ys   z"AsyncUWSGIRequest._set_body_reader csH|jdkrdSt||j}|j|IdH}|r"|jt|8_|S)zRead body chunk asynchronously. Args: size: Maximum bytes to read Returns: bytes: Body data, empty bytes when body is exhausted rN)r!minr r9r/)r"r<to_readdatar#r#r$ read_bodys  zAsyncUWSGIRequest.read_bodycs6|jdkr|dIdH}|sdS|jdksdSdS)zcDrain unread body data. Should be called before reusing connection for keepalive. rrBN)r!rG)r"rFr#r#r$ drain_bodys  zAsyncUWSGIRequest.drain_bodycCs,|}|jD] \}}||kr|SqdS)zGet header by name (case-insensitive). Args: name: Header name to look up Returns: Header value if found, None otherwise N)upperr)r"namehvr#r#r$ get_headers  zAsyncUWSGIRequest.get_headerN)r)rB) __name__ __module__ __qualname____doc__r% classmethodr*r'r.r4rGrHrMr#r#r#r$rs #   rN)rQgunicorn.uwsgi.messagergunicorn.uwsgi.errorsrrrr#r#r#r$s