ڊjKkdZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZdadadadadadZGd d Zd gZy) zc HTTP/2 server connection implementation. Uses the hyper-h2 library for HTTP/2 protocol handling. N)BytesIO) 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 ImportErrorr^/var/www/io.vulcan-creative.com/venv/lib/python3.12/site-packages/gunicorn/http2/connection.py _import_h2r s8 "#&&.* "!!"s(<ceZdZdZdZdZdZddZddZdZ d Z d Z d Z d Z d ZdZdZdZ ddZddZddZdZddZdZddZd dZd!dZdZedZdZdZy)"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|_t j |_g|_|j|_ |j|_ |j|_|j|_t"j%dd}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 collectionsdeque_deferred_events_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   & !, 1 1 3"$$'#@#@ &)&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)r;r8update_settingsr SettingCodesMAX_CONCURRENT_STREAMSr1INITIAL_WINDOW_SIZEr/MAX_FRAME_SIZEr3MAX_HEADER_LIST_SIZEr5initiate_connection_send_pending_datar<s rrEz)HTTP2ServerConnection.initiate_connectionks     $$  % % < Y>Y  % % 9 94;S;S  % % 4 4d6I6I  % % : :DY>Y  % % 9 94;S;S  % % 4 4d6I6I  % % : :D.>??Os Hc*|& |jj|j}|s d|_gS |jj|}g}|j0r;t3|j0t3|z}|j0j5|D]'}|j7|}||j9|)|j;|S#ttf$r}t d|d}~wwxYw#tj$r9}|jtjtt|d}~wtj $r9}|jtj"tt|d}~wtj$$r[}t'|dd}||j|n |jtj(tt|d}~wtj*$r9}|jtj,tt|d}~wtj.$r9}|jtj(tt|d}~wwxYw)aAProcess received data and return completed requests. Args: data: Optional bytes to process. If None, reads from socket. Returns: list: List of HTTP2Request objects for completed requests Raises: HTTP2ConnectionError: On protocol or connection errors NzSocket read error: T error_coderf)r'recvREAD_BUFFER_SIZEOSErrorIOErrorrr:r8r[rFlowControlErrorcloser FLOW_CONTROL_ERRORrstrFrameTooLargeErrorFRAME_SIZE_ERRORInvalidSettingsValueErrorgetattrPROTOCOL_ERRORTooManyStreamsErrorREFUSED_STREAM ProtocolErrorr,listclear _handle_eventrVrF)r<dataerrfcompleted_requestseventrequests rr[z"HTTP2ServerConnection.receive_datas' < Fyy~~d&;&;<DLI -\\..t4F<   $//04<?F  ! ! ' ' ) 3E((/G""))'2 3 !!!kW% F*-@+DEE F.. - JJ."C"CJ D$SV, ,00 - JJ."A"AJ B$SV, ,77 -!L$7J% j 1 n&C&C D$SV, ,11 - JJ."?"?J @$SV, ,++ - JJ."?"?J @$SV, , -s_%CC<C9&C44C9<J4EJ4F  J#AG99J4IJ4J  Jct|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<r}s rryz#HTTP2ServerConnection._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). Args: event: RequestReceived event with headers FrON)rJrRr r)rZ)r<r}rJrRr]s rrz.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) NFrOr)rJre)rJrzr)getr[lenr8increment_flow_control_windowrF ValueErrorrrkrlr rm)r<r}rJrzr]s rrz+HTTP2ServerConnection._handle_data_received/sOO zz!!), >DU3 t9q= I ::3t9PY:Z ::3t9PT:U'')   ? ?@ I n&G&G H  IsAB779C32C3c|j}|jj|}|yd|_d|_|j r|j j t||j|jS)zHandle StreamEnded event. Args: event: StreamEnded event Returns: HTTP2Request for the completed request NT) rJr)rrequest_complete_body_complete _body_eventsetr r&r(r<r}rJr]s rrz*HTTP2ServerConnection._handle_stream_endedQsqOO !!), > #' $       " " $FDHHd.>.>??rc|j}|jj|}||j|jyy)ziHandle StreamReset event (RST_STREAM frame). Args: event: StreamReset event N)rJr)rresetrfrs rrz*HTTP2ServerConnection._handle_stream_resetks> OO !!),   LL)) * rcd|_y)zwHandle ConnectionTerminated event (GOAWAY frame). Args: event: ConnectionTerminated event TNr:rs rrz3HTTP2ServerConnection._handle_connection_terminatedxs  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)rJr)rreceive_trailersrRr r&r(rs rrz/HTTP2ServerConnection._handle_trailers_receivedsVOO !!), > .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)rrJupdate_priorityrrr)r<r}r]s rrz.HTTP2ServerConnection._handle_priority_updatedsQ !!%//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:statusFrO) rr)rrnrVrWr8 send_headersrF)r<rJstatusrRr]response_headersr_r`s rsend_informationalz(HTTP2ServerConnection.send_informationals C<6S==fXFG G!!), >wyk<= ='F 45" @KD%  # #TZZ\3u:$> ? @ !!)-=%!P !rcF|jj|}|ydt|fg}|D]/\}}|j|j t|f1|j j ||||j |||jy)aSend response headers on a stream without ending it. Returns False if the stream is already gone. Split out of send_response() so a response can be streamed: headers first, then any number of data frames, then end_stream(). FrrOT)r)rrnrVrWr8rrF) r<rJrrRrPr]rr_r`s rsend_response_headersz+HTTP2ServerConnection.send_response_headerss!!), >'F 45" @KD%  # #TZZ\3u:$> ? @ !!)-=-7 " 9,D !rcX|jj|y|r|j||y |jj |dd|j|j dd|j y#t j$r|j|YywxYw)z;Close the sending half of a stream, with trailers if given.FTrrO) r)r send_trailersr8 send_datarFrStreamClosedErrorcleanup_stream)r<rJtrailerss rrPz HTTP2ServerConnection.end_streams <<  I & .    y( 3  LL " "9cd " C LL # - -cd - C  # # %//     * sAB$B)(B)cP|duxst|dk(} |j||||sy|r"t|dkDr|j||dy#tj$rA|j j |}||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 NrrOFT) rrrrrr)rrlr)r<rJrrRrXrPr]s r send_responsez#HTTP2ServerConnection.send_responsesT\3SY!^  --i9C.ED A y$4@// \\%%i0F!     *  sA$AAB%$B%cd} tj}|j|jtjd} t|D]}|jj|}|dkDr|}n|jd}|s> |jj|j}|s d|_n |jj%|} | D]p} t'| t(j*r| j,|k(s-d}nTt'| t(j.r d|_d}n/|j0j3| r|j5n|jj|}|j7|S#t t f$rYywxYw#tttj f$rYHwxYw#tj $rYbwxYw#|j7wxYw)zWait for flow control window to become positive. Returns: int: Available window size, or -1 if waiting failed 2rg?)timeoutT) selectorsDefaultSelectorregisterr' EVENT_READ TypeErrorrranger8local_flow_control_windowselectrgrhrirjrrvr:r[rrrrJrr,rVrFrl) r<rJmax_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! !1188?! //1 K( KP?? J IIK e:&   $Wn.J.JK*772 IIKsm>FAG&%F'4 G&G .G& A6G&F$#F$'G G&G  G& G#G&"G##G&&G8c|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 NFrrOT)r)rr8rminr3rrFrrrrrkrlr) r<rJrzrPr] data_to_sendr chunk_sizechunkis_finals rrzHTTP2ServerConnection.send_dataSsH!!), >   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 trailersTrO)r)rresponse_headers_sentrW startswithrrVrnr8rrrFrrrlr)r<rJrr]trailer_headersr_r`lnames 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)encodernrrVr)r<rJ status_codemessagerXrRs r send_errorz HTTP2ServerConnection.send_errorsM$+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) Nre)r)rrr8 reset_streamrF)r<rJrfr]s rrz"HTTP2ServerConnection.reset_streamsM!!),   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) NTrre)r:r)maxkeysr8close_connectionrF Exception)r<rflast_stream_ids rrlzHTTP2ServerConnection.closesr <<   !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)r8rr'sendallrirjr:r)r<rzr{s rrFz(HTTP2ServerConnection._send_pending_datasg||((*  G !!$' W% G# *-A!+EFF Gs;A$ AA$c|jS)zCheck if connection is closed.rrGs r is_closedzHTTP2ServerConnection.is_closeds||rc<|jj|dy)zvRemove a stream after processing is complete. Args: stream_id: The stream ID to clean up N)r)pop)r<rJs rrz$HTTP2ServerConnection.cleanup_streams D)rcNdt|jd|jdS)Nz)rr)r:rGs r__repr__zHTTP2ServerConnection.__repr__s04<<()*ll^1 & r)N)F))rN) __name__ __module__ __qualname____doc__rhr=rErcr[ryrrrrrrrrrrPrrrrrrrlrFpropertyrrrrrrr"r"1s-"^!*0@dD"L$L:" D@4 +@( "B*/20@=~*X*X B ",G* rr")rr*riorerrorsrrrrr r]r r~r r rrrrr r"__all__rrrrs^   !     ""I  I  X # #r