You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
3.5 KiB
108 lines
3.5 KiB
# µMail (MicroMail) for MicroPython
|
|
# Copyright (c) 2018 Shawwwn <shawwwn1@gmai.com>
|
|
# License: MIT
|
|
import usocket
|
|
|
|
DEFAULT_TIMEOUT = 10 # sec
|
|
LOCAL_DOMAIN = '127.0.0.1'
|
|
CMD_EHLO = 'EHLO'
|
|
CMD_STARTTLS = 'STARTTLS'
|
|
CMD_AUTH = 'AUTH'
|
|
CMD_MAIL = 'MAIL'
|
|
AUTH_PLAIN = 'PLAIN'
|
|
AUTH_LOGIN = 'LOGIN'
|
|
|
|
class SMTP:
|
|
def cmd(self, cmd_str):
|
|
sock = self._sock;
|
|
sock.write('%s\r\n' % cmd_str)
|
|
resp = []
|
|
next = True
|
|
while next:
|
|
code = sock.read(3)
|
|
next = sock.read(1) == b'-'
|
|
resp.append(sock.readline().strip().decode())
|
|
return int(code), resp
|
|
|
|
def __init__(self, host, port, ssl=False, username=None, password=None):
|
|
import ussl
|
|
self.username = username
|
|
addr = usocket.getaddrinfo(host, port)[0][-1]
|
|
sock = usocket.socket(usocket.AF_INET, usocket.SOCK_STREAM)
|
|
sock.settimeout(DEFAULT_TIMEOUT)
|
|
sock.connect(addr)
|
|
if ssl:
|
|
sock = ussl.wrap_socket(sock)
|
|
code = int(sock.read(3))
|
|
sock.readline()
|
|
assert code==220, 'cant connect to server %d' % code
|
|
self._sock = sock
|
|
|
|
code, resp = self.cmd(CMD_EHLO + ' ' + LOCAL_DOMAIN)
|
|
assert code==250, '%d' % code
|
|
if CMD_STARTTLS in resp:
|
|
code, resp = self.cmd(CMD_STARTTLS)
|
|
assert code==220, 'start tls failed, %d' % code
|
|
self._sock = ussl.wrap_socket(sock)
|
|
|
|
if username and password:
|
|
self.login(username, password)
|
|
|
|
def login(self, username, password):
|
|
self.username = username
|
|
code, resp = self.cmd(CMD_EHLO + ' ' + LOCAL_DOMAIN)
|
|
assert code==250, '%d' % code
|
|
|
|
auths = None
|
|
for feature in resp:
|
|
if feature[:4].upper() == CMD_AUTH:
|
|
auths = feature[4:].upper().split()
|
|
assert auths!=None, "no auth method"
|
|
|
|
from ubinascii import b2a_base64 as b64
|
|
if AUTH_PLAIN in auths:
|
|
cren = b64("\0%s\0%s" % (username, password))[:-1].decode()
|
|
code, resp = self.cmd('%s %s %s' % (CMD_AUTH, AUTH_PLAIN, cren))
|
|
elif AUTH_LOGIN in auths:
|
|
code, resp = self.cmd("%s %s %s" % (CMD_AUTH, AUTH_LOGIN, b64(username)[:-1].decode()))
|
|
assert code==334, 'wrong username %d' % code
|
|
code, resp = self.cmd(b64(password)[:-1])
|
|
else:
|
|
raise Exception("auth(%s) not supported " % ', '.join(auths))
|
|
|
|
assert code==235 or code==503, 'auth error %d' % code
|
|
return code, resp
|
|
|
|
def to(self, addrs):
|
|
code, resp = self.cmd(CMD_EHLO + ' ' + LOCAL_DOMAIN)
|
|
assert code==250, '%d' % code
|
|
code, resp = self.cmd('MAIL FROM: <%s>' % self.username)
|
|
assert code==250, 'sender refused %d' % code
|
|
|
|
if isinstance(addrs, str):
|
|
addrs = [addrs]
|
|
count = 0
|
|
for addr in addrs:
|
|
code, resp = self.cmd('RCPT TO: <%s>' % addr)
|
|
if code!=250 and code!=251:
|
|
print('%s refused' % addr)
|
|
count += 1
|
|
assert count!=len(addrs), 'recipient refused, %d' % code
|
|
|
|
code, resp = self.cmd('DATA')
|
|
assert code==354, 'data refused, %d' % code
|
|
return code, resp
|
|
|
|
def write(self, content):
|
|
self._sock.write(content)
|
|
|
|
def send(self, content=''):
|
|
if content:
|
|
self.write(content)
|
|
self._sock.write('\r\n.\r\n') # the five letter sequence marked for ending
|
|
line = self._sock.readline()
|
|
return (int(line[:3]), line[4:].strip().decode())
|
|
|
|
def quit(self):
|
|
self.cmd("QUIT")
|
|
self._sock.close()
|