快捷導航

python 用struct模塊解決黏包問題

更新時間：2020年11月07日 15:27:52 作者：手可摘星辰。

這篇文章主要介紹了python 用struct模塊解決黏包問題，幫助大家更好的理解和使用python，感興趣的朋友可以了解下

為什么會出現(xiàn)黏包現(xiàn)象：

　　首先只有在TCP協(xié)議中才會出現(xiàn)黏包現(xiàn)象，是因為TCP協(xié)議是面向流的協(xié)議，在發(fā)送的數(shù)據(jù)傳輸?shù)倪^程中還有緩存機制來避免數(shù)據(jù)丟失，因此，在連續(xù)發(fā)送小數(shù)據(jù)的時候，以及接收大小不符的時候容易出現(xiàn)黏包現(xiàn)象。本質(zhì)還是因為我們在接收數(shù)據(jù)的時候不知道發(fā)送的數(shù)據(jù)的長短。

解決黏包問題

　　在傳輸大量數(shù)據(jù)之前首先告訴接收端要發(fā)送的數(shù)據(jù)大小，如果想更漂亮的解決問題，可以通過struct模塊來定制協(xié)議。

struct模塊：

　　功能：可以把一個類型，如數(shù)字，轉(zhuǎn)成固定長度的bytes。

import struct

ret = struct.pack('i',456872783)   #'i'代表int，就是即將要把一共數(shù)字轉(zhuǎn)換成固定長度（4個字節(jié)）的bystes類型
print(ret)

num = struct.unpack('i',ret)  #轉(zhuǎn)換回來，返回一個元組
print(num[0])  #提前元組中的值得到4096

解決黏包問題：

　　服務(wù)端：

import struct
import socket
sk = socket.socket()
sk.bind(('127.0.0.1',8080))
sk.listen()
conn,addr = sk.accept()
while True:
  cmd = input('>>>')
  if cmd == 'q':   #當輸入‘q'時，結(jié)束，并向客戶端發(fā)送一個'q'。
    conn.send(b'q')
    break
  conn.send(cmd.encode('gbk'))  #將輸入的cmd命令發(fā)送給客戶端
  num = conn.recv(4)   #接收字節(jié)信息（返回的消息長度信息）。
  num = struct.unpack('i',num)[0]   #將接收的字節(jié)碼轉(zhuǎn)化為原來的類型并放在一個元組里面，后面加[0]是提前出元組中的值。
  res = conn.recv(int(num)).decode('gbk')   #接收長度為num 的消息。
  print(res)   #打印
conn.close()
sk.close()

　　客戶端：

import struct
import socket
import subprocess

sk = socket.socket()
sk.connect(('127.0.0.1',8080))
while True:
  cmd = sk.recv(1024).decode('gbk')  #接收服務(wù)端發(fā)送來的cmd命令
  if cmd == 'q': #當接收到‘q'時，結(jié)束。
    break
  # 在客戶端執(zhí)行接收到的cmd命令。并將正確的消息和錯誤的消息分別放入stdout和stderr管道。
  res = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
  std_out = res.stdout.read()   #讀取管道內(nèi)正確的消息
  std_err = res.stderr.read()   #讀取管道內(nèi)錯誤的消息
  len_num = len(std_out)+len(std_err)   #計算正確和錯誤消息的總長度
  num_by = struct.pack('i',len_num)    #將消息總長度轉(zhuǎn)換成長度為4的字節(jié)碼
  sk.send(num_by)   #發(fā)送消息長度信息
  sk.send(std_out)  #發(fā)送正確消息
  sk.send(std_err)   #發(fā)送錯誤消息

sk.close()

以上就是python 用struct模塊解決黏包問題的詳細內(nèi)容，更多關(guān)于python struct模塊的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: