arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-26 09:28:13 +00:00
Code Issues Releases Wiki Activity

apps.common.seed: simplify, remove root caching

Browse Source
This commit is contained in:
Jan Pochyla 2016-12-12 15:18:46 +01:00
parent 47994e35f4
commit d2e31e24ee
1 changed files with 7 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/apps/common/seed.py
View File

@ -1,29 +1,16 @@
from trezor import wire from trezor import wire
from trezor.crypto import bip32
from trezor.crypto import bip39
# FIXME: this is a stub _DEFAULT_CURVE = 'secp256k1'
_cached_seed = None _cached_seed = None
_cached_root_node = None
async def get_node(session_id: int, path: list): async def get_root(session_id: int, curve_name=_DEFAULT_CURVE):
root = await get_root_node(session_id)
node = root.clone()
node.derive_path(path)
return node
async def get_root_node(session_id: int):
global _cached_root_node
if _cached_root_node is None:
_cached_root_node = await compute_root_node(session_id)
return _cached_root_node
async def compute_root_node(session_id: int):
from trezor.crypto import bip32
seed = await get_seed(session_id) seed = await get_seed(session_id)
return bip32.from_seed(seed, 'secp256k1') root = bip32.from_seed(seed, curve_name)
return root
async def get_seed(session_id: int) -> bytes: async def get_seed(session_id: int) -> bytes:
@ -33,8 +20,7 @@ async def get_seed(session_id: int) -> bytes:
return _cached_seed return _cached_seed
async def compute_seed(session_id): async def compute_seed(session_id: int) -> bytes:
from trezor.crypto import bip39
from trezor.messages.FailureType import Other from trezor.messages.FailureType import Other
from .request_passphrase import request_passphrase from .request_passphrase import request_passphrase
from .request_pin import protect_by_pin from .request_pin import protect_by_pin