AaronFiore/electrum
1
0
Fork 0
Code Activity

mnemonic.make_seed: de-duplicate num_bits default magic number

Browse Source
This commit is contained in:
SomberNight
2020-12-08 10:00:34 +01:00
parent 376ee395f8
commit 933d8861ce
6 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
electrum/base_wizard.py
View File

@@ -707,7 +707,7 @@ class BaseWizard(Logger):
def create_seed(self, seed_type):
from . import mnemonic
self.seed_type = seed_type
seed = mnemonic.Mnemonic('en').make_seed(self.seed_type)
seed = mnemonic.Mnemonic('en').make_seed(seed_type=self.seed_type)
self.opt_bip39 = False
f = lambda x: self.request_passphrase(seed, x)
self.show_seed_dialog(run_next=f, seed_text=seed)

4
electrum/commands.py
View File

@@ -318,10 +318,10 @@ class Commands:
return self.config.get_ssl_domain()
@command('')
async def make_seed(self, nbits=132, language=None, seed_type=None):
async def make_seed(self, nbits=None, language=None, seed_type=None):
"""Create a seed"""
from .mnemonic import Mnemonic
s = Mnemonic(language).make_seed(seed_type, num_bits=nbits)
s = Mnemonic(language).make_seed(seed_type=seed_type, num_bits=nbits)
return s
@command('n')

13
electrum/mnemonic.py
View File

@@ -187,19 +187,20 @@ class Mnemonic(Logger):
i = i*n + k
return i
def make_seed(self, seed_type=None, *, num_bits=132) -> str:
def make_seed(self, *, seed_type=None, num_bits=None) -> str:
if seed_type is None:
seed_type = 'segwit'
if num_bits is None:
num_bits = 132
prefix = version.seed_prefix(seed_type)
# increase num_bits in order to obtain a uniform distribution for the last word
bpw = math.log(len(self.wordlist), 2)
# rounding
n = int(math.ceil(num_bits/bpw) * bpw)
self.logger.info(f"make_seed. prefix: '{prefix}', entropy: {n} bits")
num_bits = int(math.ceil(num_bits/bpw) * bpw)
self.logger.info(f"make_seed. prefix: '{prefix}', entropy: {num_bits} bits")
entropy = 1
while entropy < pow(2, n - bpw):
while entropy < pow(2, num_bits - bpw):
# try again if seed would not contain enough words
entropy = randrange(pow(2, n))
entropy = randrange(pow(2, num_bits))
nonce = 0
while True:
nonce += 1

2
electrum/plugins/trustedcoin/trustedcoin.py
View File

@@ -524,7 +524,7 @@ class TrustedCoinPlugin(BasePlugin):
def make_seed(self, seed_type):
if not is_any_2fa_seed_type(seed_type):
raise Exception(f'unexpected seed type: {seed_type}')
return Mnemonic('english').make_seed(seed_type=seed_type, num_bits=128)
return Mnemonic('english').make_seed(seed_type=seed_type)
@hook
def do_clear(self, window):

2
electrum/tests/test_mnemonic.py
View File

@@ -120,7 +120,7 @@ class Test_NewMnemonic(ElectrumTestCase):
iters = 10
m = mnemonic.Mnemonic(lang='en')
for _ in range(iters):
seed = m.make_seed("standard")
seed = m.make_seed(seed_type="standard")
i = m.mnemonic_decode(seed)
self.assertEqual(m.mnemonic_encode(i), seed)

2
electrum/wallet.py
View File

@@ -2800,7 +2800,7 @@ def create_new_wallet(*, path, config: SimpleConfig, passphrase=None, password=N
raise Exception("Remove the existing wallet first!")
db = WalletDB('', manual_upgrades=False)
seed = Mnemonic('en').make_seed(seed_type)
seed = Mnemonic('en').make_seed(seed_type=seed_type)
k = keystore.from_seed(seed, passphrase)
db.put('keystore', k.dump())
db.put('wallet_type', 'standard')