python socket聊天室(tcp server、client)

python socket聊天室(tcp server、client),第1张

概述importsocket,os,hmac,pickle,threading,datetimeclassServer:def__init__(self,secret_key):self.sock=socket.socket(socket.AF_INET,socket.SOCK_STREAM)self.addr=('127.0.0.1',8088)self.users={}

import socket,os,hmac,pickle,threading,datetimeclass Server: def __init__(self,secret_key): self.sock = socket.socket(socket.AF_INET, socket.soCK_STREAM) self.addr = ('127.0.0.1', 8088) self.users = {} self.secret_key = secret_key def start_server(self): try: self.sock.bind(self.addr) except Exception as e: print(e) self.sock.Listen(5) print("服务器已开启,ip地址为{}等待连接...".format(self.addr)) self.accept_cont() def __auth(self,sock,addr,secret_key): print('来自{}的客户端发起连接,开始验证客户端合法性...'.format(addr)) msg = os.urandom(32) sock.sendall(msg) h = hmac.new(secret_key, msg, digestmod='md5') digest = h.digest() response = sock.recv(len(digest)) return hmac.compare_digest(digest, response) def accept_cont(self): while True: sock,addr = self.sock.accept() if not self.__auth(sock,addr,self.secret_key): print('来自{}的客户端验证失败,关闭连接!'.format(addr)) sock.close() else: print('来自{}的客户端验证通过!'.format(addr)) self.users[addr] = sock number = len(self.users) print("用户{}连接成功!现在共有{}位用户".format(addr, number)) threading.Thread(target=self.recv_send, args=(sock, addr)).start() def recv_send(self,sock,addr): while True: try: response = pickle.loads(sock.recv(4096)) msg = "{}用户{}发来消息:{}".format(self.get_time(), addr, response) for clIEnt in self.users.values(): clIEnt.send(pickle.dumps(msg)) except ConnectionresetError: print("用户{}已经退出!".format(addr)) self.users[addr].close() self.users.pop(addr) msg = "{}用户{}已经退出!".format(self.get_time(), addr) for clIEnt in self.users.values(): clIEnt.send(pickle.dumps(msg)) break def close_server(self): for clIEnt in self.users.values(): clIEnt.close() self.sock.close() os._exit(0) def get_time(self): Now = datetime.datetime.Now() send_time = Now.strftime("%Y-%m-%d %H:%M:%s") return send_time def run(self): threading.Thread(target=self.start_server).start() while True: cmd = input("input 'exit' to stop sever! >>>") if cmd == "exit": self.close_server() else: print("输入命令无效,请重新输入!")secret_key = b'I have a red phone!'server = Server(secret_key)server.run()

 

客户端:

import socket,hmac,pickle,threading,osclass ClIEnt:    """"""    def __init__(self,ip_port,secret_key):        self.ip_port = ip_port        self.secret_key = secret_key        self.clIEnt = self.conn_server()        self.active = True        def conn_server(self):        clIEnt = socket.socket(socket.AF_INET,socket.soCK_STREAM)        clIEnt.setsockopt(socket.soL_SOCKET,socket.so_KEEPAliVE,True)        err = clIEnt.connect_ex(self.ip_port)        if err != 0:            print("please check sever's ip_port!")            return None        msg = clIEnt.recv(32)        h = hmac.new(self.secret_key, msg, digestmod='md5')        digest = h.digest()        clIEnt.sendall(digest)        return clIEnt        def send_msg(self):        def send_msg2(clIEnt):            msg = input('input you words:>>>').strip()            if msg in ['quit','exit']:                print('you just closed the connection.')                clIEnt.close()                self.active = False            else:                clIEnt.sendall(pickle.dumps(msg))        while self.active:            send_msg2(self.clIEnt)    def recv_msg(self):        while True:            try:                data = self.clIEnt.recv(1024)            except Exception as err:                print('you just lost the connection.err:',err)                self.clIEnt.close()                break            print('recv msg:',pickle.loads(data))        os._exit(0)    def run(self):        threading.Thread(target=self.recv_msg).start()        self.send_msg()ip_port = ('127.0.0.1',8088)secret_key = b'I have a red phone!'clIEnt0 = ClIEnt(ip_port,secret_key)clIEnt0.run()

 

总结

以上是内存溢出为你收集整理的python socket聊天室(tcp server、client)全部内容,希望文章能够帮你解决python socket聊天室(tcp server、client)所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/1159530.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-06-01
下一篇 2022-06-01

发表评论

登录后才能评论

评论列表(0条)

保存