利用Python线程池以及队列模块实现PING扫描

　　本工具开始编写的时候使用多线程模块threading，但是发现运行的时候会报出很多的错误，其中一个解决办法是在创建的新的线程的时候,time.sleep(）一定的时间，但这就从一定程度上失去了多线程的意义，还有一个比较简便的办法是利用线程池,ThreadPoolExecutor，可以指定线程池中的线程数，然后由ThreadPoolExecutor进行调度。

　　此外，子线程与主线程之间的变量是不能共享的，比如起初设想用列表存放活跃主机self.alive_hosts = []，但最后这个结果是错误的，因此需要用queue模块来存储这样的共享变量，然后在主线程中打印显示出来。

from scapy.all import *
import logging
import optparse
import sys
import ipaddress
import threading    #开始用的是threading模块，后来改用线程池模块
import time
import queue
 9 from concurrent.futures import ThreadPoolExecutor

logging.getLogger('scapy.runtime').setLevel(logging.ERROR)

class PingNetwork:
    def __init__(self) -> None:
        self.network = self.get_params()   #目标网络（格式：192.168.140.0/24）
        self.alive_hosts = []              #如果不用多线程，那么这个列表可以作为存放活跃主机的变量，但是如果用多线程，需要用队列
        self.q = queue.Queue()
        

    def get_params(self):
        parser = optparse.OptionParser('Usage: < Program > -n network')
        parser.add_option('-n', '--network', dest='network', type='string', help='Specify network to scan')
        options, args = parser.parse_args()
        if options.network is None:
            print(parser.usage)
            sys.exit(0)
        try:
            network = ipaddress.ip_network(options.network)   #对于用户输入的网络参数通过ipaddress模块转换为相应的迭代对象
            return network
        except:
            print("[-] Input ip network wirth right format. e.g 192.168.4.0/24")
            sys.exit()
    
    def icmp_host(self, ip):
        try:
            packet = IP(dst=ip)/ICMP()
            res = sr1(packet, timeout=2, verbose=False)
            # print(res, type(res))
            if res:
                if res.type == 0:                 #判断是否为echo-reply
                    # print("="*100)
                    self.q.put(ip)
        except:
            pass
    
    def run(self):
        with ThreadPoolExecutor(30) as t:   #这里用到线程池，并指定了线程数为30
            for ip in self.network:
                ip = str(ip)             
                t.submit(self.icmp_host, ip=ip)          #submit方法与threading.Thread中方法的传入参数的方式不一样
        

        if self.q.empty():
        
            print("[0] No online host found!")
        else:
            while not self.q.empty():
                print("[-] Online hosts as follows: \n")
            
                print("\t%s" % self.q.get())

if __name__ == "__main__":
    pinger = PingNetwork()
    pinger.run()