o iP@stdZddlZddlmZmZmZmZmZddlm Z ddl m Z da da dadadaddZGd d d Zd gZdS) 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) HTTP2RequestcCsdtdurdSz!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 ImportErrorrrrS/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/http2/async_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)9AsyncHTTP2ConnectionzAsync 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. icCst||_||_||_||_i|_t|_|j |_ |j |_ |j |_|j|_tjddd}tj|d|_d|_d|_d|_dS)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#rrrr__init__:s&   zAsyncHTTP2Connection.__init__c sb|jrdS|jtjj|jtjj|jtjj |j tjj |j i|j |IdHd|_dS)zSend initial HTTP/2 settings to client. Should be called after the SSL handshake completes and before processing any data. NT)r4r2update_settingsr SettingCodesMAX_CONCURRENT_STREAMSr+INITIAL_WINDOW_SIZEr)MAX_FRAME_SIZEr-MAX_HEADER_LIST_SIZEr/initiate_connection_send_pending_datar6rrrr>es      z(AsyncHTTP2Connection.initiate_connectionNc s z |durtj|j|j|dIdH}n |j|jIdH}Wnttfy6}ztd|d}~ww|s>d|_gSz|j |}Wnt j yc}z|j tjdIdHtt|d}~wt jy}z|j tjdIdHtt|d}~wt jy}z$t|dd}|dur|j |dIdHn |j tjdIdHtt|d}~wt jy}z|j tjdIdHtt|d}~wt jy}z|j tjdIdHtt|d}~wwg}|D]}||}|dur||q|IdH|S)ahReceive data and return completed requests. Args: timeout: Optional timeout in seconds for read operation Returns: list: List of HTTP2Request objects for completed requests Raises: HTTP2ConnectionError: On protocol or connection errors asyncio.TimeoutError: If timeout expires NtimeoutzSocket read error: T error_coderD)r%wait_forr!readREAD_BUFFER_SIZEOSErrorIOErrorrr3r2 receive_datarFlowControlErrorcloserFLOW_CONTROL_ERRORrstrFrameTooLargeErrorFRAME_SIZE_ERRORInvalidSettingsValueErrorgetattrPROTOCOL_ERRORTooManyStreamsErrorREFUSED_STREAM ProtocolError _handle_eventappendr?) r6rBdataerrDcompleted_requestseventrequestrrrrJzsd          z!AsyncHTTP2Connection.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_receivedr6r\rrrrWs0            z"AsyncHTTP2Connection._handle_eventcCs2|j}|j}t||}||j|<|j|dddS)z-Handle RequestReceived event (HEADERS frame).F end_streamN) stream_idheadersrr$receive_headers)r6r\rrrsstreamrrrr`s   z-AsyncHTTP2Connection._handle_request_receivedc Cs|j}|j}|j|}|durdS|j|ddt|dkr\z|jjt||d|jjt|ddWdStt j fy[d|_ z |jj t jdWYdStyZYYdSwwdS)zHandle DataReceived event.NFrpr)rrTrC)rrrYr$getrJlenr2increment_flow_control_window ValueErrorrrKr3close_connectionrrM Exception)r6r\rrrYrurrrrbs,     z*AsyncHTTP2Connection._handle_data_receivedcCs4|j}|j|}|durdSd|_t||j|jS)zHandle StreamEnded event.NT)rrr$rvrequest_completer r r#r6r\rrrurrrrds  z)AsyncHTTP2Connection._handle_stream_endedcCs.|j}|j|}|dur||jdSdS)zHandle StreamReset event.N)rrr$rvresetrDr}rrrrfs  z)AsyncHTTP2Connection._handle_stream_resetcCs d|_dS)z"Handle ConnectionTerminated event.TNr3rorrrrl$s z2AsyncHTTP2Connection._handle_connection_terminatedcCs:|j}|j|}|durdS||jt||j|jS)zHandle TrailersReceived event.N)rrr$rvreceive_trailersrsr r r#r}rrrrn(s   z.AsyncHTTP2Connection._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$rvrrupdate_priorityrrr)r6r\rurrrri3s z-AsyncHTTP2Connection._handle_priority_updatedcs|dks |dkrtd||j|}|dur"td|ddt|fg}|D]\}}||t|fq+|jj||dd |IdHdS) 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:statusFrp) rr$rvrNrXlowerr2 send_headersr?)r6rrstatusrsruresponse_headersnamevaluerrrsend_informationalAs  z'AsyncHTTP2Connection.send_informationalc s|j|}|dur dSdt|fg}|D]\}}||t|fq|dup/t|dk} z-|jj||| d|j|| d|IdH|r[t|dkr[|j ||ddIdHWdSt j yq| | |YdSw)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 NFrrrpT)r$rvrNrXrrwr2rr? send_datarStreamClosedErrorrLcleanup_stream) r6rrrrsbodyrurrrrqrrr send_responseas(   z"AsyncHTTP2Connection.send_responsec sd}t|D]q}|j|}|dkr|SzLtj|j|jddIdH}|r[|j|}|D]!}t |t j rD|j |krCWdSq1t |t j rRd|_WdSq1|IdHnd|_WdSWqtjymYqtjyxYdSw|j|S)zWait for flow control window to become positive. Returns: int: Available window size, or -1 if waiting failed 2rg?rANT)ranger2local_flow_control_windowr%rEr!rFrGrJr^rrerrrkr3r? TimeoutErrorrrV)r6rrmax_wait_attempts_ availableincomingrr\rrr_wait_for_flow_control_windows@           z2AsyncHTTP2Connection._wait_for_flow_control_windowFc s|j|}|dur dS|}zd|rj|j|}t||jt|}|dkrD|IdH||IdH}|dkr;WdSt||jt|}|d|}||d}|oWt|dk} |jj ||| d|IdH|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 NFrrpT)r$rvr2rminr-rwr?rrrrrKrLr) r6rrrYrqru data_to_sendr chunk_sizechunkis_finalrrrrs8     zAsyncHTTP2Connection.send_datacs|j|}|dur dS|jsdSg}|D]\}}|}|dr+td|d||t|fqz|jj ||dd| || IdHWdSt j ya|||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 trailersTrp)r$rvresponse_headers_sentr startswithrrXrNr2r send_trailersr?rrrLr)r6rrtrailersrutrailer_headersrrlnamerrrrs,     z"AsyncHTTP2Connection.send_trailerscsL|r|nd}dtt|fg}|r|d|||||IdHdS)z#Send an error response on a stream.zcontent-length)z content-typeztext/plain; charset=utf-8N)encoderNrwrXr)r6rr status_codemessagerrsrrr send_errors  zAsyncHTTP2Connection.send_errorcsB|j|}|dur|||jj||d|IdHdS)zReset a stream with RST_STREAM.NrC)r$rvr~r2 reset_streamr?)r6rrrDrurrrrs   z!AsyncHTTP2Connection.reset_streamrcs|jrdSd|_|dur|jrt|jnd}z|jj|d|IdHWn ty2Ynwz|j |j IdHWdStyMYdSw)z,Close the connection gracefully with GOAWAY.NTrrC) r3r$maxkeysr2rzr?r{r"rL wait_closed)r6rDlast_stream_idrrrrLs$   zAsyncHTTP2Connection.closec sh|j}|r2z|j||jIdHWdSttfy1}z d|_td|d}~wwdS)z,Send any pending data from h2 to the socket.NTzSocket write error: ) r2rr"writedrainrHrIr3r)r6rYrZrrrr?0s  z'AsyncHTTP2Connection._send_pending_datacCs|jS)zCheck if connection is closed.rr@rrr is_closed;szAsyncHTTP2Connection.is_closedcCs|j|ddS)z-Remove a stream after processing is complete.N)r$pop)r6rrrrrr@sz#AsyncHTTP2Connection.cleanup_streamcCsdt|jd|jdS)Nz)rwr$r3r@rrr__repr__Ds zAsyncHTTP2Connection.__repr__)N)F)r)rN)__name__ __module__ __qualname____doc__rGr7r>rJrWr`rbrdrfrlrnrirrrrrrrrLr?propertyrrrrrrrr0s6+ H&    ( (+ ,   r)rr%errorsrrrrrrurr]r r rrrrrr__all__rrrrs"