一,验证客户端合法性
#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 Trueelse:print('不合法的连接') #成功closereturn Falsesk = 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.requestwhile 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)breakexcept BlockingIOError:pass conn.recv(1024)