向Python smtpd库添加SMTP AUTH支持...无法重写方法？
我想扩展Python的smtpd库里的SMTPServer类，让它能够处理SMTP认证连接。听起来挺简单的……
所以，我想我可以这样开始：
def smtp_EHLO(self, arg):
print 'got in arg: ', arg
# do stuff here...
但不知为什么，这个方法从来没有被调用。Python的smtpd库会调用其他类似的方法，比如：
method = None
i = line.find(' ')
if i < 0:
command = line.upper()
arg = None
else:
command = line[:i].upper()
arg = line[i+1:].strip()
method = getattr(self, 'smtp_' + command, None)
为什么我的方法不被调用呢？
之后，我想我可能可以重写整个found_terminator(self):方法，但这似乎也不管用。
def found_terminator(self):
# I add this to my child class and it never gets called...
我是不是做错了什么，还是……也许我今天还没完全清醒……
import smtpd
import asyncore
class CustomSMTPServer(smtpd.SMTPServer):
def smtp_EHLO(self, arg):
print 'got in arg: ', arg
def process_message(self, peer, mailfrom, rcpttos, data):
print 'Receiving message from:', peer
print 'Message addressed from:', mailfrom
print 'Message addressed to :', rcpttos
print 'Message length :', len(data)
print 'HERE WE ARE MAN!'
return
# Implementation of base class abstract method
def found_terminator(self):
print 'THIS GOT CALLED RIGHT HERE!'
line = EMPTYSTRING.join(self.__line)
print >> DEBUGSTREAM, 'Data:', repr(line)
self.__line = []
if self.__state == self.COMMAND:
if not line:
self.push('500 Error: bad syntax')
return
method = None
i = line.find(' ')
if i < 0:
command = line.upper()
arg = None
else:
command = line[:i].upper()
arg = line[i+1:].strip()
method = getattr(self, 'smtp_' + command, None)
print 'looking for: ', command
print 'method is: ', method
if not method:
self.push('502 Error: command "%s" not implemented' % command)
return
method(arg)
return
else:
if self.__state != self.DATA:
self.push('451 Internal confusion')
return
# Remove extraneous carriage returns and de-transparency according
# to RFC 821, Section 4.5.2.
data = []
for text in line.split('\r\n'):
if text and text[0] == '.':
data.append(text[1:])
else:
data.append(text)
self.__data = NEWLINE.join(data)
status = self.__server.process_message(self.__peer,
self.__mailfrom,
self.__rcpttos,
self.__data)
self.__rcpttos = []
self.__mailfrom = None
self.__state = self.COMMAND
self.set_terminator('\r\n')
if not status:
self.push('250 Ok')
else:
self.push(status)
server = CustomSMTPServer(('127.0.0.1', 1025), None)
asyncore.loop()
2 个回答
1
简而言之：要给SMTPChannel增加额外的功能，你只需要声明一个函数，然后直接把它添加到smtpd.SMTPChannel里。
详细说明：
SMTPChannel类是用来响应用户在开放端口（通常是25号端口）输入的命令的。它查找可以响应的命令的方式是通过一种叫“自省”的方法，也就是检查所有可用的函数属性。
需要注意的是，SMTPChannel里的函数必须以“smtp_”开头。例如，如果你想要响应HELP命令，你就需要创建一个叫做smtpd.SMTPChannel.smtp_HELP的函数。
下面的函数来自源代码，详细说明了自省的过程。
class SMTPChannel(asynchat.async_chat):
method = getattr(self, 'smtp_' + command, None)
有效代码
步骤1：声明一个将被调用的函数
def smtp_HELP(self,arg):
self.push("[8675309] GPT Answers to HELP")
步骤2：将以下函数添加到smtpd.SMTPChannel
class FakeSMTPServer(smtpd.SMTPServer):
"""A Fake smtp server"""
smtpd.SMTPChannel.smtp_HELP = smtp_HELP
步骤3：使用Telnet连接到localhost的25号端口进行测试
Trying 127.0.0.1...
Connected to localhost.
Escape character is '^]'.
220 vics-imac.fios-router.home ESMTP Sendmail 6.7.4 Sunday 17 March 2019
HELP
[8675309] GPT Answers to HELP
3
你需要扩展一下 SMTPChannel 这个东西——因为 smtp_ 开头的方法就是在这里实现的。你对 SMTPServer 的扩展只需要返回你自己创建的这个通道的子类就可以了。