Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.bin/ssh/ssh-agent/../ssh-pkcs11-client.c	Lines:	0	111	0.0 %
Date:	2017-11-07	Branches:	0	40	0.0 %


/* $OpenBSD: ssh-pkcs11-client.c,v 1.7 2017/05/30 08:52:19 markus Exp $ */
/*
 * Copyright (c) 2010 Markus Friedl.  All rights reserved.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>
#include <sys/time.h>
#include <sys/socket.h>

#include <stdarg.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>

#include <openssl/rsa.h>

#include "pathnames.h"
#include "xmalloc.h"
#include "buffer.h"
#include "log.h"
#include "misc.h"
#include "key.h"
#include "authfd.h"
#include "atomicio.h"
#include "ssh-pkcs11.h"

/* borrows code from sftp-server and ssh-agent */

int fd = -1;
pid_t pid = -1;

static void
send_msg(Buffer *m)
{
	u_char buf[4];
	int mlen = buffer_len(m);

	put_u32(buf, mlen);
	if (atomicio(vwrite, fd, buf, 4) != 4 ||
	    atomicio(vwrite, fd, buffer_ptr(m),
	    buffer_len(m)) != buffer_len(m))
		error("write to helper failed");
	buffer_consume(m, mlen);
}

static int
recv_msg(Buffer *m)
{
	u_int l, len;
	u_char buf[1024];

	if ((len = atomicio(read, fd, buf, 4)) != 4) {
		error("read from helper failed: %u", len);
		return (0); /* XXX */
	}
	len = get_u32(buf);
	if (len > 256 * 1024)
		fatal("response too long: %u", len);
	/* read len bytes into m */
	buffer_clear(m);
	while (len > 0) {
		l = len;
		if (l > sizeof(buf))
			l = sizeof(buf);
		if (atomicio(read, fd, buf, l) != l) {
			error("response from helper failed.");
			return (0); /* XXX */
		}
		buffer_append(m, buf, l);
		len -= l;
	}
	return (buffer_get_char(m));
}

int
pkcs11_init(int interactive)
{
	return (0);
}

void
pkcs11_terminate(void)
{
	close(fd);
}

static int
pkcs11_rsa_private_encrypt(int flen, const u_char *from, u_char *to, RSA *rsa,
    int padding)
{
	struct sshkey key;	/* XXX */
	u_char *blob, *signature = NULL;
	u_int blen, slen = 0;
	int ret = -1;
	Buffer msg;

	if (padding != RSA_PKCS1_PADDING)
		return (-1);
	key.type = KEY_RSA;
	key.rsa = rsa;
	if (key_to_blob(&key, &blob, &blen) == 0)
		return -1;
	buffer_init(&msg);
	buffer_put_char(&msg, SSH2_AGENTC_SIGN_REQUEST);
	buffer_put_string(&msg, blob, blen);
	buffer_put_string(&msg, from, flen);
	buffer_put_int(&msg, 0);
	free(blob);
	send_msg(&msg);
	buffer_clear(&msg);

	if (recv_msg(&msg) == SSH2_AGENT_SIGN_RESPONSE) {
		signature = buffer_get_string(&msg, &slen);
		if (slen <= (u_int)RSA_size(rsa)) {
			memcpy(to, signature, slen);
			ret = slen;
		}
		free(signature);
	}
	buffer_free(&msg);
	return (ret);
}

/* redirect the private key encrypt operation to the ssh-pkcs11-helper */
static int
wrap_key(RSA *rsa)
{
	static RSA_METHOD helper_rsa;

	memcpy(&helper_rsa, RSA_get_default_method(), sizeof(helper_rsa));
	helper_rsa.name = "ssh-pkcs11-helper";
	helper_rsa.rsa_priv_enc = pkcs11_rsa_private_encrypt;
	RSA_set_method(rsa, &helper_rsa);
	return (0);
}

static int
pkcs11_start_helper(void)
{
	int pair[2];

	if (socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1) {
		error("socketpair: %s", strerror(errno));
		return (-1);
	}
	if ((pid = fork()) == -1) {
		error("fork: %s", strerror(errno));
		return (-1);
	} else if (pid == 0) {
		if ((dup2(pair[1], STDIN_FILENO) == -1) ||
		    (dup2(pair[1], STDOUT_FILENO) == -1)) {
			fprintf(stderr, "dup2: %s\n", strerror(errno));
			_exit(1);
		}
		close(pair[0]);
		close(pair[1]);
		execlp(_PATH_SSH_PKCS11_HELPER, _PATH_SSH_PKCS11_HELPER,
		    (char *)NULL);
		fprintf(stderr, "exec: %s: %s\n", _PATH_SSH_PKCS11_HELPER,
		    strerror(errno));
		_exit(1);
	}
	close(pair[1]);
	fd = pair[0];
	return (0);
}

