AҐi7dZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z d ZGd d ZGd d ej Zy)z ASGI protocol handler for gunicorn. Implements asyncio.Protocol to handle HTTP/1.x and HTTP/2 connections and dispatch to ASGI applications. N)datetime) AsyncUnreader) AsyncRequest)AsyncUWSGIRequest) NoMoreData)UWSGIParseExceptionc&|rt|ddSdS)zNormalize socket address to ASGI-compatible (host, port) tuple. ASGI spec requires server/client to be (host, port) tuples. IPv6 sockets return 4-tuples (host, port, flowinfo, scope_id), so we extract just the first two elements. N)tuple)sockaddrs O/var/www/descvideos/venv/lib/python3.12/site-packages/gunicorn/asgi/protocol.py_normalize_sockaddrrs#+5"1 44ceZdZdZdZy)ASGIResponseInfoz;Simple container for ASGI response info for access logging.c||_||_g|_|D]d\}}t|tr|j d}t|tr|j d}|jj ||ffy)Nlatin-1)statussentheaders isinstancebytesdecodeappend)selfrrrnamevalues r __init__zASGIResponseInfo.__init__$sp   " /KD%$&{{9-%' Y/ LL  u .  /rN)__name__ __module__ __qualname____doc__rrr rr!s E /rrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZddZdZdZdZdZdZdZdZy) ASGIProtocolzHTTP/1.1 protocol handler for ASGI applications. Handles connection lifecycle, request parsing, and ASGI app invocation. c||_|j|_|j|_|j|_d|_d|_d|_d|_d|_ d|_ d|_ y)NrF) workercfglogasgiapp transportreaderwriter_task req_count_closed_receive_queue)rr's r rzASGIProtocol.__init__7s^ ::::;;    "rc||_|jxjdz c_|jd}|rut |dri|j }|dk(rTt j|_|jjj|j|||_ yt j|_||_ |jjj|j|_ y)z(Called when a connection is established. ssl_objectselected_alpn_protocolh2N)r,r'nr_connsget_extra_infohasattrr6asyncio StreamReaderr-loop create_task_handle_http2_connectionr/r._handle_connection)rr,r5alpns r connection_madezASGIProtocol.connection_madeGs" !--l; '*.FG446Dt|&224 ![[--9911)ZH **,  [[%%11$2I2I2KL rcT|jr|jj|yy)z/Called when data is received on the connection.N)r- feed_data)rdatas r data_receivedzASGIProtocol.data_receivedas ;; KK ! !$ ' rc&|jryd|_|jxjdzc_|jr|jj |j |j j ddi|jr|jjsht|jdd}|dkDr1|jjj||jy|jjyyy) aCalled when the connection is lost or closed. Instead of immediately cancelling the task, we signal a disconnect event and send an http.disconnect message to the receive queue. This allows the ASGI app to clean up resources (like database connections) gracefully before the task is cancelled. See: https://github.com/benoitc/gunicorn/issues/3484 NTr4typehttp.disconnectasgi_disconnect_grace_periodr)r1r'r8r-feed_eofr2 put_nowaitr/donegetattrr(r= call_later_cancel_task_if_pendingcancel)rexc grace_periods r connection_lostzASGIProtocol.connection_lostfs <<   ! ;; KK "    *    * *F4E+F G ::djjoo/"488-KQOLa   ++ 00  !!#0:rc|jr6|jjs|jjyyy)z9Cancel the task if it's still pending after grace period.N)r/rNrRrs r rQz$ASGIProtocol._cancel_task_if_pendings- ::djjoo/ JJ   0:rc> |jj|y#t$re}|jtjtj tj fvr |jjdYd}~yYd}~yd}~wttf$rYywxYw)axWrite data to transport, handling connection errors gracefully. Catches exceptions that occur when the client has disconnected: - OSError with errno EPIPE, ECONNRESET, ENOTCONN - RuntimeError when transport is closing/closed - AttributeError when transport is None These are silently ignored since the client is already gone. zSocket error writing response.N) r,writeOSErrorerrnoEPIPE ECONNRESETENOTCONNr) exception RuntimeErrorAttributeError)rrEes r _safe_writezASGIProtocol._safe_writesw  NN  & Ewwu{{E,<,>T[[%=%==HHMM"PQ(-DKK% %I  (9(9xx))((***mllz  ! ! #m## "*HHNN#:A> N*+%%   C HH  > B B C  ! ! #sKAI*2A H&=H >4H&2H#3H&7'I*I"I*:I%;CI*<I(=I*K H&#H&& I/I*1I9II*II*%I*(I**J-=J0?J-J(#J0(J--J00KKc|jdk7ryd}d}|jD]1\}}|dk(r|j}|dk(s"|j}3|dk(xr|xrd|vS)a Check if request is a WebSocket upgrade. Per RFC 6455 Section 4.1, the opening handshake requires: - HTTP method MUST be GET - Upgrade header MUST be "websocket" (case-insensitive) - Connection header MUST contain "Upgrade" GETFNUPGRADE CONNECTION websocketupgrade)methodrlower)rrzr connectionrrs r rmz"ASGIProtocol._is_websocket_upgradesv >>U " "?? +KD%y ++-%"[[]  + +%P*Pj9PPrcKddlm}|j|||}||j|j||j |j }|jd{y7w)z!Handle WebSocket upgrade request.r)WebSocketProtocolN)gunicorn.asgi.websocketr_build_websocket_scoper,r-r+r)run)rrzrfrerscope ws_protocols r rnzASGIProtocol._handle_websocketsT=++GXxH' NNDKK$(( oosA!A+#A)$A+c   Kj||}ddd ddgdtj _d jdk(r,j s j ddddd{d n%tjj  fd } f d }j||}d} tj} jjjj|||d{  sj!dd d{dd_ tj| z } t)}j$j+||| jj-j||j3ryjj4xrjj6S777#tj"$rj$j'd Yd_ tj z } t)}j$j+||| jj-j||y#t.$rj$j1d YywxYwt.$rj$j1dsj!dd d{7dYd_ tj z } t)}j$j+||| jj-j||y#t.$rj$j1d YywxYwwxYw#t.$rj$j1d YwxYw#d_ tj z } t)}j$j+||| jj-j||w#t.$rj$j1d YwwxYwxYww)zHandle a single HTTP request.FNr http.requestrrHbody more_bodyTcKjrrddiSjd{}|jddk(r|jddsd|S7.w)NrHrIrrT)r1get)msg body_complete receive_queuers r receivez2ASGIProtocol._handle_http_request..receive%s]||  122%))++Cwwv.0d9S $ J ,s&AA/Ac Kjry|d}|dk(r?|jd}|jdg}j||d{y|dk(rw r tdyd |d |jdg t d D}|s j d k\rdt d gz j d{y|d k(r s td y r td y|jdd}|jdd}|r)j|d{ t|z |srjdd yyy7 771w)NrHhttp.response.informationalrrhttp.response.startResponse already startedTc3K|]o\}}t|tr|jn|jdk(xs3t|tr|jn|jdk(qyw)scontent-lengthzcontent-lengthN)rstrr).0r_s r zBASGIProtocol._handle_http_request..send..Nsg) a&0c%:TZZ\ Qbbe(24(= 4::.send3s ||vH88%kk(3 &{{9b9 ..{L'RRR00##/0J#KL#' ")("3#*;;y"#= &))$4)&"*goo.G"&K'+,<'=Ac@d'd$//AQSZ[[[11'#/0F#GL$#/0K#LL{{63/#KK U; //$ /DDD!SY.M "((6(,% !23S.\Es8AE'E A;E'E#A"E'3E%4-E'#E'%E'Internal Server Errorz'Request cancelled (client disconnected)Exception in post_request hookError in ASGI application)_build_http_scoper;Queuer2content_lengthrputr>_read_body_to_queue_build_environrnowr( pre_requestr'r+_send_error_responservr)rlraccess post_requestrwr_ should_closersrt)rrzrfrerrrenvironresp request_start request_timerrrrrrrrrs`` @@@@@@@@@r roz!ASGIProtocol._handle_http_requests&&w(C !     +   ! !Q &w##&"%   !M    8 8- P Q ; -; -|%%gxB# E$LLNM HH g 6((5'40 0 0'""$//5LMMM"%#'D  E'||~ = '9I=Ygw E%%dkk7GTJ    !{{  7TXX%7%77E z 1N%%  HHNND E#'D  E'||~ = '9I=Ygw E%%dkk7GTJ E""#CD E!  HH  : ;#//5LMMM"%#'D  E'||~ = '9I=Ygw E%%dkk7GTJ E""#CD E!   E""#CD E#'D  E'||~ = '9I=Ygw E%%dkk7GTJ E""#CD Es!A2R?G>ARAH&H'HHH RA*N5?ARHH.N23O 4R)A*QQ>$Q;8Q>:Q;;Q>>RcjK |jdd{}|r|jd|ddd{n|jddddd{yX7C7&7 #t$rD}|jj d||jddddd{7Yd}~yd}~wwxYww) z.Read request body and put chunks on the queue.TiNrrrFzError reading body: %s) read_bodyrrwr)rl)rrzqueuechunkrbs r rz ASGIProtocol._read_body_to_queues %//66)) . %%)%  )) . #%*% 6    HHNN3Q 7))&"   slB3A#AA#AA#A!A#B3A#A#!A## B0,4B+ B#!B+&B3+B00B3c g}|jD]D\}}|j|jjd|jdfFt |}t |}dddd|j dd|j d|j |j|j|jr|jjdnd |jr|jjdnd |jjxsd |||d } t|jd r|jj| d <t|d rd|j|j di| d<| S)z*Build ASGI HTTP scope from parsed request.rrh3.02.4r spec_versionr.r4r rHr* http_versionrschemepathraw_path query_string root_pathrserverclientstatepriority_weighthttp.response.priorityweight depends_on extensions)rrrencoderrrrrqueryr(rr:r'rrpriority_depends_on) rrzrfrerrrrrrs r rzASGIProtocol._build_http_scopes["?? VKD% NNDJJL// :ELLEO  '  & sAAcK|jd}d|d|dt|d}|j|jd|j|yw)zSend an error response.zutf-8z HTTP/1.1 rz, Content-Type: text/plain Content-Length: z Connection: close rN)rrrc)rrrrrs r rz!ASGIProtocol._send_error_responseUse~~g&xq *"4yk*   34 sAAciddddddddd d d d d ddddddddddddddddddd d!d"id#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdB}|j|dCS)Dz'Get HTTP reason phrase for status code.dContinueezSwitching Protocolsgz Early HintsOKCreatedAcceptedz No ContentzPartial Contenti-zMoved Permanentlyi.Foundi/z See Otheri0z Not Modifiedi3zTemporary Redirecti4zPermanent Redirectiz Bad Requesti Unauthorizedi Forbiddeniz Not FoundizMethod Not AllowedizRequest TimeoutiConflictiGoneizLength RequiredizPayload Too Largeiz URI Too LongizUnsupported Media TypeizUnprocessable EntityizToo Many RequestsrrizNot Implementediz Bad GatewayizService UnavailableizGateway TimeoutUnknown)r)rrreasonss r rzASGIProtocol._get_reason_phrasebs" " &"  "   "  "  "  "  ""  $"  "  "  "  %"  %"  " !" " #" $ %" & %'" ( ")" * +" , -" . "/" 0 $1" 2 3" 4 )5" 6 '7" 8 $9" : (;" < "=" > ?" @ &A" B "C" F{{69--rc|jrd|jsW |jjr|jj|jj d|_yyy#t $rYwxYw)zClose the transport safely. Calls write_eof() first if supported to signal end of writing, which helps ensure buffered data is flushed before closing. TN)r,r1 can_write_eof write_eofcloserwrWs r rxzASGIProtocol._close_transportsj >>$,, >>//1NN,,.$$& DL#/>   sAA22 A>=A>c K ddlm}|jd}|jd}|j}t j |}t j ||||jj}||j|||} | jd{| |_ | js!|jjr | jdd{} | D]D} |j'| | ||d{| j3| j,j.F|jxj4t7| z c_|jj4|jj8k\r-|j"j;d d |j_ n$| js|jjr t?|dr# |jjAd{|jCy7z77#tj$rYt $r&} |j"j%d| Yd} ~ d} ~ wwxYw7_#t $rp} |j"j)d  | j+| j,j.d t1| d{7n#t $rYnwxYwYd} ~ d} ~ wwxYw#| j3| j,j.wxYw#tj<$rYLt $r'} |j"j)d | Yd} ~ vd} ~ wwxYw7T#t $rY^wxYw#t?|dr4 |jjAd{7n#t $rYnwxYw|jCwxYww)zHandle an HTTP/2 connection.r)AsyncHTTP2ConnectionrerfNg?)timeoutzHTTP/2 receive error: %szError handling HTTP/2 requestrrjFzHTTP/2 connection error: %s_h2_conn)"gunicorn.http2.async_connectionr)r9r-r;StreamReaderProtocol StreamWriterr'r=r(initiate_connectionr+ is_closedrs receive_data TimeoutErrorrwr)rl_handle_http2_requestr_ send_errorstream stream_idrcleanup_streamrprrqrrrvr:r'rx) rr,r5r)rerfr-rgr.h2_connrequestsrbreqs r r?z%ASGIProtocol._handle_http2_connections0C $ L // ;H // ;H[[F33F;H))8VT[[-=-=F +&&(G--/ / /#DM''DKK,=,=%,%9%9#%9%FFH$EC E"88(H ..szz/C/CDE" #h-/;;>>T[[%=%==HHMM"PQ(-DKK%A''DKK,=,=NtZ(----///  ! ! #c 0 G++ HHNN#=qA %!**+JK!")"4"4 # 4 4c3q6# )! ! ! ..szz/C/CD%%   A HH  :NNNN> N*'N>)N**N>>Oc Kjj|j||}ddddgdgfd}f d}|j||}t j } |j j|j|j|||d{rg} D]Z\} } t| tr| jd} t| tr| jd} | j| | f\rdtfg} | D]/\} } | j| jt| f1j j#| d j$}|j#| d j'd{rIj j)d |j)d j'd{j+d{n:j-| d{nj/dd d{d t j | z }t7t9}|j2j;||||j j=|j||y777777#t0$r>|j2j5d sj/dd d{7dYwxYw#t0$r|j2j5d YywxYw# t j | z }t7t9}|j2j;||||j j=|j||w#t0$r|j2j5d YwwxYwxYww) zHandle a single HTTP/2 request.FNrrcLKjj}d|ddSw)NrFr)rread)rrzs r rz3ASGIProtocol._handle_http2_request..receives,<<$$&D&" s!$c K|d}|dk(r|jd}|jdg}g}|D]Z\}}t|tr|jd}t|tr|jd}|j ||f\ j ||d{y|dk(r(r t d yd|d|jdgy|d k(rMs t d  yr t d  y|jd d }|jdd}|r |z |sdyy|dk(rs t d y|jdg} g} | D]Z\}}t|tr|jd}t|tr|jd}| j ||f\j| yy7w)NrHrrrrrrTrrrrrrFhttp.response.trailersz)Cannot send trailers before body complete)rrrrrsend_informationalr`extend)rrrrrrrrrtrailer_headerstrailersrr8 response_bodyrrrrresponse_trailersr6s r rz0ASGIProtocol._handle_http2_request..sendsvH88%kk(3 &{{9b9 #/2KD%!$.#{{95!%/ % Y 7NND%=1 2 00KQQQ00##/0J#KL#' ")("3#*;;y"#= 11'#/0F#GL$#/0K#LL{{63/#KK U; !T)M (,%!55(#/0[#\L")++i"<#23KD%!$.#{{95!%/ % Y 7OOT5M2 3 "((269RsB%G (G)DG rz:status) end_streamrrr)r5r6_build_http2_scope_build_http2_environrrr(rr'r+rrrrrrr8 send_headersstreams_send_pending_data send_data send_trailers send_responser4rwr)r_rrrr)rrzr8rfrerrrrrrrr response_hdrsr5rrrrDrrrrrEr6s `` @@@@@@@@r r3z"ASGIProtocol._handle_http2_requestsNN,, ''8D !   ; 3; 3|++GXxH  < E HH g 6((5'40 0 0'"" #32KD%!$.#{{95!%/ % Y 7NND%=1 2%&/_1E%F$GM'.I e%,,djjlCJ-GHIOO00MV[0\$__Y7F'' %'H!44666%11)]W\1]((5(I%88:::"// ;LMMM!//!?G]((C9PQQQ"% E'||~ = '#%5s=7Igw E%%dkk7GTJq 147 ;NR & HH  : ;#((C9PQQQ"%  & E""#CD E E'||~ = '#%5s=7Igw E%%dkk7GTJ E""#CD EsA1O>>K'K'K'!K'#K'%K''MO>O;A3OO;$O85O;7O88O;;O>c g}|jD]D\}}|j|jjd|jdfFt |}t |}ddddd|j |j |j|jr|jjdnd|jr|jjdnd|jjxsd|||d } t|jd r|jj| d <i} t|d r|j|jd | d <i| d<| | d<| S)z*Build ASGI HTTP scope from HTTP/2 request.rrhrrr2rrrrrrrr?r)rrrrrrrrrr(rr:r'rrr) rrzrfrerrrrrrrs r rGzASGIProtocol._build_http2_scopevsQ"?? KD% NN ##I. Y'   %X.$X. %u=nnnnLL:A,, ++I6C?F}}GMM00;RU++1r   4;; (![[..E'N 7- .!11%994J/ 0 02 +,(l rc|j|j|j|jxsdd|r|dndd}|jD]\}}d|j ddz}|||<!|S)z.Build minimal environ dict for access logging.rzHTTP/2rrrrr)rrrrrrrs r rHz!ASGIProtocol._build_http2_environsx&nn{{ #MM/R'*28A;  #?? !KD%DLLc22C GCL !rN)F)rr r!r"rrBrFrUrQrcr@rmrnrorrrrrrrrrrxr?r3rGrHr#rr r%r%1s # M4( !$F &E$NQ* V8p4&P$$L585,' %.N E$NTEl)Vrr%)r"r;r[rgunicorn.asgi.unreaderrgunicorn.asgi.messagergunicorn.asgi.uwsgirgunicorn.http.errorsrgunicorn.uwsgi.errorsrrrProtocolr%r#rr rYsG  0.1+55 / /  7## r