1/* $OpenBSD: sshkey.c,v 1.72 2018/10/11 00:52:46 djm Exp $ */
2/*
3 * Copyright (c) 2000, 2001 Markus Friedl.  All rights reserved.
4 * Copyright (c) 2008 Alexander von Gernler.  All rights reserved.
5 * Copyright (c) 2010,2011 Damien Miller.  All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28#include "includes.h"
29
30#include <sys/types.h>
31#include <netinet/in.h>
32
33#ifdef WITH_OPENSSL
34#include <openssl/evp.h>
35#include <openssl/err.h>
36#include <openssl/pem.h>
37#endif
38
39#include "crypto_api.h"
40
41#include <errno.h>
42#include <limits.h>
43#include <stdio.h>
44#include <string.h>
45#include <resolv.h>
46#ifdef HAVE_UTIL_H
47#include <util.h>
48#endif /* HAVE_UTIL_H */
49
50#include "ssh2.h"
51#include "ssherr.h"
52#include "misc.h"
53#include "sshbuf.h"
54#include "cipher.h"
55#include "digest.h"
56#define SSHKEY_INTERNAL
57#include "sshkey.h"
58#include "sshkey-xmss.h"
59#include "match.h"
60
61#include "xmss_fast.h"
62
63#include "openbsd-compat/openssl-compat.h"
64
65/* openssh private key file format */
66#define MARK_BEGIN		"-----BEGIN OPENSSH PRIVATE KEY-----\n"
67#define MARK_END		"-----END OPENSSH PRIVATE KEY-----\n"
68#define MARK_BEGIN_LEN		(sizeof(MARK_BEGIN) - 1)
69#define MARK_END_LEN		(sizeof(MARK_END) - 1)
70#define KDFNAME			"bcrypt"
71#define AUTH_MAGIC		"openssh-key-v1"
72#define SALT_LEN		16
73#define DEFAULT_CIPHERNAME	"aes256-ctr"
74#define	DEFAULT_ROUNDS		16
75
76/* Version identification string for SSH v1 identity files. */
77#define LEGACY_BEGIN		"SSH PRIVATE KEY FILE FORMAT 1.1\n"
78
79int	sshkey_private_serialize_opt(const struct sshkey *key,
80    struct sshbuf *buf, enum sshkey_serialize_rep);
81static int sshkey_from_blob_internal(struct sshbuf *buf,
82    struct sshkey **keyp, int allow_cert);
83static int get_sigtype(const u_char *sig, size_t siglen, char **sigtypep);
84
85/* Supported key types */
86struct keytype {
87	const char *name;
88	const char *shortname;
89	const char *sigalg;
90	int type;
91	int nid;
92	int cert;
93	int sigonly;
94};
95static const struct keytype keytypes[] = {
96	{ "ssh-ed25519", "ED25519", NULL, KEY_ED25519, 0, 0, 0 },
97	{ "ssh-ed25519-cert-v01@openssh.com", "ED25519-CERT", NULL,
98	    KEY_ED25519_CERT, 0, 1, 0 },
99#ifdef WITH_XMSS
100	{ "ssh-xmss@openssh.com", "XMSS", NULL, KEY_XMSS, 0, 0, 0 },
101	{ "ssh-xmss-cert-v01@openssh.com", "XMSS-CERT", NULL,
102	    KEY_XMSS_CERT, 0, 1, 0 },
103#endif /* WITH_XMSS */
104#ifdef WITH_OPENSSL
105	{ "ssh-rsa", "RSA", NULL, KEY_RSA, 0, 0, 0 },
106	{ "rsa-sha2-256", "RSA", NULL, KEY_RSA, 0, 0, 1 },
107	{ "rsa-sha2-512", "RSA", NULL, KEY_RSA, 0, 0, 1 },
108	{ "ssh-dss", "DSA", NULL, KEY_DSA, 0, 0, 0 },
109# ifdef OPENSSL_HAS_ECC
110	{ "ecdsa-sha2-nistp256", "ECDSA", NULL,
111	    KEY_ECDSA, NID_X9_62_prime256v1, 0, 0 },
112	{ "ecdsa-sha2-nistp384", "ECDSA", NULL,
113	    KEY_ECDSA, NID_secp384r1, 0, 0 },
114#  ifdef OPENSSL_HAS_NISTP521
115	{ "ecdsa-sha2-nistp521", "ECDSA", NULL,
116	    KEY_ECDSA, NID_secp521r1, 0, 0 },
117#  endif /* OPENSSL_HAS_NISTP521 */
118# endif /* OPENSSL_HAS_ECC */
119	{ "ssh-rsa-cert-v01@openssh.com", "RSA-CERT", NULL,
120	    KEY_RSA_CERT, 0, 1, 0 },
121	{ "rsa-sha2-256-cert-v01@openssh.com", "RSA-CERT",
122	    "rsa-sha2-256", KEY_RSA_CERT, 0, 1, 1 },
123	{ "rsa-sha2-512-cert-v01@openssh.com", "RSA-CERT",
124	    "rsa-sha2-512", KEY_RSA_CERT, 0, 1, 1 },
125	{ "ssh-dss-cert-v01@openssh.com", "DSA-CERT", NULL,
126	    KEY_DSA_CERT, 0, 1, 0 },
127# ifdef OPENSSL_HAS_ECC
128	{ "ecdsa-sha2-nistp256-cert-v01@openssh.com", "ECDSA-CERT", NULL,
129	    KEY_ECDSA_CERT, NID_X9_62_prime256v1, 1, 0 },
130	{ "ecdsa-sha2-nistp384-cert-v01@openssh.com", "ECDSA-CERT", NULL,
131	    KEY_ECDSA_CERT, NID_secp384r1, 1, 0 },
132#  ifdef OPENSSL_HAS_NISTP521
133	{ "ecdsa-sha2-nistp521-cert-v01@openssh.com", "ECDSA-CERT", NULL,
134	   KEY_ECDSA_CERT, NID_secp521r1, 1, 0 },
135#  endif /* OPENSSL_HAS_NISTP521 */
136# endif /* OPENSSL_HAS_ECC */
137#endif /* WITH_OPENSSL */
138	{ NULL, NULL, NULL, -1, -1, 0, 0 }
139};
140
141const char *
142sshkey_type(const struct sshkey *k)
143{
144	const struct keytype *kt;
145
146	for (kt = keytypes; kt->type != -1; kt++) {
147		if (kt->type == k->type)
148			return kt->shortname;
149	}
150	return "unknown";
151}
152
153static const char *
154sshkey_ssh_name_from_type_nid(int type, int nid)
155{
156	const struct keytype *kt;
157
158	for (kt = keytypes; kt->type != -1; kt++) {
159		if (kt->type == type && (kt->nid == 0 || kt->nid == nid))
160			return kt->name;
161	}
162	return "ssh-unknown";
163}
164
165int
166sshkey_type_is_cert(int type)
167{
168	const struct keytype *kt;
169
170	for (kt = keytypes; kt->type != -1; kt++) {
171		if (kt->type == type)
172			return kt->cert;
173	}
174	return 0;
175}
176
177const char *
178sshkey_ssh_name(const struct sshkey *k)
179{
180	return sshkey_ssh_name_from_type_nid(k->type, k->ecdsa_nid);
181}
182
183const char *
184sshkey_ssh_name_plain(const struct sshkey *k)
185{
186	return sshkey_ssh_name_from_type_nid(sshkey_type_plain(k->type),
187	    k->ecdsa_nid);
188}
189
190int
191sshkey_type_from_name(const char *name)
192{
193	const struct keytype *kt;
194
195	for (kt = keytypes; kt->type != -1; kt++) {
196		/* Only allow shortname matches for plain key types */
197		if ((kt->name != NULL && strcmp(name, kt->name) == 0) ||
198		    (!kt->cert && strcasecmp(kt->shortname, name) == 0))
199			return kt->type;
200	}
201	return KEY_UNSPEC;
202}
203
204int
205sshkey_ecdsa_nid_from_name(const char *name)
206{
207	const struct keytype *kt;
208
209	for (kt = keytypes; kt->type != -1; kt++) {
210		if (kt->type != KEY_ECDSA && kt->type != KEY_ECDSA_CERT)
211			continue;
212		if (kt->name != NULL && strcmp(name, kt->name) == 0)
213			return kt->nid;
214	}
215	return -1;
216}
217
218char *
219sshkey_alg_list(int certs_only, int plain_only, int include_sigonly, char sep)
220{
221	char *tmp, *ret = NULL;
222	size_t nlen, rlen = 0;
223	const struct keytype *kt;
224
225	for (kt = keytypes; kt->type != -1; kt++) {
226		if (kt->name == NULL)
227			continue;
228		if (!include_sigonly && kt->sigonly)
229			continue;
230		if ((certs_only && !kt->cert) || (plain_only && kt->cert))
231			continue;
232		if (ret != NULL)
233			ret[rlen++] = sep;
234		nlen = strlen(kt->name);
235		if ((tmp = realloc(ret, rlen + nlen + 2)) == NULL) {
236			free(ret);
237			return NULL;
238		}
239		ret = tmp;
240		memcpy(ret + rlen, kt->name, nlen + 1);
241		rlen += nlen;
242	}
243	return ret;
244}
245
246int
247sshkey_names_valid2(const char *names, int allow_wildcard)
248{
249	char *s, *cp, *p;
250	const struct keytype *kt;
251	int type;
252
253	if (names == NULL || strcmp(names, "") == 0)
254		return 0;
255	if ((s = cp = strdup(names)) == NULL)
256		return 0;
257	for ((p = strsep(&cp, ",")); p && *p != '\0';
258	    (p = strsep(&cp, ","))) {
259		type = sshkey_type_from_name(p);
260		if (type == KEY_UNSPEC) {
261			if (allow_wildcard) {
262				/*
263				 * Try matching key types against the string.
264				 * If any has a positive or negative match then
265				 * the component is accepted.
266				 */
267				for (kt = keytypes; kt->type != -1; kt++) {
268					if (match_pattern_list(kt->name,
269					    p, 0) != 0)
270						break;
271				}
272				if (kt->type != -1)
273					continue;
274			}
275			free(s);
276			return 0;
277		}
278	}
279	free(s);
280	return 1;
281}
282
283u_int
284sshkey_size(const struct sshkey *k)
285{
286#ifdef WITH_OPENSSL
287	const BIGNUM *rsa_n, *dsa_p;
288#endif /* WITH_OPENSSL */
289
290	switch (k->type) {
291#ifdef WITH_OPENSSL
292	case KEY_RSA:
293	case KEY_RSA_CERT:
294		if (k->rsa == NULL)
295			return 0;
296		RSA_get0_key(k->rsa, &rsa_n, NULL, NULL);
297		return BN_num_bits(rsa_n);
298	case KEY_DSA:
299	case KEY_DSA_CERT:
300		if (k->dsa == NULL)
301			return 0;
302		DSA_get0_pqg(k->dsa, &dsa_p, NULL, NULL);
303		return BN_num_bits(dsa_p);
304	case KEY_ECDSA:
305	case KEY_ECDSA_CERT:
306		return sshkey_curve_nid_to_bits(k->ecdsa_nid);
307#endif /* WITH_OPENSSL */
308	case KEY_ED25519:
309	case KEY_ED25519_CERT:
310	case KEY_XMSS:
311	case KEY_XMSS_CERT:
312		return 256;	/* XXX */
313	}
314	return 0;
315}
316
317static int
318sshkey_type_is_valid_ca(int type)
319{
320	switch (type) {
321	case KEY_RSA:
322	case KEY_DSA:
323	case KEY_ECDSA:
324	case KEY_ED25519:
325	case KEY_XMSS:
326		return 1;
327	default:
328		return 0;
329	}
330}
331
332int
333sshkey_is_cert(const struct sshkey *k)
334{
335	if (k == NULL)
336		return 0;
337	return sshkey_type_is_cert(k->type);
338}
339
340/* Return the cert-less equivalent to a certified key type */
341int
342sshkey_type_plain(int type)
343{
344	switch (type) {
345	case KEY_RSA_CERT:
346		return KEY_RSA;
347	case KEY_DSA_CERT:
348		return KEY_DSA;
349	case KEY_ECDSA_CERT:
350		return KEY_ECDSA;
351	case KEY_ED25519_CERT:
352		return KEY_ED25519;
353	case KEY_XMSS_CERT:
354		return KEY_XMSS;
355	default:
356		return type;
357	}
358}
359
360#ifdef WITH_OPENSSL
361/* XXX: these are really begging for a table-driven approach */
362int
363sshkey_curve_name_to_nid(const char *name)
364{
365	if (strcmp(name, "nistp256") == 0)
366		return NID_X9_62_prime256v1;
367	else if (strcmp(name, "nistp384") == 0)
368		return NID_secp384r1;
369# ifdef OPENSSL_HAS_NISTP521
370	else if (strcmp(name, "nistp521") == 0)
371		return NID_secp521r1;
372# endif /* OPENSSL_HAS_NISTP521 */
373	else
374		return -1;
375}
376
377u_int
378sshkey_curve_nid_to_bits(int nid)
379{
380	switch (nid) {
381	case NID_X9_62_prime256v1:
382		return 256;
383	case NID_secp384r1:
384		return 384;
385# ifdef OPENSSL_HAS_NISTP521
386	case NID_secp521r1:
387		return 521;
388# endif /* OPENSSL_HAS_NISTP521 */
389	default:
390		return 0;
391	}
392}
393
394int
395sshkey_ecdsa_bits_to_nid(int bits)
396{
397	switch (bits) {
398	case 256:
399		return NID_X9_62_prime256v1;
400	case 384:
401		return NID_secp384r1;
402# ifdef OPENSSL_HAS_NISTP521
403	case 521:
404		return NID_secp521r1;
405# endif /* OPENSSL_HAS_NISTP521 */
406	default:
407		return -1;
408	}
409}
410
411const char *
412sshkey_curve_nid_to_name(int nid)
413{
414	switch (nid) {
415	case NID_X9_62_prime256v1:
416		return "nistp256";
417	case NID_secp384r1:
418		return "nistp384";
419# ifdef OPENSSL_HAS_NISTP521
420	case NID_secp521r1:
421		return "nistp521";
422# endif /* OPENSSL_HAS_NISTP521 */
423	default:
424		return NULL;
425	}
426}
427
428int
429sshkey_ec_nid_to_hash_alg(int nid)
430{
431	int kbits = sshkey_curve_nid_to_bits(nid);
432
433	if (kbits <= 0)
434		return -1;
435
436	/* RFC5656 section 6.2.1 */
437	if (kbits <= 256)
438		return SSH_DIGEST_SHA256;
439	else if (kbits <= 384)
440		return SSH_DIGEST_SHA384;
441	else
442		return SSH_DIGEST_SHA512;
443}
444#endif /* WITH_OPENSSL */
445
446static void
447cert_free(struct sshkey_cert *cert)
448{
449	u_int i;
450
451	if (cert == NULL)
452		return;
453	sshbuf_free(cert->certblob);
454	sshbuf_free(cert->critical);
455	sshbuf_free(cert->extensions);
456	free(cert->key_id);
457	for (i = 0; i < cert->nprincipals; i++)
458		free(cert->principals[i]);
459	free(cert->principals);
460	sshkey_free(cert->signature_key);
461	free(cert->signature_type);
462	freezero(cert, sizeof(*cert));
463}
464
465static struct sshkey_cert *
466cert_new(void)
467{
468	struct sshkey_cert *cert;
469
470	if ((cert = calloc(1, sizeof(*cert))) == NULL)
471		return NULL;
472	if ((cert->certblob = sshbuf_new()) == NULL ||
473	    (cert->critical = sshbuf_new()) == NULL ||
474	    (cert->extensions = sshbuf_new()) == NULL) {
475		cert_free(cert);
476		return NULL;
477	}
478	cert->key_id = NULL;
479	cert->principals = NULL;
480	cert->signature_key = NULL;
481	cert->signature_type = NULL;
482	return cert;
483}
484
485struct sshkey *
486sshkey_new(int type)
487{
488	struct sshkey *k;
489#ifdef WITH_OPENSSL
490	RSA *rsa;
491	DSA *dsa;
492#endif /* WITH_OPENSSL */
493
494	if ((k = calloc(1, sizeof(*k))) == NULL)
495		return NULL;
496	k->type = type;
497	k->ecdsa = NULL;
498	k->ecdsa_nid = -1;
499	k->dsa = NULL;
500	k->rsa = NULL;
501	k->cert = NULL;
502	k->ed25519_sk = NULL;
503	k->ed25519_pk = NULL;
504	k->xmss_sk = NULL;
505	k->xmss_pk = NULL;
506	switch (k->type) {
507#ifdef WITH_OPENSSL
508	case KEY_RSA:
509	case KEY_RSA_CERT:
510		if ((rsa = RSA_new()) == NULL) {
511			free(k);
512			return NULL;
513		}
514		k->rsa = rsa;
515		break;
516	case KEY_DSA:
517	case KEY_DSA_CERT:
518		if ((dsa = DSA_new()) == NULL) {
519			free(k);
520			return NULL;
521		}
522		k->dsa = dsa;
523		break;
524	case KEY_ECDSA:
525	case KEY_ECDSA_CERT:
526		/* Cannot do anything until we know the group */
527		break;
528#endif /* WITH_OPENSSL */
529	case KEY_ED25519:
530	case KEY_ED25519_CERT:
531	case KEY_XMSS:
532	case KEY_XMSS_CERT:
533		/* no need to prealloc */
534		break;
535	case KEY_UNSPEC:
536		break;
537	default:
538		free(k);
539		return NULL;
540	}
541
542	if (sshkey_is_cert(k)) {
543		if ((k->cert = cert_new()) == NULL) {
544			sshkey_free(k);
545			return NULL;
546		}
547	}
548
549	return k;
550}
551
552void
553sshkey_free(struct sshkey *k)
554{
555	if (k == NULL)
556		return;
557	switch (k->type) {
558#ifdef WITH_OPENSSL
559	case KEY_RSA:
560	case KEY_RSA_CERT:
561		RSA_free(k->rsa);
562		k->rsa = NULL;
563		break;
564	case KEY_DSA:
565	case KEY_DSA_CERT:
566		DSA_free(k->dsa);
567		k->dsa = NULL;
568		break;
569# ifdef OPENSSL_HAS_ECC
570	case KEY_ECDSA:
571	case KEY_ECDSA_CERT:
572		EC_KEY_free(k->ecdsa);
573		k->ecdsa = NULL;
574		break;
575# endif /* OPENSSL_HAS_ECC */
576#endif /* WITH_OPENSSL */
577	case KEY_ED25519:
578	case KEY_ED25519_CERT:
579		freezero(k->ed25519_pk, ED25519_PK_SZ);
580		k->ed25519_pk = NULL;
581		freezero(k->ed25519_sk, ED25519_SK_SZ);
582		k->ed25519_sk = NULL;
583		break;
584#ifdef WITH_XMSS
585	case KEY_XMSS:
586	case KEY_XMSS_CERT:
587		freezero(k->xmss_pk, sshkey_xmss_pklen(k));
588		k->xmss_pk = NULL;
589		freezero(k->xmss_sk, sshkey_xmss_sklen(k));
590		k->xmss_sk = NULL;
591		sshkey_xmss_free_state(k);
592		free(k->xmss_name);
593		k->xmss_name = NULL;
594		free(k->xmss_filename);
595		k->xmss_filename = NULL;
596		break;
597#endif /* WITH_XMSS */
598	case KEY_UNSPEC:
599		break;
600	default:
601		break;
602	}
603	if (sshkey_is_cert(k))
604		cert_free(k->cert);
605	freezero(k, sizeof(*k));
606}
607
608static int
609cert_compare(struct sshkey_cert *a, struct sshkey_cert *b)
610{
611	if (a == NULL && b == NULL)
612		return 1;
613	if (a == NULL || b == NULL)
614		return 0;
615	if (sshbuf_len(a->certblob) != sshbuf_len(b->certblob))
616		return 0;
617	if (timingsafe_bcmp(sshbuf_ptr(a->certblob), sshbuf_ptr(b->certblob),
618	    sshbuf_len(a->certblob)) != 0)
619		return 0;
620	return 1;
621}
622
623/*
624 * Compare public portions of key only, allowing comparisons between
625 * certificates and plain keys too.
626 */
627int
628sshkey_equal_public(const struct sshkey *a, const struct sshkey *b)
629{
630#if defined(WITH_OPENSSL)
631	const BIGNUM *rsa_e_a, *rsa_n_a;
632	const BIGNUM *rsa_e_b, *rsa_n_b;
633	const BIGNUM *dsa_p_a, *dsa_q_a, *dsa_g_a, *dsa_pub_key_a;
634	const BIGNUM *dsa_p_b, *dsa_q_b, *dsa_g_b, *dsa_pub_key_b;
635# if defined(OPENSSL_HAS_ECC)
636	BN_CTX *bnctx;
637# endif /* OPENSSL_HAS_ECC */
638#endif /* WITH_OPENSSL */
639
640	if (a == NULL || b == NULL ||
641	    sshkey_type_plain(a->type) != sshkey_type_plain(b->type))
642		return 0;
643
644	switch (a->type) {
645#ifdef WITH_OPENSSL
646	case KEY_RSA_CERT:
647	case KEY_RSA:
648		if (a->rsa == NULL || b->rsa == NULL)
649			return 0;
650		RSA_get0_key(a->rsa, &rsa_n_a, &rsa_e_a, NULL);
651		RSA_get0_key(b->rsa, &rsa_n_b, &rsa_e_b, NULL);
652		return BN_cmp(rsa_e_a, rsa_e_b) == 0 &&
653		    BN_cmp(rsa_n_a, rsa_n_b) == 0;
654	case KEY_DSA_CERT:
655	case KEY_DSA:
656		if (a->dsa == NULL || b->dsa == NULL)
657			return 0;
658		DSA_get0_pqg(a->dsa, &dsa_p_a, &dsa_q_a, &dsa_g_a);
659		DSA_get0_pqg(b->dsa, &dsa_p_b, &dsa_q_b, &dsa_g_b);
660		DSA_get0_key(a->dsa, &dsa_pub_key_a, NULL);
661		DSA_get0_key(b->dsa, &dsa_pub_key_b, NULL);
662		return BN_cmp(dsa_p_a, dsa_p_b) == 0 &&
663		    BN_cmp(dsa_q_a, dsa_q_b) == 0 &&
664		    BN_cmp(dsa_g_a, dsa_g_b) == 0 &&
665		    BN_cmp(dsa_pub_key_a, dsa_pub_key_b) == 0;
666# ifdef OPENSSL_HAS_ECC
667	case KEY_ECDSA_CERT:
668	case KEY_ECDSA:
669		if (a->ecdsa == NULL || b->ecdsa == NULL ||
670		    EC_KEY_get0_public_key(a->ecdsa) == NULL ||
671		    EC_KEY_get0_public_key(b->ecdsa) == NULL)
672			return 0;
673		if ((bnctx = BN_CTX_new()) == NULL)
674			return 0;
675		if (EC_GROUP_cmp(EC_KEY_get0_group(a->ecdsa),
676		    EC_KEY_get0_group(b->ecdsa), bnctx) != 0 ||
677		    EC_POINT_cmp(EC_KEY_get0_group(a->ecdsa),
678		    EC_KEY_get0_public_key(a->ecdsa),
679		    EC_KEY_get0_public_key(b->ecdsa), bnctx) != 0) {
680			BN_CTX_free(bnctx);
681			return 0;
682		}
683		BN_CTX_free(bnctx);
684		return 1;
685# endif /* OPENSSL_HAS_ECC */
686#endif /* WITH_OPENSSL */
687	case KEY_ED25519:
688	case KEY_ED25519_CERT:
689		return a->ed25519_pk != NULL && b->ed25519_pk != NULL &&
690		    memcmp(a->ed25519_pk, b->ed25519_pk, ED25519_PK_SZ) == 0;
691#ifdef WITH_XMSS
692	case KEY_XMSS:
693	case KEY_XMSS_CERT:
694		return a->xmss_pk != NULL && b->xmss_pk != NULL &&
695		    sshkey_xmss_pklen(a) == sshkey_xmss_pklen(b) &&
696		    memcmp(a->xmss_pk, b->xmss_pk, sshkey_xmss_pklen(a)) == 0;
697#endif /* WITH_XMSS */
698	default:
699		return 0;
700	}
701	/* NOTREACHED */
702}
703
704int
705sshkey_equal(const struct sshkey *a, const struct sshkey *b)
706{
707	if (a == NULL || b == NULL || a->type != b->type)
708		return 0;
709	if (sshkey_is_cert(a)) {
710		if (!cert_compare(a->cert, b->cert))
711			return 0;
712	}
713	return sshkey_equal_public(a, b);
714}
715
716static int
717to_blob_buf(const struct sshkey *key, struct sshbuf *b, int force_plain,
718  enum sshkey_serialize_rep opts)
719{
720	int type, ret = SSH_ERR_INTERNAL_ERROR;
721	const char *typename;
722#ifdef WITH_OPENSSL
723	const BIGNUM *rsa_n, *rsa_e, *dsa_p, *dsa_q, *dsa_g, *dsa_pub_key;
724#endif /* WITH_OPENSSL */
725
726	if (key == NULL)
727		return SSH_ERR_INVALID_ARGUMENT;
728
729	if (sshkey_is_cert(key)) {
730		if (key->cert == NULL)
731			return SSH_ERR_EXPECTED_CERT;
732		if (sshbuf_len(key->cert->certblob) == 0)
733			return SSH_ERR_KEY_LACKS_CERTBLOB;
734	}
735	type = force_plain ? sshkey_type_plain(key->type) : key->type;
736	typename = sshkey_ssh_name_from_type_nid(type, key->ecdsa_nid);
737
738	switch (type) {
739#ifdef WITH_OPENSSL
740	case KEY_DSA_CERT:
741	case KEY_ECDSA_CERT:
742	case KEY_RSA_CERT:
743#endif /* WITH_OPENSSL */
744	case KEY_ED25519_CERT:
745#ifdef WITH_XMSS
746	case KEY_XMSS_CERT:
747#endif /* WITH_XMSS */
748		/* Use the existing blob */
749		/* XXX modified flag? */
750		if ((ret = sshbuf_putb(b, key->cert->certblob)) != 0)
751			return ret;
752		break;
753#ifdef WITH_OPENSSL
754	case KEY_DSA:
755		if (key->dsa == NULL)
756			return SSH_ERR_INVALID_ARGUMENT;
757		DSA_get0_pqg(key->dsa, &dsa_p, &dsa_q, &dsa_g);
758		DSA_get0_key(key->dsa, &dsa_pub_key, NULL);
759		if ((ret = sshbuf_put_cstring(b, typename)) != 0 ||
760		    (ret = sshbuf_put_bignum2(b, dsa_p)) != 0 ||
761		    (ret = sshbuf_put_bignum2(b, dsa_q)) != 0 ||
762		    (ret = sshbuf_put_bignum2(b, dsa_g)) != 0 ||
763		    (ret = sshbuf_put_bignum2(b, dsa_pub_key)) != 0)
764			return ret;
765		break;
766# ifdef OPENSSL_HAS_ECC
767	case KEY_ECDSA:
768		if (key->ecdsa == NULL)
769			return SSH_ERR_INVALID_ARGUMENT;
770		if ((ret = sshbuf_put_cstring(b, typename)) != 0 ||
771		    (ret = sshbuf_put_cstring(b,
772		    sshkey_curve_nid_to_name(key->ecdsa_nid))) != 0 ||
773		    (ret = sshbuf_put_eckey(b, key->ecdsa)) != 0)
774			return ret;
775		break;
776# endif
777	case KEY_RSA:
778		if (key->rsa == NULL)
779			return SSH_ERR_INVALID_ARGUMENT;
780		RSA_get0_key(key->rsa, &rsa_n, &rsa_e, NULL);
781		if ((ret = sshbuf_put_cstring(b, typename)) != 0 ||
782		    (ret = sshbuf_put_bignum2(b, rsa_e)) != 0 ||
783		    (ret = sshbuf_put_bignum2(b, rsa_n)) != 0)
784			return ret;
785		break;
786#endif /* WITH_OPENSSL */
787	case KEY_ED25519:
788		if (key->ed25519_pk == NULL)
789			return SSH_ERR_INVALID_ARGUMENT;
790		if ((ret = sshbuf_put_cstring(b, typename)) != 0 ||
791		    (ret = sshbuf_put_string(b,
792		    key->ed25519_pk, ED25519_PK_SZ)) != 0)
793			return ret;
794		break;
795#ifdef WITH_XMSS
796	case KEY_XMSS:
797		if (key->xmss_name == NULL || key->xmss_pk == NULL ||
798		    sshkey_xmss_pklen(key) == 0)
799			return SSH_ERR_INVALID_ARGUMENT;
800		if ((ret = sshbuf_put_cstring(b, typename)) != 0 ||
801		    (ret = sshbuf_put_cstring(b, key->xmss_name)) != 0 ||
802		    (ret = sshbuf_put_string(b,
803		    key->xmss_pk, sshkey_xmss_pklen(key))) != 0 ||
804		    (ret = sshkey_xmss_serialize_pk_info(key, b, opts)) != 0)
805			return ret;
806		break;
807#endif /* WITH_XMSS */
808	default:
809		return SSH_ERR_KEY_TYPE_UNKNOWN;
810	}
811	return 0;
812}
813
814int
815sshkey_putb(const struct sshkey *key, struct sshbuf *b)
816{
817	return to_blob_buf(key, b, 0, SSHKEY_SERIALIZE_DEFAULT);
818}
819
820int
821sshkey_puts_opts(const struct sshkey *key, struct sshbuf *b,
822    enum sshkey_serialize_rep opts)
823{
824	struct sshbuf *tmp;
825	int r;
826
827	if ((tmp = sshbuf_new()) == NULL)
828		return SSH_ERR_ALLOC_FAIL;
829	r = to_blob_buf(key, tmp, 0, opts);
830	if (r == 0)
831		r = sshbuf_put_stringb(b, tmp);
832	sshbuf_free(tmp);
833	return r;
834}
835
836int
837sshkey_puts(const struct sshkey *key, struct sshbuf *b)
838{
839	return sshkey_puts_opts(key, b, SSHKEY_SERIALIZE_DEFAULT);
840}
841
842int
843sshkey_putb_plain(const struct sshkey *key, struct sshbuf *b)
844{
845	return to_blob_buf(key, b, 1, SSHKEY_SERIALIZE_DEFAULT);
846}
847
848static int
849to_blob(const struct sshkey *key, u_char **blobp, size_t *lenp, int force_plain,
850    enum sshkey_serialize_rep opts)
851{
852	int ret = SSH_ERR_INTERNAL_ERROR;
853	size_t len;
854	struct sshbuf *b = NULL;
855
856	if (lenp != NULL)
857		*lenp = 0;
858	if (blobp != NULL)
859		*blobp = NULL;
860	if ((b = sshbuf_new()) == NULL)
861		return SSH_ERR_ALLOC_FAIL;
862	if ((ret = to_blob_buf(key, b, force_plain, opts)) != 0)
863		goto out;
864	len = sshbuf_len(b);
865	if (lenp != NULL)
866		*lenp = len;
867	if (blobp != NULL) {
868		if ((*blobp = malloc(len)) == NULL) {
869			ret = SSH_ERR_ALLOC_FAIL;
870			goto out;
871		}
872		memcpy(*blobp, sshbuf_ptr(b), len);
873	}
874	ret = 0;
875 out:
876	sshbuf_free(b);
877	return ret;
878}
879
880int
881sshkey_to_blob(const struct sshkey *key, u_char **blobp, size_t *lenp)
882{
883	return to_blob(key, blobp, lenp, 0, SSHKEY_SERIALIZE_DEFAULT);
884}
885
886int
887sshkey_plain_to_blob(const struct sshkey *key, u_char **blobp, size_t *lenp)
888{
889	return to_blob(key, blobp, lenp, 1, SSHKEY_SERIALIZE_DEFAULT);
890}
891
892int
893sshkey_fingerprint_raw(const struct sshkey *k, int dgst_alg,
894    u_char **retp, size_t *lenp)
895{
896	u_char *blob = NULL, *ret = NULL;
897	size_t blob_len = 0;
898	int r = SSH_ERR_INTERNAL_ERROR;
899
900	if (retp != NULL)
901		*retp = NULL;
902	if (lenp != NULL)
903		*lenp = 0;
904	if (ssh_digest_bytes(dgst_alg) == 0) {
905		r = SSH_ERR_INVALID_ARGUMENT;
906		goto out;
907	}
908	if ((r = to_blob(k, &blob, &blob_len, 1, SSHKEY_SERIALIZE_DEFAULT))
909	    != 0)
910		goto out;
911	if ((ret = calloc(1, SSH_DIGEST_MAX_LENGTH)) == NULL) {
912		r = SSH_ERR_ALLOC_FAIL;
913		goto out;
914	}
915	if ((r = ssh_digest_memory(dgst_alg, blob, blob_len,
916	    ret, SSH_DIGEST_MAX_LENGTH)) != 0)
917		goto out;
918	/* success */
919	if (retp != NULL) {
920		*retp = ret;
921		ret = NULL;
922	}
923	if (lenp != NULL)
924		*lenp = ssh_digest_bytes(dgst_alg);
925	r = 0;
926 out:
927	free(ret);
928	if (blob != NULL) {
929		explicit_bzero(blob, blob_len);
930		free(blob);
931	}
932	return r;
933}
934
935static char *
936fingerprint_b64(const char *alg, u_char *dgst_raw, size_t dgst_raw_len)
937{
938	char *ret;
939	size_t plen = strlen(alg) + 1;
940	size_t rlen = ((dgst_raw_len + 2) / 3) * 4 + plen + 1;
941	int r;
942
943	if (dgst_raw_len > 65536 || (ret = calloc(1, rlen)) == NULL)
944		return NULL;
945	strlcpy(ret, alg, rlen);
946	strlcat(ret, ":", rlen);
947	if (dgst_raw_len == 0)
948		return ret;
949	if ((r = b64_ntop(dgst_raw, dgst_raw_len,
950	    ret + plen, rlen - plen)) == -1) {
951		freezero(ret, rlen);
952		return NULL;
953	}
954	/* Trim padding characters from end */
955	ret[strcspn(ret, "=")] = '\0';
956	return ret;
957}
958
959static char *
960fingerprint_hex(const char *alg, u_char *dgst_raw, size_t dgst_raw_len)
961{
962	char *retval, hex[5];
963	size_t i, rlen = dgst_raw_len * 3 + strlen(alg) + 2;
964
965	if (dgst_raw_len > 65536 || (retval = calloc(1, rlen)) == NULL)
966		return NULL;
967	strlcpy(retval, alg, rlen);
968	strlcat(retval, ":", rlen);
969	for (i = 0; i < dgst_raw_len; i++) {
970		snprintf(hex, sizeof(hex), "%s%02x",
971		    i > 0 ? ":" : "", dgst_raw[i]);
972		strlcat(retval, hex, rlen);
973	}
974	return retval;
975}
976
977static char *
978fingerprint_bubblebabble(u_char *dgst_raw, size_t dgst_raw_len)
979{
980	char vowels[] = { 'a', 'e', 'i', 'o', 'u', 'y' };
981	char consonants[] = { 'b', 'c', 'd', 'f', 'g', 'h', 'k', 'l', 'm',
982	    'n', 'p', 'r', 's', 't', 'v', 'z', 'x' };
983	u_int i, j = 0, rounds, seed = 1;
984	char *retval;
985
986	rounds = (dgst_raw_len / 2) + 1;
987	if ((retval = calloc(rounds, 6)) == NULL)
988		return NULL;
989	retval[j++] = 'x';
990	for (i = 0; i < rounds; i++) {
991		u_int idx0, idx1, idx2, idx3, idx4;
992		if ((i + 1 < rounds) || (dgst_raw_len % 2 != 0)) {
993			idx0 = (((((u_int)(dgst_raw[2 * i])) >> 6) & 3) +
994			    seed) % 6;
995			idx1 = (((u_int)(dgst_raw[2 * i])) >> 2) & 15;
996			idx2 = ((((u_int)(dgst_raw[2 * i])) & 3) +
997			    (seed / 6)) % 6;
998			retval[j++] = vowels[idx0];
999			retval[j++] = consonants[idx1];
1000			retval[j++] = vowels[idx2];
1001			if ((i + 1) < rounds) {
1002				idx3 = (((u_int)(dgst_raw[(2 * i) + 1])) >> 4) & 15;
1003				idx4 = (((u_int)(dgst_raw[(2 * i) + 1]))) & 15;
1004				retval[j++] = consonants[idx3];
1005				retval[j++] = '-';
1006				retval[j++] = consonants[idx4];
1007				seed = ((seed * 5) +
1008				    ((((u_int)(dgst_raw[2 * i])) * 7) +
1009				    ((u_int)(dgst_raw[(2 * i) + 1])))) % 36;
1010			}
1011		} else {
1012			idx0 = seed % 6;
1013			idx1 = 16;
1014			idx2 = seed / 6;
1015			retval[j++] = vowels[idx0];
1016			retval[j++] = consonants[idx1];
1017			retval[j++] = vowels[idx2];
1018		}
1019	}
1020	retval[j++] = 'x';
1021	retval[j++] = '\0';
1022	return retval;
1023}
1024
1025/*
1026 * Draw an ASCII-Art representing the fingerprint so human brain can
1027 * profit from its built-in pattern recognition ability.
1028 * This technique is called "random art" and can be found in some
1029 * scientific publications like this original paper:
1030 *
1031 * "Hash Visualization: a New Technique to improve Real-World Security",
1032 * Perrig A. and Song D., 1999, International Workshop on Cryptographic
1033 * Techniques and E-Commerce (CrypTEC '99)
1034 * sparrow.ece.cmu.edu/~adrian/projects/validation/validation.pdf
1035 *
1036 * The subject came up in a talk by Dan Kaminsky, too.
1037 *
1038 * If you see the picture is different, the key is different.
1039 * If the picture looks the same, you still know nothing.
1040 *
1041 * The algorithm used here is a worm crawling over a discrete plane,
1042 * leaving a trace (augmenting the field) everywhere it goes.
1043 * Movement is taken from dgst_raw 2bit-wise.  Bumping into walls
1044 * makes the respective movement vector be ignored for this turn.
1045 * Graphs are not unambiguous, because circles in graphs can be
1046 * walked in either direction.
1047 */
1048
1049/*
1050 * Field sizes for the random art.  Have to be odd, so the starting point
1051 * can be in the exact middle of the picture, and FLDBASE should be >=8 .
1052 * Else pictures would be too dense, and drawing the frame would
1053 * fail, too, because the key type would not fit in anymore.
1054 */
1055#define	FLDBASE		8
1056#define	FLDSIZE_Y	(FLDBASE + 1)
1057#define	FLDSIZE_X	(FLDBASE * 2 + 1)
1058static char *
1059fingerprint_randomart(const char *alg, u_char *dgst_raw, size_t dgst_raw_len,
1060    const struct sshkey *k)
1061{
1062	/*
1063	 * Chars to be used after each other every time the worm
1064	 * intersects with itself.  Matter of taste.
1065	 */
1066	char	*augmentation_string = " .o+=*BOX@%&#/^SE";
1067	char	*retval, *p, title[FLDSIZE_X], hash[FLDSIZE_X];
1068	u_char	 field[FLDSIZE_X][FLDSIZE_Y];
1069	size_t	 i, tlen, hlen;
1070	u_int	 b;
1071	int	 x, y, r;
1072	size_t	 len = strlen(augmentation_string) - 1;
1073
1074	if ((retval = calloc((FLDSIZE_X + 3), (FLDSIZE_Y + 2))) == NULL)
1075		return NULL;
1076
1077	/* initialize field */
1078	memset(field, 0, FLDSIZE_X * FLDSIZE_Y * sizeof(char));
1079	x = FLDSIZE_X / 2;
1080	y = FLDSIZE_Y / 2;
1081
1082	/* process raw key */
1083	for (i = 0; i < dgst_raw_len; i++) {
1084		int input;
1085		/* each byte conveys four 2-bit move commands */
1086		input = dgst_raw[i];
1087		for (b = 0; b < 4; b++) {
1088			/* evaluate 2 bit, rest is shifted later */
1089			x += (input & 0x1) ? 1 : -1;
1090			y += (input & 0x2) ? 1 : -1;
1091
1092			/* assure we are still in bounds */
1093			x = MAXIMUM(x, 0);
1094			y = MAXIMUM(y, 0);
1095			x = MINIMUM(x, FLDSIZE_X - 1);
1096			y = MINIMUM(y, FLDSIZE_Y - 1);
1097
1098			/* augment the field */
1099			if (field[x][y] < len - 2)
1100				field[x][y]++;
1101			input = input >> 2;
1102		}
1103	}
1104
1105	/* mark starting point and end point*/
1106	field[FLDSIZE_X / 2][FLDSIZE_Y / 2] = len - 1;
1107	field[x][y] = len;
1108
1109	/* assemble title */
1110	r = snprintf(title, sizeof(title), "[%s %u]",
1111		sshkey_type(k), sshkey_size(k));
1112	/* If [type size] won't fit, then try [type]; fits "[ED25519-CERT]" */
1113	if (r < 0 || r > (int)sizeof(title))
1114		r = snprintf(title, sizeof(title), "[%s]", sshkey_type(k));
1115	tlen = (r <= 0) ? 0 : strlen(title);
1116
1117	/* assemble hash ID. */
1118	r = snprintf(hash, sizeof(hash), "[%s]", alg);
1119	hlen = (r <= 0) ? 0 : strlen(hash);
1120
1121	/* output upper border */
1122	p = retval;
1123	*p++ = '+';
1124	for (i = 0; i < (FLDSIZE_X - tlen) / 2; i++)
1125		*p++ = '-';
1126	memcpy(p, title, tlen);
1127	p += tlen;
1128	for (i += tlen; i < FLDSIZE_X; i++)
1129		*p++ = '-';
1130	*p++ = '+';
1131	*p++ = '\n';
1132
1133	/* output content */
1134	for (y = 0; y < FLDSIZE_Y; y++) {
1135		*p++ = '|';
1136		for (x = 0; x < FLDSIZE_X; x++)
1137			*p++ = augmentation_string[MINIMUM(field[x][y], len)];
1138		*p++ = '|';
1139		*p++ = '\n';
1140	}
1141
1142	/* output lower border */
1143	*p++ = '+';
1144	for (i = 0; i < (FLDSIZE_X - hlen) / 2; i++)
1145		*p++ = '-';
1146	memcpy(p, hash, hlen);
1147	p += hlen;
1148	for (i += hlen; i < FLDSIZE_X; i++)
1149		*p++ = '-';
1150	*p++ = '+';
1151
1152	return retval;
1153}
1154
1155char *
1156sshkey_fingerprint(const struct sshkey *k, int dgst_alg,
1157    enum sshkey_fp_rep dgst_rep)
1158{
1159	char *retval = NULL;
1160	u_char *dgst_raw;
1161	size_t dgst_raw_len;
1162
1163	if (sshkey_fingerprint_raw(k, dgst_alg, &dgst_raw, &dgst_raw_len) != 0)
1164		return NULL;
1165	switch (dgst_rep) {
1166	case SSH_FP_DEFAULT:
1167		if (dgst_alg == SSH_DIGEST_MD5) {
1168			retval = fingerprint_hex(ssh_digest_alg_name(dgst_alg),
1169			    dgst_raw, dgst_raw_len);
1170		} else {
1171			retval = fingerprint_b64(ssh_digest_alg_name(dgst_alg),
1172			    dgst_raw, dgst_raw_len);
1173		}
1174		break;
1175	case SSH_FP_HEX:
1176		retval = fingerprint_hex(ssh_digest_alg_name(dgst_alg),
1177		    dgst_raw, dgst_raw_len);
1178		break;
1179	case SSH_FP_BASE64:
1180		retval = fingerprint_b64(ssh_digest_alg_name(dgst_alg),
1181		    dgst_raw, dgst_raw_len);
1182		break;
1183	case SSH_FP_BUBBLEBABBLE:
1184		retval = fingerprint_bubblebabble(dgst_raw, dgst_raw_len);
1185		break;
1186	case SSH_FP_RANDOMART:
1187		retval = fingerprint_randomart(ssh_digest_alg_name(dgst_alg),
1188		    dgst_raw, dgst_raw_len, k);
1189		break;
1190	default:
1191		explicit_bzero(dgst_raw, dgst_raw_len);
1192		free(dgst_raw);
1193		return NULL;
1194	}
1195	explicit_bzero(dgst_raw, dgst_raw_len);
1196	free(dgst_raw);
1197	return retval;
1198}
1199
1200static int
1201peek_type_nid(const char *s, size_t l, int *nid)
1202{
1203	const struct keytype *kt;
1204
1205	for (kt = keytypes; kt->type != -1; kt++) {
1206		if (kt->name == NULL || strlen(kt->name) != l)
1207			continue;
1208		if (memcmp(s, kt->name, l) == 0) {
1209			*nid = -1;
1210			if (kt->type == KEY_ECDSA || kt->type == KEY_ECDSA_CERT)
1211				*nid = kt->nid;
1212			return kt->type;
1213		}
1214	}
1215	return KEY_UNSPEC;
1216}
1217
1218/* XXX this can now be made const char * */
1219int
1220sshkey_read(struct sshkey *ret, char **cpp)
1221{
1222	struct sshkey *k;
1223	char *cp, *blobcopy;
1224	size_t space;
1225	int r, type, curve_nid = -1;
1226	struct sshbuf *blob;
1227
1228	if (ret == NULL)
1229		return SSH_ERR_INVALID_ARGUMENT;
1230
1231	switch (ret->type) {
1232	case KEY_UNSPEC:
1233	case KEY_RSA:
1234	case KEY_DSA:
1235	case KEY_ECDSA:
1236	case KEY_ED25519:
1237	case KEY_DSA_CERT:
1238	case KEY_ECDSA_CERT:
1239	case KEY_RSA_CERT:
1240	case KEY_ED25519_CERT:
1241#ifdef WITH_XMSS
1242	case KEY_XMSS:
1243	case KEY_XMSS_CERT:
1244#endif /* WITH_XMSS */
1245		break; /* ok */
1246	default:
1247		return SSH_ERR_INVALID_ARGUMENT;
1248	}
1249
1250	/* Decode type */
1251	cp = *cpp;
1252	space = strcspn(cp, " \t");
1253	if (space == strlen(cp))
1254		return SSH_ERR_INVALID_FORMAT;
1255	if ((type = peek_type_nid(cp, space, &curve_nid)) == KEY_UNSPEC)
1256		return SSH_ERR_INVALID_FORMAT;
1257
1258	/* skip whitespace */
1259	for (cp += space; *cp == ' ' || *cp == '\t'; cp++)
1260		;
1261	if (*cp == '\0')
1262		return SSH_ERR_INVALID_FORMAT;
1263	if (ret->type != KEY_UNSPEC && ret->type != type)
1264		return SSH_ERR_KEY_TYPE_MISMATCH;
1265	if ((blob = sshbuf_new()) == NULL)
1266		return SSH_ERR_ALLOC_FAIL;
1267
1268	/* find end of keyblob and decode */
1269	space = strcspn(cp, " \t");
1270	if ((blobcopy = strndup(cp, space)) == NULL) {
1271		sshbuf_free(blob);
1272		return SSH_ERR_ALLOC_FAIL;
1273	}
1274	if ((r = sshbuf_b64tod(blob, blobcopy)) != 0) {
1275		free(blobcopy);
1276		sshbuf_free(blob);
1277		return r;
1278	}
1279	free(blobcopy);
1280	if ((r = sshkey_fromb(blob, &k)) != 0) {
1281		sshbuf_free(blob);
1282		return r;
1283	}
1284	sshbuf_free(blob);
1285
1286	/* skip whitespace and leave cp at start of comment */
1287	for (cp += space; *cp == ' ' || *cp == '\t'; cp++)
1288		;
1289
1290	/* ensure type of blob matches type at start of line */
1291	if (k->type != type) {
1292		sshkey_free(k);
1293		return SSH_ERR_KEY_TYPE_MISMATCH;
1294	}
1295	if (sshkey_type_plain(type) == KEY_ECDSA && curve_nid != k->ecdsa_nid) {
1296		sshkey_free(k);
1297		return SSH_ERR_EC_CURVE_MISMATCH;
1298	}
1299
1300	/* Fill in ret from parsed key */
1301	ret->type = type;
1302	if (sshkey_is_cert(ret)) {
1303		if (!sshkey_is_cert(k)) {
1304			sshkey_free(k);
1305			return SSH_ERR_EXPECTED_CERT;
1306		}
1307		if (ret->cert != NULL)
1308			cert_free(ret->cert);
1309		ret->cert = k->cert;
1310		k->cert = NULL;
1311	}
1312	switch (sshkey_type_plain(ret->type)) {
1313#ifdef WITH_OPENSSL
1314	case KEY_RSA:
1315		RSA_free(ret->rsa);
1316		ret->rsa = k->rsa;
1317		k->rsa = NULL;
1318#ifdef DEBUG_PK
1319		RSA_print_fp(stderr, ret->rsa, 8);
1320#endif
1321		break;
1322	case KEY_DSA:
1323		DSA_free(ret->dsa);
1324		ret->dsa = k->dsa;
1325		k->dsa = NULL;
1326#ifdef DEBUG_PK
1327		DSA_print_fp(stderr, ret->dsa, 8);
1328#endif
1329		break;
1330# ifdef OPENSSL_HAS_ECC
1331	case KEY_ECDSA:
1332		EC_KEY_free(ret->ecdsa);
1333		ret->ecdsa = k->ecdsa;
1334		ret->ecdsa_nid = k->ecdsa_nid;
1335		k->ecdsa = NULL;
1336		k->ecdsa_nid = -1;
1337#ifdef DEBUG_PK
1338		sshkey_dump_ec_key(ret->ecdsa);
1339#endif
1340		break;
1341# endif /* OPENSSL_HAS_ECC */
1342#endif /* WITH_OPENSSL */
1343	case KEY_ED25519:
1344		freezero(ret->ed25519_pk, ED25519_PK_SZ);
1345		ret->ed25519_pk = k->ed25519_pk;
1346		k->ed25519_pk = NULL;
1347#ifdef DEBUG_PK
1348		/* XXX */
1349#endif
1350		break;
1351#ifdef WITH_XMSS
1352	case KEY_XMSS:
1353		free(ret->xmss_pk);
1354		ret->xmss_pk = k->xmss_pk;
1355		k->xmss_pk = NULL;
1356		free(ret->xmss_state);
1357		ret->xmss_state = k->xmss_state;
1358		k->xmss_state = NULL;
1359		free(ret->xmss_name);
1360		ret->xmss_name = k->xmss_name;
1361		k->xmss_name = NULL;
1362		free(ret->xmss_filename);
1363		ret->xmss_filename = k->xmss_filename;
1364		k->xmss_filename = NULL;
1365#ifdef DEBUG_PK
1366		/* XXX */
1367#endif
1368		break;
1369#endif /* WITH_XMSS */
1370	default:
1371		sshkey_free(k);
1372		return SSH_ERR_INTERNAL_ERROR;
1373	}
1374	sshkey_free(k);
1375
1376	/* success */
1377	*cpp = cp;
1378	return 0;
1379}
1380
1381
1382int
1383sshkey_to_base64(const struct sshkey *key, char **b64p)
1384{
1385	int r = SSH_ERR_INTERNAL_ERROR;
1386	struct sshbuf *b = NULL;
1387	char *uu = NULL;
1388
1389	if (b64p != NULL)
1390		*b64p = NULL;
1391	if ((b = sshbuf_new()) == NULL)
1392		return SSH_ERR_ALLOC_FAIL;
1393	if ((r = sshkey_putb(key, b)) != 0)
1394		goto out;
1395	if ((uu = sshbuf_dtob64(b)) == NULL) {
1396		r = SSH_ERR_ALLOC_FAIL;
1397		goto out;
1398	}
1399	/* Success */
1400	if (b64p != NULL) {
1401		*b64p = uu;
1402		uu = NULL;
1403	}
1404	r = 0;
1405 out:
1406	sshbuf_free(b);
1407	free(uu);
1408	return r;
1409}
1410
1411int
1412sshkey_format_text(const struct sshkey *key, struct sshbuf *b)
1413{
1414	int r = SSH_ERR_INTERNAL_ERROR;
1415	char *uu = NULL;
1416
1417	if ((r = sshkey_to_base64(key, &uu)) != 0)
1418		goto out;
1419	if ((r = sshbuf_putf(b, "%s %s",
1420	    sshkey_ssh_name(key), uu)) != 0)
1421		goto out;
1422	r = 0;
1423 out:
1424	free(uu);
1425	return r;
1426}
1427
1428int
1429sshkey_write(const struct sshkey *key, FILE *f)
1430{
1431	struct sshbuf *b = NULL;
1432	int r = SSH_ERR_INTERNAL_ERROR;
1433
1434	if ((b = sshbuf_new()) == NULL)
1435		return SSH_ERR_ALLOC_FAIL;
1436	if ((r = sshkey_format_text(key, b)) != 0)
1437		goto out;
1438	if (fwrite(sshbuf_ptr(b), sshbuf_len(b), 1, f) != 1) {
1439		if (feof(f))
1440			errno = EPIPE;
1441		r = SSH_ERR_SYSTEM_ERROR;
1442		goto out;
1443	}
1444	/* Success */
1445	r = 0;
1446 out:
1447	sshbuf_free(b);
1448	return r;
1449}
1450
1451const char *
1452sshkey_cert_type(const struct sshkey *k)
1453{
1454	switch (k->cert->type) {
1455	case SSH2_CERT_TYPE_USER:
1456		return "user";
1457	case SSH2_CERT_TYPE_HOST:
1458		return "host";
1459	default:
1460		return "unknown";
1461	}
1462}
1463
1464#ifdef WITH_OPENSSL
1465static int
1466rsa_generate_private_key(u_int bits, RSA **rsap)
1467{
1468	RSA *private = NULL;
1469	BIGNUM *f4 = NULL;
1470	int ret = SSH_ERR_INTERNAL_ERROR;
1471
1472	if (rsap == NULL)
1473		return SSH_ERR_INVALID_ARGUMENT;
1474	if (bits < SSH_RSA_MINIMUM_MODULUS_SIZE ||
1475	    bits > SSHBUF_MAX_BIGNUM * 8)
1476		return SSH_ERR_KEY_LENGTH;
1477	*rsap = NULL;
1478	if ((private = RSA_new()) == NULL || (f4 = BN_new()) == NULL) {
1479		ret = SSH_ERR_ALLOC_FAIL;
1480		goto out;
1481	}
1482	if (!BN_set_word(f4, RSA_F4) ||
1483	    !RSA_generate_key_ex(private, bits, f4, NULL)) {
1484		ret = SSH_ERR_LIBCRYPTO_ERROR;
1485		goto out;
1486	}
1487	*rsap = private;
1488	private = NULL;
1489	ret = 0;
1490 out:
1491	RSA_free(private);
1492	BN_free(f4);
1493	return ret;
1494}
1495
1496static int
1497dsa_generate_private_key(u_int bits, DSA **dsap)
1498{
1499	DSA *private;
1500	int ret = SSH_ERR_INTERNAL_ERROR;
1501
1502	if (dsap == NULL)
1503		return SSH_ERR_INVALID_ARGUMENT;
1504	if (bits != 1024)
1505		return SSH_ERR_KEY_LENGTH;
1506	if ((private = DSA_new()) == NULL) {
1507		ret = SSH_ERR_ALLOC_FAIL;
1508		goto out;
1509	}
1510	*dsap = NULL;
1511	if (!DSA_generate_parameters_ex(private, bits, NULL, 0, NULL,
1512	    NULL, NULL) || !DSA_generate_key(private)) {
1513		ret = SSH_ERR_LIBCRYPTO_ERROR;
1514		goto out;
1515	}
1516	*dsap = private;
1517	private = NULL;
1518	ret = 0;
1519 out:
1520	DSA_free(private);
1521	return ret;
1522}
1523
1524# ifdef OPENSSL_HAS_ECC
1525int
1526sshkey_ecdsa_key_to_nid(EC_KEY *k)
1527{
1528	EC_GROUP *eg;
1529	int nids[] = {
1530		NID_X9_62_prime256v1,
1531		NID_secp384r1,
1532#  ifdef OPENSSL_HAS_NISTP521
1533		NID_secp521r1,
1534#  endif /* OPENSSL_HAS_NISTP521 */
1535		-1
1536	};
1537	int nid;
1538	u_int i;
1539	BN_CTX *bnctx;
1540	const EC_GROUP *g = EC_KEY_get0_group(k);
1541
1542	/*
1543	 * The group may be stored in a ASN.1 encoded private key in one of two
1544	 * ways: as a "named group", which is reconstituted by ASN.1 object ID
1545	 * or explicit group parameters encoded into the key blob. Only the
1546	 * "named group" case sets the group NID for us, but we can figure
1547	 * it out for the other case by comparing against all the groups that
1548	 * are supported.
1549	 */
1550	if ((nid = EC_GROUP_get_curve_name(g)) > 0)
1551		return nid;
1552	if ((bnctx = BN_CTX_new()) == NULL)
1553		return -1;
1554	for (i = 0; nids[i] != -1; i++) {
1555		if ((eg = EC_GROUP_new_by_curve_name(nids[i])) == NULL) {
1556			BN_CTX_free(bnctx);
1557			return -1;
1558		}
1559		if (EC_GROUP_cmp(g, eg, bnctx) == 0)
1560			break;
1561		EC_GROUP_free(eg);
1562	}
1563	BN_CTX_free(bnctx);
1564	if (nids[i] != -1) {
1565		/* Use the group with the NID attached */
1566		EC_GROUP_set_asn1_flag(eg, OPENSSL_EC_NAMED_CURVE);
1567		if (EC_KEY_set_group(k, eg) != 1) {
1568			EC_GROUP_free(eg);
1569			return -1;
1570		}
1571	}
1572	return nids[i];
1573}
1574
1575static int
1576ecdsa_generate_private_key(u_int bits, int *nid, EC_KEY **ecdsap)
1577{
1578	EC_KEY *private;
1579	int ret = SSH_ERR_INTERNAL_ERROR;
1580
1581	if (nid == NULL || ecdsap == NULL)
1582		return SSH_ERR_INVALID_ARGUMENT;
1583	if ((*nid = sshkey_ecdsa_bits_to_nid(bits)) == -1)
1584		return SSH_ERR_KEY_LENGTH;
1585	*ecdsap = NULL;
1586	if ((private = EC_KEY_new_by_curve_name(*nid)) == NULL) {
1587		ret = SSH_ERR_ALLOC_FAIL;
1588		goto out;
1589	}
1590	if (EC_KEY_generate_key(private) != 1) {
1591		ret = SSH_ERR_LIBCRYPTO_ERROR;
1592		goto out;
1593	}
1594	EC_KEY_set_asn1_flag(private, OPENSSL_EC_NAMED_CURVE);
1595	*ecdsap = private;
1596	private = NULL;
1597	ret = 0;
1598 out:
1599	EC_KEY_free(private);
1600	return ret;
1601}
1602# endif /* OPENSSL_HAS_ECC */
1603#endif /* WITH_OPENSSL */
1604
1605int
1606sshkey_generate(int type, u_int bits, struct sshkey **keyp)
1607{
1608	struct sshkey *k;
1609	int ret = SSH_ERR_INTERNAL_ERROR;
1610
1611	if (keyp == NULL)
1612		return SSH_ERR_INVALID_ARGUMENT;
1613	*keyp = NULL;
1614	if ((k = sshkey_new(KEY_UNSPEC)) == NULL)
1615		return SSH_ERR_ALLOC_FAIL;
1616	switch (type) {
1617	case KEY_ED25519:
1618		if ((k->ed25519_pk = malloc(ED25519_PK_SZ)) == NULL ||
1619		    (k->ed25519_sk = malloc(ED25519_SK_SZ)) == NULL) {
1620			ret = SSH_ERR_ALLOC_FAIL;
1621			break;
1622		}
1623		crypto_sign_ed25519_keypair(k->ed25519_pk, k->ed25519_sk);
1624		ret = 0;
1625		break;
1626#ifdef WITH_XMSS
1627	case KEY_XMSS:
1628		ret = sshkey_xmss_generate_private_key(k, bits);
1629		break;
1630#endif /* WITH_XMSS */
1631#ifdef WITH_OPENSSL
1632	case KEY_DSA:
1633		ret = dsa_generate_private_key(bits, &k->dsa);
1634		break;
1635# ifdef OPENSSL_HAS_ECC
1636	case KEY_ECDSA:
1637		ret = ecdsa_generate_private_key(bits, &k->ecdsa_nid,
1638		    &k->ecdsa);
1639		break;
1640# endif /* OPENSSL_HAS_ECC */
1641	case KEY_RSA:
1642		ret = rsa_generate_private_key(bits, &k->rsa);
1643		break;
1644#endif /* WITH_OPENSSL */
1645	default:
1646		ret = SSH_ERR_INVALID_ARGUMENT;
1647	}
1648	if (ret == 0) {
1649		k->type = type;
1650		*keyp = k;
1651	} else
1652		sshkey_free(k);
1653	return ret;
1654}
1655
1656int
1657sshkey_cert_copy(const struct sshkey *from_key, struct sshkey *to_key)
1658{
1659	u_int i;
1660	const struct sshkey_cert *from;
1661	struct sshkey_cert *to;
1662	int r = SSH_ERR_INTERNAL_ERROR;
1663
1664	if (to_key == NULL || (from = from_key->cert) == NULL)
1665		return SSH_ERR_INVALID_ARGUMENT;
1666
1667	if ((to = cert_new()) == NULL)
1668		return SSH_ERR_ALLOC_FAIL;
1669
1670	if ((r = sshbuf_putb(to->certblob, from->certblob)) != 0 ||
1671	    (r = sshbuf_putb(to->critical, from->critical)) != 0 ||
1672	    (r = sshbuf_putb(to->extensions, from->extensions)) != 0)
1673		goto out;
1674
1675	to->serial = from->serial;
1676	to->type = from->type;
1677	if (from->key_id == NULL)
1678		to->key_id = NULL;
1679	else if ((to->key_id = strdup(from->key_id)) == NULL) {
1680		r = SSH_ERR_ALLOC_FAIL;
1681		goto out;
1682	}
1683	to->valid_after = from->valid_after;
1684	to->valid_before = from->valid_before;
1685	if (from->signature_key == NULL)
1686		to->signature_key = NULL;
1687	else if ((r = sshkey_from_private(from->signature_key,
1688	    &to->signature_key)) != 0)
1689		goto out;
1690	if (from->signature_type != NULL &&
1691	    (to->signature_type = strdup(from->signature_type)) == NULL) {
1692		r = SSH_ERR_ALLOC_FAIL;
1693		goto out;
1694	}
1695	if (from->nprincipals > SSHKEY_CERT_MAX_PRINCIPALS) {
1696		r = SSH_ERR_INVALID_ARGUMENT;
1697		goto out;
1698	}
1699	if (from->nprincipals > 0) {
1700		if ((to->principals = calloc(from->nprincipals,
1701		    sizeof(*to->principals))) == NULL) {
1702			r = SSH_ERR_ALLOC_FAIL;
1703			goto out;
1704		}
1705		for (i = 0; i < from->nprincipals; i++) {
1706			to->principals[i] = strdup(from->principals[i]);
1707			if (to->principals[i] == NULL) {
1708				to->nprincipals = i;
1709				r = SSH_ERR_ALLOC_FAIL;
1710				goto out;
1711			}
1712		}
1713	}
1714	to->nprincipals = from->nprincipals;
1715
1716	/* success */
1717	cert_free(to_key->cert);
1718	to_key->cert = to;
1719	to = NULL;
1720	r = 0;
1721 out:
1722	cert_free(to);
1723	return r;
1724}
1725
1726int
1727sshkey_from_private(const struct sshkey *k, struct sshkey **pkp)
1728{
1729	struct sshkey *n = NULL;
1730	int r = SSH_ERR_INTERNAL_ERROR;
1731#ifdef WITH_OPENSSL
1732	const BIGNUM *rsa_n, *rsa_e;
1733	BIGNUM *rsa_n_dup = NULL, *rsa_e_dup = NULL;
1734	const BIGNUM *dsa_p, *dsa_q, *dsa_g, *dsa_pub_key;
1735	BIGNUM *dsa_p_dup = NULL, *dsa_q_dup = NULL, *dsa_g_dup = NULL;
1736	BIGNUM *dsa_pub_key_dup = NULL;
1737#endif /* WITH_OPENSSL */
1738
1739	*pkp = NULL;
1740	switch (k->type) {
1741#ifdef WITH_OPENSSL
1742	case KEY_DSA:
1743	case KEY_DSA_CERT:
1744		if ((n = sshkey_new(k->type)) == NULL) {
1745			r = SSH_ERR_ALLOC_FAIL;
1746			goto out;
1747		}
1748
1749		DSA_get0_pqg(k->dsa, &dsa_p, &dsa_q, &dsa_g);
1750		DSA_get0_key(k->dsa, &dsa_pub_key, NULL);
1751		if ((dsa_p_dup = BN_dup(dsa_p)) == NULL ||
1752		    (dsa_q_dup = BN_dup(dsa_q)) == NULL ||
1753		    (dsa_g_dup = BN_dup(dsa_g)) == NULL ||
1754		    (dsa_pub_key_dup = BN_dup(dsa_pub_key)) == NULL) {
1755			r = SSH_ERR_ALLOC_FAIL;
1756			goto out;
1757		}
1758		if (!DSA_set0_pqg(n->dsa, dsa_p_dup, dsa_q_dup, dsa_g_dup)) {
1759			r = SSH_ERR_LIBCRYPTO_ERROR;
1760			goto out;
1761		}
1762		dsa_p_dup = dsa_q_dup = dsa_g_dup = NULL; /* transferred */
1763		if (!DSA_set0_key(n->dsa, dsa_pub_key_dup, NULL)) {
1764			r = SSH_ERR_LIBCRYPTO_ERROR;
1765			goto out;
1766		}
1767		dsa_pub_key_dup = NULL; /* transferred */
1768
1769		break;
1770# ifdef OPENSSL_HAS_ECC
1771	case KEY_ECDSA:
1772	case KEY_ECDSA_CERT:
1773		if ((n = sshkey_new(k->type)) == NULL) {
1774			r = SSH_ERR_ALLOC_FAIL;
1775			goto out;
1776		}
1777		n->ecdsa_nid = k->ecdsa_nid;
1778		n->ecdsa = EC_KEY_new_by_curve_name(k->ecdsa_nid);
1779		if (n->ecdsa == NULL) {
1780			r = SSH_ERR_ALLOC_FAIL;
1781			goto out;
1782		}
1783		if (EC_KEY_set_public_key(n->ecdsa,
1784		    EC_KEY_get0_public_key(k->ecdsa)) != 1) {
1785			r = SSH_ERR_LIBCRYPTO_ERROR;
1786			goto out;
1787		}
1788		break;
1789# endif /* OPENSSL_HAS_ECC */
1790	case KEY_RSA:
1791	case KEY_RSA_CERT:
1792		if ((n = sshkey_new(k->type)) == NULL) {
1793			r = SSH_ERR_ALLOC_FAIL;
1794			goto out;
1795		}
1796		RSA_get0_key(k->rsa, &rsa_n, &rsa_e, NULL);
1797		if ((rsa_n_dup = BN_dup(rsa_n)) == NULL ||
1798		    (rsa_e_dup = BN_dup(rsa_e)) == NULL) {
1799			r = SSH_ERR_ALLOC_FAIL;
1800			goto out;
1801		}
1802		if (!RSA_set0_key(n->rsa, rsa_n_dup, rsa_e_dup, NULL)) {
1803			r = SSH_ERR_LIBCRYPTO_ERROR;
1804			goto out;
1805		}
1806		rsa_n_dup = rsa_e_dup = NULL; /* transferred */
1807		break;
1808#endif /* WITH_OPENSSL */
1809	case KEY_ED25519:
1810	case KEY_ED25519_CERT:
1811		if ((n = sshkey_new(k->type)) == NULL) {
1812			r = SSH_ERR_ALLOC_FAIL;
1813			goto out;
1814		}
1815		if (k->ed25519_pk != NULL) {
1816			if ((n->ed25519_pk = malloc(ED25519_PK_SZ)) == NULL) {
1817				r = SSH_ERR_ALLOC_FAIL;
1818				goto out;
1819			}
1820			memcpy(n->ed25519_pk, k->ed25519_pk, ED25519_PK_SZ);
1821		}
1822		break;
1823#ifdef WITH_XMSS
1824	case KEY_XMSS:
1825	case KEY_XMSS_CERT:
1826		if ((n = sshkey_new(k->type)) == NULL) {
1827			r = SSH_ERR_ALLOC_FAIL;
1828			goto out;
1829		}
1830		if ((r = sshkey_xmss_init(n, k->xmss_name)) != 0)
1831			goto out;
1832		if (k->xmss_pk != NULL) {
1833			size_t pklen = sshkey_xmss_pklen(k);
1834			if (pklen == 0 || sshkey_xmss_pklen(n) != pklen) {
1835				r = SSH_ERR_INTERNAL_ERROR;
1836				goto out;
1837			}
1838			if ((n->xmss_pk = malloc(pklen)) == NULL) {
1839				r = SSH_ERR_ALLOC_FAIL;
1840				goto out;
1841			}
1842			memcpy(n->xmss_pk, k->xmss_pk, pklen);
1843		}
1844		break;
1845#endif /* WITH_XMSS */
1846	default:
1847		r = SSH_ERR_KEY_TYPE_UNKNOWN;
1848		goto out;
1849	}
1850	if (sshkey_is_cert(k) && (r = sshkey_cert_copy(k, n)) != 0)
1851		goto out;
1852	/* success */
1853	*pkp = n;
1854	n = NULL;
1855	r = 0;
1856 out:
1857	sshkey_free(n);
1858#ifdef WITH_OPENSSL
1859	BN_clear_free(rsa_n_dup);
1860	BN_clear_free(rsa_e_dup);
1861	BN_clear_free(dsa_p_dup);
1862	BN_clear_free(dsa_q_dup);
1863	BN_clear_free(dsa_g_dup);
1864	BN_clear_free(dsa_pub_key_dup);
1865#endif
1866
1867	return r;
1868}
1869
1870static int
1871cert_parse(struct sshbuf *b, struct sshkey *key, struct sshbuf *certbuf)
1872{
1873	struct sshbuf *principals = NULL, *crit = NULL;
1874	struct sshbuf *exts = NULL, *ca = NULL;
1875	u_char *sig = NULL;
1876	size_t signed_len = 0, slen = 0, kidlen = 0;
1877	int ret = SSH_ERR_INTERNAL_ERROR;
1878
1879	/* Copy the entire key blob for verification and later serialisation */
1880	if ((ret = sshbuf_putb(key->cert->certblob, certbuf)) != 0)
1881		return ret;
1882
1883	/* Parse body of certificate up to signature */
1884	if ((ret = sshbuf_get_u64(b, &key->cert->serial)) != 0 ||
1885	    (ret = sshbuf_get_u32(b, &key->cert->type)) != 0 ||
1886	    (ret = sshbuf_get_cstring(b, &key->cert->key_id, &kidlen)) != 0 ||
1887	    (ret = sshbuf_froms(b, &principals)) != 0 ||
1888	    (ret = sshbuf_get_u64(b, &key->cert->valid_after)) != 0 ||
1889	    (ret = sshbuf_get_u64(b, &key->cert->valid_before)) != 0 ||
1890	    (ret = sshbuf_froms(b, &crit)) != 0 ||
1891	    (ret = sshbuf_froms(b, &exts)) != 0 ||
1892	    (ret = sshbuf_get_string_direct(b, NULL, NULL)) != 0 ||
1893	    (ret = sshbuf_froms(b, &ca)) != 0) {
1894		/* XXX debug print error for ret */
1895		ret = SSH_ERR_INVALID_FORMAT;
1896		goto out;
1897	}
1898
1899	/* Signature is left in the buffer so we can calculate this length */
1900	signed_len = sshbuf_len(key->cert->certblob) - sshbuf_len(b);
1901
1902	if ((ret = sshbuf_get_string(b, &sig, &slen)) != 0) {
1903		ret = SSH_ERR_INVALID_FORMAT;
1904		goto out;
1905	}
1906
1907	if (key->cert->type != SSH2_CERT_TYPE_USER &&
1908	    key->cert->type != SSH2_CERT_TYPE_HOST) {
1909		ret = SSH_ERR_KEY_CERT_UNKNOWN_TYPE;
1910		goto out;
1911	}
1912
1913	/* Parse principals section */
1914	while (sshbuf_len(principals) > 0) {
1915		char *principal = NULL;
1916		char **oprincipals = NULL;
1917
1918		if (key->cert->nprincipals >= SSHKEY_CERT_MAX_PRINCIPALS) {
1919			ret = SSH_ERR_INVALID_FORMAT;
1920			goto out;
1921		}
1922		if ((ret = sshbuf_get_cstring(principals, &principal,
1923		    NULL)) != 0) {
1924			ret = SSH_ERR_INVALID_FORMAT;
1925			goto out;
1926		}
1927		oprincipals = key->cert->principals;
1928		key->cert->principals = recallocarray(key->cert->principals,
1929		    key->cert->nprincipals, key->cert->nprincipals + 1,
1930		    sizeof(*key->cert->principals));
1931		if (key->cert->principals == NULL) {
1932			free(principal);
1933			key->cert->principals = oprincipals;
1934			ret = SSH_ERR_ALLOC_FAIL;
1935			goto out;
1936		}
1937		key->cert->principals[key->cert->nprincipals++] = principal;
1938	}
1939
1940	/*
1941	 * Stash a copies of the critical options and extensions sections
1942	 * for later use.
1943	 */
1944	if ((ret = sshbuf_putb(key->cert->critical, crit)) != 0 ||
1945	    (exts != NULL &&
1946	    (ret = sshbuf_putb(key->cert->extensions, exts)) != 0))
1947		goto out;
1948
1949	/*
1950	 * Validate critical options and extensions sections format.
1951	 */
1952	while (sshbuf_len(crit) != 0) {
1953		if ((ret = sshbuf_get_string_direct(crit, NULL, NULL)) != 0 ||
1954		    (ret = sshbuf_get_string_direct(crit, NULL, NULL)) != 0) {
1955			sshbuf_reset(key->cert->critical);
1956			ret = SSH_ERR_INVALID_FORMAT;
1957			goto out;
1958		}
1959	}
1960	while (exts != NULL && sshbuf_len(exts) != 0) {
1961		if ((ret = sshbuf_get_string_direct(exts, NULL, NULL)) != 0 ||
1962		    (ret = sshbuf_get_string_direct(exts, NULL, NULL)) != 0) {
1963			sshbuf_reset(key->cert->extensions);
1964			ret = SSH_ERR_INVALID_FORMAT;
1965			goto out;
1966		}
1967	}
1968
1969	/* Parse CA key and check signature */
1970	if (sshkey_from_blob_internal(ca, &key->cert->signature_key, 0) != 0) {
1971		ret = SSH_ERR_KEY_CERT_INVALID_SIGN_KEY;
1972		goto out;
1973	}
1974	if (!sshkey_type_is_valid_ca(key->cert->signature_key->type)) {
1975		ret = SSH_ERR_KEY_CERT_INVALID_SIGN_KEY;
1976		goto out;
1977	}
1978	if ((ret = sshkey_verify(key->cert->signature_key, sig, slen,
1979	    sshbuf_ptr(key->cert->certblob), signed_len, NULL, 0)) != 0)
1980		goto out;
1981	if ((ret = get_sigtype(sig, slen, &key->cert->signature_type)) != 0)
1982		goto out;
1983
1984	/* Success */
1985	ret = 0;
1986 out:
1987	sshbuf_free(ca);
1988	sshbuf_free(crit);
1989	sshbuf_free(exts);
1990	sshbuf_free(principals);
1991	free(sig);
1992	return ret;
1993}
1994
1995#ifdef WITH_OPENSSL
1996static int
1997check_rsa_length(const RSA *rsa)
1998{
1999	const BIGNUM *rsa_n;
2000
2001	RSA_get0_key(rsa, &rsa_n, NULL, NULL);
2002	if (BN_num_bits(rsa_n) < SSH_RSA_MINIMUM_MODULUS_SIZE)
2003		return SSH_ERR_KEY_LENGTH;
2004	return 0;
2005}
2006#endif
2007
2008static int
2009sshkey_from_blob_internal(struct sshbuf *b, struct sshkey **keyp,
2010    int allow_cert)
2011{
2012	int type, ret = SSH_ERR_INTERNAL_ERROR;
2013	char *ktype = NULL, *curve = NULL, *xmss_name = NULL;
2014	struct sshkey *key = NULL;
2015	size_t len;
2016	u_char *pk = NULL;
2017	struct sshbuf *copy;
2018#if defined(WITH_OPENSSL)
2019	BIGNUM *rsa_n = NULL, *rsa_e = NULL;
2020	BIGNUM *dsa_p = NULL, *dsa_q = NULL, *dsa_g = NULL, *dsa_pub_key = NULL;
2021# if defined(OPENSSL_HAS_ECC)
2022	EC_POINT *q = NULL;
2023# endif /* OPENSSL_HAS_ECC */
2024#endif /* WITH_OPENSSL */
2025
2026#ifdef DEBUG_PK /* XXX */
2027	sshbuf_dump(b, stderr);
2028#endif
2029	if (keyp != NULL)
2030		*keyp = NULL;
2031	if ((copy = sshbuf_fromb(b)) == NULL) {
2032		ret = SSH_ERR_ALLOC_FAIL;
2033		goto out;
2034	}
2035	if (sshbuf_get_cstring(b, &ktype, NULL) != 0) {
2036		ret = SSH_ERR_INVALID_FORMAT;
2037		goto out;
2038	}
2039
2040	type = sshkey_type_from_name(ktype);
2041	if (!allow_cert && sshkey_type_is_cert(type)) {
2042		ret = SSH_ERR_KEY_CERT_INVALID_SIGN_KEY;
2043		goto out;
2044	}
2045	switch (type) {
2046#ifdef WITH_OPENSSL
2047	case KEY_RSA_CERT:
2048		/* Skip nonce */
2049		if (sshbuf_get_string_direct(b, NULL, NULL) != 0) {
2050			ret = SSH_ERR_INVALID_FORMAT;
2051			goto out;
2052		}
2053		/* FALLTHROUGH */
2054	case KEY_RSA:
2055		if ((key = sshkey_new(type)) == NULL) {
2056			ret = SSH_ERR_ALLOC_FAIL;
2057			goto out;
2058		}
2059		if ((rsa_e = BN_new()) == NULL ||
2060		    (rsa_n = BN_new()) == NULL) {
2061			ret = SSH_ERR_ALLOC_FAIL;
2062			goto out;
2063		}
2064		if (sshbuf_get_bignum2(b, rsa_e) != 0 ||
2065		    sshbuf_get_bignum2(b, rsa_n) != 0) {
2066			ret = SSH_ERR_INVALID_FORMAT;
2067			goto out;
2068		}
2069		if (!RSA_set0_key(key->rsa, rsa_n, rsa_e, NULL)) {
2070			ret = SSH_ERR_LIBCRYPTO_ERROR;
2071			goto out;
2072		}
2073		rsa_n = rsa_e = NULL; /* transferred */
2074		if ((ret = check_rsa_length(key->rsa)) != 0)
2075			goto out;
2076#ifdef DEBUG_PK
2077		RSA_print_fp(stderr, key->rsa, 8);
2078#endif
2079		break;
2080	case KEY_DSA_CERT:
2081		/* Skip nonce */
2082		if (sshbuf_get_string_direct(b, NULL, NULL) != 0) {
2083			ret = SSH_ERR_INVALID_FORMAT;
2084			goto out;
2085		}
2086		/* FALLTHROUGH */
2087	case KEY_DSA:
2088		if ((key = sshkey_new(type)) == NULL) {
2089			ret = SSH_ERR_ALLOC_FAIL;
2090			goto out;
2091		}
2092		if ((dsa_p = BN_new()) == NULL ||
2093		    (dsa_q = BN_new()) == NULL ||
2094		    (dsa_g = BN_new()) == NULL ||
2095		    (dsa_pub_key = BN_new()) == NULL) {
2096			ret = SSH_ERR_ALLOC_FAIL;
2097			goto out;
2098		}
2099		if (sshbuf_get_bignum2(b, dsa_p) != 0 ||
2100		    sshbuf_get_bignum2(b, dsa_q) != 0 ||
2101		    sshbuf_get_bignum2(b, dsa_g) != 0 ||
2102		    sshbuf_get_bignum2(b, dsa_pub_key) != 0) {
2103			ret = SSH_ERR_INVALID_FORMAT;
2104			goto out;
2105		}
2106		if (!DSA_set0_pqg(key->dsa, dsa_p, dsa_q, dsa_g)) {
2107			ret = SSH_ERR_LIBCRYPTO_ERROR;
2108			goto out;
2109		}
2110		dsa_p = dsa_q = dsa_g = NULL; /* transferred */
2111		if (!DSA_set0_key(key->dsa, dsa_pub_key, NULL)) {
2112			ret = SSH_ERR_LIBCRYPTO_ERROR;
2113			goto out;
2114		}
2115		dsa_pub_key = NULL; /* transferred */
2116#ifdef DEBUG_PK
2117		DSA_print_fp(stderr, key->dsa, 8);
2118#endif
2119		break;
2120	case KEY_ECDSA_CERT:
2121		/* Skip nonce */
2122		if (sshbuf_get_string_direct(b, NULL, NULL) != 0) {
2123			ret = SSH_ERR_INVALID_FORMAT;
2124			goto out;
2125		}
2126		/* FALLTHROUGH */
2127# ifdef OPENSSL_HAS_ECC
2128	case KEY_ECDSA:
2129		if ((key = sshkey_new(type)) == NULL) {
2130			ret = SSH_ERR_ALLOC_FAIL;
2131			goto out;
2132		}
2133		key->ecdsa_nid = sshkey_ecdsa_nid_from_name(ktype);
2134		if (sshbuf_get_cstring(b, &curve, NULL) != 0) {
2135			ret = SSH_ERR_INVALID_FORMAT;
2136			goto out;
2137		}
2138		if (key->ecdsa_nid != sshkey_curve_name_to_nid(curve)) {
2139			ret = SSH_ERR_EC_CURVE_MISMATCH;
2140			goto out;
2141		}
2142		EC_KEY_free(key->ecdsa);
2143		if ((key->ecdsa = EC_KEY_new_by_curve_name(key->ecdsa_nid))
2144		    == NULL) {
2145			ret = SSH_ERR_EC_CURVE_INVALID;
2146			goto out;
2147		}
2148		if ((q = EC_POINT_new(EC_KEY_get0_group(key->ecdsa))) == NULL) {
2149			ret = SSH_ERR_ALLOC_FAIL;
2150			goto out;
2151		}
2152		if (sshbuf_get_ec(b, q, EC_KEY_get0_group(key->ecdsa)) != 0) {
2153			ret = SSH_ERR_INVALID_FORMAT;
2154			goto out;
2155		}
2156		if (sshkey_ec_validate_public(EC_KEY_get0_group(key->ecdsa),
2157		    q) != 0) {
2158			ret = SSH_ERR_KEY_INVALID_EC_VALUE;
2159			goto out;
2160		}
2161		if (EC_KEY_set_public_key(key->ecdsa, q) != 1) {
2162			/* XXX assume it is a allocation error */
2163			ret = SSH_ERR_ALLOC_FAIL;
2164			goto out;
2165		}
2166#ifdef DEBUG_PK
2167		sshkey_dump_ec_point(EC_KEY_get0_group(key->ecdsa), q);
2168#endif
2169		break;
2170# endif /* OPENSSL_HAS_ECC */
2171#endif /* WITH_OPENSSL */
2172	case KEY_ED25519_CERT:
2173		/* Skip nonce */
2174		if (sshbuf_get_string_direct(b, NULL, NULL) != 0) {
2175			ret = SSH_ERR_INVALID_FORMAT;
2176			goto out;
2177		}
2178		/* FALLTHROUGH */
2179	case KEY_ED25519:
2180		if ((ret = sshbuf_get_string(b, &pk, &len)) != 0)
2181			goto out;
2182		if (len != ED25519_PK_SZ) {
2183			ret = SSH_ERR_INVALID_FORMAT;
2184			goto out;
2185		}
2186		if ((key = sshkey_new(type)) == NULL) {
2187			ret = SSH_ERR_ALLOC_FAIL;
2188			goto out;
2189		}
2190		key->ed25519_pk = pk;
2191		pk = NULL;
2192		break;
2193#ifdef WITH_XMSS
2194	case KEY_XMSS_CERT:
2195		/* Skip nonce */
2196		if (sshbuf_get_string_direct(b, NULL, NULL) != 0) {
2197			ret = SSH_ERR_INVALID_FORMAT;
2198			goto out;
2199		}
2200		/* FALLTHROUGH */
2201	case KEY_XMSS:
2202		if ((ret = sshbuf_get_cstring(b, &xmss_name, NULL)) != 0)
2203			goto out;
2204		if ((key = sshkey_new(type)) == NULL) {
2205			ret = SSH_ERR_ALLOC_FAIL;
2206			goto out;
2207		}
2208		if ((ret = sshkey_xmss_init(key, xmss_name)) != 0)
2209			goto out;
2210		if ((ret = sshbuf_get_string(b, &pk, &len)) != 0)
2211			goto out;
2212		if (len == 0 || len != sshkey_xmss_pklen(key)) {
2213			ret = SSH_ERR_INVALID_FORMAT;
2214			goto out;
2215		}
2216		key->xmss_pk = pk;
2217		pk = NULL;
2218		if (type != KEY_XMSS_CERT &&
2219		    (ret = sshkey_xmss_deserialize_pk_info(key, b)) != 0)
2220			goto out;
2221		break;
2222#endif /* WITH_XMSS */
2223	case KEY_UNSPEC:
2224	default:
2225		ret = SSH_ERR_KEY_TYPE_UNKNOWN;
2226		goto out;
2227	}
2228
2229	/* Parse certificate potion */
2230	if (sshkey_is_cert(key) && (ret = cert_parse(b, key, copy)) != 0)
2231		goto out;
2232
2233	if (key != NULL && sshbuf_len(b) != 0) {
2234		ret = SSH_ERR_INVALID_FORMAT;
2235		goto out;
2236	}
2237	ret = 0;
2238	if (keyp != NULL) {
2239		*keyp = key;
2240		key = NULL;
2241	}
2242 out:
2243	sshbuf_free(copy);
2244	sshkey_free(key);
2245	free(xmss_name);
2246	free(ktype);
2247	free(curve);
2248	free(pk);
2249#if defined(WITH_OPENSSL)
2250	BN_clear_free(rsa_n);
2251	BN_clear_free(rsa_e);
2252	BN_clear_free(dsa_p);
2253	BN_clear_free(dsa_q);
2254	BN_clear_free(dsa_g);
2255	BN_clear_free(dsa_pub_key);
2256# if defined(OPENSSL_HAS_ECC)
2257	EC_POINT_free(q);
2258# endif /* OPENSSL_HAS_ECC */
2259#endif /* WITH_OPENSSL */
2260	return ret;
2261}
2262
2263int
2264sshkey_from_blob(const u_char *blob, size_t blen, struct sshkey **keyp)
2265{
2266	struct sshbuf *b;
2267	int r;
2268
2269	if ((b = sshbuf_from(blob, blen)) == NULL)
2270		return SSH_ERR_ALLOC_FAIL;
2271	r = sshkey_from_blob_internal(b, keyp, 1);
2272	sshbuf_free(b);
2273	return r;
2274}
2275
2276int
2277sshkey_fromb(struct sshbuf *b, struct sshkey **keyp)
2278{
2279	return sshkey_from_blob_internal(b, keyp, 1);
2280}
2281
2282int
2283sshkey_froms(struct sshbuf *buf, struct sshkey **keyp)
2284{
2285	struct sshbuf *b;
2286	int r;
2287
2288	if ((r = sshbuf_froms(buf, &b)) != 0)
2289		return r;
2290	r = sshkey_from_blob_internal(b, keyp, 1);
2291	sshbuf_free(b);
2292	return r;
2293}
2294
2295static int
2296get_sigtype(const u_char *sig, size_t siglen, char **sigtypep)
2297{
2298	int r;
2299	struct sshbuf *b = NULL;
2300	char *sigtype = NULL;
2301
2302	if (sigtypep != NULL)
2303		*sigtypep = NULL;
2304	if ((b = sshbuf_from(sig, siglen)) == NULL)
2305		return SSH_ERR_ALLOC_FAIL;
2306	if ((r = sshbuf_get_cstring(b, &sigtype, NULL)) != 0)
2307		goto out;
2308	/* success */
2309	if (sigtypep != NULL) {
2310		*sigtypep = sigtype;
2311		sigtype = NULL;
2312	}
2313	r = 0;
2314 out:
2315	free(sigtype);
2316	sshbuf_free(b);
2317	return r;
2318}
2319
2320/*
2321 *
2322 * Checks whether a certificate's signature type is allowed.
2323 * Returns 0 (success) if the certificate signature type appears in the
2324 * "allowed" pattern-list, or the key is not a certificate to begin with.
2325 * Otherwise returns a ssherr.h code.
2326 */
2327int
2328sshkey_check_cert_sigtype(const struct sshkey *key, const char *allowed)
2329{
2330	if (key == NULL || allowed == NULL)
2331		return SSH_ERR_INVALID_ARGUMENT;
2332	if (!sshkey_type_is_cert(key->type))
2333		return 0;
2334	if (key->cert == NULL || key->cert->signature_type == NULL)
2335		return SSH_ERR_INVALID_ARGUMENT;
2336	if (match_pattern_list(key->cert->signature_type, allowed, 0) != 1)
2337		return SSH_ERR_SIGN_ALG_UNSUPPORTED;
2338	return 0;
2339}
2340
2341/*
2342 * Returns the expected signature algorithm for a given public key algorithm.
2343 */
2344const char *
2345sshkey_sigalg_by_name(const char *name)
2346{
2347	const struct keytype *kt;
2348
2349	for (kt = keytypes; kt->type != -1; kt++) {
2350		if (strcmp(kt->name, name) != 0)
2351			continue;
2352		if (kt->sigalg != NULL)
2353			return kt->sigalg;
2354		if (!kt->cert)
2355			return kt->name;
2356		return sshkey_ssh_name_from_type_nid(
2357		    sshkey_type_plain(kt->type), kt->nid);
2358	}
2359	return NULL;
2360}
2361
2362/*
2363 * Verifies that the signature algorithm appearing inside the signature blob
2364 * matches that which was requested.
2365 */
2366int
2367sshkey_check_sigtype(const u_char *sig, size_t siglen,
2368    const char *requested_alg)
2369{
2370	const char *expected_alg;
2371	char *sigtype = NULL;
2372	int r;
2373
2374	if (requested_alg == NULL)
2375		return 0;
2376	if ((expected_alg = sshkey_sigalg_by_name(requested_alg)) == NULL)
2377		return SSH_ERR_INVALID_ARGUMENT;
2378	if ((r = get_sigtype(sig, siglen, &sigtype)) != 0)
2379		return r;
2380	r = strcmp(expected_alg, sigtype) == 0;
2381	free(sigtype);
2382	return r ? 0 : SSH_ERR_SIGN_ALG_UNSUPPORTED;
2383}
2384
2385int
2386sshkey_sign(const struct sshkey *key,
2387    u_char **sigp, size_t *lenp,
2388    const u_char *data, size_t datalen, const char *alg, u_int compat)
2389{
2390	if (sigp != NULL)
2391		*sigp = NULL;
2392	if (lenp != NULL)
2393		*lenp = 0;
2394	if (datalen > SSH_KEY_MAX_SIGN_DATA_SIZE)
2395		return SSH_ERR_INVALID_ARGUMENT;
2396	switch (key->type) {
2397#ifdef WITH_OPENSSL
2398	case KEY_DSA_CERT:
2399	case KEY_DSA:
2400		return ssh_dss_sign(key, sigp, lenp, data, datalen, compat);
2401# ifdef OPENSSL_HAS_ECC
2402	case KEY_ECDSA_CERT:
2403	case KEY_ECDSA:
2404		return ssh_ecdsa_sign(key, sigp, lenp, data, datalen, compat);
2405# endif /* OPENSSL_HAS_ECC */
2406	case KEY_RSA_CERT:
2407	case KEY_RSA:
2408		return ssh_rsa_sign(key, sigp, lenp, data, datalen, alg);
2409#endif /* WITH_OPENSSL */
2410	case KEY_ED25519:
2411	case KEY_ED25519_CERT:
2412		return ssh_ed25519_sign(key, sigp, lenp, data, datalen, compat);
2413#ifdef WITH_XMSS
2414	case KEY_XMSS:
2415	case KEY_XMSS_CERT:
2416		return ssh_xmss_sign(key, sigp, lenp, data, datalen, compat);
2417#endif /* WITH_XMSS */
2418	default:
2419		return SSH_ERR_KEY_TYPE_UNKNOWN;
2420	}
2421}
2422
2423/*
2424 * ssh_key_verify returns 0 for a correct signature  and < 0 on error.
2425 * If "alg" specified, then the signature must use that algorithm.
2426 */
2427int
2428sshkey_verify(const struct sshkey *key,
2429    const u_char *sig, size_t siglen,
2430    const u_char *data, size_t dlen, const char *alg, u_int compat)
2431{
2432	if (siglen == 0 || dlen > SSH_KEY_MAX_SIGN_DATA_SIZE)
2433		return SSH_ERR_INVALID_ARGUMENT;
2434	switch (key->type) {
2435#ifdef WITH_OPENSSL
2436	case KEY_DSA_CERT:
2437	case KEY_DSA:
2438		return ssh_dss_verify(key, sig, siglen, data, dlen, compat);
2439# ifdef OPENSSL_HAS_ECC
2440	case KEY_ECDSA_CERT:
2441	case KEY_ECDSA:
2442		return ssh_ecdsa_verify(key, sig, siglen, data, dlen, compat);
2443# endif /* OPENSSL_HAS_ECC */
2444	case KEY_RSA_CERT:
2445	case KEY_RSA:
2446		return ssh_rsa_verify(key, sig, siglen, data, dlen, alg);
2447#endif /* WITH_OPENSSL */
2448	case KEY_ED25519:
2449	case KEY_ED25519_CERT:
2450		return ssh_ed25519_verify(key, sig, siglen, data, dlen, compat);
2451#ifdef WITH_XMSS
2452	case KEY_XMSS:
2453	case KEY_XMSS_CERT:
2454		return ssh_xmss_verify(key, sig, siglen, data, dlen, compat);
2455#endif /* WITH_XMSS */
2456	default:
2457		return SSH_ERR_KEY_TYPE_UNKNOWN;
2458	}
2459}
2460
2461/* Convert a plain key to their _CERT equivalent */
2462int
2463sshkey_to_certified(struct sshkey *k)
2464{
2465	int newtype;
2466
2467	switch (k->type) {
2468#ifdef WITH_OPENSSL
2469	case KEY_RSA:
2470		newtype = KEY_RSA_CERT;
2471		break;
2472	case KEY_DSA:
2473		newtype = KEY_DSA_CERT;
2474		break;
2475	case KEY_ECDSA:
2476		newtype = KEY_ECDSA_CERT;
2477		break;
2478#endif /* WITH_OPENSSL */
2479	case KEY_ED25519:
2480		newtype = KEY_ED25519_CERT;
2481		break;
2482#ifdef WITH_XMSS
2483	case KEY_XMSS:
2484		newtype = KEY_XMSS_CERT;
2485		break;
2486#endif /* WITH_XMSS */
2487	default:
2488		return SSH_ERR_INVALID_ARGUMENT;
2489	}
2490	if ((k->cert = cert_new()) == NULL)
2491		return SSH_ERR_ALLOC_FAIL;
2492	k->type = newtype;
2493	return 0;
2494}
2495
2496/* Convert a certificate to its raw key equivalent */
2497int
2498sshkey_drop_cert(struct sshkey *k)
2499{
2500	if (!sshkey_type_is_cert(k->type))
2501		return SSH_ERR_KEY_TYPE_UNKNOWN;
2502	cert_free(k->cert);
2503	k->cert = NULL;
2504	k->type = sshkey_type_plain(k->type);
2505	return 0;
2506}
2507
2508/* Sign a certified key, (re-)generating the signed certblob. */
2509int
2510sshkey_certify_custom(struct sshkey *k, struct sshkey *ca, const char *alg,
2511    sshkey_certify_signer *signer, void *signer_ctx)
2512{
2513	struct sshbuf *principals = NULL;
2514	u_char *ca_blob = NULL, *sig_blob = NULL, nonce[32];
2515	size_t i, ca_len, sig_len;
2516	int ret = SSH_ERR_INTERNAL_ERROR;
2517	struct sshbuf *cert = NULL;
2518	char *sigtype = NULL;
2519#ifdef WITH_OPENSSL
2520	const BIGNUM *rsa_n, *rsa_e, *dsa_p, *dsa_q, *dsa_g, *dsa_pub_key;
2521#endif /* WITH_OPENSSL */
2522
2523	if (k == NULL || k->cert == NULL ||
2524	    k->cert->certblob == NULL || ca == NULL)
2525		return SSH_ERR_INVALID_ARGUMENT;
2526	if (!sshkey_is_cert(k))
2527		return SSH_ERR_KEY_TYPE_UNKNOWN;
2528	if (!sshkey_type_is_valid_ca(ca->type))
2529		return SSH_ERR_KEY_CERT_INVALID_SIGN_KEY;
2530
2531	/*
2532	 * If no alg specified as argument but a signature_type was set,
2533	 * then prefer that. If both were specified, then they must match.
2534	 */
2535	if (alg == NULL)
2536		alg = k->cert->signature_type;
2537	else if (k->cert->signature_type != NULL &&
2538	    strcmp(alg, k->cert->signature_type) != 0)
2539		return SSH_ERR_INVALID_ARGUMENT;
2540
2541	if ((ret = sshkey_to_blob(ca, &ca_blob, &ca_len)) != 0)
2542		return SSH_ERR_KEY_CERT_INVALID_SIGN_KEY;
2543
2544	cert = k->cert->certblob; /* for readability */
2545	sshbuf_reset(cert);
2546	if ((ret = sshbuf_put_cstring(cert, sshkey_ssh_name(k))) != 0)
2547		goto out;
2548
2549	/* -v01 certs put nonce first */
2550	arc4random_buf(&nonce, sizeof(nonce));
2551	if ((ret = sshbuf_put_string(cert, nonce, sizeof(nonce))) != 0)
2552		goto out;
2553
2554	/* XXX this substantially duplicates to_blob(); refactor */
2555	switch (k->type) {
2556#ifdef WITH_OPENSSL
2557	case KEY_DSA_CERT:
2558		DSA_get0_pqg(k->dsa, &dsa_p, &dsa_q, &dsa_g);
2559		DSA_get0_key(k->dsa, &dsa_pub_key, NULL);
2560		if ((ret = sshbuf_put_bignum2(cert, dsa_p)) != 0 ||
2561		    (ret = sshbuf_put_bignum2(cert, dsa_q)) != 0 ||
2562		    (ret = sshbuf_put_bignum2(cert, dsa_g)) != 0 ||
2563		    (ret = sshbuf_put_bignum2(cert, dsa_pub_key)) != 0)
2564			goto out;
2565		break;
2566# ifdef OPENSSL_HAS_ECC
2567	case KEY_ECDSA_CERT:
2568		if ((ret = sshbuf_put_cstring(cert,
2569		    sshkey_curve_nid_to_name(k->ecdsa_nid))) != 0 ||
2570		    (ret = sshbuf_put_ec(cert,
2571		    EC_KEY_get0_public_key(k->ecdsa),
2572		    EC_KEY_get0_group(k->ecdsa))) != 0)
2573			goto out;
2574		break;
2575# endif /* OPENSSL_HAS_ECC */
2576	case KEY_RSA_CERT:
2577		RSA_get0_key(k->rsa, &rsa_n, &rsa_e, NULL);
2578		if ((ret = sshbuf_put_bignum2(cert, rsa_e)) != 0 ||
2579		    (ret = sshbuf_put_bignum2(cert, rsa_n)) != 0)
2580			goto out;
2581		break;
2582#endif /* WITH_OPENSSL */
2583	case KEY_ED25519_CERT:
2584		if ((ret = sshbuf_put_string(cert,
2585		    k->ed25519_pk, ED25519_PK_SZ)) != 0)
2586			goto out;
2587		break;
2588#ifdef WITH_XMSS
2589	case KEY_XMSS_CERT:
2590		if (k->xmss_name == NULL) {
2591			ret = SSH_ERR_INVALID_ARGUMENT;
2592			goto out;
2593		}
2594		if ((ret = sshbuf_put_cstring(cert, k->xmss_name)) ||
2595		    (ret = sshbuf_put_string(cert,
2596		    k->xmss_pk, sshkey_xmss_pklen(k))) != 0)
2597			goto out;
2598		break;
2599#endif /* WITH_XMSS */
2600	default:
2601		ret = SSH_ERR_INVALID_ARGUMENT;
2602		goto out;
2603	}
2604
2605	if ((ret = sshbuf_put_u64(cert, k->cert->serial)) != 0 ||
2606	    (ret = sshbuf_put_u32(cert, k->cert->type)) != 0 ||
2607	    (ret = sshbuf_put_cstring(cert, k->cert->key_id)) != 0)
2608		goto out;
2609
2610	if ((principals = sshbuf_new()) == NULL) {
2611		ret = SSH_ERR_ALLOC_FAIL;
2612		goto out;
2613	}
2614	for (i = 0; i < k->cert->nprincipals; i++) {
2615		if ((ret = sshbuf_put_cstring(principals,
2616		    k->cert->principals[i])) != 0)
2617			goto out;
2618	}
2619	if ((ret = sshbuf_put_stringb(cert, principals)) != 0 ||
2620	    (ret = sshbuf_put_u64(cert, k->cert->valid_after)) != 0 ||
2621	    (ret = sshbuf_put_u64(cert, k->cert->valid_before)) != 0 ||
2622	    (ret = sshbuf_put_stringb(cert, k->cert->critical)) != 0 ||
2623	    (ret = sshbuf_put_stringb(cert, k->cert->extensions)) != 0 ||
2624	    (ret = sshbuf_put_string(cert, NULL, 0)) != 0 || /* Reserved */
2625	    (ret = sshbuf_put_string(cert, ca_blob, ca_len)) != 0)
2626		goto out;
2627
2628	/* Sign the whole mess */
2629	if ((ret = signer(ca, &sig_blob, &sig_len, sshbuf_ptr(cert),
2630	    sshbuf_len(cert), alg, 0, signer_ctx)) != 0)
2631		goto out;
2632	/* Check and update signature_type against what was actually used */
2633	if ((ret = get_sigtype(sig_blob, sig_len, &sigtype)) != 0)
2634		goto out;
2635	if (alg != NULL && strcmp(alg, sigtype) != 0) {
2636		ret = SSH_ERR_SIGN_ALG_UNSUPPORTED;
2637		goto out;
2638	}
2639	if (k->cert->signature_type == NULL) {
2640		k->cert->signature_type = sigtype;
2641		sigtype = NULL;
2642	}
2643	/* Append signature and we are done */
2644	if ((ret = sshbuf_put_string(cert, sig_blob, sig_len)) != 0)
2645		goto out;
2646	ret = 0;
2647 out:
2648	if (ret != 0)
2649		sshbuf_reset(cert);
2650	free(sig_blob);
2651	free(ca_blob);
2652	free(sigtype);
2653	sshbuf_free(principals);
2654	return ret;
2655}
2656
2657static int
2658default_key_sign(const struct sshkey *key, u_char **sigp, size_t *lenp,
2659    const u_char *data, size_t datalen,
2660    const char *alg, u_int compat, void *ctx)
2661{
2662	if (ctx != NULL)
2663		return SSH_ERR_INVALID_ARGUMENT;
2664	return sshkey_sign(key, sigp, lenp, data, datalen, alg, compat);
2665}
2666
2667int
2668sshkey_certify(struct sshkey *k, struct sshkey *ca, const char *alg)
2669{
2670	return sshkey_certify_custom(k, ca, alg, default_key_sign, NULL);
2671}
2672
2673int
2674sshkey_cert_check_authority(const struct sshkey *k,
2675    int want_host, int require_principal,
2676    const char *name, const char **reason)
2677{
2678	u_int i, principal_matches;
2679	time_t now = time(NULL);
2680
2681	if (reason != NULL)
2682		*reason = NULL;
2683
2684	if (want_host) {
2685		if (k->cert->type != SSH2_CERT_TYPE_HOST) {
2686			*reason = "Certificate invalid: not a host certificate";
2687			return SSH_ERR_KEY_CERT_INVALID;
2688		}
2689	} else {
2690		if (k->cert->type != SSH2_CERT_TYPE_USER) {
2691			*reason = "Certificate invalid: not a user certificate";
2692			return SSH_ERR_KEY_CERT_INVALID;
2693		}
2694	}
2695	if (now < 0) {
2696		/* yikes - system clock before epoch! */
2697		*reason = "Certificate invalid: not yet valid";
2698		return SSH_ERR_KEY_CERT_INVALID;
2699	}
2700	if ((u_int64_t)now < k->cert->valid_after) {
2701		*reason = "Certificate invalid: not yet valid";
2702		return SSH_ERR_KEY_CERT_INVALID;
2703	}
2704	if ((u_int64_t)now >= k->cert->valid_before) {
2705		*reason = "Certificate invalid: expired";
2706		return SSH_ERR_KEY_CERT_INVALID;
2707	}
2708	if (k->cert->nprincipals == 0) {
2709		if (require_principal) {
2710			*reason = "Certificate lacks principal list";
2711			return SSH_ERR_KEY_CERT_INVALID;
2712		}
2713	} else if (name != NULL) {
2714		principal_matches = 0;
2715		for (i = 0; i < k->cert->nprincipals; i++) {
2716			if (strcmp(name, k->cert->principals[i]) == 0) {
2717				principal_matches = 1;
2718				break;
2719			}
2720		}
2721		if (!principal_matches) {
2722			*reason = "Certificate invalid: name is not a listed "
2723			    "principal";
2724			return SSH_ERR_KEY_CERT_INVALID;
2725		}
2726	}
2727	return 0;
2728}
2729
2730size_t
2731sshkey_format_cert_validity(const struct sshkey_cert *cert, char *s, size_t l)
2732{
2733	char from[32], to[32], ret[64];
2734	time_t tt;
2735	struct tm *tm;
2736
2737	*from = *to = '\0';
2738	if (cert->valid_after == 0 &&
2739	    cert->valid_before == 0xffffffffffffffffULL)
2740		return strlcpy(s, "forever", l);
2741
2742	if (cert->valid_after != 0) {
2743		/* XXX revisit INT_MAX in 2038 :) */
2744		tt = cert->valid_after > INT_MAX ?
2745		    INT_MAX : cert->valid_after;
2746		tm = localtime(&tt);
2747		strftime(from, sizeof(from), "%Y-%m-%dT%H:%M:%S", tm);
2748	}
2749	if (cert->valid_before != 0xffffffffffffffffULL) {
2750		/* XXX revisit INT_MAX in 2038 :) */
2751		tt = cert->valid_before > INT_MAX ?
2752		    INT_MAX : cert->valid_before;
2753		tm = localtime(&tt);
2754		strftime(to, sizeof(to), "%Y-%m-%dT%H:%M:%S", tm);
2755	}
2756
2757	if (cert->valid_after == 0)
2758		snprintf(ret, sizeof(ret), "before %s", to);
2759	else if (cert->valid_before == 0xffffffffffffffffULL)
2760		snprintf(ret, sizeof(ret), "after %s", from);
2761	else
2762		snprintf(ret, sizeof(ret), "from %s to %s", from, to);
2763
2764	return strlcpy(s, ret, l);
2765}
2766
2767int
2768sshkey_private_serialize_opt(const struct sshkey *key, struct sshbuf *b,
2769    enum sshkey_serialize_rep opts)
2770{
2771	int r = SSH_ERR_INTERNAL_ERROR;
2772#ifdef WITH_OPENSSL
2773	const BIGNUM *rsa_n, *rsa_e, *rsa_d, *rsa_iqmp, *rsa_p, *rsa_q;
2774	const BIGNUM *dsa_p, *dsa_q, *dsa_g, *dsa_pub_key, *dsa_priv_key;
2775#endif /* WITH_OPENSSL */
2776
2777	if ((r = sshbuf_put_cstring(b, sshkey_ssh_name(key))) != 0)
2778		goto out;
2779	switch (key->type) {
2780#ifdef WITH_OPENSSL
2781	case KEY_RSA:
2782		RSA_get0_key(key->rsa, &rsa_n, &rsa_e, &rsa_d);
2783		RSA_get0_factors(key->rsa, &rsa_p, &rsa_q);
2784		RSA_get0_crt_params(key->rsa, NULL, NULL, &rsa_iqmp);
2785		if ((r = sshbuf_put_bignum2(b, rsa_n)) != 0 ||
2786		    (r = sshbuf_put_bignum2(b, rsa_e)) != 0 ||
2787		    (r = sshbuf_put_bignum2(b, rsa_d)) != 0 ||
2788		    (r = sshbuf_put_bignum2(b, rsa_iqmp)) != 0 ||
2789		    (r = sshbuf_put_bignum2(b, rsa_p)) != 0 ||
2790		    (r = sshbuf_put_bignum2(b, rsa_q)) != 0)
2791			goto out;
2792		break;
2793	case KEY_RSA_CERT:
2794		if (key->cert == NULL || sshbuf_len(key->cert->certblob) == 0) {
2795			r = SSH_ERR_INVALID_ARGUMENT;
2796			goto out;
2797		}
2798		RSA_get0_key(key->rsa, NULL, NULL, &rsa_d);
2799		RSA_get0_factors(key->rsa, &rsa_p, &rsa_q);
2800		RSA_get0_crt_params(key->rsa, NULL, NULL, &rsa_iqmp);
2801		if ((r = sshbuf_put_stringb(b, key->cert->certblob)) != 0 ||
2802		    (r = sshbuf_put_bignum2(b, rsa_d)) != 0 ||
2803		    (r = sshbuf_put_bignum2(b, rsa_iqmp)) != 0 ||
2804		    (r = sshbuf_put_bignum2(b, rsa_p)) != 0 ||
2805		    (r = sshbuf_put_bignum2(b, rsa_q)) != 0)
2806			goto out;
2807		break;
2808	case KEY_DSA:
2809		DSA_get0_pqg(key->dsa, &dsa_p, &dsa_q, &dsa_g);
2810		DSA_get0_key(key->dsa, &dsa_pub_key, &dsa_priv_key);
2811		if ((r = sshbuf_put_bignum2(b, dsa_p)) != 0 ||
2812		    (r = sshbuf_put_bignum2(b, dsa_q)) != 0 ||
2813		    (r = sshbuf_put_bignum2(b, dsa_g)) != 0 ||
2814		    (r = sshbuf_put_bignum2(b, dsa_pub_key)) != 0 ||
2815		    (r = sshbuf_put_bignum2(b, dsa_priv_key)) != 0)
2816			goto out;
2817		break;
2818	case KEY_DSA_CERT:
2819		if (key->cert == NULL || sshbuf_len(key->cert->certblob) == 0) {
2820			r = SSH_ERR_INVALID_ARGUMENT;
2821			goto out;
2822		}
2823		DSA_get0_key(key->dsa, NULL, &dsa_priv_key);
2824		if ((r = sshbuf_put_stringb(b, key->cert->certblob)) != 0 ||
2825		    (r = sshbuf_put_bignum2(b, dsa_priv_key)) != 0)
2826			goto out;
2827		break;
2828# ifdef OPENSSL_HAS_ECC
2829	case KEY_ECDSA:
2830		if ((r = sshbuf_put_cstring(b,
2831		    sshkey_curve_nid_to_name(key->ecdsa_nid))) != 0 ||
2832		    (r = sshbuf_put_eckey(b, key->ecdsa)) != 0 ||
2833		    (r = sshbuf_put_bignum2(b,
2834		    EC_KEY_get0_private_key(key->ecdsa))) != 0)
2835			goto out;
2836		break;
2837	case KEY_ECDSA_CERT:
2838		if (key->cert == NULL || sshbuf_len(key->cert->certblob) == 0) {
2839			r = SSH_ERR_INVALID_ARGUMENT;
2840			goto out;
2841		}
2842		if ((r = sshbuf_put_stringb(b, key->cert->certblob)) != 0 ||
2843		    (r = sshbuf_put_bignum2(b,
2844		    EC_KEY_get0_private_key(key->ecdsa))) != 0)
2845			goto out;
2846		break;
2847# endif /* OPENSSL_HAS_ECC */
2848#endif /* WITH_OPENSSL */
2849	case KEY_ED25519:
2850		if ((r = sshbuf_put_string(b, key->ed25519_pk,
2851		    ED25519_PK_SZ)) != 0 ||
2852		    (r = sshbuf_put_string(b, key->ed25519_sk,
2853		    ED25519_SK_SZ)) != 0)
2854			goto out;
2855		break;
2856	case KEY_ED25519_CERT:
2857		if (key->cert == NULL || sshbuf_len(key->cert->certblob) == 0) {
2858			r = SSH_ERR_INVALID_ARGUMENT;
2859			goto out;
2860		}
2861		if ((r = sshbuf_put_stringb(b, key->cert->certblob)) != 0 ||
2862		    (r = sshbuf_put_string(b, key->ed25519_pk,
2863		    ED25519_PK_SZ)) != 0 ||
2864		    (r = sshbuf_put_string(b, key->ed25519_sk,
2865		    ED25519_SK_SZ)) != 0)
2866			goto out;
2867		break;
2868#ifdef WITH_XMSS
2869	case KEY_XMSS:
2870		if (key->xmss_name == NULL) {
2871			r = SSH_ERR_INVALID_ARGUMENT;
2872			goto out;
2873		}
2874		if ((r = sshbuf_put_cstring(b, key->xmss_name)) != 0 ||
2875		    (r = sshbuf_put_string(b, key->xmss_pk,
2876		    sshkey_xmss_pklen(key))) != 0 ||
2877		    (r = sshbuf_put_string(b, key->xmss_sk,
2878		    sshkey_xmss_sklen(key))) != 0 ||
2879		    (r = sshkey_xmss_serialize_state_opt(key, b, opts)) != 0)
2880			goto out;
2881		break;
2882	case KEY_XMSS_CERT:
2883		if (key->cert == NULL || sshbuf_len(key->cert->certblob) == 0 ||
2884		    key->xmss_name == NULL) {
2885			r = SSH_ERR_INVALID_ARGUMENT;
2886			goto out;
2887		}
2888		if ((r = sshbuf_put_stringb(b, key->cert->certblob)) != 0 ||
2889		    (r = sshbuf_put_cstring(b, key->xmss_name)) != 0 ||
2890		    (r = sshbuf_put_string(b, key->xmss_pk,
2891		    sshkey_xmss_pklen(key))) != 0 ||
2892		    (r = sshbuf_put_string(b, key->xmss_sk,
2893		    sshkey_xmss_sklen(key))) != 0 ||
2894		    (r = sshkey_xmss_serialize_state_opt(key, b, opts)) != 0)
2895			goto out;
2896		break;
2897#endif /* WITH_XMSS */
2898	default:
2899		r = SSH_ERR_INVALID_ARGUMENT;
2900		goto out;
2901	}
2902	/* success */
2903	r = 0;
2904 out:
2905	return r;
2906}
2907
2908int
2909sshkey_private_serialize(const struct sshkey *key, struct sshbuf *b)
2910{
2911	return sshkey_private_serialize_opt(key, b,
2912	    SSHKEY_SERIALIZE_DEFAULT);
2913}
2914
2915int
2916sshkey_private_deserialize(struct sshbuf *buf, struct sshkey **kp)
2917{
2918	char *tname = NULL, *curve = NULL, *xmss_name = NULL;
2919	struct sshkey *k = NULL;
2920	size_t pklen = 0, sklen = 0;
2921	int type, r = SSH_ERR_INTERNAL_ERROR;
2922	u_char *ed25519_pk = NULL, *ed25519_sk = NULL;
2923	u_char *xmss_pk = NULL, *xmss_sk = NULL;
2924#ifdef WITH_OPENSSL
2925	BIGNUM *exponent = NULL;
2926	BIGNUM *rsa_n = NULL, *rsa_e = NULL, *rsa_d = NULL;
2927	BIGNUM *rsa_iqmp = NULL, *rsa_p = NULL, *rsa_q = NULL;
2928	BIGNUM *dsa_p = NULL, *dsa_q = NULL, *dsa_g = NULL;
2929	BIGNUM *dsa_pub_key = NULL, *dsa_priv_key = NULL;
2930#endif /* WITH_OPENSSL */
2931
2932	if (kp != NULL)
2933		*kp = NULL;
2934	if ((r = sshbuf_get_cstring(buf, &tname, NULL)) != 0)
2935		goto out;
2936	type = sshkey_type_from_name(tname);
2937	switch (type) {
2938#ifdef WITH_OPENSSL
2939	case KEY_DSA:
2940		if ((k = sshkey_new(type)) == NULL) {
2941			r = SSH_ERR_ALLOC_FAIL;
2942			goto out;
2943		}
2944		if ((dsa_p = BN_new()) == NULL ||
2945		    (dsa_q = BN_new()) == NULL ||
2946		    (dsa_g = BN_new()) == NULL ||
2947		    (dsa_pub_key = BN_new()) == NULL ||
2948		    (dsa_priv_key = BN_new()) == NULL) {
2949			r = SSH_ERR_ALLOC_FAIL;
2950			goto out;
2951		}
2952		if ((r = sshbuf_get_bignum2(buf, dsa_p)) != 0 ||
2953		    (r = sshbuf_get_bignum2(buf, dsa_q)) != 0 ||
2954		    (r = sshbuf_get_bignum2(buf, dsa_g)) != 0 ||
2955		    (r = sshbuf_get_bignum2(buf, dsa_pub_key)) != 0 ||
2956		    (r = sshbuf_get_bignum2(buf, dsa_priv_key)) != 0)
2957			goto out;
2958		if (!DSA_set0_pqg(k->dsa, dsa_p, dsa_q, dsa_g)) {
2959			r = SSH_ERR_LIBCRYPTO_ERROR;
2960			goto out;
2961		}
2962		dsa_p = dsa_q = dsa_g = NULL; /* transferred */
2963		if (!DSA_set0_key(k->dsa, dsa_pub_key, dsa_priv_key)) {
2964			r = SSH_ERR_LIBCRYPTO_ERROR;
2965			goto out;
2966		}
2967		dsa_pub_key = dsa_priv_key = NULL; /* transferred */
2968		break;
2969	case KEY_DSA_CERT:
2970		if ((dsa_priv_key = BN_new()) == NULL) {
2971			r = SSH_ERR_ALLOC_FAIL;
2972			goto out;
2973		}
2974		if ((r = sshkey_froms(buf, &k)) != 0 ||
2975		    (r = sshbuf_get_bignum2(buf, dsa_priv_key)) != 0)
2976			goto out;
2977		if (!DSA_set0_key(k->dsa, NULL, dsa_priv_key)) {
2978			r = SSH_ERR_LIBCRYPTO_ERROR;
2979			goto out;
2980		}
2981		dsa_priv_key = NULL; /* transferred */
2982		break;
2983# ifdef OPENSSL_HAS_ECC
2984	case KEY_ECDSA:
2985		if ((k = sshkey_new(type)) == NULL) {
2986			r = SSH_ERR_ALLOC_FAIL;
2987			goto out;
2988		}
2989		if ((k->ecdsa_nid = sshkey_ecdsa_nid_from_name(tname)) == -1) {
2990			r = SSH_ERR_INVALID_ARGUMENT;
2991			goto out;
2992		}
2993		if ((r = sshbuf_get_cstring(buf, &curve, NULL)) != 0)
2994			goto out;
2995		if (k->ecdsa_nid != sshkey_curve_name_to_nid(curve)) {
2996			r = SSH_ERR_EC_CURVE_MISMATCH;
2997			goto out;
2998		}
2999		k->ecdsa = EC_KEY_new_by_curve_name(k->ecdsa_nid);
3000		if (k->ecdsa  == NULL || (exponent = BN_new()) == NULL) {
3001			r = SSH_ERR_LIBCRYPTO_ERROR;
3002			goto out;
3003		}
3004		if ((r = sshbuf_get_eckey(buf, k->ecdsa)) != 0 ||
3005		    (r = sshbuf_get_bignum2(buf, exponent)))
3006			goto out;
3007		if (EC_KEY_set_private_key(k->ecdsa, exponent) != 1) {
3008			r = SSH_ERR_LIBCRYPTO_ERROR;
3009			goto out;
3010		}
3011		if ((r = sshkey_ec_validate_public(EC_KEY_get0_group(k->ecdsa),
3012		    EC_KEY_get0_public_key(k->ecdsa))) != 0 ||
3013		    (r = sshkey_ec_validate_private(k->ecdsa)) != 0)
3014			goto out;
3015		break;
3016	case KEY_ECDSA_CERT:
3017		if ((exponent = BN_new()) == NULL) {
3018			r = SSH_ERR_LIBCRYPTO_ERROR;
3019			goto out;
3020		}
3021		if ((r = sshkey_froms(buf, &k)) != 0 ||
3022		    (r = sshbuf_get_bignum2(buf, exponent)) != 0)
3023			goto out;
3024		if (EC_KEY_set_private_key(k->ecdsa, exponent) != 1) {
3025			r = SSH_ERR_LIBCRYPTO_ERROR;
3026			goto out;
3027		}
3028		if ((r = sshkey_ec_validate_public(EC_KEY_get0_group(k->ecdsa),
3029		    EC_KEY_get0_public_key(k->ecdsa))) != 0 ||
3030		    (r = sshkey_ec_validate_private(k->ecdsa)) != 0)
3031			goto out;
3032		break;
3033# endif /* OPENSSL_HAS_ECC */
3034	case KEY_RSA:
3035		if ((k = sshkey_new(type)) == NULL) {
3036			r = SSH_ERR_ALLOC_FAIL;
3037			goto out;
3038		}
3039		if ((rsa_n = BN_new()) == NULL ||
3040		    (rsa_e = BN_new()) == NULL ||
3041		    (rsa_d = BN_new()) == NULL ||
3042		    (rsa_iqmp = BN_new()) == NULL ||
3043		    (rsa_p = BN_new()) == NULL ||
3044		    (rsa_q = BN_new()) == NULL) {
3045			r = SSH_ERR_ALLOC_FAIL;
3046			goto out;
3047		}
3048		if ((r = sshbuf_get_bignum2(buf, rsa_n)) != 0 ||
3049		    (r = sshbuf_get_bignum2(buf, rsa_e)) != 0 ||
3050		    (r = sshbuf_get_bignum2(buf, rsa_d)) != 0 ||
3051		    (r = sshbuf_get_bignum2(buf, rsa_iqmp)) != 0 ||
3052		    (r = sshbuf_get_bignum2(buf, rsa_p)) != 0 ||
3053		    (r = sshbuf_get_bignum2(buf, rsa_q)) != 0)
3054			goto out;
3055		if (!RSA_set0_key(k->rsa, rsa_n, rsa_e, rsa_d)) {
3056			r = SSH_ERR_LIBCRYPTO_ERROR;
3057			goto out;
3058		}
3059		rsa_n = rsa_e = rsa_d = NULL; /* transferred */
3060		if (!RSA_set0_factors(k->rsa, rsa_p, rsa_q)) {
3061			r = SSH_ERR_LIBCRYPTO_ERROR;
3062			goto out;
3063		}
3064		rsa_p = rsa_q = NULL; /* transferred */
3065		if ((r = check_rsa_length(k->rsa)) != 0)
3066			goto out;
3067		if ((r = ssh_rsa_complete_crt_parameters(k, rsa_iqmp)) != 0)
3068			goto out;
3069		break;
3070	case KEY_RSA_CERT:
3071		if ((rsa_d = BN_new()) == NULL ||
3072		    (rsa_iqmp = BN_new()) == NULL ||
3073		    (rsa_p = BN_new()) == NULL ||
3074		    (rsa_q = BN_new()) == NULL) {
3075			r = SSH_ERR_ALLOC_FAIL;
3076			goto out;
3077		}
3078		if ((r = sshkey_froms(buf, &k)) != 0 ||
3079		    (r = sshbuf_get_bignum2(buf, rsa_d)) != 0 ||
3080		    (r = sshbuf_get_bignum2(buf, rsa_iqmp)) != 0 ||
3081		    (r = sshbuf_get_bignum2(buf, rsa_p)) != 0 ||
3082		    (r = sshbuf_get_bignum2(buf, rsa_q)) != 0)
3083			goto out;
3084		if (!RSA_set0_key(k->rsa, NULL, NULL, rsa_d)) {
3085			r = SSH_ERR_LIBCRYPTO_ERROR;
3086			goto out;
3087		}
3088		rsa_d = NULL; /* transferred */
3089		if (!RSA_set0_factors(k->rsa, rsa_p, rsa_q)) {
3090			r = SSH_ERR_LIBCRYPTO_ERROR;
3091			goto out;
3092		}
3093		rsa_p = rsa_q = NULL; /* transferred */
3094		if ((r = check_rsa_length(k->rsa)) != 0)
3095			goto out;
3096		if ((r = ssh_rsa_complete_crt_parameters(k, rsa_iqmp)) != 0)
3097			goto out;
3098		break;
3099#endif /* WITH_OPENSSL */
3100	case KEY_ED25519:
3101		if ((k = sshkey_new(type)) == NULL) {
3102			r = SSH_ERR_ALLOC_FAIL;
3103			goto out;
3104		}
3105		if ((r = sshbuf_get_string(buf, &ed25519_pk, &pklen)) != 0 ||
3106		    (r = sshbuf_get_string(buf, &ed25519_sk, &sklen)) != 0)
3107			goto out;
3108		if (pklen != ED25519_PK_SZ || sklen != ED25519_SK_SZ) {
3109			r = SSH_ERR_INVALID_FORMAT;
3110			goto out;
3111		}
3112		k->ed25519_pk = ed25519_pk;
3113		k->ed25519_sk = ed25519_sk;
3114		ed25519_pk = ed25519_sk = NULL;
3115		break;
3116	case KEY_ED25519_CERT:
3117		if ((r = sshkey_froms(buf, &k)) != 0 ||
3118		    (r = sshbuf_get_string(buf, &ed25519_pk, &pklen)) != 0 ||
3119		    (r = sshbuf_get_string(buf, &ed25519_sk, &sklen)) != 0)
3120			goto out;
3121		if (pklen != ED25519_PK_SZ || sklen != ED25519_SK_SZ) {
3122			r = SSH_ERR_INVALID_FORMAT;
3123			goto out;
3124		}
3125		k->ed25519_pk = ed25519_pk;
3126		k->ed25519_sk = ed25519_sk;
3127		ed25519_pk = ed25519_sk = NULL;
3128		break;
3129#ifdef WITH_XMSS
3130	case KEY_XMSS:
3131		if ((k = sshkey_new(type)) == NULL) {
3132			r = SSH_ERR_ALLOC_FAIL;
3133			goto out;
3134		}
3135		if ((r = sshbuf_get_cstring(buf, &xmss_name, NULL)) != 0 ||
3136		    (r = sshkey_xmss_init(k, xmss_name)) != 0 ||
3137		    (r = sshbuf_get_string(buf, &xmss_pk, &pklen)) != 0 ||
3138		    (r = sshbuf_get_string(buf, &xmss_sk, &sklen)) != 0)
3139			goto out;
3140		if (pklen != sshkey_xmss_pklen(k) ||
3141		    sklen != sshkey_xmss_sklen(k)) {
3142			r = SSH_ERR_INVALID_FORMAT;
3143			goto out;
3144		}
3145		k->xmss_pk = xmss_pk;
3146		k->xmss_sk = xmss_sk;
3147		xmss_pk = xmss_sk = NULL;
3148		/* optional internal state */
3149		if ((r = sshkey_xmss_deserialize_state_opt(k, buf)) != 0)
3150			goto out;
3151		break;
3152	case KEY_XMSS_CERT:
3153		if ((r = sshkey_froms(buf, &k)) != 0 ||
3154		    (r = sshbuf_get_cstring(buf, &xmss_name, NULL)) != 0 ||
3155		    (r = sshbuf_get_string(buf, &xmss_pk, &pklen)) != 0 ||
3156		    (r = sshbuf_get_string(buf, &xmss_sk, &sklen)) != 0)
3157			goto out;
3158		if (strcmp(xmss_name, k->xmss_name)) {
3159			r = SSH_ERR_INVALID_FORMAT;
3160			goto out;
3161		}
3162		if (pklen != sshkey_xmss_pklen(k) ||
3163		    sklen != sshkey_xmss_sklen(k)) {
3164			r = SSH_ERR_INVALID_FORMAT;
3165			goto out;
3166		}
3167		k->xmss_pk = xmss_pk;
3168		k->xmss_sk = xmss_sk;
3169		xmss_pk = xmss_sk = NULL;
3170		/* optional internal state */
3171		if ((r = sshkey_xmss_deserialize_state_opt(k, buf)) != 0)
3172			goto out;
3173		break;
3174#endif /* WITH_XMSS */
3175	default:
3176		r = SSH_ERR_KEY_TYPE_UNKNOWN;
3177		goto out;
3178	}
3179#ifdef WITH_OPENSSL
3180	/* enable blinding */
3181	switch (k->type) {
3182	case KEY_RSA:
3183	case KEY_RSA_CERT:
3184		if (RSA_blinding_on(k->rsa, NULL) != 1) {
3185			r = SSH_ERR_LIBCRYPTO_ERROR;
3186			goto out;
3187		}
3188		break;
3189	}
3190#endif /* WITH_OPENSSL */
3191	/* success */
3192	r = 0;
3193	if (kp != NULL) {
3194		*kp = k;
3195		k = NULL;
3196	}
3197 out:
3198	free(tname);
3199	free(curve);
3200#ifdef WITH_OPENSSL
3201	BN_clear_free(exponent);
3202	BN_clear_free(dsa_p);
3203	BN_clear_free(dsa_q);
3204	BN_clear_free(dsa_g);
3205	BN_clear_free(dsa_pub_key);
3206	BN_clear_free(dsa_priv_key);
3207	BN_clear_free(rsa_n);
3208	BN_clear_free(rsa_e);
3209	BN_clear_free(rsa_d);
3210	BN_clear_free(rsa_p);
3211	BN_clear_free(rsa_q);
3212	BN_clear_free(rsa_iqmp);
3213#endif /* WITH_OPENSSL */
3214	sshkey_free(k);
3215	freezero(ed25519_pk, pklen);
3216	freezero(ed25519_sk, sklen);
3217	free(xmss_name);
3218	freezero(xmss_pk, pklen);
3219	freezero(xmss_sk, sklen);
3220	return r;
3221}
3222
3223#if defined(WITH_OPENSSL) && defined(OPENSSL_HAS_ECC)
3224int
3225sshkey_ec_validate_public(const EC_GROUP *group, const EC_POINT *public)
3226{
3227	BN_CTX *bnctx;
3228	EC_POINT *nq = NULL;
3229	BIGNUM *order, *x, *y, *tmp;
3230	int ret = SSH_ERR_KEY_INVALID_EC_VALUE;
3231
3232	/*
3233	 * NB. This assumes OpenSSL has already verified that the public
3234	 * point lies on the curve. This is done by EC_POINT_oct2point()
3235	 * implicitly calling EC_POINT_is_on_curve(). If this code is ever
3236	 * reachable with public points not unmarshalled using
3237	 * EC_POINT_oct2point then the caller will need to explicitly check.
3238	 */
3239
3240	if ((bnctx = BN_CTX_new()) == NULL)
3241		return SSH_ERR_ALLOC_FAIL;
3242	BN_CTX_start(bnctx);
3243
3244	/*
3245	 * We shouldn't ever hit this case because bignum_get_ecpoint()
3246	 * refuses to load GF2m points.
3247	 */
3248	if (EC_METHOD_get_field_type(EC_GROUP_method_of(group)) !=
3249	    NID_X9_62_prime_field)
3250		goto out;
3251
3252	/* Q != infinity */
3253	if (EC_POINT_is_at_infinity(group, public))
3254		goto out;
3255
3256	if ((x = BN_CTX_get(bnctx)) == NULL ||
3257	    (y = BN_CTX_get(bnctx)) == NULL ||
3258	    (order = BN_CTX_get(bnctx)) == NULL ||
3259	    (tmp = BN_CTX_get(bnctx)) == NULL) {
3260		ret = SSH_ERR_ALLOC_FAIL;
3261		goto out;
3262	}
3263
3264	/* log2(x) > log2(order)/2, log2(y) > log2(order)/2 */
3265	if (EC_GROUP_get_order(group, order, bnctx) != 1 ||
3266	    EC_POINT_get_affine_coordinates_GFp(group, public,
3267	    x, y, bnctx) != 1) {
3268		ret = SSH_ERR_LIBCRYPTO_ERROR;
3269		goto out;
3270	}
3271	if (BN_num_bits(x) <= BN_num_bits(order) / 2 ||
3272	    BN_num_bits(y) <= BN_num_bits(order) / 2)
3273		goto out;
3274
3275	/* nQ == infinity (n == order of subgroup) */
3276	if ((nq = EC_POINT_new(group)) == NULL) {
3277		ret = SSH_ERR_ALLOC_FAIL;
3278		goto out;
3279	}
3280	if (EC_POINT_mul(group, nq, NULL, public, order, bnctx) != 1) {
3281		ret = SSH_ERR_LIBCRYPTO_ERROR;
3282		goto out;
3283	}
3284	if (EC_POINT_is_at_infinity(group, nq) != 1)
3285		goto out;
3286
3287	/* x < order - 1, y < order - 1 */
3288	if (!BN_sub(tmp, order, BN_value_one())) {
3289		ret = SSH_ERR_LIBCRYPTO_ERROR;
3290		goto out;
3291	}
3292	if (BN_cmp(x, tmp) >= 0 || BN_cmp(y, tmp) >= 0)
3293		goto out;
3294	ret = 0;
3295 out:
3296	BN_CTX_free(bnctx);
3297	EC_POINT_free(nq);
3298	return ret;
3299}
3300
3301int
3302sshkey_ec_validate_private(const EC_KEY *key)
3303{
3304	BN_CTX *bnctx;
3305	BIGNUM *order, *tmp;
3306	int ret = SSH_ERR_KEY_INVALID_EC_VALUE;
3307
3308	if ((bnctx = BN_CTX_new()) == NULL)
3309		return SSH_ERR_ALLOC_FAIL;
3310	BN_CTX_start(bnctx);
3311
3312	if ((order = BN_CTX_get(bnctx)) == NULL ||
3313	    (tmp = BN_CTX_get(bnctx)) == NULL) {
3314		ret = SSH_ERR_ALLOC_FAIL;
3315		goto out;
3316	}
3317
3318	/* log2(private) > log2(order)/2 */
3319	if (EC_GROUP_get_order(EC_KEY_get0_group(key), order, bnctx) != 1) {
3320		ret = SSH_ERR_LIBCRYPTO_ERROR;
3321		goto out;
3322	}
3323	if (BN_num_bits(EC_KEY_get0_private_key(key)) <=
3324	    BN_num_bits(order) / 2)
3325		goto out;
3326
3327	/* private < order - 1 */
3328	if (!BN_sub(tmp, order, BN_value_one())) {
3329		ret = SSH_ERR_LIBCRYPTO_ERROR;
3330		goto out;
3331	}
3332	if (BN_cmp(EC_KEY_get0_private_key(key), tmp) >= 0)
3333		goto out;
3334	ret = 0;
3335 out:
3336	BN_CTX_free(bnctx);
3337	return ret;
3338}
3339
3340void
3341sshkey_dump_ec_point(const EC_GROUP *group, const EC_POINT *point)
3342{
3343	BIGNUM *x, *y;
3344	BN_CTX *bnctx;
3345
3346	if (point == NULL) {
3347		fputs("point=(NULL)\n", stderr);
3348		return;
3349	}
3350	if ((bnctx = BN_CTX_new()) == NULL) {
3351		fprintf(stderr, "%s: BN_CTX_new failed\n", __func__);
3352		return;
3353	}
3354	BN_CTX_start(bnctx);
3355	if ((x = BN_CTX_get(bnctx)) == NULL ||
3356	    (y = BN_CTX_get(bnctx)) == NULL) {
3357		fprintf(stderr, "%s: BN_CTX_get failed\n", __func__);
3358		return;
3359	}
3360	if (EC_METHOD_get_field_type(EC_GROUP_method_of(group)) !=
3361	    NID_X9_62_prime_field) {
3362		fprintf(stderr, "%s: group is not a prime field\n", __func__);
3363		return;
3364	}
3365	if (EC_POINT_get_affine_coordinates_GFp(group, point, x, y,
3366	    bnctx) != 1) {
3367		fprintf(stderr, "%s: EC_POINT_get_affine_coordinates_GFp\n",
3368		    __func__);
3369		return;
3370	}
3371	fputs("x=", stderr);
3372	BN_print_fp(stderr, x);
3373	fputs("\ny=", stderr);
3374	BN_print_fp(stderr, y);
3375	fputs("\n", stderr);
3376	BN_CTX_free(bnctx);
3377}
3378
3379void
3380sshkey_dump_ec_key(const EC_KEY *key)
3381{
3382	const BIGNUM *exponent;
3383
3384	sshkey_dump_ec_point(EC_KEY_get0_group(key),
3385	    EC_KEY_get0_public_key(key));
3386	fputs("exponent=", stderr);
3387	if ((exponent = EC_KEY_get0_private_key(key)) == NULL)
3388		fputs("(NULL)", stderr);
3389	else
3390		BN_print_fp(stderr, EC_KEY_get0_private_key(key));
3391	fputs("\n", stderr);
3392}
3393#endif /* WITH_OPENSSL && OPENSSL_HAS_ECC */
3394
3395static int
3396sshkey_private_to_blob2(const struct sshkey *prv, struct sshbuf *blob,
3397    const char *passphrase, const char *comment, const char *ciphername,
3398    int rounds)
3399{
3400	u_char *cp, *key = NULL, *pubkeyblob = NULL;
3401	u_char salt[SALT_LEN];
3402	char *b64 = NULL;
3403	size_t i, pubkeylen, keylen, ivlen, blocksize, authlen;
3404	u_int check;
3405	int r = SSH_ERR_INTERNAL_ERROR;
3406	struct sshcipher_ctx *ciphercontext = NULL;
3407	const struct sshcipher *cipher;
3408	const char *kdfname = KDFNAME;
3409	struct sshbuf *encoded = NULL, *encrypted = NULL, *kdf = NULL;
3410
3411	if (rounds <= 0)
3412		rounds = DEFAULT_ROUNDS;
3413	if (passphrase == NULL || !strlen(passphrase)) {
3414		ciphername = "none";
3415		kdfname = "none";
3416	} else if (ciphername == NULL)
3417		ciphername = DEFAULT_CIPHERNAME;
3418	if ((cipher = cipher_by_name(ciphername)) == NULL) {
3419		r = SSH_ERR_INVALID_ARGUMENT;
3420		goto out;
3421	}
3422
3423	if ((kdf = sshbuf_new()) == NULL ||
3424	    (encoded = sshbuf_new()) == NULL ||
3425	    (encrypted = sshbuf_new()) == NULL) {
3426		r = SSH_ERR_ALLOC_FAIL;
3427		goto out;
3428	}
3429	blocksize = cipher_blocksize(cipher);
3430	keylen = cipher_keylen(cipher);
3431	ivlen = cipher_ivlen(cipher);
3432	authlen = cipher_authlen(cipher);
3433	if ((key = calloc(1, keylen + ivlen)) == NULL) {
3434		r = SSH_ERR_ALLOC_FAIL;
3435		goto out;
3436	}
3437	if (strcmp(kdfname, "bcrypt") == 0) {
3438		arc4random_buf(salt, SALT_LEN);
3439		if (bcrypt_pbkdf(passphrase, strlen(passphrase),
3440		    salt, SALT_LEN, key, keylen + ivlen, rounds) < 0) {
3441			r = SSH_ERR_INVALID_ARGUMENT;
3442			goto out;
3443		}
3444		if ((r = sshbuf_put_string(kdf, salt, SALT_LEN)) != 0 ||
3445		    (r = sshbuf_put_u32(kdf, rounds)) != 0)
3446			goto out;
3447	} else if (strcmp(kdfname, "none") != 0) {
3448		/* Unsupported KDF type */
3449		r = SSH_ERR_KEY_UNKNOWN_CIPHER;
3450		goto out;
3451	}
3452	if ((r = cipher_init(&ciphercontext, cipher, key, keylen,
3453	    key + keylen, ivlen, 1)) != 0)
3454		goto out;
3455
3456	if ((r = sshbuf_put(encoded, AUTH_MAGIC, sizeof(AUTH_MAGIC))) != 0 ||
3457	    (r = sshbuf_put_cstring(encoded, ciphername)) != 0 ||
3458	    (r = sshbuf_put_cstring(encoded, kdfname)) != 0 ||
3459	    (r = sshbuf_put_stringb(encoded, kdf)) != 0 ||
3460	    (r = sshbuf_put_u32(encoded, 1)) != 0 ||	/* number of keys */
3461	    (r = sshkey_to_blob(prv, &pubkeyblob, &pubkeylen)) != 0 ||
3462	    (r = sshbuf_put_string(encoded, pubkeyblob, pubkeylen)) != 0)
3463		goto out;
3464
3465	/* set up the buffer that will be encrypted */
3466
3467	/* Random check bytes */
3468	check = arc4random();
3469	if ((r = sshbuf_put_u32(encrypted, check)) != 0 ||
3470	    (r = sshbuf_put_u32(encrypted, check)) != 0)
3471		goto out;
3472
3473	/* append private key and comment*/
3474	if ((r = sshkey_private_serialize_opt(prv, encrypted,
3475	     SSHKEY_SERIALIZE_FULL)) != 0 ||
3476	    (r = sshbuf_put_cstring(encrypted, comment)) != 0)
3477		goto out;
3478
3479	/* padding */
3480	i = 0;
3481	while (sshbuf_len(encrypted) % blocksize) {
3482		if ((r = sshbuf_put_u8(encrypted, ++i & 0xff)) != 0)
3483			goto out;
3484	}
3485
3486	/* length in destination buffer */
3487	if ((r = sshbuf_put_u32(encoded, sshbuf_len(encrypted))) != 0)
3488		goto out;
3489
3490	/* encrypt */
3491	if ((r = sshbuf_reserve(encoded,
3492	    sshbuf_len(encrypted) + authlen, &cp)) != 0)
3493		goto out;
3494	if ((r = cipher_crypt(ciphercontext, 0, cp,
3495	    sshbuf_ptr(encrypted), sshbuf_len(encrypted), 0, authlen)) != 0)
3496		goto out;
3497
3498	/* uuencode */
3499	if ((b64 = sshbuf_dtob64(encoded)) == NULL) {
3500		r = SSH_ERR_ALLOC_FAIL;
3501		goto out;
3502	}
3503
3504	sshbuf_reset(blob);
3505	if ((r = sshbuf_put(blob, MARK_BEGIN, MARK_BEGIN_LEN)) != 0)
3506		goto out;
3507	for (i = 0; i < strlen(b64); i++) {
3508		if ((r = sshbuf_put_u8(blob, b64[i])) != 0)
3509			goto out;
3510		/* insert line breaks */
3511		if (i % 70 == 69 && (r = sshbuf_put_u8(blob, '\n')) != 0)
3512			goto out;
3513	}
3514	if (i % 70 != 69 && (r = sshbuf_put_u8(blob, '\n')) != 0)
3515		goto out;
3516	if ((r = sshbuf_put(blob, MARK_END, MARK_END_LEN)) != 0)
3517		goto out;
3518
3519	/* success */
3520	r = 0;
3521
3522 out:
3523	sshbuf_free(kdf);
3524	sshbuf_free(encoded);
3525	sshbuf_free(encrypted);
3526	cipher_free(ciphercontext);
3527	explicit_bzero(salt, sizeof(salt));
3528	if (key != NULL) {
3529		explicit_bzero(key, keylen + ivlen);
3530		free(key);
3531	}
3532	if (pubkeyblob != NULL) {
3533		explicit_bzero(pubkeyblob, pubkeylen);
3534		free(pubkeyblob);
3535	}
3536	if (b64 != NULL) {
3537		explicit_bzero(b64, strlen(b64));
3538		free(b64);
3539	}
3540	return r;
3541}
3542
3543static int
3544sshkey_parse_private2(struct sshbuf *blob, int type, const char *passphrase,
3545    struct sshkey **keyp, char **commentp)
3546{
3547	char *comment = NULL, *ciphername = NULL, *kdfname = NULL;
3548	const struct sshcipher *cipher = NULL;
3549	const u_char *cp;
3550	int r = SSH_ERR_INTERNAL_ERROR;
3551	size_t encoded_len;
3552	size_t i, keylen = 0, ivlen = 0, authlen = 0, slen = 0;
3553	struct sshbuf *encoded = NULL, *decoded = NULL;
3554	struct sshbuf *kdf = NULL, *decrypted = NULL;
3555	struct sshcipher_ctx *ciphercontext = NULL;
3556	struct sshkey *k = NULL;
3557	u_char *key = NULL, *salt = NULL, *dp, pad, last;
3558	u_int blocksize, rounds, nkeys, encrypted_len, check1, check2;
3559
3560	if (keyp != NULL)
3561		*keyp = NULL;
3562	if (commentp != NULL)
3563		*commentp = NULL;
3564
3565	if ((encoded = sshbuf_new()) == NULL ||
3566	    (decoded = sshbuf_new()) == NULL ||
3567	    (decrypted = sshbuf_new()) == NULL) {
3568		r = SSH_ERR_ALLOC_FAIL;
3569		goto out;
3570	}
3571
3572	/* check preamble */
3573	cp = sshbuf_ptr(blob);
3574	encoded_len = sshbuf_len(blob);
3575	if (encoded_len < (MARK_BEGIN_LEN + MARK_END_LEN) ||
3576	    memcmp(cp, MARK_BEGIN, MARK_BEGIN_LEN) != 0) {
3577		r = SSH_ERR_INVALID_FORMAT;
3578		goto out;
3579	}
3580	cp += MARK_BEGIN_LEN;
3581	encoded_len -= MARK_BEGIN_LEN;
3582
3583	/* Look for end marker, removing whitespace as we go */
3584	while (encoded_len > 0) {
3585		if (*cp != '\n' && *cp != '\r') {
3586			if ((r = sshbuf_put_u8(encoded, *cp)) != 0)
3587				goto out;
3588		}
3589		last = *cp;
3590		encoded_len--;
3591		cp++;
3592		if (last == '\n') {
3593			if (encoded_len >= MARK_END_LEN &&
3594			    memcmp(cp, MARK_END, MARK_END_LEN) == 0) {
3595				/* \0 terminate */
3596				if ((r = sshbuf_put_u8(encoded, 0)) != 0)
3597					goto out;
3598				break;
3599			}
3600		}
3601	}
3602	if (encoded_len == 0) {
3603		r = SSH_ERR_INVALID_FORMAT;
3604		goto out;
3605	}
3606
3607	/* decode base64 */
3608	if ((r = sshbuf_b64tod(decoded, (char *)sshbuf_ptr(encoded))) != 0)
3609		goto out;
3610
3611	/* check magic */
3612	if (sshbuf_len(decoded) < sizeof(AUTH_MAGIC) ||
3613	    memcmp(sshbuf_ptr(decoded), AUTH_MAGIC, sizeof(AUTH_MAGIC))) {
3614		r = SSH_ERR_INVALID_FORMAT;
3615		goto out;
3616	}
3617	/* parse public portion of key */
3618	if ((r = sshbuf_consume(decoded, sizeof(AUTH_MAGIC))) != 0 ||
3619	    (r = sshbuf_get_cstring(decoded, &ciphername, NULL)) != 0 ||
3620	    (r = sshbuf_get_cstring(decoded, &kdfname, NULL)) != 0 ||
3621	    (r = sshbuf_froms(decoded, &kdf)) != 0 ||
3622	    (r = sshbuf_get_u32(decoded, &nkeys)) != 0 ||
3623	    (r = sshbuf_skip_string(decoded)) != 0 || /* pubkey */
3624	    (r = sshbuf_get_u32(decoded, &encrypted_len)) != 0)
3625		goto out;
3626
3627	if ((cipher = cipher_by_name(ciphername)) == NULL) {
3628		r = SSH_ERR_KEY_UNKNOWN_CIPHER;
3629		goto out;
3630	}
3631	if ((passphrase == NULL || strlen(passphrase) == 0) &&
3632	    strcmp(ciphername, "none") != 0) {
3633		/* passphrase required */
3634		r = SSH_ERR_KEY_WRONG_PASSPHRASE;
3635		goto out;
3636	}
3637	if (strcmp(kdfname, "none") != 0 && strcmp(kdfname, "bcrypt") != 0) {
3638		r = SSH_ERR_KEY_UNKNOWN_CIPHER;
3639		goto out;
3640	}
3641	if (!strcmp(kdfname, "none") && strcmp(ciphername, "none") != 0) {
3642		r = SSH_ERR_INVALID_FORMAT;
3643		goto out;
3644	}
3645	if (nkeys != 1) {
3646		/* XXX only one key supported */
3647		r = SSH_ERR_INVALID_FORMAT;
3648		goto out;
3649	}
3650
3651	/* check size of encrypted key blob */
3652	blocksize = cipher_blocksize(cipher);
3653	if (encrypted_len < blocksize || (encrypted_len % blocksize) != 0) {
3654		r = SSH_ERR_INVALID_FORMAT;
3655		goto out;
3656	}
3657
3658	/* setup key */
3659	keylen = cipher_keylen(cipher);
3660	ivlen = cipher_ivlen(cipher);
3661	authlen = cipher_authlen(cipher);
3662	if ((key = calloc(1, keylen + ivlen)) == NULL) {
3663		r = SSH_ERR_ALLOC_FAIL;
3664		goto out;
3665	}
3666	if (strcmp(kdfname, "bcrypt") == 0) {
3667		if ((r = sshbuf_get_string(kdf, &salt, &slen)) != 0 ||
3668		    (r = sshbuf_get_u32(kdf, &rounds)) != 0)
3669			goto out;
3670		if (bcrypt_pbkdf(passphrase, strlen(passphrase), salt, slen,
3671		    key, keylen + ivlen, rounds) < 0) {
3672			r = SSH_ERR_INVALID_FORMAT;
3673			goto out;
3674		}
3675	}
3676
3677	/* check that an appropriate amount of auth data is present */
3678	if (sshbuf_len(decoded) < encrypted_len + authlen) {
3679		r = SSH_ERR_INVALID_FORMAT;
3680		goto out;
3681	}
3682
3683	/* decrypt private portion of key */
3684	if ((r = sshbuf_reserve(decrypted, encrypted_len, &dp)) != 0 ||
3685	    (r = cipher_init(&ciphercontext, cipher, key, keylen,
3686	    key + keylen, ivlen, 0)) != 0)
3687		goto out;
3688	if ((r = cipher_crypt(ciphercontext, 0, dp, sshbuf_ptr(decoded),
3689	    encrypted_len, 0, authlen)) != 0) {
3690		/* an integrity error here indicates an incorrect passphrase */
3691		if (r == SSH_ERR_MAC_INVALID)
3692			r = SSH_ERR_KEY_WRONG_PASSPHRASE;
3693		goto out;
3694	}
3695	if ((r = sshbuf_consume(decoded, encrypted_len + authlen)) != 0)
3696		goto out;
3697	/* there should be no trailing data */
3698	if (sshbuf_len(decoded) != 0) {
3699		r = SSH_ERR_INVALID_FORMAT;
3700		goto out;
3701	}
3702
3703	/* check check bytes */
3704	if ((r = sshbuf_get_u32(decrypted, &check1)) != 0 ||
3705	    (r = sshbuf_get_u32(decrypted, &check2)) != 0)
3706		goto out;
3707	if (check1 != check2) {
3708		r = SSH_ERR_KEY_WRONG_PASSPHRASE;
3709		goto out;
3710	}
3711
3712	/* Load the private key and comment */
3713	if ((r = sshkey_private_deserialize(decrypted, &k)) != 0 ||
3714	    (r = sshbuf_get_cstring(decrypted, &comment, NULL)) != 0)
3715		goto out;
3716
3717	/* Check deterministic padding */
3718	i = 0;
3719	while (sshbuf_len(decrypted)) {
3720		if ((r = sshbuf_get_u8(decrypted, &pad)) != 0)
3721			goto out;
3722		if (pad != (++i & 0xff)) {
3723			r = SSH_ERR_INVALID_FORMAT;
3724			goto out;
3725		}
3726	}
3727
3728	/* XXX decode pubkey and check against private */
3729
3730	/* success */
3731	r = 0;
3732	if (keyp != NULL) {
3733		*keyp = k;
3734		k = NULL;
3735	}
3736	if (commentp != NULL) {
3737		*commentp = comment;
3738		comment = NULL;
3739	}
3740 out:
3741	pad = 0;
3742	cipher_free(ciphercontext);
3743	free(ciphername);
3744	free(kdfname);
3745	free(comment);
3746	if (salt != NULL) {
3747		explicit_bzero(salt, slen);
3748		free(salt);
3749	}
3750	if (key != NULL) {
3751		explicit_bzero(key, keylen + ivlen);
3752		free(key);
3753	}
3754	sshbuf_free(encoded);
3755	sshbuf_free(decoded);
3756	sshbuf_free(kdf);
3757	sshbuf_free(decrypted);
3758	sshkey_free(k);
3759	return r;
3760}
3761
3762
3763#ifdef WITH_OPENSSL
3764/* convert SSH v2 key in OpenSSL PEM format */
3765static int
3766sshkey_private_pem_to_blob(struct sshkey *key, struct sshbuf *blob,
3767    const char *_passphrase, const char *comment)
3768{
3769	int success, r;
3770	int blen, len = strlen(_passphrase);
3771	u_char *passphrase = (len > 0) ? (u_char *)_passphrase : NULL;
3772	const EVP_CIPHER *cipher = (len > 0) ? EVP_aes_128_cbc() : NULL;
3773	char *bptr;
3774	BIO *bio = NULL;
3775
3776	if (len > 0 && len <= 4)
3777		return SSH_ERR_PASSPHRASE_TOO_SHORT;
3778	if ((bio = BIO_new(BIO_s_mem())) == NULL)
3779		return SSH_ERR_ALLOC_FAIL;
3780
3781	switch (key->type) {
3782	case KEY_DSA:
3783		success = PEM_write_bio_DSAPrivateKey(bio, key->dsa,
3784		    cipher, passphrase, len, NULL, NULL);
3785		break;
3786#ifdef OPENSSL_HAS_ECC
3787	case KEY_ECDSA:
3788		success = PEM_write_bio_ECPrivateKey(bio, key->ecdsa,
3789		    cipher, passphrase, len, NULL, NULL);
3790		break;
3791#endif
3792	case KEY_RSA:
3793		success = PEM_write_bio_RSAPrivateKey(bio, key->rsa,
3794		    cipher, passphrase, len, NULL, NULL);
3795		break;
3796	default:
3797		success = 0;
3798		break;
3799	}
3800	if (success == 0) {
3801		r = SSH_ERR_LIBCRYPTO_ERROR;
3802		goto out;
3803	}
3804	if ((blen = BIO_get_mem_data(bio, &bptr)) <= 0) {
3805		r = SSH_ERR_INTERNAL_ERROR;
3806		goto out;
3807	}
3808	if ((r = sshbuf_put(blob, bptr, blen)) != 0)
3809		goto out;
3810	r = 0;
3811 out:
3812	BIO_free(bio);
3813	return r;
3814}
3815#endif /* WITH_OPENSSL */
3816
3817/* Serialise "key" to buffer "blob" */
3818int
3819sshkey_private_to_fileblob(struct sshkey *key, struct sshbuf *blob,
3820    const char *passphrase, const char *comment,
3821    int force_new_format, const char *new_format_cipher, int new_format_rounds)
3822{
3823	switch (key->type) {
3824#ifdef WITH_OPENSSL
3825	case KEY_DSA:
3826	case KEY_ECDSA:
3827	case KEY_RSA:
3828		if (force_new_format) {
3829			return sshkey_private_to_blob2(key, blob, passphrase,
3830			    comment, new_format_cipher, new_format_rounds);
3831		}
3832		return sshkey_private_pem_to_blob(key, blob,
3833		    passphrase, comment);
3834#endif /* WITH_OPENSSL */
3835	case KEY_ED25519:
3836#ifdef WITH_XMSS
3837	case KEY_XMSS:
3838#endif /* WITH_XMSS */
3839		return sshkey_private_to_blob2(key, blob, passphrase,
3840		    comment, new_format_cipher, new_format_rounds);
3841	default:
3842		return SSH_ERR_KEY_TYPE_UNKNOWN;
3843	}
3844}
3845
3846
3847#ifdef WITH_OPENSSL
3848static int
3849translate_libcrypto_error(unsigned long pem_err)
3850{
3851	int pem_reason = ERR_GET_REASON(pem_err);
3852
3853	switch (ERR_GET_LIB(pem_err)) {
3854	case ERR_LIB_PEM:
3855		switch (pem_reason) {
3856		case PEM_R_BAD_PASSWORD_READ:
3857		case PEM_R_PROBLEMS_GETTING_PASSWORD:
3858		case PEM_R_BAD_DECRYPT:
3859			return SSH_ERR_KEY_WRONG_PASSPHRASE;
3860		default:
3861			return SSH_ERR_INVALID_FORMAT;
3862		}
3863	case ERR_LIB_EVP:
3864		switch (pem_reason) {
3865		case EVP_R_BAD_DECRYPT:
3866			return SSH_ERR_KEY_WRONG_PASSPHRASE;
3867#ifdef EVP_R_BN_DECODE_ERROR
3868		case EVP_R_BN_DECODE_ERROR:
3869#endif
3870		case EVP_R_DECODE_ERROR:
3871#ifdef EVP_R_PRIVATE_KEY_DECODE_ERROR
3872		case EVP_R_PRIVATE_KEY_DECODE_ERROR:
3873#endif
3874			return SSH_ERR_INVALID_FORMAT;
3875		default:
3876			return SSH_ERR_LIBCRYPTO_ERROR;
3877		}
3878	case ERR_LIB_ASN1:
3879		return SSH_ERR_INVALID_FORMAT;
3880	}
3881	return SSH_ERR_LIBCRYPTO_ERROR;
3882}
3883
3884static void
3885clear_libcrypto_errors(void)
3886{
3887	while (ERR_get_error() != 0)
3888		;
3889}
3890
3891/*
3892 * Translate OpenSSL error codes to determine whether
3893 * passphrase is required/incorrect.
3894 */
3895static int
3896convert_libcrypto_error(void)
3897{
3898	/*
3899	 * Some password errors are reported at the beginning
3900	 * of the error queue.
3901	 */
3902	if (translate_libcrypto_error(ERR_peek_error()) ==
3903	    SSH_ERR_KEY_WRONG_PASSPHRASE)
3904		return SSH_ERR_KEY_WRONG_PASSPHRASE;
3905	return translate_libcrypto_error(ERR_peek_last_error());
3906}
3907
3908static int
3909pem_passphrase_cb(char *buf, int size, int rwflag, void *u)
3910{
3911	char *p = (char *)u;
3912	size_t len;
3913
3914	if (p == NULL || (len = strlen(p)) == 0)
3915		return -1;
3916	if (size < 0 || len > (size_t)size)
3917		return -1;
3918	memcpy(buf, p, len);
3919	return (int)len;
3920}
3921
3922static int
3923sshkey_parse_private_pem_fileblob(struct sshbuf *blob, int type,
3924    const char *passphrase, struct sshkey **keyp)
3925{
3926	EVP_PKEY *pk = NULL;
3927	struct sshkey *prv = NULL;
3928	BIO *bio = NULL;
3929	int r;
3930
3931	if (keyp != NULL)
3932		*keyp = NULL;
3933
3934	if ((bio = BIO_new(BIO_s_mem())) == NULL || sshbuf_len(blob) > INT_MAX)
3935		return SSH_ERR_ALLOC_FAIL;
3936	if (BIO_write(bio, sshbuf_ptr(blob), sshbuf_len(blob)) !=
3937	    (int)sshbuf_len(blob)) {
3938		r = SSH_ERR_ALLOC_FAIL;
3939		goto out;
3940	}
3941
3942	clear_libcrypto_errors();
3943	if ((pk = PEM_read_bio_PrivateKey(bio, NULL, pem_passphrase_cb,
3944	    (char *)passphrase)) == NULL) {
3945	       /*
3946		* libcrypto may return various ASN.1 errors when attempting
3947		* to parse a key with an incorrect passphrase.
3948		* Treat all format errors as "incorrect passphrase" if a
3949		* passphrase was supplied.
3950		*/
3951		if (passphrase != NULL && *passphrase != '\0')
3952			r = SSH_ERR_KEY_WRONG_PASSPHRASE;
3953		else
3954			r = convert_libcrypto_error();
3955		goto out;
3956	}
3957	if (EVP_PKEY_base_id(pk) == EVP_PKEY_RSA &&
3958	    (type == KEY_UNSPEC || type == KEY_RSA)) {
3959		if ((prv = sshkey_new(KEY_UNSPEC)) == NULL) {
3960			r = SSH_ERR_ALLOC_FAIL;
3961			goto out;
3962		}
3963		prv->rsa = EVP_PKEY_get1_RSA(pk);
3964		prv->type = KEY_RSA;
3965#ifdef DEBUG_PK
3966		RSA_print_fp(stderr, prv->rsa, 8);
3967#endif
3968		if (RSA_blinding_on(prv->rsa, NULL) != 1) {
3969			r = SSH_ERR_LIBCRYPTO_ERROR;
3970			goto out;
3971		}
3972		if ((r = check_rsa_length(prv->rsa)) != 0)
3973			goto out;
3974	} else if (EVP_PKEY_base_id(pk) == EVP_PKEY_DSA &&
3975	    (type == KEY_UNSPEC || type == KEY_DSA)) {
3976		if ((prv = sshkey_new(KEY_UNSPEC)) == NULL) {
3977			r = SSH_ERR_ALLOC_FAIL;
3978			goto out;
3979		}
3980		prv->dsa = EVP_PKEY_get1_DSA(pk);
3981		prv->type = KEY_DSA;
3982#ifdef DEBUG_PK
3983		DSA_print_fp(stderr, prv->dsa, 8);
3984#endif
3985#ifdef OPENSSL_HAS_ECC
3986	} else if (EVP_PKEY_base_id(pk) == EVP_PKEY_EC &&
3987	    (type == KEY_UNSPEC || type == KEY_ECDSA)) {
3988		if ((prv = sshkey_new(KEY_UNSPEC)) == NULL) {
3989			r = SSH_ERR_ALLOC_FAIL;
3990			goto out;
3991		}
3992		prv->ecdsa = EVP_PKEY_get1_EC_KEY(pk);
3993		prv->type = KEY_ECDSA;
3994		prv->ecdsa_nid = sshkey_ecdsa_key_to_nid(prv->ecdsa);
3995		if (prv->ecdsa_nid == -1 ||
3996		    sshkey_curve_nid_to_name(prv->ecdsa_nid) == NULL ||
3997		    sshkey_ec_validate_public(EC_KEY_get0_group(prv->ecdsa),
3998		    EC_KEY_get0_public_key(prv->ecdsa)) != 0 ||
3999		    sshkey_ec_validate_private(prv->ecdsa) != 0) {
4000			r = SSH_ERR_INVALID_FORMAT;
4001			goto out;
4002		}
4003# ifdef DEBUG_PK
4004		if (prv != NULL && prv->ecdsa != NULL)
4005			sshkey_dump_ec_key(prv->ecdsa);
4006# endif
4007#endif /* OPENSSL_HAS_ECC */
4008	} else {
4009		r = SSH_ERR_INVALID_FORMAT;
4010		goto out;
4011	}
4012	r = 0;
4013	if (keyp != NULL) {
4014		*keyp = prv;
4015		prv = NULL;
4016	}
4017 out:
4018	BIO_free(bio);
4019	EVP_PKEY_free(pk);
4020	sshkey_free(prv);
4021	return r;
4022}
4023#endif /* WITH_OPENSSL */
4024
4025int
4026sshkey_parse_private_fileblob_type(struct sshbuf *blob, int type,
4027    const char *passphrase, struct sshkey **keyp, char **commentp)
4028{
4029	int r = SSH_ERR_INTERNAL_ERROR;
4030
4031	if (keyp != NULL)
4032		*keyp = NULL;
4033	if (commentp != NULL)
4034		*commentp = NULL;
4035
4036	switch (type) {
4037#ifdef WITH_OPENSSL
4038	case KEY_DSA:
4039	case KEY_ECDSA:
4040	case KEY_RSA:
4041		return sshkey_parse_private_pem_fileblob(blob, type,
4042		    passphrase, keyp);
4043#endif /* WITH_OPENSSL */
4044	case KEY_ED25519:
4045#ifdef WITH_XMSS
4046	case KEY_XMSS:
4047#endif /* WITH_XMSS */
4048		return sshkey_parse_private2(blob, type, passphrase,
4049		    keyp, commentp);
4050	case KEY_UNSPEC:
4051		r = sshkey_parse_private2(blob, type, passphrase, keyp,
4052		    commentp);
4053		/* Do not fallback to PEM parser if only passphrase is wrong. */
4054		if (r == 0 || r == SSH_ERR_KEY_WRONG_PASSPHRASE)
4055			return r;
4056#ifdef WITH_OPENSSL
4057		return sshkey_parse_private_pem_fileblob(blob, type,
4058		    passphrase, keyp);
4059#else
4060		return SSH_ERR_INVALID_FORMAT;
4061#endif /* WITH_OPENSSL */
4062	default:
4063		return SSH_ERR_KEY_TYPE_UNKNOWN;
4064	}
4065}
4066
4067int
4068sshkey_parse_private_fileblob(struct sshbuf *buffer, const char *passphrase,
4069    struct sshkey **keyp, char **commentp)
4070{
4071	if (keyp != NULL)
4072		*keyp = NULL;
4073	if (commentp != NULL)
4074		*commentp = NULL;
4075
4076	return sshkey_parse_private_fileblob_type(buffer, KEY_UNSPEC,
4077	    passphrase, keyp, commentp);
4078}
4079
4080#ifdef WITH_XMSS
4081/*
4082 * serialize the key with the current state and forward the state
4083 * maxsign times.
4084 */
4085int
4086sshkey_private_serialize_maxsign(const struct sshkey *k, struct sshbuf *b,
4087    u_int32_t maxsign, sshkey_printfn *pr)
4088{
4089	int r, rupdate;
4090
4091	if (maxsign == 0 ||
4092	    sshkey_type_plain(k->type) != KEY_XMSS)
4093		return sshkey_private_serialize_opt(k, b,
4094		    SSHKEY_SERIALIZE_DEFAULT);
4095	if ((r = sshkey_xmss_get_state(k, pr)) != 0 ||
4096	    (r = sshkey_private_serialize_opt(k, b,
4097	    SSHKEY_SERIALIZE_STATE)) != 0 ||
4098	    (r = sshkey_xmss_forward_state(k, maxsign)) != 0)
4099		goto out;
4100	r = 0;
4101out:
4102	if ((rupdate = sshkey_xmss_update_state(k, pr)) != 0) {
4103		if (r == 0)
4104			r = rupdate;
4105	}
4106	return r;
4107}
4108
4109u_int32_t
4110sshkey_signatures_left(const struct sshkey *k)
4111{
4112	if (sshkey_type_plain(k->type) == KEY_XMSS)
4113		return sshkey_xmss_signatures_left(k);
4114	return 0;
4115}
4116
4117int
4118sshkey_enable_maxsign(struct sshkey *k, u_int32_t maxsign)
4119{
4120	if (sshkey_type_plain(k->type) != KEY_XMSS)
4121		return SSH_ERR_INVALID_ARGUMENT;
4122	return sshkey_xmss_enable_maxsign(k, maxsign);
4123}
4124
4125int
4126sshkey_set_filename(struct sshkey *k, const char *filename)
4127{
4128	if (k == NULL)
4129		return SSH_ERR_INVALID_ARGUMENT;
4130	if (sshkey_type_plain(k->type) != KEY_XMSS)
4131		return 0;
4132	if (filename == NULL)
4133		return SSH_ERR_INVALID_ARGUMENT;
4134	if ((k->xmss_filename = strdup(filename)) == NULL)
4135		return SSH_ERR_ALLOC_FAIL;
4136	return 0;
4137}
4138#else
4139int
4140sshkey_private_serialize_maxsign(const struct sshkey *k, struct sshbuf *b,
4141    u_int32_t maxsign, sshkey_printfn *pr)
4142{
4143	return sshkey_private_serialize_opt(k, b, SSHKEY_SERIALIZE_DEFAULT);
4144}
4145
4146u_int32_t
4147sshkey_signatures_left(const struct sshkey *k)
4148{
4149	return 0;
4150}
4151
4152int
4153sshkey_enable_maxsign(struct sshkey *k, u_int32_t maxsign)
4154{
4155	return SSH_ERR_INVALID_ARGUMENT;
4156}
4157
4158int
4159sshkey_set_filename(struct sshkey *k, const char *filename)
4160{
4161	if (k == NULL)
4162		return SSH_ERR_INVALID_ARGUMENT;
4163	return 0;
4164}
4165#endif /* WITH_XMSS */
4166