arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-03 20:11:00 +00:00
Code Issues Releases Wiki Activity

core: return root_fingerprint in PublicKey

Browse Source
This commit is contained in:
Pavol Rusnak 2021-01-05 15:41:26 +01:00
parent 4d45a68fd0
commit 5728f54b78
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/src/apps/bitcoin/get_public_key.py
View File

@ -60,4 +60,8 @@ async def get_public_key(ctx: wire.Context, msg: GetPublicKey) -> PublicKey:
if msg.show_display: if msg.show_display:
await layout.show_xpub(ctx, node_xpub, "XPUB", "Cancel") await layout.show_xpub(ctx, node_xpub, "XPUB", "Cancel")
return PublicKey(node=node_type, xpub=node_xpub) return PublicKey(
node=node_type,
xpub=node_xpub,
root_fingerprint=keychain.root_fingerprint(),
)

12
core/src/apps/common/keychain.py
View File

@ -98,6 +98,7 @@ class Keychain:
self.slip21_namespaces = tuple(slip21_namespaces) self.slip21_namespaces = tuple(slip21_namespaces)
self._cache = LRUCache(10) self._cache = LRUCache(10)
self._root_fingerprint: Optional[int] = None
def __del__(self) -> None: def __del__(self) -> None:
self._cache.__del__() self._cache.__del__()
@ -136,6 +137,17 @@ class Keychain:
node.derive_path(path[prefix_len:]) node.derive_path(path[prefix_len:])
return node return node
def root_fingerprint(self) -> int:
if self._root_fingerprint is None:
# derive m/0' to obtain root_fingerprint
n = self._derive_with_cache(
prefix_len=0,
path=[0 | paths.HARDENED],
new_root=lambda: bip32.from_seed(self.seed, self.curve),
)
self._root_fingerprint = n.fingerprint()
return self._root_fingerprint
def derive(self, path: paths.Bip32Path) -> bip32.HDNode: def derive(self, path: paths.Bip32Path) -> bip32.HDNode:
self.verify_path(path) self.verify_path(path)
return self._derive_with_cache( return self._derive_with_cache(