Changeset View
Changeset View
Standalone View
Standalone View
test/functional/abc-schnorrmultisig.py
| Show First 20 Lines • Show All 62 Lines • ▼ Show 20 Lines | |||||
| class SchnorrMultisigTest(BitcoinTestFramework): | class SchnorrMultisigTest(BitcoinTestFramework): | ||||
| def set_test_params(self): | def set_test_params(self): | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| self.block_heights = {} | self.block_heights = {} | ||||
| self.extra_args = [["-acceptnonstdtxn=1"]] | self.extra_args = [["-acceptnonstdtxn=1"]] | ||||
| def bootstrap_p2p(self, *, num_connections=1): | def reconnect_p2p(self): | ||||
| """Add a P2P connection to the node. | |||||
| Helper to connect and wait for version handshake.""" | |||||
| for _ in range(num_connections): | |||||
| self.nodes[0].add_p2p_connection(P2PDataStore()) | |||||
| def reconnect_p2p(self, **kwargs): | |||||
| """Tear down and bootstrap the P2P connection to the node. | """Tear down and bootstrap the P2P connection to the node. | ||||
| The node gets disconnected several times in this test. This helper | The node gets disconnected several times in this test. This helper | ||||
| method reconnects the p2p and restarts the network thread.""" | method reconnects the p2p and restarts the network thread.""" | ||||
| self.nodes[0].disconnect_p2ps() | self.nodes[0].disconnect_p2ps() | ||||
| self.bootstrap_p2p(**kwargs) | self.nodes[0].add_p2p_connection(P2PDataStore()) | ||||
| def getbestblock(self, node): | def getbestblock(self, node): | ||||
| """Get the best block. Register its height so we can use build_block.""" | """Get the best block. Register its height so we can use build_block.""" | ||||
| block_height = node.getblockcount() | block_height = node.getblockcount() | ||||
| blockhash = node.getblockhash(block_height) | blockhash = node.getblockhash(block_height) | ||||
| block = FromHex(CBlock(), node.getblock(blockhash, 0)) | block = FromHex(CBlock(), node.getblock(blockhash, 0)) | ||||
| block.calc_sha256() | block.calc_sha256() | ||||
| self.block_heights[block.sha256] = block_height | self.block_heights[block.sha256] = block_height | ||||
| Show All 15 Lines | def build_block(self, parent, transactions=(), nTime=None): | ||||
| block.solve() | block.solve() | ||||
| self.block_heights[block.sha256] = block_height | self.block_heights[block.sha256] = block_height | ||||
| return block | return block | ||||
| def check_for_ban_on_rejected_tx(self, tx, reject_reason=None): | def check_for_ban_on_rejected_tx(self, tx, reject_reason=None): | ||||
| """Check we are disconnected when sending a txn that the node rejects. | """Check we are disconnected when sending a txn that the node rejects. | ||||
| (Can't actually get banned, since bitcoind won't ban local peers.)""" | (Can't actually get banned, since bitcoind won't ban local peers.)""" | ||||
| self.nodes[0].p2p.send_txs_and_test( | self.nodes[0].p2ps[0].send_txs_and_test( | ||||
| [tx], self.nodes[0], success=False, expect_disconnect=True, reject_reason=reject_reason) | [tx], self.nodes[0], success=False, expect_disconnect=True, reject_reason=reject_reason) | ||||
| self.reconnect_p2p() | self.reconnect_p2p() | ||||
| def check_for_ban_on_rejected_block(self, block, reject_reason=None): | def check_for_ban_on_rejected_block(self, block, reject_reason=None): | ||||
| """Check we are disconnected when sending a block that the node rejects. | """Check we are disconnected when sending a block that the node rejects. | ||||
| (Can't actually get banned, since bitcoind won't ban local peers.)""" | (Can't actually get banned, since bitcoind won't ban local peers.)""" | ||||
| self.nodes[0].p2p.send_blocks_and_test( | self.nodes[0].p2ps[0].send_blocks_and_test( | ||||
| [block], self.nodes[0], success=False, reject_reason=reject_reason, expect_disconnect=True) | [block], self.nodes[0], success=False, reject_reason=reject_reason, expect_disconnect=True) | ||||
| self.reconnect_p2p() | self.reconnect_p2p() | ||||
| def run_test(self): | def run_test(self): | ||||
| node, = self.nodes | node, = self.nodes | ||||
| self.bootstrap_p2p() | self.nodes[0].add_p2p_connection(P2PDataStore()) | ||||
| tip = self.getbestblock(node) | tip = self.getbestblock(node) | ||||
| self.log.info("Create some blocks with OP_1 coinbase for spending.") | self.log.info("Create some blocks with OP_1 coinbase for spending.") | ||||
| blocks = [] | blocks = [] | ||||
| for _ in range(10): | for _ in range(10): | ||||
| tip = self.build_block(tip) | tip = self.build_block(tip) | ||||
| blocks.append(tip) | blocks.append(tip) | ||||
| node.p2p.send_blocks_and_test(blocks, node, success=True) | node.p2ps[0].send_blocks_and_test(blocks, node, success=True) | ||||
| spendable_outputs = [block.vtx[0] for block in blocks] | spendable_outputs = [block.vtx[0] for block in blocks] | ||||
| self.log.info("Mature the blocks and get out of IBD.") | self.log.info("Mature the blocks and get out of IBD.") | ||||
| node.generatetoaddress(100, node.get_deterministic_priv_key().address) | node.generatetoaddress(100, node.get_deterministic_priv_key().address) | ||||
| tip = self.getbestblock(node) | tip = self.getbestblock(node) | ||||
| self.log.info("Setting up spends to test and mining the fundings.") | self.log.info("Setting up spends to test and mining the fundings.") | ||||
| ▲ Show 20 Lines • Show All 47 Lines • ▼ Show 20 Lines | def run_test(self): | ||||
| # This is invalid. | # This is invalid. | ||||
| schnorr0tx = create_fund_and_spend_tx(OP_0, 'schnorr') | schnorr0tx = create_fund_and_spend_tx(OP_0, 'schnorr') | ||||
| # This is valid. | # This is valid. | ||||
| schnorr1tx = create_fund_and_spend_tx(OP_1, 'schnorr') | schnorr1tx = create_fund_and_spend_tx(OP_1, 'schnorr') | ||||
| tip = self.build_block(tip, fundings) | tip = self.build_block(tip, fundings) | ||||
| node.p2p.send_blocks_and_test([tip], node) | node.p2ps[0].send_blocks_and_test([tip], node) | ||||
| self.log.info("Send a legacy ECDSA multisig into mempool.") | self.log.info("Send a legacy ECDSA multisig into mempool.") | ||||
| node.p2p.send_txs_and_test([ecdsa0tx], node) | node.p2ps[0].send_txs_and_test([ecdsa0tx], node) | ||||
| assert_equal(node.getrawmempool(), [ecdsa0tx.hash]) | assert_equal(node.getrawmempool(), [ecdsa0tx.hash]) | ||||
| self.log.info("Trying to mine a non-null-dummy ECDSA.") | self.log.info("Trying to mine a non-null-dummy ECDSA.") | ||||
| self.check_for_ban_on_rejected_block( | self.check_for_ban_on_rejected_block( | ||||
| self.build_block(tip, [ecdsa1tx]), BADINPUTS_ERROR) | self.build_block(tip, [ecdsa1tx]), BADINPUTS_ERROR) | ||||
| self.log.info( | self.log.info( | ||||
| "If we try to submit it by mempool or RPC, it is rejected and we are banned") | "If we try to submit it by mempool or RPC, it is rejected and we are banned") | ||||
| assert_raises_rpc_error(-26, ECDSA_NULLDUMMY_ERROR, | assert_raises_rpc_error(-26, ECDSA_NULLDUMMY_ERROR, | ||||
| node.sendrawtransaction, ToHex(ecdsa1tx)) | node.sendrawtransaction, ToHex(ecdsa1tx)) | ||||
| self.check_for_ban_on_rejected_tx( | self.check_for_ban_on_rejected_tx( | ||||
| ecdsa1tx, ECDSA_NULLDUMMY_ERROR) | ecdsa1tx, ECDSA_NULLDUMMY_ERROR) | ||||
| self.log.info( | self.log.info( | ||||
| "Submitting a Schnorr-multisig via net, and mining it in a block") | "Submitting a Schnorr-multisig via net, and mining it in a block") | ||||
| node.p2p.send_txs_and_test([schnorr1tx], node) | node.p2ps[0].send_txs_and_test([schnorr1tx], node) | ||||
| assert_equal(set(node.getrawmempool()), { | assert_equal(set(node.getrawmempool()), { | ||||
| ecdsa0tx.hash, schnorr1tx.hash}) | ecdsa0tx.hash, schnorr1tx.hash}) | ||||
| tip = self.build_block(tip, [schnorr1tx]) | tip = self.build_block(tip, [schnorr1tx]) | ||||
| node.p2p.send_blocks_and_test([tip], node) | node.p2ps[0].send_blocks_and_test([tip], node) | ||||
| self.log.info( | self.log.info( | ||||
| "That legacy ECDSA multisig is still in mempool, let's mine it") | "That legacy ECDSA multisig is still in mempool, let's mine it") | ||||
| assert_equal(node.getrawmempool(), [ecdsa0tx.hash]) | assert_equal(node.getrawmempool(), [ecdsa0tx.hash]) | ||||
| tip = self.build_block(tip, [ecdsa0tx]) | tip = self.build_block(tip, [ecdsa0tx]) | ||||
| node.p2p.send_blocks_and_test([tip], node) | node.p2ps[0].send_blocks_and_test([tip], node) | ||||
| assert_equal(node.getrawmempool(), []) | assert_equal(node.getrawmempool(), []) | ||||
| self.log.info( | self.log.info( | ||||
| "Trying Schnorr in legacy multisig is invalid and banworthy.") | "Trying Schnorr in legacy multisig is invalid and banworthy.") | ||||
| self.check_for_ban_on_rejected_tx( | self.check_for_ban_on_rejected_tx( | ||||
| schnorr0tx, SCHNORR_LEGACY_MULTISIG_ERROR) | schnorr0tx, SCHNORR_LEGACY_MULTISIG_ERROR) | ||||
| self.check_for_ban_on_rejected_block( | self.check_for_ban_on_rejected_block( | ||||
| self.build_block(tip, [schnorr0tx]), BADINPUTS_ERROR) | self.build_block(tip, [schnorr0tx]), BADINPUTS_ERROR) | ||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| SchnorrMultisigTest().main() | SchnorrMultisigTest().main() | ||||