python扭曲-如何获取客户端IP

Question

How can i get the IP address of clients that are connecting to this dns server.

from twisted.internet import reactor
from twisted.names import client, dns, server


def main():
    """
    Run the server.
    """
    factory = server.DNSServerFactory(
        clients=[client.Resolver(resolv='/etc/resolv.conf')]
    )

    protocol = dns.DNSDatagramProtocol(controller=factory)

    reactor.listenUDP(10053, protocol)
    reactor.listenTCP(10053, factory)

    reactor.run()


if __name__ == '__main__':
    raise SystemExit(main())

Thank you in advance :P

Answer 1

from twisted.internet import reactor
from twisted.names import client, dns, server

class PrintClientAddressDNSServerFactory(server.DNSServerFactory):
    def buildProtocol(self, addr):
        print("Connection to DNSServerFactory from {}".format(addr))
        return server.DNSServerFactory.buildProtocol(self, addr)

class PrintClientAddressDNSDatagramProtocol(dns.DNSDatagramProtocol):
    def datagramReceived(self, datagram, addr):
        print("Datagram to DNSDatagramProtocol from {}".format(addr))
        return dns.DNSDatagramProtocol.datagramReceived(self, datagram, addr)


def main():
    """
    Run the server.
    """
    factory = PrintClientAddressDNSServerFactory(
        clients=[client.Resolver(resolv='/etc/resolv.conf')]
    )

    protocol = PrintClientAddressDNSDatagramProtocol(controller=factory)

    reactor.listenUDP(10053, protocol)
    reactor.listenTCP(10053, factory)

    reactor.run()


if __name__ == '__main__':
    raise SystemExit(main())