socketserver模块实现并发和连接合法性验证
程序员文章站
2022-06-30 13:54:47
一.socketserver模块 1.sockeserver的源码流程 2.简单的使用 socketserver服务端 socket客户端 二.连接的合法性验证 1.os.urandom(n)加密 os.urandom(n)是一种bytes类型的随机生成n个字节字符串,而且每次生成的值都不相同,再加 ......
一.socketserver模块
2.简单的使用
socketserver服务端
import socketserver
class myserver(socketserver.baserequesthandler):
def handle(self):
msg = self.request.recv(1024)
print(msg)
if __name__ == '__main__':
server = socketserver.threadingtcpserver(("127.0.0.1",8888),myserver) # 实例化对象
server.serve_forever()
socket客户端
import socket
client = socket.socket()
ip_port = ("127.0.0.1",8888)
client.connect(ip_port)
client.send(b"hello")
二.连接的合法性验证
1.os.urandom(n)加密
os.urandom(n)是一种bytes类型的随机生成n个字节字符串,而且每次生成的值都不相同,再加上md5等加密处理,九惩了内容不同长度相同的字符串了
2.hmac
python自带的hmac模块实现了标准的hmac算法
import hmac
message = b'hello world' # 加密信息
key = b'secret' # 加密密匙
h = hmac.new(key,message,digestmod='md5') # 密匙,信息,模式
print(h.hexdigest())
比较两个密文是否相同,可以用hmac.compare_digest(密文、密文),然会true或者false。
3.连接合法性验证服务端和客户端
服务端
import socket import hmac , os secret_key = b"this is key!" # 只做验证 返回bool def conn_auth(conn): print("连接认证") msg = os.urandom(32) conn.send(msg) h = hmac.new(secret_key,msg) server_key = h.digest() client_key = conn.recv(32) # print(server_key,client_key) return hmac.compare_digest(server_key,client_key) # 判断验证 def data(conn): if conn_auth(conn): print("验证成功!") conn.send("验证成功".encode("utf-8")) while 1: data = conn.recv(1024) print(data) else: print("验证失败") conn.send("验证失败".encode("utf-8")) conn.close() return # 启动 def tcp_server(ip_port,): server = socket.socket() server.bind(ip_port) server.listen() while 1: conn,addr = server.accept() print(f"新连接: {addr[0]}:{addr[1]}") data(conn) if __name__ == '__main__': ip_port = ("127.0.0.1",8888) tcp_server(ip_port)
客户端
import socket import os, hmac # 密匙 sercet_key = b"this is key!" # 使用密匙验证连接是否合法 def conn_auth(client): msg = client.recv(32) h = hmac.new(sercet_key, msg) client_key = h.digest() client.send(client_key) # 验证判断 def tcp_client(ip_port): client = socket.socket() client.connect(ip_port) conn_auth(client) verify = client.recv(12).decode("utf-8") while verify == "验证成功": data = input("给服务端发送:").encode("utf-8") client.send(data) else: print("连接被拒绝") client.close() if __name__ == '__main__': ip_port = ("127.0.0.1", 8888) tcp_client(ip_port)
上一篇: CSS实现文字两端对齐
下一篇: golang环境搭建