You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.9 KiB
86 lines
2.9 KiB
5 years ago
|
from trezor.crypto import bip32
|
||
6 years ago
|
from trezor.messages import InputScriptType
|
||
6 years ago
|
from trezor.messages.Address import Address
|
||
6 years ago
|
|
||
4 years ago
|
from . import addresses
|
||
4 years ago
|
from .keychain import with_keychain
|
||
4 years ago
|
from .multisig import multisig_pubkey_index
|
||
4 years ago
|
|
||
5 years ago
|
from apps.common.layout import address_n_to_str, show_address, show_qr, show_xpub
|
||
6 years ago
|
from apps.common.paths import validate_path
|
||
8 years ago
|
|
||
5 years ago
|
if False:
|
||
|
from typing import List
|
||
|
from trezor.messages import HDNodeType
|
||
|
from trezor import wire
|
||
|
from apps.common.coininfo import CoinInfo
|
||
|
|
||
|
|
||
|
async def show_xpubs(
|
||
4 years ago
|
ctx: wire.Context, coin: CoinInfo, pubnodes: List[HDNodeType], multisig_index: int
|
||
5 years ago
|
) -> bool:
|
||
5 years ago
|
for i, pubnode in enumerate(pubnodes):
|
||
5 years ago
|
cancel = "Next" if i < len(pubnodes) - 1 else "Address"
|
||
|
node = bip32.HDNode(
|
||
5 years ago
|
depth=pubnode.depth,
|
||
|
fingerprint=pubnode.fingerprint,
|
||
|
child_num=pubnode.child_num,
|
||
|
chain_code=pubnode.chain_code,
|
||
|
public_key=pubnode.public_key,
|
||
5 years ago
|
curve_name=coin.curve_name,
|
||
|
)
|
||
|
xpub = node.serialize_public(coin.xpub_magic)
|
||
4 years ago
|
desc = "XPUB #%d" % (i + 1)
|
||
|
desc += " (yours)" if i == multisig_index else " (others)"
|
||
|
if await show_xpub(ctx, xpub, desc=desc, cancel=cancel):
|
||
5 years ago
|
return True
|
||
|
return False
|
||
|
|
||
8 years ago
|
|
||
4 years ago
|
@with_keychain
|
||
|
async def get_address(ctx, msg, keychain, coin):
|
||
6 years ago
|
await validate_path(
|
||
|
ctx,
|
||
|
addresses.validate_full_path,
|
||
5 years ago
|
keychain,
|
||
|
msg.address_n,
|
||
5 years ago
|
coin.curve_name,
|
||
6 years ago
|
coin=coin,
|
||
|
script_type=msg.script_type,
|
||
|
)
|
||
|
|
||
4 years ago
|
node = keychain.derive(msg.address_n)
|
||
6 years ago
|
address = addresses.get_address(msg.script_type, coin, node, msg.multisig)
|
||
6 years ago
|
address_short = addresses.address_short(coin, address)
|
||
5 years ago
|
if msg.script_type == InputScriptType.SPENDWITNESS:
|
||
|
address_qr = address.upper() # bech32 address
|
||
5 years ago
|
elif coin.cashaddr_prefix is not None:
|
||
|
address_qr = address.upper() # cashaddr address
|
||
5 years ago
|
else:
|
||
|
address_qr = address # base58 address
|
||
8 years ago
|
|
||
8 years ago
|
if msg.show_display:
|
||
6 years ago
|
if msg.multisig:
|
||
5 years ago
|
if msg.multisig.nodes:
|
||
|
pubnodes = msg.multisig.nodes
|
||
|
else:
|
||
|
pubnodes = [hd.node for hd in msg.multisig.pubkeys]
|
||
4 years ago
|
multisig_index = multisig_pubkey_index(msg.multisig, node.public_key())
|
||
5 years ago
|
desc = "Multisig %d of %d" % (msg.multisig.m, len(pubnodes))
|
||
|
while True:
|
||
|
if await show_address(ctx, address_short, desc=desc):
|
||
|
break
|
||
|
if await show_qr(ctx, address_qr, desc=desc, cancel="XPUBs"):
|
||
|
break
|
||
4 years ago
|
if await show_xpubs(ctx, coin, pubnodes, multisig_index):
|
||
5 years ago
|
break
|
||
6 years ago
|
else:
|
||
|
desc = address_n_to_str(msg.address_n)
|
||
5 years ago
|
while True:
|
||
|
if await show_address(ctx, address_short, desc=desc):
|
||
|
break
|
||
|
if await show_qr(ctx, address_qr, desc=desc):
|
||
|
break
|
||
8 years ago
|
|
||
8 years ago
|
return Address(address=address)
|