如何在Python socket模块中全局禁用DNS解析及区分IP与主机名

嘿，针对你的无DNS需求，我给你两个问题的解决方案，你可以按需选用——不过把两者结合起来用其实更贴合你的“完全不解析域名、直接报错”的场景：

一、如何区分字符串格式的IP地址与主机名（域名）？ #

Python标准库的ipaddress模块是做这个判断最靠谱的工具，它能准确识别合法的IPv4和IPv6地址，剩下的自然就是主机名/域名了。

实现代码 #

import ipaddress

def is_valid_ip(s: str) -> bool:
    # 先处理IPv6常见的方括号格式
    cleaned_host = s.strip('[]')
    try:
        # 尝试解析为IPv4
        ipaddress.IPv4Address(cleaned_host)
        return True
    except ipaddress.AddressValueError:
        try:
            # 尝试解析为IPv6
            ipaddress.IPv6Address(cleaned_host)
            return True
        except ipaddress.AddressValueError:
            # 两种都不是，判定为主机名
            return False

# 测试用例
print(is_valid_ip("192.168.1.1"))    # True
print(is_valid_ip("[2001:db8::1]"))  # True
print(is_valid_ip("example.com"))    # False
print(is_valid_ip("localhost"))      # False

为什么选这个方法？ #

• 它是标准库自带的，不需要额外安装依赖
• 能覆盖所有合法的IP格式，包括压缩格式的IPv6
• 错误处理逻辑清晰，不会误判特殊格式的主机名

二、如何在Python socket模块中全局禁用DNS解析？ #

如果你的需求是任何情况下都不允许解析域名，遇到主机名直接报错退出，那么通过猴子补丁替换socket的核心解析函数getaddrinfo是最直接的方案——因为Python所有涉及域名解析的网络操作，底层都会调用这个函数。

实现代码 #

import socket
import ipaddress

# 先保存原始的getaddrinfo方法，后面要调用它处理IP地址
original_getaddrinfo = socket.getaddrinfo

def block_dns_resolution(host, port, family=0, type=0, proto=0, flags=0):
    # 先检查传入的host是不是合法IP
    try:
        cleaned_host = host.strip('[]')
        ipaddress.ip_address(cleaned_host)
    except ValueError:
        # 不是IP地址，直接抛出异常终止操作
        raise RuntimeError("DNS resolution is disabled. Only IP addresses are allowed.") from None
    
    # 如果是合法IP，调用原始方法完成后续连接准备
    return original_getaddrinfo(host, port, family, type, proto, flags)

# 全局替换socket的getaddrinfo
socket.getaddrinfo = block_dns_resolution

# 测试验证
# 尝试用域名连接，会直接报错
try:
    sock = socket.socket()
    sock.connect(("example.com", 80))
except RuntimeError as e:
    print(e)  # 输出: DNS resolution is disabled. Only IP addresses are allowed.

# 用IP连接则正常执行（如果网络通畅的话）
try:
    sock = socket.socket()
    sock.connect(("8.8.8.8", 53))
    print("Successfully connected to IP address.")
    sock.close()
except Exception as e:
    print(f"Connection failed: {str(e)}")

方案优势 #

• 全局生效：所有依赖socket模块的库（比如requests、urllib3、httpx等）都会自动遵守这个规则，不需要逐个修改
• 严格拦截：任何试图解析域名的操作都会立即抛出明确的错误，符合你“完全不解析域名”的要求
• 兼容性好：适用于Python 3.3+的所有版本，没有版本限制

内容的提问来源于stack exchange，提问作者ArekBulski