day29-struct模块解决黏包问题

#struct模块可以把一个数据类型,例如数字int,转化成固定长度(4个字节)的bytes。int转为4个bytes。
#在大量传输数据之前先告诉接收端即将接收数据的大小,方可解决黏包问题:
#利用struct模块打包要发送的数据的长度,接收端接收这个包,解包之后,接收这个长度的数据。
import struct
pack = struct.pack('i',4096) #打包,'i'代表int,把数字转化成固定长度(4个字节)的bytes。
print(pack)                     #b'\x00\x10\x00\x00',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()

 

posted @ 2019-08-27 13:19  梁劲雄  阅读(159)  评论(0编辑  收藏  举报