mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-01-02 11:30:58 +00:00
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from trezor import wire
|
|
from trezor.crypto import bip32
|
|
from trezor.crypto import bip39
|
|
|
|
_DEFAULT_CURVE = 'secp256k1'
|
|
|
|
|
|
async def derive_node(ctx: wire.Context, path=[], curve_name=_DEFAULT_CURVE):
|
|
seed = await _get_seed(ctx)
|
|
node = bip32.from_seed(seed, curve_name)
|
|
if len(path) > 0:
|
|
node.derive_path(path)
|
|
return node
|
|
|
|
|
|
async def _get_seed(ctx: wire.Context) -> bytes:
|
|
from . import cache
|
|
if cache.seed is None:
|
|
cache.seed = await _compute_seed(ctx)
|
|
return cache.seed
|
|
|
|
|
|
async def _compute_seed(ctx: wire.Context) -> bytes:
|
|
from trezor.messages.FailureType import ProcessError
|
|
from .request_passphrase import protect_by_passphrase
|
|
from . import storage
|
|
|
|
if not storage.is_initialized():
|
|
raise wire.FailureError(ProcessError, 'Device is not initialized')
|
|
|
|
passphrase = await protect_by_passphrase(ctx)
|
|
return bip39.seed(storage.get_mnemonic(), passphrase)
|
|
|
|
|
|
def derive_node_without_passphrase(path, curve_name=_DEFAULT_CURVE):
|
|
from . import storage
|
|
if not storage.is_initialized():
|
|
raise Exception('Device is not initialized')
|
|
seed = bip39.seed(storage.get_mnemonic(), '')
|
|
node = bip32.from_seed(seed, curve_name)
|
|
node.derive_path(path)
|
|
return node
|