基于python编写一个简单的多线程端口扫描脚本

基于python编写一个简单的多线程端口扫描脚本

代码如下：

#author bz 
'''
我想起夕阳下的奔跑，那是我逝去的青春。
'''
import threading
import queue
import socket
import optparse
import sys
#创建端口扫描类
class PortScaner(threading.Thread):
#需要传入端口队列，目标ip，探测超时信息
    def __init__(self,portqueue,ip,timeout=3):
        threading.Thread.__init__(self)
        self._portqueue = portqueue
        self._ip = ip
        self._timeout = timeout
    def run(self):
        while True:
        #判断端口列表是否为空
            if self._portqueue.empty():
             break
            port = self._portqueue.get(timeout=0.5)
            try:
                #创建socket对象为sock，并设置其套接字家族和类型
                '''socket.socket ( family ，type ，proto)
                family: 套接字家族可以使 AF_UNIX 或者 AF_INET
                type: 套接字类型，可以根据是面向连接的 SOCK_STREAM 还是非连接的 SOCK_DGRAM
                protocol: 一般不填，默认为 0
                '''
                sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
                #sock.settimeout(timeout):设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
                sock.settimeout(self._timeout)
                #sock.connect_ex() :主动初始化TCP服务器连接；一般address的格式为元组（hostname,port），如果连接出错，返回出错码。
                result_code = sock.connect_ex((self._ip, port))
                #若端口开发，返回0
                if result_code == 0:
                    sys.stdout.write("[%d]open \n" % port)
            except Exception as e:
                print(e)
            finally:
                sock.close()
#定义命令标准函数
def get_information():
    #创建命令解析模块
    parser = optparse.OptionParser()
    # ip 参数 -i
    parser.add_option('-i', '--ip', dest='targetip', default='127.0.0.1', type='string', help='target IP')
    # 端口参数 -p
    parser.add_option('-p', '--port', dest='port', default='80', type='string', help='scan_port')
    # 线程参数 -t
    parser.add_option('-t', '--thread', dest='threadNum', default='100', type='int', help='thread number')
    (options, args) = parser.parse_args()
    StartScan(options.targetip, options.port, options.threadNum)
#定义扫描函数
def StartScan(targetip,port,threadNum):
#端口列表
    portList = []
    portNumb = port
#判断单端口还是范围端口
    '''
    eg:192.168.1.1-100
    用split函数通过"-"将ip段划分为192.168.1.1和100,再将192.168.1.1进行分割，取出最后一位数字与100组成int型数字范围
    for循环遍历
    '''
    if '-' in port:
            for i in range(int(port.split('-')[0]),int(port.split('-')[1])+1):
                portList.append(i)
    else:
        portList.append(int(port))
    #目标ip地址
    ip = targetip
    #线程列表
    threads = []
    #线程数量
    threadNumber = threadNum
    #端口队列
    portQueue = queue.Queue()
    #生成端口，加入端口队列
    for port in portList:
        portQueue.put(port)
    for t in range(threadNumber):
        threads.append(PortScaner(portQueue,ip,timeout =3))
    #启动线程
    for thread in threads:
        thread.start()
    #阻塞线程
    for thread in threads:
        thread.join()
#主函数
if __name__ =='__main__':
    get_information()

测试：