利用Python实现批量获取端口Banner的工具

主要用到的模块包括：

• 　　Socket 通讯模块
• 　　threading多线程模块
• 　　ipaddress ip地址模块
• 　　optparse 命令行参数处理模块

 

import socket
import sys
import optparse
import ipaddress
import threading

 

class BannerGrab:
def __init__(self) -> None:
self.target = self.get_params()

def get_params(self):
parser = optparse.OptionParser('Usage: <Program> -t target ')
parser.add_option('-t','--target', dest='target', type='string', help='Specify target IP address')
options, args = parser.parse_args()
if options.target is None:
print(parser.usage)
sys.exit(0)
try:
ipaddress.ip_address(options.target) #判断用户输入的IP地址是否合法，如果不是，则退出程序
except:
print('[-] Please enter IP address')
sys.exit(0)

return options.target

def banner_grab(self, port):
try:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.settimeout(1)
s.connect((self.target, port))
banner = s.recv(1024).decode('utf-8')
print("Port %d is open \t %s" % (port, banner))
except:
pass　　　　　　　　　　　　　　　　　　#如果连接出错，则表明端口关闭等，直接忽略即可

def run(self):
try:
for port in range(1,65535):
t = threading.Thread(target=self.banner_grab, args=(port, ))
t.start()

except KeyboardInterrupt:
print("[-] Exit the program now!")
sys.exit(0)

if __name__ == "__main__":
banner = BannerGrab()
banner.run()