Python 学习笔记 - socket(2)

前面学习了基本的最原始的单线程的socket的原理，下面学习一个新的知识点-粘包。由于我们接受的命令是recv(1024),那么如果当另一端发送的数据大于1024个字节的时候，他就会出现粘包的问题。每次只能发送1024个字节，如果我们直接放在一个循环里面不断发送，不断接受，那么当数据发完以后，他就会卡住在那里，因为我们知道在正常连接状态里，socket的accept和recv都是会进入阻塞的状态（换句话说，没有客户连接或者客户发空包，那么就会卡住！）。如何处理这个问题呢？一个思路是发送之前，先打个招呼，告诉对方自己要发送的字节长度，这样对方可以根据长度判断什么时候终止接受。

下面看一个模拟SSH操作的实例，客户端发送命令，服务器端执行之后返回结果给客户端

Server.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import socket
import subprocess #导入执行命令模块
ip_port=(‘127.0.0.1‘,9999) #定义元祖
#买手机
s=socket.socket()  #绑定协议，生成套接字
s.bind(ip_port)    #绑定ip+协议+端口：用来唯一标识一个进程，ip_port必须是元组格式
s.listen(5)        #定义最大可以挂起胡链接数

while True:  #用来重复接收新的链接
    conn,addr=s.accept()   #接收客户端胡链接请求，返回conn（相当于一个特定胡链接），addr是客户端ip+port
    #收消息
    while True: #用来基于一个链接重复收发消息
            try: #捕捉客户端异常关闭（ctrl+c）
                recv_data=conn.recv(1024) #收消息，阻塞
                if len(recv_data) == 0:break #客户端如果退出，服务端将收到空消息，退出
                #发消息
                p=subprocess.Popen(str(recv_data,encoding=‘utf8‘),shell=True,stdout=subprocess.PIPE) #执行系统命令，windows平
                                                                                                      # 台命令的标准输出是gbk编码，需要转换
                res=p.stdout.read()   #获取标准输出
                print(res,type(res))
                if len(res) == 0:   #执行错误命令，标准输出为空，
                    send_data=‘cmd err‘
                else:
                    send_data=str(res,encoding=‘gbk‘)  #命令执行ok，字节gbk---->str---->字节utf-8
                send_data=bytes(send_data,encoding=‘utf8‘)
                #解决粘包问题
                ready_tag=‘Ready|%s‘ %len(send_data)
                conn.send(bytes(ready_tag,encoding=‘utf8‘)) #发送数据长度
                feedback=conn.recv(1024)  #接收确认信息
                feedback=str(feedback,encoding=‘utf8‘)
                if feedback.startswith(‘Start‘):
                    conn.send(send_data)  #发送命令的执行结果
            except Exception:
                break
 
    conn.close()

client.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import socket
ip_port=(‘127.0.0.1‘,9999)
s=socket.socket()
s.connect(ip_port)  #链接服务端，如果服务已经存在一个好的连接，那么挂起
while True:        #基于connect建立的连接来循环发送消息
    send_data=input(">>: ").strip()
    if send_data == ‘exit‘:break
    if len(send_data) == 0:continue
    s.send(bytes(send_data,encoding=‘utf8‘))
    #解决粘包问题
    ready_tag=s.recv(1024) #收取带数据长度的字节：Ready|9998
    ready_tag=str(ready_tag,encoding=‘utf8‘)
    if ready_tag.startswith(‘Ready‘):#Ready|9998
        msg_size=int(ready_tag.split(‘|‘)[-1])  #获取待接收数据长度
    start_tag=‘Start‘
    s.send(bytes(start_tag,encoding=‘utf8‘)) #发送确认信息
    #基于已经收到的待接收数据长度，循环接收数据
    recv_size=0
    recv_msg=b‘‘
    while recv_size < msg_size:
        recv_data=s.recv(1024)
        recv_msg+=recv_data
        recv_size+=len(recv_data)
        print(‘MSG SIZE %s RECE SIZE %s‘ %(msg_size,recv_size))
    print(str(recv_msg,encoding=‘utf8‘))
    
s.close()