հj:dZddlmZddlmZmZGddeZy)zAsync uWSGI protocol parser for ASGI workers. Reuses the parsing logic from gunicorn/uwsgi/message.py, only async I/O differs. ) UWSGIRequest)InvalidUWSGIHeaderUnsupportedModifiercPeZdZdZd dZed dZdZdZdZ d dZ dZ d Z y ) 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 cb||_||_||_||_||_d|_d|_d|_d|_d|_ d|_ g|_ g|_ d|_ |jrdnd|_d|_i|_d|_d|_d|_d|_d|_d|_y)N)r httpshttpFr)cfgunreader peer_addr remote_addr req_numbermethoduripathqueryfragmentversionheaderstrailersbodyis_sslscheme must_close uwsgi_vars modifier1 modifier2proxy_protocol_infocontent_lengthchunked_body_remaining)selfr rrrs I/var/www/mission/venv/lib/python3.12/site-packages/gunicorn/asgi/uwsgi.py__init__zAsyncUWSGIRequest.__init__s  "$$        !$g #'    cxK|||||}|j|jd{|S7w)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 rrrreqs r&parsezAsyncUWSGIRequest.parse@s@$#xJ7      !s /:8:cK|jdd{}t|dkr td|d|_tj |ddd}|d|_|jdk7rt|j|dkDrC|j|d{}t||kr td|j||j|jy77Qw) z=Async version of parse() - reads data then uses sync parsing.Nzincomplete headerrr littlezincomplete vars block) _async_read_exactlenrrint from_bytesr r _parse_vars_extract_request_info_set_body_reader)r%headerdatasize vars_datas r&r+zAsyncUWSGIRequest._async_parseWs--a00 v;?$%89 9>>&1+x8 >>Q %dnn5 5 a<"44X>>I9~(()@AA   Y ' ""$ '1?s#C3C/BC3C1 AC31C3cKt}t||kr]|jj|t|z d{}|s t |S|j |t||kr]t |S7=w)z'Read exactly size bytes asynchronously.N) bytearrayr4rreadextendbytes)r%sizebufchunks r&r3z#AsyncUWSGIRequest._async_read_exactossk#ho--,,TCH_==ESz JJu  #ho Sz >sABB1B8 Bcd}d|jvr# tt|jdd}||_||_y#t$rd}YwxYw)z$Set up body state for async reading.rCONTENT_LENGTHN)rmaxr5 ValueErrorr"r$)r%r"s r&r9z"AsyncUWSGIRequest._set_body_readerys[ t . #!$S9I)J%KQ!O-- #!" #s"A AAcK|jdkryt||j}|jj|d{}|r|xjt |zc_|S7&w)zRead body chunk asynchronously. Args: size: Maximum bytes to read Returns: bytes: Body data, empty bytes when body is exhausted rr(N)r$minrr?r4)r%rBto_readdatas r& read_bodyzAsyncUWSGIRequest.read_bodysd   1 $dD001]]''00   CI -  1sAA0A.'A0cK|jdkDr-|jdd{}|sy|jdkDr,yy7w)zcDrain unread body data. Should be called before reusing connection for keepalive. r N)r$rM)r%rLs r& drain_bodyzAsyncUWSGIRequest.drain_bodys@ ""Q&--D""Q&-s$AAAAc^|j}|jD]\}}||k(s |cSy)zGet header by name (case-insensitive). Args: name: Header name to look up Returns: Header value if found, None otherwise N)upperr)r%namehvs r& get_headerzAsyncUWSGIRequest.get_headers7zz|LL DAqDy r(N)r )rO) __name__ __module__ __qualname____doc__r' classmethodr.r+r3r9rMrPrVr(r&rrs?!F, 0 ." r(rN)rZgunicorn.uwsgi.messagergunicorn.uwsgi.errorsrrrr\r(r&r_s"  0 [ [r(