[openssh.git] / key.c

/*
 * read_bignum():
 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
 *
 * As far as I am concerned, the code I have written for this software
 * can be used freely for any purpose.  Any derived versions of this
 * software must be clearly marked as such, and if the derived work is
 * incompatible with the protocol description in the RFC file, it must be
 * called by a name other than "ssh" or "Secure Shell".
 *
 *
 * Copyright (c) 2000 Markus Friedl.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "includes.h"
#include "ssh.h"
#include <openssl/rsa.h>
#include <openssl/dsa.h>
#include <openssl/evp.h>
#include "xmalloc.h"
#include "key.h"
#include "rsa.h"
#include "ssh-dss.h"
#include "ssh-rsa.h"
#include "uuencode.h"
#include "buffer.h"
#include "bufaux.h"

RCSID("$OpenBSD: key.c,v 1.14 2001/01/16 19:20:06 markus Exp $");

Key *
key_new(int type)
{
	Key *k;
	RSA *rsa;
	DSA *dsa;
	k = xmalloc(sizeof(*k));
	k->type = type;
	k->dsa = NULL;
	k->rsa = NULL;
	switch (k->type) {
	case KEY_RSA1:
	case KEY_RSA:
		rsa = RSA_new();
		rsa->n = BN_new();
		rsa->e = BN_new();
		k->rsa = rsa;
		break;
	case KEY_DSA:
		dsa = DSA_new();
		dsa->p = BN_new();
		dsa->q = BN_new();
		dsa->g = BN_new();
		dsa->pub_key = BN_new();
		k->dsa = dsa;
		break;
	case KEY_UNSPEC:
		break;
	default:
		fatal("key_new: bad key type %d", k->type);
		break;
	}
	return k;
}
Key *
key_new_private(int type)
{
	Key *k = key_new(type);
	switch (k->type) {
	case KEY_RSA1:
	case KEY_RSA:
		k->rsa->d = BN_new();
		k->rsa->iqmp = BN_new();
		k->rsa->q = BN_new();
		k->rsa->p = BN_new();
		k->rsa->dmq1 = BN_new();
		k->rsa->dmp1 = BN_new();
		break;
	case KEY_DSA:
		k->dsa->priv_key = BN_new();
		break;
	case KEY_UNSPEC:
		break;
	default:
		break;
	}
	return k;
}
void
key_free(Key *k)
{
	switch (k->type) {
	case KEY_RSA1:
	case KEY_RSA:
		if (k->rsa != NULL)
			RSA_free(k->rsa);
		k->rsa = NULL;
		break;
	case KEY_DSA:
		if (k->dsa != NULL)
			DSA_free(k->dsa);
		k->dsa = NULL;
		break;
	case KEY_UNSPEC:
		break;
	default:
		fatal("key_free: bad key type %d", k->type);
		break;
	}
	xfree(k);
}
int
key_equal(Key *a, Key *b)
{
	if (a == NULL || b == NULL || a->type != b->type)
		return 0;
	switch (a->type) {
	case KEY_RSA1:
	case KEY_RSA:
		return a->rsa != NULL && b->rsa != NULL &&
		    BN_cmp(a->rsa->e, b->rsa->e) == 0 &&
		    BN_cmp(a->rsa->n, b->rsa->n) == 0;
		break;
	case KEY_DSA:
		return a->dsa != NULL && b->dsa != NULL &&
		    BN_cmp(a->dsa->p, b->dsa->p) == 0 &&
		    BN_cmp(a->dsa->q, b->dsa->q) == 0 &&
		    BN_cmp(a->dsa->g, b->dsa->g) == 0 &&
		    BN_cmp(a->dsa->pub_key, b->dsa->pub_key) == 0;
		break;
	default:
		fatal("key_equal: bad key type %d", a->type);
		break;
	}
	return 0;
}

/*
 * Generate key fingerprint in ascii format.
 * Based on ideas and code from Bjoern Groenvall <bg@sics.se>
 */
char *
key_fingerprint(Key *k)
{
	static char retval[(EVP_MAX_MD_SIZE+1)*3];
	u_char *blob = NULL;
	int len = 0;
	int nlen, elen;

	retval[0] = '\0';
	switch (k->type) {
	case KEY_RSA1:
		nlen = BN_num_bytes(k->rsa->n);
		elen = BN_num_bytes(k->rsa->e);
		len = nlen + elen;
		blob = xmalloc(len);
		BN_bn2bin(k->rsa->n, blob);
		BN_bn2bin(k->rsa->e, blob + nlen);
		break;
	case KEY_DSA:
	case KEY_RSA:
		key_to_blob(k, &blob, &len);
		break;
	case KEY_UNSPEC:
		return retval;
		break;
	default:
		fatal("key_fingerprint: bad key type %d", k->type);
		break;
	}
	if (blob != NULL) {
		int i;
		u_char digest[EVP_MAX_MD_SIZE];
		EVP_MD *md = EVP_md5();
		EVP_MD_CTX ctx;
		EVP_DigestInit(&ctx, md);
		EVP_DigestUpdate(&ctx, blob, len);
		EVP_DigestFinal(&ctx, digest, NULL);
		for(i = 0; i < md->md_size; i++) {
			char hex[4];
			snprintf(hex, sizeof(hex), "%02x:", digest[i]);
			strlcat(retval, hex, sizeof(retval));
		}
		retval[strlen(retval) - 1] = '\0';
		memset(blob, 0, len);
		xfree(blob);
	}
	return retval;
}

