File: //usr/local/CyberCP/lib/python3.10/site-packages/tornado/__pycache__/tcpclient.cpython-310.pyc
o
�hx/ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZm Z ddl
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ dd lmZmZmZmZmZmZmZmZ ej r]dd
lm Z d Z G d d
� d
e �Z G dd� de �Z!dS )z'A non-blocking TCP connection factory.
� N)�Future�future_add_done_callback)�IOLoop)�IOStream)�gen)�Resolver)� TimeoutError)�Any�Union�Dict�Tuple�List�Callable�Iterator�Optional)�Setg333333�?c
@ sV e Zd ZdZdee deejegee df f ddfdd�Z
e dee deeeejef eeejef f fd d
��Z e
dfd ed eeeejf dd
fdd�Zdeeejef ddfdd�Zdeeejef dejdeddddf
dd�Zd eddfdd�Zd&dd�Zd&d d �Zd eeejf ddfd d �Zd&d d!�Zd&d"d#�Z d&d$d%�Z dS )'�
_Connectora� A stateless implementation of the "Happy Eyeballs" algorithm.
"Happy Eyeballs" is documented in RFC6555 as the recommended practice
for when both IPv4 and IPv6 addresses are available.
In this implementation, we partition the addresses by family, and
make the first connection attempt to whichever address was
returned first by ``getaddrinfo``. If that connection fails or
times out, we begin a connection in parallel to the first address
of the other family. If there are additional failures we retry
with other addresses, keeping one connection attempt per family
in flight at a time.
http://tools.ietf.org/html/rfc6555
�addrinfo�connect�Future[IOStream]�returnNc C sR t �� | _|| _t� | _d | _d | _d | _t |�| _
| � |�\| _ | _
t� | _d S �N)r �current�io_loopr r �future�timeout�connect_timeout�
last_error�len� remaining�split�
primary_addrs�secondary_addrs�set�streams)�selfr r � r& �D/usr/local/CyberCP/lib/python3.10/site-packages/tornado/tcpclient.py�__init__; s
�
z_Connector.__init__c C sP g }g }| d d }| D ]\}}||kr |� ||f� q |� ||f� q ||fS )a� Partition the ``addrinfo`` list by address family.
Returns two lists. The first list contains the first entry from
``addrinfo`` and all others with the same family, and the
second list contains all other addresses (normally one list will
be AF_INET and the other AF_INET6, although non-standard resolvers
may return additional families).
r )�append)r �primary� secondary�
primary_af�af�addrr& r& r' r O s z_Connector.splitr r z2Future[Tuple[socket.AddressFamily, Any, IOStream]]c C s2 | � t| j�� | �|� |d ur| �|� | jS r )� try_connect�iterr! � set_timeout�set_connect_timeoutr )r% r r r& r& r' �starth s
z_Connector.start�addrsc C s� zt |�\}}W n ty' | jdkr$| j�� s$| j�| jp"td�� Y d S w | �||�\}}| j �
|� t |t �
| j|||�� d S )Nr zconnection failed)�next�
StopIterationr r �done�
set_exceptionr �IOErrorr r$ �addr � functools�partial�on_connect_done)r% r4 r- r. �streamr r& r& r' r/ s s �� �z_Connector.try_connectr- r. r c
C s� | j d8 _ z|�� }W n7 tyD } z+| j�� r!W Y d }~d S || _| �|� | jd ur9| j� | j� | �
� W Y d }~d S d }~ww | � � | j�� rT|� � d S | j
�|� | j�|||f� | �� d S )N� )r �result� Exceptionr r7 r r/ r r �remove_timeout�
on_timeout�clear_timeouts�closer$ �discard�
set_result�
close_streams)r% r4 r- r. r r> �er& r& r' r= � s(
��
z_Connector.on_connect_donec C s | j �| j �� | | j�| _d S r )r � add_timeout�timerC r )r% r r& r& r' r1 � s
�z_Connector.set_timeoutc C s( d | _ | j�� s| �t| j�� d S d S r )r r r7 r/ r0 r" �r% r& r& r' rC � s
�z_Connector.on_timeoutc C s | j d ur| j�| j � d S d S r )r r rB rL r& r& r' �
clear_timeout� s
�z_Connector.clear_timeoutc C s | j �|| j�| _d S r )r rJ �on_connect_timeoutr )r% r r& r& r' r2 � s
�z _Connector.set_connect_timeoutc C s$ | j �� s | j �t� � | �� d S r )r r7 r8 r rH rL r& r& r' rN � s
z _Connector.on_connect_timeoutc C s8 | j d ur | j�| j � | jd ur| j�| j� d S d S r )r r rB r rL r& r& r' rD � s
�z_Connector.clear_timeoutsc C s | j D ]}|�� qd S r )r$ rE )r% r> r& r& r' rH � s
�z_Connector.close_streams�r N) �__name__�
__module__� __qualname__�__doc__r
r r �socket�
AddressFamilyr r( � staticmethodr �_INITIAL_CONNECT_TIMEOUT�floatr r
�datetime� timedeltar3 r r/ r= r1 rC rM r2 rN rD rH r& r& r& r' r ) sj ���
���� ����
� ����
�
�
�
r c @ s� e Zd ZdZddee ddfdd�Zddd�Zej dddddfd e
d
e d ej d ee
ee
ef ejf d
ee dee
dee dee
eejf defdd�Z dd
e d ej dedee
dee deedf f dd�ZdS )� TCPClientz�A non-blocking TCP connection factory.
.. versionchanged:: 5.0
The ``io_loop`` argument (deprecated since version 4.1) has been removed.
N�resolverr c C s* |d ur || _ d| _d S t� | _ d| _d S )NFT)r\ �
_own_resolverr )r% r\ r& r& r' r( � s
zTCPClient.__init__c C s | j r
| j�� d S d S r )r] r\ rE rL r& r& r' rE � s �zTCPClient.close�host�portr- � ssl_options�max_buffer_size� source_ip� source_portr c
� s �|dur+t |tj�rt�� �� | }nt |tj�r%t�� �� |�� }nt d| ��|dur?t
� || j �
|||��I dH } n | j �
|||�I dH } t| tj| j|||d��}
|
j|d�I dH \}} } |dur�|dur|t
� || jd||d��I dH } | S | jd||d�I dH } | S )a� Connect to the given host and port.
Asynchronously returns an `.IOStream` (or `.SSLIOStream` if
``ssl_options`` is not None).
Using the ``source_ip`` kwarg, one can specify the source
IP address to use when establishing the connection.
In case the user needs to resolve and
use a specific interface, it has to be handled outside
of Tornado as this depends very much on the platform.
Raises `TimeoutError` if the input future does not complete before
``timeout``, which may be specified in any form allowed by
`.IOLoop.add_timeout` (i.e. a `datetime.timedelta` or an absolute time
relative to `.IOLoop.time`)
Similarly, when the user requires a certain source port, it can
be specified using the ``source_port`` arg.
.. versionchanged:: 4.5
Added the ``source_ip`` and ``source_port`` arguments.
.. versionchanged:: 5.0
Added the ``timeout`` argument.
NzUnsupported timeout %r)rb rc )r F)r` �server_hostname)�
isinstance�numbers�Realr r rK rY rZ �
total_seconds� TypeErrorr � with_timeoutr\ �resolver r; r< �_create_streamr3 � start_tls)
r% r^ r_ r- r` ra rb rc r r � connectorr. r> r& r& r' r � sF �$ ��� �
�
� �zTCPClient.connectr. r c
C s� t |t�r|nd}|}|r|s|tjkrdnd}t�|�}|s!|r8z |�||f� W n
tjy7 |�� � w zt||d�} W n tjy_ }
zt� } | � |
� | | fW Y d }
~
S d }
~
ww | | �
|�fS )Nr z::1z 127.0.0.1)ra ) re �intrT �AF_INET6�bind�errorrE r r r8 r ) r% ra r- r. rb rc �source_port_bind�source_ip_bind�
socket_objr> rI �fur& r& r' rl ) s*
�
��zTCPClient._create_streamr rO )NN)rP rQ rR rS r r r( rE rT � AF_UNSPEC�strro rU r
r r �ssl�
SSLContextrX rY rZ r r r rl r& r& r& r' r[ � sX
�������� �
�Q������
�r[ )"rS r; rT rf rY ry �typing�tornado.concurrentr r �tornado.ioloopr �tornado.iostreamr �tornador �tornado.netutilr � tornado.genr r r
r r r
r r r �
TYPE_CHECKINGr rW �objectr r[ r& r& r& r' �<module> s( ( "