make work with the cli-test user 'ct' too

1 files changed, 3 insertions, 1 deletions

diff --git a/bin/cryptech_upload b/bin/cryptech_upload
index 680a0ac..3222e2e 100755
--- a/bin/cryptech_upload
+++ b/bin/cryptech_upload
@@ -77,7 +77,7 @@ def parse_args():
                         )
 
     parser.add_argument("--username",
-                        choices = ("so", "wheel"),
+                        choices = ("so", "wheel", "ct"),
                         default = "so",
                         help = "Username to use when logging into the HSM",
                         )
@@ -134,6 +134,8 @@ def _execute(dst, cmd):
     pin = None
     if args.username == 'wheel':
         pin = wheel_pin
+    if args.username == 'ct':
+        pin = 'ct'
     _write(dst, "\r")
     prompt = _read(dst)
     if prompt.endswith("Username: "):