python基于socketserver实现并发,验证客户端的合法性

时间:2022-12-01 12:38:30

一、socketserver实现并发

tcp协议的socket是只能和一个客户端通信的,使用socketserver可以实现和多个客户端通信,他是在socket的基础上进行的封装,底层还是调用的socket。

socket是底层模块

socketserver是基于socket完成的

socketserver代码格式:

服务端:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import socketserver     # 引入模块
import time
# 类名随便定义,但是必须继承socketserver.BaseRequestHandler此类。
class Myserver(socketserver.BaseRequestHandler):
    def handle(self):           # handle方法,固定名字
        conn = self.request     # conn相当于每个客户端连接过来的,客户端信息
        '''
        以下就是基于conn客户端信息,对每个客户端链接到服务端必须做的操作
        conn——>等于客户端的连接信息
        以此实现并发连接
        '''
        while True:
            try:
                content = conn.recv(1024).decode('utf-8')
                conn.send(content.upper().encode('utf-8'))
                time.sleep(1)
            except ConnectionResetError:
                break
# 对 socketserver.ThreadingTCPServer 类实例化对象,将IP地址,端口号,以及自己定义的类名传入,并返回一个对象
server = socketserver.ThreadingTCPServer(('127.0.0.1', 4444), Myserver)
server.serve_forever()      # 执行对象server_forever方法,开启服务端

客户端:

?
1
2
3
4
5
6
7
8
9
import socket
sk = socket.socket()
sk.connect(('127.0.0.1', 4444))
while True:
    sk.send(b'heelo')
    content = sk.recv(1024).decode('utf-8')
    print(content)

二、验证客户端合法性

在分布式系统中实现一个简单的客户端链接认证功能,又不像SSL那么复杂,那么利用hmac+加盐的方法实现。

思路:

服务端发送随机字符串(并用客户端和服务端都知道的秘钥对随机字符串进行加密)——>客户端接收到服务端发送的随机字符串(并用相同的秘钥进行加密发送回服务端)——>服务端接收到客户端发送的加密字符串和自己加密的字符串做对比(如果一致,就是合法客户端,不一致,就立即关闭)

使用hashlib方法:

服务端

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import socket
import hashlib
import os
# 设置双方都拥有的秘钥
secret = '秘钥'.encode('utf-8')
sk = socket.socket()
sk.bind(('127.0.0.1', 4444))
sk.listen()
conn, _ = sk.accept()
# 给服务端发送32位随机字节串
count = os.urandom(32)
conn.send(count)
# 并对字节串进行加盐加密
hs = hashlib.sha1(secret)
hs.update(count)
info_server = hs.hexdigest()
# 接收客户端发送的对字节串加密后的信息
info_client = conn.recv(1024).decode('utf-8')
# 客户端加密后的信息和服务端加密后的信息做对比
if info_server == info_client:
    # 相等则连接合法,继续操作
    print('合法连接')
else:
    # 不相等则断开链接
    print('非合法用户')
    conn.close()

客户端

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import socket
import hashlib
# 双方共有的秘钥
secret = '秘钥'.encode('utf-8')
sk = socket.socket()
sk.connect(('127.0.0.1', 4444))
# 接收服务端发送的随机字节
count = sk.recv(1024)
# 对随机字节进行加密
hs = hashlib.sha1(secret)
hs.update(count)
info = hs.hexdigest().encode('utf-8')
# 把加密后的字节发送回服务端做判断
sk.send(info)

以上就是python基于socketserver实现并发,验证客户端的合法性的详细内容,更多关于python socketserver实现并发的资料请关注服务器之家其它相关文章!

原文链接:https://www.cnblogs.com/XiaoYang-sir/p/14767541.html