1/* $OpenBSD: ssh-keygen.c,v 1.322 2018/09/14 04:17:44 djm Exp $ */
2/*
3 * Author: Tatu Ylonen <ylo@cs.hut.fi>
4 * Copyright (c) 1994 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
5 *                    All rights reserved
6 * Identity and host key generation and maintenance.
7 *
8 * As far as I am concerned, the code I have written for this software
9 * can be used freely for any purpose.  Any derived versions of this
10 * software must be clearly marked as such, and if the derived work is
11 * incompatible with the protocol description in the RFC file, it must be
12 * called by a name other than "ssh" or "Secure Shell".
13 */
14
15#include "includes.h"
16
17#include <sys/types.h>
18#include <sys/socket.h>
19#include <sys/stat.h>
20
21#ifdef WITH_OPENSSL
22#include <openssl/evp.h>
23#include <openssl/pem.h>
24#include "openbsd-compat/openssl-compat.h"
25#endif
26
27#include <errno.h>
28#include <fcntl.h>
29#include <netdb.h>
30#ifdef HAVE_PATHS_H
31# include <paths.h>
32#endif
33#include <pwd.h>
34#include <stdarg.h>
35#include <stdio.h>
36#include <stdlib.h>
37#include <string.h>
38#include <unistd.h>
39#include <limits.h>
40#include <locale.h>
41#include <time.h>
42
43#include "xmalloc.h"
44#include "sshkey.h"
45#include "authfile.h"
46#include "uuencode.h"
47#include "sshbuf.h"
48#include "pathnames.h"
49#include "log.h"
50#include "misc.h"
51#include "match.h"
52#include "hostfile.h"
53#include "dns.h"
54#include "ssh.h"
55#include "ssh2.h"
56#include "ssherr.h"
57#include "ssh-pkcs11.h"
58#include "atomicio.h"
59#include "krl.h"
60#include "digest.h"
61#include "utf8.h"
62#include "authfd.h"
63
64#ifdef WITH_OPENSSL
65# define DEFAULT_KEY_TYPE_NAME "rsa"
66#else
67# define DEFAULT_KEY_TYPE_NAME "ed25519"
68#endif
69
70/* Number of bits in the RSA/DSA key.  This value can be set on the command line. */
71#define DEFAULT_BITS		2048
72#define DEFAULT_BITS_DSA	1024
73#define DEFAULT_BITS_ECDSA	256
74u_int32_t bits = 0;
75
76/*
77 * Flag indicating that we just want to change the passphrase.  This can be
78 * set on the command line.
79 */
80int change_passphrase = 0;
81
82/*
83 * Flag indicating that we just want to change the comment.  This can be set
84 * on the command line.
85 */
86int change_comment = 0;
87
88int quiet = 0;
89
90int log_level = SYSLOG_LEVEL_INFO;
91
92/* Flag indicating that we want to hash a known_hosts file */
93int hash_hosts = 0;
94/* Flag indicating that we want lookup a host in known_hosts file */
95int find_host = 0;
96/* Flag indicating that we want to delete a host from a known_hosts file */
97int delete_host = 0;
98
99/* Flag indicating that we want to show the contents of a certificate */
100int show_cert = 0;
101
102/* Flag indicating that we just want to see the key fingerprint */
103int print_fingerprint = 0;
104int print_bubblebabble = 0;
105
106/* Hash algorithm to use for fingerprints. */
107int fingerprint_hash = SSH_FP_HASH_DEFAULT;
108
109/* The identity file name, given on the command line or entered by the user. */
110char identity_file[1024];
111int have_identity = 0;
112
113/* This is set to the passphrase if given on the command line. */
114char *identity_passphrase = NULL;
115
116/* This is set to the new passphrase if given on the command line. */
117char *identity_new_passphrase = NULL;
118
119/* This is set to the new comment if given on the command line. */
120char *identity_comment = NULL;
121
122/* Path to CA key when certifying keys. */
123char *ca_key_path = NULL;
124
125/* Prefer to use agent keys for CA signing */
126int prefer_agent = 0;
127
128/* Certificate serial number */
129unsigned long long cert_serial = 0;
130
131/* Key type when certifying */
132u_int cert_key_type = SSH2_CERT_TYPE_USER;
133
134/* "key ID" of signed key */
135char *cert_key_id = NULL;
136
137/* Comma-separated list of principal names for certifying keys */
138char *cert_principals = NULL;
139
140/* Validity period for certificates */
141u_int64_t cert_valid_from = 0;
142u_int64_t cert_valid_to = ~0ULL;
143
144/* Certificate options */
145#define CERTOPT_X_FWD	(1)
146#define CERTOPT_AGENT_FWD	(1<<1)
147#define CERTOPT_PORT_FWD	(1<<2)
148#define CERTOPT_PTY		(1<<3)
149#define CERTOPT_USER_RC	(1<<4)
150#define CERTOPT_DEFAULT	(CERTOPT_X_FWD|CERTOPT_AGENT_FWD| \
151			 CERTOPT_PORT_FWD|CERTOPT_PTY|CERTOPT_USER_RC)
152u_int32_t certflags_flags = CERTOPT_DEFAULT;
153char *certflags_command = NULL;
154char *certflags_src_addr = NULL;
155
156/* Arbitrary extensions specified by user */
157struct cert_userext {
158	char *key;
159	char *val;
160	int crit;
161};
162struct cert_userext *cert_userext;
163size_t ncert_userext;
164
165/* Conversion to/from various formats */
166int convert_to = 0;
167int convert_from = 0;
168enum {
169	FMT_RFC4716,
170	FMT_PKCS8,
171	FMT_PEM
172} convert_format = FMT_RFC4716;
173int print_public = 0;
174int print_generic = 0;
175
176char *key_type_name = NULL;
177
178/* Load key from this PKCS#11 provider */
179char *pkcs11provider = NULL;
180
181/* Use new OpenSSH private key format when writing SSH2 keys instead of PEM */
182int use_new_format = 1;
183
184/* Cipher for new-format private keys */
185char *new_format_cipher = NULL;
186
187/*
188 * Number of KDF rounds to derive new format keys /
189 * number of primality trials when screening moduli.
190 */
191int rounds = 0;
192
193/* argv0 */
194extern char *__progname;
195
196char hostname[NI_MAXHOST];
197
198#ifdef WITH_OPENSSL
199/* moduli.c */
200int gen_candidates(FILE *, u_int32_t, u_int32_t, BIGNUM *);
201int prime_test(FILE *, FILE *, u_int32_t, u_int32_t, char *, unsigned long,
202    unsigned long);
203#endif
204
205static void
206type_bits_valid(int type, const char *name, u_int32_t *bitsp)
207{
208#ifdef WITH_OPENSSL
209	u_int maxbits, nid;
210#endif
211
212	if (type == KEY_UNSPEC)
213		fatal("unknown key type %s", key_type_name);
214	if (*bitsp == 0) {
215#ifdef WITH_OPENSSL
216		if (type == KEY_DSA)
217			*bitsp = DEFAULT_BITS_DSA;
218		else if (type == KEY_ECDSA) {
219			if (name != NULL &&
220			    (nid = sshkey_ecdsa_nid_from_name(name)) > 0)
221				*bitsp = sshkey_curve_nid_to_bits(nid);
222			if (*bitsp == 0)
223				*bitsp = DEFAULT_BITS_ECDSA;
224		} else
225#endif
226			*bitsp = DEFAULT_BITS;
227	}
228#ifdef WITH_OPENSSL
229	maxbits = (type == KEY_DSA) ?
230	    OPENSSL_DSA_MAX_MODULUS_BITS : OPENSSL_RSA_MAX_MODULUS_BITS;
231	if (*bitsp > maxbits)
232		fatal("key bits exceeds maximum %d", maxbits);
233	switch (type) {
234	case KEY_DSA:
235		if (*bitsp != 1024)
236			fatal("Invalid DSA key length: must be 1024 bits");
237		break;
238	case KEY_RSA:
239		if (*bitsp < SSH_RSA_MINIMUM_MODULUS_SIZE)
240			fatal("Invalid RSA key length: minimum is %d bits",
241			    SSH_RSA_MINIMUM_MODULUS_SIZE);
242		break;
243	case KEY_ECDSA:
244		if (sshkey_ecdsa_bits_to_nid(*bitsp) == -1)
245			fatal("Invalid ECDSA key length: valid lengths are "
246			    "256, 384 or 521 bits");
247	}
248#endif
249}
250
251static void
252ask_filename(struct passwd *pw, const char *prompt)
253{
254	char buf[1024];
255	char *name = NULL;
256
257	if (key_type_name == NULL)
258		name = _PATH_SSH_CLIENT_ID_RSA;
259	else {
260		switch (sshkey_type_from_name(key_type_name)) {
261		case KEY_DSA_CERT:
262		case KEY_DSA:
263			name = _PATH_SSH_CLIENT_ID_DSA;
264			break;
265#ifdef OPENSSL_HAS_ECC
266		case KEY_ECDSA_CERT:
267		case KEY_ECDSA:
268			name = _PATH_SSH_CLIENT_ID_ECDSA;
269			break;
270#endif
271		case KEY_RSA_CERT:
272		case KEY_RSA:
273			name = _PATH_SSH_CLIENT_ID_RSA;
274			break;
275		case KEY_ED25519:
276		case KEY_ED25519_CERT:
277			name = _PATH_SSH_CLIENT_ID_ED25519;
278			break;
279		case KEY_XMSS:
280		case KEY_XMSS_CERT:
281			name = _PATH_SSH_CLIENT_ID_XMSS;
282			break;
283		default:
284			fatal("bad key type");
285		}
286	}
287	snprintf(identity_file, sizeof(identity_file),
288	    "%s/%s", pw->pw_dir, name);
289	printf("%s (%s): ", prompt, identity_file);
290	fflush(stdout);
291	if (fgets(buf, sizeof(buf), stdin) == NULL)
292		exit(1);
293	buf[strcspn(buf, "\n")] = '\0';
294	if (strcmp(buf, "") != 0)
295		strlcpy(identity_file, buf, sizeof(identity_file));
296	have_identity = 1;
297}
298
299static struct sshkey *
300load_identity(char *filename)
301{
302	char *pass;
303	struct sshkey *prv;
304	int r;
305
306	if ((r = sshkey_load_private(filename, "", &prv, NULL)) == 0)
307		return prv;
308	if (r != SSH_ERR_KEY_WRONG_PASSPHRASE)
309		fatal("Load key \"%s\": %s", filename, ssh_err(r));
310	if (identity_passphrase)
311		pass = xstrdup(identity_passphrase);
312	else
313		pass = read_passphrase("Enter passphrase: ", RP_ALLOW_STDIN);
314	r = sshkey_load_private(filename, pass, &prv, NULL);
315	explicit_bzero(pass, strlen(pass));
316	free(pass);
317	if (r != 0)
318		fatal("Load key \"%s\": %s", filename, ssh_err(r));
319	return prv;
320}
321
322#define SSH_COM_PUBLIC_BEGIN		"---- BEGIN SSH2 PUBLIC KEY ----"
323#define SSH_COM_PUBLIC_END		"---- END SSH2 PUBLIC KEY ----"
324#define SSH_COM_PRIVATE_BEGIN		"---- BEGIN SSH2 ENCRYPTED PRIVATE KEY ----"
325#define	SSH_COM_PRIVATE_KEY_MAGIC	0x3f6ff9eb
326
327#ifdef WITH_OPENSSL
328static void
329do_convert_to_ssh2(struct passwd *pw, struct sshkey *k)
330{
331	size_t len;
332	u_char *blob;
333	char comment[61];
334	int r;
335
336	if ((r = sshkey_to_blob(k, &blob, &len)) != 0)
337		fatal("key_to_blob failed: %s", ssh_err(r));
338	/* Comment + surrounds must fit into 72 chars (RFC 4716 sec 3.3) */
339	snprintf(comment, sizeof(comment),
340	    "%u-bit %s, converted by %s@%s from OpenSSH",
341	    sshkey_size(k), sshkey_type(k),
342	    pw->pw_name, hostname);
343
344	fprintf(stdout, "%s\n", SSH_COM_PUBLIC_BEGIN);
345	fprintf(stdout, "Comment: \"%s\"\n", comment);
346	dump_base64(stdout, blob, len);
347	fprintf(stdout, "%s\n", SSH_COM_PUBLIC_END);
348	sshkey_free(k);
349	free(blob);
350	exit(0);
351}
352
353static void
354do_convert_to_pkcs8(struct sshkey *k)
355{
356	switch (sshkey_type_plain(k->type)) {
357	case KEY_RSA:
358		if (!PEM_write_RSA_PUBKEY(stdout, k->rsa))
359			fatal("PEM_write_RSA_PUBKEY failed");
360		break;
361	case KEY_DSA:
362		if (!PEM_write_DSA_PUBKEY(stdout, k->dsa))
363			fatal("PEM_write_DSA_PUBKEY failed");
364		break;
365#ifdef OPENSSL_HAS_ECC
366	case KEY_ECDSA:
367		if (!PEM_write_EC_PUBKEY(stdout, k->ecdsa))
368			fatal("PEM_write_EC_PUBKEY failed");
369		break;
370#endif
371	default:
372		fatal("%s: unsupported key type %s", __func__, sshkey_type(k));
373	}
374	exit(0);
375}
376
377static void
378do_convert_to_pem(struct sshkey *k)
379{
380	switch (sshkey_type_plain(k->type)) {
381	case KEY_RSA:
382		if (!PEM_write_RSAPublicKey(stdout, k->rsa))
383			fatal("PEM_write_RSAPublicKey failed");
384		break;
385	default:
386		fatal("%s: unsupported key type %s", __func__, sshkey_type(k));
387	}
388	exit(0);
389}
390
391static void
392do_convert_to(struct passwd *pw)
393{
394	struct sshkey *k;
395	struct stat st;
396	int r;
397
398	if (!have_identity)
399		ask_filename(pw, "Enter file in which the key is");
400	if (stat(identity_file, &st) < 0)
401		fatal("%s: %s: %s", __progname, identity_file, strerror(errno));
402	if ((r = sshkey_load_public(identity_file, &k, NULL)) != 0)
403		k = load_identity(identity_file);
404	switch (convert_format) {
405	case FMT_RFC4716:
406		do_convert_to_ssh2(pw, k);
407		break;
408	case FMT_PKCS8:
409		do_convert_to_pkcs8(k);
410		break;
411	case FMT_PEM:
412		do_convert_to_pem(k);
413		break;
414	default:
415		fatal("%s: unknown key format %d", __func__, convert_format);
416	}
417	exit(0);
418}
419
420/*
421 * This is almost exactly the bignum1 encoding, but with 32 bit for length
422 * instead of 16.
423 */
424static void
425buffer_get_bignum_bits(struct sshbuf *b, BIGNUM *value)
426{
427	u_int bytes, bignum_bits;
428	int r;
429
430	if ((r = sshbuf_get_u32(b, &bignum_bits)) != 0)
431		fatal("%s: buffer error: %s", __func__, ssh_err(r));
432	bytes = (bignum_bits + 7) / 8;
433	if (sshbuf_len(b) < bytes)
434		fatal("%s: input buffer too small: need %d have %zu",
435		    __func__, bytes, sshbuf_len(b));
436	if (BN_bin2bn(sshbuf_ptr(b), bytes, value) == NULL)
437		fatal("%s: BN_bin2bn failed", __func__);
438	if ((r = sshbuf_consume(b, bytes)) != 0)
439		fatal("%s: buffer error: %s", __func__, ssh_err(r));
440}
441
442static struct sshkey *
443do_convert_private_ssh2_from_blob(u_char *blob, u_int blen)
444{
445	struct sshbuf *b;
446	struct sshkey *key = NULL;
447	char *type, *cipher;
448	u_char e1, e2, e3, *sig = NULL, data[] = "abcde12345";
449	int r, rlen, ktype;
450	u_int magic, i1, i2, i3, i4;
451	size_t slen;
452	u_long e;
453	BIGNUM *dsa_p = NULL, *dsa_q = NULL, *dsa_g = NULL;
454	BIGNUM *dsa_pub_key = NULL, *dsa_priv_key = NULL;
455	BIGNUM *rsa_n = NULL, *rsa_e = NULL, *rsa_d = NULL;
456	BIGNUM *rsa_p = NULL, *rsa_q = NULL, *rsa_iqmp = NULL;
457	if ((b = sshbuf_from(blob, blen)) == NULL)
458		fatal("%s: sshbuf_from failed", __func__);
459	if ((r = sshbuf_get_u32(b, &magic)) != 0)
460		fatal("%s: buffer error: %s", __func__, ssh_err(r));
461
462	if (magic != SSH_COM_PRIVATE_KEY_MAGIC) {
463		error("bad magic 0x%x != 0x%x", magic,
464		    SSH_COM_PRIVATE_KEY_MAGIC);
465		sshbuf_free(b);
466		return NULL;
467	}
468	if ((r = sshbuf_get_u32(b, &i1)) != 0 ||
469	    (r = sshbuf_get_cstring(b, &type, NULL)) != 0 ||
470	    (r = sshbuf_get_cstring(b, &cipher, NULL)) != 0 ||
471	    (r = sshbuf_get_u32(b, &i2)) != 0 ||
472	    (r = sshbuf_get_u32(b, &i3)) != 0 ||
473	    (r = sshbuf_get_u32(b, &i4)) != 0)
474		fatal("%s: buffer error: %s", __func__, ssh_err(r));
475	debug("ignore (%d %d %d %d)", i1, i2, i3, i4);
476	if (strcmp(cipher, "none") != 0) {
477		error("unsupported cipher %s", cipher);
478		free(cipher);
479		sshbuf_free(b);
480		free(type);
481		return NULL;
482	}
483	free(cipher);
484
485	if (strstr(type, "dsa")) {
486		ktype = KEY_DSA;
487	} else if (strstr(type, "rsa")) {
488		ktype = KEY_RSA;
489	} else {
490		sshbuf_free(b);
491		free(type);
492		return NULL;
493	}
494	if ((key = sshkey_new(ktype)) == NULL)
495		fatal("sshkey_new failed");
496	free(type);
497
498	switch (key->type) {
499	case KEY_DSA:
500		if ((dsa_p = BN_new()) == NULL ||
501		    (dsa_q = BN_new()) == NULL ||
502		    (dsa_g = BN_new()) == NULL ||
503		    (dsa_pub_key = BN_new()) == NULL ||
504		    (dsa_priv_key = BN_new()) == NULL)
505			fatal("%s: BN_new", __func__);
506		buffer_get_bignum_bits(b, dsa_p);
507		buffer_get_bignum_bits(b, dsa_g);
508		buffer_get_bignum_bits(b, dsa_q);
509		buffer_get_bignum_bits(b, dsa_pub_key);
510		buffer_get_bignum_bits(b, dsa_priv_key);
511		if (!DSA_set0_pqg(key->dsa, dsa_p, dsa_q, dsa_g))
512			fatal("%s: DSA_set0_pqg failed", __func__);
513		dsa_p = dsa_q = dsa_g = NULL; /* transferred */
514		if (!DSA_set0_key(key->dsa, dsa_pub_key, dsa_priv_key))
515			fatal("%s: DSA_set0_key failed", __func__);
516		dsa_pub_key = dsa_priv_key = NULL; /* transferred */
517		break;
518	case KEY_RSA:
519		if ((r = sshbuf_get_u8(b, &e1)) != 0 ||
520		    (e1 < 30 && (r = sshbuf_get_u8(b, &e2)) != 0) ||
521		    (e1 < 30 && (r = sshbuf_get_u8(b, &e3)) != 0))
522			fatal("%s: buffer error: %s", __func__, ssh_err(r));
523		e = e1;
524		debug("e %lx", e);
525		if (e < 30) {
526			e <<= 8;
527			e += e2;
528			debug("e %lx", e);
529			e <<= 8;
530			e += e3;
531			debug("e %lx", e);
532		}
533		if ((rsa_e = BN_new()) == NULL)
534			fatal("%s: BN_new", __func__);
535		if (!BN_set_word(rsa_e, e)) {
536			BN_clear_free(rsa_e);
537			sshbuf_free(b);
538			sshkey_free(key);
539			return NULL;
540		}
541		if ((rsa_n = BN_new()) == NULL ||
542		    (rsa_d = BN_new()) == NULL ||
543		    (rsa_p = BN_new()) == NULL ||
544		    (rsa_q = BN_new()) == NULL ||
545		    (rsa_iqmp = BN_new()) == NULL)
546			fatal("%s: BN_new", __func__);
547		buffer_get_bignum_bits(b, rsa_d);
548		buffer_get_bignum_bits(b, rsa_n);
549		buffer_get_bignum_bits(b, rsa_iqmp);
550		buffer_get_bignum_bits(b, rsa_q);
551		buffer_get_bignum_bits(b, rsa_p);
552		if (!RSA_set0_key(key->rsa, rsa_n, rsa_e, rsa_d))
553			fatal("%s: RSA_set0_key failed", __func__);
554		rsa_n = rsa_e = rsa_d = NULL; /* transferred */
555		if (!RSA_set0_factors(key->rsa, rsa_p, rsa_q))
556			fatal("%s: RSA_set0_factors failed", __func__);
557		rsa_p = rsa_q = NULL; /* transferred */
558		if ((r = ssh_rsa_complete_crt_parameters(key, rsa_iqmp)) != 0)
559			fatal("generate RSA parameters failed: %s", ssh_err(r));
560		BN_clear_free(rsa_iqmp);
561		break;
562	}
563	rlen = sshbuf_len(b);
564	if (rlen != 0)
565		error("do_convert_private_ssh2_from_blob: "
566		    "remaining bytes in key blob %d", rlen);
567	sshbuf_free(b);
568
569	/* try the key */
570	if (sshkey_sign(key, &sig, &slen, data, sizeof(data), NULL, 0) != 0 ||
571	    sshkey_verify(key, sig, slen, data, sizeof(data), NULL, 0) != 0) {
572		sshkey_free(key);
573		free(sig);
574		return NULL;
575	}
576	free(sig);
577	return key;
578}
579
580static int
581get_line(FILE *fp, char *line, size_t len)
582{
583	int c;
584	size_t pos = 0;
585
586	line[0] = '\0';
587	while ((c = fgetc(fp)) != EOF) {
588		if (pos >= len - 1)
589			fatal("input line too long.");
590		switch (c) {
591		case '\r':
592			c = fgetc(fp);
593			if (c != EOF && c != '\n' && ungetc(c, fp) == EOF)
594				fatal("unget: %s", strerror(errno));
595			return pos;
596		case '\n':
597			return pos;
598		}
599		line[pos++] = c;
600		line[pos] = '\0';
601	}
602	/* We reached EOF */
603	return -1;
604}
605
606static void
607do_convert_from_ssh2(struct passwd *pw, struct sshkey **k, int *private)
608{
609	int r, blen, escaped = 0;
610	u_int len;
611	char line[1024];
612	u_char blob[8096];
613	char encoded[8096];
614	FILE *fp;
615
616	if ((fp = fopen(identity_file, "r")) == NULL)
617		fatal("%s: %s: %s", __progname, identity_file, strerror(errno));
618	encoded[0] = '\0';
619	while ((blen = get_line(fp, line, sizeof(line))) != -1) {
620		if (blen > 0 && line[blen - 1] == '\\')
621			escaped++;
622		if (strncmp(line, "----", 4) == 0 ||
623		    strstr(line, ": ") != NULL) {
624			if (strstr(line, SSH_COM_PRIVATE_BEGIN) != NULL)
625				*private = 1;
626			if (strstr(line, " END ") != NULL) {
627				break;
628			}
629			/* fprintf(stderr, "ignore: %s", line); */
630			continue;
631		}
632		if (escaped) {
633			escaped--;
634			/* fprintf(stderr, "escaped: %s", line); */
635			continue;
636		}
637		strlcat(encoded, line, sizeof(encoded));
638	}
639	len = strlen(encoded);
640	if (((len % 4) == 3) &&
641	    (encoded[len-1] == '=') &&
642	    (encoded[len-2] == '=') &&
643	    (encoded[len-3] == '='))
644		encoded[len-3] = '\0';
645	blen = uudecode(encoded, blob, sizeof(blob));
646	if (blen < 0)
647		fatal("uudecode failed.");
648	if (*private)
649		*k = do_convert_private_ssh2_from_blob(blob, blen);
650	else if ((r = sshkey_from_blob(blob, blen, k)) != 0)
651		fatal("decode blob failed: %s", ssh_err(r));
652	fclose(fp);
653}
654
655static void
656do_convert_from_pkcs8(struct sshkey **k, int *private)
657{
658	EVP_PKEY *pubkey;
659	FILE *fp;
660
661	if ((fp = fopen(identity_file, "r")) == NULL)
662		fatal("%s: %s: %s", __progname, identity_file, strerror(errno));
663	if ((pubkey = PEM_read_PUBKEY(fp, NULL, NULL, NULL)) == NULL) {
664		fatal("%s: %s is not a recognised public key format", __func__,
665		    identity_file);
666	}
667	fclose(fp);
668	switch (EVP_PKEY_base_id(pubkey)) {
669	case EVP_PKEY_RSA:
670		if ((*k = sshkey_new(KEY_UNSPEC)) == NULL)
671			fatal("sshkey_new failed");
672		(*k)->type = KEY_RSA;
673		(*k)->rsa = EVP_PKEY_get1_RSA(pubkey);
674		break;
675	case EVP_PKEY_DSA:
676		if ((*k = sshkey_new(KEY_UNSPEC)) == NULL)
677			fatal("sshkey_new failed");
678		(*k)->type = KEY_DSA;
679		(*k)->dsa = EVP_PKEY_get1_DSA(pubkey);
680		break;
681#ifdef OPENSSL_HAS_ECC
682	case EVP_PKEY_EC:
683		if ((*k = sshkey_new(KEY_UNSPEC)) == NULL)
684			fatal("sshkey_new failed");
685		(*k)->type = KEY_ECDSA;
686		(*k)->ecdsa = EVP_PKEY_get1_EC_KEY(pubkey);
687		(*k)->ecdsa_nid = sshkey_ecdsa_key_to_nid((*k)->ecdsa);
688		break;
689#endif
690	default:
691		fatal("%s: unsupported pubkey type %d", __func__,
692		    EVP_PKEY_base_id(pubkey));
693	}
694	EVP_PKEY_free(pubkey);
695	return;
696}
697
698static void
699do_convert_from_pem(struct sshkey **k, int *private)
700{
701	FILE *fp;
702	RSA *rsa;
703
704	if ((fp = fopen(identity_file, "r")) == NULL)
705		fatal("%s: %s: %s", __progname, identity_file, strerror(errno));
706	if ((rsa = PEM_read_RSAPublicKey(fp, NULL, NULL, NULL)) != NULL) {
707		if ((*k = sshkey_new(KEY_UNSPEC)) == NULL)
708			fatal("sshkey_new failed");
709		(*k)->type = KEY_RSA;
710		(*k)->rsa = rsa;
711		fclose(fp);
712		return;
713	}
714	fatal("%s: unrecognised raw private key format", __func__);
715}
716
717static void
718do_convert_from(struct passwd *pw)
719{
720	struct sshkey *k = NULL;
721	int r, private = 0, ok = 0;
722	struct stat st;
723
724	if (!have_identity)
725		ask_filename(pw, "Enter file in which the key is");
726	if (stat(identity_file, &st) < 0)
727		fatal("%s: %s: %s", __progname, identity_file, strerror(errno));
728
729	switch (convert_format) {
730	case FMT_RFC4716:
731		do_convert_from_ssh2(pw, &k, &private);
732		break;
733	case FMT_PKCS8:
734		do_convert_from_pkcs8(&k, &private);
735		break;
736	case FMT_PEM:
737		do_convert_from_pem(&k, &private);
738		break;
739	default:
740		fatal("%s: unknown key format %d", __func__, convert_format);
741	}
742
743	if (!private) {
744		if ((r = sshkey_write(k, stdout)) == 0)
745			ok = 1;
746		if (ok)
747			fprintf(stdout, "\n");
748	} else {
749		switch (k->type) {
750		case KEY_DSA:
751			ok = PEM_write_DSAPrivateKey(stdout, k->dsa, NULL,
752			    NULL, 0, NULL, NULL);
753			break;
754#ifdef OPENSSL_HAS_ECC
755		case KEY_ECDSA:
756			ok = PEM_write_ECPrivateKey(stdout, k->ecdsa, NULL,
757			    NULL, 0, NULL, NULL);
758			break;
759#endif
760		case KEY_RSA:
761			ok = PEM_write_RSAPrivateKey(stdout, k->rsa, NULL,
762			    NULL, 0, NULL, NULL);
763			break;
764		default:
765			fatal("%s: unsupported key type %s", __func__,
766			    sshkey_type(k));
767		}
768	}
769
770	if (!ok)
771		fatal("key write failed");
772	sshkey_free(k);
773	exit(0);
774}
775#endif
776
777static void
778do_print_public(struct passwd *pw)
779{
780	struct sshkey *prv;
781	struct stat st;
782	int r;
783
784	if (!have_identity)
785		ask_filename(pw, "Enter file in which the key is");
786	if (stat(identity_file, &st) < 0)
787		fatal("%s: %s", identity_file, strerror(errno));
788	prv = load_identity(identity_file);
789	if ((r = sshkey_write(prv, stdout)) != 0)
790		error("sshkey_write failed: %s", ssh_err(r));
791	sshkey_free(prv);
792	fprintf(stdout, "\n");
793	exit(0);
794}
795
796static void
797do_download(struct passwd *pw)
798{
799#ifdef ENABLE_PKCS11
800	struct sshkey **keys = NULL;
801	int i, nkeys;
802	enum sshkey_fp_rep rep;
803	int fptype;
804	char *fp, *ra;
805
806	fptype = print_bubblebabble ? SSH_DIGEST_SHA1 : fingerprint_hash;
807	rep =    print_bubblebabble ? SSH_FP_BUBBLEBABBLE : SSH_FP_DEFAULT;
808
809	pkcs11_init(0);
810	nkeys = pkcs11_add_provider(pkcs11provider, NULL, &keys);
811	if (nkeys <= 0)
812		fatal("cannot read public key from pkcs11");
813	for (i = 0; i < nkeys; i++) {
814		if (print_fingerprint) {
815			fp = sshkey_fingerprint(keys[i], fptype, rep);
816			ra = sshkey_fingerprint(keys[i], fingerprint_hash,
817			    SSH_FP_RANDOMART);
818			if (fp == NULL || ra == NULL)
819				fatal("%s: sshkey_fingerprint fail", __func__);
820			printf("%u %s %s (PKCS11 key)\n", sshkey_size(keys[i]),
821			    fp, sshkey_type(keys[i]));
822			if (log_level >= SYSLOG_LEVEL_VERBOSE)
823				printf("%s\n", ra);
824			free(ra);
825			free(fp);
826		} else {
827			(void) sshkey_write(keys[i], stdout); /* XXX check */
828			fprintf(stdout, "\n");
829		}
830		sshkey_free(keys[i]);
831	}
832	free(keys);
833	pkcs11_terminate();
834	exit(0);
835#else
836	fatal("no pkcs11 support");
837#endif /* ENABLE_PKCS11 */
838}
839
840static struct sshkey *
841try_read_key(char **cpp)
842{
843	struct sshkey *ret;
844	int r;
845
846	if ((ret = sshkey_new(KEY_UNSPEC)) == NULL)
847		fatal("sshkey_new failed");
848	if ((r = sshkey_read(ret, cpp)) == 0)
849		return ret;
850	/* Not a key */
851	sshkey_free(ret);
852	return NULL;
853}
854
855static void
856fingerprint_one_key(const struct sshkey *public, const char *comment)
857{
858	char *fp = NULL, *ra = NULL;
859	enum sshkey_fp_rep rep;
860	int fptype;
861
862	fptype = print_bubblebabble ? SSH_DIGEST_SHA1 : fingerprint_hash;
863	rep =    print_bubblebabble ? SSH_FP_BUBBLEBABBLE : SSH_FP_DEFAULT;
864	fp = sshkey_fingerprint(public, fptype, rep);
865	ra = sshkey_fingerprint(public, fingerprint_hash, SSH_FP_RANDOMART);
866	if (fp == NULL || ra == NULL)
867		fatal("%s: sshkey_fingerprint failed", __func__);
868	mprintf("%u %s %s (%s)\n", sshkey_size(public), fp,
869	    comment ? comment : "no comment", sshkey_type(public));
870	if (log_level >= SYSLOG_LEVEL_VERBOSE)
871		printf("%s\n", ra);
872	free(ra);
873	free(fp);
874}
875
876static void
877fingerprint_private(const char *path)
878{
879	struct stat st;
880	char *comment = NULL;
881	struct sshkey *public = NULL;
882	int r;
883
884	if (stat(identity_file, &st) < 0)
885		fatal("%s: %s", path, strerror(errno));
886	if ((r = sshkey_load_public(path, &public, &comment)) != 0) {
887		debug("load public \"%s\": %s", path, ssh_err(r));
888		if ((r = sshkey_load_private(path, NULL,
889		    &public, &comment)) != 0) {
890			debug("load private \"%s\": %s", path, ssh_err(r));
891			fatal("%s is not a key file.", path);
892		}
893	}
894
895	fingerprint_one_key(public, comment);
896	sshkey_free(public);
897	free(comment);
898}
899
900static void
901do_fingerprint(struct passwd *pw)
902{
903	FILE *f;
904	struct sshkey *public = NULL;
905	char *comment = NULL, *cp, *ep, *line = NULL;
906	size_t linesize = 0;
907	int i, invalid = 1;
908	const char *path;
909	u_long lnum = 0;
910
911	if (!have_identity)
912		ask_filename(pw, "Enter file in which the key is");
913	path = identity_file;
914
915	if (strcmp(identity_file, "-") == 0) {
916		f = stdin;
917		path = "(stdin)";
918	} else if ((f = fopen(path, "r")) == NULL)
919		fatal("%s: %s: %s", __progname, path, strerror(errno));
920
921	while (getline(&line, &linesize, f) != -1) {
922		lnum++;
923		cp = line;
924		cp[strcspn(cp, "\n")] = '\0';
925		/* Trim leading space and comments */
926		cp = line + strspn(line, " \t");
927		if (*cp == '#' || *cp == '\0')
928			continue;
929
930		/*
931		 * Input may be plain keys, private keys, authorized_keys
932		 * or known_hosts.
933		 */
934
935		/*
936		 * Try private keys first. Assume a key is private if
937		 * "SSH PRIVATE KEY" appears on the first line and we're
938		 * not reading from stdin (XXX support private keys on stdin).
939		 */
940		if (lnum == 1 && strcmp(identity_file, "-") != 0 &&
941		    strstr(cp, "PRIVATE KEY") != NULL) {
942			free(line);
943			fclose(f);
944			fingerprint_private(path);
945			exit(0);
946		}
947
948		/*
949		 * If it's not a private key, then this must be prepared to
950		 * accept a public key prefixed with a hostname or options.
951		 * Try a bare key first, otherwise skip the leading stuff.
952		 */
953		if ((public = try_read_key(&cp)) == NULL) {
954			i = strtol(cp, &ep, 10);
955			if (i == 0 || ep == NULL ||
956			    (*ep != ' ' && *ep != '\t')) {
957				int quoted = 0;
958
959				comment = cp;
960				for (; *cp && (quoted || (*cp != ' ' &&
961				    *cp != '\t')); cp++) {
962					if (*cp == '\\' && cp[1] == '"')
963						cp++;	/* Skip both */
964					else if (*cp == '"')
965						quoted = !quoted;
966				}
967				if (!*cp)
968					continue;
969				*cp++ = '\0';
970			}
971		}
972		/* Retry after parsing leading hostname/key options */
973		if (public == NULL && (public = try_read_key(&cp)) == NULL) {
974			debug("%s:%lu: not a public key", path, lnum);
975			continue;
976		}
977
978		/* Find trailing comment, if any */
979		for (; *cp == ' ' || *cp == '\t'; cp++)
980			;
981		if (*cp != '\0' && *cp != '#')
982			comment = cp;
983
984		fingerprint_one_key(public, comment);
985		sshkey_free(public);
986		invalid = 0; /* One good key in the file is sufficient */
987	}
988	fclose(f);
989	free(line);
990
991	if (invalid)
992		fatal("%s is not a public key file.", path);
993	exit(0);
994}
995
996static void
997do_gen_all_hostkeys(struct passwd *pw)
998{
999	struct {
1000		char *key_type;
1001		char *key_type_display;
1002		char *path;
1003	} key_types[] = {
1004#ifdef WITH_OPENSSL
1005		{ "rsa", "RSA" ,_PATH_HOST_RSA_KEY_FILE },
1006		{ "dsa", "DSA", _PATH_HOST_DSA_KEY_FILE },
1007#ifdef OPENSSL_HAS_ECC
1008		{ "ecdsa", "ECDSA",_PATH_HOST_ECDSA_KEY_FILE },
1009#endif /* OPENSSL_HAS_ECC */
1010#endif /* WITH_OPENSSL */
1011		{ "ed25519", "ED25519",_PATH_HOST_ED25519_KEY_FILE },
1012#ifdef WITH_XMSS
1013		{ "xmss", "XMSS",_PATH_HOST_XMSS_KEY_FILE },
1014#endif /* WITH_XMSS */
1015		{ NULL, NULL, NULL }
1016	};
1017
1018	int first = 0;
1019	struct stat st;
1020	struct sshkey *private, *public;
1021	char comment[1024], *prv_tmp, *pub_tmp, *prv_file, *pub_file;
1022	int i, type, fd, r;
1023	FILE *f;
1024
1025	for (i = 0; key_types[i].key_type; i++) {
1026		public = private = NULL;
1027		prv_tmp = pub_tmp = prv_file = pub_file = NULL;
1028
1029		xasprintf(&prv_file, "%s%s",
1030		    identity_file, key_types[i].path);
1031
1032		/* Check whether private key exists and is not zero-length */
1033		if (stat(prv_file, &st) == 0) {
1034			if (st.st_size != 0)
1035				goto next;
1036		} else if (errno != ENOENT) {
1037			error("Could not stat %s: %s", key_types[i].path,
1038			    strerror(errno));
1039			goto failnext;
1040		}
1041
1042		/*
1043		 * Private key doesn't exist or is invalid; proceed with
1044		 * key generation.
1045		 */
1046		xasprintf(&prv_tmp, "%s%s.XXXXXXXXXX",
1047		    identity_file, key_types[i].path);
1048		xasprintf(&pub_tmp, "%s%s.pub.XXXXXXXXXX",
1049		    identity_file, key_types[i].path);
1050		xasprintf(&pub_file, "%s%s.pub",
1051		    identity_file, key_types[i].path);
1052
1053		if (first == 0) {
1054			first = 1;
1055			printf("%s: generating new host keys: ", __progname);
1056		}
1057		printf("%s ", key_types[i].key_type_display);
1058		fflush(stdout);
1059		type = sshkey_type_from_name(key_types[i].key_type);
1060		if ((fd = mkstemp(prv_tmp)) == -1) {
1061			error("Could not save your public key in %s: %s",
1062			    prv_tmp, strerror(errno));
1063			goto failnext;
1064		}
1065		close(fd); /* just using mkstemp() to generate/reserve a name */
1066		bits = 0;
1067		type_bits_valid(type, NULL, &bits);
1068		if ((r = sshkey_generate(type, bits, &private)) != 0) {
1069			error("sshkey_generate failed: %s", ssh_err(r));
1070			goto failnext;
1071		}
1072		if ((r = sshkey_from_private(private, &public)) != 0)
1073			fatal("sshkey_from_private failed: %s", ssh_err(r));
1074		snprintf(comment, sizeof comment, "%s@%s", pw->pw_name,
1075		    hostname);
1076		if ((r = sshkey_save_private(private, prv_tmp, "",
1077		    comment, use_new_format, new_format_cipher, rounds)) != 0) {
1078			error("Saving key \"%s\" failed: %s",
1079			    prv_tmp, ssh_err(r));
1080			goto failnext;
1081		}
1082		if ((fd = mkstemp(pub_tmp)) == -1) {
1083			error("Could not save your public key in %s: %s",
1084			    pub_tmp, strerror(errno));
1085			goto failnext;
1086		}
1087		(void)fchmod(fd, 0644);
1088		f = fdopen(fd, "w");
1089		if (f == NULL) {
1090			error("fdopen %s failed: %s", pub_tmp, strerror(errno));
1091			close(fd);
1092			goto failnext;
1093		}
1094		if ((r = sshkey_write(public, f)) != 0) {
1095			error("write key failed: %s", ssh_err(r));
1096			fclose(f);
1097			goto failnext;
1098		}
1099		fprintf(f, " %s\n", comment);
1100		if (ferror(f) != 0) {
1101			error("write key failed: %s", strerror(errno));
1102			fclose(f);
1103			goto failnext;
1104		}
1105		if (fclose(f) != 0) {
1106			error("key close failed: %s", strerror(errno));
1107			goto failnext;
1108		}
1109
1110		/* Rename temporary files to their permanent locations. */
1111		if (rename(pub_tmp, pub_file) != 0) {
1112			error("Unable to move %s into position: %s",
1113			    pub_file, strerror(errno));
1114			goto failnext;
1115		}
1116		if (rename(prv_tmp, prv_file) != 0) {
1117			error("Unable to move %s into position: %s",
1118			    key_types[i].path, strerror(errno));
1119 failnext:
1120			first = 0;
1121			goto next;
1122		}
1123 next:
1124		sshkey_free(private);
1125		sshkey_free(public);
1126		free(prv_tmp);
1127		free(pub_tmp);
1128		free(prv_file);
1129		free(pub_file);
1130	}
1131	if (first != 0)
1132		printf("\n");
1133}
1134
1135struct known_hosts_ctx {
1136	const char *host;	/* Hostname searched for in find/delete case */
1137	FILE *out;		/* Output file, stdout for find_hosts case */
1138	int has_unhashed;	/* When hashing, original had unhashed hosts */
1139	int found_key;		/* For find/delete, host was found */
1140	int invalid;		/* File contained invalid items; don't delete */
1141};
1142
1143static int
1144known_hosts_hash(struct hostkey_foreach_line *l, void *_ctx)
1145{
1146	struct known_hosts_ctx *ctx = (struct known_hosts_ctx *)_ctx;
1147	char *hashed, *cp, *hosts, *ohosts;
1148	int has_wild = l->hosts && strcspn(l->hosts, "*?!") != strlen(l->hosts);
1149	int was_hashed = l->hosts && l->hosts[0] == HASH_DELIM;
1150
1151	switch (l->status) {
1152	case HKF_STATUS_OK:
1153	case HKF_STATUS_MATCHED:
1154		/*
1155		 * Don't hash hosts already already hashed, with wildcard
1156		 * characters or a CA/revocation marker.
1157		 */
1158		if (was_hashed || has_wild || l->marker != MRK_NONE) {
1159			fprintf(ctx->out, "%s\n", l->line);
1160			if (has_wild && !find_host) {
1161				logit("%s:%lu: ignoring host name "
1162				    "with wildcard: %.64s", l->path,
1163				    l->linenum, l->hosts);
1164			}
1165			return 0;
1166		}
1167		/*
1168		 * Split any comma-separated hostnames from the host list,
1169		 * hash and store separately.
1170		 */
1171		ohosts = hosts = xstrdup(l->hosts);
1172		while ((cp = strsep(&hosts, ",")) != NULL && *cp != '\0') {
1173			lowercase(cp);
1174			if ((hashed = host_hash(cp, NULL, 0)) == NULL)
1175				fatal("hash_host failed");
1176			fprintf(ctx->out, "%s %s\n", hashed, l->rawkey);
1177			ctx->has_unhashed = 1;
1178		}
1179		free(ohosts);
1180		return 0;
1181	case HKF_STATUS_INVALID:
1182		/* Retain invalid lines, but mark file as invalid. */
1183		ctx->invalid = 1;
1184		logit("%s:%lu: invalid line", l->path, l->linenum);
1185		/* FALLTHROUGH */
1186	default:
1187		fprintf(ctx->out, "%s\n", l->line);
1188		return 0;
1189	}
1190	/* NOTREACHED */
1191	return -1;
1192}
1193
1194static int
1195known_hosts_find_delete(struct hostkey_foreach_line *l, void *_ctx)
1196{
1197	struct known_hosts_ctx *ctx = (struct known_hosts_ctx *)_ctx;
1198	enum sshkey_fp_rep rep;
1199	int fptype;
1200	char *fp;
1201
1202	fptype = print_bubblebabble ? SSH_DIGEST_SHA1 : fingerprint_hash;
1203	rep =    print_bubblebabble ? SSH_FP_BUBBLEBABBLE : SSH_FP_DEFAULT;
1204
1205	if (l->status == HKF_STATUS_MATCHED) {
1206		if (delete_host) {
1207			if (l->marker != MRK_NONE) {
1208				/* Don't remove CA and revocation lines */
1209				fprintf(ctx->out, "%s\n", l->line);
1210			} else {
1211				/*
1212				 * Hostname matches and has no CA/revoke
1213				 * marker, delete it by *not* writing the
1214				 * line to ctx->out.
1215				 */
1216				ctx->found_key = 1;
1217				if (!quiet)
1218					printf("# Host %s found: line %lu\n",
1219					    ctx->host, l->linenum);
1220			}
1221			return 0;
1222		} else if (find_host) {
1223			ctx->found_key = 1;
1224			if (!quiet) {
1225				printf("# Host %s found: line %lu %s\n",
1226				    ctx->host,
1227				    l->linenum, l->marker == MRK_CA ? "CA" :
1228				    (l->marker == MRK_REVOKE ? "REVOKED" : ""));
1229			}
1230			if (hash_hosts)
1231				known_hosts_hash(l, ctx);
1232			else if (print_fingerprint) {
1233				fp = sshkey_fingerprint(l->key, fptype, rep);
1234				mprintf("%s %s %s %s\n", ctx->host,
1235				    sshkey_type(l->key), fp, l->comment);
1236				free(fp);
1237			} else
1238				fprintf(ctx->out, "%s\n", l->line);
1239			return 0;
1240		}
1241	} else if (delete_host) {
1242		/* Retain non-matching hosts when deleting */
1243		if (l->status == HKF_STATUS_INVALID) {
1244			ctx->invalid = 1;
1245			logit("%s:%lu: invalid line", l->path, l->linenum);
1246		}
1247		fprintf(ctx->out, "%s\n", l->line);
1248	}
1249	return 0;
1250}
1251
1252static void
1253do_known_hosts(struct passwd *pw, const char *name)
1254{
1255	char *cp, tmp[PATH_MAX], old[PATH_MAX];
1256	int r, fd, oerrno, inplace = 0;
1257	struct known_hosts_ctx ctx;
1258	u_int foreach_options;
1259
1260	if (!have_identity) {
1261		cp = tilde_expand_filename(_PATH_SSH_USER_HOSTFILE, pw->pw_uid);
1262		if (strlcpy(identity_file, cp, sizeof(identity_file)) >=
1263		    sizeof(identity_file))
1264			fatal("Specified known hosts path too long");
1265		free(cp);
1266		have_identity = 1;
1267	}
1268
1269	memset(&ctx, 0, sizeof(ctx));
1270	ctx.out = stdout;
1271	ctx.host = name;
1272
1273	/*
1274	 * Find hosts goes to stdout, hash and deletions happen in-place
1275	 * A corner case is ssh-keygen -HF foo, which should go to stdout
1276	 */
1277	if (!find_host && (hash_hosts || delete_host)) {
1278		if (strlcpy(tmp, identity_file, sizeof(tmp)) >= sizeof(tmp) ||
1279		    strlcat(tmp, ".XXXXXXXXXX", sizeof(tmp)) >= sizeof(tmp) ||
1280		    strlcpy(old, identity_file, sizeof(old)) >= sizeof(old) ||
1281		    strlcat(old, ".old", sizeof(old)) >= sizeof(old))
1282			fatal("known_hosts path too long");
1283		umask(077);
1284		if ((fd = mkstemp(tmp)) == -1)
1285			fatal("mkstemp: %s", strerror(errno));
1286		if ((ctx.out = fdopen(fd, "w")) == NULL) {
1287			oerrno = errno;
1288			unlink(tmp);
1289			fatal("fdopen: %s", strerror(oerrno));
1290		}
1291		inplace = 1;
1292	}
1293	/* XXX support identity_file == "-" for stdin */
1294	foreach_options = find_host ? HKF_WANT_MATCH : 0;
1295	foreach_options |= print_fingerprint ? HKF_WANT_PARSE_KEY : 0;
1296	if ((r = hostkeys_foreach(identity_file, (find_host || !hash_hosts) ?
1297	    known_hosts_find_delete : known_hosts_hash, &ctx, name, NULL,
1298	    foreach_options)) != 0) {
1299		if (inplace)
1300			unlink(tmp);
1301		fatal("%s: hostkeys_foreach failed: %s", __func__, ssh_err(r));
1302	}
1303
1304	if (inplace)
1305		fclose(ctx.out);
1306
1307	if (ctx.invalid) {
1308		error("%s is not a valid known_hosts file.", identity_file);
1309		if (inplace) {
1310			error("Not replacing existing known_hosts "
1311			    "file because of errors");
1312			unlink(tmp);
1313		}
1314		exit(1);
1315	} else if (delete_host && !ctx.found_key) {
1316		logit("Host %s not found in %s", name, identity_file);
1317		if (inplace)
1318			unlink(tmp);
1319	} else if (inplace) {
1320		/* Backup existing file */
1321		if (unlink(old) == -1 && errno != ENOENT)
1322			fatal("unlink %.100s: %s", old, strerror(errno));
1323		if (link(identity_file, old) == -1)
1324			fatal("link %.100s to %.100s: %s", identity_file, old,
1325			    strerror(errno));
1326		/* Move new one into place */
1327		if (rename(tmp, identity_file) == -1) {
1328			error("rename\"%s\" to \"%s\": %s", tmp, identity_file,
1329			    strerror(errno));
1330			unlink(tmp);
1331			unlink(old);
1332			exit(1);
1333		}
1334
1335		printf("%s updated.\n", identity_file);
1336		printf("Original contents retained as %s\n", old);
1337		if (ctx.has_unhashed) {
1338			logit("WARNING: %s contains unhashed entries", old);
1339			logit("Delete this file to ensure privacy "
1340			    "of hostnames");
1341		}
1342	}
1343
1344	exit (find_host && !ctx.found_key);
1345}
1346
1347/*
1348 * Perform changing a passphrase.  The argument is the passwd structure
1349 * for the current user.
1350 */
1351static void
1352do_change_passphrase(struct passwd *pw)
1353{
1354	char *comment;
1355	char *old_passphrase, *passphrase1, *passphrase2;
1356	struct stat st;
1357	struct sshkey *private;
1358	int r;
1359
1360	if (!have_identity)
1361		ask_filename(pw, "Enter file in which the key is");
1362	if (stat(identity_file, &st) < 0)
1363		fatal("%s: %s", identity_file, strerror(errno));
1364	/* Try to load the file with empty passphrase. */
1365	r = sshkey_load_private(identity_file, "", &private, &comment);
1366	if (r == SSH_ERR_KEY_WRONG_PASSPHRASE) {
1367		if (identity_passphrase)
1368			old_passphrase = xstrdup(identity_passphrase);
1369		else
1370			old_passphrase =
1371			    read_passphrase("Enter old passphrase: ",
1372			    RP_ALLOW_STDIN);
1373		r = sshkey_load_private(identity_file, old_passphrase,
1374		    &private, &comment);
1375		explicit_bzero(old_passphrase, strlen(old_passphrase));
1376		free(old_passphrase);
1377		if (r != 0)
1378			goto badkey;
1379	} else if (r != 0) {
1380 badkey:
1381		fatal("Failed to load key %s: %s", identity_file, ssh_err(r));
1382	}
1383	if (comment)
1384		mprintf("Key has comment '%s'\n", comment);
1385
1386	/* Ask the new passphrase (twice). */
1387	if (identity_new_passphrase) {
1388		passphrase1 = xstrdup(identity_new_passphrase);
1389		passphrase2 = NULL;
1390	} else {
1391		passphrase1 =
1392			read_passphrase("Enter new passphrase (empty for no "
1393			    "passphrase): ", RP_ALLOW_STDIN);
1394		passphrase2 = read_passphrase("Enter same passphrase again: ",
1395		    RP_ALLOW_STDIN);
1396
1397		/* Verify that they are the same. */
1398		if (strcmp(passphrase1, passphrase2) != 0) {
1399			explicit_bzero(passphrase1, strlen(passphrase1));
1400			explicit_bzero(passphrase2, strlen(passphrase2));
1401			free(passphrase1);
1402			free(passphrase2);
1403			printf("Pass phrases do not match.  Try again.\n");
1404			exit(1);
1405		}
1406		/* Destroy the other copy. */
1407		explicit_bzero(passphrase2, strlen(passphrase2));
1408		free(passphrase2);
1409	}
1410
1411	/* Save the file using the new passphrase. */
1412	if ((r = sshkey_save_private(private, identity_file, passphrase1,
1413	    comment, use_new_format, new_format_cipher, rounds)) != 0) {
1414		error("Saving key \"%s\" failed: %s.",
1415		    identity_file, ssh_err(r));
1416		explicit_bzero(passphrase1, strlen(passphrase1));
1417		free(passphrase1);
1418		sshkey_free(private);
1419		free(comment);
1420		exit(1);
1421	}
1422	/* Destroy the passphrase and the copy of the key in memory. */
1423	explicit_bzero(passphrase1, strlen(passphrase1));
1424	free(passphrase1);
1425	sshkey_free(private);		 /* Destroys contents */
1426	free(comment);
1427
1428	printf("Your identification has been saved with the new passphrase.\n");
1429	exit(0);
1430}
1431
1432/*
1433 * Print the SSHFP RR.
1434 */
1435static int
1436do_print_resource_record(struct passwd *pw, char *fname, char *hname)
1437{
1438	struct sshkey *public;
1439	char *comment = NULL;
1440	struct stat st;
1441	int r;
1442
1443	if (fname == NULL)
1444		fatal("%s: no filename", __func__);
1445	if (stat(fname, &st) < 0) {
1446		if (errno == ENOENT)
1447			return 0;
1448		fatal("%s: %s", fname, strerror(errno));
1449	}
1450	if ((r = sshkey_load_public(fname, &public, &comment)) != 0)
1451		fatal("Failed to read v2 public key from \"%s\": %s.",
1452		    fname, ssh_err(r));
1453	export_dns_rr(hname, public, stdout, print_generic);
1454	sshkey_free(public);
1455	free(comment);
1456	return 1;
1457}
1458
1459/*
1460 * Change the comment of a private key file.
1461 */
1462static void
1463do_change_comment(struct passwd *pw)
1464{
1465	char new_comment[1024], *comment, *passphrase;
1466	struct sshkey *private;
1467	struct sshkey *public;
1468	struct stat st;
1469	FILE *f;
1470	int r, fd;
1471
1472	if (!have_identity)
1473		ask_filename(pw, "Enter file in which the key is");
1474	if (stat(identity_file, &st) < 0)
1475		fatal("%s: %s", identity_file, strerror(errno));
1476	if ((r = sshkey_load_private(identity_file, "",
1477	    &private, &comment)) == 0)
1478		passphrase = xstrdup("");
1479	else if (r != SSH_ERR_KEY_WRONG_PASSPHRASE)
1480		fatal("Cannot load private key \"%s\": %s.",
1481		    identity_file, ssh_err(r));
1482	else {
1483		if (identity_passphrase)
1484			passphrase = xstrdup(identity_passphrase);
1485		else if (identity_new_passphrase)
1486			passphrase = xstrdup(identity_new_passphrase);
1487		else
1488			passphrase = read_passphrase("Enter passphrase: ",
1489			    RP_ALLOW_STDIN);
1490		/* Try to load using the passphrase. */
1491		if ((r = sshkey_load_private(identity_file, passphrase,
1492		    &private, &comment)) != 0) {
1493			explicit_bzero(passphrase, strlen(passphrase));
1494			free(passphrase);
1495			fatal("Cannot load private key \"%s\": %s.",
1496			    identity_file, ssh_err(r));
1497		}
1498	}
1499
1500	if (private->type != KEY_ED25519 && private->type != KEY_XMSS &&
1501	    !use_new_format) {
1502		error("Comments are only supported for keys stored in "
1503		    "the new format (-o).");
1504		explicit_bzero(passphrase, strlen(passphrase));
1505		sshkey_free(private);
1506		exit(1);
1507	}
1508	if (comment)
1509		printf("Key now has comment '%s'\n", comment);
1510	else
1511		printf("Key now has no comment\n");
1512
1513	if (identity_comment) {
1514		strlcpy(new_comment, identity_comment, sizeof(new_comment));
1515	} else {
1516		printf("Enter new comment: ");
1517		fflush(stdout);
1518		if (!fgets(new_comment, sizeof(new_comment), stdin)) {
1519			explicit_bzero(passphrase, strlen(passphrase));
1520			sshkey_free(private);
1521			exit(1);
1522		}
1523		new_comment[strcspn(new_comment, "\n")] = '\0';
1524	}
1525
1526	/* Save the file using the new passphrase. */
1527	if ((r = sshkey_save_private(private, identity_file, passphrase,
1528	    new_comment, use_new_format, new_format_cipher, rounds)) != 0) {
1529		error("Saving key \"%s\" failed: %s",
1530		    identity_file, ssh_err(r));
1531		explicit_bzero(passphrase, strlen(passphrase));
1532		free(passphrase);
1533		sshkey_free(private);
1534		free(comment);
1535		exit(1);
1536	}
1537	explicit_bzero(passphrase, strlen(passphrase));
1538	free(passphrase);
1539	if ((r = sshkey_from_private(private, &public)) != 0)
1540		fatal("sshkey_from_private failed: %s", ssh_err(r));
1541	sshkey_free(private);
1542
1543	strlcat(identity_file, ".pub", sizeof(identity_file));
1544	fd = open(identity_file, O_WRONLY | O_CREAT | O_TRUNC, 0644);
1545	if (fd == -1)
1546		fatal("Could not save your public key in %s", identity_file);
1547	f = fdopen(fd, "w");
1548	if (f == NULL)
1549		fatal("fdopen %s failed: %s", identity_file, strerror(errno));
1550	if ((r = sshkey_write(public, f)) != 0)
1551		fatal("write key failed: %s", ssh_err(r));
1552	sshkey_free(public);
1553	fprintf(f, " %s\n", new_comment);
1554	fclose(f);
1555
1556	free(comment);
1557
1558	printf("The comment in your key file has been changed.\n");
1559	exit(0);
1560}
1561
1562static void
1563add_flag_option(struct sshbuf *c, const char *name)
1564{
1565	int r;
1566
1567	debug3("%s: %s", __func__, name);
1568	if ((r = sshbuf_put_cstring(c, name)) != 0 ||
1569	    (r = sshbuf_put_string(c, NULL, 0)) != 0)
1570		fatal("%s: buffer error: %s", __func__, ssh_err(r));
1571}
1572
1573static void
1574add_string_option(struct sshbuf *c, const char *name, const char *value)
1575{
1576	struct sshbuf *b;
1577	int r;
1578
1579	debug3("%s: %s=%s", __func__, name, value);
1580	if ((b = sshbuf_new()) == NULL)
1581		fatal("%s: sshbuf_new failed", __func__);
1582	if ((r = sshbuf_put_cstring(b, value)) != 0 ||
1583	    (r = sshbuf_put_cstring(c, name)) != 0 ||
1584	    (r = sshbuf_put_stringb(c, b)) != 0)
1585		fatal("%s: buffer error: %s", __func__, ssh_err(r));
1586
1587	sshbuf_free(b);
1588}
1589
1590#define OPTIONS_CRITICAL	1
1591#define OPTIONS_EXTENSIONS	2
1592static void
1593prepare_options_buf(struct sshbuf *c, int which)
1594{
1595	size_t i;
1596
1597	sshbuf_reset(c);
1598	if ((which & OPTIONS_CRITICAL) != 0 &&
1599	    certflags_command != NULL)
1600		add_string_option(c, "force-command", certflags_command);
1601	if ((which & OPTIONS_EXTENSIONS) != 0 &&
1602	    (certflags_flags & CERTOPT_X_FWD) != 0)
1603		add_flag_option(c, "permit-X11-forwarding");
1604	if ((which & OPTIONS_EXTENSIONS) != 0 &&
1605	    (certflags_flags & CERTOPT_AGENT_FWD) != 0)
1606		add_flag_option(c, "permit-agent-forwarding");
1607	if ((which & OPTIONS_EXTENSIONS) != 0 &&
1608	    (certflags_flags & CERTOPT_PORT_FWD) != 0)
1609		add_flag_option(c, "permit-port-forwarding");
1610	if ((which & OPTIONS_EXTENSIONS) != 0 &&
1611	    (certflags_flags & CERTOPT_PTY) != 0)
1612		add_flag_option(c, "permit-pty");
1613	if ((which & OPTIONS_EXTENSIONS) != 0 &&
1614	    (certflags_flags & CERTOPT_USER_RC) != 0)
1615		add_flag_option(c, "permit-user-rc");
1616	if ((which & OPTIONS_CRITICAL) != 0 &&
1617	    certflags_src_addr != NULL)
1618		add_string_option(c, "source-address", certflags_src_addr);
1619	for (i = 0; i < ncert_userext; i++) {
1620		if ((cert_userext[i].crit && (which & OPTIONS_EXTENSIONS)) ||
1621		    (!cert_userext[i].crit && (which & OPTIONS_CRITICAL)))
1622			continue;
1623		if (cert_userext[i].val == NULL)
1624			add_flag_option(c, cert_userext[i].key);
1625		else {
1626			add_string_option(c, cert_userext[i].key,
1627			    cert_userext[i].val);
1628		}
1629	}
1630}
1631
1632static struct sshkey *
1633load_pkcs11_key(char *path)
1634{
1635#ifdef ENABLE_PKCS11
1636	struct sshkey **keys = NULL, *public, *private = NULL;
1637	int r, i, nkeys;
1638
1639	if ((r = sshkey_load_public(path, &public, NULL)) != 0)
1640		fatal("Couldn't load CA public key \"%s\": %s",
1641		    path, ssh_err(r));
1642
1643	nkeys = pkcs11_add_provider(pkcs11provider, identity_passphrase, &keys);
1644	debug3("%s: %d keys", __func__, nkeys);
1645	if (nkeys <= 0)
1646		fatal("cannot read public key from pkcs11");
1647	for (i = 0; i < nkeys; i++) {
1648		if (sshkey_equal_public(public, keys[i])) {
1649			private = keys[i];
1650			continue;
1651		}
1652		sshkey_free(keys[i]);
1653	}
1654	free(keys);
1655	sshkey_free(public);
1656	return private;
1657#else
1658	fatal("no pkcs11 support");
1659#endif /* ENABLE_PKCS11 */
1660}
1661
1662/* Signer for sshkey_certify_custom that uses the agent */
1663static int
1664agent_signer(const struct sshkey *key, u_char **sigp, size_t *lenp,
1665    const u_char *data, size_t datalen,
1666    const char *alg, u_int compat, void *ctx)
1667{
1668	int *agent_fdp = (int *)ctx;
1669
1670	return ssh_agent_sign(*agent_fdp, key, sigp, lenp,
1671	    data, datalen, alg, compat);
1672}
1673
1674static void
1675do_ca_sign(struct passwd *pw, int argc, char **argv)
1676{
1677	int r, i, fd, found, agent_fd = -1;
1678	u_int n;
1679	struct sshkey *ca, *public;
1680	char valid[64], *otmp, *tmp, *cp, *out, *comment, **plist = NULL;
1681	FILE *f;
1682	struct ssh_identitylist *agent_ids;
1683	size_t j;
1684
1685#ifdef ENABLE_PKCS11
1686	pkcs11_init(1);
1687#endif
1688	tmp = tilde_expand_filename(ca_key_path, pw->pw_uid);
1689	if (pkcs11provider != NULL) {
1690		/* If a PKCS#11 token was specified then try to use it */
1691		if ((ca = load_pkcs11_key(tmp)) == NULL)
1692			fatal("No PKCS#11 key matching %s found", ca_key_path);
1693	} else if (prefer_agent) {
1694		/*
1695		 * Agent signature requested. Try to use agent after making
1696		 * sure the public key specified is actually present in the
1697		 * agent.
1698		 */
1699		if ((r = sshkey_load_public(tmp, &ca, NULL)) != 0)
1700			fatal("Cannot load CA public key %s: %s",
1701			    tmp, ssh_err(r));
1702		if ((r = ssh_get_authentication_socket(&agent_fd)) != 0)
1703			fatal("Cannot use public key for CA signature: %s",
1704			    ssh_err(r));
1705		if ((r = ssh_fetch_identitylist(agent_fd, &agent_ids)) != 0)
1706			fatal("Retrieve agent key list: %s", ssh_err(r));
1707		found = 0;
1708		for (j = 0; j < agent_ids->nkeys; j++) {
1709			if (sshkey_equal(ca, agent_ids->keys[j])) {
1710				found = 1;
1711				break;
1712			}
1713		}
1714		if (!found)
1715			fatal("CA key %s not found in agent", tmp);
1716		ssh_free_identitylist(agent_ids);
1717		ca->flags |= SSHKEY_FLAG_EXT;
1718	} else {
1719		/* CA key is assumed to be a private key on the filesystem */
1720		ca = load_identity(tmp);
1721	}
1722	free(tmp);
1723
1724	if (key_type_name != NULL &&
1725	    sshkey_type_from_name(key_type_name) != ca->type)  {
1726		fatal("CA key type %s doesn't match specified %s",
1727		    sshkey_ssh_name(ca), key_type_name);
1728	}
1729
1730	for (i = 0; i < argc; i++) {
1731		/* Split list of principals */
1732		n = 0;
1733		if (cert_principals != NULL) {
1734			otmp = tmp = xstrdup(cert_principals);
1735			plist = NULL;
1736			for (; (cp = strsep(&tmp, ",")) != NULL; n++) {
1737				plist = xreallocarray(plist, n + 1, sizeof(*plist));
1738				if (*(plist[n] = xstrdup(cp)) == '\0')
1739					fatal("Empty principal name");
1740			}
1741			free(otmp);
1742		}
1743		if (n > SSHKEY_CERT_MAX_PRINCIPALS)
1744			fatal("Too many certificate principals specified");
1745
1746		tmp = tilde_expand_filename(argv[i], pw->pw_uid);
1747		if ((r = sshkey_load_public(tmp, &public, &comment)) != 0)
1748			fatal("%s: unable to open \"%s\": %s",
1749			    __func__, tmp, ssh_err(r));
1750		if (public->type != KEY_RSA && public->type != KEY_DSA &&
1751		    public->type != KEY_ECDSA && public->type != KEY_ED25519 &&
1752		    public->type != KEY_XMSS)
1753			fatal("%s: key \"%s\" type %s cannot be certified",
1754			    __func__, tmp, sshkey_type(public));
1755
1756		/* Prepare certificate to sign */
1757		if ((r = sshkey_to_certified(public)) != 0)
1758			fatal("Could not upgrade key %s to certificate: %s",
1759			    tmp, ssh_err(r));
1760		public->cert->type = cert_key_type;
1761		public->cert->serial = (u_int64_t)cert_serial;
1762		public->cert->key_id = xstrdup(cert_key_id);
1763		public->cert->nprincipals = n;
1764		public->cert->principals = plist;
1765		public->cert->valid_after = cert_valid_from;
1766		public->cert->valid_before = cert_valid_to;
1767		prepare_options_buf(public->cert->critical, OPTIONS_CRITICAL);
1768		prepare_options_buf(public->cert->extensions,
1769		    OPTIONS_EXTENSIONS);
1770		if ((r = sshkey_from_private(ca,
1771		    &public->cert->signature_key)) != 0)
1772			fatal("sshkey_from_private (ca key): %s", ssh_err(r));
1773
1774		if (agent_fd != -1 && (ca->flags & SSHKEY_FLAG_EXT) != 0) {
1775			if ((r = sshkey_certify_custom(public, ca,
1776			    key_type_name, agent_signer, &agent_fd)) != 0)
1777				fatal("Couldn't certify key %s via agent: %s",
1778				    tmp, ssh_err(r));
1779		} else {
1780			if ((sshkey_certify(public, ca, key_type_name)) != 0)
1781				fatal("Couldn't certify key %s: %s",
1782				    tmp, ssh_err(r));
1783		}
1784
1785		if ((cp = strrchr(tmp, '.')) != NULL && strcmp(cp, ".pub") == 0)
1786			*cp = '\0';
1787		xasprintf(&out, "%s-cert.pub", tmp);
1788		free(tmp);
1789
1790		if ((fd = open(out, O_WRONLY|O_CREAT|O_TRUNC, 0644)) == -1)
1791			fatal("Could not open \"%s\" for writing: %s", out,
1792			    strerror(errno));
1793		if ((f = fdopen(fd, "w")) == NULL)
1794			fatal("%s: fdopen: %s", __func__, strerror(errno));
1795		if ((r = sshkey_write(public, f)) != 0)
1796			fatal("Could not write certified key to %s: %s",
1797			    out, ssh_err(r));
1798		fprintf(f, " %s\n", comment);
1799		fclose(f);
1800
1801		if (!quiet) {
1802			sshkey_format_cert_validity(public->cert,
1803			    valid, sizeof(valid));
1804			logit("Signed %s key %s: id \"%s\" serial %llu%s%s "
1805			    "valid %s", sshkey_cert_type(public),
1806			    out, public->cert->key_id,
1807			    (unsigned long long)public->cert->serial,
1808			    cert_principals != NULL ? " for " : "",
1809			    cert_principals != NULL ? cert_principals : "",
1810			    valid);
1811		}
1812
1813		sshkey_free(public);
1814		free(out);
1815	}
1816#ifdef ENABLE_PKCS11
1817	pkcs11_terminate();
1818#endif
1819	exit(0);
1820}
1821
1822static u_int64_t
1823parse_relative_time(const char *s, time_t now)
1824{
1825	int64_t mul, secs;
1826
1827	mul = *s == '-' ? -1 : 1;
1828
1829	if ((secs = convtime(s + 1)) == -1)
1830		fatal("Invalid relative certificate time %s", s);
1831	if (mul == -1 && secs > now)
1832		fatal("Certificate time %s cannot be represented", s);
1833	return now + (u_int64_t)(secs * mul);
1834}
1835
1836static void
1837parse_cert_times(char *timespec)
1838{
1839	char *from, *to;
1840	time_t now = time(NULL);
1841	int64_t secs;
1842
1843	/* +timespec relative to now */
1844	if (*timespec == '+' && strchr(timespec, ':') == NULL) {
1845		if ((secs = convtime(timespec + 1)) == -1)
1846			fatal("Invalid relative certificate life %s", timespec);
1847		cert_valid_to = now + secs;
1848		/*
1849		 * Backdate certificate one minute to avoid problems on hosts
1850		 * with poorly-synchronised clocks.
1851		 */
1852		cert_valid_from = ((now - 59)/ 60) * 60;
1853		return;
1854	}
1855
1856	/*
1857	 * from:to, where
1858	 * from := [+-]timespec | YYYYMMDD | YYYYMMDDHHMMSS | "always"
1859	 *   to := [+-]timespec | YYYYMMDD | YYYYMMDDHHMMSS | "forever"
1860	 */
1861	from = xstrdup(timespec);
1862	to = strchr(from, ':');
1863	if (to == NULL || from == to || *(to + 1) == '\0')
1864		fatal("Invalid certificate life specification %s", timespec);
1865	*to++ = '\0';
1866
1867	if (*from == '-' || *from == '+')
1868		cert_valid_from = parse_relative_time(from, now);
1869	else if (strcmp(from, "always") == 0)
1870		cert_valid_from = 0;
1871	else if (parse_absolute_time(from, &cert_valid_from) != 0)
1872		fatal("Invalid from time \"%s\"", from);
1873
1874	if (*to == '-' || *to == '+')
1875		cert_valid_to = parse_relative_time(to, now);
1876	else if (strcmp(to, "forever") == 0)
1877		cert_valid_to = ~(u_int64_t)0;
1878	else if (parse_absolute_time(to, &cert_valid_to) != 0)
1879		fatal("Invalid to time \"%s\"", to);
1880
1881	if (cert_valid_to <= cert_valid_from)
1882		fatal("Empty certificate validity interval");
1883	free(from);
1884}
1885
1886static void
1887add_cert_option(char *opt)
1888{
1889	char *val, *cp;
1890	int iscrit = 0;
1891
1892	if (strcasecmp(opt, "clear") == 0)
1893		certflags_flags = 0;
1894	else if (strcasecmp(opt, "no-x11-forwarding") == 0)
1895		certflags_flags &= ~CERTOPT_X_FWD;
1896	else if (strcasecmp(opt, "permit-x11-forwarding") == 0)
1897		certflags_flags |= CERTOPT_X_FWD;
1898	else if (strcasecmp(opt, "no-agent-forwarding") == 0)
1899		certflags_flags &= ~CERTOPT_AGENT_FWD;
1900	else if (strcasecmp(opt, "permit-agent-forwarding") == 0)
1901		certflags_flags |= CERTOPT_AGENT_FWD;
1902	else if (strcasecmp(opt, "no-port-forwarding") == 0)
1903		certflags_flags &= ~CERTOPT_PORT_FWD;
1904	else if (strcasecmp(opt, "permit-port-forwarding") == 0)
1905		certflags_flags |= CERTOPT_PORT_FWD;
1906	else if (strcasecmp(opt, "no-pty") == 0)
1907		certflags_flags &= ~CERTOPT_PTY;
1908	else if (strcasecmp(opt, "permit-pty") == 0)
1909		certflags_flags |= CERTOPT_PTY;
1910	else if (strcasecmp(opt, "no-user-rc") == 0)
1911		certflags_flags &= ~CERTOPT_USER_RC;
1912	else if (strcasecmp(opt, "permit-user-rc") == 0)
1913		certflags_flags |= CERTOPT_USER_RC;
1914	else if (strncasecmp(opt, "force-command=", 14) == 0) {
1915		val = opt + 14;
1916		if (*val == '\0')
1917			fatal("Empty force-command option");
1918		if (certflags_command != NULL)
1919			fatal("force-command already specified");
1920		certflags_command = xstrdup(val);
1921	} else if (strncasecmp(opt, "source-address=", 15) == 0) {
1922		val = opt + 15;
1923		if (*val == '\0')
1924			fatal("Empty source-address option");
1925		if (certflags_src_addr != NULL)
1926			fatal("source-address already specified");
1927		if (addr_match_cidr_list(NULL, val) != 0)
1928			fatal("Invalid source-address list");
1929		certflags_src_addr = xstrdup(val);
1930	} else if (strncasecmp(opt, "extension:", 10) == 0 ||
1931		   (iscrit = (strncasecmp(opt, "critical:", 9) == 0))) {
1932		val = xstrdup(strchr(opt, ':') + 1);
1933		if ((cp = strchr(val, '=')) != NULL)
1934			*cp++ = '\0';
1935		cert_userext = xreallocarray(cert_userext, ncert_userext + 1,
1936		    sizeof(*cert_userext));
1937		cert_userext[ncert_userext].key = val;
1938		cert_userext[ncert_userext].val = cp == NULL ?
1939		    NULL : xstrdup(cp);
1940		cert_userext[ncert_userext].crit = iscrit;
1941		ncert_userext++;
1942	} else
1943		fatal("Unsupported certificate option \"%s\"", opt);
1944}
1945
1946static void
1947show_options(struct sshbuf *optbuf, int in_critical)
1948{
1949	char *name, *arg;
1950	struct sshbuf *options, *option = NULL;
1951	int r;
1952
1953	if ((options = sshbuf_fromb(optbuf)) == NULL)
1954		fatal("%s: sshbuf_fromb failed", __func__);
1955	while (sshbuf_len(options) != 0) {
1956		sshbuf_free(option);
1957		option = NULL;
1958		if ((r = sshbuf_get_cstring(options, &name, NULL)) != 0 ||
1959		    (r = sshbuf_froms(options, &option)) != 0)
1960			fatal("%s: buffer error: %s", __func__, ssh_err(r));
1961		printf("                %s", name);
1962		if (!in_critical &&
1963		    (strcmp(name, "permit-X11-forwarding") == 0 ||
1964		    strcmp(name, "permit-agent-forwarding") == 0 ||
1965		    strcmp(name, "permit-port-forwarding") == 0 ||
1966		    strcmp(name, "permit-pty") == 0 ||
1967		    strcmp(name, "permit-user-rc") == 0))
1968			printf("\n");
1969		else if (in_critical &&
1970		    (strcmp(name, "force-command") == 0 ||
1971		    strcmp(name, "source-address") == 0)) {
1972			if ((r = sshbuf_get_cstring(option, &arg, NULL)) != 0)
1973				fatal("%s: buffer error: %s",
1974				    __func__, ssh_err(r));
1975			printf(" %s\n", arg);
1976			free(arg);
1977		} else {
1978			printf(" UNKNOWN OPTION (len %zu)\n",
1979			    sshbuf_len(option));
1980			sshbuf_reset(option);
1981		}
1982		free(name);
1983		if (sshbuf_len(option) != 0)
1984			fatal("Option corrupt: extra data at end");
1985	}
1986	sshbuf_free(option);
1987	sshbuf_free(options);
1988}
1989
1990static void
1991print_cert(struct sshkey *key)
1992{
1993	char valid[64], *key_fp, *ca_fp;
1994	u_int i;
1995
1996	key_fp = sshkey_fingerprint(key, fingerprint_hash, SSH_FP_DEFAULT);
1997	ca_fp = sshkey_fingerprint(key->cert->signature_key,
1998	    fingerprint_hash, SSH_FP_DEFAULT);
1999	if (key_fp == NULL || ca_fp == NULL)
2000		fatal("%s: sshkey_fingerprint fail", __func__);
2001	sshkey_format_cert_validity(key->cert, valid, sizeof(valid));
2002
2003	printf("        Type: %s %s certificate\n", sshkey_ssh_name(key),
2004	    sshkey_cert_type(key));
2005	printf("        Public key: %s %s\n", sshkey_type(key), key_fp);
2006	printf("        Signing CA: %s %s\n",
2007	    sshkey_type(key->cert->signature_key), ca_fp);
2008	printf("        Key ID: \"%s\"\n", key->cert->key_id);
2009	printf("        Serial: %llu\n", (unsigned long long)key->cert->serial);
2010	printf("        Valid: %s\n", valid);
2011	printf("        Principals: ");
2012	if (key->cert->nprincipals == 0)
2013		printf("(none)\n");
2014	else {
2015		for (i = 0; i < key->cert->nprincipals; i++)
2016			printf("\n                %s",
2017			    key->cert->principals[i]);
2018		printf("\n");
2019	}
2020	printf("        Critical Options: ");
2021	if (sshbuf_len(key->cert->critical) == 0)
2022		printf("(none)\n");
2023	else {
2024		printf("\n");
2025		show_options(key->cert->critical, 1);
2026	}
2027	printf("        Extensions: ");
2028	if (sshbuf_len(key->cert->extensions) == 0)
2029		printf("(none)\n");
2030	else {
2031		printf("\n");
2032		show_options(key->cert->extensions, 0);
2033	}
2034}
2035
2036static void
2037do_show_cert(struct passwd *pw)
2038{
2039	struct sshkey *key = NULL;
2040	struct stat st;
2041	int r, is_stdin = 0, ok = 0;
2042	FILE *f;
2043	char *cp, *line = NULL;
2044	const char *path;
2045	size_t linesize = 0;
2046	u_long lnum = 0;
2047
2048	if (!have_identity)
2049		ask_filename(pw, "Enter file in which the key is");
2050	if (strcmp(identity_file, "-") != 0 && stat(identity_file, &st) < 0)
2051		fatal("%s: %s: %s", __progname, identity_file, strerror(errno));
2052
2053	path = identity_file;
2054	if (strcmp(path, "-") == 0) {
2055		f = stdin;
2056		path = "(stdin)";
2057		is_stdin = 1;
2058	} else if ((f = fopen(identity_file, "r")) == NULL)
2059		fatal("fopen %s: %s", identity_file, strerror(errno));
2060
2061	while (getline(&line, &linesize, f) != -1) {
2062		lnum++;
2063		sshkey_free(key);
2064		key = NULL;
2065		/* Trim leading space and comments */
2066		cp = line + strspn(line, " \t");
2067		if (*cp == '#' || *cp == '\0')
2068			continue;
2069		if ((key = sshkey_new(KEY_UNSPEC)) == NULL)
2070			fatal("sshkey_new");
2071		if ((r = sshkey_read(key, &cp)) != 0) {
2072			error("%s:%lu: invalid key: %s", path,
2073			    lnum, ssh_err(r));
2074			continue;
2075		}
2076		if (!sshkey_is_cert(key)) {
2077			error("%s:%lu is not a certificate", path, lnum);
2078			continue;
2079		}
2080		ok = 1;
2081		if (!is_stdin && lnum == 1)
2082			printf("%s:\n", path);
2083		else
2084			printf("%s:%lu:\n", path, lnum);
2085		print_cert(key);
2086	}
2087	free(line);
2088	sshkey_free(key);
2089	fclose(f);
2090	exit(ok ? 0 : 1);
2091}
2092
2093static void
2094load_krl(const char *path, struct ssh_krl **krlp)
2095{
2096	struct sshbuf *krlbuf;
2097	int r, fd;
2098
2099	if ((krlbuf = sshbuf_new()) == NULL)
2100		fatal("sshbuf_new failed");
2101	if ((fd = open(path, O_RDONLY)) == -1)
2102		fatal("open %s: %s", path, strerror(errno));
2103	if ((r = sshkey_load_file(fd, krlbuf)) != 0)
2104		fatal("Unable to load KRL: %s", ssh_err(r));
2105	close(fd);
2106	/* XXX check sigs */
2107	if ((r = ssh_krl_from_blob(krlbuf, krlp, NULL, 0)) != 0 ||
2108	    *krlp == NULL)
2109		fatal("Invalid KRL file: %s", ssh_err(r));
2110	sshbuf_free(krlbuf);
2111}
2112
2113static void
2114hash_to_blob(const char *cp, u_char **blobp, size_t *lenp,
2115    const char *file, u_long lnum)
2116{
2117	char *tmp;
2118	size_t tlen;
2119	struct sshbuf *b;
2120	int r;
2121
2122	if (strncmp(cp, "SHA256:", 7) != 0)
2123		fatal("%s:%lu: unsupported hash algorithm", file, lnum);
2124	cp += 7;
2125
2126	/*
2127	 * OpenSSH base64 hashes omit trailing '='
2128	 * characters; put them back for decode.
2129	 */
2130	tlen = strlen(cp);
2131	tmp = xmalloc(tlen + 4 + 1);
2132	strlcpy(tmp, cp, tlen + 1);
2133	while ((tlen % 4) != 0) {
2134		tmp[tlen++] = '=';
2135		tmp[tlen] = '\0';
2136	}
2137	if ((b = sshbuf_new()) == NULL)
2138		fatal("%s: sshbuf_new failed", __func__);
2139	if ((r = sshbuf_b64tod(b, tmp)) != 0)
2140		fatal("%s:%lu: decode hash failed: %s", file, lnum, ssh_err(r));
2141	free(tmp);
2142	*lenp = sshbuf_len(b);
2143	*blobp = xmalloc(*lenp);
2144	memcpy(*blobp, sshbuf_ptr(b), *lenp);
2145	sshbuf_free(b);
2146}
2147
2148static void
2149update_krl_from_file(struct passwd *pw, const char *file, int wild_ca,
2150    const struct sshkey *ca, struct ssh_krl *krl)
2151{
2152	struct sshkey *key = NULL;
2153	u_long lnum = 0;
2154	char *path, *cp, *ep, *line = NULL;
2155	u_char *blob = NULL;
2156	size_t blen = 0, linesize = 0;
2157	unsigned long long serial, serial2;
2158	int i, was_explicit_key, was_sha1, was_sha256, was_hash, r;
2159	FILE *krl_spec;
2160
2161	path = tilde_expand_filename(file, pw->pw_uid);
2162	if (strcmp(path, "-") == 0) {
2163		krl_spec = stdin;
2164		free(path);
2165		path = xstrdup("(standard input)");
2166	} else if ((krl_spec = fopen(path, "r")) == NULL)
2167		fatal("fopen %s: %s", path, strerror(errno));
2168
2169	if (!quiet)
2170		printf("Revoking from %s\n", path);
2171	while (getline(&line, &linesize, krl_spec) != -1) {
2172		lnum++;
2173		was_explicit_key = was_sha1 = was_sha256 = was_hash = 0;
2174		cp = line + strspn(line, " \t");
2175		/* Trim trailing space, comments and strip \n */
2176		for (i = 0, r = -1; cp[i] != '\0'; i++) {
2177			if (cp[i] == '#' || cp[i] == '\n') {
2178				cp[i] = '\0';
2179				break;
2180			}
2181			if (cp[i] == ' ' || cp[i] == '\t') {
2182				/* Remember the start of a span of whitespace */
2183				if (r == -1)
2184					r = i;
2185			} else
2186				r = -1;
2187		}
2188		if (r != -1)
2189			cp[r] = '\0';
2190		if (*cp == '\0')
2191			continue;
2192		if (strncasecmp(cp, "serial:", 7) == 0) {
2193			if (ca == NULL && !wild_ca) {
2194				fatal("revoking certificates by serial number "
2195				    "requires specification of a CA key");
2196			}
2197			cp += 7;
2198			cp = cp + strspn(cp, " \t");
2199			errno = 0;
2200			serial = strtoull(cp, &ep, 0);
2201			if (*cp == '\0' || (*ep != '\0' && *ep != '-'))
2202				fatal("%s:%lu: invalid serial \"%s\"",
2203				    path, lnum, cp);
2204			if (errno == ERANGE && serial == ULLONG_MAX)
2205				fatal("%s:%lu: serial out of range",
2206				    path, lnum);
2207			serial2 = serial;
2208			if (*ep == '-') {
2209				cp = ep + 1;
2210				errno = 0;
2211				serial2 = strtoull(cp, &ep, 0);
2212				if (*cp == '\0' || *ep != '\0')
2213					fatal("%s:%lu: invalid serial \"%s\"",
2214					    path, lnum, cp);
2215				if (errno == ERANGE && serial2 == ULLONG_MAX)
2216					fatal("%s:%lu: serial out of range",
2217					    path, lnum);
2218				if (serial2 <= serial)
2219					fatal("%s:%lu: invalid serial range "
2220					    "%llu:%llu", path, lnum,
2221					    (unsigned long long)serial,
2222					    (unsigned long long)serial2);
2223			}
2224			if (ssh_krl_revoke_cert_by_serial_range(krl,
2225			    ca, serial, serial2) != 0) {
2226				fatal("%s: revoke serial failed",
2227				    __func__);
2228			}
2229		} else if (strncasecmp(cp, "id:", 3) == 0) {
2230			if (ca == NULL && !wild_ca) {
2231				fatal("revoking certificates by key ID "
2232				    "requires specification of a CA key");
2233			}
2234			cp += 3;
2235			cp = cp + strspn(cp, " \t");
2236			if (ssh_krl_revoke_cert_by_key_id(krl, ca, cp) != 0)
2237				fatal("%s: revoke key ID failed", __func__);
2238		} else if (strncasecmp(cp, "hash:", 5) == 0) {
2239			cp += 5;
2240			cp = cp + strspn(cp, " \t");
2241			hash_to_blob(cp, &blob, &blen, file, lnum);
2242			r = ssh_krl_revoke_key_sha256(krl, blob, blen);
2243		} else {
2244			if (strncasecmp(cp, "key:", 4) == 0) {
2245				cp += 4;
2246				cp = cp + strspn(cp, " \t");
2247				was_explicit_key = 1;
2248			} else if (strncasecmp(cp, "sha1:", 5) == 0) {
2249				cp += 5;
2250				cp = cp + strspn(cp, " \t");
2251				was_sha1 = 1;
2252			} else if (strncasecmp(cp, "sha256:", 7) == 0) {
2253				cp += 7;
2254				cp = cp + strspn(cp, " \t");
2255				was_sha256 = 1;
2256				/*
2257				 * Just try to process the line as a key.
2258				 * Parsing will fail if it isn't.
2259				 */
2260			}
2261			if ((key = sshkey_new(KEY_UNSPEC)) == NULL)
2262				fatal("sshkey_new");
2263			if ((r = sshkey_read(key, &cp)) != 0)
2264				fatal("%s:%lu: invalid key: %s",
2265				    path, lnum, ssh_err(r));
2266			if (was_explicit_key)
2267				r = ssh_krl_revoke_key_explicit(krl, key);
2268			else if (was_sha1) {
2269				if (sshkey_fingerprint_raw(key,
2270				    SSH_DIGEST_SHA1, &blob, &blen) != 0) {
2271					fatal("%s:%lu: fingerprint failed",
2272					    file, lnum);
2273				}
2274				r = ssh_krl_revoke_key_sha1(krl, blob, blen);
2275			} else if (was_sha256) {
2276				if (sshkey_fingerprint_raw(key,
2277				    SSH_DIGEST_SHA256, &blob, &blen) != 0) {
2278					fatal("%s:%lu: fingerprint failed",
2279					    file, lnum);
2280				}
2281				r = ssh_krl_revoke_key_sha256(krl, blob, blen);
2282			} else
2283				r = ssh_krl_revoke_key(krl, key);
2284			if (r != 0)
2285				fatal("%s: revoke key failed: %s",
2286				    __func__, ssh_err(r));
2287			freezero(blob, blen);
2288			blob = NULL;
2289			blen = 0;
2290			sshkey_free(key);
2291		}
2292	}
2293	if (strcmp(path, "-") != 0)
2294		fclose(krl_spec);
2295	free(line);
2296	free(path);
2297}
2298
2299static void
2300do_gen_krl(struct passwd *pw, int updating, int argc, char **argv)
2301{
2302	struct ssh_krl *krl;
2303	struct stat sb;
2304	struct sshkey *ca = NULL;
2305	int fd, i, r, wild_ca = 0;
2306	char *tmp;
2307	struct sshbuf *kbuf;
2308
2309	if (*identity_file == '\0')
2310		fatal("KRL generation requires an output file");
2311	if (stat(identity_file, &sb) == -1) {
2312		if (errno != ENOENT)
2313			fatal("Cannot access KRL \"%s\": %s",
2314			    identity_file, strerror(errno));
2315		if (updating)
2316			fatal("KRL \"%s\" does not exist", identity_file);
2317	}
2318	if (ca_key_path != NULL) {
2319		if (strcasecmp(ca_key_path, "none") == 0)
2320			wild_ca = 1;
2321		else {
2322			tmp = tilde_expand_filename(ca_key_path, pw->pw_uid);
2323			if ((r = sshkey_load_public(tmp, &ca, NULL)) != 0)
2324				fatal("Cannot load CA public key %s: %s",
2325				    tmp, ssh_err(r));
2326			free(tmp);
2327		}
2328	}
2329
2330	if (updating)
2331		load_krl(identity_file, &krl);
2332	else if ((krl = ssh_krl_init()) == NULL)
2333		fatal("couldn't create KRL");
2334
2335	if (cert_serial != 0)
2336		ssh_krl_set_version(krl, cert_serial);
2337	if (identity_comment != NULL)
2338		ssh_krl_set_comment(krl, identity_comment);
2339
2340	for (i = 0; i < argc; i++)
2341		update_krl_from_file(pw, argv[i], wild_ca, ca, krl);
2342
2343	if ((kbuf = sshbuf_new()) == NULL)
2344		fatal("sshbuf_new failed");
2345	if (ssh_krl_to_blob(krl, kbuf, NULL, 0) != 0)
2346		fatal("Couldn't generate KRL");
2347	if ((fd = open(identity_file, O_WRONLY|O_CREAT|O_TRUNC, 0644)) == -1)
2348		fatal("open %s: %s", identity_file, strerror(errno));
2349	if (atomicio(vwrite, fd, sshbuf_mutable_ptr(kbuf), sshbuf_len(kbuf)) !=
2350	    sshbuf_len(kbuf))
2351		fatal("write %s: %s", identity_file, strerror(errno));
2352	close(fd);
2353	sshbuf_free(kbuf);
2354	ssh_krl_free(krl);
2355	sshkey_free(ca);
2356}
2357
2358static void
2359do_check_krl(struct passwd *pw, int argc, char **argv)
2360{
2361	int i, r, ret = 0;
2362	char *comment;
2363	struct ssh_krl *krl;
2364	struct sshkey *k;
2365
2366	if (*identity_file == '\0')
2367		fatal("KRL checking requires an input file");
2368	load_krl(identity_file, &krl);
2369	for (i = 0; i < argc; i++) {
2370		if ((r = sshkey_load_public(argv[i], &k, &comment)) != 0)
2371			fatal("Cannot load public key %s: %s",
2372			    argv[i], ssh_err(r));
2373		r = ssh_krl_check_key(krl, k);
2374		printf("%s%s%s%s: %s\n", argv[i],
2375		    *comment ? " (" : "", comment, *comment ? ")" : "",
2376		    r == 0 ? "ok" : "REVOKED");
2377		if (r != 0)
2378			ret = 1;
2379		sshkey_free(k);
2380		free(comment);
2381	}
2382	ssh_krl_free(krl);
2383	exit(ret);
2384}
2385
2386static void
2387usage(void)
2388{
2389	fprintf(stderr,
2390	    "usage: ssh-keygen [-q] [-b bits] [-t dsa | ecdsa | ed25519 | rsa]\n"
2391	    "                  [-N new_passphrase] [-C comment] [-f output_keyfile]\n"
2392	    "       ssh-keygen -p [-P old_passphrase] [-N new_passphrase] [-f keyfile]\n"
2393	    "       ssh-keygen -i [-m key_format] [-f input_keyfile]\n"
2394	    "       ssh-keygen -e [-m key_format] [-f input_keyfile]\n"
2395	    "       ssh-keygen -y [-f input_keyfile]\n"
2396	    "       ssh-keygen -c [-P passphrase] [-C comment] [-f keyfile]\n"
2397	    "       ssh-keygen -l [-v] [-E fingerprint_hash] [-f input_keyfile]\n"
2398	    "       ssh-keygen -B [-f input_keyfile]\n");
2399#ifdef ENABLE_PKCS11
2400	fprintf(stderr,
2401	    "       ssh-keygen -D pkcs11\n");
2402#endif
2403	fprintf(stderr,
2404	    "       ssh-keygen -F hostname [-f known_hosts_file] [-l]\n"
2405	    "       ssh-keygen -H [-f known_hosts_file]\n"
2406	    "       ssh-keygen -R hostname [-f known_hosts_file]\n"
2407	    "       ssh-keygen -r hostname [-f input_keyfile] [-g]\n"
2408#ifdef WITH_OPENSSL
2409	    "       ssh-keygen -G output_file [-v] [-b bits] [-M memory] [-S start_point]\n"
2410	    "       ssh-keygen -T output_file -f input_file [-v] [-a rounds] [-J num_lines]\n"
2411	    "                  [-j start_line] [-K checkpt] [-W generator]\n"
2412#endif
2413	    "       ssh-keygen -s ca_key -I certificate_identity [-h] [-U]\n"
2414	    "                  [-D pkcs11_provider] [-n principals] [-O option]\n"
2415	    "                  [-V validity_interval] [-z serial_number] file ...\n"
2416	    "       ssh-keygen -L [-f input_keyfile]\n"
2417	    "       ssh-keygen -A\n"
2418	    "       ssh-keygen -k -f krl_file [-u] [-s ca_public] [-z version_number]\n"
2419	    "                  file ...\n"
2420	    "       ssh-keygen -Q -f krl_file file ...\n");
2421	exit(1);
2422}
2423
2424/*
2425 * Main program for key management.
2426 */
2427int
2428main(int argc, char **argv)
2429{
2430	char dotsshdir[PATH_MAX], comment[1024], *passphrase1, *passphrase2;
2431	char *rr_hostname = NULL, *ep, *fp, *ra;
2432	struct sshkey *private, *public;
2433	struct passwd *pw;
2434	struct stat st;
2435	int r, opt, type, fd;
2436	int gen_all_hostkeys = 0, gen_krl = 0, update_krl = 0, check_krl = 0;
2437	FILE *f;
2438	const char *errstr;
2439#ifdef WITH_OPENSSL
2440	/* Moduli generation/screening */
2441	char out_file[PATH_MAX], *checkpoint = NULL;
2442	u_int32_t memory = 0, generator_wanted = 0;
2443	int do_gen_candidates = 0, do_screen_candidates = 0;
2444	unsigned long start_lineno = 0, lines_to_process = 0;
2445	BIGNUM *start = NULL;
2446#endif
2447
2448	extern int optind;
2449	extern char *optarg;
2450
2451	ssh_malloc_init();	/* must be called before any mallocs */
2452	/* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
2453	sanitise_stdfd();
2454
2455	__progname = ssh_get_progname(argv[0]);
2456
2457#ifdef WITH_OPENSSL
2458	OpenSSL_add_all_algorithms();
2459#endif
2460	log_init(argv[0], SYSLOG_LEVEL_INFO, SYSLOG_FACILITY_USER, 1);
2461
2462	seed_rng();
2463
2464	msetlocale();
2465
2466	/* we need this for the home * directory.  */
2467	pw = getpwuid(getuid());
2468	if (!pw)
2469		fatal("No user exists for uid %lu", (u_long)getuid());
2470	if (gethostname(hostname, sizeof(hostname)) < 0)
2471		fatal("gethostname: %s", strerror(errno));
2472
2473	/* Remaining characters: Ydw */
2474	while ((opt = getopt(argc, argv, "ABHLQUXceghiklopquvxy"
2475	    "C:D:E:F:G:I:J:K:M:N:O:P:R:S:T:V:W:Z:"
2476	    "a:b:f:g:j:m:n:r:s:t:z:")) != -1) {
2477		switch (opt) {
2478		case 'A':
2479			gen_all_hostkeys = 1;
2480			break;
2481		case 'b':
2482			bits = (u_int32_t)strtonum(optarg, 10, 32768, &errstr);
2483			if (errstr)
2484				fatal("Bits has bad value %s (%s)",
2485					optarg, errstr);
2486			break;
2487		case 'E':
2488			fingerprint_hash = ssh_digest_alg_by_name(optarg);
2489			if (fingerprint_hash == -1)
2490				fatal("Invalid hash algorithm \"%s\"", optarg);
2491			break;
2492		case 'F':
2493			find_host = 1;
2494			rr_hostname = optarg;
2495			break;
2496		case 'H':
2497			hash_hosts = 1;
2498			break;
2499		case 'I':
2500			cert_key_id = optarg;
2501			break;
2502		case 'R':
2503			delete_host = 1;
2504			rr_hostname = optarg;
2505			break;
2506		case 'L':
2507			show_cert = 1;
2508			break;
2509		case 'l':
2510			print_fingerprint = 1;
2511			break;
2512		case 'B':
2513			print_bubblebabble = 1;
2514			break;
2515		case 'm':
2516			if (strcasecmp(optarg, "RFC4716") == 0 ||
2517			    strcasecmp(optarg, "ssh2") == 0) {
2518				convert_format = FMT_RFC4716;
2519				break;
2520			}
2521			if (strcasecmp(optarg, "PKCS8") == 0) {
2522				convert_format = FMT_PKCS8;
2523				break;
2524			}
2525			if (strcasecmp(optarg, "PEM") == 0) {
2526				convert_format = FMT_PEM;
2527				use_new_format = 0;
2528				break;
2529			}
2530			fatal("Unsupported conversion format \"%s\"", optarg);
2531		case 'n':
2532			cert_principals = optarg;
2533			break;
2534		case 'o':
2535			/* no-op; new format is already the default */
2536			break;
2537		case 'p':
2538			change_passphrase = 1;
2539			break;
2540		case 'c':
2541			change_comment = 1;
2542			break;
2543		case 'f':
2544			if (strlcpy(identity_file, optarg,
2545			    sizeof(identity_file)) >= sizeof(identity_file))
2546				fatal("Identity filename too long");
2547			have_identity = 1;
2548			break;
2549		case 'g':
2550			print_generic = 1;
2551			break;
2552		case 'P':
2553			identity_passphrase = optarg;
2554			break;
2555		case 'N':
2556			identity_new_passphrase = optarg;
2557			break;
2558		case 'Q':
2559			check_krl = 1;
2560			break;
2561		case 'O':
2562			add_cert_option(optarg);
2563			break;
2564		case 'Z':
2565			new_format_cipher = optarg;
2566			break;
2567		case 'C':
2568			identity_comment = optarg;
2569			break;
2570		case 'q':
2571			quiet = 1;
2572			break;
2573		case 'e':
2574		case 'x':
2575			/* export key */
2576			convert_to = 1;
2577			break;
2578		case 'h':
2579			cert_key_type = SSH2_CERT_TYPE_HOST;
2580			certflags_flags = 0;
2581			break;
2582		case 'k':
2583			gen_krl = 1;
2584			break;
2585		case 'i':
2586		case 'X':
2587			/* import key */
2588			convert_from = 1;
2589			break;
2590		case 'y':
2591			print_public = 1;
2592			break;
2593		case 's':
2594			ca_key_path = optarg;
2595			break;
2596		case 't':
2597			key_type_name = optarg;
2598			break;
2599		case 'D':
2600			pkcs11provider = optarg;
2601			break;
2602		case 'U':
2603			prefer_agent = 1;
2604			break;
2605		case 'u':
2606			update_krl = 1;
2607			break;
2608		case 'v':
2609			if (log_level == SYSLOG_LEVEL_INFO)
2610				log_level = SYSLOG_LEVEL_DEBUG1;
2611			else {
2612				if (log_level >= SYSLOG_LEVEL_DEBUG1 &&
2613				    log_level < SYSLOG_LEVEL_DEBUG3)
2614					log_level++;
2615			}
2616			break;
2617		case 'r':
2618			rr_hostname = optarg;
2619			break;
2620		case 'a':
2621			rounds = (int)strtonum(optarg, 1, INT_MAX, &errstr);
2622			if (errstr)
2623				fatal("Invalid number: %s (%s)",
2624					optarg, errstr);
2625			break;
2626		case 'V':
2627			parse_cert_times(optarg);
2628			break;
2629		case 'z':
2630			errno = 0;
2631			cert_serial = strtoull(optarg, &ep, 10);
2632			if (*optarg < '0' || *optarg > '9' || *ep != '\0' ||
2633			    (errno == ERANGE && cert_serial == ULLONG_MAX))
2634				fatal("Invalid serial number \"%s\"", optarg);
2635			break;
2636#ifdef WITH_OPENSSL
2637		/* Moduli generation/screening */
2638		case 'G':
2639			do_gen_candidates = 1;
2640			if (strlcpy(out_file, optarg, sizeof(out_file)) >=
2641			    sizeof(out_file))
2642				fatal("Output filename too long");
2643			break;
2644		case 'J':
2645			lines_to_process = strtoul(optarg, NULL, 10);
2646			break;
2647		case 'j':
2648			start_lineno = strtoul(optarg, NULL, 10);
2649			break;
2650		case 'K':
2651			if (strlen(optarg) >= PATH_MAX)
2652				fatal("Checkpoint filename too long");
2653			checkpoint = xstrdup(optarg);
2654			break;
2655		case 'M':
2656			memory = (u_int32_t)strtonum(optarg, 1, UINT_MAX,
2657			    &errstr);
2658			if (errstr)
2659				fatal("Memory limit is %s: %s", errstr, optarg);
2660			break;
2661		case 'S':
2662			/* XXX - also compare length against bits */
2663			if (BN_hex2bn(&start, optarg) == 0)
2664				fatal("Invalid start point.");
2665			break;
2666		case 'T':
2667			do_screen_candidates = 1;
2668			if (strlcpy(out_file, optarg, sizeof(out_file)) >=
2669			    sizeof(out_file))
2670				fatal("Output filename too long");
2671			break;
2672		case 'W':
2673			generator_wanted = (u_int32_t)strtonum(optarg, 1,
2674			    UINT_MAX, &errstr);
2675			if (errstr != NULL)
2676				fatal("Desired generator invalid: %s (%s)",
2677				    optarg, errstr);
2678			break;
2679#endif /* WITH_OPENSSL */
2680		case '?':
2681		default:
2682			usage();
2683		}
2684	}
2685
2686	/* reinit */
2687	log_init(argv[0], log_level, SYSLOG_FACILITY_USER, 1);
2688
2689	argv += optind;
2690	argc -= optind;
2691
2692	if (ca_key_path != NULL) {
2693		if (argc < 1 && !gen_krl) {
2694			error("Too few arguments.");
2695			usage();
2696		}
2697	} else if (argc > 0 && !gen_krl && !check_krl) {
2698		error("Too many arguments.");
2699		usage();
2700	}
2701	if (change_passphrase && change_comment) {
2702		error("Can only have one of -p and -c.");
2703		usage();
2704	}
2705	if (print_fingerprint && (delete_host || hash_hosts)) {
2706		error("Cannot use -l with -H or -R.");
2707		usage();
2708	}
2709	if (gen_krl) {
2710		do_gen_krl(pw, update_krl, argc, argv);
2711		return (0);
2712	}
2713	if (check_krl) {
2714		do_check_krl(pw, argc, argv);
2715		return (0);
2716	}
2717	if (ca_key_path != NULL) {
2718		if (cert_key_id == NULL)
2719			fatal("Must specify key id (-I) when certifying");
2720		do_ca_sign(pw, argc, argv);
2721	}
2722	if (show_cert)
2723		do_show_cert(pw);
2724	if (delete_host || hash_hosts || find_host)
2725		do_known_hosts(pw, rr_hostname);
2726	if (pkcs11provider != NULL)
2727		do_download(pw);
2728	if (print_fingerprint || print_bubblebabble)
2729		do_fingerprint(pw);
2730	if (change_passphrase)
2731		do_change_passphrase(pw);
2732	if (change_comment)
2733		do_change_comment(pw);
2734#ifdef WITH_OPENSSL
2735	if (convert_to)
2736		do_convert_to(pw);
2737	if (convert_from)
2738		do_convert_from(pw);
2739#endif
2740	if (print_public)
2741		do_print_public(pw);
2742	if (rr_hostname != NULL) {
2743		unsigned int n = 0;
2744
2745		if (have_identity) {
2746			n = do_print_resource_record(pw,
2747			    identity_file, rr_hostname);
2748			if (n == 0)
2749				fatal("%s: %s", identity_file, strerror(errno));
2750			exit(0);
2751		} else {
2752
2753			n += do_print_resource_record(pw,
2754			    _PATH_HOST_RSA_KEY_FILE, rr_hostname);
2755			n += do_print_resource_record(pw,
2756			    _PATH_HOST_DSA_KEY_FILE, rr_hostname);
2757			n += do_print_resource_record(pw,
2758			    _PATH_HOST_ECDSA_KEY_FILE, rr_hostname);
2759			n += do_print_resource_record(pw,
2760			    _PATH_HOST_ED25519_KEY_FILE, rr_hostname);
2761			n += do_print_resource_record(pw,
2762			    _PATH_HOST_XMSS_KEY_FILE, rr_hostname);
2763			if (n == 0)
2764				fatal("no keys found.");
2765			exit(0);
2766		}
2767	}
2768
2769#ifdef WITH_OPENSSL
2770	if (do_gen_candidates) {
2771		FILE *out = fopen(out_file, "w");
2772
2773		if (out == NULL) {
2774			error("Couldn't open modulus candidate file \"%s\": %s",
2775			    out_file, strerror(errno));
2776			return (1);
2777		}
2778		if (bits == 0)
2779			bits = DEFAULT_BITS;
2780		if (gen_candidates(out, memory, bits, start) != 0)
2781			fatal("modulus candidate generation failed");
2782
2783		return (0);
2784	}
2785
2786	if (do_screen_candidates) {
2787		FILE *in;
2788		FILE *out = fopen(out_file, "a");
2789
2790		if (have_identity && strcmp(identity_file, "-") != 0) {
2791			if ((in = fopen(identity_file, "r")) == NULL) {
2792				fatal("Couldn't open modulus candidate "
2793				    "file \"%s\": %s", identity_file,
2794				    strerror(errno));
2795			}
2796		} else
2797			in = stdin;
2798
2799		if (out == NULL) {
2800			fatal("Couldn't open moduli file \"%s\": %s",
2801			    out_file, strerror(errno));
2802		}
2803		if (prime_test(in, out, rounds == 0 ? 100 : rounds,
2804		    generator_wanted, checkpoint,
2805		    start_lineno, lines_to_process) != 0)
2806			fatal("modulus screening failed");
2807		return (0);
2808	}
2809#endif
2810
2811	if (gen_all_hostkeys) {
2812		do_gen_all_hostkeys(pw);
2813		return (0);
2814	}
2815
2816	if (key_type_name == NULL)
2817		key_type_name = DEFAULT_KEY_TYPE_NAME;
2818
2819	type = sshkey_type_from_name(key_type_name);
2820	type_bits_valid(type, key_type_name, &bits);
2821
2822	if (!quiet)
2823		printf("Generating public/private %s key pair.\n",
2824		    key_type_name);
2825	if ((r = sshkey_generate(type, bits, &private)) != 0)
2826		fatal("sshkey_generate failed");
2827	if ((r = sshkey_from_private(private, &public)) != 0)
2828		fatal("sshkey_from_private failed: %s\n", ssh_err(r));
2829
2830	if (!have_identity)
2831		ask_filename(pw, "Enter file in which to save the key");
2832
2833	/* Create ~/.ssh directory if it doesn't already exist. */
2834	snprintf(dotsshdir, sizeof dotsshdir, "%s/%s",
2835	    pw->pw_dir, _PATH_SSH_USER_DIR);
2836	if (strstr(identity_file, dotsshdir) != NULL) {
2837		if (stat(dotsshdir, &st) < 0) {
2838			if (errno != ENOENT) {
2839				error("Could not stat %s: %s", dotsshdir,
2840				    strerror(errno));
2841			} else if (mkdir(dotsshdir, 0700) < 0) {
2842				error("Could not create directory '%s': %s",
2843				    dotsshdir, strerror(errno));
2844			} else if (!quiet)
2845				printf("Created directory '%s'.\n", dotsshdir);
2846		}
2847	}
2848	/* If the file already exists, ask the user to confirm. */
2849	if (stat(identity_file, &st) >= 0) {
2850		char yesno[3];
2851		printf("%s already exists.\n", identity_file);
2852		printf("Overwrite (y/n)? ");
2853		fflush(stdout);
2854		if (fgets(yesno, sizeof(yesno), stdin) == NULL)
2855			exit(1);
2856		if (yesno[0] != 'y' && yesno[0] != 'Y')
2857			exit(1);
2858	}
2859	/* Ask for a passphrase (twice). */
2860	if (identity_passphrase)
2861		passphrase1 = xstrdup(identity_passphrase);
2862	else if (identity_new_passphrase)
2863		passphrase1 = xstrdup(identity_new_passphrase);
2864	else {
2865passphrase_again:
2866		passphrase1 =
2867			read_passphrase("Enter passphrase (empty for no "
2868			    "passphrase): ", RP_ALLOW_STDIN);
2869		passphrase2 = read_passphrase("Enter same passphrase again: ",
2870		    RP_ALLOW_STDIN);
2871		if (strcmp(passphrase1, passphrase2) != 0) {
2872			/*
2873			 * The passphrases do not match.  Clear them and
2874			 * retry.
2875			 */
2876			explicit_bzero(passphrase1, strlen(passphrase1));
2877			explicit_bzero(passphrase2, strlen(passphrase2));
2878			free(passphrase1);
2879			free(passphrase2);
2880			printf("Passphrases do not match.  Try again.\n");
2881			goto passphrase_again;
2882		}
2883		/* Clear the other copy of the passphrase. */
2884		explicit_bzero(passphrase2, strlen(passphrase2));
2885		free(passphrase2);
2886	}
2887
2888	if (identity_comment) {
2889		strlcpy(comment, identity_comment, sizeof(comment));
2890	} else {
2891		/* Create default comment field for the passphrase. */
2892		snprintf(comment, sizeof comment, "%s@%s", pw->pw_name, hostname);
2893	}
2894
2895	/* Save the key with the given passphrase and comment. */
2896	if ((r = sshkey_save_private(private, identity_file, passphrase1,
2897	    comment, use_new_format, new_format_cipher, rounds)) != 0) {
2898		error("Saving key \"%s\" failed: %s",
2899		    identity_file, ssh_err(r));
2900		explicit_bzero(passphrase1, strlen(passphrase1));
2901		free(passphrase1);
2902		exit(1);
2903	}
2904	/* Clear the passphrase. */
2905	explicit_bzero(passphrase1, strlen(passphrase1));
2906	free(passphrase1);
2907
2908	/* Clear the private key and the random number generator. */
2909	sshkey_free(private);
2910
2911	if (!quiet)
2912		printf("Your identification has been saved in %s.\n", identity_file);
2913
2914	strlcat(identity_file, ".pub", sizeof(identity_file));
2915	if ((fd = open(identity_file, O_WRONLY|O_CREAT|O_TRUNC, 0644)) == -1)
2916		fatal("Unable to save public key to %s: %s",
2917		    identity_file, strerror(errno));
2918	if ((f = fdopen(fd, "w")) == NULL)
2919		fatal("fdopen %s failed: %s", identity_file, strerror(errno));
2920	if ((r = sshkey_write(public, f)) != 0)
2921		error("write key failed: %s", ssh_err(r));
2922	fprintf(f, " %s\n", comment);
2923	if (ferror(f) || fclose(f) != 0)
2924		fatal("write public failed: %s", strerror(errno));
2925
2926	if (!quiet) {
2927		fp = sshkey_fingerprint(public, fingerprint_hash,
2928		    SSH_FP_DEFAULT);
2929		ra = sshkey_fingerprint(public, fingerprint_hash,
2930		    SSH_FP_RANDOMART);
2931		if (fp == NULL || ra == NULL)
2932			fatal("sshkey_fingerprint failed");
2933		printf("Your public key has been saved in %s.\n",
2934		    identity_file);
2935		printf("The key fingerprint is:\n");
2936		printf("%s %s\n", fp, comment);
2937		printf("The key's randomart image is:\n");
2938		printf("%s\n", ra);
2939		free(ra);
2940		free(fp);
2941	}
2942
2943	sshkey_free(public);
2944	exit(0);
2945}
2946