/*
 * Reads a multiple-precision integer in decimal from the buffer, and advances
 * the pointer.  The integer must already be initialized.  This function is
 * permitted to modify the buffer.  This leaves *cpp to point just beyond the
 * last processed (and maybe modified) character.  Note that this may modify
 * the buffer containing the number.
 */
int
read_bignum(char **cpp, BIGNUM * value)
{
	char *cp = *cpp;
	int old;

	/* Skip any leading whitespace. */
	for (; *cp == ' ' || *cp == '\t'; cp++)
		;

	/* Check that it begins with a decimal digit. */
	if (*cp < '0' || *cp > '9')
		return 0;

	/* Save starting position. */
	*cpp = cp;

	/* Move forward until all decimal digits skipped. */
	for (; *cp >= '0' && *cp <= '9'; cp++)
		;

	/* Save the old terminating character, and replace it by \0. */
	old = *cp;
	*cp = 0;

	/* Parse the number. */
	if (BN_dec2bn(&value, *cpp) == 0)
		return 0;

	/* Restore old terminating character. */
	*cp = old;

	/* Move beyond the number and return success. */
	*cpp = cp;
	return 1;
}
int
write_bignum(FILE *f, BIGNUM *num)
{
	char *buf = BN_bn2dec(num);
	if (buf == NULL) {
		error("write_bignum: BN_bn2dec() failed");
		return 0;
	}
	fprintf(f, " %s", buf);
	free(buf);
	return 1;
}

/* returns 1 ok, -1 error, 0 type mismatch */
int
key_read(Key *ret, char **cpp)
{
	Key *k;
	int success = -1;
	char *cp, *space;
	int len, n, type;
	u_int bits;
	u_char *blob;

	cp = *cpp;

	switch(ret->type) {
	case KEY_RSA1:
		/* Get number of bits. */
		if (*cp < '0' || *cp > '9')
			return -1;	/* Bad bit count... */
		for (bits = 0; *cp >= '0' && *cp <= '9'; cp++)
			bits = 10 * bits + *cp - '0';
		if (bits == 0)
			return -1;
		*cpp = cp;
		/* Get public exponent, public modulus. */
		if (!read_bignum(cpp, ret->rsa->e))
			return -1;
		if (!read_bignum(cpp, ret->rsa->n))
			return -1;
		success = 1;
		break;
	case KEY_UNSPEC:
	case KEY_RSA:
	case KEY_DSA:
		space = strchr(cp, ' ');
		if (space == NULL) {
			debug3("key_read: no space");
			return -1;
		}
		*space = '\0';
		type = key_type_from_name(cp);
		*space = ' ';
		if (type == KEY_UNSPEC) {
			debug3("key_read: no key found");
			return -1;
		}
		cp = space+1;
		if (*cp == '\0') {
			debug3("key_read: short string");
			return -1;
		}
		if (ret->type == KEY_UNSPEC) {
			ret->type = type;
		} else if (ret->type != type) {
			/* is a key, but different type */
			debug3("key_read: type mismatch");
			return 0;
		}
		len = 2*strlen(cp);
		blob = xmalloc(len);
		n = uudecode(cp, blob, len);
		if (n < 0) {
			error("key_read: uudecode %s failed", cp);
			return -1;
		}
		k = key_from_blob(blob, n);
		if (k == NULL) {
			error("key_read: key_from_blob %s failed", cp);
			return -1;
		}
		xfree(blob);
		if (k->type != type) {
			error("key_read: type mismatch: encoding error");
			key_free(k);
			return -1;
		}
/*XXXX*/
		if (ret->type == KEY_RSA) {
			if (ret->rsa != NULL)
				RSA_free(ret->rsa);
			ret->rsa = k->rsa;
			k->rsa = NULL;
			success = 1;
#ifdef DEBUG_PK
			RSA_print_fp(stderr, ret->rsa, 8);
#endif
		} else {
			if (ret->dsa != NULL)
				DSA_free(ret->dsa);
			ret->dsa = k->dsa;
			k->dsa = NULL;
			success = 1;
#ifdef DEBUG_PK
			DSA_print_fp(stderr, ret->dsa, 8);
#endif
		}
/*XXXX*/
		if (success != 1)
			break;
		key_free(k);
		/* advance cp: skip whitespace and data */
		while (*cp == ' ' || *cp == '\t')
			cp++;
		while (*cp != '\0' && *cp != ' ' && *cp != '\t')
			cp++;
		*cpp = cp;
		break;
	default:
		fatal("key_read: bad key type: %d", ret->type);
		break;
	}
	return success;
}
int
key_write(Key *key, FILE *f)
{
	int success = 0;
	u_int bits = 0;

	if (key->type == KEY_RSA1 && key->rsa != NULL) {
		/* size of modulus 'n' */
		bits = BN_num_bits(key->rsa->n);
		fprintf(f, "%u", bits);
		if (write_bignum(f, key->rsa->e) &&
		    write_bignum(f, key->rsa->n)) {
			success = 1;
		} else {
			error("key_write: failed for RSA key");
		}
	} else if ((key->type == KEY_DSA && key->dsa != NULL) ||
	    (key->type == KEY_RSA && key->rsa != NULL)) {
		int len, n;
		u_char *blob, *uu;
		key_to_blob(key, &blob, &len);
		uu = xmalloc(2*len);
		n = uuencode(blob, len, uu, 2*len);
		if (n > 0) {
			fprintf(f, "%s %s", key_ssh_name(key), uu);
			success = 1;
		}
		xfree(blob);
		xfree(uu);
	}
	return success;
}
char *
key_type(Key *k)
{
	switch (k->type) {
	case KEY_RSA1:
		return "RSA1";
		break;
	case KEY_RSA:
		return "RSA";
		break;
	case KEY_DSA:
		return "DSA";
		break;
	}
	return "unknown";
}
char *
key_ssh_name(Key *k)
{
	switch (k->type) {
	case KEY_RSA:
		return "ssh-rsa";
		break;
	case KEY_DSA:
		return "ssh-dss";
		break;
	}
	return "ssh-unknown";
}
u_int
key_size(Key *k){
	switch (k->type) {
	case KEY_RSA1:
	case KEY_RSA:
		return BN_num_bits(k->rsa->n);
		break;
	case KEY_DSA:
		return BN_num_bits(k->dsa->p);
		break;
	}
	return 0;
}

