3.2.0

sop-sdk/sdk-python/common/SignUtil.py

@@ -0,0 +1,102 @@
+#!/usr/bin/python
+# -*- coding: UTF-8 -*-
+import rsa
+import base64
+
+__pem_begin = '-----BEGIN RSA PRIVATE KEY-----\n'
+__pem_end = '\n-----END RSA PRIVATE KEY-----'
+
+
+def create_sign(all_params, private_key, sign_type):
+    """创建签名
+
+    :param all_params: 参数
+    :type all_params: dict
+
+    :param private_key: 私钥字符串
+    :type private_key: str
+
+    :param sign_type: 签名类型，'RSA', 'RSA2'二选一
+    :type sign_type: str
+
+    :return: 返回签名内容
+    :rtype: str
+    """
+    sign_content = get_sign_content(all_params)
+    private_key = _format_private_key(private_key)
+    return sign(sign_content, private_key, sign_type)
+
+
+def _format_private_key(private_key):
+    if not private_key.startswith(__pem_begin):
+        private_key = __pem_begin + private_key
+    if not private_key.endswith(__pem_end):
+        private_key = private_key + __pem_end
+    return private_key
+
+
+def get_sign_content(params):
+    """构建签名内容
+
+    1.筛选并排序
+    获取所有请求参数，不包括字节类型参数，如文件、字节流，剔除sign字段，剔除值为空的参数，并按照参数名ASCII码递增排序（字母升序排序），
+    如果遇到相同字符则按照第二个字符的键值ASCII码递增排序，以此类推。
+
+    2.拼接
+    将排序后的参数与其对应值，组合成“参数=参数值”的格式，并且把这些参数用&字符连接起来，此时生成的字符串为待签名字符串。
+
+    :param params: 参数
+    :type params: dict
+
+    :return: 返回签名内容
+    :rtype: str
+    """
+    keys = params.keys()
+    keys.sort()
+    result = []
+    for key in keys:
+        value = str(params.get(key))
+        if len(value) > 0:
+            result.append(key + '=' + value)
+
+    return '&'.join(result)
+
+
+def sign(content, private_key, sign_type):
+    """签名
+
+    :param content: 签名内容
+    :type content: str
+
+    :param private_key: 私钥字符串
+    :type private_key: str
+
+    :param sign_type: 签名类型，'RSA', 'RSA2'二选一
+    :type sign_type: str
+
+    :return: 返回签名内容
+    :rtype: str
+    """
+    if sign_type.upper() == 'RSA':
+        return rsa_sign(content, private_key, 'SHA-1')
+    elif sign_type.upper() == 'RSA2':
+        return rsa_sign(content, private_key, 'SHA-256')
+    else:
+        raise Exception('sign_type错误')
+
+
+def rsa_sign(content, private_key, hash):
+    """SHAWithRSA
+
+    :param content: 签名内容
+    :type content: str
+
+    :param private_key: 私钥
+    :type private_key: str
+
+    :return: 签名内容
+    :rtype: str
+    """
+    pri_key = rsa.PrivateKey.load_pkcs1(private_key.encode('utf-8'))
+    sign_result = rsa.sign(content, pri_key, hash)
+    return base64.b64encode(sign_result)