最近在使用python進行網絡編程開發一個通用的tcpclient測試小工具。在使用socket進行網絡編程中,如何判定對端發送一條報文是否接收完成,是進行socket網絡開發必須要考慮的一個問題。這里,簡要介紹一下判別數據接收接收完成常用的三種方法:
1.基礎數據接收法:
使用基礎數據接收法時,當與服務socket斷開連接時,會接收到空字符串。因此,可以根據此特點,在程序中加入循環,一直接收數據,直到數據發送端關閉socket連接。適用場景:客戶端和服務器的鏈接為短鏈接(即一次socket通訊后連接就會關閉)。代碼如下:
importsocket,struct,sys,time
Port=22220
#assumeasocketdisconnect(datareturnedisemptystring)meansalldatawas#donebeingsent.
defrecv_basic(the_socket):
total_data=[]
whileTrue:
data=the_socket.recv(20480)
ifnotdata:break
total_data.append(data)
return''.join(total_data)
2.尾標識方法
通過尋找接收的協議數據中的尾標識字符串,獲取完整的數據的數據報文。適用場景:接收的協議數據中包含相關的尾標識。代碼如下:
End='somethinguseableasanendmarker'
defrecv_end(the_socket):
total_data=[];data=''
whileTrue:
data=the_socket.recv(8192)
ifEndindata:
total_data.append(data[:data.find(End)])
break
total_data.append(data)
iflen(total_data)>1:
#checkifend_of_datawassplit
last_pair=total_data[-2]+total_data[-1]
ifEndinlast_pair:
total_data[-2]=last_pair[:last_pair.find(End)]
total_data.pop()
break
return''.join(total_data)
3.負載長度方法
即通過協議數據中的負載長度值,確定有效報文的長度,適用場景:協議數據中包含負載協議字段。這種方法也是比較常用和通用的一種方法,但是需要一邊接收數據一邊解析數據。代碼如下:
defrecv_size(the_socket):
#datalengthispackedinto4bytes
total_len=0;total_data=[];size=sys.maxint
size_data=sock_data='';recv_size=8192
whiletotal_len
sock_data=the_socket.recv(recv_size)
ifnottotal_data:
iflen(sock_data)>4:
size_data+=sock_data
size=struct.unpack('>i',size_data[:4])[0]
recv_size=size
ifrecv_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)foriintotal_data])
return''.join(total_data)
以上內容為大家介紹了Python網絡編程調用接收數據的三種方法,希望對大家有所幫助,如果想要了解更多Python相關知識,請關注IT培訓機構:千鋒教育。http://www.dietsnews.net/