import argparse # 用于解析命令行参数
from socket import * # 导入 socket 库的所有内容,用于网络通信
from threading import * # 导入 threading 库的所有内容,用于多线程操作
# 创建一个信号量,初始值为 1,用于线程同步,防止多个线程同时打印导致输出混乱
screenLock = Semaphore(value=1)
# 定义 connScan 函数,用于扫描单个端口
def connScan(tgtHost, tgtport):
try:
# 创建一个 socket 对象,AF_INET 表示使用 IPv4,SOCK_STREAM 表示使用 TCP
connSkt = socket(AF_INET, SOCK_STREAM)
# 尝试连接到目标主机的指定端口
connSkt.connect((tgtHost, tgtport))
# 发送一个字符串,表示连接成功
connSkt.send('ViolentPython\r\n'.encode())
# 接收返回的数据,最多接收 100 字节
results = connSkt.recv(100)
# 获取屏幕锁,确保打印输出时不会与其他线程冲突
screenLock.acquire()
# 打印端口开放的信息
print(f'[+]{tgtport}/tcp open')
# 打印接收到的数据
print(f'[+]{results.decode("utf-8")}')
except:
# 如果发生异常,获取屏幕锁并打印端口关闭的信息
screenLock.acquire()
print(f'[-]{tgtport}/tcp closed')
finally:
# 释放屏幕锁
screenLock.release()
# 关闭 socket 连接
connSkt.close()
# 定义 portScan 函数,用于扫描多个端口
def portScan(tgtHost, tgtPorts):
try:
# 将目标主机名解析为 IP 地址
tgtIP = gethostbyname(tgtHost)
except:
# 如果解析失败,打印错误信息并返回
print(f"[-] Cannot resolve '{tgtHost}':Unknown host")
return
try:
# 将 IP 地址解析为主机名
tgtName = gethostbyaddr(tgtIP)
# 打印扫描结果的标题
print(f'\n[+]Scan Results for:{tgtName[0]}')
except:
# 如果解析失败,打印扫描结果的标题(使用 IP 地址)
print(f'\n[+]Scan Results for:{tgtIP}')
# 设置 socket 的默认超时时间为 1 秒
setdefaulttimeout(1)
# 遍历所有要扫描的端口
for tgtPort in tgtPorts:
# 为每个端口创建一个新的线程,调用 connScan 函数进行扫描
t = Thread(target=connScan, args=(tgtHost, int(tgtPort)))
t.start()
# 定义 main 函数,用于解析命令行参数并启动扫描
def main():
# 创建一个命令行参数解析器
parser = argparse.ArgumentParser(description='Port scanner usage')
# 添加 -H 参数,用于指定目标主机
parser.add_argument('-H', dest='tgtHost', type=str, help='specify target host')
# 添加 -p 参数,用于指定要扫描的端口,端口之间用逗号分隔
parser.add_argument('-p', dest='tgtPort', type=str, help='specify target port[s] separated by comma')
# 解析命令行参数
options = parser.parse_args()
# 获取目标主机和端口
tgtHost = options.tgtHost
tgtPorts = options.tgtPort
# 检查是否指定了目标主机和端口
if tgtHost is None:
print("请正确指定目标主机,使用 -H 参数输入目标主机信息。")
exit(0)
if tgtPorts is None:
print("请正确指定目标端口,使用 -p 参数输入目标端口信息(多个端口用逗号隔开)。")
exit(0)
# 将端口字符串分割为列表
tgtPorts = tgtPorts.split(',')
# 调用 portScan 函数开始扫描
portScan(tgtHost, tgtPorts)
# 如果直接运行这个脚本,则调用 main 函数
if __name__ == "__main__":
main()
输出结果