Python Socket库：如何2秒后终止连接及处理无效IP连接卡住问题

原脚本的核心问题 #

• s.connect()没有返回值，不能用if判断逻辑：连接成功时返回None，失败直接抛出异常，导致原脚本的else分支永远不会执行。
• 全局Socket对象复用容易引发状态混乱：多个连接共用同一个Socket实例，可能导致超时设置不生效或连接状态异常。
• 超时设置不符合需求：原代码设置1秒超时，但你需要的是2秒，且全局默认超时的优先级低于实例级超时，容易失效。

两种可行的超时控制方案 #

方案1：实例级超时+异常捕获 #

在函数内创建独立的Socket实例，设置超时时间并捕获各类连接异常，确保连接超时后能及时终止并返回结果。

修正后的代码：

import socket

def getBanner(ip, port, timeout=2):
    # 每个连接用独立的Socket实例，避免全局冲突
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        s.settimeout(timeout)  # 设置连接、接收操作的超时时间为2秒
        s.connect((ip, port))
        # 接收Banner，忽略编码错误避免崩溃
        banner = s.recv(1024).decode(errors='ignore')
        return banner.strip() if banner else "无Banner返回"
    except socket.timeout:
        return "连接超时（超过2秒）"
    except (ConnectionRefusedError, socket.gaierror):
        return "连接失败或IP无效"
    finally:
        s.close()  # 无论成败，确保Socket关闭

def main():
    ip = "192.168.152.0"
    banner = getBanner(ip, 21)
    print(banner)

if __name__ == "__main__":
    main()

方案2：用connect_ex替代connect #

connect_ex不会直接抛出异常，而是返回错误码（0表示成功，非0表示失败），配合超时设置可以更灵活处理连接结果：

import socket

def getBanner(ip, port, timeout=2):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.settimeout(timeout)
    try:
        # connect_ex返回0表示连接成功，其他值为错误码
        if s.connect_ex((ip, port)) == 0:
            banner = s.recv(1024).decode(errors='ignore')
            return banner.strip() if banner else "无Banner返回"
        else:
            return "端口拒绝连接"
    except socket.timeout:
        return "连接超时（超过2秒）"
    finally:
        s.close()

def main():
    ip = "192.168.152.0"
    banner = getBanner(ip, 21)
    print(banner)

if __name__ == "__main__":
    main()

关键细节说明 #

• settimeout(n)：针对当前Socket实例的所有操作（连接、接收、发送）设置超时，超时后触发socket.timeout异常，确保不会卡住。
• 避免全局Socket：每个连接用独立实例，防止因复用导致的状态残留问题。
• 异常覆盖：必须捕获socket.timeout处理超时，同时覆盖ConnectionRefusedError（端口拒绝）、socket.gaierror（IP解析失败）等常见连接异常。
• 编码容错：decode(errors='ignore')避免Banner包含非UTF-8字符时解码崩溃。

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