jDXdZddlmZddlmZmZmZmZmZddl m Z ddl m Z da dadadadadZGd d Zd gZdS) zc HTTP/2 server connection implementation. Uses the hyper-h2 library for HTTP/2 protocol handling. )BytesIO) HTTP2ErrorHTTP2ProtocolErrorHTTP2ConnectionErrorHTTP2NotAvailableHTTP2ErrorCode) HTTP2Stream) HTTP2RequestNctdS 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 ImportErrorrf/var/www/finuniver-perm.ru/html/student/venv/lib/python3.11/site-packages/gunicorn/http2/connection.py _import_h2r s "######&&&&&&&&&&&&......******** """!!!"s +AceZdZdZdZdZdZddZdZdZ d Z d Z d Z d Z d ZdZdZddZdZddZdZddZd dZd!dZdZedZdZdZdS)"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 |_ tdd}t ||_t'|_d|_d|_dS)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#!6$'$B!++#,  ''v'66 $II !rc d|jrdS|jtjj|jtjj|jtjj |j tjj |j i|j |d|_dS)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    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). Args: event: RequestReceived event with headers F end_streamN) stream_idheadersr r)receive_headers)r9r^rvrwstreams rrcz.HTTP2ServerConnection._handle_request_receivedsP O -Y--"( Y w599999rc|j}|j}|j|}|dS||dt |dkr |jt |||jt |d|n<#ttj f$r#| tjYnwxYwdS)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_receivedsO z!!),, >4DU333 t99q== I ::3t99PY:ZZZ ::3t99PT:UUU'')))) ?@ I I I n&G HHHHH ItsA&B??6C87C8c|j}|j|}|dSd|_d|_|jr|jt||j|j S)zHandle StreamEnded event. Args: event: StreamEnded event Returns: HTTP2Request for the completed request NT) rvr)r{request_complete_body_complete _body_eventsetr r&r(r9r^rvrys rrgz*HTTP2ServerConnection._handle_stream_endedsuO !!),, >4 #' $   %   " " $ $ $FDHd.>???rc|j}|j|}|||jdSdS)ziHandle StreamReset event (RST_STREAM frame). Args: event: StreamReset event N)rvr)r{resetrGrs rriz*HTTP2ServerConnection._handle_stream_reset/sI O !!),,   LL) * * * * *  rcd|_dS)zwHandle ConnectionTerminated event (GOAWAY frame). Args: event: ConnectionTerminated event TNr7rrs rroz3HTTP2ServerConnection._handle_connection_terminated<s  rc|j}|j|}|dS||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_receivedEsYO !!),, >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)r{rvupdate_priorityrrr)r9r^rys rrlz.HTTP2ServerConnection._handle_priority_updatedYs` !!%/22    " "| +/ #       rc|dks|dkrtd||j|}|td|ddt|fg}|D];\}}||t|f<|j||d |dS) 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_informationalgs C<<6S==FfFFGG G!!),, > ? ? ? ? !!)-=%!PPP !!!!!rcz|j|}|dSdt|fg}|D];\}}||t|f<|dupt |dk} |j||| ||| ||r+t |dkr| ||ddS#tj $r-| | |YdSwxYw)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[!!),, >5'F 45" @ @KD%  # #TZZ\\3u::$> ? ? ? ?T\3SYY!^   L % %i1Aj % Y Y Y    0Z  H H H  # # % % % AD A y$4@@@4/    LLNNN    * * *55  sA5C>>8D:9D:cddl}d} |}||j|jn#t t f$rYdSwxYwd} t|D]-}|j |}|dkr|}n!| d}|r |j |j } n##tttjf$rYnwxYw| s d|_n |j| } n#tj$rYnwxYw| D]Q} t'| t(jr| j|krd}n=,t'| t(jr d|_d}nR|+n/|j |}|n#|wxYw|S)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_windowsA  ++--C LLI$8 9 9 9 9:&   22 % ,--" K" K LBB9MM q==&FE 3 //#'9>>$2G#H#H#Wn.JK#'+ !%!:!:8!D!D)7!' ! !%eZ-CDD"$);;)+ % <(z/NOO"+/DL%'F!E" //111 E16?? JJ IIKKKKCIIKKKK se4=AAAG(CGC($G'C((G7DGD%!G$D%%B GGFc|j|}|dS|} |r|j|}t ||jt |}|dkrT|||}|dkrdSt ||jt |}|d|}||d}|ot |dk} |j ||| ||| ||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 NFrrtT)r)r{r5rminr0r|rCrrrrrMrNr) r9rvr[rury data_to_sendr chunk_sizechunkis_finals rrzHTTP2ServerConnection.send_datas!!),, >5   * LBB9MM  D,?\ARARSS ??++--- $ B B9 M MI A~~$u!$Y0CSEVEV!W!WJ$[j[1+JKK8 %@#l*;*;q*@ &&y%H&MMM'')))# *&   Tj  9 9 940.2QR    LLNNN    * * *55  sA4D0BD00AE87E8c4|j|}|dS|jsdSg}|D]e\}}|}|drt d|d||t|ff |j ||d| || dS#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 trailersTrt)r)r{response_headers_sentr startswithrrZrPr5r send_trailersrCrrrNr)r9rvtrailersrytrailer_headersrrlnames rrz#HTTP2ServerConnection.send_trailerss@ !!),, >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  sAC8DDc|r|nd}dtt|fg}|r|d|||||dS)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_errorEsr$+3w~~$c#d))nn56  J NNH I I I 9k7DAAAAArc|j|}||||j|||dS)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_streamTsd!!),,   LL $ $ $ !!) !CCC !!!!!rrc |jrdSd|_|/|jr&t|jnd} |j||dS#t$rYdSwxYw)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.closebs <  F  !9=LS!2!2!4!45551N  L ) )Z ) @ @ @  # # % % % % %    DD s/A44 BBc|j}|rO |j|dS#tt f$r}d|_td|d}~wwxYwdS)z,Send any pending data from h2 to the socket.TzSocket write error: N)r5rr'sendallrJrKr7r)r9r[r\s rrCz(HTTP2ServerConnection._send_pending_dataxs|((**  G G !!$'''''W% G G G# *+E!+E+EFFF G G Gs9A( A##A(c|jS)zCheck if connection is closed.rrDs r is_closedzHTTP2ServerConnection.is_closeds |rc<|j|ddS)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)))))rcBdt|jd|jdS)Nz)r|r)r7rDs r__repr__zHTTP2ServerConnection.__repr__s7 &4<(( & &l & & & r)N)F)r)rN)__name__ __module__ __qualname____doc__rIr:rBrLrYrcrergrirorqrlrrrrrrrrNrCpropertyrrrrrrr"r"/s)")")"V!!!*@"@"@"@"D$$$L:::"   D@@@4 + + +@@@(   """@****X888t****X***X B B B B " " " ",GGGX***     rr")riorerrorsrrrrr ryr r_r r rrrrr r"__all__rrrrs   !!!!!!     """"e  e  e  e  e  e  e  e  P # #r