Регистрация SMTP-соединений с помощью Twisted

Question

Питон новичок здесь. Я пишу SMTP-сервер, используя Twisted и Twisted.mail.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

Answer 1

connectionMade это часть twisted.internet.interfaces.IProtocol не является частью twisted.mail.smtp.IMessageDelivery, В реализации почтового сервера нет кода, который заботится о connectionMade метод реализации доставки сообщений.

Лучшее место для каждой логики подключения находится на заводе. И, в частности, хорошим способом решения этой проблемы является использование фабричной обертки, чтобы изолировать логику об ограничениях соединений и ведение журналов от логики об обслуживании SMTP-соединений.

Twisted поставляется с несколькими заводскими обертками. В частности, пара, которая может быть интересна для вас 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)