1  #!/usr/bin/env python3
 2  # Copyright (c) 2020-2022 The Bitcoin Core developers
 3  # Distributed under the MIT software license, see the accompanying
 4  # file COPYING or http://www.opensource.org/licenses/mit-license.php.
 5  
 6  from test_framework.test_framework import BitcoinTestFramework
 7  from test_framework.util import assert_raises_rpc_error
 8  
 9  class WalletCrossChain(BitcoinTestFramework):
10      def add_options(self, parser):
11          self.add_wallet_options(parser)
12  
13      def set_test_params(self):
14          self.num_nodes = 2
15          self.setup_clean_chain = True
16  
17      def skip_test_if_missing_module(self):
18          self.skip_if_no_wallet()
19  
20      def setup_network(self):
21          self.add_nodes(self.num_nodes)
22  
23          # Switch node 1 to testnet before starting it.
24          self.nodes[1].chain = 'testnet3'
25          self.nodes[1].extra_args = ['-maxconnections=0', '-prune=550'] # disable testnet sync
26          self.nodes[1].replace_in_config([('regtest=', 'testnet='), ('[regtest]', '[test]')])
27          self.start_nodes()
28  
29      def run_test(self):
30          self.log.info("Creating wallets")
31  
32          node0_wallet = self.nodes[0].datadir_path / 'node0_wallet'
33          node0_wallet_backup = self.nodes[0].datadir_path / 'node0_wallet.bak'
34          self.nodes[0].createwallet(node0_wallet)
35          self.nodes[0].backupwallet(node0_wallet_backup)
36          self.nodes[0].unloadwallet(node0_wallet)
37          node1_wallet = self.nodes[1].datadir_path / 'node1_wallet'
38          node1_wallet_backup = self.nodes[0].datadir_path / 'node1_wallet.bak'
39          self.nodes[1].createwallet(node1_wallet)
40          self.nodes[1].backupwallet(node1_wallet_backup)
41          self.nodes[1].unloadwallet(node1_wallet)
42  
43          self.log.info("Loading/restoring wallets into nodes with a different genesis block")
44  
45          if self.options.descriptors:
46              assert_raises_rpc_error(-18, 'Wallet file verification failed.', self.nodes[0].loadwallet, node1_wallet)
47              assert_raises_rpc_error(-18, 'Wallet file verification failed.', self.nodes[1].loadwallet, node0_wallet)
48              assert_raises_rpc_error(-18, 'Wallet file verification failed.', self.nodes[0].restorewallet, 'w', node1_wallet_backup)
49              assert_raises_rpc_error(-18, 'Wallet file verification failed.', self.nodes[1].restorewallet, 'w', node0_wallet_backup)
50          else:
51              assert_raises_rpc_error(-4, 'Wallet files should not be reused across chains.', self.nodes[0].loadwallet, node1_wallet)
52              assert_raises_rpc_error(-4, 'Wallet files should not be reused across chains.', self.nodes[1].loadwallet, node0_wallet)
53              assert_raises_rpc_error(-4, 'Wallet files should not be reused across chains.', self.nodes[0].restorewallet, 'w', node1_wallet_backup)
54              assert_raises_rpc_error(-4, 'Wallet files should not be reused across chains.', self.nodes[1].restorewallet, 'w', node0_wallet_backup)
55  
56          if not self.options.descriptors:
57              self.log.info("Override cross-chain wallet load protection")
58              self.stop_nodes()
59              self.start_nodes([['-walletcrosschain', '-prune=550']] * self.num_nodes)
60              self.nodes[0].loadwallet(node1_wallet)
61              self.nodes[1].loadwallet(node0_wallet)
62  
63  
64  if __name__ == '__main__':
65      WalletCrossChain().main()