我有一个处理数据流的 TCP。它一次需要 4096 个字节,但我需要一种方法在接收之前找出整个套接字的大小。这样我就可以确定在切换套接字之前套接字将被读取多少次。它还会通知这是该套接字的最后一次接收,以便我可以在选择切换到另一个套接字(当它可用时)之前对其执行某些操作。
def handle_tcp(self, sock, remote):
fdset = [sock, remote]
while True:
r, w, e = select.select(fdset, [], [])
# How to find out how much bytes need to be read from socket?
if sock in r:
non_blocking_calculation()
if remote.send(sock.recv(4096)) <= 0: break
if remote in r:
non_blocking_calculation()
if sock.send(remote.recv(4096)) <= 0: break
你比我先一步。按照Igor Kalinosvkiy的方式。您可以使用FIONREAD执行此操作或使用MSG_PEEK并继续阅读直到空为止。例如,之后select()
.
import fcntl, termios, array, math
...
sock_size = array.array('i', [0]) # initialise outside While loop
if sock in r:
fcntl.ioctl(sock, termios.FIONREAD, sock_size)
count = ceil(sock_size[0] / float(BUF_SIZE))
print("Socket size: {} and count: {}".format(sock_size[0], count)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)