这里实现的端口扫描器有两种,一种是使用socket模块通过与目标主机相应端口建立TCP连接(完成完整的三次握手),来确定端口是否开放。
核心代码:
import socket
sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
sock.settimeout(2)
sock.connect((host,port))
sock.send('test socket\n')
result = sock.recv(100)
if result:
print "某某端口开放"
else:
print "某某端口关闭"
sock.close()
另一种是直接利用nmap模块,直接用它内置的方法PortScanner(),输出返回信息,nmap支持使用ACK、FIN、SYN-ACK等扫描方式,如向指定端口发送TCP SYN包,等待TCP ACk响应来确定该端口是否开放。
核心代码:
import nmap
nm = nmap.PortScanner()
nm.scan(host,port)
#nmap扫描结果有一个state变量
state = nm['host']['tcp'][int(port)]['state']
print host+" "+port+" &