| Current File : //proc/self/root/lib/python3/dist-packages/twisted/mail/__pycache__/relaymanager.cpython-312.pyc |
�
Ϫ�ff� � �� � d Z ddlZddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl m
Z
mZ ddlmZ ddlmZ dd lmZmZmZ dd
lmZ dd lmZ G d � d
� Z G d� deej8 � Z G d� deej< � Z G d� de j@ � Z! G d� de!� Z" G d� d� Z# G d� d� Z$ G d� d� Z% G d � d e%� Z&d � Z'd � Z( G d � d!e)� Z* G d"� d#e)� Z+ G d$� d%� Z,y)&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 �d � e Zd ZdZd� Zed� � Zej d� � Zd� Ze fde
ddfd �Z y)
�ManagedRelayerMixinz�
SMTP Relayer which notifies a manager
Notify the manager about successful mail, failed mail
and broken connections
c � � || _ y �N��manager)�selfr s �;/usr/lib/python3/dist-packages/twisted/mail/relaymanager.py�__init__z ManagedRelayerMixin.__init__& s � � �� � c � � | j S r ��_factory�r s r �factoryzManagedRelayerMixin.factory) s � ��}�}� r c � � || _ y r r )r �values r r zManagedRelayerMixin.factory- s � � ��
r c � � | j d }|t j v r'| j j | j
|� n&| j j
| j
|� | j d= | j d= y)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
�messages r �sentMailz ManagedRelayerMixin.sentMail1 sd � � �*�*�Q�-�� �4�<�<� � �L�L� &� &�t�|�|�W� =� �L�L� &� &�t�|�|�W� =� �M�M�!� � �J�J�q�Mr �reason�returnNc �N � | j j | j � y)zm
called when connection is broken
notify manager we will try to send no more e-mail
N)r �
notifyDoner )r r, s r �connectionLostz"ManagedRelayerMixin.connectionLost? s � �
� � � � �� � �-r ) �__name__�
__module__� __qualname__�__doc__r �propertyr �setterr+ r r r0 � r r r r sU � �� � � � � �
�^�^� � � � � 0>� .�W� .�$� .r r c � � e Zd ZdZd� Zy)�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.
c �x � t j | |� t j j | |g|��i |�� y)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 r � SMTPRelayer�r r% r �args�kws r r zSMTPManagedRelayer.__init__T s4 � � �$�$�T�7�3�
���"�"�4��?�D�?�B�?r N�r1 r2 r3 r4 r r7 r r r9 r9 H s
� � �@r r9 c � � e Zd ZdZd� Zy)�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.
c �x � t j | |� t j j | |g|��i |�� y)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)r r r � ESMTPRelayerr< s r r z ESMTPManagedRelayer.__init__o s4 � �( �$�$�T�7�3�
���#�#�D�(�@�T�@�R�@r Nr? r7 r r rA rA g s
� ��Ar rA c �2 � e Zd ZU dZeZded< d� Zd� Zd� Z y)�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]r c �< � || _ || _ || _ || _ y)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�pKwArgsr< s r r z"SMTPManagedRelayerFactory.__init__� s � � !��
� �� ���
��� r c � � | j | j | j g| j ��i | 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.
)r r% r rG rH r )r �addrr s r �
buildProtocolz'SMTPManagedRelayerFactory.buildProtocol� sH � � !�4�=�=� �M�M�4�<�<�
�*.�*�*�
�8<� � �
�� ����r c �p � | j j | � | j j | � y)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, s r �clientConnectionFailedz0SMTPManagedRelayerFactory.clientConnectionFailed� s( � �
� � �'�'��-� � � � � ��%r N)
r1 r2 r3 r4 r9 r �__annotations__r rK rO r7 r r rE rE � s# � ��* +=�H�'�<��(� &r rE c � � e Zd ZdZeZd� Zd� Zy)�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__}
c �V � || _ || _ t j | ||g|��i |�� y)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�contextFactoryrE r )r r% r rT rU r= r> s r r z#ESMTPManagedRelayerFactory.__init__� s/ � �* �� � ,���!�*�*�4��7�P�T�P�R�Pr c �� � | j xr | j |� } | j | j | j || j g| j
��i | 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.
)rT r r% r rU rG rH r )r rJ �sr s r rK z(ESMTPManagedRelayerFactory.buildProtocol� sp � �
�K�K� -�D�K�K�� -�� �4�=�=� �M�M� �L�L�
� � � �
�
�Z�Z�
� �l�l�
�� ����r N)r1 r2 r3 r4 rA r r rK r7 r r rR rR � s � � � #�H�Q�2r rR c �t � e Zd ZdZdZd� Zd� Zd� Zd� Zd� Z d� Z
d � Z d
� Z d � Z
d � Zd
� Zd� Zd� Zd� Zd� Zd� Zy)�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}).
Tc �2 � || _ | j � y)z�
Initialize non-volatile state.
@type directory: L{bytes}
@param directory: The pathname of the directory holding messages in the
queue.
N)� directory�_init)r r[ s r r zQueue.__init__* s � � #��� �
�
� r c �N � d| _ i | _ i | _ | j � y)z,
Initialize volatile state.
r N)�n�waiting�relayed�
readDirectoryr s r r\ z Queue._init5 s% � � ����� ��� � ��� r c � � d| j iS )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.
r[ )r[ r s r � __getstate__zQueue.__getstate__> s � � �T�^�^�,�,r c �Z � | j j |� | j � y)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__�updater\ �r �states r � __setstate__zQueue.__setstate__G s � �
�
�
� � �U�#� �
�
� r c � � t j | j � D ] }|dd dk7 r� | j |dd � �! y)z>
Scan the message directory for new messages.
���N�-D)�os�listdirr[ �
addMessage�r r* s r ra zQueue.readDirectoryR sC � � �z�z�$�.�.�1� *�G��r�s�|�t�#�� �O�O�G�C�R�L� )� *r c �6 � | j 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.
)r_ �keysr s r �
getWaitingzQueue.getWaiting\ s � � �|�|� � �"�"r c �2 � t | j � dkD S )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 )�lenr_ r s r �
hasWaitingzQueue.hasWaitinge s � � �4�<�<� �1�$�$r c �6 � | j j � S )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.
)r` rr r s r �
getRelayedzQueue.getRelayedp s � � �|�|� � �"�"r c �< � | j |= d| j |< y)z�
Mark a message as being relayed.
@type message: L{bytes}
@param message: The base filename of a message.
� N)r_ r` rp s r � setRelayingzQueue.setRelayingz � � �
�L�L�� !� !�� � �W� r c �< � | j |= d| j |< y)z�
Mark a message as waiting to be relayed.
@type message: L{bytes}
@param message: The base filename of a message.
rz N)r` r_ rp s r �
setWaitingzQueue.setWaiting� r| r c � � || j vr8d| j |< | j r t j d|z dz � yyy)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.
rz zSet � waitingN)r` r_ �noisyr
�msgrp s r ro zQueue.addMessage� sE � � �$�,�,� &�$%�D�L�L�� !��z�z������(�:�5�6� � 'r c � � t j j |� }t j | j |� dz � t j | j |� dz � | j
|= y)z�
Remove a message from the queue.
@type message: L{bytes}
@param message: The base filename of a message.
rl �-HN)rm �path�basename�remove�getPathr` rp s r �donez
Queue.done� sX � � �'�'�"�"�7�+��
� � �$�,�,�w�'�$�.�/�
� � �$�,�,�w�'�$�.�/� �L�L�� !r c �V � t j j | 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.
)rm r� �joinr[ rp s r r� z
Queue.getPath� s � � �w�w�|�|�D�N�N�G�4�4r c �z � | j |� 5 }t j |� cddd� S # 1 sw Y yxY w)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)r r* �fs r � getEnvelopezQueue.getEnvelope� s4 � � �
!�
!�'�
*� "�a��;�;�q�>� "� "� "�s �1�:c �p � t t j j | j |dz � 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.
r� �rb)�openrm r� r� r[ rp s r r� zQueue.getEnvelopeFile� s( � � �B�G�G�L�L�� � ��4� �@�$�G�Gr c � � t j � � dt j � � d| j � dt | � � �}| j dz | _ t t j j | j |dz � d� }t j j | j |dz � }t j j | j |dz � }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.
�_rz r� �wbz-Crl r )� FileMessage) rm �getpid�timer^ �idr� r� r� r[ �twisted.mail.mailr� )r �fname�
headerFile� tempFilename�
finalFilename� messageFiler� s r �createNewMessagezQueue.createNewMessage� s� � � �9�9�;�-�q�����
�Q�t�v�v�h�a��4��z�B�����!�����"�'�'�,�,�t�~�~�u�t�|�D�d�K�
��w�w�|�|�D�N�N�E�D�L�A� ���� � �T�^�^�U�T�\�B�
��<��.� �1��;�{�L�-�P�P�Pr N)r1 r2 r3 r4 r� r r\ rc ri ra rs rv rx r{ r~ ro r� r� r� r� r� r7 r r rY rY s` � ��(
�E� � �-� �*�#� %�#�"�"� 7�
"�
5� "�
H�Qr rY c �< � e Zd ZdZd
d�Zd� Zd� Zd� Zd� Zd� Z d � Z
y) �_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.
Nc �L � || _ g | _ || _ |sddlm} || _ y)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 �_completionDeferredsr� �twisted.internetr� )r r r� r� s r r z_AttemptManager.__init__� s' � � �� �$&��!���
�� 0� �� r c �f � | j j t � � | j d 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� �appendr r s r �getCompletionDeferredz%_AttemptManager.getCompletionDeferred s+ � �
�!�!�(�(���4��(�(��,�,r c �� � | j j | j t j j |� � | j j j |� y)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 �managedr� rm r� r� �queuer� �r r r* s r �_finishz_AttemptManager._finish sH � �
� � � � �U�#�*�*�2�7�7�+;�+;�G�+D�E� � � ��� � ��(r c �p � | j rt j d|z � | j ||� y)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)r� r
r� r� r� s r r# z _AttemptManager.notifySuccess s+ � � �:�:� �G�G�=��G� H� � � �U�G�$r c �F � | j rt j d|z � t j j |� }| j j j |� 5 }t j |� \ }}ddd� t j t | j j j |� dz � � \ }}}| j j j � \ }}|5 t j ||g|� ddd� |j# � D ] }|j% |� � |j' � | j) || j j j |� � y# 1 sw Y ��
xY w# 1 sw Y ��xY w)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 Nrl )r� r
r� rm r� r� r r� r� r� r� r
�generateBouncer� r� r� �dump�
splitlines� lineReceived� eomReceivedr� ) r r r* �fp�from_�to�
bounceMessage�outgoingMessage�lines r r$ z _AttemptManager.notifyFailure+ sN � � �:�:� �G�G�&��0� 1� �'�'�"�"�7�+��
�\�\�
�
�
/�
/��
8� (�B� � � �B��I�E�2� (�#)�#8�#8� �� � �#�#�+�+�G�4�t�;� <�e�R�$
� ��r�=� #�l�l� 0� 0� A� A� C���O�
� )� �K�K��� �R� (� )�!�,�,�.� /�D� � (� (�� .� /��#�#�%� � � �U�D�L�L� .� .� 6� 6�w� ?�@� (� (�� )� )�s �)F
�=F�
F�F c � � | j j j |d� D ]N }| j rt j
d|z dz � | j j j |� �P | j j |= | j }d| _ |D ] }|j d� � y# t $ r Y �7w xY w)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.
r7 zSetting r� N) r r� �getr� r
r� r� r~ �KeyErrorr� �callback)r r r* �
notifications�ds r r/ z_AttemptManager.notifyDoneH s� � � �|�|�+�+�/�/��r�:� 3�G��z�z����
�W�,�z�9�:� �L�L� � � )� )�'� 2� 3� �� � �$�$�U�+� �1�1�
�$(��!� � �A�
�J�J�t� � �� � � � �s �9B< �< C�Cc �r � | j j | }| j
r!t j dt
|� z � d� }| j j d|| j j |� | j j |= y# t $ r t j d� Y yw xY w)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 c �0 � t | j |� y r ��mapr~ �r� r% s r r~ z6_AttemptManager.notifyNoConnection.<locals>.setWaitingp � � � �� � �(� +r � )
r r� r� r
r� r� �strr� � callLaterr� )r r �msgsr~ s r rM z"_AttemptManager.notifyNoConnection^ s� � � ��<�<�'�'��.�D�
�:�:� �G�G�1�C��I�=� >� ,�
� � � � �r�:�t�|�|�/A�/A�4�H� �L�L� � �� '�� � � �G�G�>� ?� � �s �B � B6�5B6)TN) r1 r2 r3 r4 r r� r� r# r$ r/ rM r7 r r r� r� � s+ � � � �*
-� )�
%� A�: �,(r r� c �n � e Zd ZU dZeZeej e d< dZ
dZ d d�Z d� Z
d� Zd� Zd � Zd
� Zd � Zy)
� 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 � Nc �X � || _ || _ i | _ || _ d| _ i | _ y)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.
r7 N)�maxConnections�maxMessagesPerConnectionr� r� �fArgs�fKwArgs)r r� r� r� s r r z%SmartHostSMTPRelayingManager.__init__� s0 � �$ -���(@��%��� ���
���
��� r c �@ � | j 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� )re �copy)r �dcts r rc z)SmartHostSMTPRelayingManager.__getstate__� s" � � �m�m� � �"�� � �N��
r c �H � | j j |� i | _ y)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)re rf r� rg s r ri z)SmartHostSMTPRelayingManager.__setstate__� s � �
�
�
� � �U�#��� r c �� � | j j � t | j � | j k\ ry| j j � sy| j
� 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� ra ru r� r� rv �
_checkStateMXr s r �
checkStatez'SmartHostSMTPRelayingManager.checkState� sP � �
�
�
� � �"� �t�|�|� �� 3� 3� 3� ��z�z�$�$�&� ��!�!�#�#r c �< �� | j j � }|j � i }|D �] }| j j |� \ }}t j
j
|� \ }}|j dd� }t |� dk7 rt j d|z � �}|d } | j j |� |j | g � j | j j |� � t |� | j t | j � z
k\ s�� n | j" �t% � | _ g }
|j' � D �]( \ } } t) | | j j* � } | j, | | g| j. ��i | j0 ��}
t3 t4 j6 j8 | � | j |
< | j; � �| j" j= | � }|j? d� � |j? | j@ | jB |
� |jE �fd�� |jE | jF |
| � |
j �� ��+ tI |
� S )N�@rz � z Illegal message destination: c �, � t | j � S r )r� �name)�mxs r �<lambda>z<SmartHostSMTPRelayingManager._checkStateMX.<locals>.<lambda> s � ��B�G�G� � r c �0 �� �j | � | fd S )Nrz )�errback)�err�relayAttemptDeferreds �r r� z<SmartHostSMTPRelayingManager._checkStateMX.<locals>.<lambda> s �� � 1� 9� 9�#� >�� D�Q� G� r )%r� rs �reverser� �email�utils� parseaddr�splitru r
r� r{ �
setdefaultr� r� r� r� �mxcalc� MXCalculator� iteritemsr� r� r r� r� r� rm r� r� r� �getMX� addCallback� _cbExchange�PORT�
addErrback� _ebExchanger )r � nextMessages� exchangesr� r� r� r� rJ �parts�domain�relaysr� r r �connectSetupDeferredr� s @r r� z*SmartHostSMTPRelayingManager._checkStateMX� s �� ��z�z�,�,�.� �� � � �� � � �C� �
�
�.�.�s�3�I�E�2� ���.�.�r�2�J�D�$��J�J�s�A�&�E��5�z�Q�����7�"�<�=���1�X�F� �J�J� "� "�3� '� � � ��� ,� 3� 3�D�J�J�4F�4F�s�4K� L��9�~�$�"5�"5��D�L�L�8I�"I�J�� � �;�;� �&�.�D�K��� %� /� /� 1� 0�L�F�D�%�d�D�J�J�,<�,<�=�G�"�d�l�l�4��N�4�:�:�N�� � �N�G�$'����(8�(8�$�$?�D�L�L�� !�#*�#@�#@�#B� �#'�;�;�#4�#4�V�#<� � � ,� ,�-D� E� � ,� ,�T�-=�-=�t�y�y�'� R� � +� +�G�
�
!� +� +�D�,<�,<�g�v� N� �M�M�.� /� 0� �F�#�#r c �6 � ddl m} |j |||� y)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�portr r� s r r� z(SmartHostSMTPRelayingManager._cbExchange s � �" -����7�D�'�2r c �� � t j d|z � t j |� d� }ddlm} |j d|| j
| j | � | j |= y)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 c �0 � t | j |� y r r� r� s r r~ z<SmartHostSMTPRelayingManager._ebExchange.<locals>.setWaiting; r� r r r� r� N)r
r� r� r� r� r� r� )r �failurer r� r~ r� s r r� z(SmartHostSMTPRelayingManager._ebExchange' sY � �" ���=��F�G� ����� ,� -����"�j�$�*�*�d�l�l�7�6K�L� �L�L�� !r )r� �
)r1 r2 r3 r4 rE r r r �
ClientFactoryrP r� r� r rc ri r� r� r� r� r7 r r r� r� w sN � �+�Z -F�G�T�(�(�(�
)�E�
�D�
�F��2 � �$�$$$�L3�*"r r� c � � e Zd ZdZeZy)� 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)r1 r2 r3 r4 rR r r7 r r r r D s � �� )�Gr r c �$ � | j � y)z�
Prompt a relaying manager to check state.
@type manager: L{SmartHostSMTPRelayingManager}
@param manager: A relaying manager.
N)r� r s r � _checkStater R s � � ���r c �8 � t j |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 �delays r �RelayStateHelperr \ s � � � � �� �W� =�=r c � � e Zd ZdZy)�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�r1 r2 r3 r4 r7 r r r r m � � �r r c � � e Zd ZdZy)�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 r7 r r r r u r r r c �F � e Zd ZdZdZdZd d�Zd� Zd� Zd
d�Z d � Z
d
� Z d � Z y)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}).
i TNc �\ � i | _ |�
ddlm} |� }|| _ |�ddlm} || _ y)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)r r r r s r r zMXCalculator.__init__� s2 � � �� � � � ;�%�'�H� ��
� �=� 9���
r c �~ � | j j � | j z | j t |� < y)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� �r r� s r �markBadzMXCalculator.markBad� s. � � $�z�z� 1� 1� 3�d�6G�6G� G�� � �C��G� r c �> � | j |= y# t $ r Y yw xY w)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 s r �markGoodzMXCalculator.markGood� s$ � � �� � �B���� � � �s �
� � c �� � | j j |� }|j | j � |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)r r� �maximumCanonicalChainLength�mailExchangeDeferreds r r� zMXCalculator.getMX� sc � �" $�}�}� ?� ?�� G� � �(�(��)<�)<�=� �(�(� �J�J�� ;�
� �'�'��
�
�F�;�#�#r c � � i }|d D ]@ }|j t |j � g � j |j � �B |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�answers r r& zMXCalculator._filterRecords� sL � � � � �a�j� N�F� � � ��V�[�[�!1�2� 6� =� =�f�n�n� M� N��r c �P � ddl m}m} t � }g }|j |g � }|r�|j � } | j |j k( ru|j |� t | j � }
|
|v r |
|v rt t | � � S ||
}g }n+|r| j |
|dz
� S t t | � � S | j |j k( r |j! | j" | f� |r��|r�|j% � |D ]l \ } } t | j � } | | j& vr| c S | j( j+ � | j& | z
}
|
dk\ s�]| j& | = | c S |d d S t |j- 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�errorrz zNo MX records for )�
twisted.namesr2 r3 �setr� �pop�TYPE�CNAME�addr� r� r r r� r �MXr� �
preference�sortr r r � DNSNameError)r �answersr� �
cnamesLeftr2 r3 � seenAliasesr� �pertinentRecords�record�
canonicalNamer; �host�ts r r' zMXCalculator._cbMX� s� � �4 -��e� �� �"�;�;�v�r�2�� �%�)�)�+�F� �{�{�c�i�i�'�����'� #�F�K�K� 0�
� �G�+� %� �3� &�'8��'@� A�A�'.�}�'=�$� "�I�!� #�z�z�-��a� � H�H� '�'@��'H� I�I� �{�{�c�f�f�$�� � �&�"3�"3�V�!<�=�9 �< � �N�N� �&/�
"�"�
�F��6�;�;�'���t�{�{�*�!�M��J�J�&�&�(�4�;�;�t�+<�<����6� � � �D�)�!�M�
"� �Q�<��?� "� �5�-�-�0B�6�*�.M�N�O� Or c �t ��� ddl m�m� | j rx|j �j
� t
j dj |� � | j j |� }�fd�}�fd�}|j ||� |S |j �j
� rt 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 r1 z:MX lookup failed; attempting to use hostname ({}) directlyc �( �� �j | �� S )N)r� )� Record_MX)rJ r2 s �r �
cbResolvedz&MXCalculator._ebMX.<locals>.cbResolvedd s �� ��}�}�$�}�/�/r c �N �� | j �j � t � �r )�trapr= r )r� r3 s �r �
ebResolvedz&MXCalculator._ebMX.<locals>.ebResolvedg s �� �����+�+�,�$�&�&r zNo MX found for )r4 r2 r3 �fallbackToDomainrK r= r
r� �formatr �
getHostByName� addCallbacks�check�OSError)r r r� r� rI rL r2 r3 s @@r r( zMXCalculator._ebMX@ s� �� �. -� � � � �L�L��+�+� ,� �G�G�L�S�S���
� �
�
�+�+�F�3�A�
0�
'�
�N�N�:�z� 2��H�
�]�]�5�-�-�
.�� ,�V�J�7�8� 8��r )NN)� )
r1 r2 r3 r4 r rM r r! r# r� r& r' r( r7 r r r� r� } s; � ��2 �L����(H�
�$�2�(NP�`/r r� )-r4 � email.utilsr� rm r� r� �typingr �twisted.applicationr r� r �twisted.internet.deferr r �twisted.internet.errorr �twisted.internet.protocolr � twisted.mailr
r r �twisted.pythonr
�twisted.python.failurer r r; r9 rC rA r rE rR rY r� r� r r r � Exceptionr r r� r7 r r �<module>r^ s� ��
� � �
� � � (� %� 9� 1� 4� ,� ,� � *�'.� '.�T @�,�e�.?�.?� @�> A�-�u�/A�/A� A�@H&�� 6� 6� H&�V>�!:� >�BKQ� KQ�\S(� S(�lJ"� J"�Z )�$@� )� �>�"� � �� � �r� rr