python高级技术（网络编程二）

一粘包现象（基于TCP协议实现远程执行命令）

1、TCP协议，会出现粘包现象

例：ipconfig命令，客户端收到的字符串比较短，客户端能够收完整，

　　tasklist命令，客户端收到的字符串超过1024，客户端收到信息不完整，就是粘包问题

服务端：

# 服务端应该满足两个特点

# 1、一直对外提供服务

# 2、并发的服务多个客户端

import subprocess

from socket import *

server = socket(AF_INET, SOCK_STREAM)

server.bind(('127.0.0.1', 8081))

server.listen(5)

# 服务端应该做两件事

# 第一件事：循环地从半链接池中取出链接请求与其建立双向链接，拿到链接对象

while True:

    conn, client_addr = server.accept()

    # 第二件事：拿到链接对象，与其进行通信循环

    while True:

        try:

            cmd = conn.recv(1024)

            if len(cmd) == 0:

                break

            obj = subprocess.Popen(cmd.decode('utf-8'),

                                   shell=True,

                                   stdout=subprocess.PIPE,

                                   stderr=subprocess.PIPE

                                   )

            stdout_res = obj.stdout.read()

            stderr_res = obj.stderr.read()

            print(len(stdout_res)+len(stderr_res))

            print(stderr_res.decode('gbk'))

            print(stdout_res.decode('gbk'))

            conn.send(stdout_res+stderr_res)

            # with open('1.mp4',model='rb') as f:

            #   for line in f:

            #       conn.send(line)

        except Exception:

            break

    conn.close()

客户端：

from socket import *

client = socket(AF_INET, SOCK_STREAM)

client.connect(('127.0.0.1', 8081))

while True:

    msg = input('请输入命令>>>:').strip()

    if len(msg) == 0:

        continue

    client.send(msg.encode('utf-8'))

    # 解决粘包问题思路：

    # 1、拿到数据的总大小total_size

    # 2、recv_size=0,循环接受，每接受一次，recv_size+=接受的长度

    # 3、直到recv_size=total_size,结束循环

    cmd_res = client.recv(1024)  # 本次接受，最大接受1024Bytes

    print(cmd_res.decode('gbk'))  # windows系统用gbk

# 粘包问题出现的原因

# 1、tcp是流式协议，数据像水流一样粘在一起，没有任何边界区分

# 2、收数据没收干净，有残余，就会下一次结果混淆在一起。

# 解决的核心方法就是：每次都收干净，不要任何残留

2、UDP协议，不会出现粘包现象

一个sendto对应一个recvfrom，不会出现粘包的底层原理是，不需要建立三次握手，发信息类似于集装箱

超出的集装箱部分的信息，就扔掉，所以不会出现粘包现象，是不可靠传输。

服务端：

import socket

server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

server.bind(('127.0.0.1', 8080))

res1 = server.recvfrom(1024)

print(res1)

res2 = server.recvfrom(1024)

print(res2)

server.close()

客户端：

import socket

client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

client.sendto(b'hello', ('127.0.0.1', 8080))

client.sendto(b'world', ('127.0.0.1', 8080))

client.close()

# UDP协议一个sendto 对应 一个服务端的recvfrom

二什么是粘包

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

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

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

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

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

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

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

两种情况下会发生粘包。

发送端需要等缓冲区满才发送出去，造成粘包（发送数据时间间隔很短，数据了很小，会合到一起，产生粘包）

服务端：

from socket import *

ip_port=('127.0.0.1',8080)

tcp_socket_server=socket(AF_INET,SOCK_STREAM)

tcp_socket_server.bind(ip_port)

tcp_socket_server.listen(5)

conn,addr=tcp_socket_server.accept()

data1=conn.recv(10)

data2=conn.recv(10)

print('----->',data1.decode('utf-8'))

print('----->',data2.decode('utf-8'))

conn.close()

客户端：

import socket

BUFSIZE=1024

ip_port=('127.0.0.1',8080)

s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

res=s.connect_ex(ip_port)

s.send('hello'.encode('utf-8'))

s.send('feng'.encode('utf-8'))

接收方不及时接收缓冲区的包，造成多个包接收（客户端发送了一段数据，服务端只收了一小部分，服务端下次再收的时候还是从缓冲区拿上次遗留的数据，产生粘包）

服务端：

from socket import *

ip_port=('127.0.0.1',8080)

tcp_socket_server=socket(AF_INET,SOCK_STREAM)

