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