sign_verify.py 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. import ctypes
  2. # 加载共享库
  3. lib = ctypes.CDLL('libgmsign.so')
  4. # 定义函数的参数和返回类型
  5. # 签名函数签名如下:
  6. # int sign(char *key_value, char *hash_value, char *public_key)
  7. lib.sign.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p]
  8. lib.sign.restype = ctypes.c_int
  9. # 验签函数签名如下:
  10. # int sign_verify(char *content, char *sign_value, char *public_key)
  11. lib.sign_verify.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p]
  12. lib.sign_verify.restype = ctypes.c_int
  13. def get_sign(raw_data: str) -> (str, str):
  14. """
  15. 获取签名值
  16. :param raw_data: 原文字符串
  17. :return: tuple(str,str):(签名值base64字符串,公钥base64字符串)
  18. """
  19. key_value = raw_data.replace(" ", "").encode('utf-8')
  20. hash_value = ctypes.create_string_buffer(256) # 设置签名值输出缓冲区大小为 256 字节
  21. public_key = ctypes.create_string_buffer(256) # 设置公钥输出缓冲区大小为 256 字节
  22. result = lib.sign(key_value, hash_value, public_key)
  23. if result == 0:
  24. return hash_value.value.decode(), public_key.value.decode()
  25. else:
  26. return None, None
  27. def verify_sign(raw_data: str, sign_data: str, public_key: str) -> bool:
  28. """
  29. 验证签名
  30. :param raw_data: 原文字符串
  31. :param sign_data: 签名值base64字符串
  32. :param public_key: 公钥base64字符串
  33. :return:
  34. """
  35. content = raw_data.replace(" ", "").encode('utf-8')
  36. sign_value = sign_data.encode('utf-8')
  37. public_key = public_key.encode('utf-8')
  38. verify_result = lib.sign_verify(content, sign_value, public_key)
  39. verify_result = True if verify_result == 1 else False
  40. return verify_result
  41. if __name__ == '__main__':
  42. raw_data = "hello world test sign"
  43. sign_data, public_key = get_sign(raw_data)
  44. print(f"sign_data: {sign_data}\npublic_key: {public_key}")
  45. verify_result = verify_sign(raw_data, sign_data, public_key)
  46. print(f"verify_result: {verify_result}")