Adds personalization options to ecdh.c

cryptotronix · Jun 30, 2015 · 6bf92dd · 6bf92dd
1 parent d0f06d5
commit 6bf92dd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/ecdh.c b/test/ecdh.c
@@ -60,6 +60,9 @@ parse_opt (int key, char *arg, struct argp_state *state)
     case 'g':
         arguments->gen_key = 1;
         break;
+    case 'p':
+        arguments->personalize = 1;
+        break;
     case 'f':
       arguments->input_file = arg;
       break;
@@ -154,6 +157,12 @@ main (int argc, char **argv)
 
   lca_print_hex_string ("cz:", cz.ptr, cz.len);
 
+  if (arguments.personalize)
+    {
+      rc = personalize (fd, arguments.input_file);
+      exit (rc);
+    }
+
   if (arguments.gen_key)
       genkey(arguments.slot, fd);