def test_single_witness() -> None:
# 4e52f7848dab7dd89ef7ba477939574198a170bfcb2fb34355c69f5e0169f63c
tx_bytes = "010000000001019bdea7abb2fa14dead47dd14d03cf82212a25b6096a8da6b14feec3658dbcf9d0100000000ffffffff02a02526000000000017a914f987c321394968be164053d352fc49763b2be55c874361610000000000220020701a8d401c84fb13e6baf169d59684e17abd9fa216c8cc5b9fc63d622ff8c58d04004730440220421fbbedf2ee096d6289b99973509809d5e09589040d5e0d453133dd11b2f78a02205686dbdb57e0c44e49421e9400dd4e931f1655332e8d078260c9295ba959e05d014730440220398f141917e4525d3e9e0d1c6482cb19ca3188dc5516a3a5ac29a0f4017212d902204ea405fae3a58b1fc30c5ad8ac70a76ab4f4d876e8af706a6a7b4cd6fa100f44016952210375e00eb72e29da82b89367947f29ef34afb75e8654f6ea368e0acdfd92976b7c2103a1b26313f430c4b15bb1fdce663207659d8cac749a0e53d70eff01874496feff2103c96d495bfdd5ba4145e3e046fee45e84a8a48ad05bd8dbb395c011a32cf9f88053ae00000000"
tx = Tx.parse(tx_bytes)
assert tx.serialize(include_witness=True).hex() == tx_bytes
assert tx == Tx.from_dict(tx.to_dict())
assert tx.version == 1
assert tx.lock_time == 0
assert len(tx.vin) == 1
assert len(tx.vout) == 2
stack = [
"",
"30440220421fbbedf2ee096d6289b99973509809d5e09589040d5e0d453133dd11b2f78a02205686dbdb57e0c44e49421e9400dd4e931f1655332e8d078260c9295ba959e05d01",
"30440220398f141917e4525d3e9e0d1c6482cb19ca3188dc5516a3a5ac29a0f4017212d902204ea405fae3a58b1fc30c5ad8ac70a76ab4f4d876e8af706a6a7b4cd6fa100f4401",
"52210375e00eb72e29da82b89367947f29ef34afb75e8654f6ea368e0acdfd92976b7c2103a1b26313f430c4b15bb1fdce663207659d8cac749a0e53d70eff01874496feff2103c96d495bfdd5ba4145e3e046fee45e84a8a48ad05bd8dbb395c011a32cf9f88053ae",
]
witness = Witness(stack)
assert tx.vin[0].script_witness == witness
tx_id = "4e52f7848dab7dd89ef7ba477939574198a170bfcb2fb34355c69f5e0169f63c"
assert tx.id.hex() == tx_id
hash_ = "d39eb3e3954be4bdc0b3be2d980124b1e1e11fb414b886b52939b07d95a58a8f"
assert tx.hash.hex() == hash_
assert tx.size == 380
assert tx.vsize == 190
assert tx.weight == 758
assert tx.is_segwit()
assert any(bool(w) for w in tx.vwitness)
assert not tx.is_coinbase()
def test_exceptions() -> None:
tx_bytes = "010000000001019bdea7abb2fa14dead47dd14d03cf82212a25b6096a8da6b14feec3658dbcf9d0100000000ffffffff02a02526000000000017a914f987c321394968be164053d352fc49763b2be55c874361610000000000220020701a8d401c84fb13e6baf169d59684e17abd9fa216c8cc5b9fc63d622ff8c58d04004730440220421fbbedf2ee096d6289b99973509809d5e09589040d5e0d453133dd11b2f78a02205686dbdb57e0c44e49421e9400dd4e931f1655332e8d078260c9295ba959e05d014730440220398f141917e4525d3e9e0d1c6482cb19ca3188dc5516a3a5ac29a0f4017212d902204ea405fae3a58b1fc30c5ad8ac70a76ab4f4d876e8af706a6a7b4cd6fa100f44016952210375e00eb72e29da82b89367947f29ef34afb75e8654f6ea368e0acdfd92976b7c2103a1b26313f430c4b15bb1fdce663207659d8cac749a0e53d70eff01874496feff2103c96d495bfdd5ba4145e3e046fee45e84a8a48ad05bd8dbb395c011a32cf9f88053ae00000000"
tx = Tx.parse(tx_bytes)
tx.version = 0
with pytest.raises(BTClibValueError, match="invalid version: "):
tx.assert_valid()
tx = Tx.parse(tx_bytes)
tx.lock_time = 0xFFFFFFFF + 1
with pytest.raises(BTClibValueError, match="invalid lock time: "):
tx.assert_valid()
def test_p2pk() -> None:
pub_key = "0479BE667EF9DCBBAC55A06295CE870B07029BFCDB2DCE28D959F2815B16F81798483ADA7726A3C4655DA4FBFC0E1108A8FD17B448A68554199C47D08FFB10D4B8"
signature = "304402200A5C6163F07B8D3B013C4D1D6DBA25E780B39658D79BA37AF7057A3B7F15FFA102201FD9B4EAA9943F734928B99A83592C2E7BF342EA2680F6A2BB705167966B742001"
script_pub_key = serialize([pub_key, "OP_CHECKSIG"])
script_sig = serialize([signature])
founding_tx_script = serialize(["OP_0", "OP_0"])
tx_in = TxIn(OutPoint(b"\x00" * 32, 0xFFFFFFFF), founding_tx_script, 0xFFFFFFFF)
funding_tx = Tx(1, 0, [tx_in], [TxOut(0, script_pub_key)])
tx_in = TxIn(OutPoint(funding_tx.id, 0), script_sig, 0xFFFFFFFF)
receiving_tx = Tx(1, 0, [tx_in], [TxOut(0, b"")])
hash_ = sig_hash.from_tx(funding_tx.vout, receiving_tx, 0, sig_hash.ALL)
assert dsa.verify_(hash_, pub_key, bytes.fromhex(signature)[:-1])
def deserialize_tx(k: bytes, v: bytes, type_: str) -> Tx:
"Return the dataclass element from its binary representation."
if len(k) != 1:
err_msg = f"invalid {type_} key length: {len(k)}"
raise BTClibValueError(err_msg)
return Tx.parse(v)
def test_dataclasses_json_dict() -> None:
fname = "d4f3c2c3c218be868c77ae31bedb497e2f908d6ee5bbbe91e4933e6da680c970.bin"
filename = path.join(path.dirname(__file__), "_data", fname)
with open(filename, "rb") as binary_file_:
temp = Tx.parse(binary_file_.read())
tx_out_data = temp.vout[0]
# dataclass
assert isinstance(tx_out_data, TxOut)
# Tx to/from dict
tx_out_dict = tx_out_data.to_dict()
assert isinstance(tx_out_dict, dict)
assert tx_out_data == TxOut.from_dict(tx_out_dict)
datadir = path.join(path.dirname(__file__), "_generated_files")
# Tx dict to/from dict file
filename = path.join(datadir, "tx_out.json")
with open(filename, "w", encoding="ascii") as file_:
json.dump(tx_out_dict, file_, indent=4)
with open(filename, "r", encoding="ascii") as file_:
tx_dict2 = json.load(file_)
assert isinstance(tx_dict2, dict)
assert tx_out_dict == tx_dict2
def test_valid_taproot_script_path() -> None:
tx_data = "26dc279d02d8b1a203b653fc4e0f27f408432f3f540136d33f8f930eaeba655910095142980402000000fd697cd4eb5278f1e34545cd57b6670df806fa3a0a064fd8e385a19f1a53d9ce8d8971a30f02000000378d5fb502335dbe02000000001976a9140053a23441c8478caac4c6b769c51f8476cd4b4b88ac58020000000000001976a914f2aae94a43e0d173354201d7832b46c5269c8a2488ac4a08671e"
prevouts_data = [
"91ca4c010000000017a9145658b58602cdf7b7e962cfe44e024cb0e366f27087",
"cb127401000000002251201ebe8b90363bd097aa9f352c8b21914e1886bc09fe9e70c09f33ef2d2abdf4bc",
]
witness_data = [
"9675a9982c6398ea9d441cb7a943bcd6ff033cc3a2e01a0178a7d3be4575be863871c6bf3eef5ecd34721c784259385ca9101c3a313e010ac942c99de05aaaa602",
"5799cf4b193b730fb99580b186f7477c2cca4d28957326f6f1a5d14116438530e7ec0ce1cd465ad96968ae8a6a09d4d37a060a115919f56fcfebe7b2277cc2df5cc08fb6cda9105ee2512b2e22635aba",
"7520c7b5db9562078049719228db2ac80cb9643ec96c8055aa3b29c2c03d4d99edb0ac",
"c1a7957acbaaf7b444c53d9e0c9436e8a8a3247fd515095d66ddf6201918b40a3668f9a4ccdffcf778da624dca2dda0b08e763ec52fd4ad403ec7563a3504d0cc168b9a77a410029e01dac89567c9b2e6cd726e840351df3f2f58fefe976200a19244150d04153909f660184d656ee95fa7bf8e1d4ec83da1fca34f64bc279b76d257ec623e08baba2cfa4ea9e99646e88f1eb1668c00c0f15b7443c8ab83481611cc3ae85eb89a7bfc40067eb1d2e6354a32426d0ce710e88bc4cc0718b99c325509c9d02a6a980d675a8969be10ee9bef82cafee2fc913475667ccda37b1bc7f13f64e56c449c532658ba8481631c02ead979754c809584a875951619cec8fb040c33f06468ae0266cd8693d6a64cea5912be32d8de95a6da6300b0c50fdcd6001ea41126e7b7e5280d455054a816560028f5ca53c9a50ee52f10e15c5337315bad1f5277acb109a1418649dc6ead2fe14699742fee7182f2f15e54279c7d932ed2799d01d73c97e68bbc94d6f7f56ee0a80efd7c76e3169e10d1a1ba3b5f1eb02369dc43af687461c7a2a3344d13eb5485dca29a67f16b4cb988923060fd3b65d0f0352bb634bcc44f2fe668836dcd0f604150049835135dc4b4fbf90fb334b3938a1f137eb32f047c65b85e6c1173b890b6d0162b48b186d1f1af8521945924ac8ac8efec321bf34f1d4b3d4a304a10313052c652d53f6ecb8a55586614e8950cde9ab6fe8e22802e93b3b9139112250b80ebc589aba231af535bb20f7eeec2e412f698c17f3fdc0a2e20924a5e38b21a628a9e3b2a61e35958e60c7f5087c",
]
tx = Tx.parse(tx_data)
prevouts = [TxOut.parse(prevout) for prevout in prevouts_data]
index = 1
witness = Witness(witness_data)
tx.vin[index].script_witness = witness
sighash_type = 0 # all
signature = witness.stack[0][:64]
if len(witness.stack[0]) == 65:
sighash_type = witness.stack[0][-1]
assert sighash_type != 0
msg_hash = sig_hash.from_tx(prevouts, tx, index, sighash_type)
tapscript = parse(witness.stack[-2])
pub_key = bytes.fromhex(str(tapscript[1]))
ssa.assert_as_valid_(msg_hash, pub_key, signature)
def test_dataclasses_json_dict() -> None:
fname = "d4f3c2c3c218be868c77ae31bedb497e2f908d6ee5bbbe91e4933e6da680c970.bin"
filename = path.join(path.dirname(__file__), "_data", fname)
with open(filename, "rb") as binary_file_:
temp = Tx.parse(binary_file_.read())
tx_in = temp.vin[0]
# TxIn dataclass
assert isinstance(tx_in, TxIn)
# TxIn dataclass to dict
tx_in_dict = tx_in.to_dict()
assert isinstance(tx_in_dict, dict)
# TxIn dataclass dict to file
datadir = path.join(path.dirname(__file__), "_generated_files")
filename = path.join(datadir, "tx_in.json")
with open(filename, "w") as file_:
json.dump(tx_in_dict, file_, indent=4)
# TxIn dataclass dict from file
with open(filename, "r") as file_:
tx_dict2 = json.load(file_)
assert isinstance(tx_dict2, dict)
assert tx_in_dict == tx_dict2
# TxIn dataclass from dict
tx_in2 = TxIn.from_dict(tx_in_dict)
assert isinstance(tx_in2, TxIn)
assert tx_in == tx_in2
def test_dataclasses_json_dict_out_point() -> None:
fname = "d4f3c2c3c218be868c77ae31bedb497e2f908d6ee5bbbe91e4933e6da680c970.bin"
filename = path.join(path.dirname(__file__), "_data", fname)
with open(filename, "rb") as binary_file_:
temp = Tx.parse(binary_file_.read())
out_point_data = temp.vin[0].prev_out
# dataclass
assert isinstance(out_point_data, OutPoint)
# Tx to/from dict
out_point_dict = out_point_data.to_dict()
assert isinstance(out_point_dict, dict)
assert out_point_data == OutPoint.from_dict(out_point_dict)
datadir = path.join(path.dirname(__file__), "_generated_files")
# Tx dict to/from dict file
filename = path.join(datadir, "out_point.json")
with open(filename, "w") as file_:
json.dump(out_point_dict, file_, indent=4)
with open(filename, "r") as file_:
out_point_dict2 = json.load(file_)
assert isinstance(out_point_dict2, dict)
assert out_point_dict == out_point_dict2
def test_legacy_p2pkh() -> None:
pub_key = "04280c8f66bf2ccaeb3f60a19ad4a06365f8bd6178aab0e709df2173df8f553366549aec336aae8742a84702b6c7c3052d89f5d76d535ec3716e72187956351613"
signature = "3045022100ea43c4800d1a860ec89b5273898a146cfb01d34ff4c364d24a110c480d0e3f7502201c82735577f932f1ca8e1c54bf653e0f8e74e408fe83666bc85cac4472ec950801"
script_sig = serialize([signature, pub_key])
out_point = OutPoint(
"d8343a35ba951684f2969eafe833d9e6fe436557b9707ae76802875952e860fc", 1)
tx_in = TxIn(out_point, script_sig, 0xFFFFFFFF)
tx_out1 = TxOut(
2017682,
bytes.fromhex("76a91413bd20236d0da56492c325dce289b4da35b4b5bd88ac"))
tx_out2 = TxOut(
1049154982,
bytes.fromhex("76a914da169b45781ca210f8c11617ba66bd843da76b1688ac"))
tx = Tx(1, 0, [tx_in], [tx_out1, tx_out2])
script_pub_key = serialize([
"OP_DUP",
"OP_HASH160",
"82ac30f58baf99ec9d14e6181eee076f4e27f69c",
"OP_EQUALVERIFY",
"OP_CHECKSIG",
])
utxo = TxOut(1051173696, script_pub_key)
hash_ = sig_hash.from_tx([utxo], tx, 0, sig_hash.ALL)
assert dsa.verify_(hash_, pub_key, bytes.fromhex(signature)[:-1])
def test_p2pk_anyonecanpay() -> None:
pub_key = "048282263212c609d9ea2a6e3e172de238d8c39cabd5ac1ca10646e23fd5f5150811f8a8098557dfe45e8256e830b60ace62d613ac2f7b17bed31b6eaff6e26caf"
signature = "304402204710a85181663b32d25c70ec2bbd14adff5ddfff6cb50d09e155ef5f541fc86c0220056b0cc949be9386ecc5f6c2ac0493269031dbb185781db90171b54ac127790281"
script_pub_key = serialize([pub_key, "OP_CHECKSIG"])
script_sig = serialize([signature])
founding_tx_script = serialize(["OP_0", "OP_0"])
tx_in = TxIn(OutPoint(b"\x00" * 32, 0xFFFFFFFF), founding_tx_script,
0xFFFFFFFF)
funding_tx = Tx(1, 0, [tx_in], [TxOut(0, script_pub_key)])
tx_in = TxIn(OutPoint(funding_tx.id, 0), script_sig, 0xFFFFFFFF)
receiving_tx = Tx(1, 0, [tx_in], [TxOut(0, b"")])
hash_ = sig_hash.from_tx(funding_tx.vout, receiving_tx, 0,
sig_hash.ANYONECANPAY | sig_hash.ALL)
assert dsa.verify_(hash_, pub_key, bytes.fromhex(signature)[:-1])
def test_creation() -> None:
psbt_str = "cHNidP8BAJoCAAAAAljoeiG1ba8MI76OcHBFbDNvfLqlyHV5JPVFiHuyq911AAAAAAD/////g40EJ9DsZQpoqka7CwmK6kQiwHGyyng1Kgd5WdB86h0BAAAAAP////8CcKrwCAAAAAAWABTYXCtx0AYLCcmIauuBXlCZHdoSTQDh9QUAAAAAFgAUAK6pouXw+HaliN9VRuh0LR2HAI8AAAAAAAAAAAA="
psbt = Psbt.b64decode(psbt_str)
assert psbt.b64encode() == psbt_str
output_1 = TxOut(
149990000,
ScriptPubKey("0014d85c2b71d0060b09c9886aeb815e50991dda124d"))
output_2 = TxOut(
100000000,
ScriptPubKey("001400aea9a2e5f0f876a588df5546e8742d1d87008f"))
input_1 = TxIn(
OutPoint(
"75ddabb27b8845f5247975c8a5ba7c6f336c4570708ebe230caf6db5217ae858",
0),
b"",
0xFFFFFFFF,
)
input_2 = TxIn(
OutPoint(
"1dea7cd05979072a3578cab271c02244ea8a090bbb46aa680a65ecd027048d83",
1),
b"",
0xFFFFFFFF,
)
transaction = Tx(2, 0, [input_1, input_2], [output_1, output_2])
psbt_from_tx_ = Psbt.from_tx(transaction)
assert psbt_from_tx_ == psbt
def test_output_scripts_serialization() -> None:
input_1 = TxIn(
OutPoint(
"75ddabb27b8845f5247975c8a5ba7c6f336c4570708ebe230caf6db5217ae858",
0),
b"",
0xFFFFFFFF,
)
output_1 = TxOut(
149990000,
bytes.fromhex("a914256b3a9ae8145e5094329537dd4d7a25dbc9452087"))
output_2 = TxOut(
100000000,
bytes.fromhex("001400aea9a2e5f0f876a588df5546e8742d1d87008f"))
tx = Tx(2, 0, [input_1], [output_1, output_2])
psbt = Psbt.from_tx(tx)
# p2sh-p2wsh
psbt.outputs[0].redeem_script = bytes.fromhex(
"003BD89EE628E6EB745F99DF1E4AEF64A0DAA814850DAA509F30C0F472E0563C7A")
psbt.outputs[0].witness_script = bytes.fromhex(
"522103dcea327ff7b2b4449413d9dc24cef0cc9e7864bad9d6291f3f1a04b639422c312103a2c5199b333adfaed4fea7ab65485b9e23a6cb317ddae7e8c1f2bd673414bdd352ae"
)
assert Psbt.b64decode(psbt.b64encode()) == psbt
def test_native_p2wsh_2() -> None:
tx_bytes = "0100000002e9b542c5176808107ff1df906f46bb1f2583b16112b95ee5380665ba7fcfc0010000000000ffffffff80e68831516392fcd100d186b3c2c7b95c80b53c77e77c35ba03a66b429a2a1b0000000000ffffffff0280969800000000001976a914de4b231626ef508c9a74a8517e6783c0546d6b2888ac80969800000000001976a9146648a8cd4531e1ec47f35916de8e259237294d1e88ac00000000"
tx = Tx.parse(tx_bytes)
tx.vin[0].script_witness = Witness([
"0063ab68210392972e2eb617b2388771abe27235fd5ac44af8e61693261550447a4c3e39da98ac"
])
tx.vin[1].script_witness = Witness([
"5163ab68210392972e2eb617b2388771abe27235fd5ac44af8e61693261550447a4c3e39da98ac"
])
previous_txout_1 = TxOut(
16777215,
"0020ba468eea561b26301e4cf69fa34bde4ad60c81e70f059f045ca9a79931004a4d")
previous_txout_2 = TxOut(
16777215,
"0020d9bbfbe56af7c4b7f960a70d7ea107156913d9e5a26b0a71429df5e097ca6537")
hash_ = sig_hash.from_tx(
[previous_txout_1, previous_txout_2],
tx,
0,
sig_hash.ANYONECANPAY | sig_hash.SINGLE,
)
assert hash_ == bytes.fromhex(
"e9071e75e25b8a1e298a72f0d2e9f4f95a0f5cdf86a533cda597eb402ed13b3a")
script_ = sig_hash.witness_v0_script(tx.vin[1].script_witness.stack[-1])[1]
hash_ = sig_hash.segwit_v0(
script_,
tx,
1,
sig_hash.ANYONECANPAY | sig_hash.SINGLE,
previous_txout_2.value,
)
assert hash_ == bytes.fromhex(
"cd72f1f1a433ee9df816857fad88d8ebd97e09a75cd481583eb841c330275e54")
def test_wrapped_p2wsh() -> None:
tx_bytes = "010000000136641869ca081e70f394c6948e8af409e18b619df2ed74aa106c1ca29787b96e0100000000ffffffff0200e9a435000000001976a914389ffce9cd9ae88dcc0631e88a821ffdbe9bfe2688acc0832f05000000001976a9147480a33f950689af511e6e84c138dbbd3c3ee41588ac00000000"
tx = Tx.parse(tx_bytes)
stack = [
"56210307b8ae49ac90a048e9b53357a2354b3334e9c8bee813ecb98e99a7e07e8c3ba32103b28f0c28bfab54554ae8c658ac5c3e0ce6e79ad336331f78c428dd43eea8449b21034b8113d703413d57761b8b9781957b8c0ac1dfe69f492580ca4195f50376ba4a21033400f6afecb833092a9a21cfdf1ed1376e58c5d1f47de74683123987e967a8f42103a6d48b1131e94ba04d9737d61acdaa1322008af9602b3b14862c07a1789aac162102d8b661b0b3302ee2f162b09e07a55ad5dfbe673a9f01d9f0c19617681024306b56ae"
]
tx.vin[0].script_witness = Witness(stack)
utxo = TxOut(
987654321,
"0020a16b5755f7f6f96dbd65f5f0d6ab9418b89af4b1f14a1bb8a09062c35f0dcb54",
)
hash_ = sig_hash.from_tx([utxo], tx, 0, sig_hash.ALL)
assert hash_ == bytes.fromhex(
"185c0be5263dce5b4bb50a047973c1b6272bfbd0103a89444597dc40b248ee7c")
hash_ = sig_hash.from_tx([utxo], tx, 0, sig_hash.NONE)
assert hash_ == bytes.fromhex(
"e9733bc60ea13c95c6527066bb975a2ff29a925e80aa14c213f686cbae5d2f36")
hash_ = sig_hash.from_tx([utxo], tx, 0, sig_hash.SINGLE)
assert hash_ == bytes.fromhex(
"1e1f1c303dc025bd664acb72e583e933fae4cff9148bf78c157d1e8f78530aea")
hash_ = sig_hash.from_tx([utxo], tx, 0,
sig_hash.ANYONECANPAY | sig_hash.ALL)
assert hash_ == bytes.fromhex(
"2a67f03e63a6a422125878b40b82da593be8d4efaafe88ee528af6e5a9955c6e")
hash_ = sig_hash.from_tx([utxo], tx, 0,
sig_hash.ANYONECANPAY | sig_hash.NONE)
assert hash_ == bytes.fromhex(
"781ba15f3779d5542ce8ecb5c18716733a5ee42a6f51488ec96154934e2c890a")
hash_ = sig_hash.from_tx([utxo], tx, 0,
sig_hash.ANYONECANPAY | sig_hash.SINGLE)
assert hash_ == bytes.fromhex(
"511e8e52ed574121fc1b654970395502128263f62662e076dc6baf05c2e6a99b")
def test_native_p2wpkh() -> None:
tx_bytes = "0100000002fff7f7881a8099afa6940d42d1e7f6362bec38171ea3edf433541db4e4ad969f0000000000eeffffffef51e1b804cc89d182d279655c3aa89e815b1b309fe287d9b2b55d57b90ec68a0100000000ffffffff02202cb206000000001976a9148280b37df378db99f66f85c95a783a76ac7a6d5988ac9093510d000000001976a9143bde42dbee7e4dbe6a21b2d50ce2f0167faa815988ac11000000"
tx = Tx.parse(tx_bytes)
utxo = TxOut(600000000, "00141d0f172a0ecb48aee1be1f2687d2963ae33f71a1")
hash_ = sign_hash.from_utxo(utxo, tx, 1, sign_hash.ALL)
assert hash_ == bytes.fromhex(
"c37af31116d1b27caf68aae9e3ac82f1477929014d5b917657d0eb49478cb670")
def from_dict(cls: Type[_Block],
dict_: Mapping[str, Any],
check_validity: bool = True) -> _Block:
return cls(
BlockHeader.from_dict(dict_["header"], False),
[Tx.from_dict(tx, False) for tx in dict_["transactions"]],
check_validity,
)
def test_dataclasses_json_dict() -> None:
fname = "d4f3c2c3c218be868c77ae31bedb497e2f908d6ee5bbbe91e4933e6da680c970.bin"
filename = path.join(path.dirname(__file__), "_data", fname)
with open(filename, "rb") as binary_file_:
tx = Tx.parse(binary_file_.read())
# Tx dataclass
assert isinstance(tx, Tx)
assert tx.is_segwit()
assert any(bool(w) for w in tx.vwitness)
assert any(bool(tx_in.script_witness) for tx_in in tx.vin)
assert tx.vin[0].script_witness
assert tx.vin[0].script_witness.stack
# Tx dataclass to dict
tx_dict = tx.to_dict()
assert isinstance(tx_dict, dict)
assert tx_dict["vin"][0]["txinwitness"]["stack"] # type: ignore
# Tx dataclass dict to file
datadir = path.join(path.dirname(__file__), "_generated_files")
filename = path.join(datadir, "tx.json")
with open(filename, "w") as file_:
json.dump(tx_dict, file_, indent=4)
# Tx dataclass dict from file
with open(filename, "r") as file_:
tx_dict2 = json.load(file_)
assert isinstance(tx_dict2, dict)
assert tx_dict2["vin"][0]["txinwitness"]["stack"] # type: ignore
assert tx_dict == tx_dict2
# Tx dataclass from dict
tx2 = Tx.from_dict(tx_dict)
assert isinstance(tx2, Tx)
assert tx.vin[0] == tx2.vin[0]
assert tx2.vin[0].script_witness
assert tx2.vin[0].script_witness.stack
assert tx2.is_segwit()
assert any(bool(w) for w in tx2.vwitness)
assert any(bool(tx_in.script_witness) for tx_in in tx2.vin)
assert tx == tx2
def test_wrapped_p2wpkh() -> None:
tx_bytes = "0100000001db6b1b20aa0fd7b23880be2ecbd4a98130974cf4748fb66092ac4d3ceb1a54770100000000feffffff02b8b4eb0b000000001976a914a457b684d7f0d539a46a45bbc043f35b59d0d96388ac0008af2f000000001976a914fd270b1ee6abcaea97fea7ad0402e8bd8ad6d77c88ac92040000"
tx = Tx.parse(tx_bytes)
tx.vin[0].script_sig = bytes.fromhex(
"001479091972186c449eb1ded22b78e40d009bdf0089")
utxo = TxOut(1000000000, "a9144733f37cf4db86fbc2efed2500b4f4e49f31202387")
hash_ = sig_hash.from_tx([utxo], tx, 0, sig_hash.ALL)
assert hash_ == bytes.fromhex(
"64f3b0f4dd2bb3aa1ce8566d220cc74dda9df97d8490cc81d89d735c92e59fb6")
def test_p2pkh() -> None:
pub_key = "038282263212C609D9EA2A6E3E172DE238D8C39CABD5AC1CA10646E23FD5F51508"
signature = "304402206E05A6FE23C59196FFE176C9DDC31E73A9885638F9D1328D47C0C703863B8876022076FEB53811AA5B04E0E79F938EB19906CC5E67548BC555A8E8B8B0FC603D840C01"
script_pub_key = serialize([
"OP_DUP",
"OP_HASH160",
"1018853670F9F3B0582C5B9EE8CE93764AC32B93",
"OP_EQUALVERIFY",
"OP_CHECKSIG",
])
script_sig = serialize([signature, pub_key])
founding_tx_script = serialize(["OP_0", "OP_0"])
tx_in = TxIn(OutPoint(b"\x00" * 32, 0xFFFFFFFF), founding_tx_script,
0xFFFFFFFF)
funding_tx = Tx(1, 0, [tx_in], [TxOut(0, script_pub_key)])
tx_in = TxIn(OutPoint(funding_tx.id, 0), script_sig, 0xFFFFFFFF)
receiving_tx = Tx(1, 0, [tx_in], [TxOut(0, b"")])
hash_ = sig_hash.from_tx(funding_tx.vout, receiving_tx, 0, sig_hash.ALL)
assert dsa.verify_(hash_, pub_key, bytes.fromhex(signature)[:-1])
def parse(cls: Type[_Block],
data: BinaryData,
check_validity: bool = True) -> _Block:
"Return a Block by parsing binary data."
stream = bytesio_from_binarydata(data)
header = BlockHeader.parse(stream)
n = var_int.parse(stream)
# TODO: is a block required to have a coinbase tx?
transactions = [Tx.parse(stream) for _ in range(n)]
return cls(header, transactions, check_validity)
def deserialize_tx(
k: bytes, v: bytes, type_: str, include_witness: Optional[bool] = True
) -> Tx:
"Return the dataclass element from its binary representation."
if len(k) != 1:
err_msg = f"invalid {type_} key length: {len(k)}"
raise BTClibValueError(err_msg)
tx = Tx.parse(v)
if not include_witness and tx.serialize(include_witness=False) != v:
raise BTClibValueError("wrong tx serialization format")
return tx
def test_coinbase_block_1() -> None:
coinbase_out = "00f2052a0100000043410496b538e853519c726a2c91e61ec11600ae1390813a627c66fb8be7947be63c52da7589379515d4e0a604f8141781e62294721166bf621e73a82cbf2342c858eeac"
tx_out = TxOut.parse(coinbase_out)
assert tx_out.serialize().hex() == coinbase_out
coinbase_inp = ( # prev_out
"0000000000000000000000000000000000000000000000000000000000000000ffffffff"
"0704ffff001d0104" # script_sig
"ffffffff" # sequence
)
tx_in = TxIn.parse(coinbase_inp)
assert tx_in.serialize().hex() == coinbase_inp
assert tx_in.prev_out.is_coinbase
coinbase = "01000000" "01" + coinbase_inp + "01" + coinbase_out + "00000000"
tx = Tx.parse(coinbase)
assert tx.serialize(include_witness=True).hex() == coinbase
assert tx == Tx.from_dict(tx.to_dict())
assert tx.version == 1
assert tx.lock_time == 0
assert len(tx.vin) == 1
assert len(tx.vout) == 1
assert tx.vin[0].script_sig == tx_in.script_sig
assert tx.vout[0].script_pub_key == tx_out.script_pub_key
tx_id = "0e3e2357e806b6cdb1f70b54c3a3a17b6714ee1f0e68bebb44a74b1efd512098"
assert tx.id.hex() == tx_id
assert tx.id == tx.hash
assert tx.size == 134
assert tx.vsize == tx.size
assert tx.weight == tx.size * 4
assert not tx.is_segwit()
assert not any(bool(w) for w in tx.vwitness)
assert not any(bool(tx_in.script_witness) for tx_in in tx.vin)
assert tx.is_coinbase()
def test_test_vectors() -> None:
fname = "sig_hash_legacy_test_vectors.json"
filename = path.join(path.dirname(__file__), "_data", fname)
with open(filename, "r") as file_:
data = json.load(file_)
data = data[1:] # skip column headers
for raw_tx, raw_script, input_index, hash_type, exp_hash in data:
script_ = sig_hash.legacy_script(raw_script)[0]
# FIXME: separate invalid transaction from the valid ones
tx = Tx.parse(raw_tx, check_validity=False)
if hash_type < 0:
hash_type += 0xFFFFFFFF + 1
actual_hash = sig_hash.legacy(script_, tx, input_index, hash_type)
assert actual_hash == bytes.fromhex(exp_hash)[::-1]
def test_wiki_transaction() -> None:
tx_bytes = "01000000016dbddb085b1d8af75184f0bc01fad58d1266e9b63b50881990e4b40d6aee3629000000008b483045022100f3581e1972ae8ac7c7367a7a253bc1135223adb9a468bb3a59233f45bc578380022059af01ca17d00e41837a1d58e97aa31bae584edec28d35bd96923690913bae9a0141049c02bfc97ef236ce6d8fe5d94013c721e915982acd2b12b65d9b7d59e20a842005f8fc4e02532e873d37b96f09d6d4511ada8f14042f46614a4c70c0f14beff5ffffffff02404b4c00000000001976a9141aa0cd1cbea6e7458a7abad512a9d9ea1afb225e88ac80fae9c7000000001976a9140eab5bea436a0484cfab12485efda0b78b4ecc5288ac00000000"
tx = Tx.parse(tx_bytes)
assert tx.serialize(include_witness=True).hex() == tx_bytes
assert tx == Tx.from_dict(tx.to_dict())
assert tx.version == 1
assert tx.lock_time == 0
assert len(tx.vin) == 1
assert len(tx.vout) == 2
assert tx.vout[0].value == 5000000
assert tx.vout[1].value == 3354000000
tx_id = "d4a73f51ab7ee7acb4cf0505d1fab34661666c461488e58ec30281e2becd93e2"
assert tx.id.hex() == tx_id
assert tx.hash == tx.id
assert tx.size == 258
assert tx.vsize == tx.size
assert tx.weight == tx.size * 4
assert not tx.is_segwit()
assert not any(bool(w) for w in tx.vwitness)
assert not any(bool(tx_in.script_witness) for tx_in in tx.vin)
assert not tx.is_coinbase()
def test_double_witness() -> None:
tx_bytes = "01000000000102322d4f05c3a4f78e97deda01bd8fc5ff96777b62c8f2daa72b02b70fa1e3e1051600000017160014e123a5263695be634abf3ad3456b4bf15f09cc6afffffffffdfee6e881f12d80cbcd6dc54c3fe390670678ebd26c3ae2dd129f41882e3efc25000000171600145946c8c3def6c79859f01b34ad537e7053cf8e73ffffffff02c763ac050000000017a9145ffd6df9bd06dedb43e7b72675388cbfc883d2098727eb180a000000001976a9145f9e96f739198f65d249ea2a0336e9aa5aa0c7ed88ac024830450221009b364c1074c602b2c5a411f4034573a486847da9c9c2467596efba8db338d33402204ccf4ac0eb7793f93a1b96b599e011fe83b3e91afdc4c7ab82d765ce1da25ace01210334d50996c36638265ad8e3cd127506994100dd7f24a5828155d531ebaf736e160247304402200c6dd55e636a2e4d7e684bf429b7800a091986479d834a8d462fbda28cf6f8010220669d1f6d963079516172f5061f923ef90099136647b38cc4b3be2a80b820bdf90121030aa2a1c2344bc8f38b7a726134501a2a45db28df8b4bee2df4428544c62d731400000000"
tx = Tx.parse(tx_bytes)
assert tx.serialize(include_witness=True).hex() == tx_bytes
assert tx == Tx.from_dict(tx.to_dict())
assert tx.version == 1
assert tx.lock_time == 0
assert len(tx.vin) == 2
assert len(tx.vout) == 2
stack1 = [
"30450221009b364c1074c602b2c5a411f4034573a486847da9c9c2467596efba8db338d33402204ccf4ac0eb7793f93a1b96b599e011fe83b3e91afdc4c7ab82d765ce1da25ace01",
"0334d50996c36638265ad8e3cd127506994100dd7f24a5828155d531ebaf736e16",
]
witness1 = Witness(stack1)
assert tx.vin[0].script_witness == witness1
stack2 = [
"304402200c6dd55e636a2e4d7e684bf429b7800a091986479d834a8d462fbda28cf6f8010220669d1f6d963079516172f5061f923ef90099136647b38cc4b3be2a80b820bdf901",
"030aa2a1c2344bc8f38b7a726134501a2a45db28df8b4bee2df4428544c62d7314",
]
witness2 = Witness(stack2)
assert tx.vin[1].script_witness == witness2
tx_id = "a4b76807519aba5740f7865396bc4c5ca0eb8aa7c3744ca2db88fcc9e345424c"
assert tx.id.hex() == tx_id
hash_ = "0936cb8dba90e11345b9c05f457f139ddce4a5329701af4708b2cf4a02d75adb"
assert tx.hash.hex() == hash_
assert tx.size == 421
assert tx.vsize == 259
assert tx.weight == 1033
assert tx.is_segwit()
assert any(bool(w) for w in tx.vwitness)
assert any(bool(tx_in.script_witness) for tx_in in tx.vin)
assert not tx.is_coinbase()
def test_empty_stack() -> None:
utxo = TxOut(
100000000,
serialize([
"OP_1",
"cc71eb30d653c0c3163990c47b976f3fb3f37cccdcbedb169a1dfef58bbfbfaf"
]),
)
tx_in = TxIn(OutPoint(), "", 1, Witness([]))
tx = Tx(vin=[tx_in], vout=[TxOut(100000000, "")])
err_msg = "Empty stack"
with pytest.raises(BTClibValueError, match=err_msg):
sig_hash.from_tx([utxo], tx, 0, 0)
def test_first_transaction() -> None:
tx_bytes = "0100000001c997a5e56e104102fa209c6a852dd90660a20b2d9c352423edce25857fcd3704000000004847304402204e45e16932b8af514961a1d3a1a25fdf3f4f7732e9d624c6c61548ab5fb8cd410220181522ec8eca07de4860a4acdd12909d831cc56cbbac4622082221a8768d1d0901ffffffff0200ca9a3b00000000434104ae1a62fe09c5f51b13905f07f06b99a2f7159b2225f374cd378d71302fa28414e7aab37397f554a7df5f142c21c1b7303b8a0626f1baded5c72a704f7e6cd84cac00286bee0000000043410411db93e1dcdb8a016b49840f8c53bc1eb68a382e97b1482ecad7b148a6909a5cb2e0eaddfb84ccf9744464f82e160bfa9b8b64f9d4c03f999b8643f656b412a3ac00000000"
tx = Tx.parse(tx_bytes)
utxo = TxOut(
value=5000000000,
script_pub_key=bytes.fromhex(
"410411db93e1dcdb8a016b49840f8c53bc1eb68a382e97b1482ecad7b148a6909a5cb2e0eaddfb84ccf9744464f82e160bfa9b8b64f9d4c03f999b8643f656b412a3ac"
),
)
hash_ = sig_hash.from_tx([utxo], tx, 0, sig_hash.ALL)
assert hash_ == bytes.fromhex(
"7a05c6145f10101e9d6325494245adf1297d80f8f38d4d576d57cdba220bcb19")
pub_key = "0411db93e1dcdb8a016b49840f8c53bc1eb68a382e97b1482ecad7b148a6909a5cb2e0eaddfb84ccf9744464f82e160bfa9b8b64f9d4c03f999b8643f656b412a3"
signature = "304402204E45E16932B8AF514961A1D3A1A25FDF3F4F7732E9D624C6C61548AB5FB8CD410220181522EC8ECA07DE4860A4ACDD12909D831CC56CBBAC4622082221A8768D1D0901"
assert dsa.verify_(hash_, pub_key, bytes.fromhex(signature)[:-1])
def test_wrapped_p2tr() -> None:
script = [
"OP_1",
"cc71eb30d653c0c3163990c47b976f3fb3f37cccdcbedb169a1dfef58bbfbfaf",
]
utxo = TxOut(
100000000,
serialize(["OP_HASH160",
hash160(serialize(script)), "OP_EQUAL"]))
tx_in = TxIn(OutPoint(), serialize(script), 1, Witness(["0A" * 32]))
tx = Tx(vin=[tx_in], vout=[TxOut(100000000, "")])
err_msg = "Taproot scripts cannot be wrapped in p2sh"
with pytest.raises(BTClibValueError, match=err_msg):
sig_hash.from_tx([utxo], tx, 0, 0)
def test_sig_hashsingle_bug() -> None:
pub_key = "02D5C25ADB51B61339D2B05315791E21BBE80EA470A49DB0135720983C905AACE0"
signature = "3045022100C9CDD08798A28AF9D1BAF44A6C77BCC7E279F47DC487C8C899911BC48FEAFFCC0220503C5C50AE3998A733263C5C0F7061B483E2B56C4C41B456E7D2F5A78A74C07703"
script_pub_key = serialize([
"OP_DUP",
"OP_HASH160",
"5b6462475454710f3c22f5fdf0b40704c92f25c3",
"OP_EQUALVERIFY",
"OP_CHECKSIGVERIFY",
"OP_1",
])
utxo = TxOut(0, script_pub_key)
tx_bytes = "01000000020002000000000000000000000000000000000000000000000000000000000000000000000151ffffffff0001000000000000000000000000000000000000000000000000000000000000000000006b483045022100c9cdd08798a28af9d1baf44a6c77bcc7e279f47dc487c8c899911bc48feaffcc0220503c5c50ae3998a733263c5c0f7061b483e2b56c4c41b456e7d2f5a78a74c077032102d5c25adb51b61339d2b05315791e21bbe80ea470a49db0135720983c905aace0ffffffff010000000000000000015100000000"
tx = Tx.parse(tx_bytes)
hash_ = sig_hash.from_tx([TxOut(0, ""), utxo], tx, 1, sig_hash.SINGLE)
assert dsa.verify_(hash_, pub_key, bytes.fromhex(signature)[:-1])
def test_psbt() -> None:
prev_out = OutPoint(
"9dcfdb5836ecfe146bdaa896605ba21222f83cd014dd47adde14fab2aba7de9b", 1)
script_sig = b""
sequence = 0xFFFFFFFF
tx_in = TxIn(prev_out, script_sig, sequence)
tx_out1 = TxOut(2500000, "a914f987c321394968be164053d352fc49763b2be55c87")
tx_out2 = TxOut(
6381891,
"0020701a8d401c84fb13e6baf169d59684e17abd9fa216c8cc5b9fc63d622ff8c58d")
version = 1
lock_time = 0
tx = Tx(version, lock_time, [tx_in], [tx_out1, tx_out2])
psbt = Psbt.from_tx(tx)
assert psbt == Psbt.parse(psbt.serialize())
assert psbt == Psbt.from_dict(psbt.to_dict())
