Search code examples
pythondjangoasynchronousreverse-dns

Python Asynchronous Reverse DNS Lookups


I am looking to do a large number of reverse DNS lookups in a small amount of time. I currently have implemented an asynchronous lookup using socket.gethostbyaddr and concurrent.futures thread pool, but am still not seeing the desired performance. For example, the script took about 22 minutes to complete on 2500 IP addresses.

I was wondering if there is any quicker way to do this without resorting to something like adns-python. I found this http://blog.schmichael.com/2007/09/18/a-lesson-on-python-dns-and-threads/ which provided some additional background.

Code Snippet:

ips = [...]
with concurrent.futures.ThreadPoolExecutor(max_workers = 16) as pool:
    list(pool.map(get_hostname_from_ip, ips))
def get_hostname_from_ip(ip):
    try:
        return socket.gethostbyaddr(ip)[0]
    except:
        return ""

I think part of the issue is that many of the IP addresses are not resolving and timing out. I tried:

socket.setdefaulttimeout(2.0)

but it seems to have no effect.


Solution

  • I discovered my main issue was IPs failing to resolve and thus sockets not obeying their set timeouts and failing after 30 seconds. See Python 2.6 urlib2 timeout issue.

    adns-python was a no-go because of its lack of support for IPv6 (without patches).

    After searching around I found this: Reverse DNS Lookups with dnspython and implemented a similar version in my code (his code also uses an optional thread pool and implements a timeout).

    In the end I used dnspython with a concurrent.futures thread pool for asynchronous reverse DNS lookups (see Python: Reverse DNS Lookup in a shared hosting and Dnspython: Setting query timeout/lifetime). With a timeout of 1 second this cut runtime from about 22 minutes to about 16 seconds on 2500 IP addresses. The large difference can probably be attributed to the Global Interpreter Lock on sockets and the 30 second timeouts.

    Code Snippet:

    import concurrent.futures
    from dns import resolver, reversename
    dns_resolver = resolver.Resolver()
    dns_resolver.timeout = 1
    dns_resolver.lifetime = 1
    ips = [...]
    results = []
    
    with concurrent.futures.ThreadPoolExecutor(max_workers = 16) as pool:
        results = list(pool.map(get_hostname_from_ip, ips))
    
    def get_hostname_from_ip(ip):
        try:
            reverse_name = reversename.from_address(ip)
            return dns_resolver.query(reverse_name, "PTR")[0].to_text()[:-1]
        except:
            return ""