python 之網路編程（基於TCP協議Socket通信的粘包問題及解決）

8.4 粘包問題

粘包問題發生的原因：

1.發送端需要等緩衝區滿才發送出去，造成粘包(發送數據時間間隔很短,數據了很小,會合到一起,產生粘包),這樣接收端，就難於分辨出來了，必須提供科學的拆包機制。 即面向流的通信是無消息保護邊界的。

2.接收方不及時接收緩衝區的包，造成多個包接收（客戶端發送了一段數據，服務端只收了一小部分，服務端下次再收的時候還是從緩衝區拿上次遺留的數據，產生粘包）

粘包問題主要還是因為接收方不知道消息之間的界限，不知道一次性提取多少位元組的數據所造成的。

8.41 stract模塊

1、把整型數字轉成bytes類型 2、轉成的bytes是固定長度的

import struct
​
res=struct.pack('i',20332)    # i:整型 
print(res,len(res))          # b'lO\x00\x00'   4
​
res2=struct.unpack('i',res)
print(res2[0])              # 20332

8.42 利用stract模塊解決粘包

為位元組流加上自定義固定長度報頭，報頭中包含位元組流長度，然後一次send到對端，對端在接收時，先從緩存中取出定長的報頭（報頭中含有真實數據長度），然後再取真實數據

服務端：

from socket import *
import subprocess
import struct
........
while True:
    conn,client_addr=server.accept() #(連接對象，客戶端的ip和埠)
    print(client_addr)
    while True:
        try:
            cmd=conn.recv(1024)
            obj=subprocess.Popen(cmd.decode('utf-8'),# dir
                                 shell=True,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE
                                 )
            stdout=obj.stdout.read()
            stderr=obj.stderr.read()
​
            total_size=len(stdout) + len(stderr)# 1、製作固定長度的報頭 # 430
            header=struct.pack('i',total_size) 
​
            conn.send(header)                  # 2、發送報頭
​
            conn.send(stdout)                  #3、發送真實的數據
            conn.send(stderr)                  #subprocess返回byte類型，但需要gbk解碼
        except ConnectionResetError:
            break
​
    conn.close()
server.close()

客戶端：

from socket import *
import struct
..........
while True:
    cmd=input('>>>: ').strip()
    if not cmd:continue
    client.send(cmd.encode('utf-8'))          # dir
​
    header=client.recv(4)                    #1、先收固定長度的報頭
​
    total_size=struct.unpack('i',header)[0]    #2、解析報頭
    print(total_size)                        # 430
   
    recv_size=0                             #3、根據報頭內的信息，收取真實的數據
    res=b''
    while recv_size < total_size:
        recv_data=client.recv(1024)
        res+=recv_data
        recv_size+=len(recv_data)
    print(res.decode('gbk'))
client.close()

8.43 自定義報頭

我們可以把報頭做成字典，字典里包含將要發送的真實數據的詳細信息，字典然後json序列化，編碼成byte類型，然後用struck將數據長度打包成4個位元組（4個自己足夠用了）

發送時：

先發報頭長度，再編碼報頭內容然後發送，最後發真實內容

接收時：先收報頭長度，用struct取出來，根據取出的長度收取報頭內容，然後解碼，反序列化，從反序列化的結果中取出待取數據的詳細信息，然後去取真實的數據內容

服務端：

from socket import *
import subprocess
import struct
import json
...........
while True:
    conn,client_addr=server.accept() #(連接對象，客戶端的ip和埠)
    print(client_addr)
    while True:
        try:
            cmd=conn.recv(1024)
            obj=subprocess.Popen(cmd.decode('utf-8'),
                                 shell=True,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE
                                 )
            stdout=obj.stdout.read()
            stderr=obj.stderr.read()
​
            header_dic={                            # 1、製作報頭
                'total_size':len(stdout) + len(stderr),
                'md5':'123svsaef123sdfasdf',
                'filename':'a.txt'
            }
            header_json = json.dumps(header_dic)
            header_bytes = header_json.encode('utf-8')
​
            header_size=len(header_bytes)             # 2、先發送報頭的長度
            conn.send(struct.pack('i',header_size))
​
            conn.send(header_bytes)                  # 3、發送報頭
​
            conn.send(stdout)                        # 4、發送真實的數據
            conn.send(stderr)
        except ConnectionResetError:
            break
    conn.close()
server.close()

客戶端：

from socket import *
import struct
import json
.........
while True:
    cmd=input('>>>: ').strip()
    if not cmd:continue
    client.send(cmd.encode('utf-8'))
    
    header_size=struct.unpack('i',client.recv(4))[0]     #1、先收報頭的長度
    header_bytes=client.recv(header_size)               #2、接收報頭
​
    header_json=header_bytes.decode('utf-8')            #3、解析報頭
    header_dic=json.loads(header_json)
    print(header_dic)
​
    total_size=header_dic[ 'total_size']                #4、根據報頭內的信息，收取真實的數據
    # print(total_size)     #1025
    
    recv_size=0
    res=b''
    while recv_size < total_size:
         recv_data=client.recv(1024)
         res+=recv_data
         recv_size+=len(recv_data)
​
    print(res.decode('gbk'))
client.close()

客戶端實現等待後重連：

import socket
import time
​
while True:
    try:
        client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
​
        client.connect(('127.0.0.1',8080))
        break
    except ConnectionRefusedError:
        time.sleep(3)
        print('等待3秒。。。')