socketserver 模块
# 使TCP协议支持并发操作 # 服务端 import socketserver class MyServar(socketserver.BaseRequestHandler): # 必须继承 BaseRequestHandler 类 def handle(self): # 必须定义handle函数. """ 收发数据的逻辑 :return: """ conn = self.request # 获取客户端消息 cli_addr = self.client_address # 获取客服端 ip+端口 while True: # 循环 # 接收消息 msg = conn.recv(1024) # 收发数据的逻辑 print(msg.decode()) # 发送消息 data = input("发送消息") conn.send(data.encode()) servar = socketserver.ThreadingTCPServer(("127.0.0.1",9006),MyServar) # 相当于创建,绑定, servar.serve_forever() # While 循环一直监听是否有客户端请求到达 ... # 客户端 import socket sk = socket.socket() sk.connect(("127.0.0.1",9006)) while True: # 发送消息 data = input("发送消息") sk.send(data.encode()) # 接收消息 msg = sk.recv(1024) print(msg.decode()) sk.close()
hashlib & hmac 加密算法模块
# hashlib 加密算法模块 import hashlib hm = hashlib.md5("加盐") # 创建md5对象并加盐 hm.updata("加密内容".encode()) # 更新内容进行算法运算 res = hm.hexdigest() # 拿到运算后的32位16进制数 # hmac 加密算法模块 import hmac hn = hmac.new("加盐".encode(),"数据".encode()) res = hn.hexdigest() print(res)
动态加密
import hashlib,os # rangdom也可以 salt = os.urandom(10000) # 获取长度为10000的随机字节流 hm = hashlib.md5(salt) hm.update("我爱你".encode()) res_hm = hm.hexdigest() print(res_hm)
文件校验
import hashlib,hmac # 针对小文件一次性读出来 def hashmd5_file(hm,filename): with open(filename, "rb") as f: data = f.read() hm.update(data) return hm.hexdigest() obj1 = hashlib.md5() obj2 = hashlib.md5() hm_ceshi1 = hashmd5_file(obj1,"ceshi1.txt") hm_ceshi2 = hashmd5_file(obj2,"ceshi2.txt") if hm_ceshi1 == hm_ceshi2: print("校验合法") else: print("不合法") # 针对大文件分批进行 def hmac_file(hn,file_name): numsize = os.path.getsize(file_name) with open(file_name, "rb") as f: while numsize: data = f.read(128) hn.update(data) numsize -= len(data) return hn.hexdigest() hn1 = hmac.new("盐".encode()) hn2 = hmac.new("盐".encode()) ceshi1 = hmac_file(hn1,"ceshi1.txt") ceshi2 = hmac_file(hn2,"ceshi2.txt") if ceshi1 == ceshi2: print("校验合法") else: print("不合法")
服务器校验
# 模拟支付宝服务端 import socketserver,os,hmac class MyServer(socketserver.BaseRequestHandler): miyu = "夜袭寡妇村" # 密语 def handle(self): conn = self.request cil_addr = self.client_address auth_code = os.urandom(128) conn.send(auth_code) cli_hmac = conn.recv(32).decode() hn = hmac.new(auth_code, MyServer.miyu.encode()) servar_hmac = hn.hexdigest() if cli_hmac == servar_hmac: conn.send("支付成功~".encode()) else: conn.send("滚~".encode()) servar = socketserver.ThreadingTCPServer(("127.0.0.1",3452),MyServer) servar.serve_forever() # 模拟合作服务端 import socket,hmac # 密语 miyu = "夜袭寡妇村" sk = socket.socket() sk.connect(("127.0.0.1",3452)) auth_code = sk.recv(128) hn = hmac.new(auth_code,miyu.encode()) msg = hn.hexdigest() sk.send(msg.encode()) print(sk.recv(1024).decode()) sk.close()