| 1234567891011121314151617181920212223242526272829303132 |
- """
- 密码标签生成与验证功能
- """
- import json
- from watermark_generate.tools import sign_verify
- def generate_secret_label(raw_data: str):
- """
- 生成密码标签
- :param raw_data: 模型版权信息
- :return: 指定格式密码标签字符串
- """
- sign_data, public_key = sign_verify.get_sign(raw_data)
- # TODO 增加public_key保存公钥的流程
- secret_label = json.dumps({'data': raw_data, 'sign': sign_data, 'public_key': public_key})
- return secret_label
- def verify_secret_label(secret_label: str, public_key: str) -> bool:
- """
- 验证密码标签
- :param secret_label: 生成的密码标签
- :param public_key: 签名公钥
- :return: 密码标签验证结果
- """
- dict_data = json.loads(secret_label)
- raw_data = dict_data.get("data")
- sign_data = dict_data.get("sign")
- verify_result = sign_verify.verify_sign(raw_data, sign_data, public_key)
- return verify_result
|
