最近在使用python進行網(wǎng)絡編程開發(fā)一個通用的tcpclient測試小工具。在使用socket進行網(wǎng)絡編程中，如何判定對端發(fā)送一條報文是否接收完成，是進行socket網(wǎng)絡開發(fā)必須要考慮的一個問題。這里，簡要介紹一下判別數(shù)據(jù)接收接收完成常用的三種方法：
1.基礎數(shù)據(jù)接收法：
使用基礎數(shù)據(jù)接收法時，當與服務socket斷開連接時，會接收到空字符串。因此，可以根據(jù)此特點，在程序中加入循環(huán)，一直接收數(shù)據(jù)，直到數(shù)據(jù)發(fā)送端關閉socket連接。適用場景：客戶端和服務器的鏈接為短鏈接（即一次socket通訊后連接就會關閉）。代碼如下：
|
1
2
3
4
5
6
7
8
9
10
|
import socket,struct,sys,timePort=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.尾標識方法
通過尋找接收的協(xié)議數(shù)據(jù)中的尾標識字符串，獲取完整的數(shù)據(jù)的數(shù)據(jù)報文。適用場景：接收的協(xié)議數(shù)據(jù)中包含相關的尾標識。代碼如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
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.負載長度方法
即通過協(xié)議數(shù)據(jù)中的負載長度值，確定有效報文的長度，適用場景：協(xié)議數(shù)據(jù)中包含負載協(xié)議字段。這種方法也是比較常用和通用的一種方法，但是需要一邊接收數(shù)據(jù)一邊解析數(shù)據(jù)。代碼如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
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) |
以上就是本文的全部內(nèi)容，希望本文的內(nèi)容對大家的學習或者工作能帶來一定的幫助，同時也希望多多支持服務器之家！
原文鏈接：http://www.cnblogs.com/litaozijin/p/6624029.html
