最近在使用python進行網路程式設計開發一個通用的tcpclient測試小工具。在使用socket進行網路程式設計中，如何判定對端發送一條封包是否接收完成，是進行socket網路開發必須要考慮的問題。這裡，簡單介紹一下判別資料接收完成常用的三種方法：
1.基礎資料接收法：
使用基礎資料接收法時，當與服務socket斷開連線時，會接收到空字串。因此，可以根據此特點，在程式中加入循環，一直接收數據，直到數據發送端關閉socket連接。適用場景：客戶端和伺服器的連結為短連結（即一次socket通訊後連線就會關閉）。 程式碼如下：
import socket,struct,sys,time
Port=22220
#assume a socket disconnect (data returned is empty string) means all data was #done being sent.
def recv_basic(the_socket):
total_data=[]
while True:
data = the_socket.recv(20480)
if not data: break
total_data.append(data)
return ''.join(total_data)2.尾標識方法
透過尋找接收的協定資料中的尾標識字串，取得完整的資料的資料報文。適用場景：接收的協定資料中包含相關的尾標識。程式碼如下：
End='something useable as an end marker'
def recv_end(the_socket):
total_data=[];data=''
while True:
data=the_socket.recv(8192)
if End in data:
total_data.append(data[:data.find(End)])
break
total_data.append(data)
if len(total_data)>1:
#check if end_of_data was split
last_pair=total_data[-2]+total_data[-1]
if End in last_pair:
total_data[-2]=last_pair[:last_pair.find(End)]
total_data.pop()
break
return ''.join(total_data)3.負載長度方法
即透過協定資料中的負載長度值，確定有效封包的長度，適用場景：協定資料中包含負載協定欄位。這種方法也是比較常用且通用的方法，但是需要一邊接收資料一邊解析資料。程式碼如下：
def recv_size(the_socket):
#data length is packed into 4 bytes
total_len=0;total_data=[];size=sys.maxint
size_data=sock_data='';recv_size=8192
while total_len<size:
sock_data=the_socket.recv(recv_size)
if not total_data:
if len(sock_data)>4:
size_data+=sock_data
size=struct.unpack('>i', size_data[:4])[0]
recv_size=size
if recv_size>524288:recv_size=524288
total_data.append(size_data[4:])
else:
size_data+=sock_data
else:
total_data.append(sock_data)
total_len=sum([len(i) for i in total_data ])
return ''.join(total_data)以上是詳解python網路程式設計呼叫recv函數完整接收資料的三種方法的詳細內容。更多資訊請關注PHP中文網其他相關文章！
