o

�
bd��@sp
dZd
dl
Zd
dlZd
dlZd
dlZd
dlmZ
d
dlm	Z	
d
dl
mZ
d
dlm
Z
mZ
d
dlmZ
d
dlmZ
d
d	lmZmZmZ
d
d
lmZ
d
dlmZ
Gdd
�d
�ZGdd�deej�ZGdd�deej�ZGdd�dej �Z!Gdd�de!�Z"Gdd�d�Z#Gdd�d�Z$Gdd�d�Z%Gdd�de%�Z&dd�Z'd d!�Z(Gd"d#�d#e)�Z*Gd$d%�d%e)�Z+Gd&d'�d'�Z,dS)(aI

Infrastructure for relaying mail through a smart host.

Traditional peer-to-peer email has been increasingly replaced by smart host
configurations.  Instead of sending mail directly to the recipient, a sender
sends mail to a smart host.  The smart host finds the mail exchange server for
the recipient and sends on the message.
�N)
�Type)
�internet)
�protocol)�Deferred�DeferredList)
�DNSLookupError)
�connectionDone)�bounce�relay�smtp)
�log)
�Failurec@sLeZ
dZd
Zdd�Zedd��
Zejdd��
Zdd�Ze	f
d	e
fd
d�ZdS)
�ManagedRelayerMixinz�
    SMTP Relayer which notifies a manager

    Notify the manager about successful mail, failed mail
    and broken connections
    cC�
|
|_dS�
N�
�manager)�selfr�r�;/usr/lib/python3/dist-packages/twisted/mail/relaymanager.py�__init__&s

zManagedRelayerMixin.__init__c


