Question

我正在尝试构建一个接受多个客户端连接的简单TCP中继。

Client1 --> TCPrelay1 --> RealServer1 --> TCPrelay1 --> Client1
Client2 --> TCPrelay1 --> RealServer1 --> TCPrelay1 --> Client2
Client3 --> TCPrelay1 --> RealServer1 --> TCPrelay1 --> Client3

那件事。对客户数量没有限制

我找到了一个UDP中继脚本here。

我尝试将其修改为TCP。我是python套接字的新手。那我的代码有什么问题呢？什么都没发生。而且它没有转发。

#SOCK_STREAM --TCP
localPort = 5000
remotePort = 5000

#SV
remoteHost = "xxxxx"

try:
    localPort = int(localPort)
except:
    fail('Invalid port number: ' + str(localPort))
try:
    remotePort = int(remotePort)
except:
    fail('Invalid port number: ' + str(remotePort))

try:
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind(('', localPort))
    s.listen(1)
except:
    fail('Failed to bind on port ' + str(localPort))

knownClient = None

while True:
    conn, addr = s.accept()
    conn2, addr2 = s.connect((remoteHost, remotePort))
    data = connection.recv(1024)
    if knownClient is None:
        knownClient = addr
    if addr == knownClient:
        s.sendall(data)
        print "Sent     : " + ":".join("{0:X}".format(ord(c)) for c in data)
    else:   
        s.sendall(data)
        print "Received : " + ":".join("{0:X}".format(ord(c)) for c in data)

[无视]
做了一些研究并在另一个SO问题上找到了这个逻辑：

import select

def fromAtoB(A, B):
    r, w = select.select([A], [B], [])
    if not r: select.select([A], [], [])
    elif not w: select.select([], [B], [])
    B.sendall(A.recv(4096))

但我仍在努力了解如何实施它。

Answer 1

您的代码接受一个连接，然后在一个方向上传输一个数据块，然后再也不会再查看该连接。

你的第二块代码有各种各样的问题，但最重要的是它可能会陷入僵局。在致电B.sendall期间，如果B拒绝阅读任何数据，直到它能够完成发送到A，这可能是您正在转发的协议，您的代理和B将永远等待彼此，因为您的代理拒绝从B读取任何数据，直到完成发送。

正确的继电器必须一次尝试做四件事，并且可能不会等到一件事完成另一件事。这四件事是：

1）从连接A读取。（除非已经备份了大量数据。）

2）从连接B读取。（除非已经备份了大量数据。）

3）写入连接A.（除非没有从连接B收到的未重映数据。）

4）写入连接B.（除非没有从连接A收到的未覆盖数据。）

特别是，您需要两个缓冲区，一个用于每个方向的未发送数据。您无法替换方向，因为这将要求您等待其中一个节点接收数据，如果该节点正在等待您接收数据，则可能会死锁。

通常，实现代理的最简单方法是每个连接有两个线程，每个方向一个。

Python简单TCP中继

1 个答案: