记录与扭曲的SMTP连接(Logging SMTP connections with Twisted

Python的新手在这里。我正在写使用双绞线和twisted.mail.smtp SMTP服务器。我想记录传入连接，并有可能甩掉他们，当有太多的并发连接。基本上，我想ConsoleMessageDelivery.connectionMade（）方法来在下文中，当一个新的连接是由被称为：

class ConsoleMessageDelivery:
    implements(smtp.IMessageDelivery)

    def connectionMade(self):
        # This never gets called

    def receivedHeader(self, helo, origin, recipients):
        myHostname, clientIP = helo
        headerValue = "by %s from %s with ESMTP ; %s" % (myHostname, clientIP, smtp.rfc822date())
        # email.Header.Header used for automatic wrapping of long lines
        return "Received: %s" % Header(headerValue)

    def validateFrom(self, helo, origin):
        # All addresses are accepted
        return origin

    def validateTo(self, user):
        if user.dest.local == "console":
            return lambda: ConsoleMessage()
        raise smtp.SMTPBadRcpt(user)

class ConsoleMessage:
    implements(smtp.IMessage)

    def __init__(self):
        self.lines = []

    def lineReceived(self, line):
        self.lines.append(line)

    def eomReceived(self):
        return defer.succeed(None)

    def connectionLost(self):
        # There was an error, throw away the stored lines
        self.lines = None

class ConsoleSMTPFactory(smtp.SMTPFactory):
    protocol = smtp.ESMTP

    def __init__(self, *a, **kw):
        smtp.SMTPFactory.__init__(self, *a, **kw)
        self.delivery = ConsoleMessageDelivery()

    def buildProtocol(self, addr):
        p = smtp.SMTPFactory.buildProtocol(self, addr)
        p.delivery = self.delivery
        return p

connectionMade是部分twisted.internet.interfaces.IProtocol ，而不是一部分twisted.mail.smtp.IMessageDelivery 。有在关心一个邮件服务器实现无代码的任何connectionMade上的消息传递的实现方法。

一个更好的地方， 每个连接的逻辑把在工厂。具体而言，接近一个很好的办法是用一个工厂的包装 ，至约连接限制和记录逻辑约维修SMTP连接的逻辑隔离。

扭曲带有一些工厂包装。一对夫妇特别是可能是有趣的，你是twisted.protocols.policies.LimitConnectionsByPeer和twisted.protocols.policies.LimitTotalConnectionsFactory 。

不幸的是，我不知道任何文件解释twisted.protocols.policies 。幸运的是，这不是太复杂。大多数模块在工厂的包装其他任意厂家添加一些片的行为。因此，例如，使用LimitConnectionsByPeer ，你做这样的事情：

from twisted.protocols.policies import LimitConnectionsByPeer
...
factory = ConsoleSMTPFactory()
wrapper = LimitConnectionsByPeer(ConsoleSMTPFactory(...))
reactor.listenTCP(465, wrapper)

这是所有的需要得到LimitConnectionsByPeer来完成其工作。

这里只有一点点参与编写自己的包装复杂性。首先，子类WrappingFactory 。然后实现两者的方法你感兴趣的定制。在你的情况，如果你要拒绝来自某个IP的连接，这将意味着压倒一切的buildProtocol 。然后，除非你也想自定义构造（你在这种情况下，不）的协议，调用基实现并返回其结果。例如：

from twisted.protocols.policies import WrappingFactory

class DenyFactory(WrappingFactory):
    def buildProtocol(self, clientAddress):
        if clientAddress.host == '1.3.3.7':
            # Reject it
            return None
         # Accept everything else
         return WrappingFactory.buildProtocol(self, clientAddress)

这些包装栈，这样你就可以将它们合并，以及：

from twisted.protocols.policies import LimitConnectionsByPeer
...
factory = ConsoleSMTPFactory()
wrapper = LimitConnectionsByPeer(DenyFactory(ConsoleSMTPFactory(...)))
reactor.listenTCP(465, wrapper)