python实现简单的端口扫描工具
主要有两部分,第一个部分就是对ip地址和port端口参数的解析;第二部分是建立socket进行connect的连接测试。
参数解析
利用getopt模块进行命令行选项解析的时候,如果只解析长选项,短选项字符串置为空即可
opts, args = getopt.getopt(sys.argv[1:], '', ["host=", "port="])
对ip地址的合法性进行检查方法:
- 简单正则表达式,检查范围是0.0.0.0 ~ 999.999.999.999
import re
re.match('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}', '127.0.0.1')
- 导入第三方库IPy
pip3 install IPy
from IPy import IP
IP('127.0.0.1') try-catch ValueError来检查是否合法
- 利用Python3自带的库ipaddress
import ipaddress
ipaddress.ip_address('127.0.0.1') try-catch ValueError来检查是否合法
socket建立
在利用Python内置模块socket进行连接时,主要是对错误的捕获上,这里使用socket.connect_ex(address)
方法,该方法与socket.connect(address)
方法相比,主要是失败时的动作,后则是直接抛出异常,前者是返回一个错误指示器,成功时该指示器为0connect_ex()方法,所以可以利用变量的值是否为0来判断端口是否开通。
for port in ports:
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
result = sock.connect_ex((addr, port))
if result == 0:
print('{} open'.format(port))
else:
print('{} closed'.format(port))
sock.close()