Python网络编程 —— 验证客户端的合法性

1、验证客户端的合法性

登录：只要有个性化设计的时候就需要登录

登录和合法性验证二选一，如果做登录的功能就不需要做合法性验证

server

import os
import socket
import hashlib
SECRET_KEY = b"yongliang"       # 设置秘钥

def check_client(conn):
    randbytes = os.urandom(32)   # 随机产生一个32位的随机字节
    conn.send(randbytes)         # 将随机字节发送给client

    # 服务端对随机字节进行hashlib动态加盐（秘钥）加密
    md5 = hashlib.md5(SECRET_KEY)
    md5.update(randbytes)
    code = md5.hexdigest()
    code_cli = conn.recv(32).decode("utf-8")   # 接收client发来的加密后的字符串
    return code == code_cli  # 进行客户端验证 返回bool值

sk = socket.socket()
sk.bind(("127.0.0.1",9001))
sk.listen()
while True:
    conn,addr = sk.accept()
    if not check_client(conn):continue  # 验证判断 如果得到False -->continue
    print("进程正常的通信了")     # 验证通过 执行其他程序

client

import os
import socket
import hashlib
SECRET_KEY = b"yongliang"

def check_client():
    randbytes = sk.recv(32)        # 接收服务端发送的32位的随机字节
    
    # 客户端对随机字节进行hashlib动态加盐（秘钥）加密
    md5 = hashlib.md5(SECRET_KEY)
    md5.update(randbytes)
    code = md5.hexdigest().encode("utf-8")
    sk.send(code)    # 将加密后的字符串发送到server
    
sk = socket.socket()
sk.connect(("127.0.0.1",9001))
check_client()
print("正常的客户端通信")    # 验证通过  则执行其他程序

明文进行加密时也可以用 hmac 模块

import os
import hmac                # 在加密内容基础上增加盐
SECRET_KEY = b"yongliang" 
randbytes = os.urandom(32)     # 随机产生一个32位的字节
mac = hmac.new(SECRET_KEY,randbytes)   # 对随机字节进行动态加密
ret = mac.digest()
print(ret)