AҐi1W|dZddlmZddlmZmZmZmZmZddl m Z ddl m Z da dadadadadZGd d Zd gZy) zc HTTP/2 server connection implementation. Uses the hyper-h2 library for HTTP/2 protocol handling. )BytesIO) HTTP2ErrorHTTP2ProtocolErrorHTTP2ConnectionErrorHTTP2NotAvailableHTTP2ErrorCode) HTTP2Stream) HTTP2RequestNc~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 ImportErrorrR/var/www/descvideos/venv/lib/python3.12/site-packages/gunicorn/http2/connection.py _import_h2r s8 "#&&.* "!!"s(<ceZdZdZdZdZdZddZdZdZ d Z d Z d Z d Z d ZdZdZddZdZddZdZddZddZddZdZedZdZdZy)HTTP2ServerConnectionzHTTP/2 server-side connection handler. Manages the HTTP/2 connection state and multiplexed streams. This class wraps the h2 library and provides a higher-level interface for gunicorn workers. ict||_||_||_i|_g|_|j |_|j|_ |j|_ |j|_ tjdd}t j#||_t'|_d|_d|_y)a.Initialize an HTTP/2 server connection. Args: cfg: Gunicorn configuration object sock: SSL socket with completed handshake client_addr: Client address tuple (host, port) Raises: HTTP2NotAvailable: If h2 library is not installed Fzutf-8) client_sideheader_encoding)rN)r cfgsock client_addrstreams_pending_requestshttp2_initial_window_sizeinitial_window_sizehttp2_max_concurrent_streamsmax_concurrent_streamshttp2_max_frame_sizemax_frame_sizehttp2_max_header_list_sizemax_header_list_sizerH2Configurationr H2Connectionh2_connr _read_buffer_closed _initialized)selfr&r'r(rs r__init__zHTTP2ServerConnection.__init__:s   & "$$'#@#@ &)&F&F#!66$'$B$B!++#, ''v'6 $I !rc |jry|jjtjj |j tjj|jtjj|jtjj|ji|jj|jd|_y)zSend initial HTTP/2 settings to client. Should be called after the SSL handshake completes and before processing any data. NT)r8r5update_settingsr SettingCodesMAX_CONCURRENT_STREAMSr.INITIAL_WINDOW_SIZEr,MAX_FRAME_SIZEr0MAX_HEADER_LIST_SIZEr2initiate_connection_send_pending_datar9s rrBz)HTTP2ServerConnection.initiate_connectiones     $$  % % < Y>Y  % % 9 94;S;S  % % 4 4d6I6I  % % : :D> ?  z>> ?  . .u 5 z:: ;11%8 8rc|j}|j}t||}||j|<|j |dy)z{Handle RequestReceived event (HEADERS frame). Args: event: RequestReceived event with headers F end_streamN) stream_idheadersr r)receive_headers)r9r^rvrwstreams rrcz.HTTP2ServerConnection._handle_request_receivedsG OO --Y-"( Y w59rc|j}|j}|jj|}|y|j |dt |dkDr^ |j jt |||j jt |d|jyy#ttjf$r#|jtjYywxYw)zHandle DataReceived event. Args: event: DataReceived event with body data Returns: None (request completion handled by StreamEnded) NFrtr)rvrF)rvr[r)getrLlenr5increment_flow_control_windowrC ValueErrorrrMrNr rO)r9r^rvr[rys rrez+HTTP2ServerConnection._handle_data_receivedsOO zz!!), >DU3 t9q= I ::3t9PY:Z ::3t9PT:U'')   ? ?@ I n&G&G H  IsAB779C32C3c|j}|jj|}|yd|_t ||j |j S)zHandle StreamEnded event. Args: event: StreamEnded event Returns: HTTP2Request for the completed request NT)rvr)r{request_completer r&r(r9r^rvrys rrgz*HTTP2ServerConnection._handle_stream_endedsNOO !!), >#'FDHHd.>.>??rc|j}|jj|}||j|jyy)ziHandle StreamReset event (RST_STREAM frame). Args: event: StreamReset event N)rvr)r{resetrGrs rriz*HTTP2ServerConnection._handle_stream_reset*s> OO !!),   LL)) * rcd|_y)zwHandle ConnectionTerminated event (GOAWAY frame). Args: event: ConnectionTerminated event TNr7rrs rroz3HTTP2ServerConnection._handle_connection_terminated7s  rc|j}|jj|}|y|j|jt ||j |jS)zHandle TrailersReceived event. Args: event: TrailersReceived event with trailer headers Returns: HTTP2Request if this completes the request N)rvr)r{receive_trailersrwr r&r(rs rrqz/HTTP2ServerConnection._handle_trailers_received@sVOO !!), > .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)r{rvupdate_priorityrrr)r9r^rys rrlz.HTTP2ServerConnection._handle_priority_updatedTsQ !!%//2    " "|| ++// #  rcl|dks|dk\rtd||jj|}|td|ddt|fg}|D]/\}}|j |j t|f1|j j||d |jy) 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:statusFrt) rr)r{rPrZlowerr5 send_headersrC)r9rvstatusrwryresponse_headersnamevalues rsend_informationalz(HTTP2ServerConnection.send_informationalbs C<6S==fXFG G!!), >wyk<= ='F 45" @KD%  # #TZZ\3u:$> ? @ !!)-=%!P !rc,|jj|}|ydt|fg}|D]/\}}|j|j t|f1|duxst |dk(} |j j||| |j|| |j|r"t |dkDr|j||dy#tj$r$|j|j|YywxYw)aSend 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 Raises: HTTP2Error: If stream not found or in invalid state Returns: bool: True if response sent, False if stream was already closed NFrrrtT)r)r{rPrZrr|r5rrC send_datarStreamClosedErrorrNcleanup_stream) r9rvrrwbodyryrrrrus r send_responsez#HTTP2ServerConnection.send_responses!!), >'F 45" @KD%  # #TZZ\3u:$> ? @T\3SY!^   LL % %i1Aj % Y    0Z  H  # # %D A y$4@//  LLN    *  s6A%C4DDcddl}d} |j}|j|j|jd} t|D]}|jj|}|dkDr|}n|jd}|s< |jj|j} | s d|_n |jj%| } | D]U} t'| t(j*r| j,|k(s-d}n8t'| t(j.sLd|_d}n|j1n|jj|}|j3|S#t t f$rYywxYw#tttj f$rYHwxYw#tj $rYbwxYw#|j3wxYw)zWait for flow control window to become positive. Returns: int: Available window size, or -1 if waiting failed rN2g?)timeoutT) selectorsDefaultSelectorregisterr' EVENT_READ TypeErrorr~ranger5local_flow_control_windowselectrHrIrJrKrrXr7rLrarrhrvrnrCrN) r9rvrmax_wait_attemptsselresult_ availablereadyincomingrr^s r_wait_for_flow_control_windowz3HTTP2ServerConnection._wait_for_flow_control_windows  ++-C LLI$8$8 9 % ,-" K LLBB9M q=&F 3 /#'99>>$2G2G#H$'+ !%!:!:8!D"( !%eZ-C-CD$);)+ %'z/N/NO+/DL%'F! !//1 ?" KD?? J IIK Y:&   $Wn.J.JK*77& IIKsr6E0AG %F. G:F*.GG#;G0FFF'#G&F''G*G=GGGGc|jj|}|y|} |r|jj|}t ||j t |}|dkrG|j|j|}|dkryt ||j t |}|d|}||d}|xrt |dk(} |jj||| |j|r|j||y#tjtjf$r$|j|j|YywxYw)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 NFrrtT)r)r{r5rminr0r|rCrrrrrMrNr) r9rvr[rury data_to_sendr chunk_sizechunkis_finals rrzHTTP2ServerConnection.send_datasH!!), >   LLBB9M  D,?,?\ARS ?++- $ B B9 MI A~$!$Y0C0CSEV!WJ$[j1+JK8 %@#l*;q*@ &&y%H&M'')#&   Tj  900.2Q2QR  LLN    *  s A(D A,D 8D AEEc|jj|}|y|jsyg}|D]Q\}}|j}|j drt d|d|j |t|fS |jj||d|j||jy#tj$r$|j|j|YywxYw)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 F:zPseudo-header 'z' not allowed in trailersTrt)r)r{response_headers_sentr startswithrrZrPr5r send_trailersrCrrrNr)r9rvtrailersrytrailer_headersrrlnames rrz#HTTP2ServerConnection.send_trailerss !!), >++# 8KD%JJLE$ ?4&8Q!RSS  " "E3u:#6 7  8  LL % %iT % R   1  # # %//  LLN    *  s?C4C<;C<c|r|jnd}dtt|fg}|r|jd|j ||||y)zSend an error response on a stream. Args: stream_id: The stream ID status_code: HTTP status code message: Optional error message body rzcontent-length)z content-typeztext/plain; charset=utf-8N)encoderPr|rZr)r9rv status_codemessagerrws r send_errorz HTTP2ServerConnection.send_error@sM$+w~~$c#d)n56  NNH I 9k7DArc|jj|}||j||jj |||j y)zReset a stream with RST_STREAM. Args: stream_id: The stream ID to reset error_code: HTTP/2 error code (default: CANCEL) NrF)r)r{rr5 reset_streamrC)r9rvrGrys rrz"HTTP2ServerConnection.reset_streamOsM!!),   LL $ !!) !C !rc |jryd|_|1|jr#t|jjnd} |jj ||j y#t$rYywxYw)zClose the connection gracefully with GOAWAY. Args: error_code: HTTP/2 error code (default: NO_ERROR) last_stream_id: Last processed stream ID (default: highest) NTrrF)r7r)maxkeysr5close_connectionrC Exception)r9rGlast_stream_ids rrNzHTTP2ServerConnection.close]sr <<   !9=S!2!2!451N  LL ) )Z ) @  # # %   s ,A66 BBc|jj}|r |jj|yy#tt f$r}d|_td|d}~wwxYw)z,Send any pending data from h2 to the socket.TzSocket write error: N)r5rr'sendallrJrKr7r)r9r[r\s rrCz(HTTP2ServerConnection._send_pending_datassg||((*  G !!$' W% G# *-A!+EFF Gs;A$ AA$c|jS)zCheck if connection is closed.rrDs r is_closedzHTTP2ServerConnection.is_closed}s||rc<|jj|dy)zvRemove a stream after processing is complete. Args: stream_id: The stream ID to clean up N)r)pop)r9rvs rrz$HTTP2ServerConnection.cleanup_streams D)rcNdt|jd|jdS)Nz)r|r)r7rDs r__repr__zHTTP2ServerConnection.__repr__s04<<()*ll^1 & r)N)F))rN)__name__ __module__ __qualname____doc__rIr:rBrLrYrcrergrirorqrlrrrrrrrrNrCpropertyrrrrrrr"r"/s)"V!*@"D$L:" D@* +@( "@*X8t*X*X B ",G* rr")riorerrorsrrrrr ryr r_r r rrrrr r"__all__rrrrsX   !     ""`  `  F # #r