Add register command for BIP388 policies

Author: Sjors

hwilib/_cli.py

@@ -12,6 +12,7 @@
     getdescriptors,
     prompt_pin,
     toggle_passphrase,
+    register,
     restore_device,
     send_pin,
     setup_device,
@@ -59,6 +60,9 @@ def backup_device_handler(args: argparse.Namespace, client: HardwareWalletClient
 def displayaddress_handler(args: argparse.Namespace, client: HardwareWalletClient) -> Dict[str, str]:
     return displayaddress(client, desc=args.desc, path=args.path, addr_type=args.addr_type)
 
+def register_handler(args: argparse.Namespace, client: HardwareWalletClient) -> Dict[str, str]:
+    return register(client, name=args.name, descriptor_template=args.desc, keys_info=args.key)
+
 def enumerate_handler(args: argparse.Namespace) -> List[Dict[str, Any]]:
     return enumerate(password=args.password, expert=args.expert, chain=args.chain, allow_emulators=args.allow_emulators)
 
@@ -202,6 +206,12 @@ def get_parser() -> HWIArgumentParser:
     displayaddr_parser.add_argument("--addr-type", help="The address type to display", type=AddressType.argparse, choices=list(AddressType), default=AddressType.WIT) # type: ignore
     displayaddr_parser.set_defaults(func=displayaddress_handler)
 
+    register_parser = subparsers.add_parser('register', help='Register a BIP388 wallet policy')
+    register_parser.add_argument('--name', help='Name for the policy')
+    register_parser.add_argument('--desc', help='Descriptor template, e.g. tr(musig(@0,@1)')
+    register_parser.add_argument('--key', help='Key information, e.g. [00000000/84h/0h/0h]xpub...', action='append')
+    register_parser.set_defaults(func=register_handler)
+
     setupdev_parser = subparsers.add_parser('setup', help='Setup a device. Passphrase protection uses the password given by -p. Requires interactive mode')
     setupdev_parser.add_argument('--label', '-l', help='The name to give to the device', default='')
     setupdev_parser.add_argument('--backup_passphrase', '-b', help='The passphrase to use for the backup, if applicable', default='')

hwilib/commands.py

@@ -499,6 +499,24 @@ def displayaddress(
             return {"address": client.display_singlesig_address(pubkey.get_full_derivation_path(0), addr_type)}
     raise BadArgumentError("Missing both path and descriptor")
 
+def register(
+    client: HardwareWalletClient,
+    name: str,
+    descriptor_template: str,
+    keys_info: List[str],
+) -> Dict[str, str]:
+    """
+    Register a BIP388 policy on the device for client.
+
+    :param name: Name for the policy
+    :param desc: Descriptor template
+    :return: A dictionary containing policy HMAC.
+        Returned as ``{"hmac": <hex string>}``.
+    :raises: BadArgumentError: if an argument is malformed, missing, or conflicts.
+    """
+
+    return {"hmac": client.register_bip388_policy(name, descriptor_template, keys_info)}
+
 def setup_device(client: HardwareWalletClient, label: str = "", backup_passphrase: str = "") -> Dict[str, bool]:
     """
     Setup a device that has not yet been initialized.

hwilib/devices/ledger.py

@@ -525,6 +525,26 @@ def format_key_info(pubkey: PubkeyProvider) -> str:
 
         return self.client.get_wallet_address(multisig_wallet, registered_hmac, change, address_index, True)
 
+    @ledger_exception
+    def register_bip388_policy(
+        self,
+        name: str,
+        descriptor_template: str,
+        keys_info: List[str],
+    ) -> str:
+        if isinstance(self.client, LegacyClient):
+            raise BadArgumentError("Registering a BIP388 policy not supported by this version of the Bitcoin App")
+
+        wallet_policy = WalletPolicy(
+            name=name,
+            descriptor_template=descriptor_template,
+            keys_info=keys_info
+        )
+
+        _, registered_hmac = self.client.register_wallet(wallet_policy)
+
+        return registered_hmac.hex()
+
     def setup_device(self, label: str = "", passphrase: str = "") -> bool:
         """
         Ledgers do not support setup via software.

hwilib/hwwclient.py

@@ -7,6 +7,7 @@
 
 from typing import (
     Dict,
+    List,
     Optional,
     Union,
 )
@@ -139,6 +140,18 @@ def display_multisig_address(
         raise NotImplementedError("The HardwareWalletClient base class "
                                   "does not implement this method")
 
+    def register_bip388_policy(
+        self,
+        name: str,
+        policy: str,
+        keys_info: List[str]) -> str:
+        """
+        Register a BIP388 policy.
+
+        :return: The policy HMAC
+        """
+        raise NotImplementedError("This device does not support BIP388 policies or it's not yet implemented")
+
     def wipe_device(self) -> bool:
         """
         Wipe the device.