tcp_socket_server.bind(ip_port)

tcp_socket_server.listen(5)

conn,addr=tcp_socket_server.accept()

data1=conn.recv(2) #一次没有收完整

data2=conn.recv(10)#下次收的时候,会先取旧的数据,然后取新的

print('----->',data1.decode('utf-8'))

print('----->',data2.decode('utf-8'))

conn.close()

客户端：

import socket

BUFSIZE=1024

ip_port=('127.0.0.1',8080)

s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

res=s.connect_ex(ip_port)

s.send('hello feng'.encode('utf-8'))

三、解决粘包的方法

1、struct模块

该模块可以把一个类型，如数字，转成固定长度的bytes

struct.pack('i',1111111111111)

2、解决办法

为字节流加上自定义固定长度报头，报头中包含字节流长度，然后一次send到对端，对端在接收时，先从缓存中取出定长的报头，然后再取真实数据

服务端：

import subprocess

import struct

from socket import *

server = socket(AF_INET, SOCK_STREAM)

server.bind(('127.0.0.1', 8081))

server.listen(5)

# 服务端应该做两件事

# 第一件事：循环地从半链接池中取出链接请求与其建立双向链接，拿到链接对象

while True:

    conn, client_addr = server.accept()

    # 第二件事：拿到链接对象，与其进行通信循环

    while True:

        try:

            cmd = conn.recv(1024)

            if len(cmd) == 0:

                break

            obj = subprocess.Popen(cmd.decode('utf-8'),

                                   shell=True,

                                   stdout=subprocess.PIPE,

                                   stderr=subprocess.PIPE

                                   )

            stdout_res = obj.stdout.read()

            stderr_res = obj.stderr.read()

            total_size = len(stdout_res) + len(stderr_res)

            # 1、先发头信息（固定长度的bytes):对数据描述信息

            # int--->固定长度的bytes

            header = struct.pack('i', total_size)

            conn.send(header)

            # 2、再发真实的数据

            conn.send(stdout_res)

            conn.send(stderr_res)

        except Exception:

            break

    conn.close()

客户端：

import struct

from socket import *

client = socket(AF_INET, SOCK_STREAM)

client.connect(('127.0.0.1', 8081))

while True:

    msg = input('请输入命令>>>:').strip()

    if len(msg) == 0:

        continue

    client.send(msg.encode('utf-8'))

    # 解决粘包问题思路：

    # 一、先收固定长度的头：解析出数据的描述信息，包括数据的总大小total_size

    header = client.recv(4)

    total_size = struct.unpack('i', header)[0]  # 解析出来，是个小元组

    # 二、根据解析出的描述信息，接受真实的数据

    # 2、recv_size=0,循环接受，每接受一次，recv_size+=接受的长度

    # 3、直到recv_size=total_size,结束循环

    recv_size = 0

    while recv_size < total_size:

        recv_data = client.recv(1024)  # 本次接受，最大接受1024Bytes

        recv_size += len(recv_data)

        print(recv_data.decode('gbk'), end='')  # windows系统用gbk

    else:

        print()

# 粘包问题出现的原因

# 1、tcp是流式协议，数据像水流一样粘在一起，没有任何边界区分

# 2、收数据没收干净，有残余，就会下一次结果混淆在一起。

# 解决的核心方法就是：每次都收干净，不要任何残留

3、自制报头，报头是字典情况

我们可以把报头做成字典，字典里包含将要发送的真实数据的详细信息，然后json序列化，然后用struck将序列化后的数据长度打包成4个字节（4个自己足够用了）

发送时：

先发报头长度

再编码报头内容然后发送

最后发真实内容

接收时：

先手报头长度，用struct取出来

根据取出的长度收取报头内容，然后解码，反序列化

从反序列化的结果中取出待取数据的详细信息，然后去取真实的数据内容

服务端：

import subprocess

import struct

import json

from socket import *

server = socket(AF_INET, SOCK_STREAM)

server.bind(('127.0.0.1', 8081))

server.listen(5)

# 服务端应该做两件事

# 第一件事：循环地从半链接池中取出链接请求与其建立双向链接，拿到链接对象

