Python之端口检测

1. 根据IP和端口检测当前参数下的端口是否开启
2. 如果参数为域名则通过DNS解析出对应的关联IP后再进行端口检测

实现代码

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import socket
import dns.resolver
import re


def __check_ip(ip):
    """
    类型判断，判断当前输入是否为ip
    :param ip: 域名或ip
    :return: 检测结果
    """
    p = re.compile('^((25[0-5]|2[0-4]\d|[01]?\d\d?)\.){3}(25[0-5]|2[0-4]\d|[01]?\d\d?)$')
    if p.match(ip):
        return True
    else:
        return False


def __resolution_a(domain):
    """
    DNS解析，根据域名解析对应的IP地址集合
    :param domain: 域名
    :return: ip集合
    """
    query_list = []
    a = dns.resolver.resolve(domain, 'A')
    for i in a.response.answer:
        for j in i.items:
            if j.rdtype == 1:
                ip = j.address
                query_list.append(ip)
    return query_list


def __detect_port(ip, port):
    """
    检测ip上的端口是否开放
    :param ip: ip或域名
    :param port: 端口
    :return: 端口是否开启
    """
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        s.connect((ip, int(port)))
        s.shutdown(2)
        return True
    except:
        return False


def detection(domain, port):
    """
    主函数
    :param domain: 域名或ip
    :param port: 端口
    :return: 检测结果
    """
    is_ip = __check_ip(domain)
    __data = []
    if is_ip:
        port_status = __detect_port(ip=domain, port=port)
        __data.append({
            "domain": domain,
            "port": port,
            "status": port_status
        })
    else:
        ips = __resolution_a(domain=domain)
        for ip in ips:
            port_status = __detect_port(ip=ip, port=port)
            __data.append({
                "domain": ip,
                "port": port,
                "status": port_status
            })
    return __data


if __name__ == '__main__':
    __domain = "baidu.com"
    __check_result = detection(domain=__domain, port=3306)
    print(__check_result)