Python 和 ZMap 结合的自动化漏洞扫描工具

import os
import subprocess

IP_RANGE = "pidancode.com"

def scan_port(ip, port):
    cmd = f"nmap -p {port} {ip}"
    result = os.system(cmd)
    if "open" in result:
        print(f"IP:{ip} Port:{port} is open.")

def scan(ip):
    ports = ["80", "443", "8080"]
    for port in ports:
        scan_port(ip, port)

def zmap_scan():
    cmd = f"zmap -p 80 -o result.txt {IP_RANGE}"
    subprocess.call(cmd, shell=True)
    with open("result.txt") as f:
        for line in f:
            ip = line.strip()
            scan(ip)

if __name__ == '__main__':
    zmap_scan()