Day31 python基础--网络编程基础-socketserver
一,验证客户端合法性
#server端 import os import hmac import socket secret_key = b'alex_sb' def auth(conn): msg = os.urandom(32) #生成一个随机的字符串 conn.send(msg) #发送到client端 result = hmac.new(secret_key,msg) #处理这个随机字符串,得到一个结果 client_digest = conn.recv(1024) #接收client端处理的结果 if result.hexdigest() == client_digest.decode('utf-8'): print('是合法的连接') #对比成功可以继续通信 return True else: print('不合法的连接') #成功close return False sk = socket.socket() sk.bind(('127.0.0.1',9000)) sk.listen() conn,addr = sk.accept() if auth(conn): print(conn.recv(1024)) #正常的和client端进行沟通了 conn.close() else: conn.close() sk.close()
#client端 import hmac import socket key = b'alex_sb' def auth(sk): msg = sk.recv(32) result = hmac.new(key,msg) res = result.hexdigest() sk.send(res.encode('utf-8')) sk = socket.socket() sk.connect(('127.0.0.1',9000)) auth(sk) sk.send(b'upload') # 进行其他正常的和server端的沟通 sk.close()
二,socketserver
import socketserver #tcp协议的server端就不需要导入socket # socketserver:进一步封装 # ↓ # socket 底层模块 class Myserver(socketserver.BaseRequestHandler): def handle(self): conn = self.request while True: conn.send(b'hello') print(conn.recv(1024)) server = socketserver.ThreadingTCPServer(('127.0.0.1',9000),Myserver) server.serve_forever()
#server端 import socket sk = socket.socket() sk.bind(('127.0.0.1',9000)) sk.listen() while True: conn,addr = sk.accept() while True: conn.send(b'hello') print(conn.recv(1024)) #client端 import socket sk = socket.socket() sk.connect(('127.0.0.1',9000)) while True: ret = sk.recv(1024) print(ret) sk.send(b'byebye') sk.close()
三,blocking阻塞
# blocking 阻塞 import socket sk = socket.socket() sk.bind(('127.0.0.1',9000)) sk.setblocking(False) sk.listen() while 1: try: conn,addr = sk.accept() #如果没有人来连接我,我就不在这里等待,非阻塞 print(sk) break except BlockingIOError: pass conn.recv(1024)