jQdZddlZddlmZmZmZmZmZddlm Z ddl m Z da da dadadadZGdd Zd gZdS) z Async HTTP/2 server connection implementation for ASGI workers. Uses the hyper-h2 library for HTTP/2 protocol handling with asyncio for non-blocking I/O. N) HTTP2ErrorHTTP2ProtocolErrorHTTP2ConnectionErrorHTTP2NotAvailableHTTP2ErrorCode) HTTP2Stream) HTTP2RequestctdS ddlmaddlmaddlmaddl m a ddl m adS#t$rt!wxYw)z$Lazily import h2 library components.Nr)_h2 h2.connection connection h2.configconfig _h2_config h2.eventsevents _h2_events h2.exceptions exceptions_h2_exceptions h2.settingssettings _h2_settings ImportErrorrR/var/www/html/venv/lib/python3.11/site-packages/gunicorn/http2/async_connection.py _import_h2rs "######&&&&&&&&&&&&......******** """!!!"s +AceZdZdZdZdZdZddZdZdZ d Z d Z d Z d Z d ZdZdZddZdZddZdZddZd dZd!dZdZedZdZdZdS)"AsyncHTTP2ConnectionzAsync HTTP/2 server-side connection handler for ASGI. Manages the HTTP/2 connection state and multiplexed streams using asyncio for non-blocking I/O operations. ict||_||_||_||_i|_t j|_|j |_ |j |_ |j |_|j|_t"dd}t&||_d|_d|_d|_dS)aPInitialize an async HTTP/2 server connection. Args: cfg: Gunicorn configuration object reader: asyncio StreamReader writer: asyncio StreamWriter client_addr: Client address tuple (host, port) Raises: HTTP2NotAvailable: If h2 library is not installed Fzutf-8) client_sideheader_encoding)rN)rcfgreaderwriter client_addrstreamsasyncioQueue_request_queuehttp2_initial_window_sizeinitial_window_sizehttp2_max_concurrent_streamsmax_concurrent_streamshttp2_max_frame_sizemax_frame_sizehttp2_max_header_list_sizemax_header_list_sizerH2Configurationr H2Connectionh2_conn_closed _initialized _receive_task)selfr%r&r'r(rs r__init__zAsyncHTTP2Connection.__init__:s    & &moo$'#@ &)&F#!6$'$B!++#,  ''v'66  !!rc tK|jrdS|jtjj|jtjj|jtjj |j tjj |j i|j |d{Vd|_dS)zSend initial HTTP/2 settings to client. Should be called after the SSL handshake completes and before processing any data. NT)r9r7update_settingsr SettingCodesMAX_CONCURRENT_STREAMSr0INITIAL_WINDOW_SIZEr.MAX_FRAME_SIZEr2MAX_HEADER_LIST_SIZEr4initiate_connection_send_pending_datar;s rrDz(AsyncHTTP2Connection.initiate_connectiones    F $$  % Y  % 94;S  % 4d6I  % :D ? ? 9  z> ? ? 9  . .u 5 5 5 5 z: ; ; 911%88 8trc|j}|j}t||}||j|<||ddS)z-Handle RequestReceived event (HEADERS frame).F end_streamN) stream_idheadersr r)receive_headers)r;rcr{r|streams rrhz-AsyncHTTP2Connection._handle_request_receivedsNO -Y--"( Y w599999rc|j}|j}|j|}|dS||dt |dkr |jt |||jt |dnZ#ttj f$rAd|_ |j tjn#t$rYnwxYwYnwxYwdS)zHandle DataReceived event.NFryr)r{TrJ)r{r`r)getrQlenr7increment_flow_control_window ValueErrorrrRr8close_connectionrrT Exception)r;rcr{r`r~s rrjz*AsyncHTTP2Connection._handle_data_receivedsO z!!),, >4DU333 t99q==  ::3t99PY:ZZZ ::3t99PT:UUUU ?@    $ L11^=^1____ D  ts7AB++D %C/.D/ C<9D;C<<DDc|j}|j|}|dSd|_d|_|jr|jt||j|j S)zHandle StreamEnded event.NT) r{r)rrequest_complete_body_complete _body_eventsetr r%r(r;rcr{r~s rrlz)AsyncHTTP2Connection._handle_stream_endedsqO !!),, >4 #' $   %   " " $ $ $FDHd.>???rc|j}|j|}|||jdSdS)zHandle StreamReset event.N)r{r)rresetrKrs rrnz)AsyncHTTP2Connection._handle_stream_reset#sGO !!),,   LL) * * * * *  rcd|_dS)z"Handle ConnectionTerminated event.TNr8rws rrtz2AsyncHTTP2Connection._handle_connection_terminated+s  rc|j}|j|}|dS||jt ||j|jS)zHandle TrailersReceived event.N)r{r)rreceive_trailersr|r r%r(rs rrvz.AsyncHTTP2Connection._handle_trailers_received/sUO !!),, >4 ...FDHd.>???rc|j|j}|)||j|j|jdSdS)zHandle PriorityUpdated event (PRIORITY frame). Args: event: PriorityUpdated event with priority info N)weight depends_on exclusive)r)rr{update_priorityrrr)r;rcr~s rrqz-AsyncHTTP2Connection._handle_priority_updated:s` !!%/22    " "| +/ #       rcK|dks|dkrtd||j|}|td|ddt|fg}|D];\}}||t|f<|j||d |d{VdS) aSend an informational response (1xx) on a stream. This is used for 103 Early Hints and other 1xx responses. Informational responses are sent before the final response and do not end the stream. Args: stream_id: The stream ID status: HTTP status code (100-199) headers: List of (name, value) header tuples Raises: HTTP2Error: If status is not in 1xx range dzInvalid informational status: NzStream z not found:statusFry) rr)rrUr_lowerr7 send_headersrE)r;r{statusr|r~response_headersnamevalues rsend_informationalz'AsyncHTTP2Connection.send_informationalHs C<<6S==FfFFGG G!!),, > ? ? ? ? !!)-=%!PPP%%'''''''''''rcK|j|}|dSdt|fg}|D];\}}||t|f<|dupt |dk} |j||| ||| |d{V|r1t |dkr| ||dd{VdS#tj $r-| | |YdSwxYw)aXSend a response on a stream. Args: stream_id: The stream ID to respond on status: HTTP status code (int) headers: List of (name, value) header tuples body: Optional response body bytes Returns: bool: True if response sent, False if stream was already closed NFrrryT)r)rrUr_rrr7rrE send_datarStreamClosedErrorrScleanup_stream) r;r{rr|bodyr~rrrrzs r send_responsez"AsyncHTTP2Connection.send_responsehs!!),, >5'F 45" @ @KD%  # #TZZ\\3u::$> ? ? ? ?T\3SYY!^   L % %i1Aj % Y Y Y    0Z  H H H))++ + + + + + + + GD A nnYnFFFFFFFFF4/    LLNNN    * * *55  s BD 8EEcKd}t|D]}|j|}|dkr|cS tj|j|jdd{V}|r|j|}|D]Q}t|tj r|j |krdS,t|tj r d|_dSR|d{Vn d|_dS#tj$rY t"j$rYdSwxYw|j|S)zWait for flow control window to become positive. Returns: int: Available window size, or -1 if waiting failed 2rg?rHNT)ranger7local_flow_control_windowr*rLr&rMrNrQrfrrmr{rsr8rE TimeoutErrorrr])r;r{max_wait_attempts_ availableincomingrrcs r_wait_for_flow_control_windowz2AsyncHTTP2Connection._wait_for_flow_control_windows ())  A >>yIII1}}     !(!1K$$T%:;;"""!\66x@@F!'&&%eZ-CDD&$);;')rrr <'z/NOO&+/DL#%222&113333333333$(DL22 4 '   !/   rrr |55i@@@s$A?D >#D $#D  D0D0/D0Fc"K|j|}|dS|} |r|j|}t ||jt |}|dkr`|d{V||d{V}|dkrdSt ||jt |}|d|}||d}|ot |dk} |j ||| |d{V|| ||dS#tj tj f$r-| ||YdSwxYw)aSend data on a stream. Args: stream_id: The stream ID data: Body data bytes end_stream: Whether this ends the stream Returns: bool: True if data sent, False if stream was already closed NFrryT)r)rr7rminr2rrErrrrrRrSr) r;r{r`rzr~ data_to_sendr chunk_sizechunkis_finals rrzAsyncHTTP2Connection.send_datas!!),, >5   0 LBB9MM  D,?\ARARSS ??11333333333&*&H&H&S&S S S S S S SI A~~$u!$Y0CSEVEV!W!WJ$[j[1+JKK8 %@#l*;*;q*@ &&y%H&MMM--/////////# 0&   Tj  9 9 940.2QR    LLNNN    * * *55 sBE'BEAF FcDK|j|}|dS|jsdSg}|D]e\}}|}|drt d|d||t|ff |j ||d| || d{VdS#tj $r-|||YdSwxYw)aSend trailing headers on a stream. Trailers are headers sent after the response body, commonly used for gRPC status codes, checksums, and timing information. Args: stream_id: The stream ID trailers: List of (name, value) trailer tuples Raises: HTTP2Error: If stream not found, headers not sent, or pseudo-headers used Returns: bool: True if trailers sent, False if stream was already closed NF:zPseudo-header 'z' not allowed in trailersTry)r)rresponse_headers_sentr startswithrr_rUr7r send_trailersrErrrSr)r;r{trailersr~trailer_headersrrlnames rrz"AsyncHTTP2Connection.send_trailerssV !!),, >5+ 5# 8 8KD%JJLLE$$ T !R4!R!R!RSSS  " "E3u::#6 7 7 7 7  L % %iT % R R R   1 1 1))++ + + + + + + +4/    LLNNN    * * *55  sA C##8DDcK|r|nd}dtt|fg}|r|d|||||d{VdS)z#Send an error response on a stream.rzcontent-length)z content-typeztext/plain; charset=utf-8N)encoderUrr_r)r;r{ status_codemessagerr|s r send_errorzAsyncHTTP2Connection.send_errors#*3w~~$c#d))nn56  J NNH I I I  K$GGGGGGGGGGGrcK|j|}||||j|||d{VdS)zReset a stream with RST_STREAM.NrJ)r)rrr7 reset_streamrE)r;r{rKr~s rrz!AsyncHTTP2Connection.reset_streamsx!!),,   LL $ $ $ !!) !CCC%%'''''''''''rrcK|jrdSd|_|/|jr&t|jnd} |j||d{Vn#t$rYnwxYw |j |j d{VdS#t$rYdSwxYw)z,Close the connection gracefully with GOAWAY.NTrrJ) r8r)maxkeysr7rrErr'rS wait_closed)r;rKlast_stream_ids rrSzAsyncHTTP2Connection.close!s <  F  !9=LS!2!2!4!45551N  L ) )Z ) @ @ @))++ + + + + + + + +    D   K     +))++ + + + + + + + + +    DD s$5A;; BB 8C CCcK|j}|rn |j||jd{VdS#t t f$r}d|_td|d}~wwxYwdS)z,Send any pending data from h2 to the socket.NTzSocket write error: ) r7rr'writedrainrOrPr8r)r;r`ras rrEz'AsyncHTTP2Connection._send_pending_data7s|((**  G G !!$'''k'')))))))))))W% G G G# *+E!+E+EFFF G  G Gs9AB +BB c|jS)zCheck if connection is closed.rrFs r is_closedzAsyncHTTP2Connection.is_closedBs |rc<|j|ddS)z-Remove a stream after processing is complete.N)r)pop)r;r{s rrz#AsyncHTTP2Connection.cleanup_streamGs  D)))))rcBdt|jd|jdS)Nz)rr)r8rFs r__repr__zAsyncHTTP2Connection.__repr__Ks7 &4<(( & &l & & & r)N)F)r)rN)__name__ __module__ __qualname____doc__rNr<rDrQr^rhrjrlrnrtrvrqrrrrrrrrSrEpropertyrrrrrrr!r!0s)")")"V!!!*F"F"F"F"P$$$L : : ::@@@$+++ @ @ @   (((@&&&&P&A&A&AP))))V***XHHHH((((, G G GX***     rr!)rr*errorsrrrrrr~r rdr r rrrrrr!__all__rrrrs  !!!!!!     """"` ` ` ` ` ` ` ` F " "r