o .jZ@sUddlmZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZmZmZmZmZdd lmZd d lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&d d lm'Z(d d lmZ)d dl*m+Z+e,dZ-dgZ.dZ/e0dZ1e0dZ2e0dZ3e0dZ4e0dZ5e0Z6de7d<e8dZ9e:ddD]Z;d e9e;<q[;GdddeZGd!d"d"eZ?Gd#d$d$eZ@Gd%d&d&eAZBGd'd(d(eBZCGd)d*d*eBZDGd+d,d,eBZEGd-d.d.eBZFGd/d0d0eBZGGd1d2d2eBZHGd3d4d4eBZIGd5d6d6eBZJGd7d8d8eBZKGd9d:d:eBZLdad@dAZMdbdDdEZNdcdGdHZOdddIdJZP dedfdQdRZQdgdTdUZRdgdVdWZSdhdYdZZTdgd[d\ZUGd]d^d^ZVGd_d`d`ZWdS)i) annotationsN)EnumIntEnum)UINT_VAR_MAX_SIZE) BufferBufferReadErrorDecoderStreamErrorDecompressionFailedEncoderStreamError QpackDecoder QpackEncoder StreamBlockedencode_uint_var)QuicConnectionstream_is_unidirectional)DatagramFrameReceived QuicEventStopSendingReceivedStreamDataReceived StreamReset)QuicLoggerTrace) DatagramReceived DataReceivedGoawayReceivedH3EventHeadersHeadersReceivedInformationalHeadersReceivedPushPromiseReceivedWebTransportStreamDataReceived) StopSending)r)NoAvailablePushIDErrorhttp3h3)rr):status):method:scheme :authority:paths :protocol)r*r,)r*r+r,r-frozenset[bytes]_TRAILERS_PSEUDOA[c@s`eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdS) ErrorCode3r0iiiiiiiii i i i i iiiiiiN)__name__ __module__ __qualname__H3_DATAGRAM_ERROR H3_NO_ERRORH3_GENERAL_PROTOCOL_ERRORH3_INTERNAL_ERRORH3_STREAM_CREATION_ERRORH3_CLOSED_CRITICAL_STREAMH3_FRAME_UNEXPECTEDH3_FRAME_ERRORH3_EXCESSIVE_LOAD H3_ID_ERRORH3_SETTINGS_ERRORH3_MISSING_SETTINGSH3_REQUEST_REJECTEDH3_REQUEST_CANCELLEDH3_REQUEST_INCOMPLETEH3_MESSAGE_ERRORH3_CONNECT_ERRORH3_VERSION_FALLBACKQPACK_DECOMPRESSION_FAILEDQPACK_ENCODER_STREAM_ERRORQPACK_DECODER_STREAM_ERRORrMrMF/home/thesage/.local/lib/python3.10/site-packages/qh3/h3/connection.pyr3Ds,r3c@s4eZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) FrameTyperrrr&r'r( r1N) r5r6r7DATAHEADERSPRIORITY CANCEL_PUSHSETTINGS PUSH_PROMISEGOAWAY MAX_PUSH_IDDUPLICATE_PUSHWEBTRANSPORT_STREAMrMrMrMrNrO\srOc@seZdZdZdZdZdS) HeadersStaterrrN)r5r6r7INITIAL AFTER_HEADERSAFTER_TRAILERSrMrMrMrNr]isr]c@s,eZdZdZdZdZdZdZdZdZ dZ d S) SettingrrPr4iwiB7`+!N) r5r6r7QPACK_MAX_TABLE_CAPACITYMAX_FIELD_SECTION_SIZEQPACK_BLOCKED_STREAMSENABLE_CONNECT_PROTOCOL H3_DATAGRAMH3_DATAGRAM_DRAFT05ENABLE_WEBTRANSPORTDUMMYrMrMrMrNraosrac@s eZdZdZdZdZdZdZdS) StreamTyperrrr&TN)r5r6r7CONTROLPUSH QPACK_ENCODER QPACK_DECODER WEBTRANSPORTrMrMrMrNrms rmc@s"eZdZdZejZdd ddZdS) ProtocolErrorz Base class for protocol errors. These errors are not exposed to the API user, they are handled in :meth:`H3Connection.handle_event`. reason_phrasestrcCs ||_dSN)rv)selfrvrMrMrN__init__s zProtocolError.__init__N)ru)rvrw)r5r6r7__doc__r3r: error_coderzrMrMrMrNrtsrtc@eZdZejZdS)QpackDecompressionFailedN)r5r6r7r3rJr|rMrMrMrNr~ r~c@r})QpackDecoderStreamErrorN)r5r6r7r3rLr|rMrMrMrNrrrc@r})QpackEncoderStreamErrorN)r5r6r7r3rKr|rMrMrMrNrrrc@r})ClosedCriticalStreamN)r5r6r7r3r=r|rMrMrMrNrrrc@r}) FrameErrorN)r5r6r7r3r?r|rMrMrMrNrrrc@r})FrameUnexpectedN)r5r6r7r3r>r|rMrMrMrNrrrc@r}) MessageErrorN)r5r6r7r3rGr|rMrMrMrNrrrc@r})MissingSettingsErrorN)r5r6r7r3rCr|rMrMrMrNrrrc@r}) SettingsErrorN)r5r6r7r3rBr|rMrMrMrNrrrc@r})StreamCreationErrorN)r5r6r7r3r<r|rMrMrMrNrrr frame_typeint frame_databytesreturncCs>t|}t|dtd}|||||||jS)Nrcapacity)lenrr push_uint_var push_bytesdata)rr frame_lengthbufrMrMrN encode_frames    rsettingsdict[int, int]cCs6tdd}|D]\}}||||q |jS)Nir)ritemsrr)rrsettingvaluerMrMrNencode_settingss   rrcCs&t|d}|}|std|S)Nrz*MAX_PUSH_ID frame has extra trailing bytes)r pull_uint_vareofr)rr max_push_idrMrMrNparse_max_push_ids rcCsrt|d}i}|s5|}|}|tvr td|dd||vr-td|dd|||<|r t|S)NrzSetting identifier 0xxz is reserved is included twice)rrrRESERVED_SETTINGSrdict)rrrrrrMrMrNparse_settingss rheadersrallowed_pseudo_headersrequired_pseudo_headersextract_header bytes | NonecCs:d}d}d}d}t}d} t} |D]e\} } | D]} | | r%td| dq|dur0| dur0| } | rt| ddkrt|rBtd| d| |vrNtd| d| |vrZtd| d || | d krf| }q| d krm| }q| d krs| }qd }q||s||}tdt|d|dvr|std|std| S)NFzHeader z contains uppercase lettersr:zPseudo-header z% is not allowed after regular headersz is not validrr,r-r+TzPseudo-headers z are missing)shttpshttpsz+Pseudo-header b':authority' cannot be emptyz&Pseudo-header b':path' cannot be empty)set_HAS_UPPERCASEraddissubset differencesorted)rrrrafter_pseudo_headers authoritypathschemeseen_pseudo_headersextracted_header_value _has_upperkeyrbmissingrMrMrNvalidate_headerssR     rNonecCt|ttddSN)rr)r_PUSH_PROMISE_PSEUDOrrMrMrNvalidate_push_promise_headers%  rcCst|ttddSr)r_REQUEST_ALLOWED_PSEUDO_REQUEST_REQUIRED_PSEUDOrrMrMrNvalidate_request_headers-rr int | NonecCsFt|ttdd}|durdSzt|WSty"td|dw)Nr))rrrzInvalid :status value )r_RESPONSE_ALLOWED_PSEUDO_RESPONSE_REQUIRED_PSEUDOr ValueErrorr)r status_coderMrMrNvalidate_response_headers5s  rcCrr)rr/rrMrMrNvalidate_trailersFrrc@s*eZdZd ddZddd Zdd d Zd S)H3Stream stream_idrrrcCs^d|_d|_t|_d|_d|_d|_tj|_ tj|_ d|_ d|_ ||_ d|_d|_d|_dS)NF)blockedblocked_frame_size bytearraybufferended frame_sizerr]r^headers_recv_stateheaders_send_statepush_id session_idr stream_typereceiving_ended sending_endedryrrMrMrNrzOs zH3Stream.__init__boolcCs|jo |jo |j Srx)rrrryrMrMrNis_ended_szH3Stream.is_endedcCs|jrtdd|_dS)Nzstream was already endedT)rrrrMrMrNfinish_sendingbs zH3Stream.finish_sendingN)rrrr)rrrr)r5r6r7rzrrrMrMrMrNrNs  rc@s6eZdZdZd[d\d d Z d[d]ddZd^ddZd_ddZd`ddZd`ddZ dad d!Z dbd$d%Z dcd'd(Z d[ddd)d*Z eded,d-Zeded.d/Zdfdgd4d5Zdhd8d9Zdid:d;Zdjd=d>Zdkd@dAZdldCdDZdmdFdGZdndIdJZdodKdLZ 0dfdpdMdNZdqdOdPZdrdRdSZdsdTdUZdsdVdWZdtdYdZZd0S)u H3Connectionzu A low-level HTTP/3 connection object. :param quic: A :class:`~qh3.connection.QuicConnection` instance. Fquicrenable_webtransportrrrcCsd|_d|_||_|jj|_d|_||_|j|_t |j|j|_ d|_ d|_ t |_d|_d|_d|_i|_|jr;dnd|_d|_d|_d|_d|_d|_d|_d|_d|_d|_i|_|dS)NidFrrc)_max_table_capacity_blocked_streams_enable_webtransport configuration is_client _is_client_is_done_quic _quic_loggerr _decoder_decoder_bytes_received_decoder_bytes_sentr _encoder_encoder_bytes_received_encoder_bytes_sent_settings_received_stream _max_push_id _next_push_id_local_control_stream_id_local_decoder_stream_id_local_encoder_stream_id_peer_control_stream_id_peer_decoder_stream_id_peer_encoder_stream_id_received_settings_sent_settings_blocked_stream_map_init_connection)ryrrrMrMrNrzps6  zH3Connection.__init__rris_unidirectionalcCs`|r|tj}|j|t||S|j}|j|tjd|j|ttj t||S)z Create a WebTransport stream and return the stream ID. :param session_id: The WebTransport session identifier. :param is_unidirectional: Whether to create a unidirectional stream. rr) _create_uni_streamrmrsrsend_stream_datarget_next_available_stream_id_log_stream_typerOr\)ryrrrrMrMrNcreate_webtransport_streams  z'H3Connection.create_webtransport_streameventr list[H3Event]c Cs|jsZz6t|tr||WSt|tr||jWSt|tr)||j |j WSt|t r7| |j |j WSWgSt yY}zd|_|jj|j |jdWYd}~gSd}~wwgS)zx Handle a QUIC event and return a list of HTTP events. :param event: The QUIC event to handle. T)r|rvN)r isinstancer_receive_stream_datar_receive_datagramrr_receive_stream_resetrr|r_receive_stop_sendingrtrcloserv)ryr excrMrMrN handle_events*     zH3Connection.handle_eventrcCs ||j|j|j|j|j|jfvS)z RFC 9114 6.2: control, QPACK encoder, and QPACK decoder streams are critical streams. Closure of any critical stream MUST be treated as a connection error of type H3_CLOSED_CRITICAL_STREAM. )rrrrrrrrMrMrN_is_critical_streamsz H3Connection._is_critical_streamr|cCst||r td|d|j|}|durt||dgSd|_|r,|j|d|j|dt||dgS)NCritical stream z reset by peerrr|T) rrrget H3StreamResetrrpoprryrr|streamrMrMrNrs  z"H3Connection._receive_stream_resetcCsf||r td|d|j|}|dur%d|_|r%|j|d|j|dt||dgS)Nrz stop-sending by peerTr) rrrrrrrr H3StopSendingrrMrMrNrs   z"H3Connection._receive_stop_sendingflow_idrrcCs|jt||dS)z Send a datagram for the specified flow. :param flow_id: The flow ID. :param data: The HTTP/3 datagram payload. N)rsend_datagram_framer)ryrrrMrMrN send_datagramszH3Connection.send_datagramrrc Cs|jrJd|jdus|j|jkrt|j}|jd7_|j|ttjt || |||j t j |d}|j|t ||S)a Send a push promise related to the specified stream. Returns the stream ID on which headers and data can be sent. :param stream_id: The stream ID on which to send the data. :param headers: The HTTP request headers for this push. z%Only servers may send a push promise.Nr)r)rrrr#rrrrOrXr_encode_headersrrmrp)ryrrrpush_stream_idrMrMrNsend_push_promises  zH3Connection.send_push_promise end_streamcCs|||}|jtjkrtd|r||||jdur0|jjdd|jj t ||dd|j |t tj||dS)a Send data on the given stream. To retrieve datagram which need to be sent over the network call the QUIC connection's :meth:`~qh3.connection.QuicConnection.datagrams_to_send` method. :param stream_id: The stream ID on which to send the data. :param data: The data to send. :param end_stream: Whether to end the stream. 'DATA frame is not allowed in this stateNhttp frame_createdlengthrcategoryr r)_get_or_create_streamrr]r_rr_maybe_cleanup_streamr log_eventencode_http3_data_framerrrrrOrS)ryrrr%rrMrMrN send_data$s"   zH3Connection.send_datac Cs||}|jtjkrtd|r||||}|jdur2|jjdd|jj t |||dd|jtj krcd}|j s\|s\|D]\}}|dkrTt |d krSd }n qA|d s[nqA|sbtj|_ntj|_|j|ttj||||dS) a Send headers on the given stream. To retrieve datagram which need to be sent over the network call the QUIC connection's :meth:`~qh3.connection.QuicConnection.datagrams_to_send` method. :param stream_id: The stream ID on which to send the headers. :param headers: The HTTP headers to send. :param end_stream: Whether to end the stream. *HEADERS frame is not allowed in this stateNr'r(r*rrr+Fr)T:)r-rr]r`rrr"rr/encode_http3_headers_framerr^rr startswithr_rrrrOrTr.) ryrrr%rris_informational_headerskvrMrMrN send_headersHsD          zH3Connection.send_headersdict[int, int] | NonecC|jS)z> Return the received SETTINGS frame, or None. )rrrMrMrNreceived_settingszH3Connection.received_settingscCr=)z: Return the sent SETTINGS frame, or None. )rrrMrMrN sent_settingsr?zH3Connection.sent_settingsNrrrcCs4|jjdd}|j|||d|j|t||S)zD Create an unidirectional stream of the given type. T)rrrr)rr r rr)ryrrrrMrMrNrs zH3Connection._create_uni_streamrrc Csz4|dur|j|j\}}|j|`n|st|j||\}}|jt|7_|j|j |W|StyD}zt |d}~ww)z Decode a HEADERS block and send decoder updates on the decoder stream. This is called with frame_data=None when a stream becomes unblocked. N) r_pendingr r feed_headerrrrrrr~)ryrrdecoderrrrMrMrN_decode_headerss zH3Connection._decode_headersc Cs^z |j||\}}Wnty}zt|d}~ww|jt|7_|j|j||S)zX Encode a HEADERS block and send encoder updates on the encoder stream. N) rencoder rrrrrr)ryrrencoderrrrMrMrNr"szH3Connection._encode_headersrcCs*|j|}|durt|}||j|<|Srx)rrr)ryrrrMrMrNr-s  z"H3Connection._get_or_create_streamrcCs0|r|j|jd|j|jddSdSrx)rrrrr)ryrrMrMrNr.sz"H3Connection._maybe_cleanup_streamrc Cs:tj|jtjdtj|jtjdtjdi}|jrd|tj <|S)z3 Return the local HTTP/3 settings. ir) rarerrfrgrrirlrrk)ryrrMrMrN_get_local_settingss z H3Connection._get_local_settingsrcCs|tjkr |js t|tjkrG|jrtdt|}||||_|jj | t j d| t j d| t j dd}|j|j|d|_gS|tjkrct|d}|}|s]tdt|dgS|tjkrv|jrotdt||_gS|tjtjtjtjfvrtd gS) zG Handle a frame received on the peer's control stream. z#SETTINGS have already been receivedr)max_table_capacitydyn_table_capacityblocked_streamsTrz%GOAWAY frame has extra trailing bytes)rz!Servers must not send MAX_PUSH_IDz$Invalid frame type on control stream)rOrWrrrr_validate_settingsrrapply_settingsrrarergrrrrYrrrrrrZrrrrSrTrXr[)ryrrrrGr goaway_idrMrMrN_handle_control_framesF         z"H3Connection._handle_control_frame stream_endedc CsX|tjkr |jtjurtd|s|rt||j||jdgSgSg}|tj kr|jtj ur1td| |j|}d}|jtj urM|j rHt|}n t|nt||jdurp|jjdd|jj|dure|jnt|||jdd|jtj ur|dus|d ks|rtj|_n tj |_|jrd }|jtj ur|dur|d kr|t||jd |S|t||j|j|d |S|tjkr |jdur |j std t|d}|} | |j||d}t||jdur|jjdd|jjt||| |jdd|t || |jd|S|tj!tj"tj#tjtj$tj%tj&fvr*t|jdur'dd|S)zF Handle a frame received on a request or push stream. r&)rrrPrr2Nr' frame_parsedr3r+r4T)rr)rrrrPz"Clients must not send PUSH_PROMISEr)r*rrr)rrrz$Invalid frame type on request streamz!Invalid frame type on push stream)'rOrSrr]r_rrrrrTr`rEr^rrrrrr/r6rrrappendrrrXrrtellrencode_http3_push_promise_framer rUrVrWrYrZr[) ryrrrrP http_eventsrr frame_bufrrMrMrN_handle_request_or_push_frames         =6    z*H3Connection._handle_request_or_push_framecCs|tj|_||_|j|jtt j t |j|j r2|j dur2|j|jtt jt|j |tj|_|tj|_dSrx)rrmrorrHrrrrrOrWrrrrZrrqrrrrrrMrMrNrs"  zH3Connection._init_connectioncCsV|jdur)dddddd|d}||d}|dur||d <|jjd d |d dSdS) Ncontrolpush qpack_encoder qpack_decoder webtransport)rrrr&rnunknown)newrassociated_push_idr'stream_type_setr+)rrr/)ryrrr type_namerrMrMrNr s$   zH3Connection._log_stream_typecCsHt|d}z|}Wn tytdwt||d|dgS)z$ Handle a datagram. rzCould not parse flow IDN)rr)rrrrtrrS)ryrrrrMrMrNrs   zH3Connection._receive_datagramrcCsL|j}||}t|r|||j|j}n |||j|j}|||Srx)rr-r_receive_stream_data_unirr%_receive_request_or_push_datar.)ryr rrresultrMrMrNrs   z!H3Connection._receive_stream_datacCsg}|j}|r d|_|jr|||S|j}|jtjkrK|jdurK|r=||| t t ||j||d| |S| t ||j||d|S|jtj kr|jdur|s}t|}||jkr||rgtd| t||j|dd|j|8_|Sn1||t|}||jkr|rtd| tt ||j|dd|j|8_| |Sd}|r|s|s|jtjurtd|j|ddgSgS|r||tt |d} d } | j} | j} | j} | j}|j}| s|jdurZz | |_| |_Wn tyYnw| } |jtjkr=|j|_d|_t || d}| |j|tjd |s/|r;| t ||j||d|S|j durZ|jtj krZ|j j!d d |j j"|j|d d|j}t#|| j$| }|jtj krr||krrnV||}|j}| } |||_|jsd|_d|_z||||||jo| dWnt%yd|_t||_&||j'|<Ynw|jtj(ur|jrt|} n| r| r|d| =|r|s|js|jtj)ur|rt*|ddds| td|j|dd|S)zC Handle data received on a request or push stream. TN)rrrrPz"DATA frame truncated by stream endF)rrrrPrrrr'rQr)r+rrrrPrP)+rrrextendrrrOr\rrRr!rclearrSrrrrrrr]r^rrSrr pull_bytesrWrr rmrsrr/r0minrrrrr`r_getattr)ryrrrPrUs_buffer s_stream_iddata_lenbuf_lenrconsumedbuf_tellbuf_eofbuf_pull_uint_varbuf_pull_bytes handle_framer s_frame_size chunk_sizerrMrMrNrcsP                     X   z*H3Connection._receive_request_or_push_datac Csvg}|j||r d|_tt|jd}d}t}|jtjtj tj fvs*| s|jdurz| |_Wn t y@Ynw|}|jtj krY|jdurTtd|j|_n'|jtjkrm|jdurhtd|j|_n|jtjkr|jdur|td|j|_|jtjkr|j|j|jd|jtj kr|rtdz| }| } || } Wn t yYnEw|}|||| n&|jtjkr|jdurz| |_Wn t yYnw|}|j|j|j|jd |jd|=||d |S|jtj krD|jdur z| |_Wn t yYnw|}t|j|d} |j| s4|rB|t| |j|j|jd |S|jtjkr}||j |}|}z|j!"|Wnt#yr} zt$| d} ~ ww|j%t&|7_%nm|jtjkr||j |}|}z|j'(|Wnt)y} zt*| d} ~ ww|j+,D]$\} } z|j'-| \}}||f| _.|/| Wqt0yYqw|j1t&|7_1n |2|j |}|jtjtj tj fvs*| r*|r|jd|=|D]4}|j3|}||j4t5j6d||jo|j d d |_7d|_8|j+|=|jr7|||d |jq|S)NTrrz"Only one control stream is allowedz(Only one QPACK decoder stream is allowedz(Only one QPACK encoder stream is allowedrz%Closing control stream is not allowedrAre)rrrPrrfF)9rrhrrrrrrmrprorsrrrrSrrrrrrrqrr rrjrOrrcrrirRr!rr feed_decoderr rrrr feed_encoderr rrr resume_headerrBrrrseekrrWrOrTrr)ryrrrPrUrrqunblocked_streamsrrrr blocked_idblocked_stream stream_datarrrMrMrNrbs(                        z%H3Connection._receive_stream_data_unircCstjtjtjtjfD]}||vr||dvrt|jdq |tj}|dur/|tj}|dkr=|jj dur=td|tjdkrM|dkrOtddSdS)N)rrz setting must be 0 or 1rz@H3_DATAGRAM requires max_datagram_frame_size transport parameterz(ENABLE_WEBTRANSPORT requires H3_DATAGRAM) rarhrkrirjrnamerr_remote_max_datagram_frame_size)ryrr h3_datagramrMrMrNrL`s$  zH3Connection._validate_settings)F)rrrrrr)rrrrrr)r rrr )rrrr)rrr|rrr )rrrrrr)rrrrrr)rrrrr%rrr)rrrrr%rrr)rr<rx)rrrrrr)rrrrrr)rrrrrr)rrrr)rrrr)rr)rrrrrr ) rrrrrrrPrrr r)rrrrrrrr)rrrr )r rrr )rrrrrPrrr )rrrr) r5r6r7r{rzr rrrrr!r$r1r;propertyr>r@rrEr"r-r.rHrOrWrr rrrcrbrLrMrMrMrNrisJ %      % 9        )    [.r)rrrrrr)rrrr)rrrr)rrrrrx) rrrr.rr.rrrr)rrrr)rrrr)X __future__rloggingenumrr_compatr_hazmatrrr r r r r rrquic.connectionrr quic.eventsrrrrr quic.loggerreventsrrrrrrrr r!r"rr exceptionsr# getLoggerloggerH3_ALPNr frozensetrrrrrr/__annotations__rrrange_ir3rOr]rarm Exceptionrtr~rrrrrrrrrrrrrrrrrrrrrMrMrMrNsn ,  ,         =