程序在套接字交互期间挂起

Question

我有两个程序，sendfile.py 和 recvfile.py，它们应该交互以通过网络发送文件。它们通过 TCP 套接字进行通信。通信应该是这样的：

sender =====filename=====> receiver

sender <===== 'ok' ======= receiver
               or
sender <===== 'no' ======= receiver

if ok:
sender ====== file ======> receiver 

我已经有了

发送者和接收者代码在这里：

发送者：

import sys
from jmm_sockets import *

if len(sys.argv) != 4:
    print "Usage:", sys.argv[0], "<host> <port> <filename>"
    sys.exit(1)

s = getClientSocket(sys.argv[1], int(sys.argv[2]))

try:
    f = open(sys.argv[3])
except IOError, msg:
    print "couldn't open file"
    sys.exit(1)

# send filename
s.send(sys.argv[3])

# receive 'ok'
buffer = None
response = str()
while 1:
    buffer = s.recv(1)
    if buffer == '':
        break
    else:
        response = response + buffer
if response == 'ok':
    print 'receiver acknowledged receipt of filename'
    # send file
    s.send(f.read())
elif response == 'no':
    print "receiver doesn't want the file"

# cleanup
f.close()
s.close()

接收者：

from jmm_sockets import *

s = getServerSocket(None, 16001)
conn, addr = s.accept()


buffer = None
filename = str()

# receive filename
while 1:
    buffer = conn.recv(1)
    if buffer == '':
        break
    else:
        filename = filename + buffer
print "sender wants to send", filename, "is that ok?"
user_choice = raw_input("ok/no: ")

if user_choice == 'ok':
    # send ok
    conn.send('ok')
    #receive file
    data = str()
    while 1:
        buffer = conn.recv(1)
        if buffer=='':
            break
        else:
            data = data + buffer
            print data
else:
    conn.send('no')
conn.close()

我确信我在死锁中遗漏了一些东西，但不知道它是什么是。

Answer 1

对于阻塞套接字（这是默认设置，我假设您正在使用它（无法确定，因为您使用的是神秘模块jmm_sockets）），recv方法是阻塞的——当它“暂时没有什么可返回的”时，它不会返回空字符串，正如您似乎假设的那样。

您可以解决此问题，例如，通过在要发送的实际字符串之后发送显式终止符（绝不能出现在文件名中），例如 '\xff'，然后等待它位于另一端，表明所有字符串现已收到。

Answer 2

TCP 是一种流式传输协议。它没有消息边界的概念。对于阻塞套接字， recv(n) 将返回仅当发送方关闭套接字或显式调用 shutdown 时，零长度字符串（SHUT_WR）。否则，它可以返回长度从 1 到 n 个字节的字符串，并且将阻塞，直到至少有一个字节要返回。

您可以设计一个协议来确定何时获得完整的消息。几种方法是：

1. 使用固定长度的消息。
2. 发送指示总消息长度的固定长度消息，后跟消息的可变部分。
3. 发送消息，后跟消息中绝不会出现的唯一终止消息。

您可能面临的另一个问题是 send() 不是保证发送所有数据。返回值指示实际发送了多少字节，发送者有责任继续调用 send 来发送剩余的消息字节，直到它们全部发送为止。您可能更愿意使用 sendall() 方法。