怎样保证socket.recv接收完数据

　　最近在使用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)