Format: application/x-python-code
1 o
2
3 7'hh[ � @ 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� )�
7 NetworkStream)�ConnectionNotAvailable�LocalProtocolError�RemoteProtocolError)�Origin�Request�Response)�Lock� Semaphore�ShieldCancellation)�Trace� )�ConnectionInterfacezhttpcore.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 �JC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_sync/http2.py� <genexpr> s
8 � �
9 �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__�
10 __module__� __qualname__�ACTIVE�IDLE�CLOSEDr r r r r! $ s r! c @ s2 e Zd ZdZejjdd�Z dWdXd d �ZdYdd�Z dZdd�Z
11 d[dd�Z d[dd�Z d\d d �Z
12 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�HTTP2Connection� F)Zvalidate_inbound_headersN�originr �streamr �keepalive_expiry� float | Nonec C s� || _ || _|| _tjj| jd�| _tj | _
13 d | _ d| _ t
14 � | _t
15 � | _t
16 � | _t
17 � | _d| _d| _d| _i | _d | _d | _d | _d S )N)�configr F)�_origin�_network_stream�_keepalive_expiry�h2�
18 connectionZ H2Connection�CONFIG� _h2_stater! r' �_state�
19 _expire_at�_request_countr �
20 _init_lock� _state_lock�
21 _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& �
22 zHTTP2Connection.__init__r r r r
23 c C s� | � |jj�std|jj� d| j� ���| j�# | jtjtj fv r0| j
24 d7 _
25 d | _ tj| _nt � �W d � n1 s=w Y | j
26 �v | js�z%d|i}tdt||�� | jdi |�� W d � n1 shw Y W n& ty� } zt� �
27 | �� W d � |�1 s�w Y |�d }~ww d| _d| _| jjj}t|�| _t|| j �D ]}| j� � q�W d � n1 s�w Y | j� � z | j� � }g | j |< W n t j j!y� d| _"| j
28 d8 _
29 t � �w z|||d�}tdt||�� | j#||d� W d � n 1 �s w Y td t||�� | j$||d� W d � n 1 �s*w Y td
30 t||��}| j%||d�\} }
31 | |
32 f|_&W d � n 1 �sPw Y t'| |
33 t(| ||d �d | j)|d
34 �d�W S t�y� } zMt� �* d|i}tdt||�� | j*|d � W d � n 1 �s�w Y W d � n 1 �s�w Y t+|t j j,��r�| j-�r�t.| 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 �close� _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_requestU s� ��
43 � �� ��
44
45 �����
46 � �
47
48 �
49 � ��
50 � � ��
51 �� �
52
53 ��z HTTP2Connection.handle_request�Nonec C sf t jjdt jjjdt jjjdt jjjdid�| j_| jjt jjj = | j�
54 � | j� d� | � |� dS )z�
55 The HTTP/2 connection requires some initial setup before we can start
56 using individual request/response streams on it.
57 Tr �d r* )�clientZinitial_values� N)
58 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
59
60
61 �
62 ��
63 z%HTTP2Connection._send_connection_initrH �intc C s� t |� }dd� |jD �d }d|jfd|fd|jjfd|jjfgdd� |jD � }| jj|||d � | jjd
64 |d � | � |� d S )
65 z@
66 Send the request headers to a given stream ID.
67 c S s g | ] \}}|� � d kr|�qS )� hostr r r r r �
68 <listcomp>� s z9HTTP2Connection._send_request_headers.<locals>.<listcomp>r s :methods
69 :authoritys :schemes :pathc S s( g | ]\}}|� � d vr|� � |f�qS ))rt r r r r r r ru � s �
70 �)�
71 end_streamrm rI N)
72 r r �methodrP �scheme�targetr6 � send_headersrq rr )rE r rH rv � authorityr r r r r] � s
73
74
75 ���z%HTTP2Connection._send_request_headersc C sH t |�sdS t|jtj�sJ �|jD ] }| �|||� q| �||� dS )zP
76 Iterate over the request body sending it to a given stream ID.
77 N)r rc r, �typing�Iterable�_send_stream_data�_send_end_stream)rE r rH �datar r r r^ � s
78 z"HTTP2Connection._send_request_bodyr� �bytesc C s\ |r,| � ||�}tt|�|�}|d|� ||d� }}| j�||� | �|� |sdS dS )zI
79 Send a single chunk of data in one or more data frames.
80 N)�_wait_for_outgoing_flow�min�lenr6 Z send_datarr )rE r rH r� Zmax_flow�
81 chunk_size�chunkr r r r~ s
82 �z!HTTP2Connection._send_stream_datac C s | j �|� | �|� dS )z`
83 Send an empty data frame on on a given stream ID with the END_STREAM flag set.
84 N)r6 rv rr )rE r rH r r r r s z HTTP2Connection._send_end_stream�%tuple[int, list[tuple[bytes, bytes]]]c C s� | � ||�}t|tjj�rnqd}g }|jdusJ �|jD ] \}}|dkr0t|jddd��}q |�d�s<|� ||f� q ||fS ) zT
85 Return the response status code and headers for a given stream ID.
86 T�� Ns :status�ascii�ignore)�errors� :)
87 �_receive_stream_eventrc r3 �events�ResponseReceivedr rs �decode�
88 startswith�append)rE r rH �event� status_coder r r r r r r_ s �
89 �z!HTTP2Connection._receive_response�typing.Iterator[bytes]c c sv � | � ||�}t|tjj�r1|jdusJ �|jdus J �|j}| j�||� | � |� |jV n t|tjj
90 �r:dS q)z]
91 Iterator that returns the bytes of the response body for a given stream ID.
92 TN) r� rc r3 r� � DataReceivedZflow_controlled_lengthr� r6 Zacknowledge_received_datarr � StreamEnded)rE r rH r� �amountr r r �_receive_response_body5 s �
93
94 �z&HTTP2Connection._receive_response_body�Kh2.events.ResponseReceived | h2.events.DataReceived | h2.events.StreamEndedc C sN | j �|�s| �||� | j �|�r| j | �d�}t|tjj�r%t|��|S )z�
95 Return the next available event for a given stream ID.
96
97 Will read more data from the network if required.
98 r ) rA �get�_receive_events�poprc r3 r� � StreamResetr )rE r rH r� r r r r� G s �z%HTTP2Connection._receive_stream_event�
99 int | Nonec
100 C s> | j �� | jdur$| jj}|r |r ||kr | jd8 _t� �t| j��|du s.| j�|�s�| �|�}|D ]S}t |t
101 j j �r]t
102 dt|��}| �|� ||_W d � n1 sWw Y q5t |t
103 j jt
104 j jt
105 j jt
106 j jf�r~|j| jv r}| j|j �|� q5t |t
107 j j�r�|| _q5W d � n1 s�w Y | �|� dS )zp
108 Read some data from the network until we see one or more events
109 for a given stream ID.
110 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 sF
111
112
113 �
114 ���� �� �'z HTTP2Connection._receive_eventsr� � h2.events.RemoteSettingsChangedc C s� |j �tjjj�}|rIt|j| jj j
115 �}|rK|| j krM|| j kr1| j �
116 � | j d7 _ || j ks || j k rO| j �� | j d8 _ || j k s6d 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( ��
117
118
119 �
120
121 ��z/HTTP2Connection._receive_remote_settings_changec C s� | j �� | j|= | j�W | jr| js| �� n3| jtjkrC| jsStj | _| j
122 d ur4t � � }|| j
123 | _
124 | jr[| �� W d � d S W d � d S W d � d S W d � d S W d � d S 1 sfw 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*
125
126
127
128 ���� "�z HTTP2Connection._response_closedc C s | j �� tj| _| j�� d S r� )r6 Zclose_connectionr! r( r7 r1 rU �rE r r r rU � s
129 zHTTP2Connection.close�list[h2.events.Event]c
130 C s� |j �di �}|�dd �}| jd ur| j�z| j�| j|�}|dkr&td��W n ty; } z|| _d| _|�d }~ww | j �
131 |�}|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"
132 � ��
133 z#HTTP2Connection._read_incoming_datac C s� |j �di �}|�dd �}| j�4 | j�� }| jd ur | j�z | j�||� W n ty; } z|| _d| _ |�d }~ww W d � d S 1 sGw Y d S )Nr� �writeT)
134 rM r� r= r6 � data_to_sendrD r1 r� r� r@ )rE r r� r� r� re r r r rr � s
135
136 ���"�z$HTTP2Connection._write_outgoing_datac C sZ | j �|�}| j j}t||�}|dkr+| �|� | j �|�}| j j}t||�}|dks|S )a
137 Returns the maximum allowable outgoing flow for a given stream.
138
139 If the allowable flow is zero, then waits on the network until
140 WindowUpdated frames have increased the flow rate.
141 https://tools.ietf.org/html/rfc7540#section-6.9
142 r )r6 Zlocal_flow_control_windowZmax_outbound_frame_sizer� r� )rE r rH Z
143 local_flowZmax_frame_sizeZflowr r r r� � s
144
145
146 �z'HTTP2Connection._wait_for_outgoing_flowr c C s
147 || j kS r� )r0 �rE r+ r r r rO � s
148 z"HTTP2Connection.can_handle_requestc C s2 | j tjko| j o| j o| jjjtj j
149 jk S r� ) r7 r! r( r@ r? r6 Z
150 state_machine�stater3 r4 ZConnectionStater� r r r � is_available� s �����z HTTP2Connection.is_availablec C s t �� }| jd uo
151 || jkS r� )r� r� r8 )rE r� r r r � has_expired
152 s zHTTP2Connection.has_expiredc C � | j tjkS r� )r7 r! r' r� r r r �is_idle � zHTTP2Connection.is_idlec C r� r� )r7 r! r( r� r r r � is_closed r� zHTTP2Connection.is_closed�strc C s$ t | j�}|�d| jj� d| j� �S )Nz
153 , HTTP/2, �, Request Count: )r� r0 r7 �namer9 r� r r r �info s
154
155 ��zHTTP2Connection.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 �
156 class_namer+ r r r �__repr__ s
157 ��zHTTP2Connection.__repr__c C s | S r� r r� r r r � __enter__& s zHTTP2Connection.__enter__�exc_type�type[BaseException] | None� exc_value�BaseException | None� traceback�types.TracebackType | Nonec C s | � � d S r� )rU )rE r� r� r� r r r �__exit__) s zHTTP2Connection.__exit__r� )r+ r r, r r- r. )r r r r
158 )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 �
159 '
160 h
161 !
162
163
164
165
166
167 �
168 0
169
170
171
172
173
174
175
176
177
178
179
180
181 �r) c @ s* e Zd Zdd d
182 �Zdd d
183 �Zddd�ZdS )ra r4 r) r r rH rs r rj c C s || _ || _|| _d| _d S )NF)� _connection�_request�
184 _stream_id�_closed)rE r4 r rH r r r rF 3 s
185 z"HTTP2ConnectionByteStream.__init__r� c c s� � | j | jd�}z-tdt| j |�� | jj| j | jd�D ]}|V q W d � W d S 1 s.w Y W d S ty[ } zt� �
186 | �� W d � |�1 sPw Y |�d }~ww )NrG Zreceive_response_body) r� r� r rR r� r� rT r
187 rU )rE rg r� re r r r �__iter__; s& �
188 ��&�
189
190 ����z"HTTP2ConnectionByteStream.__iter__c C s` | j s.d| _ d| ji}tdt| j|�� | jj| jd� W d � d S 1 s'w Y d S d S )NTrH rN rI )r� r� r rR r� r� rb )rE rg r r r rU K s
191 "��z HTTP2ConnectionByteStream.closeN)r4 r) r r rH rs r rj )r r� r� )r# r$ r% rF r� rU r r r r ra 2 s
192
193 ra )r r r r )&�
194 __future__r �enum�loggingr� �typesr| Z h2.configr3 Z
195 h2.connectionZ h2.eventsZ
196 h2.exceptionsZ h2.settingsZ_backends.baser � _exceptionsr r r Z_modelsr r r
197 Z_synchronizationr r r
198 �_tracer Z
199 interfacesr � getLoggerrR r �IntEnumr! r) ra r r r r �<module> s4
200
201