Cs|jSr�
�_factory�
rrrr�factory)szManagedRelayerMixin.factorycCrrr)r�valuerrrr-s
cCsJ|jd
}|
t
jvr|j�|j|�
n|j�|j|�
|jd
=|jd
=dS)z`
        called when e-mail has been sent

        we will always get 0 or 1 addresses.
        r
N)�namesr�SUCCESSr�
notifySuccessr�
notifyFailure�messages)r�code�resp�numOk�	addressesr�messagerrr�sentMail1s





zManagedRelayerMixin.sentMail�reasoncCs|j�
|j�

d
S)zm
        called when connection is broken

        notify manager we will try to send no more e-mail
        N)r�
notifyDoner)rr'rrr�connectionLost?sz"ManagedRelayerMixin.connectionLostN)�__name__�
__module__�__qualname__�__doc__r�propertyr�setterr&rr
r)rrrrrs




rc@�eZ
dZd
Zdd�ZdS)�SMTPManagedRelayeraV

    An SMTP managed relayer.

    This managed relayer is an SMTP client which is responsible for sending a
    set of messages and keeping an attempt manager informed about its progress.

    @type factory: L{SMTPManagedRelayerFactory}
    @ivar factory: The factory that created this relayer.  This must be set by
        the factory.
    cO�,t�
||�
tjj
||
g|�
Ri|�
�

d
S)a�

        @type messages: L{list} of L{bytes}
        @param messages: The base filenames of messages to be relayed.

        @type manager: L{_AttemptManager}
        @param manager: An attempt manager.

        @type args: 1-L{tuple} of (0) L{bytes} or 2-L{tuple} of
            (0) L{bytes}, (1) L{int}
        @param args: Positional arguments for L{SMTPClient.__init__}

        @type kw: L{dict}
        @param kw: Keyword arguments for L{SMTPClient.__init__}
        N)rrr
�SMTPRelayer�rr r�args�kwrrrrTs 
zSMTPManagedRelayer.__init__N�r*r+r,r-rrrrrr1Hs
r1c@r0)�ESMTPManagedRelayerz�
    An ESMTP managed relayer.

    This managed relayer is an ESMTP client which is responsible for sending a
    set of messages and keeping an attempt manager informed about its progress.
    cOr2)a�
        @type messages: L{list} of L{bytes}
        @param messages: The base filenames of messages to be relayed.

        @type manager: L{_AttemptManager}
        @param manager: An attempt manager.

        @type args: 3-L{tuple} of (0) L{bytes}, (1) L{None} or
            L{ClientContextFactory
            <twisted.internet.ssl.ClientContextFactory>}, (2) L{bytes} or
            4-L{tuple} of (0) L{bytes}, (1) L{None} or
            L{ClientContextFactory
            <twisted.internet.ssl.ClientContextFactory>}, (2) L{bytes},
            (3) L{int}
        @param args: Positional arguments for L{ESMTPClient.__init__}

        @type kw: L{dict}
        @param kw: Keyword arguments for L{ESMTPClient.__init__}
        N)rrr
�ESMTPRelayerr4rrrros 
zESMTPManagedRelayer.__init__Nr7rrrrr8gs
r8c@s6eZ
dZUd
ZeZded<dd�Zdd�Zdd	�Z	d
S)�SMTPManagedRelayerFactorya�
    A factory to create an L{SMTPManagedRelayer}.

    This factory creates a managed relayer which relays a set of messages over
    SMTP and informs an attempt manager of its progress.

    @ivar messages: See L{__init__}
    @ivar manager: See L{__init__}

    @type protocol: callable which returns L{SMTPManagedRelayer}
    @ivar protocol: A callable which returns a managed relayer for SMTP.  See
        L{SMTPManagedRelayer.__init__} for parameters to the callable.

    @type pArgs: 1-L{tuple} of (0) L{bytes} or 2-L{tuple} of
        (0) L{bytes}, (1), L{int}
    @ivar pArgs: Positional arguments for L{SMTPClient.__init__}

    @type pKwArgs: L{dict}
    @ivar pKwArgs: Keyword arguments for L{SMTPClient.__init__}
    zType[protocol.Protocol]rcOs|
|_||_
||_||_d
S)a�

        @type messages: L{list} of L{bytes}
        @param messages: The base filenames of messages to be relayed.

        @type manager: L{_AttemptManager}
        @param manager: An attempt manager.

        @type args: 1-L{tuple} of (0) L{bytes} or 2-L{tuple} of
            (0) L{bytes}, (1), L{int}
        @param args: Positional arguments for L{SMTPClient.__init__}

        @type kw: L{dict}
        @param kw: Keyword arguments for L{SMTPClient.__init__}
        N)r r�pArgs�pKwArgsr4rrrr�s



z"SMTPManagedRelayerFactory.__init__cCs,|j|j
|jg|j�
Ri|j�
�
}||_|S)
a

        Create an L{SMTPManagedRelayer}.

        @type addr: L{IAddress <twisted.internet.interfaces.IAddress>} provider
        @param addr: The address of the SMTP server.

        @rtype: L{SMTPManagedRelayer}
        @return: A managed relayer for SMTP.
        )rr rr;r<r)r�addrrrrr�
buildProtocol�s

�
�
�
z'SMTPManagedRelayerFactory.buildProtocolcCs|j�
|�

|j�|�

d
S)aH

        Notify the attempt manager that a connection could not be established.

        @type connector: L{IConnector <twisted.internet.interfaces.IConnector>}
            provider
        @param connector: A connector.

        @type reason: L{Failure}
        @param reason: The reason the connection attempt failed.
        N)r�notifyNoConnectionr()r�	connectorr'rrr�clientConnectionFailed�s
z0SMTPManagedRelayerFactory.clientConnectionFailedN)
r*r+r,r-r1r�__annotations__rr>rArrrrr:�s

r:c@s$eZ
dZd
ZeZdd�Zdd�ZdS)�ESMTPManagedRelayerFactorya�

    A factory to create an L{ESMTPManagedRelayer}.

    This factory creates a managed relayer which relays a set of messages over
    ESMTP and informs an attempt manager of its progress.

    @type protocol: callable which returns L{ESMTPManagedRelayer}
    @ivar protocol: A callable which returns a managed relayer for ESMTP.  See
        L{ESMTPManagedRelayer.__init__} for parameters to the callable.

    @ivar secret: See L{__init__}
    @ivar contextFactory: See L{__init__}
    cOs,||_||_
tj||
|g|�
Ri|�
�

d
S)a�
        @type messages: L{list} of L{bytes}
        @param messages: The base filenames of messages to be relayed.

        @type manager: L{_AttemptManager}
        @param manager: An attempt manager.

        @type secret: L{bytes}
        @param secret: A string for the authentication challenge response.

        @type contextFactory: L{None} or
            L{ClientContextFactory <twisted.internet.ssl.ClientContextFactory>}
        @param contextFactory: An SSL context factory.

        @type args: 1-L{tuple} of (0) L{bytes} or 2-L{tuple} of
            (0) L{bytes}, (1), L{int}
        @param args: Positional arguments for L{SMTPClient.__init__}

        @param kw: Keyword arguments for L{SMTPClient.__init__}
        N)�secret�contextFactoryr:r)rr rrDrEr5r6rrrr�s
 
z#ESMTPManagedRelayerFactory.__init__cCsB|jo|�|
�
}|j
|j|j||jg|j�
Ri|j�
�
}||_|S)
a

        Create an L{ESMTPManagedRelayer}.

        @type addr: L{IAddress <twisted.internet.interfaces.IAddress>} provider
        @param addr: The address of the ESMTP server.

        @rtype: L{ESMTPManagedRelayer}
        @return: A managed relayer for ESMTP.
        )rDrr rrEr;r<r)rr=�
srrrrr>�s





���
z(ESMTPManagedRelayerFactory.buildProtocolN)r*r+r,r-r8rrr>rrrrrC�s

rCc@s�eZ
dZd
ZdZdd�Zdd�Zdd�Zd	d
�Zdd�Z	d
d�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#S)$�Queuea.
    A queue for messages to be relayed.

    @ivar directory: See L{__init__}

    @type n: L{int}
    @ivar n: A number used to form unique filenames.

    @type waiting: L{dict} of L{bytes}
    @ivar waiting: The base filenames of messages waiting to be relayed.

    @type relayed: L{dict} of L{bytes}
    @ivar relayed: The base filenames of messages in the process of being
        relayed.

    @type noisy: L{bool}
    @ivar noisy: A flag which determines whether informational log messages
        will be generated (C{True}) or not (C{False}).
    TcCs|
|_|�
�
d
S)z�
        Initialize non-volatile state.

        @type directory: L{bytes}
        @param directory: The pathname of the directory holding messages in the
            queue.
        N)�	directory�_init)rrHrrrr*
s
zQueue.__init__c

Csd
|_i|_
i|_|��
dS)z,
        Initialize volatile state.
        r
N)�
n�waiting�relayed�
readDirectoryrrrrrI5
s


zQueue._initc

Cs
d
|ji
S)z�
        Create a representation of the non-volatile state of the queue.

        @rtype: L{dict} mapping L{bytes} to L{object}
        @return: The non-volatile state of the queue.
        rH)
rHrrrr�__getstate__>
�
zQueue.__getstate__cCs|j�
|
�

|��
d
S)z�
        Restore the non-volatile state of the queue and recreate the volatile
        state.

        @type state: L{dict} mapping L{bytes} to L{object}
        @param state: The non-volatile state of the queue.
        N)�__dict__�updaterI�r�staterrr�__setstate__G
s
zQueue.__setstate__c
Cs:t�
|j�
D]}
|
d
d�dkrq|�|
dd
��

qdS)z>
        Scan the message directory for new messages.
        ���N�-D)�os�listdirrH�
addMessage�rr%rrrrMR
s


�zQueue.readDirectoryc

C�
|j�
�S)
z�
        Return the base filenames of messages waiting to be relayed.

        @rtype: L{list} of L{bytes}
        @return: The base filenames of messages waiting to be relayed.
        )rK�keysrrrr�
getWaiting\
rOzQueue.getWaitingc

Cst|j
�
d
kS)z�
        Return an indication of whether the queue has messages waiting to be
        relayed.

        @rtype: L{bool}
        @return: C{True} if messages are waiting to be relayed.  C{False}
            otherwise.
        r
)�lenrKrrrr�
hasWaitinge
s	zQueue.hasWaitingc

Cr[)
z�
        Return the base filenames of messages in the process of being relayed.

        @rtype: L{list} of L{bytes}
        @return: The base filenames of messages in the process of being
            relayed.
        )rLr\rrrr�
getRelayedp
s
zQueue.getRelayedcC�|j|
=d
|j
|
<dS)z�
        Mark a message as being relayed.

        @type message: L{bytes}
        @param message: The base filename of a message.
        �
N)rKrLrZrrr�setRelayingz
�
zQueue.setRelayingcCra)z�
        Mark a message as waiting to be relayed.

        @type message: L{bytes}
        @param message: The base filename of a message.
        rbN)rLrKrZrrr�
setWaiting�
rdzQueue.setWaitingcCs8|
|jv
rd
|j
|
<|jrt�d|
d�

dSdSdS)z�
        Mark a message as waiting to be relayed unless it is in the process of
        being relayed.

        @type message: L{bytes}
        @param message: The base filename of a message.
        rbzSet � waitingN)rLrK�noisyr�msgrZrrrrY�
s




�zQueue.addMessagecCs@tj
�|
�
}
t�|�|
�
d
�

t�|�|
�
d�

|j|
=dS)z�
        Remove a message from the queue.

        @type message: L{bytes}
        @param message: The base filename of a message.
        rV�-HN)rW�path�basename�remove�getPathrLrZrrr�done�
s


z
Queue.donecCstj
�|j|
�S)
z�
        Return the full base pathname of a message in the queue.

        @type message: L{bytes}
        @param message: The base filename of a message.

        @rtype: L{bytes}
        @return: The full base pathname of the message.
        )rWrj�joinrHrZrrrrm�
s
z
Queue.getPathcCs8|�|
�
�
}t
�|�
Wd
�
S1sw



Y
d
S)a

        Get the envelope for a message.

        @type message: L{bytes}
        @param message: The base filename of a message.

        @rtype: L{list} of two L{bytes}
        @return: A list containing the origination and destination addresses
            for the message.
        N)�getEnvelopeFile�pickle�load)rr%�
frrr�getEnvelope�
s
$�zQueue.getEnvelopecCstt
j�|j|
d
�d�S)z�
        Return the envelope file for a message in the queue.

        @type message: L{bytes}
        @param message: The base filename of a message.

        @rtype: file
        @return: The envelope file for the message.
        ri�rb)�openrWrjrorHrZrrrrp�
s
zQueue.getEnvelopeFilec
Cs�t�
��d
t���d
|j�d
t|�
��}
|jd|_ttj�|j|
d�d�}tj�|j|
d�}tj�|j|
d�}t|d�}ddl	m
}
|||||�fS)	z�
        Create a new message in the queue.

        @rtype: 2-L{tuple} of (0) file, (1) L{FileMessage}
        @return: The envelope file and a message receiver for a new message in
            the queue.
        �
_rbri�wbz-CrVr
)
�FileMessage)rW�getpid�timerJ�idrvrjrorH�twisted.mail.mailry)r�fname�
headerFile�tempFilename�
finalFilename�messageFileryrrr�createNewMessage�
s(





zQueue.createNewMessageN)r*r+r,r-rgrrIrNrTrMr]r_r`rcrerYrnrmrtrpr�rrrrrG
s&
		
	



