app/model/key_gm.py

   1 import hmac
   2 import hashlib
   3 from gmssl import sm2, func
   4
   5 # get_gm_root_xprv create rootxprv from seed
   6 # seed_str length is 512 bits.
   7 # root_xprv length is 512 bits.
   8 # You can get more test data from: https://gist.github.com/zcc0721/8bfc1e1c92b7cd9f00743d6878097809
   9 # test data 1:
  10 #   seed_str: ecc2bbb6c0492873cdbc81edf56bd896d3b644047879840e357be735b7fa7b6f4af1be7b8d71cc649ac4ca3816f9ccaf11bf49f4effb845f3c19e16eaf8bfcda
  11 #   root_xprv_str: a61d6b741b0e74b8d0836ac22b675bbf8e108148ef018d1b000aef1a899a136bd316c0f59e7333520ae1a429504073b2773869e95aa95bb3a4fa0ec76744025c
  12 # test data 2:
  13 #   seed_str: afa3a86bbec2f40bb32833fc6324593824c4fc7821ed32eac1f762b5893e56745f66a6c6f2588b3d627680aa4e0e50efd25065097b3daa8c6a19d606838fe7d4
  14 #   root_xprv_str: 302a25c7c0a68a83fa043f594a2db8b44bc871fced553a8a33144b31bc7fb88887c9e75915bb6ba3fd0b9f94a60b7a5897ab9db6a48f888c2559132dba9152b0
  15 # test data 3:
  16 #   seed_str: b435f948bd3748ede8f9d6f59728d669939e79c6c885667a5c138e05bbabde1de0dcfcbe0c6112022fbbf0da522f4e224a9c2381016380688b51886248b3156f
  17 #   root_xprv_str: 6532adeb967ac5ccbf988cf8190817bf9040c8cfd9cdfe3d5e400effb29469e6d478b61cc6be936f367ae769eb1dc65c473ee73cac2eb43cf6d5e7c62b7f0062
  18 def get_gm_root_xprv(seed_str):
  19     hc_str = hmac.HMAC(b'Root', bytes.fromhex(seed_str), digestmod=hashlib.sha512).hexdigest()
  20     root_xprv_str = hc_str
  21     return {
  22         "root_xprv": root_xprv_str
  23     }
  24
  25 # get_gm_xpub derives new xpub from xprv
  26 # xprv length is 64 bytes.
  27 # xpub length is 65 bytes.
  28 # You can get more test data from: https://gist.github.com/zcc0721/9e5761e6a924cce3aa7cf7f72721218a
  29 # test data 1:
  30 #   xprv_str: c003f4bcccf9ad6f05ad2c84fa5ff98430eb8e73de5de232bc29334c7d074759d513bc370335cac51d77f0be5dfe84de024cfee562530b4d873b5f5e2ff4f57c
  31 #   xpub_str: 02476044353971ae0ed41cba76f27d0bd2e09d09db5c238bb74f69569bf343f742d513bc370335cac51d77f0be5dfe84de024cfee562530b4d873b5f5e2ff4f57c
  32 # test data 2:
  33 #   xprv_str: 36667587de27eec684fc4b222276f22a24d9a82e947ee0119148bedd4dec461dd4e1b1d95dfb0f78896677ea1026af7510b41fabd3bd5771311c0cb6968337b2
  34 #   xpub_str: 0396a36cd902db56eca016c213a8ac25de35a7afd78061351f1898529f0956c22ed4e1b1d95dfb0f78896677ea1026af7510b41fabd3bd5771311c0cb6968337b2
  35 # test data 3:
  36 #   xprv_str: 74a49c698dbd3c12e36b0b287447d833f74f3937ff132ebff7054baa18623c35a705bb18b82e2ac0384b5127db97016e63609f712bc90e3506cfbea97599f46f
  37 #   xpub_str: 03cafbdedea4a639d31fe4c257f1bb58303359be1a00b9f90b5c605f57e4308ed1a705bb18b82e2ac0384b5127db97016e63609f712bc90e3506cfbea97599f46f
  38 def get_gm_xpub(xprv_str):
  39     private_key_int = int(xprv_str[:64], 16)
  40     sm2_crypt = sm2.CryptSM2(private_key=xprv_str[:64], public_key="")
  41     public_key_str = sm2_crypt._kg(private_key_int, sm2.default_ecc_table['g'])
  42     pc = '0' + str(int(public_key_str[-1], 16) % 2 + 2)
  43     xpub_str = pc + public_key_str[:64] + xprv_str[64:]
  44     return {
  45         "xpub": xpub_str
  46     }