Python全栈之路-Day37
1 粘包
1.1 粘包现象
server:
#!/usr/bin/env python
# __Author__: "wanyongzhen"
# Date: 2017/5/3
import socket
# 须知:只有TCP有粘包现象,UDP永远不会粘包
# 所谓粘包问题主要还是因为接收方不知道消息之间的界限,不知道一次性提取多少字节的数据所造成的。
# 此外,发送方引起的粘包是由TCP协议本身造成的,TCP为提高传输效率,发送方往往要收集到足够多的数据后才发送一个TCP段。
# 若连续几次需要send的数据都很少,通常TCP会根据优化算法把这些数据合成一个TCP段后一次发送出去,这样接收方就收到了粘包数据。
phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
phone.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1) # 端口重用
phone.bind(('127.0.0.1',8080))
phone.listen(5)
conn,addr = phone.accept()
# 粘包现象1
# data1 = conn.recv(1024)
# data2 = conn.recv(1024)
# print(data1) # 收到粘包数据 b'helloworldSB'
# print(data2) # 收到空数据 b''
# 粘包现象2(基于客户端的:解决粘包现象1)
data1 = conn.recv(2)
data2 = conn.recv(1024)
print(data1) # 收到两个字节数据 b'he'
print(data2) # 收到粘包数据 b'lloworld' 没收到b'SB'
client:
#!/usr/bin/env python
# __Author__: "wanyongzhen"
# Date: 2017/5/3
import socket
phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 买手机
phone.connect(('127.0.0.1',8080)) # 直接打电话,发起链接
# 以下两条消息被合并成一条数据发送到服务器端
phone.send('helloworld'.encode('utf-8'))
# 解决粘包现象1
# import time
# time.sleep(3) # 通过time.sleep()简单解决粘包问题
phone.send('SB'.encode('utf-8'))
1.2 解决粘包
server:
#!/usr/bin/env python
# __Author__: "wanyongzhen"
# Date: 2017/5/3
import socket # 导入socket模块
import subprocess
# 类似生活中的打电话
phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 买手机
phone.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1) # 端口重用
phone.bind(('127.0.0.1',8080)) # 绑定手机卡 (绑定IP 端口)
phone.listen(5) # 开机 (TCP的半链接池或暂存池大小)
# while True: # 链接循环
# conn,addr = phone.accept() # 等待电话链接
# print('client conn is ',conn) # 打印电话线路
# print('client addr is ',addr) # 打印客户端地址
# while True: # 通信循环
# try: # 应对Windows系统
# cmd = conn.recv(1024) # 收消息 (从缓存中收取消息)(每次收取的消息大小为1024字节)
# if not cmd: break # 应对Linux系统
# res = subprocess.Popen(cmd.decode('utf-8'),
# shell=True,
# stderr=subprocess.PIPE,
# stdout=subprocess.PIPE)
#
# conn.send(res.stderr.read()) # 回消息
# conn.send(res.stdout.read()) # 回消息
# except Exception:
# break
# conn.close() # 挂电话
# phone.close() # 关机
# # 自定义报头解决粘包
# import struct
# while True: # 链接循环
# conn,addr = phone.accept() # 等待电话链接
# print('client conn is ',conn) # 打印电话线路
# print('client addr is ',addr) # 打印客户端地址
# while True: # 通信循环
# try: # 应对Windows系统
# cmd = conn.recv(1024) # 收消息 (从缓存中收取消息)(每次收取的消息大小为1024字节)
# if not cmd: break # 应对Linux系统
# res = subprocess.Popen(cmd.decode('utf-8'),
# shell=True,
# stderr=subprocess.PIPE,
# stdout=subprocess.PIPE)
# out_res = res.stdout.read()
# err_res = res.stderr.read()
# data_size = len(out_res) + len(err_res)
# # 发送报头
# conn.send(struct.pack('i',data_size))
# # 发送数据
# conn.send(out_res) # 回消息
# conn.send(err_res) # 回消息
# except Exception:
# break
# conn.close() # 挂电话
# phone.close() # 关机
# 自定义json报头解决粘包
import struct
import json
while True: # 链接循环
conn,addr = phone.accept() # 等待电话链接
print('client conn is ',conn) # 打印电话线路
print('client addr is ',addr) # 打印客户端地址
while True: # 通信循环
try: # 应对Windows系统
cmd = conn.recv(1024) # 收消息 (从缓存中收取消息)(每次收取的消息大小为1024字节)
if not cmd: break # 应对Linux系统
res = subprocess.Popen(cmd.decode('utf-8'),
shell=True,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE)
out_res = res.stdout.read()
err_res = res.stderr.read()
data_size = len(out_res) + len(err_res)
header_dict = {'length':data_size,'filename':None}
header_json = json.dumps(header_dict)
header_bytes = header_json.encode('utf-8')
header_len = len(header_bytes)
# 发送json header长度
conn.send(struct.pack('i',header_len))
# 发送json header
conn.send(header_bytes)
# 发送数据
conn.send(out_res) # 回消息
conn.send(err_res) # 回消息
except Exception:
break
conn.close() # 挂电话
phone.close() # 关机
client:
#!/usr/bin/env python
# __Author__: "wanyongzhen"
# Date: 2017/5/3
import socket
phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 买手机
phone.connect(('127.0.0.1',8080)) # 直接打电话,发起链接
# while True: # 通信循环
# cmd = input('>>: ').strip() # 输入消息
# if not cmd: continue
# phone.send(cmd.encode('utf-8')) # 发送消息(发送到缓存中,并由操作系统发送到服务端缓存中)
# # 粘包问题:
# # 当命令输出内容很大时(超过1024字节),客户端不能一次性接受完成
# # 就会出现这种情况:在下次接受时会收到上次服务器端发送的命令结果
# # 这就叫粘包
# data = phone.recv(1024) # 接收消息
# print(data.decode('utf-8')) # 打印消息
# phone.close() # 关机
# # 自定义报头解决粘包
# import struct
# while True: # 通信循环
# cmd = input('>>: ').strip() # 输入消息
# if not cmd: continue
# phone.send(cmd.encode('utf-8')) # 发送消息(发送到缓存中,并由操作系统发送到服务端缓存中)
# # 接收报头
# header = phone.recv(4) # 接收消息
# data_size = struct.unpack('i',header)[0]
# # 接收数据
# recv_size = 0
# recv_data = b''
# while recv_size < data_size:
# data = phone.recv(1024)
# recv_size += len(data)
# recv_data += data
# print(recv_data.decode('utf-8')) # 打印消息
# phone.close() # 关机
# 自定义json报头解决粘包
import struct
import json
while True: # 通信循环
cmd = input('>>: ').strip() # 输入消息
if not cmd: continue
phone.send(cmd.encode('utf-8')) # 发送消息(发送到缓存中,并由操作系统发送到服务端缓存中)
# 接收json header长度
header_len_json = phone.recv(4) # 接收消息
json_len = struct.unpack('i',header_len_json)[0]
# 接收json header数据
header_json = phone.recv(json_len)
header_json = header_json.decode('utf-8')
header_dict = json.loads(header_json)
data_size = header_dict['length']
# 接收数据
recv_size = 0
recv_data = b''
while recv_size < data_size:
data = phone.recv(1024)
recv_size += len(data)
recv_data += data
print(recv_data.decode('utf-8')) # 打印消息
phone.close() # 关机