blob: c7ff821bd67ed01c56205ffae19bbea4873ad4d0 [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
R David Murrayac4e5ab2011-07-02 21:03:19 -040052import copy
Barry Warsaw2cc1f6d2007-08-30 14:28:55 +000053from email.base64mime import body_encode as encode_base64
Brett Cannone6f8a892004-07-10 23:14:30 +000054from sys import stderr
Guido van Rossumbbe323e1998-01-29 17:24:40 +000055
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +000056__all__ = ["SMTPException", "SMTPServerDisconnected", "SMTPResponseException",
57 "SMTPSenderRefused", "SMTPRecipientsRefused", "SMTPDataError",
58 "SMTPConnectError", "SMTPHeloError", "SMTPAuthenticationError",
59 "quoteaddr", "quotedata", "SMTP"]
Skip Montanaro0de65802001-02-15 22:15:14 +000060
Guido van Rossumbbe323e1998-01-29 17:24:40 +000061SMTP_PORT = 25
Thomas Wouters89f507f2006-12-13 04:49:30 +000062SMTP_SSL_PORT = 465
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +000063CRLF = "\r\n"
64bCRLF = b"\r\n"
Guido van Rossumbbe323e1998-01-29 17:24:40 +000065
Piers Lauder385a77a2002-07-27 00:38:30 +000066OLDSTYLE_AUTH = re.compile(r"auth=(.*)", re.I)
67
Tim Peters495ad3c2001-01-15 01:36:40 +000068# Exception classes used by this module.
Guido van Rossum296e1431999-04-07 15:03:39 +000069class SMTPException(Exception):
70 """Base class for all exceptions raised by this module."""
71
72class SMTPServerDisconnected(SMTPException):
73 """Not connected to any SMTP server.
74
75 This exception is raised when the server unexpectedly disconnects,
76 or when an attempt is made to use the SMTP instance before
77 connecting it to a server.
78 """
79
80class SMTPResponseException(SMTPException):
81 """Base class for all exceptions that include an SMTP error code.
82
83 These exceptions are generated in some instances when the SMTP
84 server returns an error code. The error code is stored in the
85 `smtp_code' attribute of the error, and the `smtp_error' attribute
86 is set to the error message.
87 """
88
89 def __init__(self, code, msg):
90 self.smtp_code = code
91 self.smtp_error = msg
92 self.args = (code, msg)
93
94class SMTPSenderRefused(SMTPResponseException):
95 """Sender address refused.
Guido van Rossumae010462001-09-11 15:57:46 +000096
Guido van Rossum296e1431999-04-07 15:03:39 +000097 In addition to the attributes set by on all SMTPResponseException
Barry Warsawd25c1b71999-11-28 17:11:06 +000098 exceptions, this sets `sender' to the string that the SMTP refused.
Guido van Rossum296e1431999-04-07 15:03:39 +000099 """
100
101 def __init__(self, code, msg, sender):
102 self.smtp_code = code
103 self.smtp_error = msg
104 self.sender = sender
105 self.args = (code, msg, sender)
106
Guido van Rossum20c92281999-04-21 16:52:20 +0000107class SMTPRecipientsRefused(SMTPException):
Barry Warsawd25c1b71999-11-28 17:11:06 +0000108 """All recipient addresses refused.
Guido van Rossumae010462001-09-11 15:57:46 +0000109
Thomas Wouters7e474022000-07-16 12:04:32 +0000110 The errors for each recipient are accessible through the attribute
Tim Peters495ad3c2001-01-15 01:36:40 +0000111 'recipients', which is a dictionary of exactly the same sort as
112 SMTP.sendmail() returns.
Guido van Rossum296e1431999-04-07 15:03:39 +0000113 """
114
115 def __init__(self, recipients):
116 self.recipients = recipients
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000117 self.args = (recipients,)
Guido van Rossum296e1431999-04-07 15:03:39 +0000118
119
Guido van Rossum296e1431999-04-07 15:03:39 +0000120class SMTPDataError(SMTPResponseException):
121 """The SMTP server didn't accept the data."""
122
123class SMTPConnectError(SMTPResponseException):
Barry Warsawd25c1b71999-11-28 17:11:06 +0000124 """Error during connection establishment."""
Guido van Rossum296e1431999-04-07 15:03:39 +0000125
126class SMTPHeloError(SMTPResponseException):
Barry Warsawd25c1b71999-11-28 17:11:06 +0000127 """The server refused our HELO reply."""
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000128
Guido van Rossumae010462001-09-11 15:57:46 +0000129class SMTPAuthenticationError(SMTPResponseException):
130 """Authentication error.
131
132 Most probably the server didn't accept the username/password
133 combination provided.
134 """
Peter Schneider-Kamp7bc82bb2000-08-10 14:02:23 +0000135
R David Murray4c14bba2011-07-18 21:59:53 -0400136def quoteaddr(addrstring):
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000137 """Quote a subset of the email addresses defined by RFC 821.
138
Georg Brandl9f0f9602008-06-12 22:23:59 +0000139 Should be able to handle anything email.utils.parseaddr can handle.
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000140 """
R David Murray4c14bba2011-07-18 21:59:53 -0400141 displayname, addr = email.utils.parseaddr(addrstring)
142 if (displayname, addr) == ('', ''):
143 # parseaddr couldn't parse it, use it as is and hope for the best.
144 if addrstring.strip().startswith('<'):
145 return addrstring
146 return "<%s>" % addrstring
147 return "<%s>" % addr
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000148
R David Murray46346762011-07-18 21:38:54 -0400149def _addr_only(addrstring):
150 displayname, addr = email.utils.parseaddr(addrstring)
151 if (displayname, addr) == ('', ''):
152 # parseaddr couldn't parse it, so use it as is.
153 return addrstring
154 return addr
155
R. David Murray7dff9e02010-11-08 17:15:13 +0000156# Legacy method kept for backward compatibility.
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000157def quotedata(data):
158 """Quote data for email.
159
Barry Warsawd25c1b71999-11-28 17:11:06 +0000160 Double leading '.', and change Unix newline '\\n', or Mac '\\r' into
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000161 Internet CRLF end-of-line.
162 """
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000163 return re.sub(r'(?m)^\.', '..',
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000164 re.sub(r'(?:\r\n|\n|\r(?!\n))', CRLF, data))
Guido van Rossum69a79bc1998-07-13 15:18:49 +0000165
R. David Murray7dff9e02010-11-08 17:15:13 +0000166def _quote_periods(bindata):
R David Murray0f663d02011-06-09 15:05:57 -0400167 return re.sub(br'(?m)^\.', b'..', bindata)
R. David Murray7dff9e02010-11-08 17:15:13 +0000168
169def _fix_eols(data):
170 return re.sub(r'(?:\r\n|\n|\r(?!\n))', CRLF, data)
171
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000172try:
173 import ssl
174except ImportError:
175 _have_ssl = False
176else:
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000177 _have_ssl = True
178
Peter Schneider-Kamp7bc82bb2000-08-10 14:02:23 +0000179
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000180class SMTP:
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000181 """This class manages a connection to an SMTP or ESMTP server.
182 SMTP Objects:
Tim Peters495ad3c2001-01-15 01:36:40 +0000183 SMTP objects have the following attributes:
184 helo_resp
185 This is the message given by the server in response to the
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000186 most recent HELO command.
Tim Peters495ad3c2001-01-15 01:36:40 +0000187
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000188 ehlo_resp
Tim Peters495ad3c2001-01-15 01:36:40 +0000189 This is the message given by the server in response to the
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000190 most recent EHLO command. This is usually multiline.
191
Tim Peters495ad3c2001-01-15 01:36:40 +0000192 does_esmtp
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000193 This is a True value _after you do an EHLO command_, if the
194 server supports ESMTP.
195
Tim Peters495ad3c2001-01-15 01:36:40 +0000196 esmtp_features
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000197 This is a dictionary, which, if the server supports ESMTP,
Barry Warsawd25c1b71999-11-28 17:11:06 +0000198 will _after you do an EHLO command_, contain the names of the
199 SMTP service extensions this server supports, and their
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000200 parameters (if any).
Barry Warsawd25c1b71999-11-28 17:11:06 +0000201
Tim Peters495ad3c2001-01-15 01:36:40 +0000202 Note, all extension names are mapped to lower case in the
203 dictionary.
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000204
Barry Warsawd25c1b71999-11-28 17:11:06 +0000205 See each method's docstrings for details. In general, there is a
206 method of the same name to perform each SMTP command. There is also a
207 method called 'sendmail' that will do an entire mail transaction.
208 """
Guido van Rossum95e6f701998-06-25 02:15:50 +0000209 debuglevel = 0
210 file = None
211 helo_resp = None
Guido van Rossumd8faa362007-04-27 19:54:29 +0000212 ehlo_msg = "ehlo"
Guido van Rossum95e6f701998-06-25 02:15:50 +0000213 ehlo_resp = None
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000214 does_esmtp = 0
Antoine Pitrouc1d52062011-05-07 19:39:37 +0200215 default_port = SMTP_PORT
Guido van Rossum95e6f701998-06-25 02:15:50 +0000216
Georg Brandlf78e02b2008-06-10 17:40:04 +0000217 def __init__(self, host='', port=0, local_hostname=None,
218 timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000219 """Initialize a new instance.
220
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000221 If specified, `host' is the name of the remote host to which to
222 connect. If specified, `port' specifies the port to which to connect.
Barry Warsawd25c1b71999-11-28 17:11:06 +0000223 By default, smtplib.SMTP_PORT is used. An SMTPConnectError is raised
Neil Schemenauer6730f262002-03-24 15:30:40 +0000224 if the specified `host' doesn't respond correctly. If specified,
Tim Peters863ac442002-04-16 01:38:40 +0000225 `local_hostname` is used as the FQDN of the local host. By default,
226 the local hostname is found using socket.getfqdn().
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000227
228 """
Guido van Rossumd8faa362007-04-27 19:54:29 +0000229 self.timeout = timeout
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000230 self.esmtp_features = {}
Guido van Rossum296e1431999-04-07 15:03:39 +0000231 if host:
232 (code, msg) = self.connect(host, port)
233 if code != 220:
234 raise SMTPConnectError(code, msg)
Raymond Hettingerf13eb552002-06-02 00:40:05 +0000235 if local_hostname is not None:
Barry Warsaw13e34f72002-03-26 20:27:35 +0000236 self.local_hostname = local_hostname
Neil Schemenauer6730f262002-03-24 15:30:40 +0000237 else:
Barry Warsaw13e34f72002-03-26 20:27:35 +0000238 # RFC 2821 says we should use the fqdn in the EHLO/HELO verb, and
239 # if that can't be calculated, that we should use a domain literal
240 # instead (essentially an encoded IP address like [A.B.C.D]).
241 fqdn = socket.getfqdn()
242 if '.' in fqdn:
243 self.local_hostname = fqdn
244 else:
245 # We can't find an fqdn hostname, so use a domain literal
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000246 addr = '127.0.0.1'
247 try:
248 addr = socket.gethostbyname(socket.gethostname())
249 except socket.gaierror:
250 pass
Barry Warsaw13e34f72002-03-26 20:27:35 +0000251 self.local_hostname = '[%s]' % addr
Tim Peters495ad3c2001-01-15 01:36:40 +0000252
Barry Warsaw1f5c9582011-03-15 15:04:44 -0400253 def __enter__(self):
254 return self
255
256 def __exit__(self, *args):
257 try:
258 code, message = self.docmd("QUIT")
259 if code != 221:
260 raise SMTPResponseException(code, message)
261 except SMTPServerDisconnected:
262 pass
263 finally:
264 self.close()
265
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000266 def set_debuglevel(self, debuglevel):
267 """Set the debug output level.
268
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000269 A non-false value results in debug messages for connection and for all
270 messages sent to and received from the server.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000271
272 """
273 self.debuglevel = debuglevel
274
Barry Warsawc9181712008-03-19 14:25:51 +0000275 def _get_socket(self, host, port, timeout):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000276 # This makes it simpler for SMTP_SSL to use the SMTP connect code
277 # and just alter the socket connection bit.
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000278 if self.debuglevel > 0:
279 print('connect:', (host, port), file=stderr)
Barry Warsawc9181712008-03-19 14:25:51 +0000280 return socket.create_connection((host, port), timeout)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000281
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000282 def connect(self, host='localhost', port=0):
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000283 """Connect to a host on a given port.
Guido van Rossum95e6f701998-06-25 02:15:50 +0000284
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000285 If the hostname ends with a colon (`:') followed by a number, and
286 there is no port specified, that suffix will be stripped off and the
287 number interpreted as the port number to use.
Guido van Rossum95e6f701998-06-25 02:15:50 +0000288
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000289 Note: This method is automatically invoked by __init__, if a host is
290 specified during instantiation.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000291
292 """
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000293 if not port and (host.find(':') == host.rfind(':')):
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000294 i = host.rfind(':')
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000295 if i >= 0:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000296 host, port = host[:i], host[i + 1:]
297 try:
298 port = int(port)
Eric S. Raymond8d876032001-02-09 10:14:53 +0000299 except ValueError:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000300 raise socket.error("nonnumeric port")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000301 if not port:
302 port = self.default_port
303 if self.debuglevel > 0:
304 print('connect:', (host, port), file=stderr)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000305 self.sock = self._get_socket(host, port, self.timeout)
Antoine Pitrouf068ab82011-06-06 19:17:09 +0200306 self.file = None
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000307 (code, msg) = self.getreply()
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000308 if self.debuglevel > 0:
309 print("connect:", msg, file=stderr)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000310 return (code, msg)
Tim Peters495ad3c2001-01-15 01:36:40 +0000311
Guido van Rossum8a392d72007-11-21 22:09:45 +0000312 def send(self, s):
313 """Send `s' to the server."""
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000314 if self.debuglevel > 0:
315 print('send:', repr(s), file=stderr)
Christian Heimes90c3d9b2008-02-23 13:18:03 +0000316 if hasattr(self, 'sock') and self.sock:
Guido van Rossum8a392d72007-11-21 22:09:45 +0000317 if isinstance(s, str):
318 s = s.encode("ascii")
Guido van Rossum2880f6e1998-08-10 20:07:00 +0000319 try:
Guido van Rossum8a392d72007-11-21 22:09:45 +0000320 self.sock.sendall(s)
Guido van Rossum2880f6e1998-08-10 20:07:00 +0000321 except socket.error:
Barry Warsaw76750972001-12-14 20:34:20 +0000322 self.close()
Guido van Rossum40233ea1999-01-15 03:23:55 +0000323 raise SMTPServerDisconnected('Server not connected')
Guido van Rossumfc40a831998-01-29 17:26:45 +0000324 else:
Guido van Rossum40233ea1999-01-15 03:23:55 +0000325 raise SMTPServerDisconnected('please run connect() first')
Tim Peters495ad3c2001-01-15 01:36:40 +0000326
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000327 def putcmd(self, cmd, args=""):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000328 """Send a command to the server."""
Guido van Rossumdb23d3d1999-06-09 15:13:10 +0000329 if args == "":
330 str = '%s%s' % (cmd, CRLF)
331 else:
332 str = '%s %s%s' % (cmd, args, CRLF)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000333 self.send(str)
Tim Peters495ad3c2001-01-15 01:36:40 +0000334
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000335 def getreply(self):
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000336 """Get a reply from the server.
Tim Peters495ad3c2001-01-15 01:36:40 +0000337
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000338 Returns a tuple consisting of:
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000339
340 - server response code (e.g. '250', or such, if all goes well)
341 Note: returns -1 if it can't read response code.
342
343 - server response string corresponding to response code (multiline
344 responses are converted to a single, multiline string).
Guido van Rossumf123f841999-03-29 20:33:21 +0000345
346 Raises SMTPServerDisconnected if end-of-file is reached.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000347 """
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000348 resp = []
Guido van Rossum296e1431999-04-07 15:03:39 +0000349 if self.file is None:
350 self.file = self.sock.makefile('rb')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000351 while 1:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000352 try:
353 line = self.file.readline()
354 except socket.error:
355 line = ''
Guido van Rossum806c2462007-08-06 23:33:07 +0000356 if not line:
Guido van Rossum296e1431999-04-07 15:03:39 +0000357 self.close()
358 raise SMTPServerDisconnected("Connection unexpectedly closed")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000359 if self.debuglevel > 0:
360 print('reply:', repr(line), file=stderr)
Guido van Rossum806c2462007-08-06 23:33:07 +0000361 resp.append(line[4:].strip(b' \t\r\n'))
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000362 code = line[:3]
Guido van Rossum296e1431999-04-07 15:03:39 +0000363 # Check that the error code is syntactically correct.
364 # Don't attempt to read a continuation line if it is broken.
365 try:
Eric S. Raymondc013f302001-02-09 05:40:38 +0000366 errcode = int(code)
Guido van Rossum296e1431999-04-07 15:03:39 +0000367 except ValueError:
368 errcode = -1
369 break
Guido van Rossumf123f841999-03-29 20:33:21 +0000370 # Check if multiline response.
Guido van Rossum806c2462007-08-06 23:33:07 +0000371 if line[3:4] != b"-":
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000372 break
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000373
Guido van Rossumc43e79f2007-06-18 18:26:36 +0000374 errmsg = b"\n".join(resp)
Tim Peters495ad3c2001-01-15 01:36:40 +0000375 if self.debuglevel > 0:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000376 print('reply: retcode (%s); Msg: %s' % (errcode, errmsg), file=stderr)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000377 return errcode, errmsg
Tim Peters495ad3c2001-01-15 01:36:40 +0000378
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000379 def docmd(self, cmd, args=""):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000380 """Send a command, and return its response code."""
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000381 self.putcmd(cmd, args)
Guido van Rossum296e1431999-04-07 15:03:39 +0000382 return self.getreply()
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000383
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000384 # std smtp commands
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000385 def helo(self, name=''):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000386 """SMTP 'helo' command.
387 Hostname to send for this command defaults to the FQDN of the local
388 host.
389 """
Neil Schemenauer6730f262002-03-24 15:30:40 +0000390 self.putcmd("helo", name or self.local_hostname)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000391 (code, msg) = self.getreply()
392 self.helo_resp = msg
393 return (code, msg)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000394
Guido van Rossum95e6f701998-06-25 02:15:50 +0000395 def ehlo(self, name=''):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000396 """ SMTP 'ehlo' command.
397 Hostname to send for this command defaults to the FQDN of the local
398 host.
399 """
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000400 self.esmtp_features = {}
Guido van Rossumd8faa362007-04-27 19:54:29 +0000401 self.putcmd(self.ehlo_msg, name or self.local_hostname)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000402 (code, msg) = self.getreply()
Tim Peters495ad3c2001-01-15 01:36:40 +0000403 # According to RFC1869 some (badly written)
404 # MTA's will disconnect on an ehlo. Toss an exception if
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000405 # that happens -ddm
406 if code == -1 and len(msg) == 0:
Barry Warsaw76750972001-12-14 20:34:20 +0000407 self.close()
Guido van Rossum40233ea1999-01-15 03:23:55 +0000408 raise SMTPServerDisconnected("Server not connected")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000409 self.ehlo_resp = msg
Fred Drake8152d322000-12-12 23:20:45 +0000410 if code != 250:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000411 return (code, msg)
412 self.does_esmtp = 1
Thomas Wouters7e474022000-07-16 12:04:32 +0000413 #parse the ehlo response -ddm
Guido van Rossum04110fb2007-08-24 16:32:05 +0000414 assert isinstance(self.ehlo_resp, bytes), repr(self.ehlo_resp)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000415 resp = self.ehlo_resp.decode("latin-1").split('\n')
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000416 del resp[0]
Guido van Rossum2880f6e1998-08-10 20:07:00 +0000417 for each in resp:
Piers Lauder385a77a2002-07-27 00:38:30 +0000418 # To be able to communicate with as many SMTP servers as possible,
419 # we have to take the old-style auth advertisement into account,
420 # because:
421 # 1) Else our SMTP feature parser gets confused.
422 # 2) There are some servers that only advertise the auth methods we
423 # support using the old style.
424 auth_match = OLDSTYLE_AUTH.match(each)
425 if auth_match:
426 # This doesn't remove duplicates, but that's no problem
427 self.esmtp_features["auth"] = self.esmtp_features.get("auth", "") \
428 + " " + auth_match.groups(0)[0]
429 continue
430
Barry Warsawbe22ae62002-04-15 20:03:30 +0000431 # RFC 1869 requires a space between ehlo keyword and parameters.
432 # It's actually stricter, in that only spaces are allowed between
433 # parameters, but were not going to check for that here. Note
434 # that the space isn't present if there are no parameters.
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000435 m = re.match(r'(?P<feature>[A-Za-z0-9][A-Za-z0-9\-]*) ?', each)
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000436 if m:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000437 feature = m.group("feature").lower()
438 params = m.string[m.end("feature"):].strip()
Piers Lauder385a77a2002-07-27 00:38:30 +0000439 if feature == "auth":
440 self.esmtp_features[feature] = self.esmtp_features.get(feature, "") \
441 + " " + params
442 else:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000443 self.esmtp_features[feature] = params
444 return (code, msg)
Guido van Rossum95e6f701998-06-25 02:15:50 +0000445
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000446 def has_extn(self, opt):
447 """Does the server support a given SMTP service extension?"""
Raymond Hettinger54f02222002-06-01 14:18:47 +0000448 return opt.lower() in self.esmtp_features
Guido van Rossum95e6f701998-06-25 02:15:50 +0000449
Guido van Rossum18586f41998-04-03 17:03:13 +0000450 def help(self, args=''):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000451 """SMTP 'help' command.
452 Returns help text from server."""
Guido van Rossum18586f41998-04-03 17:03:13 +0000453 self.putcmd("help", args)
Kurt B. Kaiser58bd1902005-06-26 18:27:36 +0000454 return self.getreply()[1]
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000455
456 def rset(self):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000457 """SMTP 'rset' command -- resets session."""
Guido van Rossum296e1431999-04-07 15:03:39 +0000458 return self.docmd("rset")
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000459
460 def noop(self):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000461 """SMTP 'noop' command -- doesn't do anything :>"""
Guido van Rossum296e1431999-04-07 15:03:39 +0000462 return self.docmd("noop")
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000463
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000464 def mail(self, sender, options=[]):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000465 """SMTP 'mail' command -- begins mail xfer session."""
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000466 optionlist = ''
467 if options and self.does_esmtp:
Eric S. Raymondc013f302001-02-09 05:40:38 +0000468 optionlist = ' ' + ' '.join(options)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000469 self.putcmd("mail", "FROM:%s%s" % (quoteaddr(sender), optionlist))
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000470 return self.getreply()
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000471
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000472 def rcpt(self, recip, options=[]):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000473 """SMTP 'rcpt' command -- indicates 1 recipient for this mail."""
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000474 optionlist = ''
475 if options and self.does_esmtp:
Eric S. Raymondc013f302001-02-09 05:40:38 +0000476 optionlist = ' ' + ' '.join(options)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000477 self.putcmd("rcpt", "TO:%s%s" % (quoteaddr(recip), optionlist))
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000478 return self.getreply()
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000479
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000480 def data(self, msg):
Tim Peters495ad3c2001-01-15 01:36:40 +0000481 """SMTP 'DATA' command -- sends message data to server.
Guido van Rossum296e1431999-04-07 15:03:39 +0000482
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000483 Automatically quotes lines beginning with a period per rfc821.
Guido van Rossum296e1431999-04-07 15:03:39 +0000484 Raises SMTPDataError if there is an unexpected reply to the
485 DATA command; the return value from this method is the final
R. David Murray7dff9e02010-11-08 17:15:13 +0000486 response code received when the all data is sent. If msg
487 is a string, lone '\r' and '\n' characters are converted to
488 '\r\n' characters. If msg is bytes, it is transmitted as is.
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000489 """
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000490 self.putcmd("data")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000491 (code, repl) = self.getreply()
492 if self.debuglevel > 0:
493 print("data:", (code, repl), file=stderr)
Fred Drake8152d322000-12-12 23:20:45 +0000494 if code != 354:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000495 raise SMTPDataError(code, repl)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000496 else:
R. David Murray7dff9e02010-11-08 17:15:13 +0000497 if isinstance(msg, str):
498 msg = _fix_eols(msg).encode('ascii')
499 q = _quote_periods(msg)
500 if q[-2:] != bCRLF:
501 q = q + bCRLF
502 q = q + b"." + bCRLF
Guido van Rossum20c92281999-04-21 16:52:20 +0000503 self.send(q)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000504 (code, msg) = self.getreply()
505 if self.debuglevel > 0:
506 print("data:", (code, msg), file=stderr)
507 return (code, msg)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000508
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000509 def verify(self, address):
Barry Warsawa7d9bdf1998-12-22 03:24:27 +0000510 """SMTP 'verify' command -- checks for address validity."""
R David Murray46346762011-07-18 21:38:54 -0400511 self.putcmd("vrfy", _addr_only(address))
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000512 return self.getreply()
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000513 # a.k.a.
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000514 vrfy = verify
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000515
516 def expn(self, address):
Christian Heimes90c3d9b2008-02-23 13:18:03 +0000517 """SMTP 'expn' command -- expands a mailing list."""
R David Murray46346762011-07-18 21:38:54 -0400518 self.putcmd("expn", _addr_only(address))
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000519 return self.getreply()
520
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000521 # some useful methods
Guido van Rossumae010462001-09-11 15:57:46 +0000522
Christian Heimes679db4a2008-01-18 09:56:22 +0000523 def ehlo_or_helo_if_needed(self):
524 """Call self.ehlo() and/or self.helo() if needed.
525
526 If there has been no previous EHLO or HELO command this session, this
527 method tries ESMTP EHLO first.
528
529 This method may raise the following exceptions:
530
531 SMTPHeloError The server didn't reply properly to
532 the helo greeting.
533 """
534 if self.helo_resp is None and self.ehlo_resp is None:
535 if not (200 <= self.ehlo()[0] <= 299):
536 (code, resp) = self.helo()
537 if not (200 <= code <= 299):
538 raise SMTPHeloError(code, resp)
539
Guido van Rossumae010462001-09-11 15:57:46 +0000540 def login(self, user, password):
541 """Log in on an SMTP server that requires authentication.
542
543 The arguments are:
544 - user: The user name to authenticate with.
545 - password: The password for the authentication.
546
547 If there has been no previous EHLO or HELO command this session, this
548 method tries ESMTP EHLO first.
549
550 This method will return normally if the authentication was successful.
551
552 This method may raise the following exceptions:
553
554 SMTPHeloError The server didn't reply properly to
555 the helo greeting.
556 SMTPAuthenticationError The server didn't accept the username/
557 password combination.
Fred Drake2f8f4d32001-10-13 18:35:32 +0000558 SMTPException No suitable authentication method was
Guido van Rossumae010462001-09-11 15:57:46 +0000559 found.
560 """
561
562 def encode_cram_md5(challenge, user, password):
Georg Brandl706824f2009-06-04 09:42:55 +0000563 challenge = base64.decodebytes(challenge)
R. David Murrayfb123912009-05-28 18:19:00 +0000564 response = user + " " + hmac.HMAC(password.encode('ascii'),
565 challenge).hexdigest()
566 return encode_base64(response.encode('ascii'), eol='')
Guido van Rossumae010462001-09-11 15:57:46 +0000567
568 def encode_plain(user, password):
R. David Murraycaa27b72009-05-23 18:49:56 +0000569 s = "\0%s\0%s" % (user, password)
570 return encode_base64(s.encode('ascii'), eol='')
Tim Peters469cdad2002-08-08 20:19:19 +0000571
Guido van Rossumae010462001-09-11 15:57:46 +0000572 AUTH_PLAIN = "PLAIN"
573 AUTH_CRAM_MD5 = "CRAM-MD5"
Piers Lauder385a77a2002-07-27 00:38:30 +0000574 AUTH_LOGIN = "LOGIN"
Guido van Rossumae010462001-09-11 15:57:46 +0000575
Christian Heimes679db4a2008-01-18 09:56:22 +0000576 self.ehlo_or_helo_if_needed()
Guido van Rossumae010462001-09-11 15:57:46 +0000577
578 if not self.has_extn("auth"):
579 raise SMTPException("SMTP AUTH extension not supported by server.")
580
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000581 # Authentication methods the server claims to support
582 advertised_authlist = self.esmtp_features["auth"].split()
Guido van Rossumae010462001-09-11 15:57:46 +0000583
584 # List of authentication methods we support: from preferred to
585 # less preferred methods. Except for the purpose of testing the weaker
586 # ones, we prefer stronger methods like CRAM-MD5:
Piers Lauder385a77a2002-07-27 00:38:30 +0000587 preferred_auths = [AUTH_CRAM_MD5, AUTH_PLAIN, AUTH_LOGIN]
Guido van Rossumae010462001-09-11 15:57:46 +0000588
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000589 # We try the authentication methods the server advertises, but only the
590 # ones *we* support. And in our preferred order.
591 authlist = [auth for auth in preferred_auths if auth in advertised_authlist]
592 if not authlist:
Fred Drake2f8f4d32001-10-13 18:35:32 +0000593 raise SMTPException("No suitable authentication method found.")
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000594
595 # Some servers advertise authentication methods they don't really
596 # support, so if authentication fails, we continue until we've tried
597 # all methods.
598 for authmethod in authlist:
599 if authmethod == AUTH_CRAM_MD5:
600 (code, resp) = self.docmd("AUTH", AUTH_CRAM_MD5)
601 if code == 334:
602 (code, resp) = self.docmd(encode_cram_md5(resp, user, password))
603 elif authmethod == AUTH_PLAIN:
604 (code, resp) = self.docmd("AUTH",
605 AUTH_PLAIN + " " + encode_plain(user, password))
606 elif authmethod == AUTH_LOGIN:
607 (code, resp) = self.docmd("AUTH",
608 "%s %s" % (AUTH_LOGIN, encode_base64(user.encode('ascii'), eol='')))
609 if code == 334:
610 (code, resp) = self.docmd(encode_base64(password.encode('ascii'), eol=''))
611
Guido van Rossumae010462001-09-11 15:57:46 +0000612 # 235 == 'Authentication successful'
613 # 503 == 'Error: already authenticated'
Gerhard Häring1c5471f2010-08-05 14:08:44 +0000614 if code in (235, 503):
615 return (code, resp)
616
617 # We could not login sucessfully. Return result of last attempt.
618 raise SMTPAuthenticationError(code, resp)
Guido van Rossumae010462001-09-11 15:57:46 +0000619
Antoine Pitroue0650202011-05-18 18:03:09 +0200620 def starttls(self, keyfile=None, certfile=None, context=None):
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000621 """Puts the connection to the SMTP server into TLS mode.
Tim Petersb64bec32001-09-18 02:26:39 +0000622
Christian Heimes679db4a2008-01-18 09:56:22 +0000623 If there has been no previous EHLO or HELO command this session, this
624 method tries ESMTP EHLO first.
625
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000626 If the server supports TLS, this will encrypt the rest of the SMTP
627 session. If you provide the keyfile and certfile parameters,
628 the identity of the SMTP server and client can be checked. This,
629 however, depends on whether the socket module really checks the
630 certificates.
Christian Heimes679db4a2008-01-18 09:56:22 +0000631
632 This method may raise the following exceptions:
633
634 SMTPHeloError The server didn't reply properly to
635 the helo greeting.
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000636 """
Christian Heimes679db4a2008-01-18 09:56:22 +0000637 self.ehlo_or_helo_if_needed()
638 if not self.has_extn("starttls"):
639 raise SMTPException("STARTTLS extension not supported by server.")
Tim Petersb64bec32001-09-18 02:26:39 +0000640 (resp, reply) = self.docmd("STARTTLS")
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000641 if resp == 220:
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000642 if not _have_ssl:
643 raise RuntimeError("No SSL support included in this Python")
Antoine Pitroue0650202011-05-18 18:03:09 +0200644 if context is not None and keyfile is not None:
645 raise ValueError("context and keyfile arguments are mutually "
646 "exclusive")
647 if context is not None and certfile is not None:
648 raise ValueError("context and certfile arguments are mutually "
649 "exclusive")
650 if context is not None:
651 self.sock = context.wrap_socket(self.sock)
652 else:
653 self.sock = ssl.wrap_socket(self.sock, keyfile, certfile)
Antoine Pitrouf068ab82011-06-06 19:17:09 +0200654 self.file = None
Christian Heimes679db4a2008-01-18 09:56:22 +0000655 # RFC 3207:
656 # The client MUST discard any knowledge obtained from
657 # the server, such as the list of SMTP service extensions,
658 # which was not obtained from the TLS negotiation itself.
659 self.helo_resp = None
660 self.ehlo_resp = None
661 self.esmtp_features = {}
662 self.does_esmtp = 0
Guido van Rossumf7fcf5e2001-09-14 16:08:44 +0000663 return (resp, reply)
Tim Petersb64bec32001-09-18 02:26:39 +0000664
Jeremy Hylton31bb8ce1998-08-13 19:57:46 +0000665 def sendmail(self, from_addr, to_addrs, msg, mail_options=[],
Tim Peters495ad3c2001-01-15 01:36:40 +0000666 rcpt_options=[]):
667 """This command performs an entire mail transaction.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000668
Tim Peters495ad3c2001-01-15 01:36:40 +0000669 The arguments are:
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000670 - from_addr : The address sending this mail.
671 - to_addrs : A list of addresses to send this mail to. A bare
672 string will be treated as a list with 1 address.
Tim Peters495ad3c2001-01-15 01:36:40 +0000673 - msg : The message to send.
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000674 - mail_options : List of ESMTP options (such as 8bitmime) for the
675 mail command.
676 - rcpt_options : List of ESMTP options (such as DSN commands) for
677 all the rcpt commands.
678
R. David Murray7dff9e02010-11-08 17:15:13 +0000679 msg may be a string containing characters in the ASCII range, or a byte
680 string. A string is encoded to bytes using the ascii codec, and lone
R David Murrayac4e5ab2011-07-02 21:03:19 -0400681 \\r and \\n characters are converted to \\r\\n characters.
R. David Murray7dff9e02010-11-08 17:15:13 +0000682
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000683 If there has been no previous EHLO or HELO command this session, this
684 method tries ESMTP EHLO first. If the server does ESMTP, message size
685 and each of the specified options will be passed to it. If EHLO
686 fails, HELO will be tried and ESMTP options suppressed.
687
688 This method will return normally if the mail is accepted for at least
Barry Warsawd25c1b71999-11-28 17:11:06 +0000689 one recipient. It returns a dictionary, with one entry for each
690 recipient that was refused. Each entry contains a tuple of the SMTP
691 error code and the accompanying error message sent by the server.
Guido van Rossum296e1431999-04-07 15:03:39 +0000692
693 This method may raise the following exceptions:
694
695 SMTPHeloError The server didn't reply properly to
Tim Peters495ad3c2001-01-15 01:36:40 +0000696 the helo greeting.
Barry Warsawd25c1b71999-11-28 17:11:06 +0000697 SMTPRecipientsRefused The server rejected ALL recipients
Guido van Rossum296e1431999-04-07 15:03:39 +0000698 (no mail was sent).
699 SMTPSenderRefused The server didn't accept the from_addr.
700 SMTPDataError The server replied with an unexpected
701 error code (other than a refusal of
702 a recipient).
703
704 Note: the connection will be open even after an exception is raised.
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000705
Guido van Rossum95e6f701998-06-25 02:15:50 +0000706 Example:
Tim Peters495ad3c2001-01-15 01:36:40 +0000707
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000708 >>> import smtplib
709 >>> s=smtplib.SMTP("localhost")
Guido van Rossumfc40a831998-01-29 17:26:45 +0000710 >>> tolist=["one@one.org","two@two.org","three@three.org","four@four.org"]
Martin v. Löwis301b1cd2002-07-28 16:52:01 +0000711 >>> msg = '''\\
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000712 ... From: Me@my.org
713 ... Subject: testin'...
714 ...
715 ... This is a test '''
716 >>> s.sendmail("me@my.org",tolist,msg)
717 { "three@three.org" : ( 550 ,"User unknown" ) }
718 >>> s.quit()
Tim Peters495ad3c2001-01-15 01:36:40 +0000719
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000720 In the above example, the message was accepted for delivery to three
721 of the four addresses, and one was rejected, with the error code
Barry Warsawd25c1b71999-11-28 17:11:06 +0000722 550. If all addresses are accepted, then the method will return an
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000723 empty dictionary.
724
725 """
Christian Heimes679db4a2008-01-18 09:56:22 +0000726 self.ehlo_or_helo_if_needed()
Guido van Rossum95e6f701998-06-25 02:15:50 +0000727 esmtp_opts = []
R. David Murray7dff9e02010-11-08 17:15:13 +0000728 if isinstance(msg, str):
729 msg = _fix_eols(msg).encode('ascii')
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000730 if self.does_esmtp:
731 # Hmmm? what's this? -ddm
732 # self.esmtp_features['7bit']=""
733 if self.has_extn('size'):
Walter Dörwald70a6b492004-02-12 17:35:32 +0000734 esmtp_opts.append("size=%d" % len(msg))
Guido van Rossumfcfb6321998-08-04 15:29:54 +0000735 for option in mail_options:
Guido van Rossum95e6f701998-06-25 02:15:50 +0000736 esmtp_opts.append(option)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000737 (code, resp) = self.mail(from_addr, esmtp_opts)
Fred Drake8152d322000-12-12 23:20:45 +0000738 if code != 250:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000739 self.rset()
Guido van Rossum296e1431999-04-07 15:03:39 +0000740 raise SMTPSenderRefused(code, resp, from_addr)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000741 senderrs = {}
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000742 if isinstance(to_addrs, str):
Jeremy Hylton31bb8ce1998-08-13 19:57:46 +0000743 to_addrs = [to_addrs]
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000744 for each in to_addrs:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000745 (code, resp) = self.rcpt(each, rcpt_options)
Fred Drake8152d322000-12-12 23:20:45 +0000746 if (code != 250) and (code != 251):
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000747 senderrs[each] = (code, resp)
748 if len(senderrs) == len(to_addrs):
Guido van Rossum95e6f701998-06-25 02:15:50 +0000749 # the server refused all our recipients
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000750 self.rset()
Guido van Rossum296e1431999-04-07 15:03:39 +0000751 raise SMTPRecipientsRefused(senderrs)
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000752 (code, resp) = self.data(msg)
Fred Drake8152d322000-12-12 23:20:45 +0000753 if code != 250:
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000754 self.rset()
Guido van Rossum296e1431999-04-07 15:03:39 +0000755 raise SMTPDataError(code, resp)
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000756 #if we got here then somebody got our mail
Tim Peters495ad3c2001-01-15 01:36:40 +0000757 return senderrs
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000758
R. David Murray7dff9e02010-11-08 17:15:13 +0000759 def send_message(self, msg, from_addr=None, to_addrs=None,
760 mail_options=[], rcpt_options={}):
761 """Converts message to a bytestring and passes it to sendmail.
762
763 The arguments are as for sendmail, except that msg is an
R David Murrayac4e5ab2011-07-02 21:03:19 -0400764 email.message.Message object. If from_addr is None or to_addrs is
765 None, these arguments are taken from the headers of the Message as
766 described in RFC 2822 (a ValueError is raised if there is more than
767 one set of 'Resent-' headers). Regardless of the values of from_addr and
768 to_addr, any Bcc field (or Resent-Bcc field, when the Message is a
769 resent) of the Message object won't be transmitted. The Message
770 object is then serialized using email.generator.BytesGenerator and
771 sendmail is called to transmit the message.
772
R. David Murray7dff9e02010-11-08 17:15:13 +0000773 """
R David Murrayac4e5ab2011-07-02 21:03:19 -0400774 # 'Resent-Date' is a mandatory field if the Message is resent (RFC 2822
775 # Section 3.6.6). In such a case, we use the 'Resent-*' fields. However,
776 # if there is more than one 'Resent-' block there's no way to
777 # unambiguously determine which one is the most recent in all cases,
778 # so rather than guess we raise a ValueError in that case.
779 #
780 # TODO implement heuristics to guess the correct Resent-* block with an
781 # option allowing the user to enable the heuristics. (It should be
782 # possible to guess correctly almost all of the time.)
783 resent =msg.get_all('Resent-Date')
784 if resent is None:
785 header_prefix = ''
786 elif len(resent) == 1:
787 header_prefix = 'Resent-'
788 else:
789 raise ValueError("message has more than one 'Resent-' header block")
R. David Murray7dff9e02010-11-08 17:15:13 +0000790 if from_addr is None:
R David Murrayac4e5ab2011-07-02 21:03:19 -0400791 # Prefer the sender field per RFC 2822:3.6.2.
792 from_addr = (msg[header_prefix+'Sender']
793 if (header_prefix+'Sender') in msg
794 else msg[header_prefix+'From'])
R. David Murray7dff9e02010-11-08 17:15:13 +0000795 if to_addrs is None:
R David Murrayac4e5ab2011-07-02 21:03:19 -0400796 addr_fields = [f for f in (msg[header_prefix+'To'],
797 msg[header_prefix+'Bcc'],
798 msg[header_prefix+'Cc']) if f is not None]
R. David Murray7dff9e02010-11-08 17:15:13 +0000799 to_addrs = [a[1] for a in email.utils.getaddresses(addr_fields)]
R David Murrayac4e5ab2011-07-02 21:03:19 -0400800 # Make a local copy so we can delete the bcc headers.
801 msg_copy = copy.copy(msg)
802 del msg_copy['Bcc']
803 del msg_copy['Resent-Bcc']
R. David Murray7dff9e02010-11-08 17:15:13 +0000804 with io.BytesIO() as bytesmsg:
805 g = email.generator.BytesGenerator(bytesmsg)
R David Murrayac4e5ab2011-07-02 21:03:19 -0400806 g.flatten(msg_copy, linesep='\r\n')
R. David Murray7dff9e02010-11-08 17:15:13 +0000807 flatmsg = bytesmsg.getvalue()
808 return self.sendmail(from_addr, to_addrs, flatmsg, mail_options,
809 rcpt_options)
810
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000811 def close(self):
812 """Close the connection to the SMTP server."""
813 if self.file:
814 self.file.close()
815 self.file = None
816 if self.sock:
817 self.sock.close()
818 self.sock = None
819
Guido van Rossumbbe323e1998-01-29 17:24:40 +0000820 def quit(self):
Guido van Rossum95e6f701998-06-25 02:15:50 +0000821 """Terminate the SMTP session."""
Christian Heimesba4af492008-03-28 00:55:15 +0000822 res = self.docmd("quit")
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000823 self.close()
Christian Heimesba4af492008-03-28 00:55:15 +0000824 return res
Guido van Rossum95e6f701998-06-25 02:15:50 +0000825
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000826if _have_ssl:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000827
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000828 class SMTP_SSL(SMTP):
829 """ This is a subclass derived from SMTP that connects over an SSL encrypted
830 socket (to use this class you need a socket module that was compiled with SSL
831 support). If host is not specified, '' (the local host) is used. If port is
832 omitted, the standard SMTP-over-SSL port (465) is used. keyfile and certfile
833 are also optional - they can contain a PEM formatted private key and
Antoine Pitroue0650202011-05-18 18:03:09 +0200834 certificate chain file for the SSL connection. context also optional, can contain
835 a SSLContext, and is an alternative to keyfile and certfile; If it is specified both
836 keyfile and certfile must be None.
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000837 """
Antoine Pitrouc1d52062011-05-07 19:39:37 +0200838
839 default_port = SMTP_SSL_PORT
840
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000841 def __init__(self, host='', port=0, local_hostname=None,
Georg Brandlf78e02b2008-06-10 17:40:04 +0000842 keyfile=None, certfile=None,
Antoine Pitroue0650202011-05-18 18:03:09 +0200843 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, context=None):
844 if context is not None and keyfile is not None:
845 raise ValueError("context and keyfile arguments are mutually "
846 "exclusive")
847 if context is not None and certfile is not None:
848 raise ValueError("context and certfile arguments are mutually "
849 "exclusive")
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000850 self.keyfile = keyfile
851 self.certfile = certfile
Antoine Pitroue0650202011-05-18 18:03:09 +0200852 self.context = context
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000853 SMTP.__init__(self, host, port, local_hostname, timeout)
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000854
855 def _get_socket(self, host, port, timeout):
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000856 if self.debuglevel > 0:
857 print('connect:', (host, port), file=stderr)
R. David Murray87e20742009-05-23 01:30:26 +0000858 new_socket = socket.create_connection((host, port), timeout)
Antoine Pitroue0650202011-05-18 18:03:09 +0200859 if self.context is not None:
860 new_socket = self.context.wrap_socket(new_socket)
861 else:
862 new_socket = ssl.wrap_socket(new_socket, self.keyfile, self.certfile)
R. David Murray87e20742009-05-23 01:30:26 +0000863 return new_socket
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000864
865 __all__.append("SMTP_SSL")
Barry Warsaw4c4bec81998-12-22 03:02:20 +0000866
Guido van Rossumd8faa362007-04-27 19:54:29 +0000867#
868# LMTP extension
869#
870LMTP_PORT = 2003
871
872class LMTP(SMTP):
873 """LMTP - Local Mail Transfer Protocol
874
875 The LMTP protocol, which is very similar to ESMTP, is heavily based
876 on the standard SMTP client. It's common to use Unix sockets for LMTP,
877 so our connect() method must support that as well as a regular
878 host:port server. To specify a Unix socket, you must use an absolute
879 path as the host, starting with a '/'.
880
881 Authentication is supported, using the regular SMTP mechanism. When
882 using a Unix socket, LMTP generally don't support or require any
883 authentication, but your mileage might vary."""
884
885 ehlo_msg = "lhlo"
886
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000887 def __init__(self, host='', port=LMTP_PORT, local_hostname=None):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000888 """Initialize a new instance."""
889 SMTP.__init__(self, host, port, local_hostname)
890
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000891 def connect(self, host='localhost', port=0):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000892 """Connect to the LMTP daemon, on either a Unix or a TCP socket."""
893 if host[0] != '/':
894 return SMTP.connect(self, host, port)
895
896 # Handle Unix-domain sockets.
897 try:
898 self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Antoine Pitrouf068ab82011-06-06 19:17:09 +0200899 self.file = None
Guido van Rossumd8faa362007-04-27 19:54:29 +0000900 self.sock.connect(host)
901 except socket.error as msg:
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000902 if self.debuglevel > 0:
903 print('connect fail:', host, file=stderr)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000904 if self.sock:
905 self.sock.close()
906 self.sock = None
907 raise socket.error(msg)
908 (code, msg) = self.getreply()
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000909 if self.debuglevel > 0:
910 print('connect:', msg, file=stderr)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000911 return (code, msg)
912
913
Guido van Rossum95e6f701998-06-25 02:15:50 +0000914# Test the sendmail method, which tests most of the others.
915# Note: This always sends to localhost.
916if __name__ == '__main__':
Andrew M. Kuchling6be424f2001-08-13 14:41:39 +0000917 import sys
Guido van Rossum95e6f701998-06-25 02:15:50 +0000918
919 def prompt(prompt):
920 sys.stdout.write(prompt + ": ")
Eric S. Raymondc013f302001-02-09 05:40:38 +0000921 return sys.stdin.readline().strip()
Guido van Rossum95e6f701998-06-25 02:15:50 +0000922
923 fromaddr = prompt("From")
Giampaolo Rodolàbd258bd2011-02-22 15:56:20 +0000924 toaddrs = prompt("To").split(',')
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000925 print("Enter message, end with ^D:")
Guido van Rossum95e6f701998-06-25 02:15:50 +0000926 msg = ''
927 while 1:
928 line = sys.stdin.readline()
929 if not line:
930 break
931 msg = msg + line
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000932 print("Message length is %d" % len(msg))
Guido van Rossum95e6f701998-06-25 02:15:50 +0000933
934 server = SMTP('localhost')
935 server.set_debuglevel(1)
936 server.sendmail(fromaddr, toaddrs, msg)
937 server.quit()