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

feat(core): implement GetPublicKey.ignore_xpub_magic behaviour

Browse Source
This commit is contained in:
Pavol Rusnak 2020-11-21 12:16:07 +01:00
parent e660a518c6
commit 92452d54e5
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -28,15 +28,21 @@ async def get_public_key(ctx: wire.Context, msg: GetPublicKey) -> PublicKey:
elif (
coin.segwit
and script_type == InputScriptType.SPENDP2SHWITNESS
and coin.xpub_magic_segwit_p2sh is not None
and (msg.ignore_xpub_magic or coin.xpub_magic_segwit_p2sh is not None)
):
node_xpub = node.serialize_public(coin.xpub_magic_segwit_p2sh)
# TODO: resolve type: ignore below
node_xpub = node.serialize_public(
coin.xpub_magic if msg.ignore_xpub_magic else coin.xpub_magic_segwit_p2sh # type: ignore
)
elif (
coin.segwit
and script_type == InputScriptType.SPENDWITNESS
and coin.xpub_magic_segwit_native is not None
and (msg.ignore_xpub_magic or coin.xpub_magic_segwit_native is not None)
):
node_xpub = node.serialize_public(coin.xpub_magic_segwit_native)
# TODO: resolve type: ignore below
node_xpub = node.serialize_public(
coin.xpub_magic if msg.ignore_xpub_magic else coin.xpub_magic_segwit_native # type: ignore
)
else:
raise wire.DataError("Invalid combination of coin and script_type")