Server : Apache/2.4.52 (Ubuntu) System : Linux webserver 6.8.0-49-generic #49~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Wed Nov 6 17:42:15 UTC 2 x86_64 User : www-data ( 33) PHP Version : 8.1.2-1ubuntu2.21 Disable Function : NONE Directory : /lib/python3/dist-packages/twisted/names/__pycache__/ |
o
�b�# � @ sv d dl Z d dlmZ d dlmZmZ d dlmZmZ d dl m
Z
d dl m Z G dd� d�Z
eej�G d d
� d
��ZdS ) � N)� implementer)�error�
interfaces)�client�dns)� DNSNameError)� nativeStringc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �"_SRVConnector_ClientFactoryWrapperc C s || _ || _d S �N)�-_SRVConnector_ClientFactoryWrapper__connector�2_SRVConnector_ClientFactoryWrapper__wrappedFactory)�self� connector�wrappedFactory� r �:/usr/lib/python3/dist-packages/twisted/names/srvconnect.py�__init__ s
z+_SRVConnector_ClientFactoryWrapper.__init__c C s | j �| j� d S r
)r �startedConnectingr )r
r r r r r s z4_SRVConnector_ClientFactoryWrapper.startedConnectingc C � | j �|� d S r
)r �connectionFailed�r
r �reasonr r r �clientConnectionFailed � z9_SRVConnector_ClientFactoryWrapper.clientConnectionFailedc C r r
)r �connectionLostr r r r �clientConnectionLost r z7_SRVConnector_ClientFactoryWrapper.clientConnectionLostc C s t | j|�S r
)�getattrr )r
�keyr r r � __getattr__ s z._SRVConnector_ClientFactoryWrapper.__getattr__N)�__name__�
__module__� __qualname__r r r r r r r r r r s r c @ s� e Zd ZdZdZdddi dfdd�Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � ZdS ) � SRVConnectoraD
A connector that looks up DNS SRV records.
RFC 2782 details how SRV records should be interpreted and selected
for subsequent connection attempts. The algorithm for using the records'
priority and weight is implemented in L{pickServer}.
@ivar servers: List of candidate server records for future connection
attempts.
@type servers: L{list} of L{dns.Record_SRV}
@ivar orderedServers: List of server records that have already been tried
in this round of connection attempts.
@type orderedServers: L{list} of L{dns.Record_SRV}
r �tcp�
connectTCPr Nc
C s^ || _ || _|du r dnt�|�| _|| _|| _|| _|| _|| _ | | _
d| _ d| _ d| _
dS )a�
@param domain: The domain to connect to. If passed as a text
string, it will be encoded using C{idna} encoding.
@type domain: L{bytes} or L{str}
@param defaultPort: Optional default port number to be used when SRV
lookup fails and the service name is unknown. This should be the
port number associated with the service name as defined by the IANA
registry.
@type defaultPort: L{int}
N)�reactor�servicer � domainString�domain�factory�protocol�connectFuncName�connectFuncArgs�connectFuncKwArgs� _defaultPortr �servers�orderedServers)
r
r% r& r( r) r* r+ r, r- � defaultPortr r r r 6 s
zSRVConnector.__init__c C s� | j �� | j �| � | jsR| jdu r | �t�d�� dS t� dt
| j �t
| j �t
| j�f �}|�
| j| j� |�| fdd�� | jrJ|�| j� |�| j� dS | jdu r]| �� dS | j�� dS )z"Start connection to remote server.NzDomain is not defined.z
_%s._%s.%sc S s |� � S r
)�_reallyConnect)�xr
r r r �<lambda>q s z&SRVConnector.connect.<locals>.<lambda>)r) �doStartr r/ r( r r �DNSLookupErrorr �
lookupServicer r& r* � addCallbacks�
_cbGotServers�
_ebGotServers� addCallbackr. �
addErrback�_ebServiceUnknownr r2 �connect)r
�dr r r r> ] s0
����
zSRVConnector.connectc C s |� t� g | _g | _d S r
)�trapr r/ r0 �r
�failurer r r r: z s
zSRVConnector._ebGotServersc C s� |\}}}t |�dkr2|d jtjkr2|d jr2|d jjt�d�kr2t�dt | j
�t | j �f ��g | _ g | _
|D ]}|jtjksE|jsFq:| j
�|j� q:d S )N� r � .z'Service %s not available for domain %s.)�len�typer �SRV�payload�target�Namer r6 �reprr& r( r/ r0 �append)r
�result�answers�auth�add�ar r r r9 � s$
����zSRVConnector._cbGotServersc C s6 |� tj� t�dd| j| j�g| _g | _| � � dS )a.
Connect to the default port when the service name is unknown.
If no SRV records were found, the service name will be passed as the
port. If resolving the name fails with
L{error.ServiceNameUnknownError}, a final attempt is done using the
default port.
r N)
r@ r �ServiceNameUnknownErrorr �
Record_SRVr. r( r/ r0 r> rA r r r r= � s z SRVConnector._ebServiceUnknownc C s$ | j dusJ �| jdusJ �| j s | js t| j�| jfS | j s)| jr)| j| _ g | _| j s.J �| j jdd� d� | j d j}d}d}g }| j D ]}|j|kr]||j7 }|�||f� |d7 }qFt �
d|�}|D ]!\}}||kr�| j | }| j |= | j�|� t |j �|j
f S qftd| jj� d���) a;
Pick the next server.
This selects the next server from the list of SRV records according
to their priority and weight values, as set out by the default
algorithm specified in RFC 2782.
At the beginning of a round, L{servers} is populated with
L{orderedServers}, and the latter is made empty. L{servers}
is the list of candidates, and L{orderedServers} is the list of servers
that have already been tried.
First, all records are ordered by priority and weight in ascending
order. Then for each priority level, a running sum is calculated
over the sorted list of records for that priority. Then a random value
between 0 and the final sum is compared to each record in order. The
first record that is greater than or equal to that random value is
chosen and removed from the list of candidates for this round.
@return: A tuple of target hostname and port from the chosen DNS SRV
record.
@rtype: L{tuple} of native L{str} and L{int}
Nc S s | j | jfS r
)�priority�weight)�recordr r r r4 � s z)SRVConnector.pickServer.<locals>.<lambda>)r r rC z Impossible z pickServer result.)r/ r0 r r( r& �sortrT rU rL �random�randint�strrI �port� RuntimeError� __class__r ) r
� minPriority�index� weightSum� weightIndexr3 �randrU �chosenr r r �
pickServer� s8
�
�zSRVConnector.pickServerc C s� | j rd| _ d S | �� \| _| _| jd usJ d��| jd us"J d��t| j| j�}|| j| jt| | j�g| j �R i | j
��| _ d S )Nr z Must have a host to connect to.z Must have a port to connect to.) � stopAfterDNSrd �hostr[ r r% r+ r r) r, r- r )r
� connectFuncr r r r2 � s
�� �zSRVConnector._reallyConnectc C s | j r
| j �� dS d| _dS )zStop attempting to connect.rC N)r �stopConnectingre �r
r r r rh � s
zSRVConnector.stopConnectingc C s$ | j dur | j �� dS | �� dS )z%Disconnect whatever our are state is.N)r �
disconnectrh ri r r r rj � s
zSRVConnector.disconnectc C s | j sJ �| j �� S r
)r �getDestinationri r r r rk s
zSRVConnector.getDestinationc C � | j �| |� | j �� d S r
)r) r �doStop�r
r r r r r � z SRVConnector.connectionFailedc C rl r
)r) r rm rn r r r r
ro zSRVConnector.connectionLost)r r r! �__doc__re r r> r: r9 r= rd r2 rh rj rk r r r r r r r" " s(
�'
= r" )rX �zope.interfacer �twisted.internetr r �
twisted.namesr r �twisted.names.errorr �twisted.python.compatr r �
IConnectorr" r r r r �<module> s