AҐiPxdZddlZddlmZmZmZmZmZddlm Z ddl m Z da da dadadadZGdd Zd gZy) 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) HTTP2Requestc~ty ddlmaddlmaddlmaddl m a ddl m ay#t$r t!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 ImportErrorrX/var/www/descvideos/venv/lib/python3.12/site-packages/gunicorn/http2/async_connection.py _import_h2rs8 "#&&.* "!!"s(<ceZdZdZdZdZdZddZdZdZ d Z d Z d Z d Z d ZdZdZddZdZddZdZddZddZddZdZedZdZdZy)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"j%dd}t&j)||_d|_d|_d|_y)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    & &mmo$'#@#@ &)&F&F#!66$'$B$B!++#, ''v'6  !!rc K|jry|jjtjj |j tjj|jtjj|jtjj|ji|jj|jd{d|_y7 w)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     $$  % % < Y>Y  % % 9 94;S;S  % % 4 4d6I6I  % % : :D<KKK ct|tjr|j|St|tjr|j |St|tj r|j|St|tjr|j|yt|tjr yt|tjr|j|yt|tjr yt|tjr|j|yt|tj r|j#|Sy)zHandle a single h2 event. Args: event: h2 event object Returns: HTTP2Request if a request is complete, None otherwise N) isinstancerRequestReceived_handle_request_received DataReceived_handle_data_received StreamEnded_handle_stream_ended StreamReset_handle_stream_reset WindowUpdatedPriorityUpdated_handle_priority_updatedSettingsAcknowledgedConnectionTerminated_handle_connection_terminatedTrailersReceived_handle_trailers_receivedr;rcs rr^z"AsyncHTTP2Connection._handle_events. eZ77 8007 7 z66 7--e4 4 z55 6,,U3 3 z55 6  % %e ,"z77 8 z99 :  ) )% 0z>> ?  z>> ?  . .u 5 z:: ;11%8 8rc|j}|j}t||}||j|<|j |dy)z-Handle RequestReceived event (HEADERS frame).F end_streamN) stream_idheadersr r)receive_headers)r;rcr{r|streams rrhz-AsyncHTTP2Connection._handle_request_receivedsEOO --Y-"( Y w59rc|j}|j}|jj|}|y|j |dt |dkDrN |j jt |||j jt |dyy#ttjf$rEd|_ |j jtjYy#t$rYYywxYwwxYw)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_receivedsOO zz!!), >DU3 t9q=  ::3t9PY:Z ::3t9PT:U ? ?@  $ LL11^=^=^1_!  s+A B'' D*C44 D=DDDc|j}|jj|}|yd|_t ||j |j S)zHandle StreamEnded event.NT)r{r)rrequest_completer r%r(r;rcr{r~s rrlz)AsyncHTTP2Connection._handle_stream_endedsHOO !!), >"&FDHHd.>.>??rc|j}|jj|}||j|jyy)zHandle StreamReset event.N)r{r)rresetrKrs rrnz)AsyncHTTP2Connection._handle_stream_resets<OO !!),   LL)) * rcd|_y)z"Handle ConnectionTerminated event.TNr8rws rrtz2AsyncHTTP2Connection._handle_connection_terminated$s  rc|j}|jj|}|y|j|jt ||j |jS)zHandle TrailersReceived event.N)r{r)rreceive_trailersr|r r%r(rs rrvz.AsyncHTTP2Connection._handle_trailers_received(sROO !!), > .FDHHd.>.>??rc|jj|j}|3|j|j|j |j yy)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_updated3sQ !!%//2    " "|| ++// #  rcK|dks|dk\rtd||jj|}|td|ddt|fg}|D]/\}}|j |j t|f1|j j||d |jd{y7w) 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_informationalAs C<6S==fXFG G!!), >wyk<= ='F 45" @KD%  # #TZZ\3u:$> ? @ !!)-=%!P%%'''sB8C:C;Cc\K|jj|}|ydt|fg}|D]/\}}|j|j t|f1|duxst |dk(} |j j||| |j|| |jd{|r*t |dkDr|j||dd{y717#tj$r$|j|j|YywxYww)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_responseas+!!), >'F 45" @KD%  # #TZZ\3u:$> ? @T\3SY!^   LL % %i1Aj % Y    0Z  H))+ + +D A nnYnFFF ,G//  LLN    *  sOA5D,8AC2<C.=+C2(C0)C2-D,.C20C224D)&D,(D))D,cKd}t|D]}|jj|}|dkDr|cS tj|j j |jdd{}|r|jj|}|D]S}t|tjr|j|k(s-yt|tjsKd|_y|jd{n d|_y|jj|S77.#tj $rY:t"j$$rYywxYww)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_windowsK () A >>yII1}   !(!1!1KK$$T%:%:;"!\\66x@F!'&%eZ-C-CD$);')'z/N/NO+/DL#% &11333$(DL 4+ >||55i@@14 '' !//  s}6E#=D46D07AD4D4E# D4%D4-E#/D4D2 D4 E#0D42D44E E# E E#E  E#cK|jj|}|y|} |r|jj|}t ||j t |}|dkrW|jd{|j|d{}|dkryt ||j t |}|d|}||d}|xrt |dk(} |jj||| |jd{|r|j||y777#tjtjf$r$|j|j|YywxYww)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_datask!!), >   LLBB9M  D,?,?\ARS ?11333&*&H&H&S SI A~$!$Y0C0CSEV!WJ$[j1+JK8 %@#l*;q*@ &&y%H&M--///#&   Tj  94 S000.2Q2QR  LLN    * ss!E6AD,9D&:D,D( D,E6A-D, D* D,D,%E6&D,(D,*D,,AE30E62E33E6cK|jj|}|y|jsyg}|D]Q\}}|j}|j drt d|d|j |t|fS |jj||d|j||jd{y7#tj$r$|j|j|YywxYww)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_trailerss !!), >++# 8KD%JJLE$ ?4&8Q!RSS  " "E3u:#6 7  8  LL % %iT % R   1))+ + + ,//  LLN    *  s=BD AC C CD C4DD DD cK|r|jnd}dtt|fg}|r|jd|j ||||d{y7w)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_errorsU#*w~~$c#d)n56  NNH I  K$GGGsAA AA cK|jj|}||j||jj |||j d{y7w)zReset a stream with RST_STREAM.NrJ)r)rrr7 reset_streamrE)r;r{rKr~s rrz!AsyncHTTP2Connection.reset_streamsU!!),   LL $ !!) !C%%'''sAA)!A'"A)cK|jryd|_|1|jr#t|jjnd} |jj ||j d{ |jj|jjd{y7B#t$rYKwxYw7#t$rYywxYww)z,Close the connection gracefully with GOAWAY.NTrrJ) r8r)maxkeysr7rrErr'rS wait_closed)r;rKlast_stream_ids rrSzAsyncHTTP2Connection.closes <<   !9=S!2!2!451N  LL ) )Z ) @))+ + +  KK   ++))+ + + ,    ,   slAC /B?:B=;B?7C7C8C<C=B?? C C C  CC CCCCcK|jj}|r? |jj||jj d{yy7#t t f$r}d|_td|d}~wwxYww)z,Send any pending data from h2 to the socket.NTzSocket write error: ) r7rr'writedrainrOrPr8r)r;r`ras rrEz'AsyncHTTP2Connection._send_pending_data0s||((*  G !!$'kk''))) *W% G# *-A!+EFF Gs:B 8A!AA!B A!!B 0BB  B c|jS)zCheck if connection is closed.rrFs r is_closedzAsyncHTTP2Connection.is_closed;s||rc<|jj|dy)z-Remove a stream after processing is complete.N)r)pop)r;r{s rrz#AsyncHTTP2Connection.cleanup_stream@s D)rcNdt|jd|jdS)Nz)rr)r8rFs r__repr__zAsyncHTTP2Connection.__repr__Ds04<<()*ll^1 & r)N)F))rN)__name__ __module__ __qualname____doc__rNr<rDrQr^rhrjrlrnrtrvrqrrrrrrrrSrEpropertyrrrrrrr!r!0s)"V!*F"P$L :: @+ @ (@&P&AP)V*XH(, G* rr!)rr*errorsrrrrrr~r rdr r rrrrrr!__all__rrrrsX  !     ""Y Y x " "r