Verify that 'identity-v1' state machine is used

Following #426.
romanz · Sep 2, 2023 · 23c6349 · 23c6349
1 parent 6776971
commit 23c6349
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)