一,驗證客戶端合法性
#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)
?