美文网首页
day32-粘包问题和报头的定制

day32-粘包问题和报头的定制

作者: 天行_b6d0 | 来源:发表于2020-08-19 16:19 被阅读0次

    粘包问题

    TCP协议作为流式协议,只有TCP协议存在粘包问题。

    发送端可以是一K一K地发送数据,而接收端的应用程序可以两K两K地提走数据,当然也有可能一次提走3K或6K数据,或者一次只提走几个字节的数据,也就是说,应用程序所看到的数据是一个整体,或说是一个流(stream),一条消息有多少字节对应用程序是不可见的,因此TCP协议是面向流的协议,这也是容易出现粘包问题的原因。而UDP是面向消息的协议,每个UDP段都是一条消息,应用程序必须以消息为单位提取数据,不能一次提取任意字节的数据,这一点和TCP是很不同的。怎样定义消息呢?可以认为对方一次性write/send的数据为一个消息,需要明白的是当对方send一条信息的时候,无论底层怎样分段分片,TCP协议层会把构成整条消息的数据段排序完成后才呈现在内核缓冲区。

    例如基于tcp的套接字客户端往服务端上传文件,发送时文件内容是按照一段一段的字节流发送的,在接收方看了,根本不知道该文件的字节流从何处开始,在何处结束

    所谓粘包问题主要还是因为接收方不知道消息之间的界限,不知道一次性提取多少字节的数据所造成的。

    此外,发送方引起的粘包是由TCP协议本身造成的,TCP为提高传输效率,发送方往往要收集到足够多的数据后才发送一个TCP段。若连续几次需要send的数据都很少,通常TCP会根据优化算法把这些数据合成一个TCP段后一次发送出去,这样接收方就收到了粘包数据。

    1. TCP(transport control protocol,传输控制协议)是面向连接的,面向流的,提供高可靠性服务。收发两端(客户端和服务器端)都要有一一成对的socket,因此,发送端为了将多个发往接收端的包,更有效的发到对方,使用了优化方法(Nagle算法),将多次间隔较小且数据量小的数据,合并成一个大的数据块,然后进行封包。这样,接收端,就难于分辨出来了,必须提供科学的拆包机制。 即面向流的通信是无消息保护边界的。
    2. UDP(user datagram protocol,用户数据报协议)是无连接的,面向消息的,提供高效率服务。不会使用块的合并优化算法,, 由于UDP支持的是一对多的模式,所以接收端的skbuff(套接字缓冲区)采用了链式结构来记录每一个到达的UDP包,在每个UDP包中就有了消息头(消息来源地址,端口等信息),这样,对于接收端来说,就容易进行区分处理了。 即面向消息的通信是有消息保护边界的。
    3. tcp是基于数据流的,于是收发的消息不能为空,这就需要在客户端和服务端都添加空消息的处理机制,防止程序卡住,而udp是基于数据报的,即便是你输入的是空内容(直接回车),那也不是空消息,udp协议会帮你封装上消息头,实验略

    udp的recvfrom是阻塞的,一个recvfrom(x)必须对唯一一个sendinto(y),收完了x个字节的数据就算完成,若是y>x数据就丢失,这意味着udp根本不会粘包,但是会丢数据,不可靠

    tcp的协议数据不会丢,没有收完包,下次接收,会继续上次继续接收,己端总是在收到ack时才会清除缓冲区内容。数据是可靠的,但是会粘包。

    一、远程执行命令程序解决粘包问题

    server端

    import subprocess
    import struct
    from socket import *
    
    server = socket(AF_INET, SOCK_STREAM)
    # print(server)
    server.bind(('127.0.0.1', 8082))
    server.listen(5)
    while True:
        conn, client_addr = server.accept()
        print(conn)
        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.stdout.read()
                total_size = len(stdout) + len(stderr)
    
                # 先发送数据的长度
                conn.send(struct.pack('i',total_size))
                # 发送真正的数据
                conn.send(stdout)
                conn.send(stderr)
            except Exception:
                break
        conn.close()
    
    server.close()
    
    

    client端

    import struct
    from socket import *
    
    client = socket(AF_INET, SOCK_STREAM)
    # print(client)
    client.connect(('127.0.0.1', 8082))
    
    while True:
        cmd = input(">>: ").strip()
        if len(cmd) == 0:
            continue
        client.send(cmd.encode('utf-8'))
    
        # 先收数据的长度
        n = 0
        header = b''
        while n < 4:
            data = client.recv(1)
            header += data
            n += len(data)
    
        total_size = struct.unpack('i', header)[0]
    
        # 收真正的数据
        recv_size = 0
        res = b''
        while recv_size < total_size:
            data = client.recv(1024)
            res += data
            recv_size += len(data)
    
        print(res.decode('gbk'))
    
    client.close()
    

    二、定制复杂的报头

    server.py端

    import subprocess
    import os
    import struct
    import json
    from socket import *
    
    server = socket(AF_INET, SOCK_STREAM)
    # print(server)
    server.bind(('127.0.0.1', 8082))
    server.listen(5)
    while True:
        conn, client_addr = server.accept()
        print(conn)
        print(client_addr)
    
        while True:
            try:
                msg = conn.recv(1024).decode('utf-8')
                cmd,file_path=msg.split()
                if cmd == "get":
                    # 一、制作报头
                    header_dic={
                        "total_size":os.path.getsize(file_path),
                        "filename":os.path.basename(file_path),
                        "md5":"1231231231232132131232311"  # 加密
                    }
                    header_json=json.dumps(header_dic)
                    header_json_bytes=header_json.encode('utf-8')
    
    
                    # 二、发送数据
                    # 1、先发送报头的长度
                    header_size=len(header_json_bytes)
                    conn.send(struct.pack('i',header_size))
                    # 2、再发送报头
                    conn.send(header_json_bytes)
                    # 3、最后发送真实的数据
                    with open(r'%s' %file_path,mode='rb') as f:
                        for line in f:
                            conn.send(line)
            except Exception:
                break
        conn.close()
    
    server.close()
    

    client端

    import struct
    import json
    from socket import *
    
    client = socket(AF_INET, SOCK_STREAM)
    # print(client)
    client.connect(('127.0.0.1', 8082))
    
    while True:
        cmd = input(">>: ").strip()  # get 文件路径
        if len(cmd) == 0:
            continue
        client.send(cmd.encode('utf-8'))
    
        # 1、先接收报头的长度
        res=client.recv(4)
        header_size=struct.unpack('i',res)[0]
        # 2、再接收报头
        header_json_bytes=client.recv(header_size)
        header_json=header_json_bytes.decode('utf-8')
        header_dic=json.loads(header_json)
        print(header_dic)
        # 3、最后接收真实的数据
        total_size=header_dic['total_size']
        filename=header_dic['filename']
        recv_size = 0
        with open(r"D:\%s" %filename, mode='wb') as f:
            while recv_size < total_size:
                data = client.recv(1024)
                f.write(data)
                recv_size += len(data)
    
    
    client.close()
    

    相关文章

      网友评论

          本文标题:day32-粘包问题和报头的定制

          本文链接:https://www.haomeiwen.com/subject/ivccjktx.html