portscaner 多线程、多协程并发端口扫描

import socket,time,re,sys,os,threading
import gevent
from gevent import monkey
monkey.patch_all()
 
socket.setdefaulttimeout(2)
 
#该方法用来处理用户数据的port范围，并计算范围内的port，将其添加到列表中，将列表返回
def handle_port(input_ports):
    try:
        pattern = re.compile('(^\d{1,5})-(\d{1,5}$)')
        match = pattern.match(input_ports)
        if match:
            start_port = int(match.group(1))
            end_port = int(match.group(2))
            if end_port <=65535 :
                if start_port < end_port:
                    list =[]
                    for i in range(start_port, end_port+1):
                        list.append(i)
                    return list
            else:
                print("端口范围输入有误")
                exit(0)
        else:
            print("端口格式输入格式有误。")
            exit(0)
    except Exception as err:
        print(err)
        exit(0)
 
#该方法用来处理用户数据的IP地址范围，并计算范围内的IP地址，将其添加到列表中，将列表返回
def handle_ip(input_addrs):
 
    try:
        pattern = re.compile('(\d{1,3}\.\d{1,3}\.\d{1,3}\.)(\d{1,3})-(\d{1,3})')  # 匹配标准点进制的IP
        match = pattern.match(input_addrs)
        if match:
            list = []
            for i in range(int(match.group(2)),int(match.group(3))+1):
                addr = match.group(1)+str(i)
                list.append(addr)
            return list
        else:
            print("ip地址格式输入有误")
            exit(0)
    except Exception as err:
        print(err)
        exit(0)
 
#调用socket方法进行tcp端口扫描，client.connect()方法的返回值如果为None，则说明端口开放，若无返回值，说明连接超时，就没有返回值
def scaner(ip,port):
    try:
        client = socket.socket()
        res = client.connect((ip,port))
        if not res:
            open_port = []
            print(ip,":",port," is open")
            open_port.append((ip,port))
            return open_port
 
    except Exception as e:
        pass
 
    finally:
        client.close()
 
#本方法对传递来的列表ports进行循环，每次循环启动一个协程，在协程内部将port和addr进行配对，并调用scaner方法进行扫描
def coroutine_scan(addr,ports):
    list = []
    for port in ports:
        eve = gevent.spawn(scaner, addr, port)
        list.append(eve)
    gevent.joinall(list)
 
#本方法对传递来的列表addrs进行循环，每一次循环启动一个进程，并将addrs循环出来的数据和ports列表传递给coroutine_scan方法
def thread_scan(addrs,ports):
    for addr in addrs:
        t = threading.Thread(target=coroutine_scan, args=(addr, ports))
        t.start()
 
#本方法用来接收参数，并调用handle_port,handle_ip对输入的内容进行处理，然后将返回的列表数据传递给thread_scan
def main():
 
    input_addrs = input("输入IP地址范围：如'192.168.0.1-45'\n>>").strip()
    input_ports = input("输入端口范围：如'1000-1005'\n>>").strip()
 
    if len(input_addrs) and len(input_addrs)> 0:
 
        ports = handle_port(input_ports)
        addrs = handle_ip(input_addrs)
 
        print("开始扫描.......")
        thread_scan(addrs,ports)
 
    else:
        print("请输入正确的IP地址范围和端口。")
        main()
 
if __name__ == "__main__":
    main()