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

added validation of input data

Browse Source
This commit is contained in:
Alexey Karyabkin 2018-06-04 16:17:57 +04:00
parent 9003eefc79
commit 1dce8d124d
1 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pybtc/hdwallet.py
View File

@ -11,9 +11,11 @@ from .constants import *
# #
def create_passphrase(bits=256, language='english'): def create_passphrase(bits=256, language='english'):
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[::-1])
raise ValueError('Strength should be one of the following [128, 160, 192, 224, 256], but it is not (%d).' % bits)
def create_mnemonic(entropy, language='english'): def create_mnemonic(entropy, language='english'):
@ -57,6 +59,7 @@ def add_checksum(data):
def mnemonic2bytes(passphrase, language): def mnemonic2bytes(passphrase, language):
mnemonic = passphrase.split() mnemonic = passphrase.split()
if len(mnemonic) in [12, 15, 18, 21, 24]:
wordlist = create_wordlist(language) wordlist = create_wordlist(language)
codes = dict() codes = dict()
for code, word in enumerate(wordlist): for code, word in enumerate(wordlist):
@ -74,5 +77,6 @@ def mnemonic2bytes(passphrase, language):
fb = unhexlify(ent_hash)[0] fb = unhexlify(ent_hash)[0]
assert (fb >> (8 - chk_sum_bit_len)) & chk_sum assert (fb >> (8 - chk_sum_bit_len)) & chk_sum
return entropy return entropy
raise ValueError('Number of words must be one of the following: [12, 15, 18, 21, 24], but it is not (%d).' % len(mnemonic))