diff --git a/test/ecdh.c b/test/ecdh.c index 051a669..b6e6705 100644 --- 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);