Python基于socketserver实现并发及客户端合法性验证
本文详解如何利用Python的socketserver模块实现多客户端并发连接,并通过hmac结合加盐机制验证客户端身份。内容涵盖ThreadingTCPServer的使用、随机字符串加密交互流程及代码实现,帮助开发者在分布式系统中构建轻量级且安全的连接认证方案。
一、socketserver实现并发
原生TCP协议的Socket仅支持单一客户端通信,而socketserver模块在此基础上进行了封装,允许服务器同时处理多个客户端连接,其底层依然调用Socket接口。
Socket作为底层通信模块,负责具体的网络数据传输。
socketserver则是基于Socket构建的高级封装库,简化了并发处理的复杂性。
socketserver的标准代码结构如下:
服务端实现示例:
import socketserver # 引入模块
import time
# 类名随便定义,但是必须继承socketserver.BaseRequestHandler此类。
class Myserver(socketserver.BaseRequestHandler):
def handle(self): # handle方法,固定名字
conn = self.request # conn相当于每个客户端连接过来的,客户端信息
'''
以下就是基于conn客户端信息,对每个客户端链接到服务端必须做的操作
conn——>等于客户端的连接信息
以此实现并发连接
'''
while True:
try:
content = conn.recv(1024).decode('utf-8')
conn.send(content.upper().encode('utf-8'))
time.sleep(1)
except ConnectionResetError:
break
# 对 socketserver.ThreadingTCPServer 类实例化对象,将IP地址,端口号,以及自己定义的类名传入,并返回一个对象
server = socketserver.ThreadingTCPServer(('127.0.0.1', 4444), Myserver)
server.serve_forever() # 执行对象server_forever方法,开启服务端
客户端连接示例:
import socket
sk = socket.socket()
sk.connect(('127.0.0.1', 4444))
while True:
sk.send(b'heelo')
content = sk.recv(1024).decode('utf-8')
print(content)
二、验证客户端合法性
在分布式系统中,若需实现轻量级的客户端链接认证,无需采用复杂的SSL协议,可利用hmac算法配合加盐策略来完成身份验证。
验证逻辑思路:
服务端生成随机字符串,并使用客户端与服务端共享的秘钥对该字符串进行加密;客户端接收该随机字符串后,使用相同秘钥加密并回传;服务端接收回传的加密串,并与自己计算的加密结果比对,一致则判定为合法客户端,否则立即断开连接。
具体实现中需使用hashlib模块:
服务端代码示例:
import socket
import hashlib
import os
# 设置双方都拥有的秘钥
secret = '秘钥'.encode('utf-8')
sk = socket.socket()
sk.bind(('127.0.0.1', 4444))
sk.listen()
conn, _ = sk.accept()
# 给服务端发送32位随机字节串
count = os.urandom(32)
conn.send(count)
# 并对字节串进行加盐加密
hs = hashlib.sha1(secret)
hs.update(count)
info_server = hs.hexdigest()
# 接收客户端发送的对字节串加密后的信息
info_client = conn.recv(1024).decode('utf-8')
# 客户端加密后的信息和服务端加密后的信息做对比
if info_server == info_client:
# 相等则连接合法,继续操作
print('合法连接')
else:
# 不相等则断开链接
print('非合法用户')
conn.close()
客户端代码示例:
import socket
import hashlib
# 双方共有的秘钥
secret = '秘钥'.encode('utf-8')
sk = socket.socket()
sk.connect(('127.0.0.1', 4444))
# 接收服务端发送的随机字节
count = sk.recv(1024)
# 对随机字节进行加密
hs = hashlib.sha1(secret)
hs.update(count)
info = hs.hexdigest().encode('utf-8')
# 把加密后的字节发送回服务端做判断
sk.send(info)
以上为Python基于socketserver实现并发及验证客户端合法性的完整流程,更多相关技术细节可参考本站其他文章。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













