Changeset View
Changeset View
Standalone View
Standalone View
test/functional/abc-replay-protection.py
| Show All 13 Lines | |||||
| from test_framework.comptool import TestManager, TestInstance, RejectResult | from test_framework.comptool import TestManager, TestInstance, RejectResult | ||||
| from test_framework.blocktools import * | from test_framework.blocktools import * | ||||
| import time | import time | ||||
| from test_framework.key import CECKey | from test_framework.key import CECKey | ||||
| from test_framework.script import * | from test_framework.script import * | ||||
| # far into the future | # far into the future | ||||
| REPLAY_PROTECTION_START_TIME = 2000000000 | REPLAY_PROTECTION_START_TIME = 2000000000 | ||||
| MAGNETIC_ANOMALY_START_TIME = 4000000000 | |||||
| # Error due to invalid signature | # Error due to invalid signature | ||||
| INVALID_SIGNATURE_ERROR = b'mandatory-script-verify-flag-failed (Signature must be zero for failed CHECK(MULTI)SIG operation)' | INVALID_SIGNATURE_ERROR = b'mandatory-script-verify-flag-failed (Signature must be zero for failed CHECK(MULTI)SIG operation)' | ||||
| RPC_INVALID_SIGNATURE_ERROR = "16: " + \ | RPC_INVALID_SIGNATURE_ERROR = "16: " + \ | ||||
| INVALID_SIGNATURE_ERROR.decode("utf-8") | INVALID_SIGNATURE_ERROR.decode("utf-8") | ||||
| class PreviousSpendableOutput(object): | class PreviousSpendableOutput(object): | ||||
| def __init__(self, tx=CTransaction(), n=-1): | def __init__(self, tx=CTransaction(), n=-1): | ||||
| self.tx = tx | self.tx = tx | ||||
| self.n = n # the output we're spending | self.n = n # the output we're spending | ||||
| class ReplayProtectionTest(ComparisonTestFramework): | class ReplayProtectionTest(ComparisonTestFramework): | ||||
| def set_test_params(self): | def set_test_params(self): | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
| self.block_heights = {} | self.block_heights = {} | ||||
| self.tip = None | self.tip = None | ||||
| self.blocks = {} | self.blocks = {} | ||||
| self.extra_args = [['-whitelist=127.0.0.1', | self.extra_args = [['-whitelist=127.0.0.1', | ||||
| "-magneticanomalyactivationtime=%d" % MAGNETIC_ANOMALY_START_TIME, | |||||
| "-replayprotectionactivationtime=%d" % REPLAY_PROTECTION_START_TIME]] | "-replayprotectionactivationtime=%d" % REPLAY_PROTECTION_START_TIME]] | ||||
| def run_test(self): | def run_test(self): | ||||
| self.test = TestManager(self, self.options.tmpdir) | self.test = TestManager(self, self.options.tmpdir) | ||||
| self.test.add_all_connections(self.nodes) | self.test.add_all_connections(self.nodes) | ||||
| # Start up network handling in another thread | # Start up network handling in another thread | ||||
| NetworkThread().start() | NetworkThread().start() | ||||
| self.nodes[0].setmocktime(REPLAY_PROTECTION_START_TIME) | self.nodes[0].setmocktime(REPLAY_PROTECTION_START_TIME) | ||||
| ▲ Show 20 Lines • Show All 49 Lines • ▼ Show 20 Lines | def get_tests(self): | ||||
| self.tip = self.blocks[number] | self.tip = self.blocks[number] | ||||
| # adds transactions to the block and updates state | # adds transactions to the block and updates state | ||||
| def update_block(block_number, new_transactions): | def update_block(block_number, new_transactions): | ||||
| [tx.rehash() for tx in new_transactions] | [tx.rehash() for tx in new_transactions] | ||||
| block = self.blocks[block_number] | block = self.blocks[block_number] | ||||
| block.vtx.extend(new_transactions) | block.vtx.extend(new_transactions) | ||||
| old_sha256 = block.sha256 | old_sha256 = block.sha256 | ||||
| make_conform_to_ctor(block) | |||||
deadalnix: It's a good idea to do these in their own patches. It's easy wins that won't get blocked by… | |||||
| block.hashMerkleRoot = block.calc_merkle_root() | block.hashMerkleRoot = block.calc_merkle_root() | ||||
| block.solve() | block.solve() | ||||
| # Update the internal state just like in next_block | # Update the internal state just like in next_block | ||||
| self.tip = block | self.tip = block | ||||
| if block.sha256 != old_sha256: | if block.sha256 != old_sha256: | ||||
| self.block_heights[ | self.block_heights[ | ||||
| block.sha256] = self.block_heights[old_sha256] | block.sha256] = self.block_heights[old_sha256] | ||||
| del self.block_heights[old_sha256] | del self.block_heights[old_sha256] | ||||
| ▲ Show 20 Lines • Show All 164 Lines • Show Last 20 Lines | |||||
It's a good idea to do these in their own patches. It's easy wins that won't get blocked by anything else.