U  W[#@sddlmZmZddlZddlmZddlmZmZddl m Z m Z ddl m Z ddlmZmZGdd d ZeejGd d d ZdS) )absolute_importdivisionN) implementer)error interfaces)clientdns) DNSNameError) nativeStringunicodec@s4eZdZddZddZddZddZd d Zd S) "_SRVConnector_ClientFactoryWrappercCs||_||_dSN)-_SRVConnector_ClientFactoryWrapper__connector2_SRVConnector_ClientFactoryWrapper__wrappedFactory)self connectorZwrappedFactoryr:/usr/lib/python3/dist-packages/twisted/names/srvconnect.py__init__sz+_SRVConnector_ClientFactoryWrapper.__init__cCs|j|jdSr )rstartedConnectingr)rrrrrrsz4_SRVConnector_ClientFactoryWrapper.startedConnectingcCs|j|dSr )rconnectionFailedrrreasonrrrclientConnectionFailedsz9_SRVConnector_ClientFactoryWrapper.clientConnectionFailedcCs|j|dSr )rconnectionLostrrrrclientConnectionLostsz7_SRVConnector_ClientFactoryWrapper.clientConnectionLostcCs t|j|Sr )getattrr)rkeyrrr __getattr__sz._SRVConnector_ClientFactoryWrapper.__getattr__N)__name__ __module__ __qualname__rrrrrrrrrr s r c@seZdZdZdZdddidfddZd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZddZddZdS) 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} rZtcpZ connectTCPrNc Csd||_||_t|tr |d}t||_||_||_||_ ||_ ||_ | |_ d|_ d|_d|_dS)a @param domain: The domain to connect to. If passed as a unicode string, it will be encoded using C{idna} encoding. @type domain: L{bytes} or L{unicode} @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} ZidnaN)reactorservice isinstancer encoder domainfactoryprotocolconnectFuncNameconnectFuncArgsconnectFuncKwArgs _defaultPortrserversorderedServers) rr#r$r'r(r)r*r+r,Z defaultPortrrrr8s   zSRVConnector.__init__cCs|j|j||js|jdkr:|tddSt dt |j t |j |jf}| |j|j||fdd|jr||j||jn|jdkr|n |jdS)z"Start connection to remote server.NzDomain is not defined.z _%s._%s.%scSs|Sr )_reallyConnect)xrrrrhz&SRVConnector.connect..)r(ZdoStartrr.r'rrDNSLookupErrorrZ lookupServicer r$r)Z addCallbacks _cbGotServers _ebGotServersZ addCallbackr-Z addErrback_ebServiceUnknownrr0connect)rdrrrr8Zs(      zSRVConnector.connectcCs|tg|_g|_dSr )trapr r.r/rZfailurerrrr6qs zSRVConnector._ebGotServerscCs|\}}}t|dkrd|djtjkrd|djrd|djjtdkrdtdt |j t |j fg|_ g|_ |D]&}|jtjkst|jsqt|j |jqtdS)Nr.z'Service %s not available for domain %s.)lentyperZSRVZpayloadtargetNamerr4reprr$r'r.r/append)rresultZanswersZauthaddarrrr5{s zSRVConnector._cbGotServerscCs6|tjtdd|j|jg|_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. rN) r:rZServiceNameUnknownErrorrZ Record_SRVr-r'r.r/r8r;rrrr7s zSRVConnector._ebServiceUnknownc Cs"|jdk st|jdk st|js4|js4|j|jfS|jsN|jrN|j|_g|_|jsXt|jjddd|jdj}d}d}g}|jD].}|j|kr||j7}|||f|d7}qt d|}|D]B\}}||kr|j|}|j|=|j|t |j |j fSqtd|jjfdS)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} NcSs |j|jfSr )priorityweight)recordrrrr2r3z)SRVConnector.pickServer..)rrr<z Impossible %s pickServer result.)r.AssertionErrorr/r'r$sortrGrHrCrandomZrandintstrr@port RuntimeError __class__r) rZ minPriorityindexZ weightSumZ weightIndexr1ZrandrHZchosenrrr pickServers8              zSRVConnector.pickServercCs||jrd|_dS|\|_|_|jdk s2td|jdk sDtdt|j|j}||j|jt||j f|j |j |_ dS)NrzMust have a host to connect to.zMust have a port to connect to.) stopAfterDNSrRZhostrNrJrr#r*r r(r+r,r)rZ connectFuncrrrr0s  zSRVConnector._reallyConnectcCs|jr|jnd|_dS)zStop attempting to connect.r<N)rstopConnectingrSrrrrrTs zSRVConnector.stopConnectingcCs"|jdk r|jn|dS)z%Disconnect whatever our are state is.N)r disconnectrTrUrrrrVs  zSRVConnector.disconnectcCs|js t|jSr )rrJgetDestinationrUrrrrWs zSRVConnector.getDestinationcCs|j|||jdSr )r(rdoStoprrrrrrszSRVConnector.connectionFailedcCs|j|||jdSr )r(rrXrYrrrrszSRVConnector.connectionLost)rr r!__doc__rSrr8r6r5r7rRr0rTrVrWrrrrrrr"$s& " >r")Z __future__rrrLZzope.interfacerZtwisted.internetrrZ twisted.namesrrZtwisted.names.errorr Ztwisted.python.compatr r r Z IConnectorr"rrrrs