def test_smtp_edge(self):
queue = self.mox.CreateMockAnything()
queue.enqueue(IsA(Envelope)).AndReturn([(Envelope(), 'testid')])
self.mox.ReplayAll()
server = SmtpEdge(('127.0.0.1', 0), queue)
server.start()
gevent.sleep(0)
client_sock = create_connection(server.server.address)
client = Client(client_sock)
client.get_banner()
client.ehlo('there')
client.mailfrom('*****@*****.**')
client.rcptto('*****@*****.**')
client.data()
client.send_empty_data()
client.quit()
client_sock.close()
def test_smtp_edge(self):
queue = self.mox.CreateMockAnything()
queue.enqueue(IsA(Envelope)).AndReturn([(Envelope(), 'testid')])
self.mox.ReplayAll()
server = SmtpEdge(('127.0.0.1', 0), queue)
server.start()
gevent.sleep(0)
client_sock = create_connection(server.server.address)
client = Client(client_sock)
client.get_banner()
client.ehlo('there')
client.mailfrom('*****@*****.**')
client.rcptto('*****@*****.**')
client.data()
client.send_empty_data()
client.quit()
client_sock.close()
def test_send_empty_data(self):
self.sock.sendall(b".\r\n")
self.sock.recv(IsA(int)).AndReturn(b"250 2.0.0 Done\r\n")
self.mox.ReplayAll()
client = Client(self.sock)
reply = client.send_empty_data()
self.assertEqual("250", reply.code)
self.assertEqual("2.0.0 Done", reply.message)
self.assertEqual(b"[SEND_DATA]", reply.command)
def test_send_empty_data(self):
self.sock.sendall(b'.\r\n')
self.sock.recv(IsA(int)).AndReturn(b'250 2.0.0 Done\r\n')
self.mox.ReplayAll()
client = Client(self.sock)
reply = client.send_empty_data()
self.assertEqual('250', reply.code)
self.assertEqual('2.0.0 Done', reply.message)
self.assertEqual(b'[SEND_DATA]', reply.command)
def test_send_empty_data(self):
self.sock.sendall('.\r\n')
self.sock.recv(IsA(int)).AndReturn('250 2.0.0 Done\r\n')
self.mox.ReplayAll()
client = Client(self.sock)
reply = client.send_empty_data()
assert_equal('250', reply.code)
assert_equal('2.0.0 Done', reply.message)
assert_equal('[SEND_DATA]', reply.command)
class SmtpRelayClient(Greenlet):
def __init__(self, address, queue, socket_creator=None, ehlo_as=None,
tls=None, tls_immediately=False,
tls_required=False, tls_wrapper=None,
connect_timeout=None, command_timeout=None,
data_timeout=None, idle_timeout=None):
super(SmtpRelayClient, self).__init__()
self.address = address
self.queue = queue
self.idle = False
if socket_creator:
self._socket_creator = socket_creator
self.socket = None
self.client = None
self.ehlo_as = ehlo_as or hostname
self.tls = tls
self.tls_immediately = tls_immediately
self.tls_required = tls_required
self.tls_wrapper = tls_wrapper
self.connect_timeout = connect_timeout
self.command_timeout = command_timeout
self.data_timeout = data_timeout
self.idle_timeout = idle_timeout
def _socket_creator(self, address):
socket = create_connection(address)
log.connect(socket, address)
return socket
def _connect(self):
with Timeout(self.connect_timeout):
self.socket = self._socket_creator(self.address)
self.client = Client(self.socket, self.tls_wrapper)
def _banner(self):
with Timeout(self.command_timeout):
banner = self.client.get_banner()
if banner.is_error():
raise SmtpRelayError.factory(banner)
def _ehlo(self):
with Timeout(self.command_timeout):
ehlo = self.client.ehlo(self.ehlo_as)
if ehlo.is_error():
raise SmtpRelayError.factory(ehlo)
def _starttls(self):
with Timeout(self.command_timeout):
starttls = self.client.starttls(self.tls)
if starttls.is_error() and self.tls_required:
raise SmtpRelayError.factory(starttls)
def _handshake(self):
if self.tls and self.tls_immediately:
self.client.encrypt(self.tls)
self._banner()
self._ehlo()
if self.tls and not self.tls_immediately:
self._starttls()
self._ehlo()
def _rset(self):
with Timeout(self.command_timeout):
rset = self.client.rset()
def _mailfrom(self, sender):
with Timeout(self.command_timeout):
mailfrom = self.client.mailfrom(sender)
if mailfrom and mailfrom.is_error():
raise SmtpRelayError.factory(mailfrom)
return mailfrom
def _rcptto(self, rcpt):
with Timeout(self.command_timeout):
rcptto = self.client.rcptto(rcpt)
if rcptto and rcptto.is_error():
raise SmtpRelayError.factory(rcptto)
return rcptto
def _check_replies(self, mailfrom, rcpttos, data):
if mailfrom.is_error():
raise SmtpRelayError.factory(mailfrom)
for rcptto in rcpttos:
if not rcptto.is_error():
break
else:
raise SmtpRelayError.factory(rcpttos[0])
if data.is_error():
raise SmtpRelayError.factory(data)
def _send_message_data(self, envelope):
header_data, message_data = envelope.flatten()
with Timeout(self.data_timeout):
send_data = self.client.send_data(header_data, message_data)
self.client._flush_pipeline()
if send_data.is_error():
raise SmtpRelayError.factory(send_data)
return send_data
def _send_envelope(self, result, envelope):
data = None
try:
mailfrom = self._mailfrom(envelope.sender)
rcpttos = [self._rcptto(rcpt) for rcpt in envelope.recipients]
with Timeout(self.command_timeout):
data = self.client.data()
self._check_replies(mailfrom, rcpttos, data)
except SmtpRelayError as e:
if data and not data.is_error():
with Timeout(self.data_timeout):
self.client.send_empty_data()
self._rset()
result.set_exception(e)
return False
try:
self._send_message_data(envelope)
except SmtpRelayError as e:
self._rset()
result.set_exception(e)
return False
return True
def _check_server_timeout(self):
if self.client.has_reply_waiting():
with Timeout(self.command_timeout):
timeout = self.client.get_reply()
return True
def _disconnect(self):
try:
with Timeout(self.command_timeout):
self.client.quit()
except Exception:
pass
finally:
if self.client:
self.client.io.close()
def _run(self):
try:
self._connect()
self._handshake()
while True:
self.idle = True
n, result, envelope = self.queue.get(timeout=self.idle_timeout)
self.idle = False
if self._check_server_timeout():
self.queue.put((0, result, envelope))
break
if self._send_envelope(result, envelope):
result.set(True)
if self.idle_timeout is None:
break
except (Empty, Timeout):
pass
finally:
self._disconnect()