Format: application/x-python-code
1 o
2
3 7'h�] � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dl
4 Zd dl Zd dl Zddl
5 mZ ddlmZmZmZ ddlmZmZmZ ddlmZmZmZ ddlm Z d d
6 l m Z e� d �Z ddd�Z!G dd� dej"�Z#G dd� de �Z$G dd� d�Z%dS )� )� annotationsN� )�AsyncNetworkStream)�ConnectionNotAvailable�LocalProtocolError�RemoteProtocolError)�Origin�Request�Response)� AsyncLock�AsyncSemaphore�AsyncShieldCancellation)�Trace� )�AsyncConnectionInterfacezhttpcore.http2�requestr �return�boolc C s t dd� | jD ��S )Nc s s, � | ]\}}|� � d kp|� � dkV qdS )s content-length� transfer-encodingN��lower��.0�k�v� r �KC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_async/http2.py� <genexpr> s
7 � �
8 �z#has_body_headers.<locals>.<genexpr>)�any�headers)r r r r �has_body_headers s �r c @ s e Zd ZdZdZdZdS )�HTTPConnectionStater r � N)�__name__�
9 __module__� __qualname__�ACTIVE�IDLE�CLOSEDr r r r r! $ s r! c @ s2 e Zd ZdZejjdd�Z dWdXd d �ZdYdd�Z dZdd�Z
10 d[dd�Z d[dd�Z d\d d �Z
11 d[d d!�Zd]d#d$�Zd^d&d'�Zd_d)d*�Z dWd`d,d-�Zdad0d1�Zdbd2d3�Zdcd4d5�Zddd7d8�ZdZd9d:�Zded;d<�Zdfd>d?�Zdgd@dA�ZdgdBdC�ZdgdDdE�Z dgdFdG�Z dhdIdJ�Z dhdKdL�Z didMdN�Z djdkdUdV�Z!dS )l�AsyncHTTP2Connection� F)Zvalidate_inbound_headersN�originr �streamr �keepalive_expiry� float | Nonec C s� || _ || _|| _tjj| jd�| _tj | _
12 d | _ d| _ t
13 � | _t
14 � | _t
15 � | _t
16 � | _d| _d| _d| _i | _d | _d | _d | _d S )N)�configr F)�_origin�_network_stream�_keepalive_expiry�h2�
17 connectionZ H2Connection�CONFIG� _h2_stater! r' �_state�
18 _expire_at�_request_countr �
19 _init_lock� _state_lock�
20 _read_lock� _write_lock�_sent_connection_init�_used_all_stream_ids�_connection_error�_events�_connection_terminated�_read_exception�_write_exception)�selfr+ r, r- r r r �__init__. s& �
21 z AsyncHTTP2Connection.__init__r r r r
22 c
23 � sD �| � |jj�std|jj� d| j� ���| j4 I d H �& | jtjtj fv r5| j
24 d7 _
25 d | _ tj| _nt � �W d �I d H n 1 I d H sHw Y | j
26 4 I d H �� | js�z2d|i}tdt||�4 I d H � | jdi |��I d H W d �I d H n 1 I d H s�w Y W n) ty� } z t� � | �� I d H W d � |�1 s�w Y |�d }~ww d| _d| _| jjj}t|�| _t|| j �D ]
27 }| j� � I d H q�W d �I d H n 1 I d H s�w Y | j� � I d H z | j� � }g | j |< W n t j j!�y d| _"| j
28 d8 _
29 t � �w z�||d�}tdt||�4 I d H � | j#||d�I d H W d �I d H n 1 I d H �sDw Y td t||�4 I d H � | j$||d�I d H W d �I d H n 1 I d H �spw Y td
30 t||�4 I d H � }| j%||d�I d H \} }
31 | |
32 f|_&W d �I d H n 1 I d H �s�w Y t'| |
33 t(| ||d �d | j)|d
34 �d�W S t�y! } zZt� �7 d|i}tdt||�4 I d H � | j*|d �I d H W d �I d H n 1 I d H �s�w Y W d � n 1 �sw Y t+|t j j,��r| j-�rt.| j-��t/|��|�d }~ww )Nz Attempted to send request to z on connection to r r Zsend_connection_initT�r � stream_idZsend_request_headersZsend_request_bodyZreceive_response_headers�rH s HTTP/2)� http_versionZnetwork_streamrH )�statusr �content�
35 extensionsrH �response_closedr )0�can_handle_request�urlr+ � RuntimeErrorr0 r; r7 r! r& r' r9 r8 r r: r> r �logger�_send_connection_init�
36 BaseExceptionr
37 �aclose� _max_streamsr6 �local_settings�max_concurrent_streamsr �_max_streams_semaphore�range�acquireZ get_next_available_stream_idrA r3 �
38 exceptionsZNoAvailableStreamIDErrorr? �_send_request_headers�_send_request_body�_receive_response� return_valuer
39 �HTTP2ConnectionByteStreamr1 �_response_closed�
40 isinstance�
41 ProtocolErrorrB r r ) rE r Z
42 sci_kwargs�excZlocal_settings_max_streams�_rH �kwargs�tracerK r r r r �handle_async_requestU s� � ��
43 (��(��
44 �����
45 �(�
46 �
47 *�*� �� *� ��
48 *�� �
49
50 ��z)AsyncHTTP2Connection.handle_async_request�Nonec � sn �t jjdt jjjdt jjjdt jjjdid�| j_| jjt jjj = | j�
51 � | j� d� | � |�I dH dS )z�
52 The HTTP/2 connection requires some initial setup before we can start
53 using individual request/response streams on it.
54 Tr �d r* )�clientZinitial_values� N)
55 r3 �settingsZSettings� SettingCodesZ ENABLE_PUSH�MAX_CONCURRENT_STREAMSZMAX_HEADER_LIST_SIZEr6 rW ZENABLE_CONNECT_PROTOCOLZinitiate_connection� increment_flow_control_window�_write_outgoing_data)rE r r r r rS � s �
56
57
58 �
59 ��
60 z*AsyncHTTP2Connection._send_connection_initrH �intc � s� �t |� }dd� |jD �d }d|jfd|fd|jjfd|jjfgdd� |jD � }| jj|||d � | jjd
61 |d � | � |�I d H d S )
62 z@
63 Send the request headers to a given stream ID.
64 c S s g | ] \}}|� � d kr|�qS )� hostr r r r r �
65 <listcomp>� s z>AsyncHTTP2Connection._send_request_headers.<locals>.<listcomp>r s :methods
66 :authoritys :schemes :pathc S s( g | ]\}}|� � d vr|� � |f�qS ))rt r r r r r r ru � s �
67 �)�
68 end_streamrm rI N)
69 r r �methodrP �scheme�targetr6 � send_headersrq rr )rE r rH rv � authorityr r r r r] � s �
70
71
72 ���z*AsyncHTTP2Connection._send_request_headersc � s` �t |�sdS t|jtj�sJ �|j2 z3 dH W }| �|||�I dH q6 | �||�I dH dS )zP
73 Iterate over the request body sending it to a given stream ID.
74 N)r rc r, �typing�
75 AsyncIterable�_send_stream_data�_send_end_stream)rE r rH �datar r r r^ � s ��z'AsyncHTTP2Connection._send_request_bodyr� �bytesc � sj �|r3| � ||�I dH }tt|�|�}|d|� ||d� }}| j�||� | �|�I dH |sdS dS )zI
76 Send a single chunk of data in one or more data frames.
77 N)�_wait_for_outgoing_flow�min�lenr6 Z send_datarr )rE r rH r� Zmax_flow�
78 chunk_size�chunkr r r r~ s � �z&AsyncHTTP2Connection._send_stream_datac � s" �| j �|� | �|�I dH dS )z`
79 Send an empty data frame on on a given stream ID with the END_STREAM flag set.
80 N)r6 rv rr )rE r rH r r r r s � z%AsyncHTTP2Connection._send_end_stream�%tuple[int, list[tuple[bytes, bytes]]]c � s� � | � ||�I dH }t|tjj�rnqd}g }|jdus J �|jD ] \}}|dkr4t|jddd��}q"|�d�s@|� ||f� q"||fS ) zT
81 Return the response status code and headers for a given stream ID.
82 TN�� s :status�ascii�ignore)�errors� :)
83 �_receive_stream_eventrc r3 �events�ResponseReceivedr rs �decode�
84 startswith�append)rE r rH �event� status_coder r r r r r r_ s ��
85 �z&AsyncHTTP2Connection._receive_response�typing.AsyncIterator[bytes]c C s� � | � ||�I dH }t|tjj�r7|jdusJ �|jdus J �|j}| j�||� | � |�I dH |jV n t|tjj
86 �r@dS q)z]
87 Iterator that returns the bytes of the response body for a given stream ID.
88 TN) r� rc r3 r� � DataReceivedZflow_controlled_lengthr� r6 Zacknowledge_received_datarr � StreamEnded)rE r rH r� �amountr r r �_receive_response_body5 s �
89 �z+AsyncHTTP2Connection._receive_response_body�Kh2.events.ResponseReceived | h2.events.DataReceived | h2.events.StreamEndedc � sV �| j �|�s| �||�I dH | j �|�r| j | �d�}t|tjj�r)t|��|S )z�
90 Return the next available event for a given stream ID.
91
92 Will read more data from the network if required.
93 Nr ) rA �get�_receive_events�poprc r3 r� � StreamResetr )rE r rH r� r r r r� G s � �z*AsyncHTTP2Connection._receive_stream_event�
94 int | Nonec � sz �| j 4 I dH �� | jdur)| jj}|r$|r$||kr$| jd8 _t� �t| j��|du s3| j�|�s�| �|�I dH }|D ]`}t |t
95 j j �rrt
96 dt|�4 I dH �}| �|�I dH ||_W d �I dH n 1 I dH slw Y q=t |t
97 j jt
98 j jt
99 j jt
100 j jf�r�|j| jv r�| j|j �|� q=t |t
101 j j�r�|| _q=W d �I dH n 1 I dH s�w Y | �|�I dH dS )zp
102 Read some data from the network until we see one or more events
103 for a given stream ID.
104 Nr Zreceive_remote_settings)r< rB �last_stream_idr9 r r rA r� �_read_incoming_datarc r3 r� ZRemoteSettingsChangedr rR � _receive_remote_settings_changer` r� r� r� r� rH r� ZConnectionTerminatedrr )rE r rH r� r� r� rh r r r r� V sH �
105
106 �(���� ��(�'z$AsyncHTTP2Connection._receive_eventsr� � h2.events.RemoteSettingsChangedc � s� �|j �tjjj�}|rPt|j| jj j
107 �}|rR|| j krT|| j kr5| j �
108 � I d H | j d7 _ || j ks!|| j k rV| j �� I d H | j d8 _ || j k s:d S d S d S d S d S )Nr )Zchanged_settingsr� r3 rn ro rp r� � new_valuer6 rW rX rV rY �releaser[ )rE r� rX Znew_max_streamsr r r r� � s* ���
109
110 �
111 ��z4AsyncHTTP2Connection._receive_remote_settings_changec � s �| j �� I d H | j|= | j4 I d H �l | jr#| js#| �� I d H n<| jtjkrT| jsjtj | _| j
112 d ur?t � � }|| j
113 | _
114 | jru| �� I d H W d �I d H d S W d �I d H d S W d �I d H d S W d �I d H d S W d �I d H d S 1 I d H s�w Y d S �N)rY r� rA r; rB rU r7 r! r& r' r2 �time� monotonicr8 r? )rE rH �nowr r r rb � s, �
115 ���� .�z%AsyncHTTP2Connection._response_closedc � s( �| j �� tj| _| j�� I d H d S r� )r6 Zclose_connectionr! r( r7 r1 rU �rE r r r rU � s �
116 zAsyncHTTP2Connection.aclose�list[h2.events.Event]c
117 � s� �|j �di �}|�dd �}| jd ur| j�z| j�| j|�I d H }|dkr*td��W n ty? } z|| _d| _|�d }~ww | j �
118 |�}|S )N�timeout�read� zServer disconnectedT) rM r� rC r1 r� �READ_NUM_BYTESr � Exceptionr@ r6 Z receive_data)rE r �timeoutsr� r� re r� r r r r� � s$ �
119 � ��
120 z(AsyncHTTP2Connection._read_incoming_datac � s� �|j �di �}|�dd �}| j4 I d H �: | j�� }| jd ur#| j�z | j�||�I d H W n tyC } z|| _d| _ |�d }~ww W d �I d H d S 1 I d H sUw Y d S )Nr� �writeT)
121 rM r� r= r6 � data_to_sendrD r1 r� r� r@ )rE r r� r� r� re r r r rr � s" �
122
123 ���.�z)AsyncHTTP2Connection._write_outgoing_datac � sb �| j �|�}| j j}t||�}|dkr/| �|�I dH | j �|�}| j j}t||�}|dks|S )a
124 Returns the maximum allowable outgoing flow for a given stream.
125
126 If the allowable flow is zero, then waits on the network until
127 WindowUpdated frames have increased the flow rate.
128 https://tools.ietf.org/html/rfc7540#section-6.9
129 r N)r6 Zlocal_flow_control_windowZmax_outbound_frame_sizer� r� )rE r rH Z
130 local_flowZmax_frame_sizeZflowr r r r� � s �
131
132 �z,AsyncHTTP2Connection._wait_for_outgoing_flowr c C s
133 || j kS r� )r0 �rE r+ r r r rO � s
134 z'AsyncHTTP2Connection.can_handle_requestc C s2 | j tjko| j o| j o| jjjtj j
135 jk S r� ) r7 r! r( r@ r? r6 Z
136 state_machine�stater3 r4 ZConnectionStater� r r r � is_available� s �����z!AsyncHTTP2Connection.is_availablec C s t �� }| jd uo
137 || jkS r� )r� r� r8 )rE r� r r r � has_expired
138 s z AsyncHTTP2Connection.has_expiredc C � | j tjkS r� )r7 r! r' r� r r r �is_idle � z AsyncHTTP2Connection.is_idlec C r� r� )r7 r! r( r� r r r � is_closed r� z AsyncHTTP2Connection.is_closed�strc C s$ t | j�}|�d| jj� d| j� �S )Nz
139 , HTTP/2, �, Request Count: )r� r0 r7 �namer9 r� r r r �info s
140
141 ��zAsyncHTTP2Connection.infoc C s6 | j j}t| j�}d|� d|�d| jj� d| j� d� S )N�<z [z, r� z]>)� __class__r# r� r0 r7 r� r9 )rE �
142 class_namer+ r r r �__repr__ s
143 ��z AsyncHTTP2Connection.__repr__c � s �| S r� r r� r r r �
144 __aenter__& s �z AsyncHTTP2Connection.__aenter__�exc_type�type[BaseException] | None� exc_value�BaseException | None� traceback�types.TracebackType | Nonec � s �| � � I d H d S r� )rU )rE r� r� r� r r r � __aexit__) s �z AsyncHTTP2Connection.__aexit__r� )r+ r r, r r- r. )r r r r
145 )r r r rj )r r rH rs r rj )r r rH rs r� r� r rj )r r rH rs r r� )r r rH rs r r� )r r rH rs r r� )r r rH r� r rj )r� r� r rj )rH rs r rj �r rj )r r r r� )r r rH rs r rs )r+ r r r )r r )r r� )r r) )NNN)r� r� r� r� r� r� r rj )"r# r$ r% r� r3 r/ ZH2Configurationr5 rF ri rS r] r^ r~ r r_ r� r� r� r� rb rU r� rr r� rO r� r� r� r� r� r� r� r� r r r r r) * sD �
146 '
147 h
148 !
149
150
151
152
153
154 �
155 0
156
157
158
159
160
161
162
163
164
165
166
167
168 �r) c @ s* e Zd Zdd d
169 �Zdd d
170 �Zddd�ZdS )ra r4 r) r r rH rs r rj c C s || _ || _|| _d| _d S )NF)� _connection�_request�
171 _stream_id�_closed)rE r4 r rH r r r rF 3 s
172 z"HTTP2ConnectionByteStream.__init__r� c C s� �| j | jd�}z<tdt| j |�4 I d H �" | jj| j | jd�2 z 3 d H W }|V q 6 W d �I d H W d S 1 I d H s=w Y W d S tym } z t� � | �� I d H W d � |�1 sbw Y |�d }~ww )NrG Zreceive_response_body) r� r� r rR r� r� rT r
173 rU )rE rg r� re r r r � __aiter__; s& ���2�
174 ����z#HTTP2ConnectionByteStream.__aiter__c � s| �| j s<d| _ d| ji}tdt| j|�4 I d H � | jj| jd�I d H W d �I d H d S 1 I d H s5w Y d S d S )NTrH rN rI )r� r� r rR r� r� rb )rE rg r r r rU K s �
175 .��z HTTP2ConnectionByteStream.acloseN)r4 r) r r rH rs r rj )r r� r� )r# r$ r% rF r� rU r r r r ra 2 s
176
177 ra )r r r r )&�
178 __future__r �enum�loggingr� �typesr| Z h2.configr3 Z
179 h2.connectionZ h2.eventsZ
180 h2.exceptionsZ h2.settingsZ_backends.baser � _exceptionsr r r Z_modelsr r r
181 Z_synchronizationr r r
182 �_tracer Z
183 interfacesr � getLoggerrR r �IntEnumr! r) ra r r r r �<module> s4
184
185