python 验证客户端的合法性

HMAC身份验证在服务器客户端通信中的应用

最新推荐文章于 2024-08-11 18:08:47 发布

转载最新推荐文章于 2024-08-11 18:08:47 发布 · 193 阅读

2 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/wt7018/p/10991819.html

文章标签：

#python

本文介绍了一种使用HMAC模块进行客户端身份验证的方法，通过服务器与客户端之间的随机消息交换和HMAC比较，确保通信双方的合法性。此方法增强了服务器的安全性，防止非法客户端接入。

目的：对连接服务器的客户端进行判断

# Server
import socket
import hmac
import os
secret_key = bytes('tom', encoding='utf-8')
sk = socket.socket()
sk.bind(('127.0.0.1', 8010))
sk.listen()
connect, addr = sk.accept()


def check_client(conn):
    """
    使用hmac模块进行加密
    new()，里面的数据类型要是bytes
    :param conn:
    :return: True or False
    """
    msg = os.urandom(32)    # 使用os模块，随机
    conn.send(msg)
    h = hmac.new(secret_key, msg)
    server_digest = h.digest()
    client_digest = conn.recv(1024)
    return hmac.compare_digest(server_digest, client_digest)


ret = check_client(connect)
if ret:
    print('该客户端是合法的客户端')
    connect.close()
else:
    print('该客户端是不合法的客户端')
    connect.close()
sk.close()

# client
import socket
import hmac
secret_key = bytes('tom', encoding='utf-8')
sk = socket.socket()
sk.connect(('127.0.0.1', 8010))
msg = sk.recv(1024)
h = hmac.new(secret_key, msg)
client_server = h.digest()
sk.send(client_server)
sk.close()