RSA *
rsa_generate_private_key(u_int bits)
{
        RSA *private;
        private = RSA_generate_key(bits, 35, NULL, NULL);
        if (private == NULL)
                fatal("rsa_generate_private_key: key generation failed.");
        return private;
}

DSA*
dsa_generate_private_key(u_int bits)
{
	DSA *private = DSA_generate_parameters(bits, NULL, 0, NULL, NULL, NULL, NULL);
	if (private == NULL)
		fatal("dsa_generate_private_key: DSA_generate_parameters failed");
	if (!DSA_generate_key(private))
                fatal("dsa_generate_private_key: DSA_generate_key failed.");
        if (private == NULL)
                fatal("dsa_generate_private_key: NULL.");
	return private;
}

Key *
key_generate(int type, u_int bits)
{
	Key *k = key_new(KEY_UNSPEC);
	switch (type) {
        case KEY_DSA:
		k->dsa = dsa_generate_private_key(bits);
		break;
	case KEY_RSA:
	case KEY_RSA1:
		k->rsa = rsa_generate_private_key(bits);
		break;
	default:
                fatal("key_generate: unknown type %d", type);
	}
        k->type = type;
	return k;
}

Key *
key_from_private(Key *k)
{
	Key *n = NULL;
	switch (k->type) {
        case KEY_DSA:
		n = key_new(k->type);
		BN_copy(n->dsa->p, k->dsa->p);
		BN_copy(n->dsa->q, k->dsa->q);
		BN_copy(n->dsa->g, k->dsa->g);
		BN_copy(n->dsa->pub_key, k->dsa->pub_key);
		break;
	case KEY_RSA:
	case KEY_RSA1:
		n = key_new(k->type);
		BN_copy(n->rsa->n, k->rsa->n);
		BN_copy(n->rsa->e, k->rsa->e);
		break;
	default:
                fatal("key_from_private: unknown type %d", k->type);
		break;
	}
	return n;
}

int
key_type_from_name(char *name)
{
	if (strcmp(name, "rsa1") == 0){
		return KEY_RSA1;
	} else if (strcmp(name, "rsa") == 0){
		return KEY_RSA;
	} else if (strcmp(name, "dsa") == 0){
		return KEY_DSA;
	} else if (strcmp(name, "ssh-rsa") == 0){
		return KEY_RSA;
	} else if (strcmp(name, "ssh-dss") == 0){
		return KEY_DSA;
	}
	debug("key_type_from_name: unknown key type '%s'", name);
	return KEY_UNSPEC;
}

Key *
key_from_blob(char *blob, int blen)
{
	Buffer b;
	char *ktype;
	int rlen, type;
	Key *key = NULL;

#ifdef DEBUG_PK
	dump_base64(stderr, blob, blen);
#endif
	buffer_init(&b);
	buffer_append(&b, blob, blen);
	ktype = buffer_get_string(&b, NULL);
	type = key_type_from_name(ktype);

	switch(type){
	case KEY_RSA:
		key = key_new(type);
		buffer_get_bignum2(&b, key->rsa->e);
		buffer_get_bignum2(&b, key->rsa->n);
#ifdef DEBUG_PK
		RSA_print_fp(stderr, key->rsa, 8);
#endif
		break;
	case KEY_DSA:
		key = key_new(type);
		buffer_get_bignum2(&b, key->dsa->p);
		buffer_get_bignum2(&b, key->dsa->q);
		buffer_get_bignum2(&b, key->dsa->g);
		buffer_get_bignum2(&b, key->dsa->pub_key);
#ifdef DEBUG_PK
		DSA_print_fp(stderr, key->dsa, 8);
#endif
		break;
	case KEY_UNSPEC:
		key = key_new(type);
		break;
	default:
		error("key_from_blob: cannot handle type %s", ktype);
		break;
	}
	rlen = buffer_len(&b);
	if (key != NULL && rlen != 0)
		error("key_from_blob: remaining bytes in key blob %d", rlen);
	xfree(ktype);
	buffer_free(&b);
	return key;
}

