day29-struct模块解决黏包问题

#struct模块可以把一个数据类型，例如数字int，转化成固定长度(4个字节)的bytes。int转为4个bytes。
#在大量传输数据之前先告诉接收端即将接收数据的大小，方可解决黏包问题：
#利用struct模块打包要发送的数据的长度，接收端接收这个包，解包之后，接收这个长度的数据。
import struct
pack = struct.pack('i',4096) #打包，'i'代表int，把数字转化成固定长度(4个字节)的bytes。
print(pack)                     #b'x00x10x00x00'，4个字节（bytes）
unpack = struct.unpack('i',pack)#解包
print(unpack)                     #(4096,) 解包的结果是一个元祖
print(unpack[0])                  #4096，元祖的第一个元素就是字节数

#server:
import socket
import struct
sk = socket.socket()
sk.bind(('127.0.0.1',8888))
sk.listen()
conn,addr = sk.accept()
while True:
    cmd = input('>>>')
    if cmd == 'q':
        conn.send(b'q')
        break
    conn.send(cmd.encode('gbk'))     #encode编码，把str转化为bytes。
    pack = conn.recv(4)       #接收的是client的包，由4个字节组成，这里不需要decode解码，因为下面是解包。
    unpack = struct.unpack('i',pack)
    ret = conn.recv(unpack[0]).decode('gbk')
    print(ret)

conn.close()
sk.close()

#client:
import socket
import subprocess
import struct
sk = socket.socket()
sk.connect(('127.0.0.1',8888))
while True:
    cmd = sk.recv(1024).decode('gbk')
    if cmd == 'q':
        break
    ret = subprocess.Popen(cmd,shell=True,
                     stdout=subprocess.PIPE,
                     stderr=subprocess.PIPE)
    std_out = ret.stdout.read() #读取出来的内容是 bytes
    std_err = ret.stderr.read() #读取出来的内容是 bytes
    len_num = len(std_out) + len(std_err) #计算出两个输出的长度
    pack = struct.pack('i',len_num)        #打包长度，pack由4个字节组成
    sk.send(pack) #server的pack = conn.recv(4)接收这个pack。
    sk.send(std_out)
    sk.send(std_err)

sk.close()