highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2026-09-11 21:20:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

test: check joinpsbts preserves global xpubs and proprietary fields

Browse Source
This commit is contained in:
Thomas
2026-06-11 20:53:29 +02:00
parent 011094b282
commit 436921eb46
1 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
test/functional/rpc_psbt.py
View File

@@ -1158,6 +1158,45 @@ class PSBTTest(BitcoinTestFramework):
break
assert shuffled
# Check that joining preserves global xpub and proprietary records
def global_xpub_key(extended_pubkey):
xpub_data, xpub_version = base58_to_byte(extended_pubkey)
return bytes([PSBT_GLOBAL_XPUB]) + bytes([xpub_version]) + xpub_data
xpub1 = "tpubD6NzVbkrYhZ4XgiXtGrdW5XDAPFCL9h7we1vwNCpn8tGbBcgfVYjXyhWo4E1xkh56hjod1RhGjxbaTLV3X4FyWuejifB9jusQ46QzG87VKp"
xpub_key1 = global_xpub_key(xpub1)
xpub_key2 = global_xpub_key("tpubD6NzVbkrYhZ4WaWSyoBvQwbpLkojyoTZPRsgXELWz3Popb3qkjcJyJUGLnL4qHHoQvao8ESaAstxYSnhyswJ76uZPStJRJCTKvosUCJZL5B")
xpub_value = b"\x00\x00\x00\x00" # master key fingerprint with an empty derivation path
global_prop_key = bytes([PSBT_GLOBAL_PROPRIETARY]) + b"\x02\x01\x02\x00" # identifier "0102", subtype 0
global_prop_value = b"\xde\xad\xbe\xef"
psbt1_obj = PSBT.from_base64(psbt1)
psbt1_obj.g.map[xpub_key1] = xpub_value
psbt1_obj.g.map[global_prop_key] = global_prop_value
psbt2_obj = PSBT.from_base64(psbt2)
psbt2_obj.g.map[xpub_key2] = xpub_value
joined_globals = PSBT.from_base64(self.nodes[0].joinpsbts([psbt1_obj.to_base64(), psbt2_obj.to_base64()]))
assert_equal(joined_globals.g.map[xpub_key1], xpub_value)
assert_equal(joined_globals.g.map[xpub_key2], xpub_value)
assert_equal(joined_globals.g.map[global_prop_key], global_prop_value)
# Same proprietary key in both PSBTs with different values: the first PSBT's value wins
collide_key = bytes([PSBT_GLOBAL_PROPRIETARY]) + b"\x02\x03\x04\x00"
psbt_first_obj = PSBT.from_base64(psbt1)
psbt_first_obj.g.map[collide_key] = b"\x11\x11\x11\x11"
psbt_second_obj = PSBT.from_base64(psbt2)
psbt_second_obj.g.map[collide_key] = b"\x22\x22\x22\x22"
joined_collision = PSBT.from_base64(self.nodes[0].joinpsbts([psbt_first_obj.to_base64(), psbt_second_obj.to_base64()]))
assert_equal(joined_collision.g.map[collide_key], b"\x11\x11\x11\x11")
# Same xpub with conflicting origins: the first PSBT's origin is kept, avoiding duplicate keys
conflict_first_obj = PSBT.from_base64(psbt1)
conflict_first_obj.g.map[xpub_key1] = xpub_value
conflict_second_obj = PSBT.from_base64(psbt2)
conflict_second_obj.g.map[xpub_key1] = b"\x11\x11\x11\x11"
joined_conflict = self.nodes[0].joinpsbts([conflict_first_obj.to_base64(), conflict_second_obj.to_base64()])
assert_equal(self.nodes[0].decodepsbt(joined_conflict)["global_xpubs"], [{"xpub": xpub1, "master_fingerprint": "00000000", "path": "m"}])
# Newly created PSBT needs UTXOs and updating
addr = self.nodes[1].getnewaddress("", "p2sh-segwit")
utxo = self.create_outpoints(self.nodes[0], outputs=[{addr: 7}])[0]