rGc@sJeZ
dZd
Zddd�
Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dS)�_AttemptManagera*

    A manager for an attempt to relay a set of messages to a mail exchange
    server.

    @ivar manager: See L{__init__}

    @type _completionDeferreds: L{list} of L{Deferred}
    @ivar _completionDeferreds: Deferreds which are to be notified when the
        attempt to relay is finished.
    TNcCs,|
|_g|_
||_|sd
dlm}
||_dS)a�

        @type manager: L{SmartHostSMTPRelayingManager}
        @param manager: A smart host.

        @type noisy: L{bool}
        @param noisy: A flag which determines whether informational log
            messages will be generated (L{True}) or not (L{False}).

        @type reactor: L{IReactorTime
            <twisted.internet.interfaces.IReactorTime>} provider
        @param reactor: A reactor which will be used to schedule delayed calls.
        r
�
�reactorN)r�_completionDeferredsrg�twisted.internetr�)rrrgr�rrrr�
s





z_AttemptManager.__init__c

Cs|j�
t��

|jd
S)z�
        Return a deferred which will fire when the attempt to relay is
        finished.

        @rtype: L{Deferred}
        @return: A deferred which will fire when the attempt to relay is
            finished.
        ���)r��appendrrrrr�getCompletionDeferreds	

z%_AttemptManager.getCompletionDeferredcCs,|jj
|
�tj�|�
�

