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

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

Server.py

#!/usr/bin/env python# -*- coding:utf-8 -*-import socketimport 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 socketip_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()