mirror of
https://gitlab.com/pulsechaincom/staking-deposit-cli.git
synced 2025-01-10 21:11:21 +00:00
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
from os import walk
|
|
from unicodedata import normalize
|
|
from secrets import randbits
|
|
from typing import (
|
|
List,
|
|
Optional,
|
|
)
|
|
|
|
from utils.crypto import (
|
|
SHA256,
|
|
PBKDF2,
|
|
)
|
|
|
|
word_lists_path = './src/key_derivation/word_lists/'
|
|
|
|
|
|
def _get_word_list(language: str):
|
|
return open('%s%s.txt' % (word_lists_path, language)).readlines()
|
|
|
|
|
|
def get_seed(*, mnemonic: str, password: str='') -> bytes:
|
|
mnemonic = normalize('NFKD', mnemonic)
|
|
salt = normalize('NFKD', 'mnemonic' + password).encode('utf-8')
|
|
return PBKDF2(password=mnemonic, salt=salt, dklen=64, c=2048, prf='sha512')
|
|
|
|
|
|
def get_languages(path: str=word_lists_path) -> List[str]:
|
|
"""
|
|
Walk the `path` and list all the languages with word-lists available.
|
|
"""
|
|
(_, _, filenames) = next(walk(path))
|
|
filenames = [name[:-4] for name in filenames]
|
|
return filenames
|
|
|
|
|
|
def get_mnemonic(*, language: str, entropy: Optional[bytes]=None) -> str:
|
|
"""
|
|
Returns a mnemonic string in a given `language` based on `entropy`.
|
|
"""
|
|
if entropy is None:
|
|
entropy = randbits(256).to_bytes(32, 'big')
|
|
entropy_length = len(entropy) * 8
|
|
assert entropy_length in range(128, 257, 32)
|
|
checksum_length = (entropy_length // 32)
|
|
checksum = int.from_bytes(SHA256(entropy), 'big') >> 256 - checksum_length
|
|
entropy_bits = int.from_bytes(entropy, 'big') << checksum_length
|
|
entropy_bits += checksum
|
|
entropy_length += checksum_length
|
|
mnemonic = []
|
|
word_list = _get_word_list(language)
|
|
for i in range(entropy_length // 11 - 1, -1, -1):
|
|
index = (entropy_bits >> i * 11) & 2**11 - 1
|
|
mnemonic.append(word_list[index])
|
|
return ' '.join(mnemonic)
|