验证客户端链接以及socketserver模块

 1 服务端套接字函数
 2 s.bind()    绑定(主机,端口号)到套接字
 3 s.listen()  开始TCP监听
 4 s.accept()  被动接受TCP客户的连接,(阻塞式)等待连接的到来
 5 
 6 客户端套接字函数
 7 s.connect()     主动初始化TCP服务器连接
 8 s.connect_ex()  connect()函数的扩展版本,出错时返回出错码,而不是抛出异常
 9 
10 公共用途的套接字函数
11 s.recv()            接收TCP数据
12 s.send()            发送TCP数据
13 s.sendall()         发送TCP数据
14 s.recvfrom()        接收UDP数据
15 s.sendto()          发送UDP数据
16 s.getpeername()     连接到当前套接字的远端的地址
17 s.getsockname()     当前套接字的地址
18 s.getsockopt()      返回指定套接字的参数
19 s.setsockopt()      设置指定套接字的参数
20 s.close()           关闭套接字
21 
22 面向锁的套接字方法
23 s.setblocking()     设置套接字的阻塞与非阻塞模式
24 s.settimeout()      设置阻塞套接字操作的超时时间
25 s.gettimeout()      得到阻塞套接字操作的超时时间
26 
27 面向文件的套接字的函数
28 s.fileno()          套接字的文件描述符
29 s.makefile()        创建一个与该套接字相关的文件

 1 #server
 2 import os
 3 import socket
 4 import hmac
 5 def auth(conn):
 6     msg = os.urandom(32)    #生成一个随机字符串
 7     conn.send(msg)              #发送到client端
 8     result = hmac.new(secret_key, msg) #处理这个字符串，得到一个结果
 9     client_digest = conn.recv(1024)      #接收client端处理的结果
10     if result.hexdigest() == client_digest.decode(‘utf-8‘):
11         print(‘合法的连接‘)    #对比成功可以继续通信
12         return True
13     else:
14         print(‘不合法的连接‘)    #不成功  close
15         return False
16 secret_key = b‘wangwang‘
17 sk = socket.socket()
18 sk.bind((‘127.0.0.1‘, 8520))
19 sk.listen()
20 conn, addr = sk.accept()
21 if auth(conn):
22     print(conn.recv(1024))  #可以正常与client端进行通信了
23     conn.close()
24 else:
25     conn.close()
26 sk.close()
27 
28 #client
29 import hmac
30 import socket
31 def auth(sk):
32     msg = sk.recv(32)
33     result = hmac.new(key, msg)
34     res = result.hexdigest()
35     sk.send(res.encode(‘utf-8‘))
36 key =b‘wangwang‘
37 sk = socket.socket()
38 sk.connect((‘127.0.0.1‘, 8520))
39 auth(sk)
40 sk.send(b‘upload‘)    #正常的和server端进行通信
41 sk.close()

 1 #server端
 2 import socketserver
 3 #tcp协议的server端不需要导入socket
 4 class Myserver(socketserver.BaseRequestHandler):
 5     def handle(self):
 6         conn = self.request
 7         while True:
 8             msg = input(‘>>>‘)
 9             conn.send(msg.encode(‘utf-8‘))
10             re_msg = conn.recv(1024)
11             print(re_msg.decode(‘utf-8‘))
12 
13 socketserver.TCPServer.allow_reuse_address = True
14 # 设置allow_reuse_address允许服务器重用地址
15 server = socketserver.ThreadingTCPServer((‘127.0.0.1‘, 8520),Myserver)
16 # 创建一个server, 将服务地址绑定到127.0.0.1:8520
17 server.serve_forever()
18 # 让server永远运行下去，除非强制停止程序
19 
20 #client
21 import socket
22 sk = socket.socket()
23 sk.connect((‘127.0.0.1‘, 8520))
24 while True:
25     ret = sk.recv(1024)
26     print(ret.decode(‘utf-8‘))
27     msg = input(‘>>>‘)
28     sk.send(msg.encode(‘utf-8‘))
29     if msg == ‘q‘:
30         break
31 sk.close()