int
key_to_blob(Key *key, u_char **blobp, u_int *lenp)
{
	Buffer b;
	int len;
	u_char *buf;

	if (key == NULL) {
		error("key_to_blob: key == NULL");
		return 0;
	}
	buffer_init(&b);
	switch(key->type){
	case KEY_DSA:
		buffer_put_cstring(&b, key_ssh_name(key));
		buffer_put_bignum2(&b, key->dsa->p);
		buffer_put_bignum2(&b, key->dsa->q);
		buffer_put_bignum2(&b, key->dsa->g);
		buffer_put_bignum2(&b, key->dsa->pub_key);
		break;
	case KEY_RSA:
		buffer_put_cstring(&b, key_ssh_name(key));
		buffer_put_bignum2(&b, key->rsa->e);
		buffer_put_bignum2(&b, key->rsa->n);
		break;
	default:
		error("key_to_blob: illegal key type %d", key->type);
		break;
	}
	len = buffer_len(&b);
	buf = xmalloc(len);
	memcpy(buf, buffer_ptr(&b), len);
	memset(buffer_ptr(&b), 0, len);
	buffer_free(&b);
	if (lenp != NULL)
		*lenp = len;
	if (blobp != NULL)
		*blobp = buf;
	return len;
}

int
key_sign(
    Key *key,
    u_char **sigp, int *lenp,
    u_char *data, int datalen)
{
	switch(key->type){
	case KEY_DSA:
		return ssh_dss_sign(key, sigp, lenp, data, datalen);
		break;
	case KEY_RSA:
		return ssh_rsa_sign(key, sigp, lenp, data, datalen);
		break;
	default:
		error("key_sign: illegal key type %d", key->type);
		return -1;
		break;
	}
}

int
key_verify(
    Key *key,
    u_char *signature, int signaturelen,
    u_char *data, int datalen)
{
	switch(key->type){
	case KEY_DSA:
		return ssh_dss_verify(key, signature, signaturelen, data, datalen);
		break;
	case KEY_RSA:
		return ssh_rsa_verify(key, signature, signaturelen, data, datalen);
		break;
	default:
		error("key_verify: illegal key type %d", key->type);
		return -1;
		break;
	}
}
Commit	Line	Data
4fe2af09	1	/*
bcbf86ec	2	* read_bignum():
	3	* Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
	4	*
	5	* As far as I am concerned, the code I have written for this software
	6	* can be used freely for any purpose. Any derived versions of this
	7	* software must be clearly marked as such, and if the derived work is
	8	* incompatible with the protocol description in the RFC file, it must be
	9	* called by a name other than "ssh" or "Secure Shell".
	10	*
	11	*
4fe2af09	12	* Copyright (c) 2000 Markus Friedl. All rights reserved.
	13	*
	14	* Redistribution and use in source and binary forms, with or without
	15	* modification, are permitted provided that the following conditions
	16	* are met:
	17	* 1. Redistributions of source code must retain the above copyright
	18	* notice, this list of conditions and the following disclaimer.
	19	* 2. Redistributions in binary form must reproduce the above copyright
	20	* notice, this list of conditions and the following disclaimer in the
	21	* documentation and/or other materials provided with the distribution.
4fe2af09	22	*
	23	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
	24	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
	25	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
	26	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
	27	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
	28	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
	29	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
	30	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
	31	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
	32	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	33	*/
4fe2af09	34
4fe2af09	35	#include "includes.h"
35484284	36	#include "ssh.h"
4fe2af09	37	#include <openssl/rsa.h>
	38	#include <openssl/dsa.h>
	39	#include <openssl/evp.h>
4fe2af09	40	#include "xmalloc.h"
4fe2af09	41	#include "key.h"
fa08c86b	42	#include "rsa.h"
	43	#include "ssh-dss.h"
	44	#include "ssh-rsa.h"
a306f2dd	45	#include "uuencode.h"
fa08c86b	46	#include "buffer.h"
fa08c86b	47	#include "bufaux.h"
a306f2dd	48
b5c334cc	49	RCSID("$OpenBSD: key.c,v 1.14 2001/01/16 19:20:06 markus Exp $");
4fe2af09	50
	51	Key *
	52	key_new(int type)
	53	{
	54	Key *k;
	55	RSA *rsa;
	56	DSA *dsa;
	57	k = xmalloc(sizeof(*k));
	58	k->type = type;
a306f2dd	59	k->dsa = NULL;
a306f2dd	60	k->rsa = NULL;
4fe2af09	61	switch (k->type) {
fa08c86b	62	case KEY_RSA1:
4fe2af09	63	case KEY_RSA:
	64	rsa = RSA_new();
	65	rsa->n = BN_new();
	66	rsa->e = BN_new();
	67	k->rsa = rsa;
	68	break;
	69	case KEY_DSA:
	70	dsa = DSA_new();
	71	dsa->p = BN_new();
	72	dsa->q = BN_new();
	73	dsa->g = BN_new();
	74	dsa->pub_key = BN_new();
	75	k->dsa = dsa;
	76	break;
fa08c86b	77	case KEY_UNSPEC:
4fe2af09	78	break;
	79	default:
	80	fatal("key_new: bad key type %d", k->type);
	81	break;
	82	}
	83	return k;
	84	}
