python如何設(shè)計tcp數(shù)據(jù)包協(xié)議類

這篇文章主要介紹了python如何設(shè)計tcp數(shù)據(jù)包協(xié)議類，具有一定借鑒價值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

一. 問題描述

在tcp編程中，最需要解決的就是粘包分包問題。所以，我們需要在每個數(shù)據(jù)包前面加上數(shù)據(jù)包的長度用以分割粘連的包。

二. 包結(jié)構(gòu)的設(shè)計

包的組成：包長度+數(shù)據(jù)域

包長度：用4個字節(jié)存儲數(shù)據(jù)域長度，數(shù)據(jù)域長度即為其所占字節(jié)數(shù)

數(shù)據(jù)域：由若干個變量組成，如果是定長變量則不用加變量長度

定長變量：我們?nèi)藶橐?guī)定，傳輸中的int為4字節(jié)定長變量

變長變量：那就是字符串啦

文字難理解，那我就畫個圖吧：

上圖的第一行是數(shù)據(jù)包的一個總體結(jié)構(gòu)

第二行是數(shù)據(jù)域內(nèi)部的一個結(jié)構(gòu)（數(shù)據(jù)域的變量數(shù)量和位置都是我們自己定的，上圖只是舉一個例子而已）

第三行是具體變量的結(jié)構(gòu)

如果不太清楚這個結(jié)構(gòu)，不要緊，我們來舉一個具體的例子

比如我們現(xiàn)在創(chuàng)建一個數(shù)據(jù)域是這樣的數(shù)據(jù)包：

數(shù)據(jù)域：666,"你好啊","hello",888

這個數(shù)據(jù)域一共存儲了四個變量，開頭和結(jié)尾是兩個整型變量，中間是兩個字符串變量。然后我們對這個數(shù)據(jù)域構(gòu)建出來的數(shù)據(jù)包是這個樣子的：

這下搞明白了吧，那下面就看看怎么用python封裝一個類實現(xiàn)上述結(jié)構(gòu)的數(shù)據(jù)包的組裝。

三. 代碼實現(xiàn)

 class Protocol:
 """
 規(guī)定：
 數(shù)據(jù)包頭部占4字節(jié)
 整型占4字節(jié)
 字符串長度位占2字節(jié)
 字符串不定長
 """
 
 def __init__(self, bs=None):
 """
 如果bs為None則代表需要創(chuàng)建一個數(shù)據(jù)包
 否則代表需要解析一個數(shù)據(jù)包
 """
 if bs:
  self.bs = bytearray(bs)
 else:
  self.bs = bytearray(0)
 
 def get_int32(self):
 try:
  ret = self.bs[:4]
  self.bs = self.bs[4:]
  return int.from_bytes(ret, byteorder='little')
 except:
  raise Exception("數(shù)據(jù)異常！")
 
 def get_str(self):
 try:
  # 拿到字符串字節(jié)長度(字符串長度位2字節(jié))
  length = int.from_bytes(self.bs[:2], byteorder='little')
  # 再拿字符串
  ret = self.bs[2:length + 2]
  # 刪掉取出來的部分
  self.bs = self.bs[2 + length:]
  return ret.decode(encoding='utf8')
 except:
  raise Exception("數(shù)據(jù)異常！")
 
 def add_int32(self, val):
 bytes_val = bytearray(val.to_bytes(4, byteorder='little'))
 self.bs += bytes_val
 
 def add_str(self, val):
 bytes_val = bytearray(val.encode(encoding='utf8'))
 bytes_length = bytearray(len(bytes_val).to_bytes(2, byteorder='little'))
 self.bs += (bytes_length + bytes_val)
 
 def get_pck_not_head(self):
 return self.bs
 
 def get_pck_has_head(self):
 bytes_pck_length = bytearray(len(self.bs).to_bytes(4, byteorder='little'))
 return bytes_pck_length + self.bs
 
 
if __name__ == '__main__':
 p = Protocol()
 
 p.add_int32(666)
 p.add_str("你好啊")
 p.add_str("hello")
 p.add_int32(888)
 
 r = Protocol(p.get_pck_not_head())
 
 print(r.get_int32())
 print(r.get_str())
 print(r.get_str())
 print(r.get_int32())

感謝你能夠認真閱讀完這篇文章，希望小編分享的“python如何設(shè)計tcp數(shù)據(jù)包協(xié)議類”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識等著你來學習!