黏包

目录

1.黏包的问题
2.解决黏包问题
3.文件上传下载
4.struct模块使用

1.黏包的问题

#CMD服务端
import  socket
import subprocess
import struct
server = socket.socket()
server.bind(("127.0.0.1",9090))
server.listen()

while True:
    client,addr = server.accept()
    while True:
        try:
            # 接收命令
            cmd = client.recv(1024).decode("utf-8")
            p = subprocess.Popen(cmd,shell=True,stdout=-1,stderr=-1)
            # data与err_data 都是采用的系统编码 windows是GBK
            data = p.stdout.read()
            err_data = p.stderr.read()

            print("数据长度:%s" % (len(data)   + len(err_data)))

            # 先发送长度给客户端
            length = len(data)   + len(err_data)

            len_data = struct.pack("i",length)
            # 先发送长度 在发真实数据 有可能 长度数据和真实数据黏在一起 而接收方不知道长度数据的字节数 导致黏包
            # 解决的方案就是 长度信息占的字节数固定死  整数 转成一个固定长度字节

            client.send(len_data)

            client.send(data)
            client.send(err_data)
        except ConnectionResetError:
            client.close()
            print("连接中断......")
            break

#CMD客户端
import struct
import socket

c = socket.socket()
c.connect(("127.0.0.1",9090))
while True:
    cmd = input(">>:").strip()

    c.send(cmd.encode("utf-8"))

    # 先接收长度  长度固定为4个字节
    length = c.recv(4)
    len_data = struct.unpack("i",length)[0] # 转换为整型
    print("数据长度为%s" % len_data)

    all_data = b"" # 存储已接收数据
    rcv_size = 0 # 已接收长度
    # 接收真实数据
    # 循环接收 直到 接收到的长度等于总长度
    while  rcv_size < len_data:
        data = c.recv(1024)
        rcv_size += len(data)
        all_data += data

    print("接收长度%s" % rcv_size)
    print(all_data.decode("gbk"))

2.解决黏包问题

#CMD服务端
import  socket
import subprocess
import struct
import datetime
import json

server = socket.socket()
server.bind(("127.0.0.1",9090))
server.listen()

# 要求  不仅返回命令的结果 还要返回执行命令的时间  执行时间:2018/12/26

while True:
    client,addr = server.accept()
    while True:
        try:
            # 接收命令
            cmd = client.recv(1024).decode("utf-8")
            p = subprocess.Popen(cmd,shell=True,stdout=-1,stderr=-1)
            # data与err_data 都是采用的系统编码 windows是GBK
            data = p.stdout.read()
            err_data = p.stderr.read()

            print("数据长度:%s" % (len(data)   + len(err_data)))
            # 计算真实数据长度
            length = len(data)   + len(err_data)


            # 在发送数据之前发送额外的信息
            #t = "{执行时间:%s 真实数据长度:%s" % (datetime.datetime.now(),length)
            # 把要发送的数据先存到字典中
            t = {}
            t["time"] = str(datetime.datetime.now())
            t["size"] = length
            t["filename"] = "a.mp4"


            t_json = json.dumps(t) # 得到json格式字符串
            t_data = t_json.encode("utf-8") # 将json转成了字节
            t_length = struct.pack("i",len(t_data))



            # 1.先发送额外信息的长度
            client.send(t_length)
            # 2.发送额外信息
            client.send(t_data)

            # 3.发送真实数据
            client.send(data)
            client.send(err_data)
        except ConnectionResetError:
            client.close()
            print("连接中断......")
            break


#  1.发送了真实数据长度   2.发送了额外信息长度  3.发送额外信息  4.真实数据

#CMD客户端
import struct
import socket
import json

c = socket.socket()
c.connect(("127.0.0.1",9090))
while True:
    cmd = input(">>:")
    if not cmd:
        print("命令不能为空")
        continue
    c.send(cmd.encode("utf-8"))

    # 1.接收的是额外信息的长度
    length = c.recv(4)
    len_data = struct.unpack("i",length)[0] # 转换为整型

    # 2.接收额外信息
    t_data = c.recv(len_data)
    print(t_data.decode("utf-8"))

    json_dic = json.loads(t_data.decode("utf-8"))
    print("执行时间:%s" % json_dic["time"])

    data_size = json_dic["size"] # 得到数据长度


    # 3.接收真实数据
    all_data = b"" # 存储已接收数据
    rcv_size = 0 # 已接收长度
    # 接收真实数据
    # 循环接收 直到 接收到的长度等于总长度
    while  rcv_size < data_size:
        data = c.recv(1024)
        rcv_size += len(data)
        all_data += data

    print("接收长度%s" % rcv_size)
    print(all_data.decode("gbk"))

3.文件上传下载

"""
    1.应该采用TCP  必须保证数据时完整的
    2.只能传输bytes类型  刚好 文件操作 读取和写入都是bytes类型
    3.上传的思路
        自定义报头 发送文件名 文件大小 md5值
        读取文件数据  发送给对方
"""
#服务端
import  socket
import  struct
import  json
server = socket.socket()
server.bind(("127.0.0.1",9090))
server.listen()
client,addr = server.accept()

f =  open("接收到的文件",mode="wb")

head_len = client.recv(4)
json_len = struct.unpack("i",head_len)[0]

json_str = client.recv(json_len).decode("utf-8")
head = json.loads(json_str)
print(head)


recv_size = 0
while recv_size < head["size"]:
    data = client.recv(1024)
    f.write(data)
    recv_size += len(data)

print("接收完成...")

#客户端
import socket
import os
import json
import struct
c = socket.socket()
c.connect(("127.0.0.1",9090))

filepath=  r"D:\脱产5期内容\day30\视频\1.上周回顾.mp4"
f = open(filepath,mode="rb")

# 在发送数据前先发送报头
head = {"size":os.path.getsize(filepath),"filename":"回顾.mp4"}
json_data = json.dumps(head).encode("utf-8")

json_len = struct.pack("i",len(json_data))
c.send(json_len) # 发长度
c.send(json_data) # 发报头

# 发数据
while True:
    data = f.read(1024)
    if not data:
        break
    # 发送给服务器
    c.send(data)

print("上传完成...")

4.struct模块的使用

"""
    struct结构体
    可以将python中的数据类型 转换成bytes

"""
num = 1024

import struct

# 该函数 将一个python的数据转成bytes   第一个参数通常是i 其能转换的数据范围是c语言的int范围
# 如果int不够 那就用q  表示long long 型
res = struct.pack("i",num)
print(res)
print(len(res))


# 从字节转回整型
res2 = struct.unpack("i",res)
print(res2[0])