File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/connector.cpython-310.pyc
o
���gA� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
d dl m Z d dl
mZ d dlmZmZmZ d dlmZ d dlmZ d dlmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z( d dl)Z)d d
l*m+Z+m,Z, d d l-m.Z.m/Z/ d d l0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z; d d
l<m=Z= d dl>m?Z?m@Z@mAZA d dl,mBZBmCZCmDZDmEZEmFZFmGZG d dlHmIZI er�d dlJZJeJjKZKnz d dlJZJeJjKZKW n
eLy� dZJeMZKY nw eNdh�ZOeNddh�ZPeNddh�ZQePeOB ZReReQB ZSdejT k�odk n �pejTdk ZUdZVe�r'd dlWmXZX d dl>mYZY d d lZm[Z[ G d d � d �Z\G d d � d �Z]G d!d"� d"�Z^G d#d$� d$�Z_G d%d&� d&�Z`d'ead(eKfd)d*�Zbebd+�Zcebd,�ZdG d-d.� d.e_�ZeG d/d0� d0e_�ZfG d1d2� d2e_�ZgdS )3� N)� OrderedDict� defaultdict�deque)�suppress)�
HTTPStatus)�chain�cycle�islice)� monotonic)�
TracebackType)�
TYPE_CHECKING�Any� Awaitable�Callable� DefaultDict�Deque�Dict�Iterator�List�Literal�Optional�Sequence�Set�Tuple�Type�Union�cast� )�hdrs�helpers)�AbstractResolver�
ResolveResult) �ClientConnectionError� ClientConnectorCertificateError�ClientConnectorDNSError�ClientConnectorError�ClientConnectorSSLError�ClientHttpProxyError�ClientProxyConnectionError�ServerFingerprintMismatch�UnixClientConnectorError� cert_errors�
ssl_errors)�ResponseHandler)�
ClientRequest� Fingerprint�_merge_ssl_params)� ceil_timeout�
is_ip_address�noop�sentinel�
set_exception�
set_result)�DefaultResolver� �http�https�ws�wss)� �
r )r= r> r )r= � � )�
BaseConnector� TCPConnector�
UnixConnector�NamedPipeConnector)�
ClientTimeout)�
ConnectionKey)�Tracec @ s> e Zd ZdZdee ddfdd�Zdefdd�Zd d d
�ZdS ) �_DeprecationWaiter��
_awaitable�_awaited� awaitable�returnNc C s || _ d| _d S �NFrI )�selfrL � rP �</usr/local/lib/python3.10/dist-packages/aiohttp/connector.py�__init__h �
z_DeprecationWaiter.__init__c C s d| _ | j�� S �NT)rK rJ � __await__�rO rP rP rQ rU l rS z _DeprecationWaiter.__await__c C s | j s t�dt� d S d S )NzDConnector.close() is a coroutine, please use await connector.close())rK �warnings�warn�DeprecationWarningrV rP rP rQ �__del__p s ��z_DeprecationWaiter.__del__�rM N) �__name__�
__module__� __qualname__� __slots__r r
rR rU rZ rP rP rP rQ rH e s
rH c
@ s� e Zd ZdZdddddedejddf
d d
�Zdefd d �Z e
fd
e ddfdd�Z de
d fdd�Zedejfdd��Zedeej fdd��Zedee fdd��Zdeg df ddfdd�Zd$d d �Zd$d d �Zd$d d!�Zedefd"d#��ZdS )%�
ConnectionN� connectorrA �keyrF �protocol�looprM c C s@ || _ || _|| _|| _g | _|�� r t�t� d��| _
d S d S )Nr ) �_key�
_connector�_loop� _protocol�
_callbacks� get_debug� traceback�
extract_stack�sys� _getframe�_source_traceback)rO ra rb rc rd rP rP rQ rR } s �zConnection.__init__c C s d| j � d�S )Nz Connection<�>)re rV rP rP rQ �__repr__� s zConnection.__repr__� _warningsc C s� | j d ur?d| i}|jd| ��tfi |�� | j�� r d S | jj| j| j dd� | dd�}| jd ur7| j|d< | j� |� d S d S )N�sourcezUnclosed connection T�� should_closezUnclosed connection)�client_connection�message�source_traceback)
rh rX �ResourceWarningrg � is_closedrf �_releasere ro �call_exception_handler)rO rr �kwargs�contextrP rP rQ rZ � s
�zConnection.__del__Tc C � dS )z9Force subclasses to not be falsy, to make checks simpler.TrP rV rP rP rQ �__bool__� s zConnection.__bool__c C s t jdtdd� | jS )Nz%connector.loop property is deprecated� ��
stacklevel)rW rX rY rg rV rP rP rQ rd � s �zConnection.loopc C s | j d u rd S | j jS �N)rh � transportrV rP rP rQ r� � s
zConnection.transportc C � | j S r� )rh rV rP rP rQ rc � s zConnection.protocol�callbackc C s |d ur | j �|� d S d S r� )ri �append)rO r� rP rP rQ � add_callback� s �zConnection.add_callbackc C sR | j d d � g }| _ |D ]}tt�� |� W d � n1 s!w Y q
d S r� )ri r � Exception)rO � callbacks�cbrP rP rQ �_notify_release� s
���zConnection._notify_releasec C s6 | � � | jd ur| jj| j| jdd� d | _d S d S )NTrt �r� rh rf r{ re rV rP rP rQ �close� s
�zConnection.closec C s2 | � � | jd ur| j�| j| j� d | _d S d S r� r� rV rP rP rQ �release� s
�zConnection.releasec C s | j d u p
| j �� S r� )rh � is_connectedrV rP rP rQ �closed� s zConnection.closedr[ ) r\ r] r^ ro r- �asyncio�AbstractEventLooprR �strrq rW r
rZ r r� �propertyrd r � Transportr� rc r r� r� r� r� �boolr� rP rP rP rQ r` y s8 ����
�
r` c @ s e Zd ZdZdZddd�ZdS )�_TransportPlaceholderz.placeholder for BaseConnector.connect functionrP rM Nc C r )z Close the placeholder transport.NrP rV rP rP rQ r� � s z_TransportPlaceholder.closer[ )r\ r] r^ �__doc__r_ r� rP rP rP rQ r� � s r� c @ s e Zd ZdZdZdZdZeZe ddddddd �d
e
e de f d e
d ed
ede
deej de ddfdd�Zefdeddfdd�ZdNdd�Zdeddfdd�ZdNd d �Z dOd eee d ee d ee ddfd!d"�Z e de
fd#d$��Z e defd%d&��Z e defd'd(��Z dPd)d*�Z!dPd+d,�Z"de#d fd-d.�Z$dPd/d0�Z%e de
fd1d2��Z&d3d4defd5d6�Z'd7e(d8e)d9 d:d;de*fd<d=�Z+d3d4d8e)d9 ddfd>d?�Z,d3d4d8e)d9 dee* fd@dA�Z-dPdBdC�Z.d3d4dDe/ddfdEdF�Z0ddG�d3d4dHe/dIe
ddfdJdK�Z1d7e(d8e)d9 d:d;de/fdLdM�Z2dS )QrA a\ Base connector class.
keepalive_timeout - (optional) Keep-alive timeout.
force_close - Set to True to force close and do reconnect
after each request (and between redirects).
limit - The total number of simultaneous connections.
limit_per_host - Number of simultaneous connections to one host.
enable_cleanup_closed - Enables clean-up closed ssl transports.
Disabled by default.
timeout_ceil_threshold - Trigger ceiling of timeout values when
it's above timeout_ceil_threshold.
loop - Optional event loop.
TNg @F�d r � ��keepalive_timeout� force_close�limit�limit_per_host�enable_cleanup_closedrd �timeout_ceil_thresholdr� r� r� r� r� rd r� rM c C s |r|d ur|t urtd��n|t u rd}|pt�� }|| _d| _|�� r.t�t �
d��| _ t t
�| _|| _|| _t� | _t t�| _tt|�| _|| _t t�| _|| _tj t |d�| _ d | _ d | _ |rst!sst"j#dt j$� �t%dd� d}| | _&g | _'| �(� d S ) Nz6keepalive_timeout cannot be set if force_close is Trueg .@Fr �rd zoenable_cleanup_closed ignored because https://github.com/python/cpython/pull/118960 is fixed in Python version r� r� ))r4 �
ValueErrorr� �get_running_loop�_timeout_ceil_threshold�_closedrj rk rl rm rn ro r r �_conns�_limit�_limit_per_host�set� _acquired�_acquired_per_hostr �float�_keepalive_timeout� _force_closer �_waitersrg � functools�partialr- �_factory�_cleanup_handle�_cleanup_closed_handle�NEEDS_CLEANUP_CLOSEDrW rX � version_inforY �_cleanup_closed_disabled�_cleanup_closed_transports�_cleanup_closed)rO r� r� r� r� r� rd r� rP rP rQ rR � sP �� �� ��� zBaseConnector.__init__rr c C s� | j rd S | js
d S dd� | j�� D �}| �� d| i}|jd| ��tfi |�� | |dd�}| jd ur9| j|d< | j�|� d S )Nc S s g | ]}t |��qS rP )�repr)�.0�crP rP rQ �
<listcomp>@ s z)BaseConnector.__del__.<locals>.<listcomp>rs zUnclosed connector zUnclosed connector)ra � connectionsrw rx ) r� r� �values�_closerX ry ro rg r| )rO rr �connsr} r~ rP rP rQ rZ : s �
zBaseConnector.__del__c C s t �dt� | S )NzH"with Connector():" is deprecated, use "async with Connector():" instead)rW rX rY rV rP rP rQ � __enter__O s
�zBaseConnector.__enter__�excc G s | � � d S r� )r� )rO r� rP rP rQ �__exit__W s zBaseConnector.__exit__c � s �| S r� rP rV rP rP rQ �
__aenter__Z s �zBaseConnector.__aenter__�exc_type� exc_value�
exc_tracebackc � s �| � � I d H d S r� )r� )rO r� r� r� rP rP rQ � __aexit__] s �zBaseConnector.__aexit__c C r� )z1Ultimately close connection on releasing if True.)r� rV rP rP rQ r� e � zBaseConnector.force_closec C r� )z�The total number for simultaneous connections.
If limit is 0 the connector has no limit.
The default limit size is 100.
)r� rV rP rP rQ r� j � zBaseConnector.limitc C r� )z�The limit for simultaneous connections to the same endpoint.
Endpoints are the same if they are have equal
(host, port, is_ssl) triple.
)r� rV rP rP rQ r� s r� z BaseConnector.limit_per_hostc C s� | j r | j �� d| _ t� }| j}| jr^tt�}|| }| j�� D ]9\}}t� }|D ])\}} |�� r@| | dkr@|� || f� q*|j
}
|� � | j sS|j
rS| j� |
� q*|rZ|||< q!|| _| jrptj| d|| j| jd�| _ dS dS )zCleanup unused transports.Nr �_cleanup�r� )r� �cancelr
r� r� r r �itemsr� r� r� r� r� �is_sslr� r �weakref_handlerg r� ) rO �now�timeoutr� �deadlinerb r� �alive�proto�use_timer� rP rP rQ r� | s@
�� ��zBaseConnector._cleanupc C s\ | j r| j �� | jD ]
}|dur|�� q g | _| js,tj| d| j| j| j d�| _ dS dS )z~Double confirmation for transport close.
Some broken ssl servers may leave socket open without proper close.
Nr� r� )
r� r� r� �abortr� r r� �_cleanup_closed_periodrg r� )rO r� rP rP rQ r� � s
� ��z BaseConnector._cleanup_closedc C s | � � tt� �S )z Close all opened transports.)r� rH r3 rV rP rP rQ r� � s
zBaseConnector.closec
C s� | j rd S d| _ z�| j�� r<W | j�� | j�� | j�� D ] }|D ]}|�� q"q | j�� d | _ | j
�� d | _ d S | j rD| j �� | j rL| j �� | j�� D ]
}|D ]\}}|� � qUqQ| jD ]}|� � qb| j
D ]
}|d urv|�
� qlW | j�� | j�� | j�� D ] }|D ]}|�� q�q�| j�� d | _ | j
�� d | _ d S | j�� | j�� | j�� D ] }|D ]}|�� q�q�| j�� d | _ | j
�� d | _ w rT )r� rg rz r� �clearr� r� r� r� r� r� r� r� r� )rO �
keyed_waiters� keyed_waiter�datar� �t0r� rP rP rQ r� � sf
�
�
�
��
�
�
�
zBaseConnector._closec C r� )z;Is connector closed.
A readonly property.
)r� rV rP rP rQ r� � s zBaseConnector.closedrb rF c C s\ d}| j r| j t| j� }dkr|S | j }r,| j�|� }r&|t|�8 }||kr,|S |S )z�
Return number of available connections.
The limit, limit_per_host and the connection key are taken into account.
If it returns less than 1 means that there are no connections
available.
r r )r� �lenr� r� r� �get)rO rb � total_remain� host_remain�acquiredrP rP rQ �_available_connections� s
z$BaseConnector._available_connections�req�tracesrG r� rE c
� s� �|j }| �||�I dH }dur|S t|j|j�4 I dH �� | �|�dkrH| �||�I dH | �||�I dH }durH|W d �I dH S ttt � �}| j
� |� | j r_| j
| � |� z(|rn|D ] }|�� I dH qd| �|||�I dH }|r�|D ] }|�� I dH q|W n ty� | �||� � w | jr�|�� td��W d �I dH n 1 I dH s�w Y | j
�|� | j
� |� | j r�| j
| } | �|� | � |� t| ||| j�S )z'Get from pool or create new connection.Nr zConnector is closed.)�connection_key�_getr1 �connect�ceil_thresholdr� � _wait_for_available_connectionr r- r� r� �addr� r� � send_connection_create_start�_create_connection�send_connection_create_end�
BaseException�_release_acquiredr� r� r" �remover` rg )
rO r� r� r� rb �conn� placeholder�tracer� �acquired_per_hostrP rP rQ r� sL �� � �
(� #
zBaseConnector.connectc
� s� �d} | j �� }| j| }d||< |r|j|dd� z4|r*|D ] }|�� I dH q |I dH |r=|D ] }|�� I dH q3W |�|d� | j�|d�sO| j|= n|�|d� | j�|d�sa| j|= w | �|�dkrkdS |d7 }q)z&Wait for an available connection slot.r TNF��lastr ) rg �
create_futurer� � move_to_end� send_connection_queued_start�send_connection_queued_end�popr� r� )rO rb r� �attempts�futr� r� rP rP rQ r� = s8 �
� � �
�z,BaseConnector._wait_for_available_connectionc � s� �| j �|� }du r
dS t� }|ru|�� \}}|�� r`|| | jkr`|s)| j |= | j�|� | jr:| j | �|� |rX|D ]}z |�
� I dH W q> t yW | � ||� � w t
| ||| j�S |j}|�� | jss|jrs| j�|� |s| j |= dS )znGet next reusable connection for the key or None.
The connection will be marked as acquired.
N)r� r� r
�popleftr� r� r� r� r� r� �send_connection_reuseconnr� r� r` rg r� r� r� r� r� r� ) rO rb r� r� �t1r� r� r� r� rP rP rQ r� g s8 � � � zBaseConnector._getc C sv | j sdS t| j �}t�|� |D ]'}| �|�dk rq| j | }|r8|jdd�\}}|�� s6|�d� dS |s"qdS )z�
Iterates over all waiters until one to be released is found.
The one to be released is not finished and
belongs to a host that has available connections.
Nr Fr� )r� �list�random�shuffler� �popitem�doner6 )rO �queuesrb �waiters�waiter�_rP rP rQ �_release_waiter� s
���z BaseConnector._release_waiterr� c C sN | j rdS | j�|� | jr!| j�|� }r!|�|� |s!| j|= | �� dS )z Release acquired connection.N)r� r� �discardr� r� r� r )rO rb r� r� rP rP rQ r� � s
z BaseConnector._release_acquiredrt rc ru c C s� | j rd S | �||� | js|s|jr(|j}|�� |jr&| js&| j� |� d S | j
| � |t � f� | j d u rHt
j| d| j| j| jd�| _ d S d S )Nr� r� )r� r� r� ru r� r� r� r� r� r� r� r
r� r r� r� rg r� )rO rb rc ru r� rP rP rQ r{ � s&
��zBaseConnector._releasec � s �t � �r� )�NotImplementedError)rO r� r� r� rP rP rQ r� � s �z BaseConnector._create_connection)rM rA )NNNr[ )3r\ r] r^ r� r� ro r� �HIGH_LEVEL_SCHEMA_SET�allowed_protocol_schema_setr4 r �objectr� r� �intr r� r� rR rW r
rZ r� r� r� r r� r r� r� r� r� r� r� r� r r� r� r� r� r. r r` r� r� r� r r- r� r{ r� rP rP rP rQ rA � s� � ������ �
�I
�
���
�
'
(���
�5��
�*��
�
*����
� ����rA c @ s� e Zd Zddee ddfdd�Zdedefdd�Zd e e
e f d
e e
ddfd d �Zd e e
e f ddfd
d�Zddd�Zd e e
e f de e
fdd�Zd e e
e f defdd�ZdS )�_DNSCacheTableN�ttlrM c C s i | _ i | _|| _d S r� )� _addrs_rr� _timestamps�_ttl)rO r rP rP rQ rR � s
z_DNSCacheTable.__init__�hostc C s
|| j v S r� )r )rO r rP rP rQ � __contains__� s
z_DNSCacheTable.__contains__rb �addrsc C s4 t |�t|�f| j|< | jd urt� | j|< d S d S r� )r r� r r r
r )rO rb r rP rP rQ r� � s
�z_DNSCacheTable.addc C s. | j �|d � | jd ur| j�|d � d S d S r� )r r r r �rO rb rP rP rQ r� � s
�z_DNSCacheTable.removec C s | j �� | j�� d S r� )r r� r rV rP rP rQ r� � s
z_DNSCacheTable.clearc C s( | j | \}}tt||��}t|� |S r� )r r r �next)rO rb rd �lengthr rP rP rQ �
next_addrs� s z_DNSCacheTable.next_addrsc C s$ | j d u rdS | j| | j t� k S rN )r r r
r rP rP rQ �expired� s
z_DNSCacheTable.expiredr� r[ )r\ r] r^ r r� rR r r� r r r� r r r! r� r� r� r! r" rP rP rP rQ r � s "
r �verifiedrM c C sz t du rdS | r
t �� }n)t �t j�}| jt jO _| jt jO _d|_t j|_ | jt j
O _|� � |� d� |S )z�Create SSL context.
This method is not async-friendly and should be called from a thread
because it will load certificates from disk and do other blocking I/O.
NF)zhttp/1.1)
�ssl�create_default_context�
SSLContext�PROTOCOL_TLS_CLIENT�options� OP_NO_SSLv2� OP_NO_SSLv3�check_hostname� CERT_NONE� verify_mode�OP_NO_COMPRESSION�set_default_verify_paths�set_alpn_protocols)r# �
sslcontextrP rP rQ �_make_ssl_context s
r2 TFc & sJ e Zd ZdZeedh�B Zddddejj dddde
dddddd d
dd �d e d
e e
de de e dejde e dee eef de eeef de e dedeef de dedede de ej ded e e d e e f$� fd d �Zd ed f� fd!d"� Ze d efd#d$��Z e d e fd%d&��Z dZd'e e d(e e d dfd)d*�Z d[d'ed(ed+e e d, d e!e" fd-d.�Z#d/eeef d'ed(ed0e$d1 d+e e d, d e!e" f d2d3�Z%d4e&d+e!d, d5d6d e'fd7d8�Z(d4e&d e e fd9d:�Z)d4e&d e d; fd<d=�Z*e+d>�d?e,d@e!e-j. d4e&d5d6dAe/e0 dBe,d eej1e'f fdCdD�Z2e+d>�d?e,d4e&d5d6dAe/e0 dBe,d eej1e'f f dEdF�Z3d\dHdI�Z4d]dJdK�Z5d e fdLdM�Z6dNej1d4e&d dfdOdP�Z7e+fdNej1d4e&d5d6dAe/e0 d eej8e'f f
dQdR�Z9dSe!e" d e!e-j. fdTdU�Z:e+d>�d4e&d+e!d, d5d6dAe/e0 d eej1e'f f
dVdW�Z;d4e&d+e!d, d5d6d eej8e'f fdXdY�Z<� Z=S )^rB u- TCP connector.
verify_ssl - Set to True to check ssl certifications.
fingerprint - Pass the binary sha256
digest of the expected certificate in DER format to verify
that the certificate the server presents matches. See also
https://en.wikipedia.org/wiki/HTTP_Public_Key_Pinning
resolver - Enable DNS lookups and use this
resolver
use_dns_cache - Use memory cache for DNS lookups.
ttl_dns_cache - Max seconds having cached a DNS entry, None forever.
family - socket address family
local_addr - local tuple of (host, port) to bind socket to
keepalive_timeout - (optional) Keep-alive timeout.
force_close - Set to True to force close and do reconnect
after each request (and between redirects).
limit - The total number of simultaneous connections.
limit_per_host - Number of simultaneous connections to one host.
enable_cleanup_closed - Enables clean-up closed ssl transports.
Disabled by default.
happy_eyeballs_delay - This is the “Connection Attempt Delay”
as defined in RFC 8305. To disable
the happy eyeballs algorithm, set to None.
interleave - “First Address Family Count” as defined in RFC 8305
loop - Optional event loop.
�tcpTN�
Fr� r r� g �?)�
verify_ssl� fingerprint�
use_dns_cache�
ttl_dns_cache�family� ssl_contextr$ �
local_addr�resolverr� r� r� r� r� rd r� �happy_eyeballs_delay�
interleaver5 r6 r7 r8 r9 r: r$ r; r< r� r� r� r� r� rd r� r= r> c s� t � j|
| | |
|||d� t||||�| _| d u r t| jd�} | | _|| _t|d�| _ i | _
|| _ t �
|�| _|| _|| _t� | _d S )Nr� r� )r )�superrR r0 �_sslr7 rg � _resolver�_use_dns_cacher �
_cached_hosts�_throttle_dns_futures�_family�aiohappyeyeballs�addr_to_addr_infos�_local_addr_infos�_happy_eyeballs_delay� _interleaver� �_resolve_host_tasks)rO r5 r6 r7 r8 r9 r: r$ r; r< r� r� r� r� r� rd r� r= r> �� __class__rP rQ rR C s, �
� zTCPConnector.__init__rM c s< t �| j�� �D ]}|�� q| jD ]}|�� qt� �� S )z Close all ongoing DNS calls.)r �
from_iterablerD r� r� rK r? r� )rO r �trL rP rQ r� s s
zTCPConnector.closec C r� )zSocket family like AF_INET.)rE rV rP rP rQ r9 } r� zTCPConnector.familyc C r� )z%True if local DNS caching is enabled.)rB rV rP rP rQ r7 � r� zTCPConnector.use_dns_cacher �portc C sJ |dur|dur| j �||f� dS |dus|dur td��| j �� dS )z8Remove specified host/port or clear all dns local cache.Nz5either both host and port or none of them are allowed)rC r� r� r� )rO r rP rP rP rQ �clear_dns_cache� s
z TCPConnector.clear_dns_cacher� rG c � s� �t |�r|||| jddd�gS | js@|r"|D ]
}|�|�I dH q| jj||| jd�I dH }|r>|D ]
}|�|�I dH q3|S ||f}|| jv rf| j�|�sf| j� |�}|rd|D ]
}|�
|�I dH qY|S || j v r�| j | }| j �
� } |�| � |r�|D ]
}|�
|�I dH q~z | I dH W |�| � n|�| � w | j� |�S t� | j |< }| �|||||�}
t�� } tjdkr�tj|
| dd�} n| �|
�} | �� s�| j�| � | �| jj� z t�| �I dH W S tj y� d
d d �}
| �|
� � w )z*Resolve host and return list of addresses.r )�hostnamer rP r9 r� �flagsN�r9 )r= r? T)rd � eager_startr �#asyncio.Future[List[ResolveResult]]rM c S s: t ttj��
| �� W d � d S 1 sw Y d S r� )r r� r� �CancelledError�result)r rP rP rQ �drop_exception� s
"�z2TCPConnector._resolve_host.<locals>.drop_exception)r rV rM N) r2 rE rB �send_dns_resolvehost_startrA �resolve�send_dns_resolvehost_endrC r" r! �send_dns_cache_hitrD rg r� r� r r� �_resolve_host_with_throttler� r� rm r� �Task� create_taskr
rK �add_done_callback�shieldrW )rO r rP r� r� �resrb rX �futures�future�corord �resolved_host_taskrY rP rP rQ �
_resolve_host� sj ���
�zTCPConnector._resolve_hostrb rd zasyncio.Future[None]c
� s� �|r|D ]
}|� |�I dH qz^z>|r!|D ]
}|�|�I dH q| jj||| jd�I dH }|r=|D ]
}|�|�I dH q2| j�||� |D ]}t|d� qFW n t yf } z |D ]}t
|| � qY� d} ~ ww W | j � |� n| j � |� w | j�
|�S )z�Resolve host and set result for all waiters.
This method must be run in a task and shielded from cancellation
to avoid cancelling the underlying lookup.
NrT )�send_dns_cache_missrZ rA r[ rE r\ rC r� r6 r� r5 rD r r! )
rO rb r rP rd r� r� r r �erP rP rQ r^ � s2 �
� ���
z(TCPConnector._resolve_host_with_throttler� r� rE c � s@ �|j r| �|||�I dH \}}|S | �|||�I dH \}}|S )zcCreate connection.
Has same keyword arguments as BaseEventLoop.create_connection.
N)�proxy�_create_proxy_connection�_create_direct_connection)rO r� r� r� r r� rP rP rQ r� s ��z TCPConnector._create_connectionc C sd |� � sdS tdu rtd��|j}t|tj�r|S |dur tS | j}t|tj�r*|S |dur0tS tS )a* Logic to get the correct SSL context
0. if req.ssl is false, return None
1. if ssl_context is specified in req, use it
2. if _ssl_context is specified in self, use it
3. otherwise:
1. if verify_ssl is not specified in req, use self.ssl_context
(will generate a default context according to self.verify_ssl)
2. if verify_ssl is True in req, generate a default SSL context
3. if verify_ssl is False in req, generate a SSL context that
won't verify
NzSSL is not supported.T)r� r$ � RuntimeError�
isinstancer& �_SSL_CONTEXT_UNVERIFIEDr@ �_SSL_CONTEXT_VERIFIED)rO r� r1 rP rP rQ �_get_ssl_context$ s z TCPConnector._get_ssl_contextr/ c C s, |j }t|t�r
|S | j}t|t�r|S d S r� )r$ ro r/ r@ )rO r� �retrP rP rQ �_get_fingerprintE s
z TCPConnector._get_fingerprint�� client_error�args�
addr_infosrv r} c
� s �zGt |j|jd�4 I d H �- tj|| j| j| j| jd�I d H }| jj |i |�d|i��I d H W d �I d H W S 1 I d H sAw Y W d S t
y[ } zt |j |�|�d }~w t
ym } zt|j |�|�d }~w ty� } z|jd u r�t|tj�r�� ||j |�|�d }~ww )N�r� )rx �local_addr_infosr= r> rd �sock)r1 � sock_connectr� rF �start_connectionrH rI rJ rg �create_connectionr+ r# r� r, r& �OSError�errnoro r� � TimeoutError) rO rx r� r� rv rw r} r{ r� rP rP rQ �_wrap_create_connectionN s4 � � � 4� ����z$TCPConnector._wrap_create_connectionc
� s� �z2t |j|jd�4 I d H � | jj|i |��I d H W d �I d H W S 1 I d H s,w Y W d S tyF } zt|j|�|�d }~w tyX } zt |j|�|�d }~w t
yv } z|j d u rkt |t
j�rk� ||j|�|�d }~ww �Nry )r1 r| r� rg r~ r+ r# r� r, r& r r� ro r� r� )rO r� r� rv rw r} r� rP rP rQ �_wrap_existing_connectionl s& ��4�����z&TCPConnector._wrap_existing_connectionr. c C s8 |� � sdS |j}|dusJ �|jdkrdS | �� dS )z�Raise a :py:exc:`RuntimeError` on missing ``start_tls()``.
It is necessary for TLS-in-TLS so that it is possible to
send HTTPS queries through HTTPS proxies.
This doesn't affect regular HTTP requests, though.
Nr: )r� rk �scheme�_check_loop_for_start_tls)rO r� � proxy_urlrP rP rQ �_fail_on_no_start_tls� s
z"TCPConnector._fail_on_no_start_tlsc
C s2 z| j j W d S ty } ztd�|�d }~ww )Na� An HTTPS request is being sent through an HTTPS proxy. This needs support for TLS in TLS but it is not implemented in your runtime for the stdlib asyncio.
Please upgrade to Python 3.11 or higher. For more details, please see:
* https://bugs.python.org/issue37179
* https://github.com/python/cpython/pull/28073
* https://docs.aiohttp.org/en/stable/client_advanced.html#proxy-support
* https://github.com/aio-libs/aiohttp/discussions/6044
)rg � start_tls�AttributeErrorrn )rO �attr_excrP rP rQ r� � s � ���z&TCPConnector._check_loop_for_start_tlsc C s$ z| � � W dS ty Y dS w )NFT)r� rn rV rP rP rQ �_loop_supports_start_tls� s
��z%TCPConnector._loop_supports_start_tls�underlying_transportc C s< |j jjdkr dS t|dd�}|rdS tjdt| dd� dS )z6Issue a warning if the requested URL has HTTPS scheme.r: N�_start_tls_compatibleFa An HTTPS request is being sent through an HTTPS proxy. This support for TLS in TLS is known to be disabled in the stdlib asyncio (Python <3.11). This is why you'll probably see an error in the log below.
It is possible to enable it via monkeypatching. For more details, see:
* https://bugs.python.org/issue37179
* https://github.com/python/cpython/pull/28073
You can temporarily patch this as follows:
* https://docs.aiohttp.org/en/stable/client_advanced.html#proxy-support
* https://github.com/aio-libs/aiohttp/discussions/6044
r= )rs r� )� request_info�urlr� �getattrrW rX �RuntimeWarning)rO r� r� �asyncio_supports_tls_in_tlsrP rP rQ �_warn_about_tls_in_tls� s �
�z#TCPConnector._warn_about_tls_in_tlsc � s �| � � }| �|�}tr|dusJ �zpt|j|jd�4 I dH �W z| jj||||jp,|j |j
d�I dH }W n t yA |� � � w t
|tj�rl| �|�}|rlz|�|� W n tyk |� � | jsj| j�|� � w W d �I dH n 1 I dH s|w Y W nj ty� } zt|j| �| �d} ~ w ty� } zt|j| �| �d} ~ w ty� } z| j du r�t
| tj �r�� ||j| �| �d} ~ w t y� }
z t d|j �d|j d�d|j!�d|j"p�d � d
|
�d � �|
�d}
~
ww |du r�d } ||jt| ���|�#|� ||fS )
z$Wrap the raw TCP transport with TLS.Nry )�server_hostname�ssl_handshake_timeoutz2Cannot initialize a TLS-in-TLS connection to host �:�dz4 through an underlying connection to an HTTPS proxy z ssl:�defaultz [�]z:Failed to start TLS (possibly caused by closing transport))$r� rr r r1 r| r� rg r� r� r �totalr� r� ro r� r� rt �checkr) r� r� r� r+ r# r� r, r& r r� r� � TypeErrorr" rP rk r$ �connection_made) rO r� r� r� rv � tls_protor1 �
tls_transportr6 r� �type_err�msgrP rP rQ �_start_tls_connection� s� �
�
� �
��(�������������� �z"TCPConnector._start_tls_connection�hostsc C s| g }|D ]7}|d }d|v }|rt jnt j}| jr | j|kr q|r)||d ddfn||d f}|�|t jt jd|f� q|S )z�Converts the list of hosts to a list of addr_infos.
The list of hosts is the result of a DNS lookup. The list of
addr_infos is the result of a call to `socket.getaddrinfo()`.
r r� rP r r8 )�socket�AF_INET6�AF_INETrE r� � SOCK_STREAM� IPPROTO_TCP)rO r� rx �hinfor �is_ipv6r9 �addrrP rP rQ � _convert_hosts_to_addr_infos s �z)TCPConnector._convert_hosts_to_addr_infosc
� s� �| � |�}| �|�}|jj}|d usJ �|�d�r!|�d�d }|j}|d us*J �z
| j|||d�I d H } W n tyV }
z|
j d u rKt
|
t j �rK� t
|j|
�|
�d }
~
ww d } | �| �} | r�|rj|jpf|�d�nd }
z| j| j||| |
||d�I d H \}}W n tt j fy� }
z|
} t�| | j� W Y d }
~
q^d }
~
ww |�� r�|r�z
|�|� W ||fS ty� }
z$|�� | js�| j � |� |
} |� d�}|� � }t� | |� W Y d }
~
q^d }
~
ww ||fS | d us�J �| �)Nz..�.)r� )r� r$ rx r� r� rv r� )!rr rt r� �raw_host�endswith�rstriprP rh r r� ro r� r� r$ r� r� r� r� r� r% rF �pop_addr_infos_interleaverJ r� r� r) r� r� r� r� �get_extra_info� getpeername�remove_addr_infos)rO r� r� r� rv r1 r6 r rP r� r� �last_excrx r� �transpr� r{ �bad_peerrP rP rQ rm / sp �
��
�� �� �
�� z&TCPConnector._create_direct_connectionc � s� �| � |� | �� }i }|jd ur|j}|jtj |tj< |j}|d us&J �ttj|||j | j
|j d�}| j |g |t
d�I d H \}} |j�tjd �}
|
d ur_|�� sY|
|jtj< n|
|jtj< |�� �rF|rl| �||� tj|_|j|_|jjd d d d�} t| | | | j
�} |�| �I d H }
z�z| j}|d us�J �|j|| j d� |
� | �I d H }W n t y� |
� � | � � � w d | _zMz6|j dkr�|j!}|d u r�t"|j �j#}t$|
j%|j&|j ||jd��|s�|j'dd d�}|d u r�t(d ��|�)� }W n
t �y |� � � w W |�s
|� � n |�s|� � w w |�s1| �*|�}| j+| j,||||j-|d
�I d H W |
� � S | j.|||d �I d H W |
� � S |
� � w || fS ) N)�headers�authrd r$ ru )rk �
proxy_auth�proxy_headers_hash)�read_until_eofr� �� )�statusrw r� r� )r� z)Transport does not expose socket instance)r� r$ r{ r� r� )r� r� )/r� r� �
proxy_headersr� r �HOSTrk r. �METH_GETr� rg r$ rm r( r �
AUTHORIZATIONr� �PROXY_AUTHORIZATIONr� � METH_CONNECT�methodr� r� �_replacer` �sendrh �set_response_paramsr� �startr� r� r� �reasonr �phraser' r� �historyr� rn �duprr r� r� r r� )rO r� r� r� �runtime_has_start_tlsr� r� � proxy_reqr� r� r� rb r� �
proxy_resprc �resprw �rawsockr1 rP rP rQ rl z s� �
�
�
� � �
�������
�
�
� �z%TCPConnector._create_proxy_connection)NNr� )r� r. rM Nr[ )>r\ r] r^ r� r � frozensetr r� �
AddressFamily� AF_UNSPECr4 r� r �bytesr r& r r/ r r� r r� r r� r� rR r r� r� r9 r7 rQ r r r! rh r r^ r. r- r� rr rt r% r
rF � AddrInfoTyper r� r� r� r� r� r� r� r� �
BaseTransportr� r� rm rl �
__classcell__rP rP rL rQ rB $ sn ������� �
� � �
��������0
���
� ���
�
�[
����
�
�)���
�!�������
�#������
�
��
�*�����
�H�
������
�K��� �rB c � e Zd ZdZeedh�B Zdedddfdede d e
e e df d
e
d e
d eej d
df� fdd�
Zed
efdd��Zdeded ddd
efdd�Z� ZS )rC a� Unix socket connector.
path - Unix socket path.
keepalive_timeout - (optional) Keep-alive timeout.
force_close - Set to True to force close and do reconnect
after each request (and between redirects).
limit - The total number of simultaneous connections.
limit_per_host - Number of simultaneous connections to one host.
loop - Optional event loop.
�unixFr� r N�pathr� r� r� r� rd rM c s t � j|||||d� || _d S )N�r� r� r� r� rd )r? rR �_path�rO r� r� r� r� r� rd rL rP rQ rR s �
zUnixConnector.__init__c C r� )zPath to unix socket.�r� rV rP rP rQ r� r� zUnixConnector.pathr� r� rG r� rE c
� s� �z5t |j|jd�4 I d H � | j�| j| j�I d H \}}W d �I d H W |S 1 I d H s/w Y W |S tyW } z|jd u rJt |t
j �rJ� t | j
|j|�|�d }~ww r� )r1 r| r� rg �create_unix_connectionr� r� r r� ro r� r� r* r� r� �rO r� r� r� r r� r� rP rP rQ r� s$ ���� � ���z UnixConnector._create_connection�r\ r] r^ r� r r� r r4 r� r� r r r� r r r� r� rR r� r� r. r r- r� r� rP rP rL rQ rC � sD ��� ���������rC c r� )rD a� Named pipe connector.
Only supported by the proactor event loop.
See also: https://docs.python.org/3/library/asyncio-eventloop.html
path - Windows named pipe path.
keepalive_timeout - (optional) Keep-alive timeout.
force_close - Set to True to force close and do reconnect
after each request (and between redirects).
limit - The total number of simultaneous connections.
limit_per_host - Number of simultaneous connections to one host.
loop - Optional event loop.
�npipeFr� r Nr� r� r� r� r� rd rM c s6 t � j|||||d� t| jtj�std��|| _d S )Nr� z9Named Pipes only available in proactor loop under windows)r? rR ro rg r� �ProactorEventLooprn r� r� rL rP rQ rR A s ���
zNamedPipeConnector.__init__c C r� )zPath to the named pipe.r� rV rP rP rQ r� Y r� zNamedPipeConnector.pathr� r� rG r� rE c
� s� �z:t |j|jd�4 I d H �! | j�| j| j�I d H \}}t�d�I d H W d �I d H n 1 I d H s5w Y W n t yZ } z|j
d u rOt |tj �rO� t
|j|�|�d }~ww tt|�S )Nry r )r1 r| r� rg �create_pipe_connectionr� r� r� �sleepr r� ro r� r% r� r r- r� rP rP rQ r� ^ s$ ���(��
��
z%NamedPipeConnector._create_connectionr� rP rP rL rQ rD 0 sD ��� ���������rD )hr� r� r r� rm rk rW � collectionsr r r �
contextlibr r9 r � itertoolsr r r �timer
�typesr �typingr r
r r r r r r r r r r r r r r r rF r8 r r �abcr r! �client_exceptionsr" r# r$ r% r& r' r( r) r* r+ r, � client_protor- �
client_reqrepr. r/ r0 r1 r2 r3 r4 r5 r6 r< r7 r$ r& � ImportErrorr r� �EMPTY_SCHEMA_SET�HTTP_SCHEMA_SET�
WS_SCHEMA_SET�HTTP_AND_EMPTY_SCHEMA_SETr r� r� �__all__�clientrE rF �tracingrG rH r` r� rA r r� r2 rq rp rB rC rD rP rP rP rQ �<module> s� L4
�
� X ' Z7