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()