fa08c86b	85	Key *
	86	key_new_private(int type)
	87	{
	88	Key *k = key_new(type);
	89	switch (k->type) {
	90	case KEY_RSA1:
	91	case KEY_RSA:
	92	k->rsa->d = BN_new();
	93	k->rsa->iqmp = BN_new();
	94	k->rsa->q = BN_new();
	95	k->rsa->p = BN_new();
	96	k->rsa->dmq1 = BN_new();
	97	k->rsa->dmp1 = BN_new();
	98	break;
	99	case KEY_DSA:
	100	k->dsa->priv_key = BN_new();
	101	break;
	102	case KEY_UNSPEC:
	103	break;
	104	default:
	105	break;
	106	}
	107	return k;
	108	}
4fe2af09	109	void
	110	key_free(Key *k)
	111	{
	112	switch (k->type) {
fa08c86b	113	case KEY_RSA1:
4fe2af09	114	case KEY_RSA:
	115	if (k->rsa != NULL)
	116	RSA_free(k->rsa);
	117	k->rsa = NULL;
	118	break;
	119	case KEY_DSA:
	120	if (k->dsa != NULL)
	121	DSA_free(k->dsa);
	122	k->dsa = NULL;
	123	break;
fa08c86b	124	case KEY_UNSPEC:
fa08c86b	125	break;
4fe2af09	126	default:
	127	fatal("key_free: bad key type %d", k->type);
	128	break;
	129	}
	130	xfree(k);
	131	}
	132	int
	133	key_equal(Key a, Key b)
	134	{
	135	if (a == NULL \|\| b == NULL \|\| a->type != b->type)
	136	return 0;
	137	switch (a->type) {
fa08c86b	138	case KEY_RSA1:
4fe2af09	139	case KEY_RSA:
	140	return a->rsa != NULL && b->rsa != NULL &&
	141	BN_cmp(a->rsa->e, b->rsa->e) == 0 &&
	142	BN_cmp(a->rsa->n, b->rsa->n) == 0;
	143	break;
	144	case KEY_DSA:
	145	return a->dsa != NULL && b->dsa != NULL &&
	146	BN_cmp(a->dsa->p, b->dsa->p) == 0 &&
	147	BN_cmp(a->dsa->q, b->dsa->q) == 0 &&
	148	BN_cmp(a->dsa->g, b->dsa->g) == 0 &&
	149	BN_cmp(a->dsa->pub_key, b->dsa->pub_key) == 0;
	150	break;
	151	default:
a306f2dd	152	fatal("key_equal: bad key type %d", a->type);
4fe2af09	153	break;
	154	}
	155	return 0;
	156	}
	157
4fe2af09	158	/*
	159	* Generate key fingerprint in ascii format.
	160	* Based on ideas and code from Bjoern Groenvall <bg@sics.se>
	161	*/
	162	char *
	163	key_fingerprint(Key *k)
	164	{
74fc9186	165	static char retval[(EVP_MAX_MD_SIZE+1)*3];
1e3b8b07	166	u_char *blob = NULL;
4fe2af09	167	int len = 0;
a306f2dd	168	int nlen, elen;
4fe2af09	169
fa08c86b	170	retval[0] = '\0';
4fe2af09	171	switch (k->type) {
fa08c86b	172	case KEY_RSA1:
4fe2af09	173	nlen = BN_num_bytes(k->rsa->n);
	174	elen = BN_num_bytes(k->rsa->e);
	175	len = nlen + elen;
a306f2dd	176	blob = xmalloc(len);
	177	BN_bn2bin(k->rsa->n, blob);
	178	BN_bn2bin(k->rsa->e, blob + nlen);
4fe2af09	179	break;
4fe2af09	180	case KEY_DSA:
fa08c86b	181	case KEY_RSA:
	182	key_to_blob(k, &blob, &len);
	183	break;
	184	case KEY_UNSPEC:
	185	return retval;
4fe2af09	186	break;
	187	default:
	188	fatal("key_fingerprint: bad key type %d", k->type);
	189	break;
	190	}
a306f2dd	191	if (blob != NULL) {
74fc9186	192	int i;
1e3b8b07	193	u_char digest[EVP_MAX_MD_SIZE];
74fc9186	194	EVP_MD *md = EVP_md5();
	195	EVP_MD_CTX ctx;
	196	EVP_DigestInit(&ctx, md);
	197	EVP_DigestUpdate(&ctx, blob, len);
	198	EVP_DigestFinal(&ctx, digest, NULL);
	199	for(i = 0; i < md->md_size; i++) {
	200	char hex[4];
	201	snprintf(hex, sizeof(hex), "%02x:", digest[i]);
	202	strlcat(retval, hex, sizeof(retval));
	203	}
	204	retval[strlen(retval) - 1] = '\0';
a306f2dd	205	memset(blob, 0, len);
a306f2dd	206	xfree(blob);
4fe2af09	207	}
	208	return retval;
	209	}
	210
	211	/*
	212	* Reads a multiple-precision integer in decimal from the buffer, and advances
	213	* the pointer. The integer must already be initialized. This function is
	214	* permitted to modify the buffer. This leaves *cpp to point just beyond the
	215	* last processed (and maybe modified) character. Note that this may modify
	216	* the buffer containing the number.
	217	*/
	218	int
	219	read_bignum(char *cpp, BIGNUM value)
	220	{
	221	char cp = cpp;
	222	int old;
	223
	224	/* Skip any leading whitespace. */
	225	for (; cp == ' ' \|\| cp == '\t'; cp++)
	226	;
	227
	228	/* Check that it begins with a decimal digit. */
	229	if (cp < '0' \|\| cp > '9')
	230	return 0;
	231
	232	/* Save starting position. */
	233	*cpp = cp;
	234
	235	/* Move forward until all decimal digits skipped. */
	236	for (; cp >= '0' && cp <= '9'; cp++)
	237	;
	238
	239	/* Save the old terminating character, and replace it by \0. */
	240	old = *cp;
	241	*cp = 0;
	242
	243	/* Parse the number. */
	244	if (BN_dec2bn(&value, *cpp) == 0)
	245	return 0;
	246
	247	/* Restore old terminating character. */
	248	*cp = old;
	249
	250	/* Move beyond the number and return success. */
	251	*cpp = cp;
	252	return 1;
	253	}
	254	int
	255	write_bignum(FILE f, BIGNUM num)
	256	{
	257	char *buf = BN_bn2dec(num);
	258	if (buf == NULL) {
	259	error("write_bignum: BN_bn2dec() failed");
	260	return 0;
	261	}
	262	fprintf(f, " %s", buf);
	263	free(buf);
	264	return 1;
	265	}
