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