while True:

    conn, client_addr = server.accept()

    # 第二件事：拿到链接对象，与其进行通信循环

    while True:

        try:

            cmd = conn.recv(1024)

            if len(cmd) == 0:

                break

            obj = subprocess.Popen(cmd.decode('utf-8'),

                                   shell=True,

                                   stdout=subprocess.PIPE,

                                   stderr=subprocess.PIPE

                                   )

            stdout_res = obj.stdout.read()

            stderr_res = obj.stderr.read()

            total_size = len(stdout_res) + len(stderr_res)

            # 1、制作头

            header_dic = {

                'filename': 'a.txt',

                'total_size': total_size,

                'md5': '12sddeio'

            }

            json_str = json.dumps(header_dic)

            json_str_bytes = json_str.encode('utf-8')

            # 2、先发头信息（固定长度的bytes):对数据描述信息

            # int--->固定长度的bytes

            header_size = struct.pack('i', len(json_str_bytes))

            # 发头的长度信息

            conn.send(header_size)

            # 3、发头信息

            conn.send(json_str_bytes)

            # 4、再发真实的数据

            conn.send(stdout_res)

            conn.send(stderr_res)

        except Exception:

            break

    conn.close()

客户端：

import struct

import json

from socket import *

client = socket(AF_INET, SOCK_STREAM)

client.connect(('127.0.0.1', 8081))

while True:

    msg = input('请输入命令>>>:').strip()

    if len(msg) == 0:

        continue

    client.send(msg.encode('utf-8'))

    # 解决粘包问题思路：

    # 一、先收固定长度的头：解析出数据的描述信息，包括数据的总大小total_size

    header_size = client.recv(4)

    header_len = struct.unpack('i', header_size)[0]

    json_str_bytes = client.recv(header_len)

    json_str = json_str_bytes.decode('utf-8')

    header_dic = json.loads(json_str)

    print(header_dic)

    total_size = header_dic['total_size']

    # 二、根据解析出的描述信息，接受真实的数据

    # 2、recv_size=0,循环接受，每接受一次，recv_size+=接受的长度

    # 3、直到recv_size=total_size,结束循环

    recv_size = 0

    while recv_size < total_size:

        recv_data = client.recv(1024)  # 每次接受，最大接受1024Bytes

        recv_size += len(recv_data)

        print(recv_data.decode('gbk'), end='')  # windows系统用gbk

    else:

        print()

# 粘包问题出现的原因

# 1、tcp是流式协议，数据像水流一样粘在一起，没有任何边界区分

# 2、收数据没收干净，有残余，就会下一次结果混淆在一起。

# 解决的核心方法就是：每次都收干净，不要任何残留

四、socketserver实现并发

1、TCP协议的并发

服务端：

import socketserver

class MyRequestHandle(socketserver.BaseRequestHandler):

    def handle(self):

        print(self.request)  # 如果TCP协议，self.request--->conn

        print(self.client_address)

        while True:

            try:

                msg = self.request.recv(1024)

                if len(msg) == 0:

                    break

                self.request.send(msg.upper())

            except Exception:

                break

        self.request.close()

# 服务端应该做两件事

# 第一件事：循环地从半链接池中取出链接请求与其建立双向链接，拿到链接对象

s = socketserver.ThreadingTCPServer(('127.0.0.1', 8080), MyRequestHandle)

s.serve_forever()

# 等同于

# while True:

#     conn, client_addr = server.accept()

# 第二件事：拿到链接对象，与其进行通信循环----》handle方法

客户端：

from socket import *

client = socket(AF_INET, SOCK_STREAM)

client.connect(('127.0.0.1', 8080))

while True:

    msg = input('请输入命令>>>:').strip()

    if len(msg) == 0:

        continue

    client.send(msg.encode('utf-8'))

    res = client.recv(1024)

    print(res.decode('utf-8'))

2、UDP协议

服务端：

import socketserver

class MyRequesthandle(socketserver.BaseRequestHandler):

    def handle(self):

        client_data = self.request[0]

        server = self.request[1]

        client_address = self.client_address

        print('客户端发来的数据{}'.format(client_data))

        server.sendto(client_data.upper(), client_address)

s = socketserver.ThreadingUDPServer(('127.0.0.1', 8080), MyRequesthandle)

s.serve_forever()

# 相当于：只负责循环的收

# while True:

#     data, server_addr = client.recvfrom(1024)

#     启动一个线程处理后续的事情（data,client_addr)

客户端：

import socket

client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  # 数据报协议---》UDP协议

while True:

    msg = input('>>>:').strip()

    client.sendto(msg.encode('utf-8'), ('127.0.0.1', 8080))

    data, server_addr = client.recvfrom(1024)

    print(data.decode('utf-8'))

client.close()