coldcard: avoid creating keystore with testnet/mainnet mixed up

fixes #6722

electrum/base_wizard.py

@@ -480,6 +480,12 @@ class BaseWizard(Logger):
             'label': label,
             'soft_device_id': soft_device_id,
         }
+        try:
+            client.manipulate_keystore_dict_during_wizard_setup(d)
+        except Exception as e:
+            self.logger.exception('')
+            self.show_error(e)
+            raise ChooseHwDeviceAgain()
         k = hardware_keystore(d)
         self.on_keystore(k)
 

electrum/plugins/coldcard/coldcard.py

@@ -142,19 +142,19 @@ class CKCCClient(HardwareClientBase):
         else:
             lab = 'Coldcard ' + xfp2str(self.dev.master_fingerprint)
 
-        # Hack zone: during initial setup I need the xfp and master xpub but 
+        return lab
-        # very few objects are passed between the various steps of base_wizard.
+
-        # Solution: return a string with some hidden metadata
+    def manipulate_keystore_dict_during_wizard_setup(self, d: dict):
-        # - see <https://stackoverflow.com/questions/7172772/abc-for-string>
+        master_xpub = self.dev.master_xpub
-        # - needs to work w/ deepcopy
+        if master_xpub is not None:
-        class LabelStr(str):
+            try:
-            def __new__(cls, s, xfp=None, xpub=None):
+                node = BIP32Node.from_xkey(master_xpub)
-                self = super().__new__(cls, str(s))
+            except InvalidMasterKeyVersionBytes:
-                self.xfp = getattr(s, 'xfp', xfp)
+                raise UserFacingException(
-                self.xpub = getattr(s, 'xpub', xpub)
+                    _('Invalid xpub magic. Make sure your {} device is set to the correct chain.').format(self.device) + ' ' +
-                return self
+                    _('You might have to unplug and plug it in again.')
-
+                ) from None
-        return LabelStr(lab, self.dev.master_fingerprint, self.dev.master_xpub)
+            d['ckcc_xpub'] = master_xpub
 
     @runs_in_hwd_thread
     def has_usable_connection_with_device(self):
@@ -262,8 +262,7 @@ class Coldcard_KeyStore(Hardware_KeyStore):
         # we need to know at least the fingerprint of the master xpub to verify against MiTM
         # - device reports these value during encryption setup process
         # - full xpub value now optional
-        lab = d['label']
+        self.ckcc_xpub = d.get('ckcc_xpub', None)
-        self.ckcc_xpub = getattr(lab, 'xpub', None) or d.get('ckcc_xpub', None)
 
     def dump(self):
         # our additions to the stored data about keystore -- only during creation?

electrum/plugins/hw_wallet/plugin.py

@@ -266,6 +266,13 @@ class HardwareClientBase:
         """
         return None
 
+    def manipulate_keystore_dict_during_wizard_setup(self, d: dict) -> None:
+        """Called during wallet creation in the wizard, before the keystore
+        is constructed for the first time. 'd' is the dict that will be
+        passed to the keystore constructor.
+        """
+        pass
+
 
 class HardwareHandlerBase:
     """An interface between the GUI and the device handling logic for handling I/O."""