Verify that 'identity-v1' state machine is used

Following https://github.com/romanz/trezor-agent/issues/426.
2023-09-02 14:34:08 +03:00
parent 6776971b5a
commit 23c6349c98
1 changed files with 3 additions and 1 deletions
--- a/libagent/age/init.py
+++ b/libagent/age/init.py
@@ -172,8 +172,10 @@ def main(device_type):
    try:
        if args.identity:
            run_pubkey(device_type=device_type, args=args)
-        elif args.age_plugin:
+        elif args.age_plugin == 'identity-v1':
            run_decrypt(device_type=device_type, args=args)
+        else:
+            log.error("Unsupported state machine: %r", args.age_plugin)
    except Exception as e:  # pylint: disable=broad-except
        log.exception("age plugin failed: %s", e)