Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python |
| 2 | |
Barry Warsaw | a1ae884 | 2000-07-09 21:24:31 +0000 | [diff] [blame] | 3 | '''SMTP/ESMTP client class. |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 4 | |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 5 | This should follow RFC 821 (SMTP), RFC 1869 (ESMTP), RFC 2554 (SMTP |
| 6 | Authentication) and RFC 2487 (Secure SMTP over TLS). |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 7 | |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 8 | Notes: |
| 9 | |
| 10 | Please remember, when doing ESMTP, that the names of the SMTP service |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 11 | extensions are NOT the same thing as the option keywords for the RCPT |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 12 | and MAIL commands! |
| 13 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 14 | Example: |
| 15 | |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 16 | >>> import smtplib |
| 17 | >>> s=smtplib.SMTP("localhost") |
| 18 | >>> print s.help() |
| 19 | 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 Warsaw | a1ae884 | 2000-07-09 21:24:31 +0000 | [diff] [blame] | 33 | ''' |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 34 | |
Guido van Rossum | 98d9fd3 | 2000-02-28 15:12:25 +0000 | [diff] [blame] | 35 | # 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 Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 40 | # RFC 2554 (authentication) support by Gerhard Haering <gerhard@bigfoot.de>. |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 41 | # |
Guido van Rossum | 98d9fd3 | 2000-02-28 15:12:25 +0000 | [diff] [blame] | 42 | # This was modified from the Python 1.5 library HTTP lib. |
| 43 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 44 | import socket |
Barry Warsaw | 0720177 | 1998-12-22 20:37:36 +0000 | [diff] [blame] | 45 | import re |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 46 | import rfc822 |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 47 | import base64 |
| 48 | import hmac |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 49 | from email.base64MIME import encode as encode_base64 |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 50 | |
Skip Montanaro | 0de6580 | 2001-02-15 22:15:14 +0000 | [diff] [blame] | 51 | __all__ = ["SMTPException","SMTPServerDisconnected","SMTPResponseException", |
| 52 | "SMTPSenderRefused","SMTPRecipientsRefused","SMTPDataError", |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 53 | "SMTPConnectError","SMTPHeloError","SMTPAuthenticationError", |
| 54 | "quoteaddr","quotedata","SMTP"] |
Skip Montanaro | 0de6580 | 2001-02-15 22:15:14 +0000 | [diff] [blame] | 55 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 56 | SMTP_PORT = 25 |
| 57 | CRLF="\r\n" |
| 58 | |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 59 | OLDSTYLE_AUTH = re.compile(r"auth=(.*)", re.I) |
| 60 | |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 61 | # Exception classes used by this module. |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 62 | class SMTPException(Exception): |
| 63 | """Base class for all exceptions raised by this module.""" |
| 64 | |
| 65 | class SMTPServerDisconnected(SMTPException): |
| 66 | """Not connected to any SMTP server. |
| 67 | |
| 68 | This exception is raised when the server unexpectedly disconnects, |
| 69 | or when an attempt is made to use the SMTP instance before |
| 70 | connecting it to a server. |
| 71 | """ |
| 72 | |
| 73 | class SMTPResponseException(SMTPException): |
| 74 | """Base class for all exceptions that include an SMTP error code. |
| 75 | |
| 76 | These exceptions are generated in some instances when the SMTP |
| 77 | server returns an error code. The error code is stored in the |
| 78 | `smtp_code' attribute of the error, and the `smtp_error' attribute |
| 79 | is set to the error message. |
| 80 | """ |
| 81 | |
| 82 | def __init__(self, code, msg): |
| 83 | self.smtp_code = code |
| 84 | self.smtp_error = msg |
| 85 | self.args = (code, msg) |
| 86 | |
| 87 | class SMTPSenderRefused(SMTPResponseException): |
| 88 | """Sender address refused. |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 89 | |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 90 | In addition to the attributes set by on all SMTPResponseException |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 91 | exceptions, this sets `sender' to the string that the SMTP refused. |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 92 | """ |
| 93 | |
| 94 | def __init__(self, code, msg, sender): |
| 95 | self.smtp_code = code |
| 96 | self.smtp_error = msg |
| 97 | self.sender = sender |
| 98 | self.args = (code, msg, sender) |
| 99 | |
Guido van Rossum | 20c9228 | 1999-04-21 16:52:20 +0000 | [diff] [blame] | 100 | class SMTPRecipientsRefused(SMTPException): |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 101 | """All recipient addresses refused. |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 102 | |
Thomas Wouters | 7e47402 | 2000-07-16 12:04:32 +0000 | [diff] [blame] | 103 | The errors for each recipient are accessible through the attribute |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 104 | 'recipients', which is a dictionary of exactly the same sort as |
| 105 | SMTP.sendmail() returns. |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 106 | """ |
| 107 | |
| 108 | def __init__(self, recipients): |
| 109 | self.recipients = recipients |
| 110 | self.args = ( recipients,) |
| 111 | |
| 112 | |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 113 | class SMTPDataError(SMTPResponseException): |
| 114 | """The SMTP server didn't accept the data.""" |
| 115 | |
| 116 | class SMTPConnectError(SMTPResponseException): |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 117 | """Error during connection establishment.""" |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 118 | |
| 119 | class SMTPHeloError(SMTPResponseException): |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 120 | """The server refused our HELO reply.""" |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 121 | |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 122 | class SMTPAuthenticationError(SMTPResponseException): |
| 123 | """Authentication error. |
| 124 | |
| 125 | Most probably the server didn't accept the username/password |
| 126 | combination provided. |
| 127 | """ |
Peter Schneider-Kamp | 7bc82bb | 2000-08-10 14:02:23 +0000 | [diff] [blame] | 128 | |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 129 | class SSLFakeSocket: |
| 130 | """A fake socket object that really wraps a SSLObject. |
Tim Peters | b64bec3 | 2001-09-18 02:26:39 +0000 | [diff] [blame] | 131 | |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 132 | It only supports what is needed in smtplib. |
| 133 | """ |
| 134 | def __init__(self, realsock, sslobj): |
| 135 | self.realsock = realsock |
| 136 | self.sslobj = sslobj |
| 137 | |
| 138 | def send(self, str): |
| 139 | self.sslobj.write(str) |
| 140 | return len(str) |
| 141 | |
Martin v. Löwis | 9ea6c19 | 2002-06-02 12:33:22 +0000 | [diff] [blame] | 142 | sendall = send |
| 143 | |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 144 | def close(self): |
| 145 | self.realsock.close() |
| 146 | |
| 147 | class SSLFakeFile: |
| 148 | """A fake file like object that really wraps a SSLObject. |
Tim Peters | b64bec3 | 2001-09-18 02:26:39 +0000 | [diff] [blame] | 149 | |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 150 | It only supports what is needed in smtplib. |
| 151 | """ |
| 152 | def __init__( self, sslobj): |
| 153 | self.sslobj = sslobj |
| 154 | |
| 155 | def readline(self): |
| 156 | str = "" |
| 157 | chr = None |
| 158 | while chr != "\n": |
| 159 | chr = self.sslobj.read(1) |
| 160 | str += chr |
| 161 | return str |
| 162 | |
| 163 | def close(self): |
| 164 | pass |
| 165 | |
Guido van Rossum | 69a79bc | 1998-07-13 15:18:49 +0000 | [diff] [blame] | 166 | def quoteaddr(addr): |
| 167 | """Quote a subset of the email addresses defined by RFC 821. |
| 168 | |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 169 | Should be able to handle anything rfc822.parseaddr can handle. |
| 170 | """ |
Raymond Hettinger | 342456d | 2002-09-05 01:14:07 +0000 | [diff] [blame] | 171 | m = (None, None) |
Guido van Rossum | 69a79bc | 1998-07-13 15:18:49 +0000 | [diff] [blame] | 172 | try: |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 173 | m=rfc822.parseaddr(addr)[1] |
| 174 | except AttributeError: |
| 175 | pass |
Raymond Hettinger | 342456d | 2002-09-05 01:14:07 +0000 | [diff] [blame] | 176 | if m == (None, None): # Indicates parse failure or AttributeError |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 177 | #something weird here.. punt -ddm |
Raymond Hettinger | 342456d | 2002-09-05 01:14:07 +0000 | [diff] [blame] | 178 | return "<%s>" % addr |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 179 | else: |
| 180 | return "<%s>" % m |
Guido van Rossum | 69a79bc | 1998-07-13 15:18:49 +0000 | [diff] [blame] | 181 | |
| 182 | def quotedata(data): |
| 183 | """Quote data for email. |
| 184 | |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 185 | Double leading '.', and change Unix newline '\\n', or Mac '\\r' into |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 186 | Internet CRLF end-of-line. |
| 187 | """ |
Guido van Rossum | 69a79bc | 1998-07-13 15:18:49 +0000 | [diff] [blame] | 188 | return re.sub(r'(?m)^\.', '..', |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 189 | re.sub(r'(?:\r\n|\n|\r(?!\n))', CRLF, data)) |
Guido van Rossum | 69a79bc | 1998-07-13 15:18:49 +0000 | [diff] [blame] | 190 | |
Peter Schneider-Kamp | 7bc82bb | 2000-08-10 14:02:23 +0000 | [diff] [blame] | 191 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 192 | class SMTP: |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 193 | """This class manages a connection to an SMTP or ESMTP server. |
| 194 | SMTP Objects: |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 195 | SMTP objects have the following attributes: |
| 196 | helo_resp |
| 197 | This is the message given by the server in response to the |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 198 | most recent HELO command. |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 199 | |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 200 | ehlo_resp |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 201 | This is the message given by the server in response to the |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 202 | most recent EHLO command. This is usually multiline. |
| 203 | |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 204 | does_esmtp |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 205 | This is a True value _after you do an EHLO command_, if the |
| 206 | server supports ESMTP. |
| 207 | |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 208 | esmtp_features |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 209 | This is a dictionary, which, if the server supports ESMTP, |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 210 | will _after you do an EHLO command_, contain the names of the |
| 211 | SMTP service extensions this server supports, and their |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 212 | parameters (if any). |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 213 | |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 214 | Note, all extension names are mapped to lower case in the |
| 215 | dictionary. |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 216 | |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 217 | See each method's docstrings for details. In general, there is a |
| 218 | method of the same name to perform each SMTP command. There is also a |
| 219 | method called 'sendmail' that will do an entire mail transaction. |
| 220 | """ |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 221 | debuglevel = 0 |
| 222 | file = None |
| 223 | helo_resp = None |
| 224 | ehlo_resp = None |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 225 | does_esmtp = 0 |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 226 | |
Neil Schemenauer | 6730f26 | 2002-03-24 15:30:40 +0000 | [diff] [blame] | 227 | def __init__(self, host = '', port = 0, local_hostname = None): |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 228 | """Initialize a new instance. |
| 229 | |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 230 | If specified, `host' is the name of the remote host to which to |
| 231 | connect. If specified, `port' specifies the port to which to connect. |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 232 | By default, smtplib.SMTP_PORT is used. An SMTPConnectError is raised |
Neil Schemenauer | 6730f26 | 2002-03-24 15:30:40 +0000 | [diff] [blame] | 233 | if the specified `host' doesn't respond correctly. If specified, |
Tim Peters | 863ac44 | 2002-04-16 01:38:40 +0000 | [diff] [blame] | 234 | `local_hostname` is used as the FQDN of the local host. By default, |
| 235 | the local hostname is found using socket.getfqdn(). |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 236 | |
| 237 | """ |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 238 | self.esmtp_features = {} |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 239 | if host: |
| 240 | (code, msg) = self.connect(host, port) |
| 241 | if code != 220: |
| 242 | raise SMTPConnectError(code, msg) |
Raymond Hettinger | f13eb55 | 2002-06-02 00:40:05 +0000 | [diff] [blame] | 243 | if local_hostname is not None: |
Barry Warsaw | 13e34f7 | 2002-03-26 20:27:35 +0000 | [diff] [blame] | 244 | self.local_hostname = local_hostname |
Neil Schemenauer | 6730f26 | 2002-03-24 15:30:40 +0000 | [diff] [blame] | 245 | else: |
Barry Warsaw | 13e34f7 | 2002-03-26 20:27:35 +0000 | [diff] [blame] | 246 | # RFC 2821 says we should use the fqdn in the EHLO/HELO verb, and |
| 247 | # if that can't be calculated, that we should use a domain literal |
| 248 | # instead (essentially an encoded IP address like [A.B.C.D]). |
| 249 | fqdn = socket.getfqdn() |
| 250 | if '.' in fqdn: |
| 251 | self.local_hostname = fqdn |
| 252 | else: |
| 253 | # We can't find an fqdn hostname, so use a domain literal |
| 254 | addr = socket.gethostbyname(socket.gethostname()) |
| 255 | self.local_hostname = '[%s]' % addr |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 256 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 257 | def set_debuglevel(self, debuglevel): |
| 258 | """Set the debug output level. |
| 259 | |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 260 | A non-false value results in debug messages for connection and for all |
| 261 | messages sent to and received from the server. |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 262 | |
| 263 | """ |
| 264 | self.debuglevel = debuglevel |
| 265 | |
| 266 | def connect(self, host='localhost', port = 0): |
| 267 | """Connect to a host on a given port. |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 268 | |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 269 | If the hostname ends with a colon (`:') followed by a number, and |
| 270 | there is no port specified, that suffix will be stripped off and the |
| 271 | number interpreted as the port number to use. |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 272 | |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 273 | Note: This method is automatically invoked by __init__, if a host is |
| 274 | specified during instantiation. |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 275 | |
| 276 | """ |
Martin v. Löwis | 4eb5940 | 2001-07-26 13:37:33 +0000 | [diff] [blame] | 277 | if not port and (host.find(':') == host.rfind(':')): |
Martin v. Löwis | a43c2f8 | 2001-07-24 20:34:08 +0000 | [diff] [blame] | 278 | i = host.rfind(':') |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 279 | if i >= 0: |
| 280 | host, port = host[:i], host[i+1:] |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 281 | try: port = int(port) |
Eric S. Raymond | 8d87603 | 2001-02-09 10:14:53 +0000 | [diff] [blame] | 282 | except ValueError: |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 283 | raise socket.error, "nonnumeric port" |
| 284 | if not port: port = SMTP_PORT |
Martin v. Löwis | 4eb5940 | 2001-07-26 13:37:33 +0000 | [diff] [blame] | 285 | if self.debuglevel > 0: print 'connect:', (host, port) |
Martin v. Löwis | 2ad2569 | 2001-07-31 08:40:21 +0000 | [diff] [blame] | 286 | msg = "getaddrinfo returns an empty list" |
Martin v. Löwis | 322c0d1 | 2001-10-07 08:53:32 +0000 | [diff] [blame] | 287 | self.sock = None |
Martin v. Löwis | 4eb5940 | 2001-07-26 13:37:33 +0000 | [diff] [blame] | 288 | for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM): |
| 289 | af, socktype, proto, canonname, sa = res |
| 290 | try: |
| 291 | self.sock = socket.socket(af, socktype, proto) |
| 292 | if self.debuglevel > 0: print 'connect:', (host, port) |
| 293 | self.sock.connect(sa) |
| 294 | except socket.error, msg: |
| 295 | if self.debuglevel > 0: print 'connect fail:', (host, port) |
Martin v. Löwis | 322c0d1 | 2001-10-07 08:53:32 +0000 | [diff] [blame] | 296 | if self.sock: |
| 297 | self.sock.close() |
Martin v. Löwis | 4eb5940 | 2001-07-26 13:37:33 +0000 | [diff] [blame] | 298 | self.sock = None |
| 299 | continue |
| 300 | break |
| 301 | if not self.sock: |
| 302 | raise socket.error, msg |
Martin v. Löwis | a43c2f8 | 2001-07-24 20:34:08 +0000 | [diff] [blame] | 303 | (code, msg) = self.getreply() |
| 304 | if self.debuglevel > 0: print "connect:", msg |
| 305 | return (code, msg) |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 306 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 307 | def send(self, str): |
| 308 | """Send `str' to the server.""" |
| 309 | if self.debuglevel > 0: print 'send:', `str` |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 310 | if self.sock: |
Guido van Rossum | 2880f6e | 1998-08-10 20:07:00 +0000 | [diff] [blame] | 311 | try: |
Martin v. Löwis | e12454f | 2002-02-16 23:06:19 +0000 | [diff] [blame] | 312 | self.sock.sendall(str) |
Guido van Rossum | 2880f6e | 1998-08-10 20:07:00 +0000 | [diff] [blame] | 313 | except socket.error: |
Barry Warsaw | 7675097 | 2001-12-14 20:34:20 +0000 | [diff] [blame] | 314 | self.close() |
Guido van Rossum | 40233ea | 1999-01-15 03:23:55 +0000 | [diff] [blame] | 315 | raise SMTPServerDisconnected('Server not connected') |
Guido van Rossum | fc40a83 | 1998-01-29 17:26:45 +0000 | [diff] [blame] | 316 | else: |
Guido van Rossum | 40233ea | 1999-01-15 03:23:55 +0000 | [diff] [blame] | 317 | raise SMTPServerDisconnected('please run connect() first') |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 318 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 319 | def putcmd(self, cmd, args=""): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 320 | """Send a command to the server.""" |
Guido van Rossum | db23d3d | 1999-06-09 15:13:10 +0000 | [diff] [blame] | 321 | if args == "": |
| 322 | str = '%s%s' % (cmd, CRLF) |
| 323 | else: |
| 324 | str = '%s %s%s' % (cmd, args, CRLF) |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 325 | self.send(str) |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 326 | |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 327 | def getreply(self): |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 328 | """Get a reply from the server. |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 329 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 330 | Returns a tuple consisting of: |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 331 | |
| 332 | - server response code (e.g. '250', or such, if all goes well) |
| 333 | Note: returns -1 if it can't read response code. |
| 334 | |
| 335 | - server response string corresponding to response code (multiline |
| 336 | responses are converted to a single, multiline string). |
Guido van Rossum | f123f84 | 1999-03-29 20:33:21 +0000 | [diff] [blame] | 337 | |
| 338 | Raises SMTPServerDisconnected if end-of-file is reached. |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 339 | """ |
| 340 | resp=[] |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 341 | if self.file is None: |
| 342 | self.file = self.sock.makefile('rb') |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 343 | while 1: |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 344 | line = self.file.readline() |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 345 | if line == '': |
| 346 | self.close() |
| 347 | raise SMTPServerDisconnected("Connection unexpectedly closed") |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 348 | if self.debuglevel > 0: print 'reply:', `line` |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 349 | resp.append(line[4:].strip()) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 350 | code=line[:3] |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 351 | # Check that the error code is syntactically correct. |
| 352 | # Don't attempt to read a continuation line if it is broken. |
| 353 | try: |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 354 | errcode = int(code) |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 355 | except ValueError: |
| 356 | errcode = -1 |
| 357 | break |
Guido van Rossum | f123f84 | 1999-03-29 20:33:21 +0000 | [diff] [blame] | 358 | # Check if multiline response. |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 359 | if line[3:4]!="-": |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 360 | break |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 361 | |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 362 | errmsg = "\n".join(resp) |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 363 | if self.debuglevel > 0: |
Guido van Rossum | fc40a83 | 1998-01-29 17:26:45 +0000 | [diff] [blame] | 364 | print 'reply: retcode (%s); Msg: %s' % (errcode,errmsg) |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 365 | return errcode, errmsg |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 366 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 367 | def docmd(self, cmd, args=""): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 368 | """Send a command, and return its response code.""" |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 369 | self.putcmd(cmd,args) |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 370 | return self.getreply() |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 371 | |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 372 | # std smtp commands |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 373 | def helo(self, name=''): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 374 | """SMTP 'helo' command. |
| 375 | Hostname to send for this command defaults to the FQDN of the local |
| 376 | host. |
| 377 | """ |
Neil Schemenauer | 6730f26 | 2002-03-24 15:30:40 +0000 | [diff] [blame] | 378 | self.putcmd("helo", name or self.local_hostname) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 379 | (code,msg)=self.getreply() |
| 380 | self.helo_resp=msg |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 381 | return (code,msg) |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 382 | |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 383 | def ehlo(self, name=''): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 384 | """ SMTP 'ehlo' command. |
| 385 | Hostname to send for this command defaults to the FQDN of the local |
| 386 | host. |
| 387 | """ |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 388 | self.esmtp_features = {} |
Neil Schemenauer | 6730f26 | 2002-03-24 15:30:40 +0000 | [diff] [blame] | 389 | self.putcmd("ehlo", name or self.local_hostname) |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 390 | (code,msg)=self.getreply() |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 391 | # According to RFC1869 some (badly written) |
| 392 | # MTA's will disconnect on an ehlo. Toss an exception if |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 393 | # that happens -ddm |
| 394 | if code == -1 and len(msg) == 0: |
Barry Warsaw | 7675097 | 2001-12-14 20:34:20 +0000 | [diff] [blame] | 395 | self.close() |
Guido van Rossum | 40233ea | 1999-01-15 03:23:55 +0000 | [diff] [blame] | 396 | raise SMTPServerDisconnected("Server not connected") |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 397 | self.ehlo_resp=msg |
Fred Drake | 8152d32 | 2000-12-12 23:20:45 +0000 | [diff] [blame] | 398 | if code != 250: |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 399 | return (code,msg) |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 400 | self.does_esmtp=1 |
Thomas Wouters | 7e47402 | 2000-07-16 12:04:32 +0000 | [diff] [blame] | 401 | #parse the ehlo response -ddm |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 402 | resp=self.ehlo_resp.split('\n') |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 403 | del resp[0] |
Guido van Rossum | 2880f6e | 1998-08-10 20:07:00 +0000 | [diff] [blame] | 404 | for each in resp: |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 405 | # To be able to communicate with as many SMTP servers as possible, |
| 406 | # we have to take the old-style auth advertisement into account, |
| 407 | # because: |
| 408 | # 1) Else our SMTP feature parser gets confused. |
| 409 | # 2) There are some servers that only advertise the auth methods we |
| 410 | # support using the old style. |
| 411 | auth_match = OLDSTYLE_AUTH.match(each) |
| 412 | if auth_match: |
| 413 | # This doesn't remove duplicates, but that's no problem |
| 414 | self.esmtp_features["auth"] = self.esmtp_features.get("auth", "") \ |
| 415 | + " " + auth_match.groups(0)[0] |
| 416 | continue |
| 417 | |
Barry Warsaw | be22ae6 | 2002-04-15 20:03:30 +0000 | [diff] [blame] | 418 | # RFC 1869 requires a space between ehlo keyword and parameters. |
| 419 | # It's actually stricter, in that only spaces are allowed between |
| 420 | # parameters, but were not going to check for that here. Note |
| 421 | # that the space isn't present if there are no parameters. |
| 422 | m=re.match(r'(?P<feature>[A-Za-z0-9][A-Za-z0-9\-]*) ?',each) |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 423 | if m: |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 424 | feature=m.group("feature").lower() |
| 425 | params=m.string[m.end("feature"):].strip() |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 426 | if feature == "auth": |
| 427 | self.esmtp_features[feature] = self.esmtp_features.get(feature, "") \ |
| 428 | + " " + params |
| 429 | else: |
| 430 | self.esmtp_features[feature]=params |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 431 | return (code,msg) |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 432 | |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 433 | def has_extn(self, opt): |
| 434 | """Does the server support a given SMTP service extension?""" |
Raymond Hettinger | 54f0222 | 2002-06-01 14:18:47 +0000 | [diff] [blame] | 435 | return opt.lower() in self.esmtp_features |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 436 | |
Guido van Rossum | 18586f4 | 1998-04-03 17:03:13 +0000 | [diff] [blame] | 437 | def help(self, args=''): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 438 | """SMTP 'help' command. |
| 439 | Returns help text from server.""" |
Guido van Rossum | 18586f4 | 1998-04-03 17:03:13 +0000 | [diff] [blame] | 440 | self.putcmd("help", args) |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 441 | return self.getreply() |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 442 | |
| 443 | def rset(self): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 444 | """SMTP 'rset' command -- resets session.""" |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 445 | return self.docmd("rset") |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 446 | |
| 447 | def noop(self): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 448 | """SMTP 'noop' command -- doesn't do anything :>""" |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 449 | return self.docmd("noop") |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 450 | |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 451 | def mail(self,sender,options=[]): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 452 | """SMTP 'mail' command -- begins mail xfer session.""" |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 453 | optionlist = '' |
| 454 | if options and self.does_esmtp: |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 455 | optionlist = ' ' + ' '.join(options) |
Guido van Rossum | db23d3d | 1999-06-09 15:13:10 +0000 | [diff] [blame] | 456 | self.putcmd("mail", "FROM:%s%s" % (quoteaddr(sender) ,optionlist)) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 457 | return self.getreply() |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 458 | |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 459 | def rcpt(self,recip,options=[]): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 460 | """SMTP 'rcpt' command -- indicates 1 recipient for this mail.""" |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 461 | optionlist = '' |
| 462 | if options and self.does_esmtp: |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 463 | optionlist = ' ' + ' '.join(options) |
Guido van Rossum | 348fd06 | 1999-01-14 04:18:46 +0000 | [diff] [blame] | 464 | self.putcmd("rcpt","TO:%s%s" % (quoteaddr(recip),optionlist)) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 465 | return self.getreply() |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 466 | |
| 467 | def data(self,msg): |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 468 | """SMTP 'DATA' command -- sends message data to server. |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 469 | |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 470 | Automatically quotes lines beginning with a period per rfc821. |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 471 | Raises SMTPDataError if there is an unexpected reply to the |
| 472 | DATA command; the return value from this method is the final |
| 473 | response code received when the all data is sent. |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 474 | """ |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 475 | self.putcmd("data") |
| 476 | (code,repl)=self.getreply() |
| 477 | if self.debuglevel >0 : print "data:", (code,repl) |
Fred Drake | 8152d32 | 2000-12-12 23:20:45 +0000 | [diff] [blame] | 478 | if code != 354: |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 479 | raise SMTPDataError(code,repl) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 480 | else: |
Guido van Rossum | 20c9228 | 1999-04-21 16:52:20 +0000 | [diff] [blame] | 481 | q = quotedata(msg) |
| 482 | if q[-2:] != CRLF: |
| 483 | q = q + CRLF |
| 484 | q = q + "." + CRLF |
| 485 | self.send(q) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 486 | (code,msg)=self.getreply() |
| 487 | if self.debuglevel >0 : print "data:", (code,msg) |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 488 | return (code,msg) |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 489 | |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 490 | def verify(self, address): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 491 | """SMTP 'verify' command -- checks for address validity.""" |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 492 | self.putcmd("vrfy", quoteaddr(address)) |
| 493 | return self.getreply() |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 494 | # a.k.a. |
| 495 | vrfy=verify |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 496 | |
| 497 | def expn(self, address): |
Barry Warsaw | a7d9bdf | 1998-12-22 03:24:27 +0000 | [diff] [blame] | 498 | """SMTP 'verify' command -- checks for address validity.""" |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 499 | self.putcmd("expn", quoteaddr(address)) |
| 500 | return self.getreply() |
| 501 | |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 502 | # some useful methods |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 503 | |
| 504 | def login(self, user, password): |
| 505 | """Log in on an SMTP server that requires authentication. |
| 506 | |
| 507 | The arguments are: |
| 508 | - user: The user name to authenticate with. |
| 509 | - password: The password for the authentication. |
| 510 | |
| 511 | If there has been no previous EHLO or HELO command this session, this |
| 512 | method tries ESMTP EHLO first. |
| 513 | |
| 514 | This method will return normally if the authentication was successful. |
| 515 | |
| 516 | This method may raise the following exceptions: |
| 517 | |
| 518 | SMTPHeloError The server didn't reply properly to |
| 519 | the helo greeting. |
| 520 | SMTPAuthenticationError The server didn't accept the username/ |
| 521 | password combination. |
Fred Drake | 2f8f4d3 | 2001-10-13 18:35:32 +0000 | [diff] [blame] | 522 | SMTPException No suitable authentication method was |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 523 | found. |
| 524 | """ |
| 525 | |
| 526 | def encode_cram_md5(challenge, user, password): |
| 527 | challenge = base64.decodestring(challenge) |
| 528 | response = user + " " + hmac.HMAC(password, challenge).hexdigest() |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 529 | return encode_base64(response, eol="") |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 530 | |
| 531 | def encode_plain(user, password): |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 532 | return encode_base64("%s\0%s\0%s" % (user, user, password), eol="") |
Tim Peters | 469cdad | 2002-08-08 20:19:19 +0000 | [diff] [blame] | 533 | |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 534 | |
| 535 | AUTH_PLAIN = "PLAIN" |
| 536 | AUTH_CRAM_MD5 = "CRAM-MD5" |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 537 | AUTH_LOGIN = "LOGIN" |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 538 | |
| 539 | if self.helo_resp is None and self.ehlo_resp is None: |
| 540 | if not (200 <= self.ehlo()[0] <= 299): |
| 541 | (code, resp) = self.helo() |
| 542 | if not (200 <= code <= 299): |
| 543 | raise SMTPHeloError(code, resp) |
| 544 | |
| 545 | if not self.has_extn("auth"): |
| 546 | raise SMTPException("SMTP AUTH extension not supported by server.") |
| 547 | |
| 548 | # Authentication methods the server supports: |
| 549 | authlist = self.esmtp_features["auth"].split() |
| 550 | |
| 551 | # List of authentication methods we support: from preferred to |
| 552 | # less preferred methods. Except for the purpose of testing the weaker |
| 553 | # ones, we prefer stronger methods like CRAM-MD5: |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 554 | preferred_auths = [AUTH_CRAM_MD5, AUTH_PLAIN, AUTH_LOGIN] |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 555 | |
| 556 | # Determine the authentication method we'll use |
| 557 | authmethod = None |
| 558 | for method in preferred_auths: |
| 559 | if method in authlist: |
| 560 | authmethod = method |
| 561 | break |
Tim Peters | b64bec3 | 2001-09-18 02:26:39 +0000 | [diff] [blame] | 562 | |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 563 | if authmethod == AUTH_CRAM_MD5: |
| 564 | (code, resp) = self.docmd("AUTH", AUTH_CRAM_MD5) |
| 565 | if code == 503: |
| 566 | # 503 == 'Error: already authenticated' |
| 567 | return (code, resp) |
| 568 | (code, resp) = self.docmd(encode_cram_md5(resp, user, password)) |
| 569 | elif authmethod == AUTH_PLAIN: |
Tim Peters | b64bec3 | 2001-09-18 02:26:39 +0000 | [diff] [blame] | 570 | (code, resp) = self.docmd("AUTH", |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 571 | AUTH_PLAIN + " " + encode_plain(user, password)) |
Piers Lauder | 385a77a | 2002-07-27 00:38:30 +0000 | [diff] [blame] | 572 | elif authmethod == AUTH_LOGIN: |
| 573 | (code, resp) = self.docmd("AUTH", |
| 574 | "%s %s" % (AUTH_LOGIN, encode_base64(user, eol=""))) |
| 575 | if code != 334: |
| 576 | raise SMTPAuthenticationError(code, resp) |
Martin v. Löwis | 49c05d3 | 2002-10-06 17:55:08 +0000 | [diff] [blame^] | 577 | (code, resp) = self.docmd(encode_base64(password, eol="")) |
Raymond Hettinger | 7fdfc2d | 2002-05-31 17:49:10 +0000 | [diff] [blame] | 578 | elif authmethod is None: |
Fred Drake | 2f8f4d3 | 2001-10-13 18:35:32 +0000 | [diff] [blame] | 579 | raise SMTPException("No suitable authentication method found.") |
Guido van Rossum | ae01046 | 2001-09-11 15:57:46 +0000 | [diff] [blame] | 580 | if code not in [235, 503]: |
| 581 | # 235 == 'Authentication successful' |
| 582 | # 503 == 'Error: already authenticated' |
| 583 | raise SMTPAuthenticationError(code, resp) |
| 584 | return (code, resp) |
| 585 | |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 586 | def starttls(self, keyfile = None, certfile = None): |
| 587 | """Puts the connection to the SMTP server into TLS mode. |
Tim Peters | b64bec3 | 2001-09-18 02:26:39 +0000 | [diff] [blame] | 588 | |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 589 | If the server supports TLS, this will encrypt the rest of the SMTP |
| 590 | session. If you provide the keyfile and certfile parameters, |
| 591 | the identity of the SMTP server and client can be checked. This, |
| 592 | however, depends on whether the socket module really checks the |
| 593 | certificates. |
| 594 | """ |
Tim Peters | b64bec3 | 2001-09-18 02:26:39 +0000 | [diff] [blame] | 595 | (resp, reply) = self.docmd("STARTTLS") |
Guido van Rossum | f7fcf5e | 2001-09-14 16:08:44 +0000 | [diff] [blame] | 596 | if resp == 220: |
| 597 | sslobj = socket.ssl(self.sock, keyfile, certfile) |
| 598 | self.sock = SSLFakeSocket(self.sock, sslobj) |
| 599 | self.file = SSLFakeFile(sslobj) |
| 600 | return (resp, reply) |
Tim Peters | b64bec3 | 2001-09-18 02:26:39 +0000 | [diff] [blame] | 601 | |
Jeremy Hylton | 31bb8ce | 1998-08-13 19:57:46 +0000 | [diff] [blame] | 602 | def sendmail(self, from_addr, to_addrs, msg, mail_options=[], |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 603 | rcpt_options=[]): |
| 604 | """This command performs an entire mail transaction. |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 605 | |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 606 | The arguments are: |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 607 | - from_addr : The address sending this mail. |
| 608 | - to_addrs : A list of addresses to send this mail to. A bare |
| 609 | string will be treated as a list with 1 address. |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 610 | - msg : The message to send. |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 611 | - mail_options : List of ESMTP options (such as 8bitmime) for the |
| 612 | mail command. |
| 613 | - rcpt_options : List of ESMTP options (such as DSN commands) for |
| 614 | all the rcpt commands. |
| 615 | |
| 616 | If there has been no previous EHLO or HELO command this session, this |
| 617 | method tries ESMTP EHLO first. If the server does ESMTP, message size |
| 618 | and each of the specified options will be passed to it. If EHLO |
| 619 | fails, HELO will be tried and ESMTP options suppressed. |
| 620 | |
| 621 | This method will return normally if the mail is accepted for at least |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 622 | one recipient. It returns a dictionary, with one entry for each |
| 623 | recipient that was refused. Each entry contains a tuple of the SMTP |
| 624 | error code and the accompanying error message sent by the server. |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 625 | |
| 626 | This method may raise the following exceptions: |
| 627 | |
| 628 | SMTPHeloError The server didn't reply properly to |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 629 | the helo greeting. |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 630 | SMTPRecipientsRefused The server rejected ALL recipients |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 631 | (no mail was sent). |
| 632 | SMTPSenderRefused The server didn't accept the from_addr. |
| 633 | SMTPDataError The server replied with an unexpected |
| 634 | error code (other than a refusal of |
| 635 | a recipient). |
| 636 | |
| 637 | Note: the connection will be open even after an exception is raised. |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 638 | |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 639 | Example: |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 640 | |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 641 | >>> import smtplib |
| 642 | >>> s=smtplib.SMTP("localhost") |
Guido van Rossum | fc40a83 | 1998-01-29 17:26:45 +0000 | [diff] [blame] | 643 | >>> tolist=["one@one.org","two@two.org","three@three.org","four@four.org"] |
Martin v. Löwis | 301b1cd | 2002-07-28 16:52:01 +0000 | [diff] [blame] | 644 | >>> msg = '''\\ |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 645 | ... From: Me@my.org |
| 646 | ... Subject: testin'... |
| 647 | ... |
| 648 | ... This is a test ''' |
| 649 | >>> s.sendmail("me@my.org",tolist,msg) |
| 650 | { "three@three.org" : ( 550 ,"User unknown" ) } |
| 651 | >>> s.quit() |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 652 | |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 653 | In the above example, the message was accepted for delivery to three |
| 654 | of the four addresses, and one was rejected, with the error code |
Barry Warsaw | d25c1b7 | 1999-11-28 17:11:06 +0000 | [diff] [blame] | 655 | 550. If all addresses are accepted, then the method will return an |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 656 | empty dictionary. |
| 657 | |
| 658 | """ |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 659 | if self.helo_resp is None and self.ehlo_resp is None: |
| 660 | if not (200 <= self.ehlo()[0] <= 299): |
| 661 | (code,resp) = self.helo() |
| 662 | if not (200 <= code <= 299): |
| 663 | raise SMTPHeloError(code, resp) |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 664 | esmtp_opts = [] |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 665 | if self.does_esmtp: |
| 666 | # Hmmm? what's this? -ddm |
| 667 | # self.esmtp_features['7bit']="" |
| 668 | if self.has_extn('size'): |
| 669 | esmtp_opts.append("size=" + `len(msg)`) |
| 670 | for option in mail_options: |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 671 | esmtp_opts.append(option) |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 672 | |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 673 | (code,resp) = self.mail(from_addr, esmtp_opts) |
Fred Drake | 8152d32 | 2000-12-12 23:20:45 +0000 | [diff] [blame] | 674 | if code != 250: |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 675 | self.rset() |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 676 | raise SMTPSenderRefused(code, resp, from_addr) |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 677 | senderrs={} |
Walter Dörwald | 65230a2 | 2002-06-03 15:58:32 +0000 | [diff] [blame] | 678 | if isinstance(to_addrs, basestring): |
Jeremy Hylton | 31bb8ce | 1998-08-13 19:57:46 +0000 | [diff] [blame] | 679 | to_addrs = [to_addrs] |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 680 | for each in to_addrs: |
Guido van Rossum | fcfb632 | 1998-08-04 15:29:54 +0000 | [diff] [blame] | 681 | (code,resp)=self.rcpt(each, rcpt_options) |
Fred Drake | 8152d32 | 2000-12-12 23:20:45 +0000 | [diff] [blame] | 682 | if (code != 250) and (code != 251): |
Guido van Rossum | fc40a83 | 1998-01-29 17:26:45 +0000 | [diff] [blame] | 683 | senderrs[each]=(code,resp) |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 684 | if len(senderrs)==len(to_addrs): |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 685 | # the server refused all our recipients |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 686 | self.rset() |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 687 | raise SMTPRecipientsRefused(senderrs) |
Fred Drake | 8152d32 | 2000-12-12 23:20:45 +0000 | [diff] [blame] | 688 | (code,resp) = self.data(msg) |
| 689 | if code != 250: |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 690 | self.rset() |
Guido van Rossum | 296e143 | 1999-04-07 15:03:39 +0000 | [diff] [blame] | 691 | raise SMTPDataError(code, resp) |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 692 | #if we got here then somebody got our mail |
Tim Peters | 495ad3c | 2001-01-15 01:36:40 +0000 | [diff] [blame] | 693 | return senderrs |
Guido van Rossum | bbe323e | 1998-01-29 17:24:40 +0000 | [diff] [blame] | 694 | |
| 695 | |
| 696 | def close(self): |
| 697 | """Close the connection to the SMTP server.""" |
| 698 | if self.file: |
| 699 | self.file.close() |
| 700 | self.file = None |
| 701 | if self.sock: |
| 702 | self.sock.close() |
| 703 | self.sock = None |
| 704 | |
| 705 | |
| 706 | def quit(self): |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 707 | """Terminate the SMTP session.""" |
Guido van Rossum | 45e2fbc | 1998-03-26 21:13:24 +0000 | [diff] [blame] | 708 | self.docmd("quit") |
| 709 | self.close() |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 710 | |
Barry Warsaw | 4c4bec8 | 1998-12-22 03:02:20 +0000 | [diff] [blame] | 711 | |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 712 | # Test the sendmail method, which tests most of the others. |
| 713 | # Note: This always sends to localhost. |
| 714 | if __name__ == '__main__': |
Andrew M. Kuchling | 6be424f | 2001-08-13 14:41:39 +0000 | [diff] [blame] | 715 | import sys |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 716 | |
| 717 | def prompt(prompt): |
| 718 | sys.stdout.write(prompt + ": ") |
Eric S. Raymond | c013f30 | 2001-02-09 05:40:38 +0000 | [diff] [blame] | 719 | return sys.stdin.readline().strip() |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 720 | |
| 721 | fromaddr = prompt("From") |
Eric S. Raymond | 38151ed | 2001-02-09 07:40:17 +0000 | [diff] [blame] | 722 | toaddrs = prompt("To").split(',') |
Guido van Rossum | 95e6f70 | 1998-06-25 02:15:50 +0000 | [diff] [blame] | 723 | print "Enter message, end with ^D:" |
| 724 | msg = '' |
| 725 | while 1: |
| 726 | line = sys.stdin.readline() |
| 727 | if not line: |
| 728 | break |
| 729 | msg = msg + line |
| 730 | print "Message length is " + `len(msg)` |
| 731 | |
| 732 | server = SMTP('localhost') |
| 733 | server.set_debuglevel(1) |
| 734 | server.sendmail(fromaddr, toaddrs, msg) |
| 735 | server.quit() |