如何解決Python socket連接中的粘包、精確傳輸問題

小編給大家分享一下如何解決Python socket連接中的粘包、精確傳輸問題，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

粘包：

• 發生原因：

當調用send的時候，數據并不是即時發給客戶端的。而是放到了系統的socket發送緩沖區里，等緩沖區滿了、或者數據等待超時了，數據才會發送，所以有時候發送太快的話，前一份數據還沒有傳給客戶端，那么這份數據和上一份數據一起發給客戶端的時候就會造成“粘包” 。

• 解決方案：

解決根源的思想是避免不同段的數據一起發送。

1. 方案1：前一段數據send完后，等待一段時間再send第二段數據。缺點：時間效率低，而且也無法完全避免問題【因為不清楚該設置多少時間才能保證前一份數據已經發送】

2. 方案2：握手機制：前一段數據send完后，嘗試recv，等待客戶端回應，確認第一段數據發送完后，再send第二段數據。完美方案？

方案二的演示：

服務端【發送方】代碼：

import socket

server=socket.socket()
server.bind(("localhost",1234))
server.listen()

while True:
  print("正在等待。。。")
  conn,addr=server.accept()
  while True:
    try:
      conn.send(b"first info")
      ack=conn.recv(1024) #接收客戶端確認
      print(ack)
      conn.send(b"second info")
    except ConnectionResetError as e:
      print(e)
      break

server.close()

客戶端【接收方】代碼：

import socket

client=socket.socket()

client.connect(("localhost",1234))

data=client.recv(1024)
print(data.decode())
client.send(b"ack")#發送確認
data=client.recv(1024)
print(data.decode())
client.close()

不精確傳輸問題：

發生原因：

由于數據太大，發送方一次send不完，而接收方只recv一次，使得影響了后面數據的傳輸

解決方案：

解決根源的思想是改變recv的次數。

• 方案：將數據的大小發給接收方，讓接收方來決定recv的次數

方案實現代碼【以解決長數據shell命令傳輸為例】：

服務端【發送方】：

import socket,os

server=socket.socket()
server.bind(("localhost",1234))
server.listen()
while True:
  print("正在等待...")
  conn,addr=server.accept()
  print("連接成功!")
  while True:
    try:
      cmd=conn.recv(1024)
      data=os.popen(cmd.decode()).read()
      # print(data)
      cmd_len=len(data.encode())
      print(cmd_len)
      #發現這里如果cmd_len為0會導致異常，有些是沒有返回值的command
      if cmd_len==0:
        data="command has nothing return"
        cmd_len=len(data.encode())
      ##因為這里前面沒有發送操作，所以不用擔心粘包，如果有則要考慮處理
      conn.send(str(cmd_len).encode())#因為len結果是int，所以還要轉換
      #這里要處理粘包
      ack=conn.recv(1024)
      conn.send(data.encode())
    except ConnectionResetError as e:
      print(e)
      break

server.close()

客戶端【接收方】：

import socket

client=socket.socket()
client.connect(("localhost",1234))
while True:

  cmd = input(">>:")
  client.send(cmd.encode())
  data_len=client.recv(1024)
  data_len=int(data_len.decode())
  print(data_len)
  recv_len=0
  client.send(b'ack')
  total_data=b''
  while recv_len<data_len:
    data=client.recv(1024)
    recv_len+=len(data)
    total_data+=data
  print(total_data.decode())
client.close()

• 利用這個原理可以實現文件傳輸，只要能確定接受次數，就能保證文件傳輸的大小正確。

看完了這篇文章，相信你對“如何解決Python socket連接中的粘包、精確傳輸問題”有了一定的了解，如果想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！