|jj�|�

d
S)aY

        Remove a message from the relay queue and from the smart host's list of
        messages being relayed.

        @type relay: L{SMTPManagedRelayerFactory}
        @param relay: The factory for the relayer which sent the message.

        @type message: L{bytes}
        @param message: The path of the file holding the message.
        N)r�managedrlrWrjrk�queuern�rr
r%rrr�_finishs
z_AttemptManager._finishcCs$|jr
t
�d
|�

|�|
|�
dS)aD

        Remove a message from the relay queue after it has been successfully
        sent.

        @type relay: L{SMTPManagedRelayerFactory}
        @param relay: The factory for the relayer which sent the message.

        @type message: L{bytes}
        @param message: The path of the file holding the message.
        z'success sending %s, removing from queueN)rgrrhr�r�rrrrs

z_AttemptManager.notifySuccessc	Cs
|jr
t
�d
|�

tj�|�
}|jj�|�
�}t	�
|�
\}}Wd�
n1s)w



Y
t�t
|jj�|�
d�
||�\}}}|jj��\}}|�
t	�||g|�
Wd�
n1s^w



Y
|��D]}|�|�

qg|��
|�|
|jj�|�
�
dS)a7

        Generate a bounce message for a message which cannot be relayed.

        @type relay: L{SMTPManagedRelayerFactory}
        @param relay: The factory for the relayer responsible for the message.

        @type message: L{bytes}
        @param message: The path of the file holding the message.
        zcould not relay NrV)rgrrhrWrjrkrr�rprqrrr	�generateBouncervrmr��dump�
splitlines�lineReceived�eomReceivedr�)	rr
r%�fp�from_�to�
bounceMessage�outgoingMessage�linerrrr+s"



�

�

�


z_AttemptManager.notifyFailurecCs�|jj
�|
d
�D]}|jrt�d|d�

|jj�|�

qz|jj
|
=Wn	ty.


Yn
w|j	}d|_	|D]}|�
d�

q7dS)aF

        When the connection is lost or cannot be established, prepare to
        resend unsent messages and fire all deferred which are waiting for
        the completion of the attempt to relay.

        @type relay: L{SMTPManagedRelayerFactory}
        @param relay: The factory for the relayer for the connection.
        rzSetting rfN)rr��getrgrrhr�re�KeyErrorr��callback)rr
r%�
notifications�
drrrr(Hs	






�


�z_AttemptManager.notifyDonecCstz|jj
|
}Wnty


t�d
�

YdSw|jr$t�dt|�
�

dd�}|j�d||jj	|�
|jj
|
=dS)a

        When a connection to the mail exchange server cannot be established,
        prepare to resend messages later.

        @type relay: L{SMTPManagedRelayerFactory}
        @param relay: The factory for the relayer meant to use the connection.
        z(notifyNoConnection passed unknown relay!NzBacking off on delivery of cS�t|j
|
�
dSr��mapre�r�r rrrrep�
z6_AttemptManager.notifyNoConnection.<locals>.setWaiting�)
rr�r�rrhrg�strr��	callLaterr�)rr
�msgsrerrrr?^s	




�

z"_AttemptManager.notifyNoConnection)TN)r*r+r,r-rr�r�rrr(r?rrrrr��
s

r�c@sfeZ
dZUd
ZeZeeje	d<dZ
dZddd�
Zd	d
�Z
dd�Zd
d�Zdd�Zdd�Zdd�ZdS)�SmartHostSMTPRelayingManagera�
    A smart host which uses SMTP managed relayers to send messages from the
    relay queue.

    L{checkState} must be called periodically at which time the state of the
    relay queue is checked and new relayers are created as needed.

    In order to relay a set of messages to a mail exchange server, a smart host
    creates an attempt manager and a managed relayer factory for that set of
    messages.  When a connection is made with the mail exchange server, the
    managed relayer factory creates a managed relayer to send the messages.
    The managed relayer reports on its progress to the attempt manager which,
    in turn, updates the smart host's relay queue and information about its
    managed relayers.

    @ivar queue: See L{__init__}.
    @ivar maxConnections: See L{__init__}.
    @ivar maxMessagesPerConnection: See L{__init__}.

    @type fArgs: 3-L{tuple} of (0) L{list} of L{bytes},
        (1) L{_AttemptManager}, (2) L{bytes} or 4-L{tuple} of (0) L{list}
        of L{bytes}, (1) L{_AttemptManager}, (2) L{bytes}, (3) L{int}
    @ivar fArgs: Positional arguments for
        L{SMTPManagedRelayerFactory.__init__}.

    @type fKwArgs: L{dict}
    @ivar fKwArgs: Keyword arguments for L{SMTPManagedRelayerFactory.__init__}.

    @type factory: callable which returns L{SMTPManagedRelayerFactory}
    @ivar factory: A callable which creates a factory for creating a managed
        relayer. See L{SMTPManagedRelayerFactory.__init__} for parameters to
        the callable.

    @type PORT: L{int}
    @ivar PORT: The port over which to connect to the SMTP server.

    @type mxcalc: L{None} or L{MXCalculator}
    @ivar mxcalc: A resource for mail exchange host lookups.

    @type managed: L{dict} mapping L{SMTPManagedRelayerFactory} to L{list} of
        L{bytes}
    @ivar managed: A mapping of factory for a managed relayer to
        filenames of messages the managed relayer is responsible for.
    r�N��
cCs(||_||_
i|_|
|_d
|_i|_dS)a
        Initialize a smart host.

        The default values specify connection limits appropriate for a
        low-volume smart host.

        @type queue: L{Queue}
        @param queue: A relay queue.

        @type maxConnections: L{int}
        @param maxConnections: The maximum number of concurrent connections to
            SMTP servers.

        @type maxMessagesPerConnection: L{int}
        @param maxMessagesPerConnection: The maximum number of messages for
            which a relayer will be given responsibility.
        rN)�maxConnections�maxMessagesPerConnectionr�r��fArgs�fKwArgs)rr�r�r�rrrr�s





