RanchiMall/pyflo
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed bugs when create mnemonic phrase & generate seed from mnemonic phrase

Browse Source
This commit is contained in:
Alexey Karyabkin 2018-06-18 21:02:32 +04:00
parent 571793359c
commit e888bddca6
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pybtc/hdwallet.py
View File

@ -19,7 +19,7 @@ def create_passphrase(bits=256, language='english'):
if bits in [128, 160, 192, 224, 256]: if bits in [128, 160, 192, 224, 256]:
entropy = os.urandom(bits // 8) entropy = os.urandom(bits // 8)
mnemonic = create_mnemonic(entropy, language) mnemonic = create_mnemonic(entropy, language)
return ' '.join(mnemonic[::-1]) return ' '.join(mnemonic)
else: else:
raise ValueError('Strength should be one of the following [128, 160, 192, 224, 256], but it is not (%d).' % bits) raise ValueError('Strength should be one of the following [128, 160, 192, 224, 256], but it is not (%d).' % bits)
@ -61,7 +61,7 @@ def add_checksum_ent(data):
return data_int return data_int
def mnemonic2bytes(passphrase, language): def mnemonic_to_bytes(passphrase, language):
mnemonic = passphrase.split() mnemonic = passphrase.split()
if len(mnemonic) in [12, 15, 18, 21, 24]: if len(mnemonic) in [12, 15, 18, 21, 24]:
wordlist = create_wordlist(language) wordlist = create_wordlist(language)
@ -84,8 +84,8 @@ def mnemonic2bytes(passphrase, language):
raise ValueError('Number of words must be one of the following: [12, 15, 18, 21, 24], but it is not (%d).' % len(mnemonic)) raise ValueError('Number of words must be one of the following: [12, 15, 18, 21, 24], but it is not (%d).' % len(mnemonic))
def create_seed(passphrase, password=''): def create_seed(passphrase, password):
return pbkdf2_hmac('sha512', password.encode(), passphrase.encode(), 2048) return pbkdf2_hmac('sha512', passphrase.encode(), (passphrase + password).encode(), 2048)