fa08c86b	266
	267	/* returns 1 ok, -1 error, 0 type mismatch */
	268	int
a306f2dd	269	key_read(Key ret, char *cpp)
4fe2af09	270	{
a306f2dd	271	Key *k;
fa08c86b	272	int success = -1;
	273	char cp, space;
	274	int len, n, type;
	275	u_int bits;
1e3b8b07	276	u_char *blob;
a306f2dd	277
	278	cp = *cpp;
	279
4fe2af09	280	switch(ret->type) {
fa08c86b	281	case KEY_RSA1:
a306f2dd	282	/* Get number of bits. */
a306f2dd	283	if (cp < '0' \|\| cp > '9')
fa08c86b	284	return -1; /* Bad bit count... */
a306f2dd	285	for (bits = 0; cp >= '0' && cp <= '9'; cp++)
a306f2dd	286	bits = 10 * bits + *cp - '0';
4fe2af09	287	if (bits == 0)
fa08c86b	288	return -1;
a306f2dd	289	*cpp = cp;
4fe2af09	290	/* Get public exponent, public modulus. */
4fe2af09	291	if (!read_bignum(cpp, ret->rsa->e))
fa08c86b	292	return -1;
4fe2af09	293	if (!read_bignum(cpp, ret->rsa->n))
fa08c86b	294	return -1;
fa08c86b	295	success = 1;
4fe2af09	296	break;
fa08c86b	297	case KEY_UNSPEC:
fa08c86b	298	case KEY_RSA:
4fe2af09	299	case KEY_DSA:
fa08c86b	300	space = strchr(cp, ' ');
	301	if (space == NULL) {
	302	debug3("key_read: no space");
	303	return -1;
	304	}
	305	*space = '\0';
	306	type = key_type_from_name(cp);
	307	*space = ' ';
	308	if (type == KEY_UNSPEC) {
	309	debug3("key_read: no key found");
	310	return -1;
	311	}
	312	cp = space+1;
	313	if (*cp == '\0') {
	314	debug3("key_read: short string");
	315	return -1;
	316	}
	317	if (ret->type == KEY_UNSPEC) {
	318	ret->type = type;
	319	} else if (ret->type != type) {
	320	/* is a key, but different type */
	321	debug3("key_read: type mismatch");
4fe2af09	322	return 0;
fa08c86b	323	}
a306f2dd	324	len = 2*strlen(cp);
	325	blob = xmalloc(len);
	326	n = uudecode(cp, blob, len);
1d1ffb87	327	if (n < 0) {
71276795	328	error("key_read: uudecode %s failed", cp);
fa08c86b	329	return -1;
1d1ffb87	330	}
fa08c86b	331	k = key_from_blob(blob, n);
71276795	332	if (k == NULL) {
fa08c86b	333	error("key_read: key_from_blob %s failed", cp);
fa08c86b	334	return -1;
71276795	335	}
a306f2dd	336	xfree(blob);
fa08c86b	337	if (k->type != type) {
	338	error("key_read: type mismatch: encoding error");
	339	key_free(k);
	340	return -1;
	341	}
	342	/XXXX/
	343	if (ret->type == KEY_RSA) {
	344	if (ret->rsa != NULL)
	345	RSA_free(ret->rsa);
	346	ret->rsa = k->rsa;
	347	k->rsa = NULL;
	348	success = 1;
	349	#ifdef DEBUG_PK
	350	RSA_print_fp(stderr, ret->rsa, 8);
	351	#endif
	352	} else {
	353	if (ret->dsa != NULL)
	354	DSA_free(ret->dsa);
	355	ret->dsa = k->dsa;
	356	k->dsa = NULL;
	357	success = 1;
	358	#ifdef DEBUG_PK
	359	DSA_print_fp(stderr, ret->dsa, 8);
	360	#endif
	361	}
	362	/XXXX/
	363	if (success != 1)
	364	break;
a306f2dd	365	key_free(k);
71276795	366	/* advance cp: skip whitespace and data */
	367	while (cp == ' ' \|\| cp == '\t')
	368	cp++;
	369	while (cp != '\0' && cp != ' ' && *cp != '\t')
	370	cp++;
	371	*cpp = cp;
4fe2af09	372	break;
4fe2af09	373	default:
a306f2dd	374	fatal("key_read: bad key type: %d", ret->type);
4fe2af09	375	break;
4fe2af09	376	}
fa08c86b	377	return success;
4fe2af09	378	}
	379	int
	380	key_write(Key key, FILE f)
	381	{
	382	int success = 0;
1e3b8b07	383	u_int bits = 0;
4fe2af09	384
fa08c86b	385	if (key->type == KEY_RSA1 && key->rsa != NULL) {
4fe2af09	386	/* size of modulus 'n' */
	387	bits = BN_num_bits(key->rsa->n);
	388	fprintf(f, "%u", bits);
	389	if (write_bignum(f, key->rsa->e) &&
	390	write_bignum(f, key->rsa->n)) {
	391	success = 1;
	392	} else {
	393	error("key_write: failed for RSA key");
	394	}
fa08c86b	395	} else if ((key->type == KEY_DSA && key->dsa != NULL) \|\|
fa08c86b	396	(key->type == KEY_RSA && key->rsa != NULL)) {
a306f2dd	397	int len, n;
1e3b8b07	398	u_char blob, uu;
fa08c86b	399	key_to_blob(key, &blob, &len);
a306f2dd	400	uu = xmalloc(2*len);
1d1ffb87	401	n = uuencode(blob, len, uu, 2*len);
1d1ffb87	402	if (n > 0) {
fa08c86b	403	fprintf(f, "%s %s", key_ssh_name(key), uu);
1d1ffb87	404	success = 1;
1d1ffb87	405	}
a306f2dd	406	xfree(blob);
a306f2dd	407	xfree(uu);
4fe2af09	408	}
	409	return success;
	410	}
