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

wallet/address: native p2wsh multisig get_address

Browse Source
This commit is contained in:
Tomas Susanka 2018-02-12 14:08:43 +01:00 committed by Jan Pochyla
parent 4c2dec55ed
commit 16072d7011
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/apps/wallet/sign_tx/addresses.py
View File

@ -24,21 +24,26 @@ def get_address(script_type: InputScriptType, coin: CoinType, node, multisig=Non
if script_type == InputScriptType.SPENDADDRESS: # p2pkh if script_type == InputScriptType.SPENDADDRESS: # p2pkh
return node.address(coin.address_type) return node.address(coin.address_type)
elif script_type == InputScriptType.SPENDWITNESS: # native p2wpkh elif script_type == InputScriptType.SPENDWITNESS: # native p2wpkh or native p2wsh
if not coin.segwit or not coin.bech32_prefix: if not coin.segwit or not coin.bech32_prefix:
raise AddressError(FailureType.ProcessError, raise AddressError(FailureType.ProcessError,
'Segwit not enabled on this coin') 'Segwit not enabled on this coin')
# native p2wsh multisig
if multisig is not None:
return address_multisig_p2wsh(multisig_get_pubkeys(multisig), multisig.m, coin.bech32_prefix)
# native p2wpkh
return address_p2wpkh(node.public_key(), coin.bech32_prefix) return address_p2wpkh(node.public_key(), coin.bech32_prefix)
elif script_type == InputScriptType.SPENDP2SHWITNESS: # p2wpkh or p2wsh using p2sh elif script_type == InputScriptType.SPENDP2SHWITNESS: # p2wpkh or p2wsh nested in p2sh
if not coin.segwit or coin.address_type_p2sh is None: if not coin.segwit or coin.address_type_p2sh is None:
raise AddressError(FailureType.ProcessError, raise AddressError(FailureType.ProcessError,
'Segwit not enabled on this coin') 'Segwit not enabled on this coin')
# p2wsh multisig # p2wsh multisig nested in p2sh
if multisig is not None: if multisig is not None:
return address_multisig_p2wsh_in_p2sh(multisig_get_pubkeys(multisig), multisig.m, coin.address_type_p2sh) return address_multisig_p2wsh_in_p2sh(multisig_get_pubkeys(multisig), multisig.m, coin.address_type_p2sh)
# p2wpkh # p2wpkh nested in p2sh
return address_p2wpkh_in_p2sh(node.public_key(), coin.address_type_p2sh) return address_p2wpkh_in_p2sh(node.public_key(), coin.address_type_p2sh)
elif script_type == InputScriptType.SPENDMULTISIG: # p2sh multisig elif script_type == InputScriptType.SPENDMULTISIG: # p2sh multisig
@ -77,12 +82,9 @@ def address_multisig_p2wsh_in_p2sh(pubkeys: bytes, m: int, addrtype):
return address_p2wsh_in_p2sh(digest, addrtype) return address_p2wsh_in_p2sh(digest, addrtype)
def address_multisig_p2wsh(pubkeys: bytes, m: int, addrtype): def address_multisig_p2wsh(pubkeys: bytes, m: int, hrp: str):
digest = output_script_multisig_p2wsh(pubkeys, m) digest = output_script_multisig_p2wsh(pubkeys, m)
if addrtype is None: return address_p2wsh(digest, hrp)
raise AddressError(FailureType.ProcessError,
'Multisig not enabled on this coin')
return address_p2sh(digest, addrtype)
def address_p2sh(redeem_script_hash: bytes, addrtype: int) -> str: def address_p2sh(redeem_script_hash: bytes, addrtype: int) -> str: