o i1W@sxdZddlmZddlmZmZmZmZmZddl m Z ddl m Z da dadadadadd ZGd d d Zd gZdS) zc HTTP/2 server connection implementation. Uses the hyper-h2 library for HTTP/2 protocol handling. )BytesIO) HTTP2ErrorHTTP2ProtocolErrorHTTP2ConnectionErrorHTTP2NotAvailableHTTP2ErrorCode) HTTP2Stream) HTTP2RequestNcCsdtdurdSz!ddlmaddlmaddlmaddl m a ddl m aWdSty1tw)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 ImportErrorrrrM/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/http2/connection.py _import_h2s     rc@seZdZdZdZddZddZd5dd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZd5ddZddZd6d!d"Zd#d$Zd5d%d&Zd7d(d)Zd8d+d,Zd-d.Zed/d0Zd1d2Zd3d4ZdS)9HTTP2ServerConnectionzHTTP/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. icCsxt||_||_||_i|_g|_|j|_|j|_ |j |_ |j |_ tjddd}tj|d|_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)rcfgsock 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#rrrr__init__:s$  zHTTP2ServerConnection.__init__c CsZ|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)r3r0update_settingsr SettingCodesMAX_CONCURRENT_STREAMSr)INITIAL_WINDOW_SIZEr'MAX_FRAME_SIZEr+MAX_HEADER_LIST_SIZEr-initiate_connection_send_pending_datar4rrrr<es      z)HTTP2ServerConnection.initiate_connectionNc Cs|dur#z |j|j}Wnttfy"}ztd|d}~ww|s*d|_gSz|j|}Wnt j yL}z |j t j dtt|d}~wt jye}z |j t jdtt|d}~wt jy}zt|dd}|dur~|j |dn|j t jdtt|d}~wt jy}z |j t jdtt|d}~wt jy}z |j t jdtt|d}~wwg}|D]}||}|dur||q||S)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_coder@)r"recvREAD_BUFFER_SIZEOSErrorIOErrorrr2r0 receive_datarFlowControlErrorcloserFLOW_CONTROL_ERRORrstrFrameTooLargeErrorFRAME_SIZE_ERRORInvalidSettingsValueErrorgetattrPROTOCOL_ERRORTooManyStreamsErrorREFUSED_STREAM ProtocolError _handle_eventappendr=)r4dataerr@completed_requestseventrequestrrrrEzsX         z"HTTP2ServerConnection.receive_datacCst|tjr ||St|tjr||St|tjr!||St|tjr.| |dSt|tj r7 dSt|tj rD| |dSt|tj rM dSt|tjrZ||dSt|tjre||SdS)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_receivedr4rWrrrrRs0            z#HTTP2ServerConnection._handle_eventcCs2|j}|j}t||}||j|<|j|dddS)z{Handle RequestReceived event (HEADERS frame). Args: event: RequestReceived event with headers F end_streamN) stream_idheadersr r$receive_headers)r4rWrmrnstreamrrrr[s   z.HTTP2ServerConnection._handle_request_receivedc Cs|j}|j}|j|}|durdS|j|ddt|dkrOz|jjt||d|jjt|dd|WdSt t j fyN|j t jdYdSwdS)zHandle DataReceived event. Args: event: DataReceived event with body data Returns: None (request completion handled by StreamEnded) NFrkr)rmr?)rmrTr$getrElenr0increment_flow_control_windowr= ValueErrorrrFrGrrH)r4rWrmrTrprrrr]s"    z+HTTP2ServerConnection._handle_data_receivedcCs4|j}|j|}|durdSd|_t||j|jS)zHandle StreamEnded event. Args: event: StreamEnded event Returns: HTTP2Request for the completed request NT)rmr$rqrequest_completer r!r#r4rWrmrprrrr_s  z*HTTP2ServerConnection._handle_stream_endedcCs.|j}|j|}|dur||jdSdS)ziHandle StreamReset event (RST_STREAM frame). Args: event: StreamReset event N)rmr$rqresetr@rvrrrra*s  z*HTTP2ServerConnection._handle_stream_resetcCs d|_dS)zwHandle ConnectionTerminated event (GOAWAY frame). Args: event: ConnectionTerminated event TNr2rjrrrrg7s z3HTTP2ServerConnection._handle_connection_terminatedcCs:|j}|j|}|durdS||jt||j|jS)zHandle TrailersReceived event. Args: event: TrailersReceived event with trailer headers Returns: HTTP2Request if this completes the request N)rmr$rqreceive_trailersrnr r!r#rvrrrri@s   z/HTTP2ServerConnection._handle_trailers_receivedcCs4|j|j}|dur|j|j|j|jddSdS)zHandle PriorityUpdated event (PRIORITY frame). Args: event: PriorityUpdated event with priority info N)weight depends_on exclusive)r$rqrmupdate_priorityrzr{r|)r4rWrprrrrdTs z.HTTP2ServerConnection._handle_priority_updatedcCs|dks|dkrtd||j|}|dur!td|ddt|fg}|D]\}}||t|fq*|jj||dd |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:statusFrk) rr$rqrIrSlowerr0 send_headersr=)r4rmstatusrnrpresponse_headersnamevaluerrrsend_informationalbs   z(HTTP2ServerConnection.send_informationalc Cs|j|}|dur dSdt|fg}|D]\}}||t|fq|dup.t|dk} z'|jj||| d|j|| d||rTt|dkrT|j ||ddWdSt j yj| | |YdSw)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 NFrrrkT)r$rqrIrSrrrr0rr= send_datarStreamClosedErrorrGcleanup_stream) r4rmrrnbodyrprrrrlrrr send_responses&   z#HTTP2ServerConnection.send_responsec Csddl}d}z|}||j|jWn ttfy YdSwd}zt|D]t}|j |}|dkr8|}nr|j dd}|rz |j |j } Wnt ttjfyWYnZw| s_d|_nYz|j| } Wn tjyrYnMw| D]} t| tjr| j|krd}nqut| tjrd|_d}nqu|q(n*q(|j |}W||SW||SW||SW||SW||SW||S|w)zWait for flow control window to become positive. Returns: int: Available window size, or -1 if waiting failed rN2g?)timeoutT) selectorsDefaultSelectorregisterr" EVENT_READ TypeErrorrtranger0local_flow_control_windowselectrArBrCrDrrQr2rErYrr`rmrfr=rG) r4rmrmax_wait_attemptsselresult_ availablereadyincomingrrWrrr_wait_for_flow_control_windows        z3HTTP2ServerConnection._wait_for_flow_control_windowFc Cs|j|}|dur dS|}z[|r`|j|}t||jt|}|dkr=|||}|dkr4WdSt||jt|}|d|}||d}|oPt|dk} |jj ||| d||s|j ||dWdSt j t j fy| ||YdSw)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 NFrrkT)r$rqr0rminr+rrr=rrrrrFrGr) r4rmrTrlrp data_to_sendr chunk_sizechunkis_finalrrrrs6      zHTTP2ServerConnection.send_datacCs|j|}|dur dS|jsdSg}|D]\}}|}|dr*td|d||t|fqz|jj ||dd| || WdSt j y]|||YdSw)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 trailersTrk)r$rqresponse_headers_sentr startswithrrSrIr0r send_trailersr=rrrGr)r4rmtrailersrptrailer_headersrrlnamerrrrs*     z#HTTP2ServerConnection.send_trailerscCsD|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 zcontent-length)z content-typeztext/plain; charset=utf-8N)encoderIrrrSr)r4rm status_codemessagerrnrrr send_error@s  z HTTP2ServerConnection.send_errorcCs:|j|}|dur|||jj||d|dS)zReset a stream with RST_STREAM. Args: stream_id: The stream ID to reset error_code: HTTP/2 error code (default: CANCEL) Nr?)r$rqrwr0 reset_streamr=)r4rmr@rprrrrOs   z"HTTP2ServerConnection.reset_streamrcCsb|jrdSd|_|dur|jrt|jnd}z|jj|d|WdSty0YdSw)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) NTrr?)r2r$maxkeysr0close_connectionr= Exception)r4r@last_stream_idrrrrG]s zHTTP2ServerConnection.closec CsV|j}|r)z |j|WdSttfy(}z d|_td|d}~wwdS)z,Send any pending data from h2 to the socket.TzSocket write error: N)r0rr"sendallrCrDr2r)r4rTrUrrrr=ss z(HTTP2ServerConnection._send_pending_datacCs|jS)zCheck if connection is closed.rxr>rrr is_closed}szHTTP2ServerConnection.is_closedcCs|j|ddS)zvRemove a stream after processing is complete. Args: stream_id: The stream ID to clean up N)r$pop)r4rmrrrrsz$HTTP2ServerConnection.cleanup_streamcCsdt|jd|jdS)Nz)rrr$r2r>rrr__repr__s zHTTP2ServerConnection.__repr__)N)F)r)rN)__name__ __module__ __qualname____doc__rBr5r<rErRr[r]r_rargrirdrrrrrrrrGr=propertyrrrrrrrr/s6+ B&"   , :, ,    r)riorerrorsrrrrrrpr rXr r rrrrrr__all__rrrrs"    g