1d1ffb87	411	char *
	412	key_type(Key *k)
	413	{
	414	switch (k->type) {
fa08c86b	415	case KEY_RSA1:
	416	return "RSA1";
	417	break;
1d1ffb87	418	case KEY_RSA:
	419	return "RSA";
	420	break;
	421	case KEY_DSA:
	422	return "DSA";
	423	break;
	424	}
	425	return "unknown";
	426	}
fa08c86b	427	char *
	428	key_ssh_name(Key *k)
	429	{
	430	switch (k->type) {
	431	case KEY_RSA:
	432	return "ssh-rsa";
	433	break;
	434	case KEY_DSA:
	435	return "ssh-dss";
	436	break;
	437	}
	438	return "ssh-unknown";
	439	}
	440	u_int
2e73a022	441	key_size(Key *k){
2e73a022	442	switch (k->type) {
fa08c86b	443	case KEY_RSA1:
2e73a022	444	case KEY_RSA:
	445	return BN_num_bits(k->rsa->n);
	446	break;
	447	case KEY_DSA:
	448	return BN_num_bits(k->dsa->p);
	449	break;
	450	}
	451	return 0;
	452	}
fa08c86b	453
fa08c86b	454	RSA *
1e3b8b07	455	rsa_generate_private_key(u_int bits)
fa08c86b	456	{
	457	RSA *private;
	458	private = RSA_generate_key(bits, 35, NULL, NULL);
	459	if (private == NULL)
	460	fatal("rsa_generate_private_key: key generation failed.");
	461	return private;
	462	}
	463
	464	DSA*
1e3b8b07	465	dsa_generate_private_key(u_int bits)
fa08c86b	466	{
	467	DSA *private = DSA_generate_parameters(bits, NULL, 0, NULL, NULL, NULL, NULL);
	468	if (private == NULL)
	469	fatal("dsa_generate_private_key: DSA_generate_parameters failed");
	470	if (!DSA_generate_key(private))
	471	fatal("dsa_generate_private_key: DSA_generate_key failed.");
	472	if (private == NULL)
	473	fatal("dsa_generate_private_key: NULL.");
	474	return private;
	475	}
	476
	477	Key *
1e3b8b07	478	key_generate(int type, u_int bits)
fa08c86b	479	{
	480	Key *k = key_new(KEY_UNSPEC);
	481	switch (type) {
	482	case KEY_DSA:
	483	k->dsa = dsa_generate_private_key(bits);
	484	break;
	485	case KEY_RSA:
	486	case KEY_RSA1:
	487	k->rsa = rsa_generate_private_key(bits);
	488	break;
	489	default:
	490	fatal("key_generate: unknown type %d", type);
	491	}
	492	k->type = type;
	493	return k;
	494	}
	495
	496	Key *
	497	key_from_private(Key *k)
	498	{
	499	Key *n = NULL;
	500	switch (k->type) {
	501	case KEY_DSA:
	502	n = key_new(k->type);
	503	BN_copy(n->dsa->p, k->dsa->p);
	504	BN_copy(n->dsa->q, k->dsa->q);
	505	BN_copy(n->dsa->g, k->dsa->g);
	506	BN_copy(n->dsa->pub_key, k->dsa->pub_key);
	507	break;
	508	case KEY_RSA:
	509	case KEY_RSA1:
	510	n = key_new(k->type);
	511	BN_copy(n->rsa->n, k->rsa->n);
	512	BN_copy(n->rsa->e, k->rsa->e);
	513	break;
	514	default:
	515	fatal("key_from_private: unknown type %d", k->type);
	516	break;
	517	}
	518	return n;
	519	}
	520
	521	int
	522	key_type_from_name(char *name)
	523	{
	524	if (strcmp(name, "rsa1") == 0){
	525	return KEY_RSA1;
	526	} else if (strcmp(name, "rsa") == 0){
	527	return KEY_RSA;
	528	} else if (strcmp(name, "dsa") == 0){
	529	return KEY_DSA;
	530	} else if (strcmp(name, "ssh-rsa") == 0){
	531	return KEY_RSA;
	532	} else if (strcmp(name, "ssh-dss") == 0){
	533	return KEY_DSA;
	534	}
	535	debug("key_type_from_name: unknown key type '%s'", name);
	536	return KEY_UNSPEC;
	537	}
	538
	539	Key *
	540	key_from_blob(char *blob, int blen)
	541	{
	542	Buffer b;
543	char *ktype;
544	int rlen, type;
545	Key *key = NULL;
546
547	#ifdef DEBUG_PK
548	dump_base64(stderr, blob, blen);
549	#endif
550	buffer_init(&b);
551	buffer_append(&b, blob, blen);
552	ktype = buffer_get_string(&b, NULL);
553	type = key_type_from_name(ktype);
554
555	switch(type){
556	case KEY_RSA:
557	key = key_new(type);
fa08c86b	558	buffer_get_bignum2(&b, key->rsa->e);
b5c334cc	559	buffer_get_bignum2(&b, key->rsa->n);
fa08c86b	560	#ifdef DEBUG_PK
	561	RSA_print_fp(stderr, key->rsa, 8);
	562	#endif
	563	break;
	564	case KEY_DSA:
	565	key = key_new(type);
	566	buffer_get_bignum2(&b, key->dsa->p);
	567	buffer_get_bignum2(&b, key->dsa->q);
	568	buffer_get_bignum2(&b, key->dsa->g);
	569	buffer_get_bignum2(&b, key->dsa->pub_key);
	570	#ifdef DEBUG_PK
	571	DSA_print_fp(stderr, key->dsa, 8);
	572	#endif
	573	break;
	574	case KEY_UNSPEC:
	575	key = key_new(type);
	576	break;
	577	default:
	578	error("key_from_blob: cannot handle type %s", ktype);
	579	break;
	580	}
	581	rlen = buffer_len(&b);
	582	if (key != NULL && rlen != 0)
	583	error("key_from_blob: remaining bytes in key blob %d", rlen);
	584	xfree(ktype);
	585	buffer_free(&b);
	586	return key;
	587	}
	588
	589	int
1e3b8b07	590	key_to_blob(Key key, u_char blobp, u_int lenp)
fa08c86b	591	{
	592	Buffer b;
	593	int len;
1e3b8b07	594	u_char *buf;
fa08c86b	595
	596	if (key == NULL) {
	597	error("key_to_blob: key == NULL");
	598	return 0;
	599	}
	600	buffer_init(&b);
	601	switch(key->type){
	602	case KEY_DSA:
	603	buffer_put_cstring(&b, key_ssh_name(key));
	604	buffer_put_bignum2(&b, key->dsa->p);
	605	buffer_put_bignum2(&b, key->dsa->q);
	606	buffer_put_bignum2(&b, key->dsa->g);
	607	buffer_put_bignum2(&b, key->dsa->pub_key);
	608	break;
	609	case KEY_RSA:
	610	buffer_put_cstring(&b, key_ssh_name(key));
fa08c86b	611	buffer_put_bignum2(&b, key->rsa->e);
b5c334cc	612	buffer_put_bignum2(&b, key->rsa->n);
fa08c86b	613	break;
	614	default:
	615	error("key_to_blob: illegal key type %d", key->type);
	616	break;
	617	}
	618	len = buffer_len(&b);
	619	buf = xmalloc(len);
	620	memcpy(buf, buffer_ptr(&b), len);
	621	memset(buffer_ptr(&b), 0, len);
	622	buffer_free(&b);
	623	if (lenp != NULL)
	624	*lenp = len;
	625	if (blobp != NULL)
	626	*blobp = buf;
	627	return len;
	628	}
	629
	630	int
	631	key_sign(
	632	Key *key,
1e3b8b07	633	u_char *sigp, int lenp,
1e3b8b07	634	u_char *data, int datalen)
fa08c86b	635	{
	636	switch(key->type){
	637	case KEY_DSA:
	638	return ssh_dss_sign(key, sigp, lenp, data, datalen);
	639	break;
	640	case KEY_RSA:
	641	return ssh_rsa_sign(key, sigp, lenp, data, datalen);
	642	break;
	643	default:
	644	error("key_sign: illegal key type %d", key->type);
	645	return -1;
	646	break;
	647	}
	648	}
	649
	650	int
	651	key_verify(
	652	Key *key,
1e3b8b07	653	u_char *signature, int signaturelen,
1e3b8b07	654	u_char *data, int datalen)
fa08c86b	655	{
	656	switch(key->type){
	657	case KEY_DSA:
	658	return ssh_dss_verify(key, signature, signaturelen, data, datalen);
	659	break;
	660	case KEY_RSA:
	661	return ssh_rsa_verify(key, signature, signaturelen, data, datalen);
	662	break;
	663	default:
	664	error("key_verify: illegal key type %d", key->type);
	665	return -1;
	666	break;
	667	}
	668	}