resolve.py

#!/usr/bin/python3

import os
import sys
import socket
import argparse
from colored import fg, bg, attr
from threading import Thread
from queue import Queue
from multiprocessing.dummy import Pool


def banner():
	print("""
                                   _
               _ __ ___  ___  ___ | |_   _____       _ __  _   _
              | '__/ _ \/ __|/ _ \| \ \ / / _ \     | '_ \| | | |
              | | |  __/\__ \ (_) | |\ V /  __/  _  | |_) | |_| |
              |_|  \___||___/\___/|_| \_/ \___| (_) | .__/ \__, |
                                                    |_|    |___/

                                by @gwendallecoguic

""")
	pass

banner()


parser = argparse.ArgumentParser()
parser.add_argument( "-o","--host",help="set hosts file list" )
parser.add_argument( "-t","--threads",help="threads, default 10" )
parser.add_argument( "-i","--ip",help="also store the ip address", action="store_true" )
parser.parse_args()
args = parser.parse_args()

if args.threads:
    _threads = int(args.threads)
else:
    _threads = 10

if args.ip:
    _store_ip = True
else:
    _store_ip = False

t_hosts = []
if args.host:
    if os.path.isfile(args.host):
        fp = open( args.host, 'r' )
        t_hosts = fp.read().split("\n")
        fp.close()

n_host = len(t_hosts)

if not n_host:
    parser.error( 'hosts list missing' )

sys.stdout.write( '%s[+] %d hosts loaded: %s%s\n' % (fg('green'),n_host,args.host,attr(0)) )

def resolve( host, store_ip ):
    host = host.strip()
    if not len(host):
        return

    if t_multiproc['n_current']%5000 == 0:
        save( store_ip )

    sys.stdout.write( 'progress: %d/%d\r' %  (t_multiproc['n_current'],t_multiproc['n_total']) )
    t_multiproc['n_current'] = t_multiproc['n_current'] + 1

    try:
        ip = socket.gethostbyname( host )
        t_alive[host] = ip
        # print(ip)
    except Exception as e:
        t_dead.append( host )
        # sys.stdout.write( "%s[-] error occurred: %s (%s)%s\n" % (fg('red'),e,host,attr(0)) )

t_alive = {}
t_dead = []

t_multiproc = {
    'n_current': 0,
    'n_total': n_host
}

# pool = Pool( _threads )
# pool.map( resolve, t_hosts )
# pool.close()
# pool.join()

def save( store_ip ):
    fp = open( 'hosts_alive', 'w' )
    for h in sorted(t_alive.keys()):
        if store_ip:
            fp.write( "%s:%s\n" % (h,t_alive[h]) )
        else:
            fp.write( "%s\n" % h )
    fp.close()

    fp = open( 'hosts_dead', 'w' )
    for h in t_dead:
        fp.write( "%s\n" % h )
    fp.close()


def doWork():
    while True:
        host = q.get()
        resolve( host, _store_ip )
        q.task_done()


q = Queue( _threads*2 )

for i in range(_threads):
    t = Thread( target=doWork )
    t.daemon = True
    t.start()

try:
    for host in t_hosts:
        q.put( host )
    q.join()
except KeyboardInterrupt:
    sys.exit(1)


# print( t_alive)
# print( t_dead)
sys.stdout.write( '%s[+] %d hosts alive, %d dead hosts%s\n' % (fg('green'),len(t_alive),len(t_dead),attr(0)) )
sys.stdout.write( '%s[+] Output files hosts_alive and hosts_dead created%s\n' % (fg('green'),attr(0)) )

save( _store_ip )

exit()