python学习笔记day33socketserver模块解决TCP不能同时跟多个client同时通信的问题...

# server.py
import socketserver
class Myserver(socketserver.BaseRequestHandler):def handle(self):while True:ret&＃61;self.request.recv(1024).decode("utf-8")print(ret)info&＃61;input(">>>")self.request.send(bytes(info.encode("utf-8")))if __name__&＃61;&＃61;"__main__":server&＃61;socketserver.ThreadingTCPServer(("127.0.0.1",8080),Myserver)server.serve_forever()

# client.py
import socket
sk&＃61;socket.socket()
sk.connect(("127.0.0.1",8080))
while True:info&＃61;input(">>>")sk.send(bytes(("来自璇璇的消息:"&＃43;info).encode("utf-8")))ret&＃61;sk.recv(1024).decode("utf-8")print(ret)sk.close()

# client2.py
import socket
sk&＃61;socket.socket()
sk.connect(("127.0.0.1",8080))
while True:info&＃61;input(">>>")sk.send(bytes((&＃39;来自嘻嘻的消息:&＃39;&＃43;info).encode("utf-8")))ret&＃61;sk.recv(1024).decode("utf-8")print(ret)
sk.close()

# server.py
import os
import socketserver # 由于TCP起的server同一时间不能跟多个客户端通信&＃xff0c;socketserver模块刚好解决了该问题
import hmac # server端和合法的客户端有相同的密钥&＃xff0c;server端给客户端发送要加密的bytes类型# 客户端也使用hmac进行加密&＃xff0c;将结果返回给server端&＃xff0c;server端比较两个加密的结果
secret_key&＃61;b"xuanxuan" # server和合法的客户端之间的密钥&＃xff08;约定好的&＃xff0c;双方都已知的&＃xff09;class MyServer(socketserver.BaseRequestHandler): # 自己定义的类Myserver必须要继承BaseRequestHandlerdef handle(self): # 该类中必须要定义的方法msg&＃61;os.urandom(32) # 随机生成一个32位的字节&＃xff0c;用于发送给client端&＃xff0c;进行验证&＃xff08;配合密钥&＃xff09;self.request.send(msg) # self.request相当于socket中的conn表示一个连接h&＃61;hmac.new(secret_key,msg) # 获得需要加密的对象 一个是双方约定好的密钥 &＃xff0c;另一个是随机生成的bytes类型的需要加密的信息digest&＃61;h.digest() # 加密client_digest&＃61;self.request.recv(1024)ret&＃61;hmac.compare_digest(digest,client_digest)if ret:while True:ret&＃61;self.request.recv(1024).decode("utf-8")print(ret)info&＃61;input(">>>")self.request.send(bytes(info.encode("utf-8")))else:print("不合法的客户端&＃xff0c;我要断开连接")if __name__&＃61;&＃61;"__main__":server&＃61;socketserver.ThreadingTCPServer(("127.0.0.1",8080),MyServer) # 必须要传两个参数 IP地址 端口号&＃xff0c;还有上面定义的类名server.serve_forever() # 永久起一个服务

# client.py
import socket
import hmac
secret_key&＃61;b"xuanxuan" # 跟server端的密钥一样&＃xff0c;是合法的客户端
sk&＃61;socket.socket()
sk.connect(("127.0.0.1",8080))
msg&＃61;sk.recv(1024) # hmac.new()操作的对象都是bytes类型的&＃xff0c;所以先不解码啦

h&＃61;hmac.new(secret_key,msg)
client_digest&＃61;h.digest()
sk.send(client_digest) # 把加密后的给server端发过去
while True:info&＃61;input(">>>")sk.send(bytes(("来自客户端1的消息:"&＃43;info).encode("utf-8")))ret&＃61;sk.recv(1024).decode("utf-8")print(ret)
sk.close()

# client2.py
import socket
import hmac
secret_key&＃61;b"xuan" # 跟server端的密钥不一样&＃xff0c;不是合法的客户端
sk&＃61;socket.socket()
sk.connect(("127.0.0.1",8080))
msg&＃61;sk.recv(1024) # hmac.new()操作的对象都是bytes类型的&＃xff0c;所以先不解码啦
h&＃61;hmac.new(secret_key,msg)
client_digest&＃61;h.digest()
sk.send(client_digest) # 把加密后的给server端发过去
while True:info&＃61;input(">>>")sk.send(bytes(("来自客户端2的消息:"&＃43;info).encode("utf-8")))ret&＃61;sk.recv(1024).decode("utf-8")print(ret)
sk.close()