在使用 Python 的 `socket` 模块建立网络连接时，异常处理是非常关键的，因为网络通信可能受到多种因素的影响，例如目标主机不可达、端口未开放、连接超时、接收数据时连接被对方关闭等。为了确保程序的健壮性，需要对这些异常进行捕获和处理。 ### 异常类型及处理方式 在使用 `socket` 时，常见的异常包括： - `socket.timeout`：当设置了超时时间且操作超时。 - `ConnectionRefusedError`：连接被目标主机拒绝。 - `ConnectionResetError`：连接被对方重置。 - `socket.gaierror`：地址相关的错误，如主机名无法解析。 - `OSError`：操作系统层面的错误，如资源耗尽。 ### 异常处理示例 以下是一个客户端连接的异常处理示例，展示了如何使用 `try...except` 块来捕获和处理常见异常： ```python import socket # 设置全局超时时间为5秒 socket.setdefaulttimeout(5) try: # 创建TCP socket s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) print("Socket created successfully.") # 连接目标主机 print("Connecting to www.example.com:80...") s.connect(("www.example.com", 80)) print("Connected.") # 发送HTTP请求 s.sendall(b"GET / HTTP/1.1\r\nHost: www.example.com\r\n\r\n") # 接收响应数据 data = s.recv(4096) print("Received data:") print(data.decode('utf-8')) except socket.timeout: print("Connection timed out.") except ConnectionRefusedError: print("Connection refused by the remote host.") except socket.gaierror: print("Failed to resolve hostname.") except ConnectionResetError: print("Connection was reset by the peer.") except OSError as e: print(f"OS error occurred: {e}") except Exception as e: print(f"An unexpected error occurred: {e}") finally: # 确保关闭socket try: s.close() print("Socket closed.") except: pass ``` ### 异常处理的注意事项 - **设置超时机制**：可以使用 `setdefaulttimeout()` 或在 `connect()` 之前调用 `settimeout()` 来避免程序无限期阻塞。 - **使用 `finally` 确保资源释放**：即使发生异常，也应确保关闭 `socket`，防止资源泄漏。 - **区分不同异常类型**：根据不同的异常类型采取不同的恢复策略，比如重试、记录日志或提示用户。 - **捕获通用异常**：在 `except` 块中最后捕获 `Exception` 类型，用于处理未明确列出的其他异常。 ### 服务端异常处理 在服务端代码中，通常会在 `while` 循环中持续监听客户端连接，并在每个连接处理过程中使用异常捕获，防止一次异常导致整个服务崩溃： ```python import socket server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) server_socket.bind(('0.0.0.0', 8080)) server_socket.listen(5) print("Server is listening on port 8080...") while True: client_socket, addr = server_socket.accept() print(f"Connection from {addr}") try: while True: data = client_socket.recv(1024) if not data: break client_socket.sendall(data.upper()) except ConnectionResetError: print("Client disconnected unexpectedly.") except Exception as e: print(f"Error occurred: {e}") finally: client_socket.close() print("Client connection closed.") ``` 上述代码展示了服务端如何在每次客户端通信中使用 `try...except` 处理异常，确保即使一个客户端连接出错，也不会影响其他客户端的正常通信。