前面学习了基本的最原始的单线程的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()
转载于:https://blog.51cto.com/beanxyz/1860016