authfd.c revision 296781
1/* $OpenBSD: authfd.c,v 1.100 2015/12/04 16:41:28 markus Exp $ */
2/*
3 * Author: Tatu Ylonen <ylo@cs.hut.fi>
4 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
5 *                    All rights reserved
6 * Functions for connecting the local authentication agent.
7 *
8 * As far as I am concerned, the code I have written for this software
9 * can be used freely for any purpose.  Any derived versions of this
10 * software must be clearly marked as such, and if the derived work is
11 * incompatible with the protocol description in the RFC file, it must be
12 * called by a name other than "ssh" or "Secure Shell".
13 *
14 * SSH2 implementation,
15 * Copyright (c) 2000 Markus Friedl.  All rights reserved.
16 *
17 * Redistribution and use in source and binary forms, with or without
18 * modification, are permitted provided that the following conditions
19 * are met:
20 * 1. Redistributions of source code must retain the above copyright
21 *    notice, this list of conditions and the following disclaimer.
22 * 2. Redistributions in binary form must reproduce the above copyright
23 *    notice, this list of conditions and the following disclaimer in the
24 *    documentation and/or other materials provided with the distribution.
25 *
26 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
27 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
28 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
29 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
30 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
31 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
32 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
35 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36 */
37
38#include "includes.h"
39
40#include <sys/types.h>
41#include <sys/un.h>
42#include <sys/socket.h>
43
44#include <fcntl.h>
45#include <stdlib.h>
46#include <signal.h>
47#include <stdarg.h>
48#include <string.h>
49#include <unistd.h>
50#include <errno.h>
51
52#include "xmalloc.h"
53#include "ssh.h"
54#include "rsa.h"
55#include "sshbuf.h"
56#include "sshkey.h"
57#include "authfd.h"
58#include "cipher.h"
59#include "compat.h"
60#include "log.h"
61#include "atomicio.h"
62#include "misc.h"
63#include "ssherr.h"
64
65#define MAX_AGENT_IDENTITIES	2048		/* Max keys in agent reply */
66#define MAX_AGENT_REPLY_LEN	(256 * 1024) 	/* Max bytes in agent reply */
67
68/* macro to check for "agent failure" message */
69#define agent_failed(x) \
70    ((x == SSH_AGENT_FAILURE) || \
71    (x == SSH_COM_AGENT2_FAILURE) || \
72    (x == SSH2_AGENT_FAILURE))
73
74/* Convert success/failure response from agent to a err.h status */
75static int
76decode_reply(u_char type)
77{
78	if (agent_failed(type))
79		return SSH_ERR_AGENT_FAILURE;
80	else if (type == SSH_AGENT_SUCCESS)
81		return 0;
82	else
83		return SSH_ERR_INVALID_FORMAT;
84}
85
86/* Returns the number of the authentication fd, or -1 if there is none. */
87int
88ssh_get_authentication_socket(int *fdp)
89{
90	const char *authsocket;
91	int sock, oerrno;
92	struct sockaddr_un sunaddr;
93
94	if (fdp != NULL)
95		*fdp = -1;
96
97	authsocket = getenv(SSH_AUTHSOCKET_ENV_NAME);
98	if (!authsocket)
99		return SSH_ERR_AGENT_NOT_PRESENT;
100
101	memset(&sunaddr, 0, sizeof(sunaddr));
102	sunaddr.sun_family = AF_UNIX;
103	strlcpy(sunaddr.sun_path, authsocket, sizeof(sunaddr.sun_path));
104
105	if ((sock = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
106		return SSH_ERR_SYSTEM_ERROR;
107
108	/* close on exec */
109	if (fcntl(sock, F_SETFD, FD_CLOEXEC) == -1 ||
110	    connect(sock, (struct sockaddr *)&sunaddr, sizeof(sunaddr)) < 0) {
111		oerrno = errno;
112		close(sock);
113		errno = oerrno;
114		return SSH_ERR_SYSTEM_ERROR;
115	}
116	if (fdp != NULL)
117		*fdp = sock;
118	else
119		close(sock);
120	return 0;
121}
122
123/* Communicate with agent: send request and read reply */
124static int
125ssh_request_reply(int sock, struct sshbuf *request, struct sshbuf *reply)
126{
127	int r;
128	size_t l, len;
129	char buf[1024];
130
131	/* Get the length of the message, and format it in the buffer. */
132	len = sshbuf_len(request);
133	put_u32(buf, len);
134
135	/* Send the length and then the packet to the agent. */
136	if (atomicio(vwrite, sock, buf, 4) != 4 ||
137	    atomicio(vwrite, sock, (u_char *)sshbuf_ptr(request),
138	    sshbuf_len(request)) != sshbuf_len(request))
139		return SSH_ERR_AGENT_COMMUNICATION;
140	/*
141	 * Wait for response from the agent.  First read the length of the
142	 * response packet.
143	 */
144	if (atomicio(read, sock, buf, 4) != 4)
145	    return SSH_ERR_AGENT_COMMUNICATION;
146
147	/* Extract the length, and check it for sanity. */
148	len = get_u32(buf);
149	if (len > MAX_AGENT_REPLY_LEN)
150		return SSH_ERR_INVALID_FORMAT;
151
152	/* Read the rest of the response in to the buffer. */
153	sshbuf_reset(reply);
154	while (len > 0) {
155		l = len;
156		if (l > sizeof(buf))
157			l = sizeof(buf);
158		if (atomicio(read, sock, buf, l) != l)
159			return SSH_ERR_AGENT_COMMUNICATION;
160		if ((r = sshbuf_put(reply, buf, l)) != 0)
161			return r;
162		len -= l;
163	}
164	return 0;
165}
166
167/*
168 * Closes the agent socket if it should be closed (depends on how it was
169 * obtained).  The argument must have been returned by
170 * ssh_get_authentication_socket().
171 */
172void
173ssh_close_authentication_socket(int sock)
174{
175	if (getenv(SSH_AUTHSOCKET_ENV_NAME))
176		close(sock);
177}
178
179/* Lock/unlock agent */
180int
181ssh_lock_agent(int sock, int lock, const char *password)
182{
183	int r;
184	u_char type = lock ? SSH_AGENTC_LOCK : SSH_AGENTC_UNLOCK;
185	struct sshbuf *msg;
186
187	if ((msg = sshbuf_new()) == NULL)
188		return SSH_ERR_ALLOC_FAIL;
189	if ((r = sshbuf_put_u8(msg, type)) != 0 ||
190	    (r = sshbuf_put_cstring(msg, password)) != 0)
191		goto out;
192	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
193		goto out;
194	if ((r = sshbuf_get_u8(msg, &type)) != 0)
195		goto out;
196	r = decode_reply(type);
197 out:
198	sshbuf_free(msg);
199	return r;
200}
201
202#ifdef WITH_SSH1
203static int
204deserialise_identity1(struct sshbuf *ids, struct sshkey **keyp, char **commentp)
205{
206	struct sshkey *key;
207	int r, keybits;
208	u_int32_t bits;
209	char *comment = NULL;
210
211	if ((key = sshkey_new(KEY_RSA1)) == NULL)
212		return SSH_ERR_ALLOC_FAIL;
213	if ((r = sshbuf_get_u32(ids, &bits)) != 0 ||
214	    (r = sshbuf_get_bignum1(ids, key->rsa->e)) != 0 ||
215	    (r = sshbuf_get_bignum1(ids, key->rsa->n)) != 0 ||
216	    (r = sshbuf_get_cstring(ids, &comment, NULL)) != 0)
217		goto out;
218	keybits = BN_num_bits(key->rsa->n);
219	/* XXX previously we just warned here. I think we should be strict */
220	if (keybits < 0 || bits != (u_int)keybits) {
221		r = SSH_ERR_KEY_BITS_MISMATCH;
222		goto out;
223	}
224	if (keyp != NULL) {
225		*keyp = key;
226		key = NULL;
227	}
228	if (commentp != NULL) {
229		*commentp = comment;
230		comment = NULL;
231	}
232	r = 0;
233 out:
234	sshkey_free(key);
235	free(comment);
236	return r;
237}
238#endif
239
240static int
241deserialise_identity2(struct sshbuf *ids, struct sshkey **keyp, char **commentp)
242{
243	int r;
244	char *comment = NULL;
245	const u_char *blob;
246	size_t blen;
247
248	if ((r = sshbuf_get_string_direct(ids, &blob, &blen)) != 0 ||
249	    (r = sshbuf_get_cstring(ids, &comment, NULL)) != 0)
250		goto out;
251	if ((r = sshkey_from_blob(blob, blen, keyp)) != 0)
252		goto out;
253	if (commentp != NULL) {
254		*commentp = comment;
255		comment = NULL;
256	}
257	r = 0;
258 out:
259	free(comment);
260	return r;
261}
262
263/*
264 * Fetch list of identities held by the agent.
265 */
266int
267ssh_fetch_identitylist(int sock, int version, struct ssh_identitylist **idlp)
268{
269	u_char type, code1 = 0, code2 = 0;
270	u_int32_t num, i;
271	struct sshbuf *msg;
272	struct ssh_identitylist *idl = NULL;
273	int r;
274
275	/* Determine request and expected response types */
276	switch (version) {
277	case 1:
278		code1 = SSH_AGENTC_REQUEST_RSA_IDENTITIES;
279		code2 = SSH_AGENT_RSA_IDENTITIES_ANSWER;
280		break;
281	case 2:
282		code1 = SSH2_AGENTC_REQUEST_IDENTITIES;
283		code2 = SSH2_AGENT_IDENTITIES_ANSWER;
284		break;
285	default:
286		return SSH_ERR_INVALID_ARGUMENT;
287	}
288
289	/*
290	 * Send a message to the agent requesting for a list of the
291	 * identities it can represent.
292	 */
293	if ((msg = sshbuf_new()) == NULL)
294		return SSH_ERR_ALLOC_FAIL;
295	if ((r = sshbuf_put_u8(msg, code1)) != 0)
296		goto out;
297
298	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
299		goto out;
300
301	/* Get message type, and verify that we got a proper answer. */
302	if ((r = sshbuf_get_u8(msg, &type)) != 0)
303		goto out;
304	if (agent_failed(type)) {
305		r = SSH_ERR_AGENT_FAILURE;
306		goto out;
307	} else if (type != code2) {
308		r = SSH_ERR_INVALID_FORMAT;
309		goto out;
310	}
311
312	/* Get the number of entries in the response and check it for sanity. */
313	if ((r = sshbuf_get_u32(msg, &num)) != 0)
314		goto out;
315	if (num > MAX_AGENT_IDENTITIES) {
316		r = SSH_ERR_INVALID_FORMAT;
317		goto out;
318	}
319	if (num == 0) {
320		r = SSH_ERR_AGENT_NO_IDENTITIES;
321		goto out;
322	}
323
324	/* Deserialise the response into a list of keys/comments */
325	if ((idl = calloc(1, sizeof(*idl))) == NULL ||
326	    (idl->keys = calloc(num, sizeof(*idl->keys))) == NULL ||
327	    (idl->comments = calloc(num, sizeof(*idl->comments))) == NULL) {
328		r = SSH_ERR_ALLOC_FAIL;
329		goto out;
330	}
331	for (i = 0; i < num;) {
332		switch (version) {
333		case 1:
334#ifdef WITH_SSH1
335			if ((r = deserialise_identity1(msg,
336			    &(idl->keys[i]), &(idl->comments[i]))) != 0)
337				goto out;
338#endif
339			break;
340		case 2:
341			if ((r = deserialise_identity2(msg,
342			    &(idl->keys[i]), &(idl->comments[i]))) != 0) {
343				if (r == SSH_ERR_KEY_TYPE_UNKNOWN) {
344					/* Gracefully skip unknown key types */
345					num--;
346					continue;
347				} else
348					goto out;
349			}
350			break;
351		}
352		i++;
353	}
354	idl->nkeys = num;
355	*idlp = idl;
356	idl = NULL;
357	r = 0;
358 out:
359	sshbuf_free(msg);
360	if (idl != NULL)
361		ssh_free_identitylist(idl);
362	return r;
363}
364
365void
366ssh_free_identitylist(struct ssh_identitylist *idl)
367{
368	size_t i;
369
370	if (idl == NULL)
371		return;
372	for (i = 0; i < idl->nkeys; i++) {
373		if (idl->keys != NULL)
374			sshkey_free(idl->keys[i]);
375		if (idl->comments != NULL)
376			free(idl->comments[i]);
377	}
378	free(idl);
379}
380
381/*
382 * Sends a challenge (typically from a server via ssh(1)) to the agent,
383 * and waits for a response from the agent.
384 * Returns true (non-zero) if the agent gave the correct answer, zero
385 * otherwise.
386 */
387
388#ifdef WITH_SSH1
389int
390ssh_decrypt_challenge(int sock, struct sshkey* key, BIGNUM *challenge,
391    u_char session_id[16], u_char response[16])
392{
393	struct sshbuf *msg;
394	int r;
395	u_char type;
396
397	if (key->type != KEY_RSA1)
398		return SSH_ERR_INVALID_ARGUMENT;
399	if ((msg = sshbuf_new()) == NULL)
400		return SSH_ERR_ALLOC_FAIL;
401	if ((r = sshbuf_put_u8(msg, SSH_AGENTC_RSA_CHALLENGE)) != 0 ||
402	    (r = sshbuf_put_u32(msg, BN_num_bits(key->rsa->n))) != 0 ||
403	    (r = sshbuf_put_bignum1(msg, key->rsa->e)) != 0 ||
404	    (r = sshbuf_put_bignum1(msg, key->rsa->n)) != 0 ||
405	    (r = sshbuf_put_bignum1(msg, challenge)) != 0 ||
406	    (r = sshbuf_put(msg, session_id, 16)) != 0 ||
407	    (r = sshbuf_put_u32(msg, 1)) != 0) /* Response type for proto 1.1 */
408		goto out;
409	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
410		goto out;
411	if ((r = sshbuf_get_u8(msg, &type)) != 0)
412		goto out;
413	if (agent_failed(type)) {
414		r = SSH_ERR_AGENT_FAILURE;
415		goto out;
416	} else if (type != SSH_AGENT_RSA_RESPONSE) {
417		r = SSH_ERR_INVALID_FORMAT;
418		goto out;
419	}
420	if ((r = sshbuf_get(msg, response, 16)) != 0)
421		goto out;
422	r = 0;
423 out:
424	sshbuf_free(msg);
425	return r;
426}
427#endif
428
429/* encode signature algoritm in flag bits, so we can keep the msg format */
430static u_int
431agent_encode_alg(struct sshkey *key, const char *alg)
432{
433	if (alg != NULL && key->type == KEY_RSA) {
434		if (strcmp(alg, "rsa-sha2-256") == 0)
435			return SSH_AGENT_RSA_SHA2_256;
436		else if (strcmp(alg, "rsa-sha2-512") == 0)
437			return SSH_AGENT_RSA_SHA2_512;
438	}
439	return 0;
440}
441
442/* ask agent to sign data, returns err.h code on error, 0 on success */
443int
444ssh_agent_sign(int sock, struct sshkey *key,
445    u_char **sigp, size_t *lenp,
446    const u_char *data, size_t datalen, const char *alg, u_int compat)
447{
448	struct sshbuf *msg;
449	u_char *blob = NULL, type;
450	size_t blen = 0, len = 0;
451	u_int flags = 0;
452	int r = SSH_ERR_INTERNAL_ERROR;
453
454	*sigp = NULL;
455	*lenp = 0;
456
457	if (datalen > SSH_KEY_MAX_SIGN_DATA_SIZE)
458		return SSH_ERR_INVALID_ARGUMENT;
459	if (compat & SSH_BUG_SIGBLOB)
460		flags |= SSH_AGENT_OLD_SIGNATURE;
461	if ((msg = sshbuf_new()) == NULL)
462		return SSH_ERR_ALLOC_FAIL;
463	if ((r = sshkey_to_blob(key, &blob, &blen)) != 0)
464		goto out;
465	flags |= agent_encode_alg(key, alg);
466	if ((r = sshbuf_put_u8(msg, SSH2_AGENTC_SIGN_REQUEST)) != 0 ||
467	    (r = sshbuf_put_string(msg, blob, blen)) != 0 ||
468	    (r = sshbuf_put_string(msg, data, datalen)) != 0 ||
469	    (r = sshbuf_put_u32(msg, flags)) != 0)
470		goto out;
471	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
472		goto out;
473	if ((r = sshbuf_get_u8(msg, &type)) != 0)
474		goto out;
475	if (agent_failed(type)) {
476		r = SSH_ERR_AGENT_FAILURE;
477		goto out;
478	} else if (type != SSH2_AGENT_SIGN_RESPONSE) {
479		r = SSH_ERR_INVALID_FORMAT;
480		goto out;
481	}
482	if ((r = sshbuf_get_string(msg, sigp, &len)) != 0)
483		goto out;
484	*lenp = len;
485	r = 0;
486 out:
487	if (blob != NULL) {
488		explicit_bzero(blob, blen);
489		free(blob);
490	}
491	sshbuf_free(msg);
492	return r;
493}
494
495/* Encode key for a message to the agent. */
496
497#ifdef WITH_SSH1
498static int
499ssh_encode_identity_rsa1(struct sshbuf *b, RSA *key, const char *comment)
500{
501	int r;
502
503	/* To keep within the protocol: p < q for ssh. in SSL p > q */
504	if ((r = sshbuf_put_u32(b, BN_num_bits(key->n))) != 0 ||
505	    (r = sshbuf_put_bignum1(b, key->n)) != 0 ||
506	    (r = sshbuf_put_bignum1(b, key->e)) != 0 ||
507	    (r = sshbuf_put_bignum1(b, key->d)) != 0 ||
508	    (r = sshbuf_put_bignum1(b, key->iqmp)) != 0 ||
509	    (r = sshbuf_put_bignum1(b, key->q)) != 0 ||
510	    (r = sshbuf_put_bignum1(b, key->p)) != 0 ||
511	    (r = sshbuf_put_cstring(b, comment)) != 0)
512		return r;
513	return 0;
514}
515#endif
516
517static int
518ssh_encode_identity_ssh2(struct sshbuf *b, struct sshkey *key,
519    const char *comment)
520{
521	int r;
522
523	if ((r = sshkey_private_serialize(key, b)) != 0 ||
524	    (r = sshbuf_put_cstring(b, comment)) != 0)
525		return r;
526	return 0;
527}
528
529static int
530encode_constraints(struct sshbuf *m, u_int life, u_int confirm)
531{
532	int r;
533
534	if (life != 0) {
535		if ((r = sshbuf_put_u8(m, SSH_AGENT_CONSTRAIN_LIFETIME)) != 0 ||
536		    (r = sshbuf_put_u32(m, life)) != 0)
537			goto out;
538	}
539	if (confirm != 0) {
540		if ((r = sshbuf_put_u8(m, SSH_AGENT_CONSTRAIN_CONFIRM)) != 0)
541			goto out;
542	}
543	r = 0;
544 out:
545	return r;
546}
547
548/*
549 * Adds an identity to the authentication server.
550 * This call is intended only for use by ssh-add(1) and like applications.
551 */
552int
553ssh_add_identity_constrained(int sock, struct sshkey *key, const char *comment,
554    u_int life, u_int confirm)
555{
556	struct sshbuf *msg;
557	int r, constrained = (life || confirm);
558	u_char type;
559
560	if ((msg = sshbuf_new()) == NULL)
561		return SSH_ERR_ALLOC_FAIL;
562
563	switch (key->type) {
564#ifdef WITH_SSH1
565	case KEY_RSA1:
566		type = constrained ?
567		    SSH_AGENTC_ADD_RSA_ID_CONSTRAINED :
568		    SSH_AGENTC_ADD_RSA_IDENTITY;
569		if ((r = sshbuf_put_u8(msg, type)) != 0 ||
570		    (r = ssh_encode_identity_rsa1(msg, key->rsa, comment)) != 0)
571			goto out;
572		break;
573#endif
574#ifdef WITH_OPENSSL
575	case KEY_RSA:
576	case KEY_RSA_CERT:
577	case KEY_DSA:
578	case KEY_DSA_CERT:
579	case KEY_ECDSA:
580	case KEY_ECDSA_CERT:
581#endif
582	case KEY_ED25519:
583	case KEY_ED25519_CERT:
584		type = constrained ?
585		    SSH2_AGENTC_ADD_ID_CONSTRAINED :
586		    SSH2_AGENTC_ADD_IDENTITY;
587		if ((r = sshbuf_put_u8(msg, type)) != 0 ||
588		    (r = ssh_encode_identity_ssh2(msg, key, comment)) != 0)
589			goto out;
590		break;
591	default:
592		r = SSH_ERR_INVALID_ARGUMENT;
593		goto out;
594	}
595	if (constrained &&
596	    (r = encode_constraints(msg, life, confirm)) != 0)
597		goto out;
598	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
599		goto out;
600	if ((r = sshbuf_get_u8(msg, &type)) != 0)
601		goto out;
602	r = decode_reply(type);
603 out:
604	sshbuf_free(msg);
605	return r;
606}
607
608/*
609 * Removes an identity from the authentication server.
610 * This call is intended only for use by ssh-add(1) and like applications.
611 */
612int
613ssh_remove_identity(int sock, struct sshkey *key)
614{
615	struct sshbuf *msg;
616	int r;
617	u_char type, *blob = NULL;
618	size_t blen;
619
620	if ((msg = sshbuf_new()) == NULL)
621		return SSH_ERR_ALLOC_FAIL;
622
623#ifdef WITH_SSH1
624	if (key->type == KEY_RSA1) {
625		if ((r = sshbuf_put_u8(msg,
626		    SSH_AGENTC_REMOVE_RSA_IDENTITY)) != 0 ||
627		    (r = sshbuf_put_u32(msg, BN_num_bits(key->rsa->n))) != 0 ||
628		    (r = sshbuf_put_bignum1(msg, key->rsa->e)) != 0 ||
629		    (r = sshbuf_put_bignum1(msg, key->rsa->n)) != 0)
630			goto out;
631	} else
632#endif
633	if (key->type != KEY_UNSPEC) {
634		if ((r = sshkey_to_blob(key, &blob, &blen)) != 0)
635			goto out;
636		if ((r = sshbuf_put_u8(msg,
637		    SSH2_AGENTC_REMOVE_IDENTITY)) != 0 ||
638		    (r = sshbuf_put_string(msg, blob, blen)) != 0)
639			goto out;
640	} else {
641		r = SSH_ERR_INVALID_ARGUMENT;
642		goto out;
643	}
644	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
645		goto out;
646	if ((r = sshbuf_get_u8(msg, &type)) != 0)
647		goto out;
648	r = decode_reply(type);
649 out:
650	if (blob != NULL) {
651		explicit_bzero(blob, blen);
652		free(blob);
653	}
654	sshbuf_free(msg);
655	return r;
656}
657
658/*
659 * Add/remove an token-based identity from the authentication server.
660 * This call is intended only for use by ssh-add(1) and like applications.
661 */
662int
663ssh_update_card(int sock, int add, const char *reader_id, const char *pin,
664    u_int life, u_int confirm)
665{
666	struct sshbuf *msg;
667	int r, constrained = (life || confirm);
668	u_char type;
669
670	if (add) {
671		type = constrained ?
672		    SSH_AGENTC_ADD_SMARTCARD_KEY_CONSTRAINED :
673		    SSH_AGENTC_ADD_SMARTCARD_KEY;
674	} else
675		type = SSH_AGENTC_REMOVE_SMARTCARD_KEY;
676
677	if ((msg = sshbuf_new()) == NULL)
678		return SSH_ERR_ALLOC_FAIL;
679	if ((r = sshbuf_put_u8(msg, type)) != 0 ||
680	    (r = sshbuf_put_cstring(msg, reader_id)) != 0 ||
681	    (r = sshbuf_put_cstring(msg, pin)) != 0)
682		goto out;
683	if (constrained &&
684	    (r = encode_constraints(msg, life, confirm)) != 0)
685		goto out;
686	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
687		goto out;
688	if ((r = sshbuf_get_u8(msg, &type)) != 0)
689		goto out;
690	r = decode_reply(type);
691 out:
692	sshbuf_free(msg);
693	return r;
694}
695
696/*
697 * Removes all identities from the agent.
698 * This call is intended only for use by ssh-add(1) and like applications.
699 */
700int
701ssh_remove_all_identities(int sock, int version)
702{
703	struct sshbuf *msg;
704	u_char type = (version == 1) ?
705	    SSH_AGENTC_REMOVE_ALL_RSA_IDENTITIES :
706	    SSH2_AGENTC_REMOVE_ALL_IDENTITIES;
707	int r;
708
709	if ((msg = sshbuf_new()) == NULL)
710		return SSH_ERR_ALLOC_FAIL;
711	if ((r = sshbuf_put_u8(msg, type)) != 0)
712		goto out;
713	if ((r = ssh_request_reply(sock, msg, msg)) != 0)
714		goto out;
715	if ((r = sshbuf_get_u8(msg, &type)) != 0)
716		goto out;
717	r = decode_reply(type);
718 out:
719	sshbuf_free(msg);
720	return r;
721}
722