1

操作系统：CentOS
Python：2.4 和 2.7

我正在尝试为我的应用程序构建 TCP 和 UDP 流量的中继或代理。
我的设置是 app1 通过不断发送 UDP 数据包来验证 app2 是否在线。
一旦 app1 验证 app2 在线，它就会通过 TCP 协议发送 app2 数据包。

到目前为止，这是我的代码：

#-------------------------------------
#SOCK_DGRAM --UDP
u_localPort = 20006
u_remotePort = 20006
u_remoteHost = "x.x.x.x"

u_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
u_sock.bind(('', u_localPort))
u_knownClient = None
u_knownServer = (u_remoteHost, u_remotePort)
#-------------------------------------

#-------------------------------------
#SOCK_STREAM --TCP
t_localPort = 10006
t_remotePort = 10006
t_remoteHost = "x.x.x.x"

#client
c_tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
c_tcp.connect((t_remoteHost, t_remotePort))

#server
s_tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s_tcp.bind(('', u_localPort))
s_tcp.listen(1)

conn, addr = s_tcp.accept()
#-------------------------------------

while True:
    u_data, u_addr = u_sock.recvfrom(4096)
    if u_knownClient is None:
        u_knownClient = u_addr
    if u_addr == u_knownClient:
        u_sock.sendto(u_data, u_knownServer)
    else:
        u_sock.sendto(u_data, u_knownClient)


    data = conn.recv(1024)
    if not data: break
    print "received data:", data
    c_tcp.send(data)  # echo

使用此代码，UDP 数据似乎可以正确中继。但 TCP 不是。有什么问题？

4

1 回答 1

1

我在任何地方都看不到您使任何套接字成为非阻塞的。因此，任何对 recv 或 recvfrom 的调用都会挂起，直到实际收到数据。

因此，当您进入主循环时，您将永远阻塞对usock.recvfrom(4096). 在您实际收到一个 UDP 数据包之前，您的代码不会使其conn.recv(1024)排队。

考虑使用非阻塞套接字（具有正确的选择/轮询语义）。

于 2013-04-21T06:23:41.310 回答