欢迎加入QQ讨论群258996829
苹果6袋
麦子学院
python调用recv函数完整接收数据的三种方法
发布时间：2017-03-28 22:43 回复:0 查看:2696 最后回复：2017-03-28 22:43
最近在使用
python开发
一个通用的tcpclient
测试小工具。在使用
socket
进行网络编程中，如何判定对端发送一条报文是否接收完成，是进行
socket
网络开发必须要考虑的一个问题。这里，简要介绍一下判别数据接收接收完成常用的三种方法：
1.基础数据接收法：
使用基础数据接收法时，当与服务socket
断开连接时，会接收到空字符串。因此，可以根据此特点，在程序中加入循环，一直接收数据，直到数据发送端关闭
socket
连接。适用场景：客户端和服务器的链接为 短链接（即一次
socket
通讯后连接就会关闭）。 代码如下：
1
import socket,struct,sys,time
2
3 Port=22220
4 #assume a socket disconnect (data returned is empty string) means all data was #done being sent.
5
def
recv_basic(the_socket):
6 total_data=[]
7
while
True:
8 data = the_socket.recv(20480)
9
if
not data:
break
10 total_data.append(data)
11
return ''.join(total_data)
2.尾标识方法
通过寻找接收的协议数据中的尾标识字符串，获取完整的数据的数据报文。适用场景：接收的协议数据中包含相关的尾标识。代码如下：
1 End='something useable as an end marker'
2
def
recv_end(the_socket):
3 total_data=[];data=''
4
while
True:
5 data=the_socket.recv(8192)
6
if End
in data:
7 total_data.append(data[:data.find(End)])
8
break
9 total_data.append(data)
10
if len(total_data)>1:
11 #check if end_of_data was split
12 last_pair=total_data[-2]+total_data[-1]
13
if End
in last_pair:
14 total_data[-2]=last_pair[:last_pair.find(End)]
15 total_data.pop()
16
break
17
return ''.join(total_data)
3.负载长度方法
即通过协议数据中的负载长度值，确定有效报文的长度，适用场景：协议数据中包含负载协议字段。这种方法也是比较常用和通用的一种方法，但是需要一边接收数据一边解析数据。代码如下：
1
def
recv_size(the_socket):
2 #data length is packed into 4 bytes
3 total_len=0;total_data=[];size=sys.maxint
4 size_data=sock_data='';recv_size=8192
5
while total_len<size:
6 sock_data=the_socket.recv(recv_size)
7
if
not total_data:
8
if len(sock_data)>4:
9 size_data+=sock_data
10 size=struct.unpack('>i', size_data[:4])[0]
11 recv_size=size
12
if recv_size>524288:recv_size=524288
13 total_data.append(size_data[4:])
14
else:
15 size_data+=sock_data
16
else:
17 total_data.append(sock_data)
18 total_len=sum([len(i)
for i
in total_data ])
19
return ''.join(total_data)
来源：
博客园