tongsuo-project / tongsuo-python-sdk Goto Github PK

def sm2_demo():
    msg = b"message"
    key = ec.generate_private_key(ec.SM2()).private_numbers().private_key()

    # 验证 key.public_key 签名 & 验签
    signature = key.sign(msg, ec.ECDSA(hashes.SM3()))
    key.public_key().verify(signature, msg, ec.ECDSA(hashes.SM3()))

    # 将公钥导出为文本
    pem_public_key_str = key.public_key().public_bytes(
        encoding=serialization.Encoding.PEM, format=serialization.PublicFormat.SubjectPublicKeyInfo
    ).decode()

    # 通过文本加载公钥
    pubkey_from_text = serialization.load_pem_public_key(pem_public_key_str.encode())

    # 验证公钥等价
    assert pubkey_from_text.public_numbers() == key.public_key().public_numbers()
    assert pubkey_from_text.public_bytes(
        encoding=serialization.Encoding.PEM, format=serialization.PublicFormat.SubjectPublicKeyInfo
    ) == key.public_key().public_bytes(
        encoding=serialization.Encoding.PEM, format=serialization.PublicFormat.SubjectPublicKeyInfo
    )
    # 验签
    pubkey_from_text.verify(signature, msg, ec.ECDSA(hashes.SM3()))

from tongsuopy.crypto import hashes
from tongsuopy.crypto.asymciphers import ec


msg = b"message digest"
d = b"3945208F7B2144B13F36E38AC6D39F95889393692860B51A42FB81EF4DF7C5B8"
Qx = b"09F9DF311E5421A150DD7D161E4BC5C672179FAD1833FC076BB08FF356F35020"
Qy = b"CCEA490CE26775A52DC6EA718CC1AA600AED05FBF35E084A6632F6072DA9AD13"

key = ec.EllipticCurvePrivateNumbers(
    int(d, 16),
    ec.EllipticCurvePublicNumbers(int(Qx, 16), int(Qy, 16), ec.SM2()),
).private_key()
signature = key.sign(msg, ec.ECDSA(hashes.SM3()))

pubkey = key.public_key()
pubkey.verify(signature, msg, ec.ECDSA(hashes.SM3()))

from tongsuopy.crypto import serialization
from tongsuopy.crypto.asymciphers import ec

msg = b"hello"
key = ec.generate_private_key(ec.SM2())

pem = key.public_key().public_bytes(
    encoding=serialization.Encoding.PEM,
    format=serialization.PublicFormat.SubjectPublicKeyInfo,
)
pubkey = serialization.load_pem_public_key(pem)

ciphertext = pubkey.encrypt(msg)
decrypt_text = key.decrypt(ciphertext)
assert decrypt_text == msg