2018-01-29 13:41:56 +00:00
|
|
|
from trezor.crypto import bip32
|
|
|
|
|
|
|
|
|
|
from trezor.messages.MultisigRedeemScriptType import MultisigRedeemScriptType
|
|
|
|
|
from trezor.messages.HDNodePathType import HDNodePathType
|
|
|
|
|
|
|
|
|
|
from apps.wallet.sign_tx.writers import *
|
|
|
|
|
|
|
|
|
|
|
2018-01-31 11:48:18 +00:00
|
|
|
def multisig_pubkey_index(multisig: MultisigRedeemScriptType, pubkey: bytes) -> int:
|
|
|
|
|
for i, hd in enumerate(multisig.pubkeys):
|
|
|
|
|
if multisig_get_pubkey(hd) == pubkey:
|
|
|
|
|
return i
|
|
|
|
|
return -1
|
|
|
|
|
|
|
|
|
|
|
2018-01-29 13:41:56 +00:00
|
|
|
def multisig_get_pubkey(hd: HDNodePathType) -> bytes:
|
|
|
|
|
p = hd.address_n
|
|
|
|
|
n = hd.node
|
|
|
|
|
node = bip32.HDNode(
|
|
|
|
|
depth=n.depth,
|
|
|
|
|
fingerprint=n.fingerprint,
|
|
|
|
|
child_num=n.child_num,
|
|
|
|
|
chain_code=n.chain_code,
|
|
|
|
|
public_key=n.public_key)
|
|
|
|
|
for i in p:
|
|
|
|
|
node.derive(i, True)
|
|
|
|
|
return node.public_key()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def multisig_get_pubkeys(multisig: MultisigRedeemScriptType):
|
|
|
|
|
return [multisig_get_pubkey(hd) for hd in multisig.pubkeys]
|
2018-02-01 12:23:12 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def check_address_n_against_pubkeys(multisig: MultisigRedeemScriptType, address_n) -> bool:
|
|
|
|
|
for p in multisig.pubkeys:
|
|
|
|
|
if p.address_n == address_n:
|
|
|
|
|
return True
|
|
|
|
|
return False
|
