linux 模拟tcp请求超时

import socket

def simulate_tcp_timeout(host, port, timeout_seconds):
    try:
        # 创建一个TCP/IP套接字
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(timeout_seconds)  # 设置超时时间

        # 连接到服务器
        server_address = (host, port)
        print(f'连接到 {server_address}')
        sock.connect(server_address)

        # 发送数据
        message = '这是一个测试消息'
        print(f'发送: {message}')
        sock.sendall(message.encode())

        # 接收响应
        data = sock.recv(1024)
        print(f'收到: {data.decode()}')

    except socket.timeout:
        print(f'请求超时，等待了 {timeout_seconds} 秒')
    except Exception as e:
        print(f'发生错误: {e}')
    finally:
        print('关闭连接')
        sock.close()

# 使用示例
simulate_tcp_timeout('example.com', 80, 5)