z%SmartHostSMTPRelayingManager.__init__c
Cs|j�
�}
|
d
=|
S)z�
        Create a representation of the non-volatile state of this object.

        @rtype: L{dict} mapping L{bytes} to L{object}
        @return: The non-volatile state of the queue.
        r�)rP�copy)r�dctrrrrN�s


z)SmartHostSMTPRelayingManager.__getstate__cCs|j�
|
�

i|_d
S)z�
        Restore the non-volatile state of this object and recreate the volatile
        state.

        @type state: L{dict} mapping L{bytes} to L{object}
        @param state: The non-volatile state of the queue.
        N)rPrQr�rRrrrrT�s

z)SmartHostSMTPRelayingManager.__setstate__c

Cs4|j�
�
t|j�
|jkrd
S|j��sd
S|��S)af

        Check the state of the relay queue and, if possible, launch relayers to
        handle waiting messages.

        @rtype: L{None} or L{Deferred}
        @return: No return value if no further messages can be relayed or a
            deferred which fires when all of the SMTP connections initiated by
            this call have disconnected.
        N)r�rMr^r�r�r_�
_checkStateMXrrrr�
checkState�s






z'SmartHostSMTPRelayingManager.checkStatec
s�
|j�
�}
|
��
i}|
D]K}|j�|�
\}}tj�|�
\}}|�d
d�}t|�
dkr3t	�
d|�

q
|d}	|j�|�

|�|	g��
|j�|�
�

t|�
|jt|j�
krX
n
q
|jdurbt�|_g}
|��D]T\}	}t||jj�}|j||g|j�
Ri|j�
�
}
ttjj|�|j|
<|���|j�|	�
}|�dd��

|�|j |j!|
�
|�"�f
dd��

|�"|j#|
|	�
|
�
��

qht$|
�
S)N�
@rbr�zIllegal message destination: c

Ss
t|j
�
Sr)r��name)
�mxrrr�<lambda>	s
z<SmartHostSMTPRelayingManager._checkStateMX.<locals>.<lambda>c

s��|�
|fd
S)Nrb)
�errback�
�err�
�relayAttemptDeferredrrr�s)%r�r]�reversert�email�utils�	parseaddr�splitr^rr�rc�
setdefaultr�rmr�r��mxcalc�MXCalculator�	iteritemsr�rgrr�r�r�rWrjrkr��getMX�addCallback�_cbExchange�PORT�
addErrback�_ebExchanger)r�nextMessages�	exchangesrhr�r�r�r=�parts�domain�relaysr�rr�connectSetupDeferredrr�rr��sB













�












�

z*SmartHostSMTPRelayingManager._checkStateMXcCsd
dlm
}
|�|
||�
dS)a�

        Initiate a connection with a mail exchange server.

        This callback function runs after mail exchange server for the domain
        has been looked up.

        @type address: L{bytes}
        @param address: The hostname of a mail exchange server.

        @type port: L{int}
        @param port: A port number.

        @type factory: L{SMTPManagedRelayerFactory}
        @param factory: A factory which can create a relayer for the mail
            exchange server.
        r
r�N)r�r��
connectTCP)r�address�portrr�rrrr�sz(SmartHostSMTPRelayingManager._cbExchangecCsPt�
d
|�

t�
|
�

dd�}ddlm}
|�d||j|j|�
|j|=dS)a�

        Prepare to resend messages later.

        This errback function runs when no mail exchange server for the domain
        can be found.

        @type failure: L{Failure}
        @param failure: The reason the mail exchange lookup failed.

        @type factory: L{SMTPManagedRelayerFactory}
        @param factory: A factory which can create a relayer for the mail
            exchange server.

        @type domain: L{bytes}
        @param domain: A domain.
        z+Error setting up managed relay factory for cSr�rr�r�rrrre;r�z<SmartHostSMTPRelayingManager._ebExchange.<locals>.setWaitingr
r�r�N)rr�r�r�r�r�r�)r�failurerr�rer�rrrr�'s


z(SmartHostSMTPRelayingManager._ebExchange)r�r�)r*r+r,r-r:rrr�
ClientFactoryrBr�r�rrNrTr�r�r�r�rrrrr�ws

-
&r�c
@seZ
dZd
ZeZdS)�SmartHostESMTPRelayingManagera]

    A smart host which uses ESMTP managed relayers to send messages from the
    relay queue.

    @type factory: callable which returns L{ESMTPManagedRelayerFactory}
    @ivar factory: A callable which creates a factory for creating a managed
        relayer. See L{ESMTPManagedRelayerFactory.__init__} for parameters to
        the callable.
    N)r*r+r,r-rCrrrrrr�Ds

r�c

Cs|��
d
S)z�
    Prompt a relaying manager to check state.

    @type manager: L{SmartHostSMTPRelayingManager}
    @param manager: A relaying manager.
    N)
r�rrrr�_checkStateRsr�cCst�
|
t|�S)
a�

    Set up a periodic call to prompt a relaying manager to check state.

    @type manager: L{SmartHostSMTPRelayingManager}
    @param manager: A relaying manager.

    @type delay: L{float}
    @param delay: The number of seconds between calls.

    @rtype: L{TimerService <internet.TimerService>}
    @return: A service which periodically reminds a relaying manager to check
        state.
    )r�TimerServicer�)r�delayrrr�RelayStateHelper\sr�c
@�eZ
dZd
ZdS)�CanonicalNameLoopz�
    An error indicating that when trying to look up a mail exchange host, a set
    of canonical name records was found which form a cycle and resolution was
    abandoned.
    N�r*r+r,r-rrrrr�m�
r�c
@r�)�CanonicalNameChainTooLongz�
    An error indicating that when trying to look up a mail exchange host, too
    many canonical name records which point to other canonical name records
    were encountered and resolution was abandoned.
    Nr�rrrrr�ur�r�c@sTeZ
dZd
ZdZdZddd�
Zdd�Zd	d
�Zddd
�
Z	dd�Z
dd�Zdd�ZdS)r�a�
    A utility for looking up mail exchange hosts and tracking whether they are
    working or not.

    @type clock: L{IReactorTime <twisted.internet.interfaces.IReactorTime>}
        provider
    @ivar clock: A reactor which will be used to schedule timeouts.

    @type resolver: L{IResolver <twisted.internet.interfaces.IResolver>}
    @ivar resolver: A resolver.

    @type badMXs: L{dict} mapping L{bytes} to L{float}
    @ivar badMXs: A mapping of non-functioning mail exchange hostname to time
        at which another attempt at contacting it may be made.

    @type timeOutBadMX: L{int}
    @ivar timeOutBadMX: Period in seconds between attempts to contact a
        non-functioning mail exchange host.

    @type fallbackToDomain: L{bool}
    @ivar fallbackToDomain: A flag indicating whether to attempt to use the
        hostname directly when no mail exchange can be found (C{True}) or
        not (C{False}).
    iTNcCsDi|_|
d
urddl
m}
|�}
|
|_|d
urddlm}
||_d
S)aU

        @type resolver: L{IResolver <twisted.internet.interfaces.IResolver>}
            provider or L{None}
        @param resolver: A resolver.

        @type clock: L{IReactorTime <twisted.internet.interfaces.IReactorTime>}
            provider or L{None}
        @param clock: A reactor which will be used to schedule timeouts.
        Nr
)
�createResolverr�)�badMXs�twisted.names.clientr��resolverr�r��clock)rr�r�r�rrrr�s







zMXCalculator.__init__cCs|j�
�|j|jt|
�
<d
S)z�
        Record that a mail exchange host is not currently functioning.

        @type mx: L{bytes}
        @param mx: The hostname of a mail exchange host.
        N)r��seconds�timeOutBadMXr�r��rr�rrr�markBad�szMXCalculator.markBadcCs$z|j|
=Wd
St
y


Yd
Sw)z�
        Record that a mail exchange host is functioning.

        @type mx: L{bytes}
        @param mx: The hostname of a mail exchange host.
        N)r�r�r�rrr�markGood�s



�zMXCalculator.markGood�cCs:|j�
|
�
}|�|j�

|�|j|
|�
|�|j|
�
|S)
ad
        Find the name of a host that acts as a mail exchange server
        for a domain.

        @type domain: L{bytes}
        @param domain: A domain name.

        @type maximumCanonicalChainLength: L{int}
        @param maximumCanonicalChainLength: The maximum number of unique
            canonical name records to follow while looking up the mail exchange
            host.

        @rtype: L{Deferred} which successfully fires with L{Record_MX}
        @return: A deferred which succeeds with the MX record for the mail
            exchange server for the domain or fails if none can be found.
        )r��lookupMailExchanger��_filterRecords�_cbMXr��_ebMX)rr��maximumCanonicalChainLength�mailExchangeDeferredrrrr��s


�
zMXCalculator.getMXcCs0i}|
d
D]}|�t
|j�
g��|j�

q|S)af
        Organize the records of a DNS response by record name.

        @type records: 3-L{tuple} of (0) L{list} of L{RRHeader
            <twisted.names.dns.RRHeader>}, (1) L{list} of L{RRHeader
            <twisted.names.dns.RRHeader>}, (2) L{list} of L{RRHeader
            <twisted.names.dns.RRHeader>}
        @param records: Answer resource records, authority resource records and
            additional resource records.

        @rtype: L{dict} mapping L{bytes} to L{list} of L{IRecord
            <twisted.names.dns.IRecord>} provider
        @return: A mapping of record name to record payload.
        r
)r�r�r�r��payload)r�records�	recordBag�answerrrrr��s


zMXCalculator._filterRecordscCs>
d
dlm
}m}
t�}g}|
�|g�}|r^|��}	|	j|jkrN|�|�

t	|	j
�
}
|
|
vr>|
|vr7tt|	�
�
S|
|
}g}n|rH|�
|
|d�Stt|	�
�
S|	j|jkr\|�|	j|	f�

|s|r�|��
|D](\}}	t	|	j
�
}||jv
rx|	
S|j��|j|}
|
d
kr�|j|=|	
Sqf|d
dSt|�d|���
�
S)a�
        Try to find the mail exchange host for a domain from the given DNS
        records.

        This will attempt to resolve canonical name record results.  It can
        recognize loops and will give up on non-cyclic chains after a specified
        number of lookups.

        @type answers: L{dict} mapping L{bytes} to L{list} of L{IRecord
            <twisted.names.dns.IRecord>} provider
        @param answers: A mapping of record name to record payload.

        @type domain: L{bytes}
        @param domain: A domain name.

        @type cnamesLeft: L{int}
        @param cnamesLeft: The number of unique canonical name records
            left to follow while looking up the mail exchange host.

        @rtype: L{Record_MX <twisted.names.dns.Record_MX>} or L{Failure}
        @return: An MX record for the mail exchange host or a failure if one
            cannot be found.
        r
��dns�errorrbzNo MX records for )�
twisted.namesr�r��setr��pop�TYPE�CNAME�addr�r�r
r�r�r��MXr��
preference�sortr�r�r��DNSNameError)r�answersr��
cnamesLeftr�r��seenAliasesr��pertinentRecords�record�
canonicalNamer

�host�
trrrr��sB







� 










�zMXCalculator._cbMXcs�d
dlm
�m�

|jr3|
��
j�

t�d�|�
�

|j	�
|�
}�f
dd�}�
f
dd�}|�||�
|S|
��
j�
r@t
d|���
�
|
S)	aX
        Attempt to use the name of the domain directly when mail exchange
        lookup fails.

        @type failure: L{Failure}
        @param failure: The reason for the lookup failure.

        @type domain: L{bytes}
        @param domain: The domain name.

        @rtype: L{Record_MX <twisted.names.dns.Record_MX>} or L{Failure}
        @return: An MX record for the domain or a failure if the fallback to
            domain option is not in effect and an error, other than not
            finding an MX record, occurred during lookup.

        @raise IOError: When no MX record could be found and the fallback to
            domain option is not in effect.

        @raise DNSLookupError: When no MX record could be found and the
            fallback to domain option is in effect but no address for the
            domain could be found.
        r
r�z:MX lookup failed; attempting to use hostname ({}) directlyc

s�j|d
�
S)N)
r�)
�	Record_MX)
r=)
r�rr�
cbResolvedfs
z&MXCalculator._ebMX.<locals>.cbResolvedc

s|��j
�

t��
r)�trapr
rr�)
r�rr�
ebResolvedis

z&MXCalculator._ebMX.<locals>.ebResolvedzNo MX found for )r�r�r��fallbackToDomainr
r
rrh�formatr��
getHostByName�addCallbacks�check�OSError)rr�r�r�r

r
rr�rr�Bs 



��



zMXCalculator._ebMX)NN)
r�)
r*r+r,r-r�r
rr�r�r�r�r�r�rrrrr�}s


	
Rr�)-r-�email.utilsr�rWrqr{�typingr�twisted.applicationrr�r�twisted.internet.deferrr�twisted.internet.errorr�twisted.internet.protocolr�twisted.mailr	r
r�twisted.pythonr�twisted.python.failurer
rr3r1r9r8r�r:rCrGr�r�r�r�r��	Exceptionr�r�r�rrrr�<module>
s>	










* KAON