Python Socket编程详解：TCP粘包问题与UDP并发实现
本文深入解析Python网络编程中的核心概念，重点阐述TCP协议中粘包问题的成因与基于定长报头的解决方案，对比TCP与UDP在连接建立、数据传输及适用场景上的差异。同时介绍基于UDP协议的Socket编程方法，并演示如何利用socketserver模块实现服务端与多客户端的并发通信，为实际开发提供完整代码示例。
粘包问题
在使用TCP协议进行数据传输时，经常会出现数据粘包的现象。
我们需要先厘清TCP和UDP这两种传输协议的区别。它们都是用于传输数据的底层协议，本身没有绝对的好坏之分，开发者应根据具体的应用需求来选择合适的协议类型。
TCP协议适用于传输大量数据，但建立连接的过程相对复杂，且容易出现粘包问题。不过，粘包问题是可以解决的，关键在于发送端预先告知接收端数据的长度，接收端据此读取确切长度的数据，即可保证一次性完整传输。
UDP协议则适合传输数据量较小的场景。它没有粘包问题，也不需要预先建立连接，每次发送都是独立的数据包，下一次发送即视为新的数据交互。其弊端在于数据传输不安全，存在丢包风险。
以QQ为例，从理论上看，即时通讯软件似乎更适合使用UDP协议以降低延迟，但实际应用中却采用了TCP协议。这主要源于历史遗留问题：早期QQ对单次发送内容的字数进行了限制，这种机制在某种程度上规定了发送与接收的数据长度保持一致，从而规避了部分问题。
形成粘包的原因
第一种情况是发送的两个数据包非常小，且发送间隔时间极短，导致它们在网络底层被合并成一个数据包传输。
第二种情况是单个数据量过大，接收端一次读取不完，剩余部分留在缓冲区中，下一次读取时会将剩余数据与新数据一起取出，从而形成粘包。
解决粘包问题
核心策略是在传输实际数据之前，先发送一个表示数据大小的定长头部信息。
# TCP 解决粘包问题 附带处理了一下
import socket
import struct
import subprocess
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 8000))
server.listen(5)
print('start...')
# 连接循环
while True:
conn, client_address = server.accept()
print(client_address)
# 通信循环
while True:
try:
cmd = conn.recv(1024)
print(cmd)
pipeline = subprocess.Popen(cmd.decode('utf8'),
shell=True,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE)
stdout = pipeline.stdout.read()
stderr = pipeline.stderr.read()
count_len = len(stdout) + len(stderr)
guding_bytes = struct.pack('i', count_len)
conn.send(guding_bytes) # 4
conn.send(stderr + stdout)
except ConnectionResetError: # 连接循环时出现的中断
break
conn.close()
# 客户端
mport struct
from socket import *
client = socket(AF_INET, SOCK_STREAM)
client.connect(('127.0.0.1', 8000))
while True:
# cmd = input('please enter your cmd you want:')
cmd = 'dir'
client.send(cmd.encode('utf8'))
guding_bytes = client.recv(4)
count_len = struct.unpack('i', guding_bytes)[0]
data = client.recv(count_len)
print(data.decode('gbk'))
基于UDP协议的socket套接字编程
UDP协议的特点是无连接。
# UDP服务端
import socket
server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
server.bind(('127.0.0.1', 8000))
print('start...')
while True:
data, client_addr = server.recvfrom(1024)
print(client_addr)
print(data)
server.sendto(data.upper(), client_addr)
# UDP客户端
import socket
client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
while True:
msg = input('please enter your msg:')
client.sendto(msg.encode('utf8'), ('127.0.0.1', 8000))
data = client.recvfrom(1024)
print(data)
其调用功能与TCP相似，但具体方法名的表示有所不同。由于UDP不需要建立连接，因此UDP的sendto方法需要直接指定目标地址，而TCP的send方法则依赖于已建立的连接对象。
基于socketserver实现并发的socket套接字编程
为了实现服务端同时与多个客户端进行连接，我们可以将之前的模型从“一个警局五部电话但只有一人接听”升级为“五部电话各有专人接听”，即通过并发机制处理多个连接。
# 同一时刻有多个人在接听
import socketserver
import subprocess
import struct
class MyHandler(socketserver.BaseRequestHandler):
# 通信循环
def handle(self):
while True:
try:
cmd = self.request.recv(1024)
print(cmd)
pipeline = subprocess.Popen(cmd.decode('utf8'),
shell = True,
stderr = subprocess.PIPE,
stdout = subprocess.PIPE)
stdout = pipeline.stdout.read()
stderr = pipeline.stderr.read()
count_len = len(stderr) + len(stdout)
guding_bytes = struct.pack('i', count_len)
self.request.send(guding_bytes) # 4
self.request.send(stderr + stdout)
except ConnectionResetError:
break
# 使用socketserver的连接循环(并发),但是使用了自己的通信循环
# myhandler = MyHandler()
if __name__ == '__main__':
server = socketserver.ThreadingTCPServer(('127.0.0.1', 8000), MyHandler, bind_and_activate=True)
print('start...')
server.serve_forever()
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持本站。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
