在套接字上连续接收和发送

Question

我正在处理我需要连接到服务器的任务（没有透露服务器的详细信息），捕获回复，修改它并将其发回以进行验证。

我创建了以下代码来满足我的需要，但问题是在第一次正确回复后，服务器发送另一个。

代码：

# -*- encoding: utf-8 -*-

import socket
from decryptmsg import decryptmsg
from cleanmsg import cleanmsg

#connection 
ip="<IP>"
port=4000
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((ip,port))

def recvall(sock):
    BUFFER = 8192
    data = b''
    while True:
        part = sock.recv(BUFFER)
        data += part
        if len(part) < BUFFER:
            break
    return data


while True:

    print "[+] start communication"
    data = recvall(sock)
    print data
    data = cleanmsg(data)
    if data != None:
        valmis = decryptmsg(str(data))
        if valmis == None:
            print "[-] no results"
            break
        else:
            print "[+] sending message... "
            sock.send(valmis) 
            continue

当我遇到第二个问题时，我使用此代码很好地捕获了输入并按预期进行处理，但是当我尝试发送第二个回复时，我收到错误：

Traceback (most recent call last):
File "challenge.py", line 28, in <module>
     sock.send(valmis)
socket.error: [Errno 32] Broken pipe

如果我不关闭或关闭套接字，则不会向服务器发送任何回复。

如何告诉我的客户发送消息并等待没有 socket.shutdown 的回复？或者如果我需要为每个循环打开新的套接字，应该如何构造循环？服务器的回复每次都会改变，所以如果我完全打开新连接并请求数据，我会得到新的回复，并且该过程会重新开始。

更新： 问题似乎是在尝试从服务器接收第二个回复时，客户端只接收到第一行消息。

Answer 1

你怎么知道它没有发送任何东西？我稍微修改了您的代码（else: 子句中有一些奇怪的东西，我稍后会回来）。

import socket

#connection 
ip="localhost"
port=4000
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((ip,port))

while True:

    data = sock.recv(8192)
    if not data: break;
    print data
    if data != None:
        valmis = data
        if valmis == None:
            print "[-] no results"
            break
        else:
            print "[+] sending message... "
            sock.send(valmis) #this never gets sent without
            continue

基本上，这是您的代码的剥离版本 - 没有解密或外部功能。它只是发回它从服务器接收到的任何内容。

然后我用 ncat 运行了一个“服务器”：

ncat -l 4000

启动您的程序并开始输入行（1、2、3、4 等），这发生在“服务器”。客户及时回复我的信息：

test@xyzzy:/tmp$ ncat -l 4000
1
1
2
2
3
3

这发生在客户端：

test@xyzzy:/tmp$ python so.py 
1

[+] sending message... 
2

[+] sending message... 
3

[+] sending message... 

在我看来，这段代码运行良好。如果服务器没有收到你的回复，可能是服务器端有问题。例如，它可能期望响应中有一个终止符。您的cleanmsg 是否过多地清理消息，例如删除尾随换行符并且服务器希望收到一个？

当您在那里执行另一个 sock.recv() 时，您原来的 else 子句中存在问题。这意味着在收到回复后，您将在那里阻塞以等待来自服务器的下一条消息，当您收到一条消息时，您将continue 您的循环并再次点击sock.recv()。第二条消息已在您的 else 子句中使用。

如果您的服务器以某种方式确认您的解密，这可能是故意的。如果你的协议是这样的：

server -> client  (request)
client -> server  (decrypted message)
server -> client  (some kind of acknowledgement - unclear from your code)

server -> client  (request 2)
etc. 

那么您可能在 Jason 的评论中遇到了问题。 TCP 套接字与消息的概念完全无关。他们只是传输数据。当您的代码点击 sock.recv() 时，可能会发生以下五件事之一：

1. 套接字和调用块中没有任何内容
2. 有一个完整的“消息”，只有在套接字中，您会收到它
3. 有部分消息，您将收到该消息。可能是因为消息超过 8192 字节，或者您的代码只是在服务器仅传输了部分消息数据时才决定读取。
4. 有两个或更多完整的“消息”在等待，您将全部收到。
5. 四个，但最后一条消息是部分的

始终在使用 TCP 套接字时，必须满足场景 2-5。您必须解析数据，确保一切都在那里，如果没有，请等待更多。如果超出您的预期，请相应地处理它们。如果它们是完整的消息，则处理它们。如果最后一条消息是部分消息，则处理其他所有消息并等待更多消息。

如果消息似乎在使用 TCP 套接字的自制通信协议中“消失”，则 99% 的问题是由于假设套接字知道或关心您的“消息”的结构而引起的。一个非常常见的错误是在您的第一条消息之后将您的套接字读取为空并忽略您收到的所有内容。

希望这会有所帮助。从套接字通信的角度来看，没有额外 recv 的代码似乎可以正常工作 - 无需关闭套接字。可能是服务器端问题、协议问题或消息解析问题。

因此，您的套接字始终只有一个 recv 调用。即使您期待某种确认而不是新消息，也只有一个地方可以处理套接字数据。然后在那里做一些条件来检测你收到什么样的消息，然后决定如何处理它。