testing: add BIP-322 POR signed with WIF Store test
This commit is contained in:
scgbckbone
doc-hex
parent
8cf271ab4a
commit
d8f9a88bcb
@ -7,7 +7,7 @@ from ckcc_protocol.protocol import MAX_TXN_LEN
|
||||
from psbt import BasicPSBT, BasicPSBTInput, BasicPSBTOutput
|
||||
from io import BytesIO
|
||||
from helpers import hash160, taptweak, str_to_path
|
||||
from bip32 import BIP32Node
|
||||
from bip32 import BIP32Node, PublicKey
|
||||
from constants import simulator_fixed_tprv, AF_P2WSH, AF_P2WSH_P2SH, AF_P2SH
|
||||
from ctransaction import CTransaction, COutPoint, CTxIn, CTxOut, uint256_from_str
|
||||
|
||||
@ -59,6 +59,7 @@ def bip322_txn(dev, pytestconfig, create_msg_file):
|
||||
sp = f"0/{i}"
|
||||
af = addr_fmt
|
||||
ia = input_amount
|
||||
pubkey = None # public key
|
||||
try:
|
||||
if inp[0] is not None:
|
||||
af = inp[0]
|
||||
@ -66,12 +67,20 @@ def bip322_txn(dev, pytestconfig, create_msg_file):
|
||||
sp = inp[1]
|
||||
if inp[2] is not None:
|
||||
ia = inp[2]
|
||||
if inp[3] is not None:
|
||||
pubkey = inp[3]
|
||||
except:
|
||||
pass
|
||||
|
||||
int_path = str_to_path(sp)
|
||||
subkey = mk.subkey_for_path(sp)
|
||||
sec = subkey.sec()
|
||||
if pubkey:
|
||||
int_path = [0]
|
||||
sec = pubkey
|
||||
else:
|
||||
int_path = str_to_path(sp)
|
||||
sec = mk.subkey_for_path(sp).sec()
|
||||
|
||||
subkey = PublicKey.parse(sec)
|
||||
|
||||
assert len(sec) == 33, "expect compressed"
|
||||
|
||||
if af == "p2tr":
|
||||
@ -82,7 +91,7 @@ def bip322_txn(dev, pytestconfig, create_msg_file):
|
||||
|
||||
elif af in ("p2wpkh", "p2sh-p2wpkh", "p2wpkh-p2sh"):
|
||||
psbt.inputs[i].bip32_paths[sec] = mfp + struct.pack(f'<{"I" * len(int_path)}', *int_path)
|
||||
scr = bytes([0x00, 0x14]) + subkey.hash160()
|
||||
scr = bytes([0x00, 0x14]) + subkey.h160()
|
||||
|
||||
if af != "p2wpkh":
|
||||
# use classic p2wpkh (from above) as redeem script
|
||||
@ -90,8 +99,8 @@ def bip322_txn(dev, pytestconfig, create_msg_file):
|
||||
scr = bytes([0xa9, 0x14]) + hash160(scr) + bytes([0x87])
|
||||
|
||||
elif af == "p2pkh":
|
||||
psbt.inputs[i].bip32_paths[sec] = mfp + struct.pack('<II', 0, i)
|
||||
scr = bytes([0x76, 0xa9, 0x14]) + subkey.hash160() + bytes([0x88, 0xac])
|
||||
psbt.inputs[i].bip32_paths[sec] = mfp + struct.pack(f'<{"I" * len(int_path)}', *int_path)
|
||||
scr = bytes([0x76, 0xa9, 0x14]) + subkey.h160() + bytes([0x88, 0xac])
|
||||
|
||||
else:
|
||||
raise ValueError("unknown addr_fmt %s" % af)
|
||||
|
||||
@ -2,11 +2,11 @@
|
||||
#
|
||||
# BIP-322 Message Signing and Proof of Reserves
|
||||
#
|
||||
import pytest, time
|
||||
import pytest, time, os
|
||||
from io import BytesIO
|
||||
from decimal import Decimal
|
||||
from constants import SIGHASH_MAP, AF_P2SH, AF_P2WSH, AF_P2WSH_P2SH
|
||||
from bip322 import bip322_txn, bip322_ms_txn, bip322_msg_hash
|
||||
from bip322 import bip322_txn, bip322_ms_txn, bip322_msg_hash, BIP32Node
|
||||
from ctransaction import CTransaction, CTxIn, COutPoint
|
||||
from helpers import str_to_path
|
||||
from charcodes import KEY_QR, KEY_NFC
|
||||
@ -574,4 +574,58 @@ def test_bip322_msg_import_fail(bip322_txn, start_sign, end_sign, cap_story, nee
|
||||
assert "hash verification failed" in story
|
||||
press_cancel()
|
||||
|
||||
|
||||
@pytest.mark.parametrize("num_ins", [1, 12])
|
||||
@pytest.mark.parametrize("addr_fmt", ["p2pkh", "p2wpkh", "p2sh-p2wpkh"])
|
||||
def test_wif_store_sign_bip322_por(num_ins, addr_fmt, bip322_txn, goto_home, pick_menu_item,
|
||||
need_keypress, start_sign, end_sign, cap_menu, cap_story,
|
||||
press_cancel, settings_remove, press_select, import_wif_to_store):
|
||||
|
||||
settings_remove("wifs")
|
||||
|
||||
node = BIP32Node.from_master_secret(os.urandom(32))
|
||||
|
||||
ins = []
|
||||
wifs = []
|
||||
for i in range(num_ins):
|
||||
n = node.subkey_for_path("0/%d" % i)
|
||||
wifs.append(n.node.private_key.wif(testnet=True))
|
||||
if i == 0:
|
||||
amt = None
|
||||
elif i // 2 == 0:
|
||||
amt = 10000000
|
||||
else:
|
||||
amt = 900000000
|
||||
|
||||
ins.append([addr_fmt, None, amt , n.node.private_key.K.sec()])
|
||||
|
||||
msg = b"Coinkite"
|
||||
psbt, msg_challenge = bip322_txn(ins, msg=b"Coinkite")
|
||||
|
||||
import_wif_to_store(wifs)
|
||||
|
||||
menu = cap_menu()
|
||||
assert menu[0] == "Import WIF"
|
||||
|
||||
start_sign(psbt, finalize=True)
|
||||
time.sleep(.1)
|
||||
title, story = cap_story()
|
||||
assert "import message" in story
|
||||
# msg file was auto-gened on SD card
|
||||
need_keypress("1")
|
||||
time.sleep(.1)
|
||||
title, story = cap_story()
|
||||
assert title == "Message:"
|
||||
assert msg.decode() in story
|
||||
press_select()
|
||||
time.sleep(.1)
|
||||
title, story = cap_story()
|
||||
assert "Proof of Reserves" in story
|
||||
assert "warning" in story
|
||||
if num_ins == 1:
|
||||
assert "WIF store: 0" in story
|
||||
else:
|
||||
assert f"WIF store: {', '.join([str(i) for i in range(num_ins)])}" in story
|
||||
end_sign(finalize=True)
|
||||
|
||||
# EOF
|
||||
|
||||
@ -613,11 +613,8 @@ def test_wif_store_signing(num_ins, addr_fmt, fake_txn, goto_home, pick_menu_ite
|
||||
psbt = fake_txn(num_ins, 1, segwit_in=sw, wrapped=wrap, master_xpub=node.hwif())
|
||||
|
||||
wifs = []
|
||||
privkeys = []
|
||||
for i in range(num_ins):
|
||||
n = node.subkey_for_path("0/%d" % i)
|
||||
sk = n.node.private_key
|
||||
privkeys.append(sk)
|
||||
wifs.append(n.node.private_key.wif(testnet=True))
|
||||
|
||||
import_wif_to_store(wifs)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user