python从入门到精通-----------验证用户的合法性

sever端

# 使用tcp协议
# 使用os模块先生成一个随机的字符窜
import socket
import os
import hashlib

secret_key = b'hhhh'  # 创建的秘钥
sk = socket.socket()
sk.bind(('192.168.43.31', 9001))  # 申请操作系统的资源
sk.listen()  #

conn, addr = sk.accept()  # conn里储存的是一个客户端和server端的连接信息
    # 创建随机的字符窜
rand = os.urandom(32)
    # 发送随机的字符窜
conn.send(rand)
    # 更据发送的字符窜 + 秘钥 进行摘要
result = hashlib.sha1(secret_key)
result.update(rand)
res = result.hexdigest()
    # 等待摘要的结果
pwd = conn.recv(1024).decode('utf-8')
# 与摘要的结果进行对比
if pwd == res:
    print('客户端合法')
        # 可以发送内容了
    conn.send(b'hellow')

else:
    conn.close()

client端

import socket
import hashlib  # 导入hashlib 模块
secret_key = b'hhhh'  # 创建的秘钥
sk = socket.socket()  #
sk.connect(('192.168.43.31', 9001))

# 接收发送的随机字符窜
rand = sk.recv(32)
# 更据发送的字符窜 + 秘钥 进行摘要
pwd = hashlib.sha1(secret_key)
pwd.update(rand)
res = pwd.hexdigest()
# 将摘要的结果发送回server端
sk.send(res.encode('utf-8'))
# 继续开始和server端通信
msg = sk.recv(1024)
print(msg)