python高级技术(网络编程二)

发布时间 2023-05-07 02:33:32作者: coder雪山

一 粘包现象(基于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段后一次发送出去,这样接收方就收到了粘包数据。

  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时才会清除缓冲区内容。数据是可靠的,但是会粘包。

两种情况下会发生粘包。

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

服务端:

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()