int
pkcs11_add_provider(char *name, char *pin, Key ***keysp)
{
	struct sshkey *k;
	int i, nkeys;
	u_char *blob;
	u_int blen;
	Buffer msg;

	if (fd < 0 && pkcs11_start_helper() < 0)
		return (-1);

	buffer_init(&msg);
	buffer_put_char(&msg, SSH_AGENTC_ADD_SMARTCARD_KEY);
	buffer_put_cstring(&msg, name);
	buffer_put_cstring(&msg, pin);
	send_msg(&msg);
	buffer_clear(&msg);

	if (recv_msg(&msg) == SSH2_AGENT_IDENTITIES_ANSWER) {
		nkeys = buffer_get_int(&msg);
		*keysp = xcalloc(nkeys, sizeof(Key *));
		for (i = 0; i < nkeys; i++) {
			blob = buffer_get_string(&msg, &blen);
			free(buffer_get_string(&msg, NULL));
			k = key_from_blob(blob, blen);
			wrap_key(k->rsa);
			(*keysp)[i] = k;
			free(blob);
		}
	} else {
		nkeys = -1;
	}
	buffer_free(&msg);
	return (nkeys);
}

int
pkcs11_del_provider(char *name)
{
	int ret = -1;
	Buffer msg;

	buffer_init(&msg);
	buffer_put_char(&msg, SSH_AGENTC_REMOVE_SMARTCARD_KEY);
	buffer_put_cstring(&msg, name);
	buffer_put_cstring(&msg, "");
	send_msg(&msg);
	buffer_clear(&msg);

	if (recv_msg(&msg) == SSH_AGENT_SUCCESS)
		ret = 0;
	buffer_free(&msg);
	return (ret);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: ssh-pkcs11-client.c,v 1.7 2017/05/30 08:52:19 markus Exp $ */
2			/*
3			* Copyright (c) 2010 Markus Friedl. All rights reserved.
4			*
5			* Permission to use, copy, modify, and distribute this software for any
6			* purpose with or without fee is hereby granted, provided that the above
7			* copyright notice and this permission notice appear in all copies.
8			*
9			* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10			* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11			* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12			* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13			* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14			* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15			* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16			*/
17
18			#include <sys/types.h>
19			#include <sys/time.h>
20			#include <sys/socket.h>
21
22			#include <stdarg.h>
23			#include <string.h>
24			#include <unistd.h>
25			#include <errno.h>
26
27			#include <openssl/rsa.h>
28
29			#include "pathnames.h"
30			#include "xmalloc.h"
31			#include "buffer.h"
32			#include "log.h"
33			#include "misc.h"
34			#include "key.h"
35			#include "authfd.h"
36			#include "atomicio.h"
37			#include "ssh-pkcs11.h"
38
39			/* borrows code from sftp-server and ssh-agent */
40
41			int fd = -1;
42			pid_t pid = -1;
43
44			static void
45			send_msg(Buffer *m)
46			{
47			u_char buf[4];
48			int mlen = buffer_len(m);
49
50			put_u32(buf, mlen);
51			if (atomicio(vwrite, fd, buf, 4) != 4 \|\|
52			atomicio(vwrite, fd, buffer_ptr(m),
53			buffer_len(m)) != buffer_len(m))
54			error("write to helper failed");
55			buffer_consume(m, mlen);
56			}
57
58			static int
59			recv_msg(Buffer *m)
60			{
61			u_int l, len;
62			u_char buf[1024];
63
64			if ((len = atomicio(read, fd, buf, 4)) != 4) {
65			error("read from helper failed: %u", len);
66			return (0); /* XXX */
67			}
68			len = get_u32(buf);
69			if (len > 256 * 1024)
70			fatal("response too long: %u", len);
71			/* read len bytes into m */
72			buffer_clear(m);
73			while (len > 0) {
74			l = len;
75			if (l > sizeof(buf))
76			l = sizeof(buf);
77			if (atomicio(read, fd, buf, l) != l) {
78			error("response from helper failed.");
79			return (0); /* XXX */
80			}
81			buffer_append(m, buf, l);
82			len -= l;
83			}
84			return (buffer_get_char(m));
85			}
86
87			int
88			pkcs11_init(int interactive)
89			{
90			return (0);
91			}
92
93			void
94			pkcs11_terminate(void)
95			{
96			close(fd);
97			}
98
99			static int
100			pkcs11_rsa_private_encrypt(int flen, const u_char from, u_char to, RSA *rsa,
101			int padding)
102			{
103			struct sshkey key; /* XXX */
104			u_char blob, signature = NULL;
105			u_int blen, slen = 0;
106			int ret = -1;
107			Buffer msg;
108
109			if (padding != RSA_PKCS1_PADDING)
110			return (-1);
111			key.type = KEY_RSA;
112			key.rsa = rsa;
113			if (key_to_blob(&key, &blob, &blen) == 0)
114			return -1;
115			buffer_init(&msg);
116			buffer_put_char(&msg, SSH2_AGENTC_SIGN_REQUEST);
117			buffer_put_string(&msg, blob, blen);
118			buffer_put_string(&msg, from, flen);
119			buffer_put_int(&msg, 0);
120			free(blob);
121			send_msg(&msg);
122			buffer_clear(&msg);
123
124			if (recv_msg(&msg) == SSH2_AGENT_SIGN_RESPONSE) {
125			signature = buffer_get_string(&msg, &slen);
126			if (slen <= (u_int)RSA_size(rsa)) {
127			memcpy(to, signature, slen);
128			ret = slen;
129			}
130			free(signature);
131			}
132			buffer_free(&msg);
133			return (ret);
134			}
135
136			/* redirect the private key encrypt operation to the ssh-pkcs11-helper */
137			static int
138			wrap_key(RSA *rsa)
139			{
140			static RSA_METHOD helper_rsa;
141
142			memcpy(&helper_rsa, RSA_get_default_method(), sizeof(helper_rsa));
143			helper_rsa.name = "ssh-pkcs11-helper";
144			helper_rsa.rsa_priv_enc = pkcs11_rsa_private_encrypt;
145			RSA_set_method(rsa, &helper_rsa);
146			return (0);
147			}
148
149			static int
150			pkcs11_start_helper(void)
151			{
152			int pair[2];
153
154			if (socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1) {
155			error("socketpair: %s", strerror(errno));
156			return (-1);
157			}
158			if ((pid = fork()) == -1) {
159			error("fork: %s", strerror(errno));
160			return (-1);
161			} else if (pid == 0) {
162			if ((dup2(pair[1], STDIN_FILENO) == -1) \|\|
163			(dup2(pair[1], STDOUT_FILENO) == -1)) {
164			fprintf(stderr, "dup2: %s\n", strerror(errno));
165			_exit(1);
166			}
167			close(pair[0]);
168			close(pair[1]);
169			execlp(_PATH_SSH_PKCS11_HELPER, _PATH_SSH_PKCS11_HELPER,
170			(char *)NULL);
171			fprintf(stderr, "exec: %s: %s\n", _PATH_SSH_PKCS11_HELPER,
172			strerror(errno));
173			_exit(1);
174			}
175			close(pair[1]);
176			fd = pair[0];
177			return (0);
178			}
179
180			int
181			pkcs11_add_provider(char name, char pin, Key ***keysp)
182			{
183			struct sshkey *k;
184			int i, nkeys;
185			u_char *blob;
186			u_int blen;
187			Buffer msg;
188
189			if (fd < 0 && pkcs11_start_helper() < 0)
190			return (-1);
191
192			buffer_init(&msg);
193			buffer_put_char(&msg, SSH_AGENTC_ADD_SMARTCARD_KEY);
194			buffer_put_cstring(&msg, name);
195			buffer_put_cstring(&msg, pin);
196			send_msg(&msg);
197			buffer_clear(&msg);
198
199			if (recv_msg(&msg) == SSH2_AGENT_IDENTITIES_ANSWER) {
200			nkeys = buffer_get_int(&msg);
201			keysp = xcalloc(nkeys, sizeof(Key ));
202			for (i = 0; i < nkeys; i++) {
203			blob = buffer_get_string(&msg, &blen);
204			free(buffer_get_string(&msg, NULL));
205			k = key_from_blob(blob, blen);
206			wrap_key(k->rsa);
207			(*keysp)[i] = k;
208			free(blob);
209			}
210			} else {
211			nkeys = -1;
212			}
213			buffer_free(&msg);
214			return (nkeys);
215			}
216
217			int
218			pkcs11_del_provider(char *name)
219			{
220			int ret = -1;
221			Buffer msg;
222
223			buffer_init(&msg);
224			buffer_put_char(&msg, SSH_AGENTC_REMOVE_SMARTCARD_KEY);
225			buffer_put_cstring(&msg, name);
226			buffer_put_cstring(&msg, "");
227			send_msg(&msg);
228			buffer_clear(&msg);
229
230			if (recv_msg(&msg) == SSH_AGENT_SUCCESS)
231			ret = 0;
232			buffer_free(&msg);
233			return (ret);
234			}