用python編寫的簡易端口掃描器

#coding = utf-8

'''
python 3.4
通過簡單的TCP端口連接來判斷指定IP是否開放了指定端口。
'''

import socket
import optparse
import re
import threading
import sys

def anlyze_host(target_host):
#將從--host參數獲取到的目標值轉換為標准的xxx.xxx.xxx.xxx形式，其中主要是利用socket的gethostbyname函數將域名形式的值轉換為四位點進制形式
    try:
        pattern = re.compile(r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}') #匹配標准點進制的IP
        match = pattern.match(target_host)
        if match:
            return(match.group())
        else:
            try:
                target_host = socket.gethostbyname(target_host) #如果不是，就把target_host的值作為域名進行解析
                return(target_host)
            except Exception as err:
                print('地址解析錯誤：',err)
                exit(0)
    except Exception as err:
        print('請注意錯誤1：',sys.exc_info()[0],err)
        print(parser.usage)
        exit(0)
                    

def anlyze_port(target_port):
#解析--port參數傳入的值，返回端口列表
    try:
        pattern = re.compile(r'(\d+)-(\d+)')    #解析連接符-模式
        match = pattern.match(target_port)
        if match:
            start_port = int(match.group(1))
            end_port = int(match.group(2))
            return([x for x in range(start_port,end_port + 1)])
        else:
            return([int(x) for x in target_port.split(',')])
    except Exception as err:
        print('請注意錯誤2:',sys.exc_info()[0],err)
        print(parser.usage)
        exit(0)

def scanner(target_host,target_port):
#創建一個socket對象
    s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    s.settimeout(5)
    try:
        s.connect((target_host,target_port))
        #s.sendall(b'hello\r\n\r\n')
        #message = s.recv(100)
        #if message:
        print('[+]%s的%3s端口:打開' % (target_host,target_port)) #若可以建立連接，表示此端口是打開的
           # print(' %s' % message.decode('utf-8'))
    except socket.timeout:
        print('[-]%s的%3s端口:關閉' % (target_host,target_port)) #如果連接超時，表示此端口關閉
    except Exception as err:
        print('請注意錯誤3:',sys.exc_info()[0],err)
        exit(0)


        
def main():
   usage = 'Usage:%prog -h <host> -p <port>'
   parser = optparse.OptionParser(usage,version='%prog v1.0')
   parser.add_option('--host',dest='target_host',type='string',
                     help='需要掃描的主機,域名或IP')
   parser.add_option('--port',dest='target_port',type='string',
                    help='需要掃描的主機端口，支持1-100或21,53,80兩種形式')
   (options,args) = parser.parse_args()
   if options.target_host == None or options.target_port == None:
      print(parser.usage)
      exit(0)
   else:
      target_host = options.target_host
      target_port = options.target_port

   target_host = anlyze_host(target_host)
   target_port = anlyze_port(target_port)

   for port in target_port:
      t = threading.Thread(target=scanner,args=(target_host,port))#多線程掃描端口
      t.start()

if __name__ == '__main__':
   main()  

運行的結果為：

c:\Python34\python.exe TcpScanner.py --host 192.168.2.1 --port 1-1024
[+]192.168.2.1的 25端口:打開
[+]192.168.2.1的110端口:打開
[+]192.168.2.1的119端口:打開
[+]192.168.2.1的143端口:打開
[+]192.168.2.1的465端口:打開
[+]192.168.2.1的563端口:打開
[+]192.168.2.1的587端口:打開
[+]192.168.2.1的993端口:打開
[+]192.168.2.1的995端口:打開
[+]192.168.2.1的 80端口:打開
[-]192.168.2.1的  1端口:關閉
[-]192.168.2.1的 18端口:關閉
[-]192.168.2.1的  4端口:關閉
[-]192.168.2.1的  8端口:關閉
[-]192.168.2.1的 13端口:關閉
[-]192.168.2.1的  9端口:關閉
[-]192.168.2.1的 42端口:關閉
[-]192.168.2.1的 19端口:關閉
[-]192.168.2.1的 67端口:關閉
[-]192.168.2.1的 21端口:關閉
[-]192.168.2.1的 14端口:關閉
[-]192.168.2.1的 17端口:關閉
……