我可以让socket.makefile具有与常规文件相同的读取语义吗？

Question

Python文件对象有一个read方法,它接受一个可选的size参数,它基本上是要返回的最大字节数.例如:

fname = "message.txt"
open(fname, "w").write("Hello World!")
print open(fname).read()   # prints the entire file contents
print open(fname).read(5)  # print "Hello"
print open(fname).read(99) # prints "Hello World!"

因此,即使我们的文件少于99个字符,也会read(99)立即调用所有可用数据.

我想在socket.makefile返回的文件对象上获得此行为.但如果我说:

import socket
ADDR = ("localhost", 12345)

listener = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
listener.bind(ADDR)
listener.listen(1)

client = socket.create_connection(ADDR)
cf = client.makefile("r+b", bufsize=0)

server, client_addr = listener.accept()
sf = server.makefile("r+b", bufsize=0)

sf.write("Hello World!")
sf.flush()
print cf.read(99)         # hangs forever

根据socket.makefile文档,"可选模式和bufsize参数的解释方式与内置文件()函数的解释方式相同." 但是我的原始文件示例即使在我说的时候open(fname, "r+b", 0)也可以工作,而我无法找到一种方法,使用套接字伪文件将所有可用数据返回到指定的字节数.

如果我只是使用它,这似乎工作得很好socket.recv:

import socket
ADDR = ("localhost", 12345)

listener = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
listener.bind(ADDR)
listener.listen(1)

client = socket.create_connection(ADDR)
server, client_addr = listener.accept()

server.sendall("Hello World!")
print client.recv(99)           # prints "Hello World!"

那么有什么方法可以使这个工作socket.makefile,或者这种"高级"功能根本不可用？

编辑: Python 3.2似乎行为正常,虽然参数语法socket.makefile似乎已经改变:

import socket
ADDR = ("localhost", 12345)

listener = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
listener.bind(ADDR)
listener.listen(1)

client = socket.create_connection(ADDR)
cf = client.makefile("rwb", buffering=0)

server, client_addr = listener.accept()
sf = server.makefile("rwb", buffering=0)

sf.write(b"Hello World!")
sf.flush()
print(cf.read(99))         # prints "Hello World!"

我还没有深入研究源代码,以找出这两个版本之间的区别,但这可能是一个提示.

Answer 1

这里的问题是client.read()尝试从当前位置读取到EOF,但套接字的EOF仅在另一侧关闭连接时出现.recv另一方面,将返回任何准备好读取的数据(如果有的话),或者可以根据阻塞和超时设置阻止.

与此相比:

import socket
ADDR = ("localhost", 12345)

listener = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
listener.bind(ADDR)
listener.listen(1)

client = socket.create_connection(ADDR)
cf = client.makefile("r+b", bufsize=0)

server, client_addr = listener.accept()
sf = server.makefile("r+b", bufsize=0)

sf.write("Hello World!")
sf.flush()
sf.close()
server.close()
print cf.read(99)         # does not hang