blob: f9d1c99bb467318b0269012cb922ccab5a674571 [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#! /usr/bin/env python3
Barry Warsaw4c4bec81998-12-22 03:02:20 +00002
Barry Warsawa1ae8842000-07-09 21:24:31 +00003'''SMTP/ESMTP client class.
Guido van Rossumbbe323e1998-01-29 17:24:40 +00004
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +00005This should follow RFC 821 (SMTP), RFC 1869 (ESMTP), RFC 2554 (SMTP
6Authentication) and RFC 2487 (Secure SMTP over TLS).
Guido van Rossumbbe323e1998-01-29 17:24:40 +00007
Guido van Rossumfcfb6321998-08-04 15:29:54 +00008Notes:
9
10Please remember, when doing ESMTP, that the names of the SMTP service
Barry Warsaw4c4bec81998-12-22 03:02:20 +000011extensions are NOT the same thing as the option keywords for the RCPT
Guido van Rossumfcfb6321998-08-04 15:29:54 +000012and MAIL commands!
13
Guido van Rossumbbe323e1998-01-29 17:24:40 +000014Example:
15
Barry Warsawa7d9bdf1998-12-22 03:24:27 +000016 >>> import smtplib
17 >>> s=smtplib.SMTP("localhost")
Guido van Rossum7131f842007-02-09 20:13:25 +000018 >>> print(s.help())
Barry Warsawa7d9bdf1998-12-22 03:24:27 +000019 This is Sendmail version 8.8.4
20 Topics:
21 HELO EHLO MAIL RCPT DATA
22 RSET NOOP QUIT HELP VRFY
23 EXPN VERB ETRN DSN
24 For more info use "HELP <topic>".
25 To report bugs in the implementation send email to
26 sendmail-bugs@sendmail.org.
27 For local information send email to Postmaster at your site.
28 End of HELP info
29 >>> s.putcmd("vrfy","someone@here")
30 >>> s.getreply()
31 (250, "Somebody OverHere <somebody@here.my.org>")
32 >>> s.quit()
Barry Warsawa1ae8842000-07-09 21:24:31 +000033'''
Guido van Rossumbbe323e1998-01-29 17:24:40 +000034
Guido van Rossum98d9fd32000-02-28 15:12:25 +000035# Author: The Dragon De Monsyne <dragondm@integral.org>
36# ESMTP support, test code and doc fixes added by
37# Eric S. Raymond <esr@thyrsus.com>
38# Better RFC 821 compliance (MAIL and RCPT, and CRLF in data)
39# by Carey Evans <c.evans@clear.net.nz>, for picky mail servers.
Guido van Rossumae010462001-09-11 15:57:46 +000040# RFC 2554 (authentication) support by Gerhard Haering <gerhard@bigfoot.de>.
Tim Peters495ad3c2001-01-15 01:36:40 +000041#
Guido van Rossum98d9fd32000-02-28 15:12:25 +000042# This was modified from the Python 1.5 library HTTP lib.
43
Guido van Rossumbbe323e1998-01-29 17:24:40 +000044import socket
R. David Murray7dff9e02010-11-08 17:15:13 +000045import io
Barry Warsaw07201771998-12-22 20:37:36 +000046import re
Thomas Woutersb2137042007-02-01 18:02:27 +000047import email.utils
R. David Murray7dff9e02010-11-08 17:15:13 +000048import email.message
49import email.generator
Guido van Rossumae010462001-09-11 15:57:46 +000050import base64
51import hmac
Barry Warsaw2cc1f6d2007-08-30 14:28:55 +000052from email.base64mime import body_encode as encode_base64
Brett Cannone6f8a892004-07-10 23:14:30 +000053from sys import stderr
Guido van Rossumbbe323e1998-01-29 17:24:40 +000054
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +000055__all__ = ["SMTPException", "SMTPServerDisconnected", "SMTPResponseException",
56 "SMTPSenderRefused", "SMTPRecipientsRefused", "SMTPDataError",
57 "SMTPConnectError", "SMTPHeloError", "SMTPAuthenticationError",
58 "quoteaddr", "quotedata", "SMTP"]
Skip Montanaro0de65802001-02-15 22:15:14 +000059
Guido van Rossumbbe323e1998-01-29 17:24:40 +000060SMTP_PORT = 25
Thomas Wouters89f507f2006-12-13 04:49:30 +000061SMTP_SSL_PORT = 465
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +000062CRLF = "\r\n"
63bCRLF = b"\r\n"
Guido van Rossumbbe323e1998-01-29 17:24:40 +000064
Piers Lauder385a77a2002-07-27 00:38:30 +000065OLDSTYLE_AUTH = re.compile(r"auth=(.*)", re.I)
66
Tim Peters495ad3c2001-01-15 01:36:40 +000067# Exception classes used by this module.
Guido van Rossum296e1431999-04-07 15:03:39 +000068class SMTPException(Exception):
69 """Base class for all exceptions raised by this module."""
70
71class SMTPServerDisconnected(SMTPException):
72 """Not connected to any SMTP server.
73
74 This exception is raised when the server unexpectedly disconnects,
75 or when an attempt is made to use the SMTP instance before
76 connecting it to a server.
77 """
78
79class SMTPResponseException(SMTPException):
80 """Base class for all exceptions that include an SMTP error code.
81
82 These exceptions are generated in some instances when the SMTP
83 server returns an error code. The error code is stored in the
84 `smtp_code' attribute of the error, and the `smtp_error' attribute
85 is set to the error message.
86 """
87
88 def __init__(self, code, msg):
89 self.smtp_code = code
90 self.smtp_error = msg
91 self.args = (code, msg)
92
93class SMTPSenderRefused(SMTPResponseException):
94 """Sender address refused.
Guido van Rossumae010462001-09-11 15:57:46 +000095
Guido van Rossum296e1431999-04-07 15:03:39 +000096 In addition to the attributes set by on all SMTPResponseException
Barry Warsawd25c1b71999-11-28 17:11:06 +000097 exceptions, this sets `sender' to the string that the SMTP refused.
Guido van Rossum296e1431999-04-07 15:03:39 +000098 """
99
100 def __init__(self, code, msg, sender):
101 self.smtp_code = code
102 self.smtp_error = msg
103 self.sender = sender
104 self.args = (code, msg, sender)
105
Guido van Rossum20c92281999-04-21 16:52:20 +0000106class SMTPRecipientsRefused(SMTPException):
Barry Warsawd25c1b71999-11-28 17:11:06 +0000107 """All recipient addresses refused.
Guido van Rossumae010462001-09-11 15:57:46 +0000108
Thomas Wouters7e474022000-07-16 12:04:32 +0000109 The errors for each recipient are accessible through the attribute
Tim Peters495ad3c2001-01-15 01:36:40 +0000110 'recipients', which is a dictionary of exactly the same sort as
111 SMTP.sendmail() returns.
Guido van Rossum296e1431999-04-07 15:03:39 +0000112 """
113
114 def __init__(self, recipients):
115 self.recipients = recipients
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000116 self.args = (recipients,)
Guido van Rossum296e1431999-04-07 15:03:39 +0000117
118
Guido van Rossum296e1431999-04-07 15:03:39 +0000119class SMTPDataError(SMTPResponseException):
120 """The SMTP server didn't accept the data."""
121
122class SMTPConnectError(SMTPResponseException):
Barry Warsawd25c1b71999-11-28 17:11:06 +0000123 """Error during connection establishment."""
Guido van Rossum296e1431999-04-07 15:03:39 +0000124
125class SMTPHeloError(SMTPResponseException):
Barry Warsawd25c1b71999-11-28 17:11:06 +0000126 """The server refused our HELO reply."""
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000127
Guido van Rossumae010462001-09-11 15:57:46 +0000128class SMTPAuthenticationError(SMTPResponseException):
129 """Authentication error.
130
131 Most probably the server didn't accept the username/password
132 combination provided.
133 """
Peter Schneider-Kamp7bc82bb2000-08-10 14:02:23 +0000134
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000135def quoteaddr(addr):
136 """Quote a subset of the email addresses defined by RFC 821.
137
Georg Brandl9f0f9602008-06-12 22:23:59 +0000138 Should be able to handle anything email.utils.parseaddr can handle.
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000139 """
Raymond Hettinger342456d2002-09-05 01:14:07 +0000140 m = (None, None)
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000141 try:
Thomas Woutersb2137042007-02-01 18:02:27 +0000142 m = email.utils.parseaddr(addr)[1]
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000143 except AttributeError:
144 pass
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000145 if m == (None, None): # Indicates parse failure or AttributeError
Georg Brandlbd3bc4d2006-02-17 09:52:53 +0000146 # something weird here.. punt -ddm
Raymond Hettinger342456d2002-09-05 01:14:07 +0000147 return "<%s>" % addr
Georg Brandlbd3bc4d2006-02-17 09:52:53 +0000148 elif m is None:
149 # the sender wants an empty return address
150 return "<>"
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000151 else:
152 return "<%s>" % m
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000153
R. David Murray7dff9e02010-11-08 17:15:13 +0000154# Legacy method kept for backward compatibility.
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000155def quotedata(data):
156 """Quote data for email.
157
Barry Warsawd25c1b71999-11-28 17:11:06 +0000158 Double leading '.', and change Unix newline '\\n', or Mac '\\r' into
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000159 Internet CRLF end-of-line.
160 """
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000161 return re.sub(r'(?m)^\.', '..',
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000162 re.sub(r'(?:\r\n|\n|\r(?!\n))', CRLF, data))
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000163
R. David Murray7dff9e02010-11-08 17:15:13 +0000164def _quote_periods(bindata):
R David Murray0f663d02011-06-09 15:05:57 -0400165 return re.sub(br'(?m)^\.', b'..', bindata)
R. David Murray7dff9e02010-11-08 17:15:13 +0000166
167def _fix_eols(data):
168 return re.sub(r'(?:\r\n|\n|\r(?!\n))', CRLF, data)
169
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000170try:
171 import ssl
172except ImportError:
173 _have_ssl = False
174else:
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000175 _have_ssl = True
176
Peter Schneider-Kamp7bc82bb2000-08-10 14:02:23 +0000177
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000178class SMTP:
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000179 """This class manages a connection to an SMTP or ESMTP server.
180 SMTP Objects:
Tim Peters495ad3c2001-01-15 01:36:40 +0000181 SMTP objects have the following attributes:
182 helo_resp
183 This is the message given by the server in response to the
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000184 most recent HELO command.
Tim Peters495ad3c2001-01-15 01:36:40 +0000185
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000186 ehlo_resp
Tim Peters495ad3c2001-01-15 01:36:40 +0000187 This is the message given by the server in response to the
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000188 most recent EHLO command. This is usually multiline.
189
Tim Peters495ad3c2001-01-15 01:36:40 +0000190 does_esmtp
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000191 This is a True value _after you do an EHLO command_, if the
192 server supports ESMTP.
193
Tim Peters495ad3c2001-01-15 01:36:40 +0000194 esmtp_features
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000195 This is a dictionary, which, if the server supports ESMTP,
Barry Warsawd25c1b71999-11-28 17:11:06 +0000196 will _after you do an EHLO command_, contain the names of the
197 SMTP service extensions this server supports, and their
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000198 parameters (if any).
Barry Warsawd25c1b71999-11-28 17:11:06 +0000199
Tim Peters495ad3c2001-01-15 01:36:40 +0000200 Note, all extension names are mapped to lower case in the
201 dictionary.
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000202
Barry Warsawd25c1b71999-11-28 17:11:06 +0000203 See each method's docstrings for details. In general, there is a
204 method of the same name to perform each SMTP command. There is also a
205 method called 'sendmail' that will do an entire mail transaction.
206 """
Guido van Rossum95e6f701998-06-25 02:15:50 +0000207 debuglevel = 0
208 file = None
209 helo_resp = None
Guido van Rossumd8faa362007-04-27 19:54:29 +0000210 ehlo_msg = "ehlo"
Guido van Rossum95e6f701998-06-25 02:15:50 +0000211 ehlo_resp = None
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000212 does_esmtp = 0
Antoine Pitrouc1d52062011-05-07 19:39:37 +0200213 default_port = SMTP_PORT
Guido van Rossum95e6f701998-06-25 02:15:50 +0000214
Georg Brandlf78e02b2008-06-10 17:40:04 +0000215 def __init__(self, host='', port=0, local_hostname=None,
216 timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000217 """Initialize a new instance.
218
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000219 If specified, `host' is the name of the remote host to which to
220 connect. If specified, `port' specifies the port to which to connect.
Barry Warsawd25c1b71999-11-28 17:11:06 +0000221 By default, smtplib.SMTP_PORT is used. An SMTPConnectError is raised
Neil Schemenauer6730f262002-03-24 15:30:40 +0000222 if the specified `host' doesn't respond correctly. If specified,
Tim Peters863ac442002-04-16 01:38:40 +0000223 `local_hostname` is used as the FQDN of the local host. By default,
224 the local hostname is found using socket.getfqdn().
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000225
226 """
Guido van Rossumd8faa362007-04-27 19:54:29 +0000227 self.timeout = timeout
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000228 self.esmtp_features = {}
Guido van Rossum296e1431999-04-07 15:03:39 +0000229 if host:
230 (code, msg) = self.connect(host, port)
231 if code != 220:
232 raise SMTPConnectError(code, msg)
Raymond Hettingerf13eb552002-06-02 00:40:05 +0000233 if local_hostname is not None:
Barry Warsaw13e34f72002-03-26 20:27:35 +0000234 self.local_hostname = local_hostname
Neil Schemenauer6730f262002-03-24 15:30:40 +0000235 else:
Barry Warsaw13e34f72002-03-26 20:27:35 +0000236 # RFC 2821 says we should use the fqdn in the EHLO/HELO verb, and
237 # if that can't be calculated, that we should use a domain literal
238 # instead (essentially an encoded IP address like [A.B.C.D]).
239 fqdn = socket.getfqdn()
240 if '.' in fqdn:
241 self.local_hostname = fqdn
242 else:
243 # We can't find an fqdn hostname, so use a domain literal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000244 addr = '127.0.0.1'
245 try:
246 addr = socket.gethostbyname(socket.gethostname())
247 except socket.gaierror:
248 pass
Barry Warsaw13e34f72002-03-26 20:27:35 +0000249 self.local_hostname = '[%s]' % addr
Tim Peters495ad3c2001-01-15 01:36:40 +0000250
Barry Warsaw1f5c9582011-03-15 15:04:44 -0400251 def __enter__(self):
252 return self
253
254 def __exit__(self, *args):
255 try:
256 code, message = self.docmd("QUIT")
257 if code != 221:
258 raise SMTPResponseException(code, message)
259 except SMTPServerDisconnected:
260 pass
261 finally:
262 self.close()
263
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000264 def set_debuglevel(self, debuglevel):
265 """Set the debug output level.
266
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000267 A non-false value results in debug messages for connection and for all
268 messages sent to and received from the server.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000269
270 """
271 self.debuglevel = debuglevel
272
Barry Warsawc9181712008-03-19 14:25:51 +0000273 def _get_socket(self, host, port, timeout):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000274 # This makes it simpler for SMTP_SSL to use the SMTP connect code
275 # and just alter the socket connection bit.
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000276 if self.debuglevel > 0:
277 print('connect:', (host, port), file=stderr)
Barry Warsawc9181712008-03-19 14:25:51 +0000278 return socket.create_connection((host, port), timeout)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000279
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000280 def connect(self, host='localhost', port=0):
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000281 """Connect to a host on a given port.
Guido van Rossum95e6f701998-06-25 02:15:50 +0000282
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000283 If the hostname ends with a colon (`:') followed by a number, and
284 there is no port specified, that suffix will be stripped off and the
285 number interpreted as the port number to use.
Guido van Rossum95e6f701998-06-25 02:15:50 +0000286
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000287 Note: This method is automatically invoked by __init__, if a host is
288 specified during instantiation.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000289
290 """
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000291 if not port and (host.find(':') == host.rfind(':')):
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000292 i = host.rfind(':')
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000293 if i >= 0:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000294 host, port = host[:i], host[i + 1:]
295 try:
296 port = int(port)
Eric S. Raymond8d876032001-02-09 10:14:53 +0000297 except ValueError:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000298 raise socket.error("nonnumeric port")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000299 if not port:
300 port = self.default_port
301 if self.debuglevel > 0:
302 print('connect:', (host, port), file=stderr)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000303 self.sock = self._get_socket(host, port, self.timeout)
Antoine Pitrouf068ab82011-06-06 19:17:09 +0200304 self.file = None
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000305 (code, msg) = self.getreply()
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000306 if self.debuglevel > 0:
307 print("connect:", msg, file=stderr)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000308 return (code, msg)
Tim Peters495ad3c2001-01-15 01:36:40 +0000309
Guido van Rossum8a392d72007-11-21 22:09:45 +0000310 def send(self, s):
311 """Send `s' to the server."""
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000312 if self.debuglevel > 0:
313 print('send:', repr(s), file=stderr)
Christian Heimes90c3d9b2008-02-23 13:18:03 +0000314 if hasattr(self, 'sock') and self.sock:
Guido van Rossum8a392d72007-11-21 22:09:45 +0000315 if isinstance(s, str):
316 s = s.encode("ascii")
Guido van Rossum2880f6e1998-08-10 20:07:00 +0000317 try:
Guido van Rossum8a392d72007-11-21 22:09:45 +0000318 self.sock.sendall(s)
Guido van Rossum2880f6e1998-08-10 20:07:00 +0000319 except socket.error:
Barry Warsaw76750972001-12-14 20:34:20 +0000320 self.close()
Guido van Rossum40233ea1999-01-15 03:23:55 +0000321 raise SMTPServerDisconnected('Server not connected')
Guido van Rossumfc40a831998-01-29 17:26:45 +0000322 else:
Guido van Rossum40233ea1999-01-15 03:23:55 +0000323 raise SMTPServerDisconnected('please run connect() first')
Tim Peters495ad3c2001-01-15 01:36:40 +0000324
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000325 def putcmd(self, cmd, args=""):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000326 """Send a command to the server."""
Guido van Rossumdb23d3d1999-06-09 15:13:10 +0000327 if args == "":
328 str = '%s%s' % (cmd, CRLF)
329 else:
330 str = '%s %s%s' % (cmd, args, CRLF)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000331 self.send(str)
Tim Peters495ad3c2001-01-15 01:36:40 +0000332
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000333 def getreply(self):
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000334 """Get a reply from the server.
Tim Peters495ad3c2001-01-15 01:36:40 +0000335
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000336 Returns a tuple consisting of:
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000337
338 - server response code (e.g. '250', or such, if all goes well)
339 Note: returns -1 if it can't read response code.
340
341 - server response string corresponding to response code (multiline
342 responses are converted to a single, multiline string).
Guido van Rossumf123f841999-03-29 20:33:21 +0000343
344 Raises SMTPServerDisconnected if end-of-file is reached.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000345 """
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000346 resp = []
Guido van Rossum296e1431999-04-07 15:03:39 +0000347 if self.file is None:
348 self.file = self.sock.makefile('rb')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000349 while 1:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000350 try:
351 line = self.file.readline()
352 except socket.error:
353 line = ''
Guido van Rossum806c2462007-08-06 23:33:07 +0000354 if not line:
Guido van Rossum296e1431999-04-07 15:03:39 +0000355 self.close()
356 raise SMTPServerDisconnected("Connection unexpectedly closed")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000357 if self.debuglevel > 0:
358 print('reply:', repr(line), file=stderr)
Guido van Rossum806c2462007-08-06 23:33:07 +0000359 resp.append(line[4:].strip(b' \t\r\n'))
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000360 code = line[:3]
Guido van Rossum296e1431999-04-07 15:03:39 +0000361 # Check that the error code is syntactically correct.
362 # Don't attempt to read a continuation line if it is broken.
363 try:
Eric S. Raymondc013f302001-02-09 05:40:38 +0000364 errcode = int(code)
Guido van Rossum296e1431999-04-07 15:03:39 +0000365 except ValueError:
366 errcode = -1
367 break
Guido van Rossumf123f841999-03-29 20:33:21 +0000368 # Check if multiline response.
Guido van Rossum806c2462007-08-06 23:33:07 +0000369 if line[3:4] != b"-":
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000370 break
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000371
Guido van Rossumc43e79f2007-06-18 18:26:36 +0000372 errmsg = b"\n".join(resp)
Tim Peters495ad3c2001-01-15 01:36:40 +0000373 if self.debuglevel > 0:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000374 print('reply: retcode (%s); Msg: %s' % (errcode, errmsg), file=stderr)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000375 return errcode, errmsg
Tim Peters495ad3c2001-01-15 01:36:40 +0000376
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000377 def docmd(self, cmd, args=""):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000378 """Send a command, and return its response code."""
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000379 self.putcmd(cmd, args)
Guido van Rossum296e1431999-04-07 15:03:39 +0000380 return self.getreply()
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000381
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000382 # std smtp commands
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000383 def helo(self, name=''):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000384 """SMTP 'helo' command.
385 Hostname to send for this command defaults to the FQDN of the local
386 host.
387 """
Neil Schemenauer6730f262002-03-24 15:30:40 +0000388 self.putcmd("helo", name or self.local_hostname)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000389 (code, msg) = self.getreply()
390 self.helo_resp = msg
391 return (code, msg)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000392
Guido van Rossum95e6f701998-06-25 02:15:50 +0000393 def ehlo(self, name=''):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000394 """ SMTP 'ehlo' command.
395 Hostname to send for this command defaults to the FQDN of the local
396 host.
397 """
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000398 self.esmtp_features = {}
Guido van Rossumd8faa362007-04-27 19:54:29 +0000399 self.putcmd(self.ehlo_msg, name or self.local_hostname)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000400 (code, msg) = self.getreply()
Tim Peters495ad3c2001-01-15 01:36:40 +0000401 # According to RFC1869 some (badly written)
402 # MTA's will disconnect on an ehlo. Toss an exception if
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000403 # that happens -ddm
404 if code == -1 and len(msg) == 0:
Barry Warsaw76750972001-12-14 20:34:20 +0000405 self.close()
Guido van Rossum40233ea1999-01-15 03:23:55 +0000406 raise SMTPServerDisconnected("Server not connected")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000407 self.ehlo_resp = msg
Fred Drake8152d322000-12-12 23:20:45 +0000408 if code != 250:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000409 return (code, msg)
410 self.does_esmtp = 1
Thomas Wouters7e474022000-07-16 12:04:32 +0000411 #parse the ehlo response -ddm
Guido van Rossum04110fb2007-08-24 16:32:05 +0000412 assert isinstance(self.ehlo_resp, bytes), repr(self.ehlo_resp)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000413 resp = self.ehlo_resp.decode("latin-1").split('\n')
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000414 del resp[0]
Guido van Rossum2880f6e1998-08-10 20:07:00 +0000415 for each in resp:
Piers Lauder385a77a2002-07-27 00:38:30 +0000416 # To be able to communicate with as many SMTP servers as possible,
417 # we have to take the old-style auth advertisement into account,
418 # because:
419 # 1) Else our SMTP feature parser gets confused.
420 # 2) There are some servers that only advertise the auth methods we
421 # support using the old style.
422 auth_match = OLDSTYLE_AUTH.match(each)
423 if auth_match:
424 # This doesn't remove duplicates, but that's no problem
425 self.esmtp_features["auth"] = self.esmtp_features.get("auth", "") \
426 + " " + auth_match.groups(0)[0]
427 continue
428
Barry Warsawbe22ae62002-04-15 20:03:30 +0000429 # RFC 1869 requires a space between ehlo keyword and parameters.
430 # It's actually stricter, in that only spaces are allowed between
431 # parameters, but were not going to check for that here. Note
432 # that the space isn't present if there are no parameters.
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000433 m = re.match(r'(?P<feature>[A-Za-z0-9][A-Za-z0-9\-]*) ?', each)
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000434 if m:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000435 feature = m.group("feature").lower()
436 params = m.string[m.end("feature"):].strip()
Piers Lauder385a77a2002-07-27 00:38:30 +0000437 if feature == "auth":
438 self.esmtp_features[feature] = self.esmtp_features.get(feature, "") \
439 + " " + params
440 else:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000441 self.esmtp_features[feature] = params
442 return (code, msg)
Guido van Rossum95e6f701998-06-25 02:15:50 +0000443
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000444 def has_extn(self, opt):
445 """Does the server support a given SMTP service extension?"""
Raymond Hettinger54f02222002-06-01 14:18:47 +0000446 return opt.lower() in self.esmtp_features
Guido van Rossum95e6f701998-06-25 02:15:50 +0000447
Guido van Rossum18586f41998-04-03 17:03:13 +0000448 def help(self, args=''):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000449 """SMTP 'help' command.
450 Returns help text from server."""
Guido van Rossum18586f41998-04-03 17:03:13 +0000451 self.putcmd("help", args)
Kurt B. Kaiser58bd1902005-06-26 18:27:36 +0000452 return self.getreply()[1]
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000453
454 def rset(self):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000455 """SMTP 'rset' command -- resets session."""
Guido van Rossum296e1431999-04-07 15:03:39 +0000456 return self.docmd("rset")
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000457
458 def noop(self):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000459 """SMTP 'noop' command -- doesn't do anything :>"""
Guido van Rossum296e1431999-04-07 15:03:39 +0000460 return self.docmd("noop")
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000461
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000462 def mail(self, sender, options=[]):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000463 """SMTP 'mail' command -- begins mail xfer session."""
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000464 optionlist = ''
465 if options and self.does_esmtp:
Eric S. Raymondc013f302001-02-09 05:40:38 +0000466 optionlist = ' ' + ' '.join(options)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000467 self.putcmd("mail", "FROM:%s%s" % (quoteaddr(sender), optionlist))
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000468 return self.getreply()
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000469
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000470 def rcpt(self, recip, options=[]):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000471 """SMTP 'rcpt' command -- indicates 1 recipient for this mail."""
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000472 optionlist = ''
473 if options and self.does_esmtp:
Eric S. Raymondc013f302001-02-09 05:40:38 +0000474 optionlist = ' ' + ' '.join(options)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000475 self.putcmd("rcpt", "TO:%s%s" % (quoteaddr(recip), optionlist))
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000476 return self.getreply()
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000477
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000478 def data(self, msg):
Tim Peters495ad3c2001-01-15 01:36:40 +0000479 """SMTP 'DATA' command -- sends message data to server.
Guido van Rossum296e1431999-04-07 15:03:39 +0000480
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000481 Automatically quotes lines beginning with a period per rfc821.
Guido van Rossum296e1431999-04-07 15:03:39 +0000482 Raises SMTPDataError if there is an unexpected reply to the
483 DATA command; the return value from this method is the final
R. David Murray7dff9e02010-11-08 17:15:13 +0000484 response code received when the all data is sent. If msg
485 is a string, lone '\r' and '\n' characters are converted to
486 '\r\n' characters. If msg is bytes, it is transmitted as is.
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000487 """
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000488 self.putcmd("data")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000489 (code, repl) = self.getreply()
490 if self.debuglevel > 0:
491 print("data:", (code, repl), file=stderr)
Fred Drake8152d322000-12-12 23:20:45 +0000492 if code != 354:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000493 raise SMTPDataError(code, repl)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000494 else:
R. David Murray7dff9e02010-11-08 17:15:13 +0000495 if isinstance(msg, str):
496 msg = _fix_eols(msg).encode('ascii')
497 q = _quote_periods(msg)
498 if q[-2:] != bCRLF:
499 q = q + bCRLF
500 q = q + b"." + bCRLF
Guido van Rossum20c92281999-04-21 16:52:20 +0000501 self.send(q)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000502 (code, msg) = self.getreply()
503 if self.debuglevel > 0:
504 print("data:", (code, msg), file=stderr)
505 return (code, msg)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000506
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000507 def verify(self, address):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000508 """SMTP 'verify' command -- checks for address validity."""
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000509 self.putcmd("vrfy", quoteaddr(address))
510 return self.getreply()
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000511 # a.k.a.
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000512 vrfy = verify
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000513
514 def expn(self, address):
Christian Heimes90c3d9b2008-02-23 13:18:03 +0000515 """SMTP 'expn' command -- expands a mailing list."""
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000516 self.putcmd("expn", quoteaddr(address))
517 return self.getreply()
518
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000519 # some useful methods
Guido van Rossumae010462001-09-11 15:57:46 +0000520
Christian Heimes679db4a2008-01-18 09:56:22 +0000521 def ehlo_or_helo_if_needed(self):
522 """Call self.ehlo() and/or self.helo() if needed.
523
524 If there has been no previous EHLO or HELO command this session, this
525 method tries ESMTP EHLO first.
526
527 This method may raise the following exceptions:
528
529 SMTPHeloError The server didn't reply properly to
530 the helo greeting.
531 """
532 if self.helo_resp is None and self.ehlo_resp is None:
533 if not (200 <= self.ehlo()[0] <= 299):
534 (code, resp) = self.helo()
535 if not (200 <= code <= 299):
536 raise SMTPHeloError(code, resp)
537
Guido van Rossumae010462001-09-11 15:57:46 +0000538 def login(self, user, password):
539 """Log in on an SMTP server that requires authentication.
540
541 The arguments are:
542 - user: The user name to authenticate with.
543 - password: The password for the authentication.
544
545 If there has been no previous EHLO or HELO command this session, this
546 method tries ESMTP EHLO first.
547
548 This method will return normally if the authentication was successful.
549
550 This method may raise the following exceptions:
551
552 SMTPHeloError The server didn't reply properly to
553 the helo greeting.
554 SMTPAuthenticationError The server didn't accept the username/
555 password combination.
Fred Drake2f8f4d32001-10-13 18:35:32 +0000556 SMTPException No suitable authentication method was
Guido van Rossumae010462001-09-11 15:57:46 +0000557 found.
558 """
559
560 def encode_cram_md5(challenge, user, password):
Georg Brandl706824f2009-06-04 09:42:55 +0000561 challenge = base64.decodebytes(challenge)
R. David Murrayfb123912009-05-28 18:19:00 +0000562 response = user + " " + hmac.HMAC(password.encode('ascii'),
563 challenge).hexdigest()
564 return encode_base64(response.encode('ascii'), eol='')
Guido van Rossumae010462001-09-11 15:57:46 +0000565
566 def encode_plain(user, password):
R. David Murraycaa27b72009-05-23 18:49:56 +0000567 s = "\0%s\0%s" % (user, password)
568 return encode_base64(s.encode('ascii'), eol='')
Tim Peters469cdad2002-08-08 20:19:19 +0000569
Guido van Rossumae010462001-09-11 15:57:46 +0000570 AUTH_PLAIN = "PLAIN"
571 AUTH_CRAM_MD5 = "CRAM-MD5"
Piers Lauder385a77a2002-07-27 00:38:30 +0000572 AUTH_LOGIN = "LOGIN"
Guido van Rossumae010462001-09-11 15:57:46 +0000573
Christian Heimes679db4a2008-01-18 09:56:22 +0000574 self.ehlo_or_helo_if_needed()
Guido van Rossumae010462001-09-11 15:57:46 +0000575
576 if not self.has_extn("auth"):
577 raise SMTPException("SMTP AUTH extension not supported by server.")
578
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000579 # Authentication methods the server claims to support
580 advertised_authlist = self.esmtp_features["auth"].split()
Guido van Rossumae010462001-09-11 15:57:46 +0000581
582 # List of authentication methods we support: from preferred to
583 # less preferred methods. Except for the purpose of testing the weaker
584 # ones, we prefer stronger methods like CRAM-MD5:
Piers Lauder385a77a2002-07-27 00:38:30 +0000585 preferred_auths = [AUTH_CRAM_MD5, AUTH_PLAIN, AUTH_LOGIN]
Guido van Rossumae010462001-09-11 15:57:46 +0000586
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000587 # We try the authentication methods the server advertises, but only the
588 # ones *we* support. And in our preferred order.
589 authlist = [auth for auth in preferred_auths if auth in advertised_authlist]
590 if not authlist:
Fred Drake2f8f4d32001-10-13 18:35:32 +0000591 raise SMTPException("No suitable authentication method found.")
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000592
593 # Some servers advertise authentication methods they don't really
594 # support, so if authentication fails, we continue until we've tried
595 # all methods.
596 for authmethod in authlist:
597 if authmethod == AUTH_CRAM_MD5:
598 (code, resp) = self.docmd("AUTH", AUTH_CRAM_MD5)
599 if code == 334:
600 (code, resp) = self.docmd(encode_cram_md5(resp, user, password))
601 elif authmethod == AUTH_PLAIN:
602 (code, resp) = self.docmd("AUTH",
603 AUTH_PLAIN + " " + encode_plain(user, password))
604 elif authmethod == AUTH_LOGIN:
605 (code, resp) = self.docmd("AUTH",
606 "%s %s" % (AUTH_LOGIN, encode_base64(user.encode('ascii'), eol='')))
607 if code == 334:
608 (code, resp) = self.docmd(encode_base64(password.encode('ascii'), eol=''))
609
Guido van Rossumae010462001-09-11 15:57:46 +0000610 # 235 == 'Authentication successful'
611 # 503 == 'Error: already authenticated'
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000612 if code in (235, 503):
613 return (code, resp)
614
615 # We could not login sucessfully. Return result of last attempt.
616 raise SMTPAuthenticationError(code, resp)
Guido van Rossumae010462001-09-11 15:57:46 +0000617
Antoine Pitroue0650202011-05-18 18:03:09 +0200618 def starttls(self, keyfile=None, certfile=None, context=None):
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000619 """Puts the connection to the SMTP server into TLS mode.
Tim Petersb64bec32001-09-18 02:26:39 +0000620
Christian Heimes679db4a2008-01-18 09:56:22 +0000621 If there has been no previous EHLO or HELO command this session, this
622 method tries ESMTP EHLO first.
623
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000624 If the server supports TLS, this will encrypt the rest of the SMTP
625 session. If you provide the keyfile and certfile parameters,
626 the identity of the SMTP server and client can be checked. This,
627 however, depends on whether the socket module really checks the
628 certificates.
Christian Heimes679db4a2008-01-18 09:56:22 +0000629
630 This method may raise the following exceptions:
631
632 SMTPHeloError The server didn't reply properly to
633 the helo greeting.
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000634 """
Christian Heimes679db4a2008-01-18 09:56:22 +0000635 self.ehlo_or_helo_if_needed()
636 if not self.has_extn("starttls"):
637 raise SMTPException("STARTTLS extension not supported by server.")
Tim Petersb64bec32001-09-18 02:26:39 +0000638 (resp, reply) = self.docmd("STARTTLS")
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000639 if resp == 220:
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000640 if not _have_ssl:
641 raise RuntimeError("No SSL support included in this Python")
Antoine Pitroue0650202011-05-18 18:03:09 +0200642 if context is not None and keyfile is not None:
643 raise ValueError("context and keyfile arguments are mutually "
644 "exclusive")
645 if context is not None and certfile is not None:
646 raise ValueError("context and certfile arguments are mutually "
647 "exclusive")
648 if context is not None:
649 self.sock = context.wrap_socket(self.sock)
650 else:
651 self.sock = ssl.wrap_socket(self.sock, keyfile, certfile)
Antoine Pitrouf068ab82011-06-06 19:17:09 +0200652 self.file = None
Christian Heimes679db4a2008-01-18 09:56:22 +0000653 # RFC 3207:
654 # The client MUST discard any knowledge obtained from
655 # the server, such as the list of SMTP service extensions,
656 # which was not obtained from the TLS negotiation itself.
657 self.helo_resp = None
658 self.ehlo_resp = None
659 self.esmtp_features = {}
660 self.does_esmtp = 0
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000661 return (resp, reply)
Tim Petersb64bec32001-09-18 02:26:39 +0000662
Jeremy Hylton31bb8ce1998-08-13 19:57:46 +0000663 def sendmail(self, from_addr, to_addrs, msg, mail_options=[],
Tim Peters495ad3c2001-01-15 01:36:40 +0000664 rcpt_options=[]):
665 """This command performs an entire mail transaction.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000666
Tim Peters495ad3c2001-01-15 01:36:40 +0000667 The arguments are:
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000668 - from_addr : The address sending this mail.
669 - to_addrs : A list of addresses to send this mail to. A bare
670 string will be treated as a list with 1 address.
Tim Peters495ad3c2001-01-15 01:36:40 +0000671 - msg : The message to send.
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000672 - mail_options : List of ESMTP options (such as 8bitmime) for the
673 mail command.
674 - rcpt_options : List of ESMTP options (such as DSN commands) for
675 all the rcpt commands.
676
R. David Murray7dff9e02010-11-08 17:15:13 +0000677 msg may be a string containing characters in the ASCII range, or a byte
678 string. A string is encoded to bytes using the ascii codec, and lone
679 \r and \n characters are converted to \r\n characters.
680
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000681 If there has been no previous EHLO or HELO command this session, this
682 method tries ESMTP EHLO first. If the server does ESMTP, message size
683 and each of the specified options will be passed to it. If EHLO
684 fails, HELO will be tried and ESMTP options suppressed.
685
686 This method will return normally if the mail is accepted for at least
Barry Warsawd25c1b71999-11-28 17:11:06 +0000687 one recipient. It returns a dictionary, with one entry for each
688 recipient that was refused. Each entry contains a tuple of the SMTP
689 error code and the accompanying error message sent by the server.
Guido van Rossum296e1431999-04-07 15:03:39 +0000690
691 This method may raise the following exceptions:
692
693 SMTPHeloError The server didn't reply properly to
Tim Peters495ad3c2001-01-15 01:36:40 +0000694 the helo greeting.
Barry Warsawd25c1b71999-11-28 17:11:06 +0000695 SMTPRecipientsRefused The server rejected ALL recipients
Guido van Rossum296e1431999-04-07 15:03:39 +0000696 (no mail was sent).
697 SMTPSenderRefused The server didn't accept the from_addr.
698 SMTPDataError The server replied with an unexpected
699 error code (other than a refusal of
700 a recipient).
701
702 Note: the connection will be open even after an exception is raised.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000703
Guido van Rossum95e6f701998-06-25 02:15:50 +0000704 Example:
Tim Peters495ad3c2001-01-15 01:36:40 +0000705
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000706 >>> import smtplib
707 >>> s=smtplib.SMTP("localhost")
Guido van Rossumfc40a831998-01-29 17:26:45 +0000708 >>> tolist=["one@one.org","two@two.org","three@three.org","four@four.org"]
Martin v. Löwis301b1cd2002-07-28 16:52:01 +0000709 >>> msg = '''\\
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000710 ... From: Me@my.org
711 ... Subject: testin'...
712 ...
713 ... This is a test '''
714 >>> s.sendmail("me@my.org",tolist,msg)
715 { "three@three.org" : ( 550 ,"User unknown" ) }
716 >>> s.quit()
Tim Peters495ad3c2001-01-15 01:36:40 +0000717
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000718 In the above example, the message was accepted for delivery to three
719 of the four addresses, and one was rejected, with the error code
Barry Warsawd25c1b71999-11-28 17:11:06 +0000720 550. If all addresses are accepted, then the method will return an
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000721 empty dictionary.
722
723 """
Christian Heimes679db4a2008-01-18 09:56:22 +0000724 self.ehlo_or_helo_if_needed()
Guido van Rossum95e6f701998-06-25 02:15:50 +0000725 esmtp_opts = []
R. David Murray7dff9e02010-11-08 17:15:13 +0000726 if isinstance(msg, str):
727 msg = _fix_eols(msg).encode('ascii')
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000728 if self.does_esmtp:
729 # Hmmm? what's this? -ddm
730 # self.esmtp_features['7bit']=""
731 if self.has_extn('size'):
Walter Dörwald70a6b492004-02-12 17:35:32 +0000732 esmtp_opts.append("size=%d" % len(msg))
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000733 for option in mail_options:
Guido van Rossum95e6f701998-06-25 02:15:50 +0000734 esmtp_opts.append(option)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000735 (code, resp) = self.mail(from_addr, esmtp_opts)
Fred Drake8152d322000-12-12 23:20:45 +0000736 if code != 250:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000737 self.rset()
Guido van Rossum296e1431999-04-07 15:03:39 +0000738 raise SMTPSenderRefused(code, resp, from_addr)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000739 senderrs = {}
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000740 if isinstance(to_addrs, str):
Jeremy Hylton31bb8ce1998-08-13 19:57:46 +0000741 to_addrs = [to_addrs]
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000742 for each in to_addrs:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000743 (code, resp) = self.rcpt(each, rcpt_options)
Fred Drake8152d322000-12-12 23:20:45 +0000744 if (code != 250) and (code != 251):
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000745 senderrs[each] = (code, resp)
746 if len(senderrs) == len(to_addrs):
Guido van Rossum95e6f701998-06-25 02:15:50 +0000747 # the server refused all our recipients
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000748 self.rset()
Guido van Rossum296e1431999-04-07 15:03:39 +0000749 raise SMTPRecipientsRefused(senderrs)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000750 (code, resp) = self.data(msg)
Fred Drake8152d322000-12-12 23:20:45 +0000751 if code != 250:
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000752 self.rset()
Guido van Rossum296e1431999-04-07 15:03:39 +0000753 raise SMTPDataError(code, resp)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000754 #if we got here then somebody got our mail
Tim Peters495ad3c2001-01-15 01:36:40 +0000755 return senderrs
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000756
R. David Murray7dff9e02010-11-08 17:15:13 +0000757 def send_message(self, msg, from_addr=None, to_addrs=None,
758 mail_options=[], rcpt_options={}):
759 """Converts message to a bytestring and passes it to sendmail.
760
761 The arguments are as for sendmail, except that msg is an
762 email.message.Message object. If from_addr is None, the from_addr is
763 taken from the 'From' header of the Message. If to_addrs is None, its
764 value is composed from the addresses listed in the 'To', 'CC', and
765 'Bcc' fields. Regardless of the values of from_addr and to_addr, any
766 Bcc field in the Message object is deleted. The Message object is then
767 serialized using email.generator.BytesGenerator and sendmail is called
768 to transmit the message.
769 """
770 if from_addr is None:
771 from_addr = msg['From']
772 if to_addrs is None:
773 addr_fields = [f for f in (msg['To'], msg['Bcc'], msg['CC'])
774 if f is not None]
775 to_addrs = [a[1] for a in email.utils.getaddresses(addr_fields)]
776 del msg['Bcc']
777 with io.BytesIO() as bytesmsg:
778 g = email.generator.BytesGenerator(bytesmsg)
779 g.flatten(msg, linesep='\r\n')
780 flatmsg = bytesmsg.getvalue()
781 return self.sendmail(from_addr, to_addrs, flatmsg, mail_options,
782 rcpt_options)
783
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000784 def close(self):
785 """Close the connection to the SMTP server."""
786 if self.file:
787 self.file.close()
788 self.file = None
789 if self.sock:
790 self.sock.close()
791 self.sock = None
792
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000793 def quit(self):
Guido van Rossum95e6f701998-06-25 02:15:50 +0000794 """Terminate the SMTP session."""
Christian Heimesba4af492008-03-28 00:55:15 +0000795 res = self.docmd("quit")
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000796 self.close()
Christian Heimesba4af492008-03-28 00:55:15 +0000797 return res
Guido van Rossum95e6f701998-06-25 02:15:50 +0000798
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000799if _have_ssl:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000800
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000801 class SMTP_SSL(SMTP):
802 """ This is a subclass derived from SMTP that connects over an SSL encrypted
803 socket (to use this class you need a socket module that was compiled with SSL
804 support). If host is not specified, '' (the local host) is used. If port is
805 omitted, the standard SMTP-over-SSL port (465) is used. keyfile and certfile
806 are also optional - they can contain a PEM formatted private key and
Antoine Pitroue0650202011-05-18 18:03:09 +0200807 certificate chain file for the SSL connection. context also optional, can contain
808 a SSLContext, and is an alternative to keyfile and certfile; If it is specified both
809 keyfile and certfile must be None.
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000810 """
Antoine Pitrouc1d52062011-05-07 19:39:37 +0200811
812 default_port = SMTP_SSL_PORT
813
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000814 def __init__(self, host='', port=0, local_hostname=None,
Georg Brandlf78e02b2008-06-10 17:40:04 +0000815 keyfile=None, certfile=None,
Antoine Pitroue0650202011-05-18 18:03:09 +0200816 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, context=None):
817 if context is not None and keyfile is not None:
818 raise ValueError("context and keyfile arguments are mutually "
819 "exclusive")
820 if context is not None and certfile is not None:
821 raise ValueError("context and certfile arguments are mutually "
822 "exclusive")
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000823 self.keyfile = keyfile
824 self.certfile = certfile
Antoine Pitroue0650202011-05-18 18:03:09 +0200825 self.context = context
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000826 SMTP.__init__(self, host, port, local_hostname, timeout)
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000827
828 def _get_socket(self, host, port, timeout):
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000829 if self.debuglevel > 0:
830 print('connect:', (host, port), file=stderr)
R. David Murray87e20742009-05-23 01:30:26 +0000831 new_socket = socket.create_connection((host, port), timeout)
Antoine Pitroue0650202011-05-18 18:03:09 +0200832 if self.context is not None:
833 new_socket = self.context.wrap_socket(new_socket)
834 else:
835 new_socket = ssl.wrap_socket(new_socket, self.keyfile, self.certfile)
R. David Murray87e20742009-05-23 01:30:26 +0000836 return new_socket
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000837
838 __all__.append("SMTP_SSL")
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000839
Guido van Rossumd8faa362007-04-27 19:54:29 +0000840#
841# LMTP extension
842#
843LMTP_PORT = 2003
844
845class LMTP(SMTP):
846 """LMTP - Local Mail Transfer Protocol
847
848 The LMTP protocol, which is very similar to ESMTP, is heavily based
849 on the standard SMTP client. It's common to use Unix sockets for LMTP,
850 so our connect() method must support that as well as a regular
851 host:port server. To specify a Unix socket, you must use an absolute
852 path as the host, starting with a '/'.
853
854 Authentication is supported, using the regular SMTP mechanism. When
855 using a Unix socket, LMTP generally don't support or require any
856 authentication, but your mileage might vary."""
857
858 ehlo_msg = "lhlo"
859
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000860 def __init__(self, host='', port=LMTP_PORT, local_hostname=None):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000861 """Initialize a new instance."""
862 SMTP.__init__(self, host, port, local_hostname)
863
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000864 def connect(self, host='localhost', port=0):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000865 """Connect to the LMTP daemon, on either a Unix or a TCP socket."""
866 if host[0] != '/':
867 return SMTP.connect(self, host, port)
868
869 # Handle Unix-domain sockets.
870 try:
871 self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Antoine Pitrouf068ab82011-06-06 19:17:09 +0200872 self.file = None
Guido van Rossumd8faa362007-04-27 19:54:29 +0000873 self.sock.connect(host)
874 except socket.error as msg:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000875 if self.debuglevel > 0:
876 print('connect fail:', host, file=stderr)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000877 if self.sock:
878 self.sock.close()
879 self.sock = None
880 raise socket.error(msg)
881 (code, msg) = self.getreply()
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000882 if self.debuglevel > 0:
883 print('connect:', msg, file=stderr)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000884 return (code, msg)
885
886
Guido van Rossum95e6f701998-06-25 02:15:50 +0000887# Test the sendmail method, which tests most of the others.
888# Note: This always sends to localhost.
889if __name__ == '__main__':
Andrew M. Kuchling6be424f2001-08-13 14:41:39 +0000890 import sys
Guido van Rossum95e6f701998-06-25 02:15:50 +0000891
892 def prompt(prompt):
893 sys.stdout.write(prompt + ": ")
Eric S. Raymondc013f302001-02-09 05:40:38 +0000894 return sys.stdin.readline().strip()
Guido van Rossum95e6f701998-06-25 02:15:50 +0000895
896 fromaddr = prompt("From")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000897 toaddrs = prompt("To").split(',')
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000898 print("Enter message, end with ^D:")
Guido van Rossum95e6f701998-06-25 02:15:50 +0000899 msg = ''
900 while 1:
901 line = sys.stdin.readline()
902 if not line:
903 break
904 msg = msg + line
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000905 print("Message length is %d" % len(msg))
Guido van Rossum95e6f701998-06-25 02:15:50 +0000906
907 server = SMTP('localhost')
908 server.set_debuglevel(1)
909 server.sendmail(fromaddr, toaddrs, msg)
910 server.quit()