自学内容网 自学内容网

python制作一个简单的端口扫描器,用于检测目标主机上指定端口的开放状态

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()

输出结果


原文地址:https://blog.csdn.net/qq_68809241/article/details/143727232

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!