Python Socket模块基础通信与粘包处理详解
本文详细解析Python socket模块的基础用法及粘包问题的解决方案。通过对比基础版与防粘包版的Socket SSH实现,展示了服务器端绑定监听、客户端连接交互的核心逻辑。重点介绍了利用命令结果长度预发送及缓冲区确认机制来有效防止TCP粘包现象,确保数据传输的完整性与准确性。
本文深入探讨了Python中socket模块的具体实现方法,通过详细的示例代码解析,为读者在学习或实际工作中提供有价值的参考。
基础Socket SSH通信实现
首先介绍基础的Socket SSH通信实现,该版本未包含防止粘包的机制。
#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com
import socket
import os
server = socket.socket()
server.bind(('localhost', 6969)) #绑定被监听端口
server.listen(5) #监听端口
while True:
print("我要开始等电话了")
conn, addr = server.accept() # 就是等待的意思
#conn就是客户端连过来的时候,在服务器端为其生成的一个连接实例
print("电话来了%s"% [conn, addr])
while True:
data = conn.recv(1024)
if not data:
print('client is lost.')
break
# res = os.popen(data).read() #popen就是打开命令执行,read就是获取结果
# with open('filename', 'r') as ret: #这两行就 可以用过来传输文件了。
# data = ret.read()
print('receive:',data)
conn.send(data.upper())
server.close()
接着展示Socket客户端模块的基本代码结构。
#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com
import socket
client = socket.socket() #声明socket类型,同时生成socket链接对象
client.connect(('localhost',6969)) #localhost就是本机地址
while True:
msg = input('input msg >>:').strip()
if len(msg) == 0: continue #检查msg的信息,防止无输入信息
#client.send(b"Hello, world!") #发送信息
client.send(msg.encode('utf-8'))
data = client.recv(1024) #默认接受1024字节,就是1k
# with open('filename', 'w') as ret: # 这两行就 可以用过来传输文件了。
# ret = data.write()
print(data.decode())
client.close() #关闭端口
解决Socket粘包问题的服务端实现
随后提供包含防止粘包功能的socket_ssh.py服务端代码示例。
#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com
import socket
import os
server = socket.socket()
server.bind(('localhost', 6969)) #绑定被监听端口
server.listen(5) #监听端口
while True:
print("我要开始等电话了")
conn, addr = server.accept() # 就是等待的意思
#conn就是客户端连过来的时候,在服务器端为其生成的一个连接实例
while True:
data = conn.recv(1024).decode()
print("电话来了%s" % type(data))
# if type(data) is str:
# data = data.strip()
if not data:
print('client is lost.')
break
cmd_res = os.popen(data).read() #popen就是打开命令执行,read就是获取结果
cmd_res_size = str(len(cmd_res.encode("utf-8")))
print("before send",len(cmd_res),"size after encode", cmd_res_size)
if len(cmd_res) == 0:
print("there is no output.")
res_warning = "there is no output."
conn.send(res_warning.encode("utf-8"))
continue
else:
conn.send(cmd_res_size.encode("utf8"))
print(conn.recv(1024).decode()) #通过接收数据的形式来强制发送缓冲区的数据,防止粘包。
# with open('filename', 'r') as ret: #这两行就 可以用过来传输文件了。
# data = ret.read()
#print('receive:',data)
print('receive:', data)
conn.send(cmd_res.encode("utf-8"))
# conn.send(bytes(cmd_res)) #不可行。传输的时候是需要encoding
server.close()
配合防粘包机制的客户端实现
最后给出对应的socket_client.py客户端代码示例,用于配合服务端进行完整交互。
#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com
import socket
client = socket.socket() #声明socket类型,同时生成socket链接对象
client.connect(('localhost',6969)) #localhost就是本机地址
while True:
msg = input('input msg >>:').strip()
if len(msg) == 0: continue #检查msg的信息,防止无输入信息
#client.send(b"Hello, world!") #发送信息
client.send(msg.encode('utf-8'))
received_size = client.recv(1024).decode() #用来记录接受的数据大小
print("接收的数据大小", received_size)
received_cont = b''
received_cont_size = 0 # 用来判断接受数据的大小
if received_size != "there is no output." :
client.send("准备好了,可以发送。".encode()) #发送确认信息,以防止粘包
received_size = int(received_size) #数据需要变成int才能进行判断
while received_size != received_cont_size: #判断encode后的长度是否一致。
received_cont_for_test = client.recv(1024)
received_cont_size += int(len(received_cont_for_test))
received_cont = received_cont + received_cont_for_test
print("当前结束后的数据大小为:", received_cont_size)
# print(received_cont_size)
else:
print("数据接收完成,接收的数据大小为:", received_cont_size)
print("接收的内容为:\n",received_cont.decode(),"\n")
else:
print("output:\n", received_size)
# data = client.recv(1024) #默认接受1024字节,就是1k
# with open('filename', 'w') as ret: # 这两行就 可以用过来传输文件了。
# ret = data.write()
# print(data.decode())
# print(str(data))
client.close() #关闭端口
以上内容构成了本文的核心知识点,希望能为读者的学习带来帮助,同时也欢迎读者继续支持本站获取更多技术资源。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









