xref: /freebsd/crypto/openssh/ssh-agent.c (revision 19261079)
1 /* $OpenBSD: ssh-agent.c,v 1.278 2021/04/03 06:18:41 djm 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  * The authentication agent program.
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  * Copyright (c) 2000, 2001 Markus Friedl.  All rights reserved.
15  *
16  * Redistribution and use in source and binary forms, with or without
17  * modification, are permitted provided that the following conditions
18  * are met:
19  * 1. Redistributions of source code must retain the above copyright
20  *    notice, this list of conditions and the following disclaimer.
21  * 2. Redistributions in binary form must reproduce the above copyright
22  *    notice, this list of conditions and the following disclaimer in the
23  *    documentation and/or other materials provided with the distribution.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
26  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
27  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
28  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
29  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
30  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
34  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35  */
36 
37 #include "includes.h"
38 __RCSID("$FreeBSD$");
39 
40 #include <sys/types.h>
41 #include <sys/param.h>
42 #include <sys/resource.h>
43 #include <sys/stat.h>
44 #include <sys/socket.h>
45 #include <sys/wait.h>
46 #ifdef HAVE_SYS_TIME_H
47 # include <sys/time.h>
48 #endif
49 #ifdef HAVE_SYS_UN_H
50 # include <sys/un.h>
51 #endif
52 #include "openbsd-compat/sys-queue.h"
53 
54 #ifdef WITH_OPENSSL
55 #include <openssl/evp.h>
56 #include "openbsd-compat/openssl-compat.h"
57 #endif
58 
59 #include <errno.h>
60 #include <fcntl.h>
61 #include <limits.h>
62 #ifdef HAVE_PATHS_H
63 # include <paths.h>
64 #endif
65 #ifdef HAVE_POLL_H
66 # include <poll.h>
67 #endif
68 #include <signal.h>
69 #include <stdarg.h>
70 #include <stdio.h>
71 #include <stdlib.h>
72 #include <time.h>
73 #include <string.h>
74 #include <unistd.h>
75 #ifdef HAVE_UTIL_H
76 # include <util.h>
77 #endif
78 
79 #include "xmalloc.h"
80 #include "ssh.h"
81 #include "ssh2.h"
82 #include "sshbuf.h"
83 #include "sshkey.h"
84 #include "authfd.h"
85 #include "compat.h"
86 #include "log.h"
87 #include "misc.h"
88 #include "digest.h"
89 #include "ssherr.h"
90 #include "match.h"
91 #include "msg.h"
92 #include "ssherr.h"
93 #include "pathnames.h"
94 #include "ssh-pkcs11.h"
95 #include "sk-api.h"
96 
97 #ifndef DEFAULT_ALLOWED_PROVIDERS
98 # define DEFAULT_ALLOWED_PROVIDERS "/usr/lib*/*,/usr/local/lib*/*"
99 #endif
100 
101 /* Maximum accepted message length */
102 #define AGENT_MAX_LEN	(256*1024)
103 /* Maximum bytes to read from client socket */
104 #define AGENT_RBUF_LEN	(4096)
105 
106 typedef enum {
107 	AUTH_UNUSED = 0,
108 	AUTH_SOCKET = 1,
109 	AUTH_CONNECTION = 2,
110 } sock_type;
111 
112 typedef struct socket_entry {
113 	int fd;
114 	sock_type type;
115 	struct sshbuf *input;
116 	struct sshbuf *output;
117 	struct sshbuf *request;
118 } SocketEntry;
119 
120 u_int sockets_alloc = 0;
121 SocketEntry *sockets = NULL;
122 
123 typedef struct identity {
124 	TAILQ_ENTRY(identity) next;
125 	struct sshkey *key;
126 	char *comment;
127 	char *provider;
128 	time_t death;
129 	u_int confirm;
130 	char *sk_provider;
131 } Identity;
132 
133 struct idtable {
134 	int nentries;
135 	TAILQ_HEAD(idqueue, identity) idlist;
136 };
137 
138 /* private key table */
139 struct idtable *idtab;
140 
141 int max_fd = 0;
142 
143 /* pid of shell == parent of agent */
144 pid_t parent_pid = -1;
145 time_t parent_alive_interval = 0;
146 
147 /* pid of process for which cleanup_socket is applicable */
148 pid_t cleanup_pid = 0;
149 
150 /* pathname and directory for AUTH_SOCKET */
151 char socket_name[PATH_MAX];
152 char socket_dir[PATH_MAX];
153 
154 /* Pattern-list of allowed PKCS#11/Security key paths */
155 static char *allowed_providers;
156 
157 /* locking */
158 #define LOCK_SIZE	32
159 #define LOCK_SALT_SIZE	16
160 #define LOCK_ROUNDS	1
161 int locked = 0;
162 u_char lock_pwhash[LOCK_SIZE];
163 u_char lock_salt[LOCK_SALT_SIZE];
164 
165 extern char *__progname;
166 
167 /* Default lifetime in seconds (0 == forever) */
168 static int lifetime = 0;
169 
170 static int fingerprint_hash = SSH_FP_HASH_DEFAULT;
171 
172 /* Refuse signing of non-SSH messages for web-origin FIDO keys */
173 static int restrict_websafe = 1;
174 
175 /*
176  * Client connection count; incremented in new_socket() and decremented in
177  * close_socket().  When it reaches 0, ssh-agent will exit.  Since it is
178  * normally initialized to 1, it will never reach 0.  However, if the -x
179  * option is specified, it is initialized to 0 in main(); in that case,
180  * ssh-agent will exit as soon as it has had at least one client but no
181  * longer has any.
182  */
183 static int xcount = 1;
184 
185 static void
186 close_socket(SocketEntry *e)
187 {
188 	int last = 0;
189 
190 	if (e->type == AUTH_CONNECTION) {
191 		debug("xcount %d -> %d", xcount, xcount - 1);
192 		if (--xcount == 0)
193 			last = 1;
194 	}
195 	close(e->fd);
196 	sshbuf_free(e->input);
197 	sshbuf_free(e->output);
198 	sshbuf_free(e->request);
199 	memset(e, '\0', sizeof(*e));
200 	e->fd = -1;
201 	e->type = AUTH_UNUSED;
202 	if (last)
203 		cleanup_exit(0);
204 }
205 
206 static void
207 idtab_init(void)
208 {
209 	idtab = xcalloc(1, sizeof(*idtab));
210 	TAILQ_INIT(&idtab->idlist);
211 	idtab->nentries = 0;
212 }
213 
214 static void
215 free_identity(Identity *id)
216 {
217 	sshkey_free(id->key);
218 	free(id->provider);
219 	free(id->comment);
220 	free(id->sk_provider);
221 	free(id);
222 }
223 
224 /* return matching private key for given public key */
225 static Identity *
226 lookup_identity(struct sshkey *key)
227 {
228 	Identity *id;
229 
230 	TAILQ_FOREACH(id, &idtab->idlist, next) {
231 		if (sshkey_equal(key, id->key))
232 			return (id);
233 	}
234 	return (NULL);
235 }
236 
237 /* Check confirmation of keysign request */
238 static int
239 confirm_key(Identity *id, const char *extra)
240 {
241 	char *p;
242 	int ret = -1;
243 
244 	p = sshkey_fingerprint(id->key, fingerprint_hash, SSH_FP_DEFAULT);
245 	if (p != NULL &&
246 	    ask_permission("Allow use of key %s?\nKey fingerprint %s.%s%s",
247 	    id->comment, p,
248 	    extra == NULL ? "" : "\n", extra == NULL ? "" : extra))
249 		ret = 0;
250 	free(p);
251 
252 	return (ret);
253 }
254 
255 static void
256 send_status(SocketEntry *e, int success)
257 {
258 	int r;
259 
260 	if ((r = sshbuf_put_u32(e->output, 1)) != 0 ||
261 	    (r = sshbuf_put_u8(e->output, success ?
262 	    SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE)) != 0)
263 		fatal_fr(r, "compose");
264 }
265 
266 /* send list of supported public keys to 'client' */
267 static void
268 process_request_identities(SocketEntry *e)
269 {
270 	Identity *id;
271 	struct sshbuf *msg;
272 	int r;
273 
274 	debug2_f("entering");
275 
276 	if ((msg = sshbuf_new()) == NULL)
277 		fatal_f("sshbuf_new failed");
278 	if ((r = sshbuf_put_u8(msg, SSH2_AGENT_IDENTITIES_ANSWER)) != 0 ||
279 	    (r = sshbuf_put_u32(msg, idtab->nentries)) != 0)
280 		fatal_fr(r, "compose");
281 	TAILQ_FOREACH(id, &idtab->idlist, next) {
282 		if ((r = sshkey_puts_opts(id->key, msg,
283 		    SSHKEY_SERIALIZE_INFO)) != 0 ||
284 		    (r = sshbuf_put_cstring(msg, id->comment)) != 0) {
285 			error_fr(r, "compose key/comment");
286 			continue;
287 		}
288 	}
289 	if ((r = sshbuf_put_stringb(e->output, msg)) != 0)
290 		fatal_fr(r, "enqueue");
291 	sshbuf_free(msg);
292 }
293 
294 
295 static char *
296 agent_decode_alg(struct sshkey *key, u_int flags)
297 {
298 	if (key->type == KEY_RSA) {
299 		if (flags & SSH_AGENT_RSA_SHA2_256)
300 			return "rsa-sha2-256";
301 		else if (flags & SSH_AGENT_RSA_SHA2_512)
302 			return "rsa-sha2-512";
303 	} else if (key->type == KEY_RSA_CERT) {
304 		if (flags & SSH_AGENT_RSA_SHA2_256)
305 			return "rsa-sha2-256-cert-v01@openssh.com";
306 		else if (flags & SSH_AGENT_RSA_SHA2_512)
307 			return "rsa-sha2-512-cert-v01@openssh.com";
308 	}
309 	return NULL;
310 }
311 
312 /*
313  * Attempt to parse the contents of a buffer as a SSH publickey userauth
314  * request, checking its contents for consistency and matching the embedded
315  * key against the one that is being used for signing.
316  * Note: does not modify msg buffer.
317  * Optionally extract the username and session ID from the request.
318  */
319 static int
320 parse_userauth_request(struct sshbuf *msg, const struct sshkey *expected_key,
321     char **userp, struct sshbuf **sess_idp)
322 {
323 	struct sshbuf *b = NULL, *sess_id = NULL;
324 	char *user = NULL, *service = NULL, *method = NULL, *pkalg = NULL;
325 	int r;
326 	u_char t, sig_follows;
327 	struct sshkey *mkey = NULL;
328 
329 	if (userp != NULL)
330 		*userp = NULL;
331 	if (sess_idp != NULL)
332 		*sess_idp = NULL;
333 	if ((b = sshbuf_fromb(msg)) == NULL)
334 		fatal_f("sshbuf_fromb");
335 
336 	/* SSH userauth request */
337 	if ((r = sshbuf_froms(b, &sess_id)) != 0)
338 		goto out;
339 	if (sshbuf_len(sess_id) == 0) {
340 		r = SSH_ERR_INVALID_FORMAT;
341 		goto out;
342 	}
343 	if ((r = sshbuf_get_u8(b, &t)) != 0 || /* SSH2_MSG_USERAUTH_REQUEST */
344 	    (r = sshbuf_get_cstring(b, &user, NULL)) != 0 || /* server user */
345 	    (r = sshbuf_get_cstring(b, &service, NULL)) != 0 || /* service */
346 	    (r = sshbuf_get_cstring(b, &method, NULL)) != 0 || /* method */
347 	    (r = sshbuf_get_u8(b, &sig_follows)) != 0 || /* sig-follows */
348 	    (r = sshbuf_get_cstring(b, &pkalg, NULL)) != 0 || /* alg */
349 	    (r = sshkey_froms(b, &mkey)) != 0) /* key */
350 		goto out;
351 	if (t != SSH2_MSG_USERAUTH_REQUEST ||
352 	    sig_follows != 1 ||
353 	    strcmp(service, "ssh-connection") != 0 ||
354 	    !sshkey_equal(expected_key, mkey) ||
355 	    sshkey_type_from_name(pkalg) != expected_key->type) {
356 		r = SSH_ERR_INVALID_FORMAT;
357 		goto out;
358 	}
359 	if (strcmp(method, "publickey") != 0) {
360 		r = SSH_ERR_INVALID_FORMAT;
361 		goto out;
362 	}
363 	if (sshbuf_len(b) != 0) {
364 		r = SSH_ERR_INVALID_FORMAT;
365 		goto out;
366 	}
367 	/* success */
368 	r = 0;
369 	debug3_f("well formed userauth");
370 	if (userp != NULL) {
371 		*userp = user;
372 		user = NULL;
373 	}
374 	if (sess_idp != NULL) {
375 		*sess_idp = sess_id;
376 		sess_id = NULL;
377 	}
378  out:
379 	sshbuf_free(b);
380 	sshbuf_free(sess_id);
381 	free(user);
382 	free(service);
383 	free(method);
384 	free(pkalg);
385 	sshkey_free(mkey);
386 	return r;
387 }
388 
389 /*
390  * Attempt to parse the contents of a buffer as a SSHSIG signature request.
391  * Note: does not modify buffer.
392  */
393 static int
394 parse_sshsig_request(struct sshbuf *msg)
395 {
396 	int r;
397 	struct sshbuf *b;
398 
399 	if ((b = sshbuf_fromb(msg)) == NULL)
400 		fatal_f("sshbuf_fromb");
401 
402 	if ((r = sshbuf_cmp(b, 0, "SSHSIG", 6)) != 0 ||
403 	    (r = sshbuf_consume(b, 6)) != 0 ||
404 	    (r = sshbuf_get_cstring(b, NULL, NULL)) != 0 || /* namespace */
405 	    (r = sshbuf_get_string_direct(b, NULL, NULL)) != 0 || /* reserved */
406 	    (r = sshbuf_get_cstring(b, NULL, NULL)) != 0 || /* hashalg */
407 	    (r = sshbuf_get_string_direct(b, NULL, NULL)) != 0) /* H(msg) */
408 		goto out;
409 	if (sshbuf_len(b) != 0) {
410 		r = SSH_ERR_INVALID_FORMAT;
411 		goto out;
412 	}
413 	/* success */
414 	r = 0;
415  out:
416 	sshbuf_free(b);
417 	return r;
418 }
419 
420 /*
421  * This function inspects a message to be signed by a FIDO key that has a
422  * web-like application string (i.e. one that does not begin with "ssh:".
423  * It checks that the message is one of those expected for SSH operations
424  * (pubkey userauth, sshsig, CA key signing) to exclude signing challenges
425  * for the web.
426  */
427 static int
428 check_websafe_message_contents(struct sshkey *key, struct sshbuf *data)
429 {
430 	if (parse_userauth_request(data, key, NULL, NULL) == 0) {
431 		debug_f("signed data matches public key userauth request");
432 		return 1;
433 	}
434 	if (parse_sshsig_request(data) == 0) {
435 		debug_f("signed data matches SSHSIG signature request");
436 		return 1;
437 	}
438 
439 	/* XXX check CA signature operation */
440 
441 	error("web-origin key attempting to sign non-SSH message");
442 	return 0;
443 }
444 
445 /* ssh2 only */
446 static void
447 process_sign_request2(SocketEntry *e)
448 {
449 	u_char *signature = NULL;
450 	size_t slen = 0;
451 	u_int compat = 0, flags;
452 	int r, ok = -1;
453 	char *fp = NULL;
454 	struct sshbuf *msg = NULL, *data = NULL;
455 	struct sshkey *key = NULL;
456 	struct identity *id;
457 	struct notifier_ctx *notifier = NULL;
458 
459 	debug_f("entering");
460 
461 	if ((msg = sshbuf_new()) == NULL || (data = sshbuf_new()) == NULL)
462 		fatal_f("sshbuf_new failed");
463 	if ((r = sshkey_froms(e->request, &key)) != 0 ||
464 	    (r = sshbuf_get_stringb(e->request, data)) != 0 ||
465 	    (r = sshbuf_get_u32(e->request, &flags)) != 0) {
466 		error_fr(r, "parse");
467 		goto send;
468 	}
469 
470 	if ((id = lookup_identity(key)) == NULL) {
471 		verbose_f("%s key not found", sshkey_type(key));
472 		goto send;
473 	}
474 	if (id->confirm && confirm_key(id, NULL) != 0) {
475 		verbose_f("user refused key");
476 		goto send;
477 	}
478 	if (sshkey_is_sk(id->key)) {
479 		if (strncmp(id->key->sk_application, "ssh:", 4) != 0 &&
480 		    !check_websafe_message_contents(key, data)) {
481 			/* error already logged */
482 			goto send;
483 		}
484 		if ((id->key->sk_flags & SSH_SK_USER_PRESENCE_REQD)) {
485 			if ((fp = sshkey_fingerprint(key, SSH_FP_HASH_DEFAULT,
486 			    SSH_FP_DEFAULT)) == NULL)
487 				fatal_f("fingerprint failed");
488 			notifier = notify_start(0,
489 			    "Confirm user presence for key %s %s",
490 			    sshkey_type(id->key), fp);
491 		}
492 	}
493 	/* XXX support PIN required FIDO keys */
494 	if ((r = sshkey_sign(id->key, &signature, &slen,
495 	    sshbuf_ptr(data), sshbuf_len(data), agent_decode_alg(key, flags),
496 	    id->sk_provider, NULL, compat)) != 0) {
497 		error_fr(r, "sshkey_sign");
498 		goto send;
499 	}
500 	/* Success */
501 	ok = 0;
502  send:
503 	notify_complete(notifier, "User presence confirmed");
504 
505 	if (ok == 0) {
506 		if ((r = sshbuf_put_u8(msg, SSH2_AGENT_SIGN_RESPONSE)) != 0 ||
507 		    (r = sshbuf_put_string(msg, signature, slen)) != 0)
508 			fatal_fr(r, "compose");
509 	} else if ((r = sshbuf_put_u8(msg, SSH_AGENT_FAILURE)) != 0)
510 		fatal_fr(r, "compose failure");
511 
512 	if ((r = sshbuf_put_stringb(e->output, msg)) != 0)
513 		fatal_fr(r, "enqueue");
514 
515 	sshbuf_free(data);
516 	sshbuf_free(msg);
517 	sshkey_free(key);
518 	free(fp);
519 	free(signature);
520 }
521 
522 /* shared */
523 static void
524 process_remove_identity(SocketEntry *e)
525 {
526 	int r, success = 0;
527 	struct sshkey *key = NULL;
528 	Identity *id;
529 
530 	debug2_f("entering");
531 	if ((r = sshkey_froms(e->request, &key)) != 0) {
532 		error_fr(r, "parse key");
533 		goto done;
534 	}
535 	if ((id = lookup_identity(key)) == NULL) {
536 		debug_f("key not found");
537 		goto done;
538 	}
539 	/* We have this key, free it. */
540 	if (idtab->nentries < 1)
541 		fatal_f("internal error: nentries %d", idtab->nentries);
542 	TAILQ_REMOVE(&idtab->idlist, id, next);
543 	free_identity(id);
544 	idtab->nentries--;
545 	success = 1;
546  done:
547 	sshkey_free(key);
548 	send_status(e, success);
549 }
550 
551 static void
552 process_remove_all_identities(SocketEntry *e)
553 {
554 	Identity *id;
555 
556 	debug2_f("entering");
557 	/* Loop over all identities and clear the keys. */
558 	for (id = TAILQ_FIRST(&idtab->idlist); id;
559 	    id = TAILQ_FIRST(&idtab->idlist)) {
560 		TAILQ_REMOVE(&idtab->idlist, id, next);
561 		free_identity(id);
562 	}
563 
564 	/* Mark that there are no identities. */
565 	idtab->nentries = 0;
566 
567 	/* Send success. */
568 	send_status(e, 1);
569 }
570 
571 /* removes expired keys and returns number of seconds until the next expiry */
572 static time_t
573 reaper(void)
574 {
575 	time_t deadline = 0, now = monotime();
576 	Identity *id, *nxt;
577 
578 	for (id = TAILQ_FIRST(&idtab->idlist); id; id = nxt) {
579 		nxt = TAILQ_NEXT(id, next);
580 		if (id->death == 0)
581 			continue;
582 		if (now >= id->death) {
583 			debug("expiring key '%s'", id->comment);
584 			TAILQ_REMOVE(&idtab->idlist, id, next);
585 			free_identity(id);
586 			idtab->nentries--;
587 		} else
588 			deadline = (deadline == 0) ? id->death :
589 			    MINIMUM(deadline, id->death);
590 	}
591 	if (deadline == 0 || deadline <= now)
592 		return 0;
593 	else
594 		return (deadline - now);
595 }
596 
597 static int
598 parse_key_constraint_extension(struct sshbuf *m, char **sk_providerp)
599 {
600 	char *ext_name = NULL;
601 	int r;
602 
603 	if ((r = sshbuf_get_cstring(m, &ext_name, NULL)) != 0) {
604 		error_fr(r, "parse constraint extension");
605 		goto out;
606 	}
607 	debug_f("constraint ext %s", ext_name);
608 	if (strcmp(ext_name, "sk-provider@openssh.com") == 0) {
609 		if (sk_providerp == NULL) {
610 			error_f("%s not valid here", ext_name);
611 			r = SSH_ERR_INVALID_FORMAT;
612 			goto out;
613 		}
614 		if (*sk_providerp != NULL) {
615 			error_f("%s already set", ext_name);
616 			r = SSH_ERR_INVALID_FORMAT;
617 			goto out;
618 		}
619 		if ((r = sshbuf_get_cstring(m, sk_providerp, NULL)) != 0) {
620 			error_fr(r, "parse %s", ext_name);
621 			goto out;
622 		}
623 	} else {
624 		error_f("unsupported constraint \"%s\"", ext_name);
625 		r = SSH_ERR_FEATURE_UNSUPPORTED;
626 		goto out;
627 	}
628 	/* success */
629 	r = 0;
630  out:
631 	free(ext_name);
632 	return r;
633 }
634 
635 static int
636 parse_key_constraints(struct sshbuf *m, struct sshkey *k, time_t *deathp,
637     u_int *secondsp, int *confirmp, char **sk_providerp)
638 {
639 	u_char ctype;
640 	int r;
641 	u_int seconds, maxsign = 0;
642 
643 	while (sshbuf_len(m)) {
644 		if ((r = sshbuf_get_u8(m, &ctype)) != 0) {
645 			error_fr(r, "parse constraint type");
646 			goto out;
647 		}
648 		switch (ctype) {
649 		case SSH_AGENT_CONSTRAIN_LIFETIME:
650 			if (*deathp != 0) {
651 				error_f("lifetime already set");
652 				r = SSH_ERR_INVALID_FORMAT;
653 				goto out;
654 			}
655 			if ((r = sshbuf_get_u32(m, &seconds)) != 0) {
656 				error_fr(r, "parse lifetime constraint");
657 				goto out;
658 			}
659 			*deathp = monotime() + seconds;
660 			*secondsp = seconds;
661 			break;
662 		case SSH_AGENT_CONSTRAIN_CONFIRM:
663 			if (*confirmp != 0) {
664 				error_f("confirm already set");
665 				r = SSH_ERR_INVALID_FORMAT;
666 				goto out;
667 			}
668 			*confirmp = 1;
669 			break;
670 		case SSH_AGENT_CONSTRAIN_MAXSIGN:
671 			if (k == NULL) {
672 				error_f("maxsign not valid here");
673 				r = SSH_ERR_INVALID_FORMAT;
674 				goto out;
675 			}
676 			if (maxsign != 0) {
677 				error_f("maxsign already set");
678 				r = SSH_ERR_INVALID_FORMAT;
679 				goto out;
680 			}
681 			if ((r = sshbuf_get_u32(m, &maxsign)) != 0) {
682 				error_fr(r, "parse maxsign constraint");
683 				goto out;
684 			}
685 			if ((r = sshkey_enable_maxsign(k, maxsign)) != 0) {
686 				error_fr(r, "enable maxsign");
687 				goto out;
688 			}
689 			break;
690 		case SSH_AGENT_CONSTRAIN_EXTENSION:
691 			if ((r = parse_key_constraint_extension(m,
692 			    sk_providerp)) != 0)
693 				goto out; /* error already logged */
694 			break;
695 		default:
696 			error_f("Unknown constraint %d", ctype);
697 			r = SSH_ERR_FEATURE_UNSUPPORTED;
698 			goto out;
699 		}
700 	}
701 	/* success */
702 	r = 0;
703  out:
704 	return r;
705 }
706 
707 static void
708 process_add_identity(SocketEntry *e)
709 {
710 	Identity *id;
711 	int success = 0, confirm = 0;
712 	char *fp, *comment = NULL, *sk_provider = NULL;
713 	char canonical_provider[PATH_MAX];
714 	time_t death = 0;
715 	u_int seconds = 0;
716 	struct sshkey *k = NULL;
717 	int r = SSH_ERR_INTERNAL_ERROR;
718 
719 	debug2_f("entering");
720 	if ((r = sshkey_private_deserialize(e->request, &k)) != 0 ||
721 	    k == NULL ||
722 	    (r = sshbuf_get_cstring(e->request, &comment, NULL)) != 0) {
723 		error_fr(r, "parse");
724 		goto out;
725 	}
726 	if (parse_key_constraints(e->request, k, &death, &seconds, &confirm,
727 	    &sk_provider) != 0) {
728 		error_f("failed to parse constraints");
729 		sshbuf_reset(e->request);
730 		goto out;
731 	}
732 
733 	if (sk_provider != NULL) {
734 		if (!sshkey_is_sk(k)) {
735 			error("Cannot add provider: %s is not an "
736 			    "authenticator-hosted key", sshkey_type(k));
737 			goto out;
738 		}
739 		if (strcasecmp(sk_provider, "internal") == 0) {
740 			debug_f("internal provider");
741 		} else {
742 			if (realpath(sk_provider, canonical_provider) == NULL) {
743 				verbose("failed provider \"%.100s\": "
744 				    "realpath: %s", sk_provider,
745 				    strerror(errno));
746 				goto out;
747 			}
748 			free(sk_provider);
749 			sk_provider = xstrdup(canonical_provider);
750 			if (match_pattern_list(sk_provider,
751 			    allowed_providers, 0) != 1) {
752 				error("Refusing add key: "
753 				    "provider %s not allowed", sk_provider);
754 				goto out;
755 			}
756 		}
757 	}
758 	if ((r = sshkey_shield_private(k)) != 0) {
759 		error_fr(r, "shield private");
760 		goto out;
761 	}
762 	if (lifetime && !death)
763 		death = monotime() + lifetime;
764 	if ((id = lookup_identity(k)) == NULL) {
765 		id = xcalloc(1, sizeof(Identity));
766 		TAILQ_INSERT_TAIL(&idtab->idlist, id, next);
767 		/* Increment the number of identities. */
768 		idtab->nentries++;
769 	} else {
770 		/* key state might have been updated */
771 		sshkey_free(id->key);
772 		free(id->comment);
773 		free(id->sk_provider);
774 	}
775 	/* success */
776 	id->key = k;
777 	id->comment = comment;
778 	id->death = death;
779 	id->confirm = confirm;
780 	id->sk_provider = sk_provider;
781 
782 	if ((fp = sshkey_fingerprint(k, SSH_FP_HASH_DEFAULT,
783 	    SSH_FP_DEFAULT)) == NULL)
784 		fatal_f("sshkey_fingerprint failed");
785 	debug_f("add %s %s \"%.100s\" (life: %u) (confirm: %u) "
786 	    "(provider: %s)", sshkey_ssh_name(k), fp, comment, seconds,
787 	    confirm, sk_provider == NULL ? "none" : sk_provider);
788 	free(fp);
789 	/* transferred */
790 	k = NULL;
791 	comment = NULL;
792 	sk_provider = NULL;
793 	success = 1;
794  out:
795 	free(sk_provider);
796 	free(comment);
797 	sshkey_free(k);
798 	send_status(e, success);
799 }
800 
801 /* XXX todo: encrypt sensitive data with passphrase */
802 static void
803 process_lock_agent(SocketEntry *e, int lock)
804 {
805 	int r, success = 0, delay;
806 	char *passwd;
807 	u_char passwdhash[LOCK_SIZE];
808 	static u_int fail_count = 0;
809 	size_t pwlen;
810 
811 	debug2_f("entering");
812 	/*
813 	 * This is deliberately fatal: the user has requested that we lock,
814 	 * but we can't parse their request properly. The only safe thing to
815 	 * do is abort.
816 	 */
817 	if ((r = sshbuf_get_cstring(e->request, &passwd, &pwlen)) != 0)
818 		fatal_fr(r, "parse");
819 	if (pwlen == 0) {
820 		debug("empty password not supported");
821 	} else if (locked && !lock) {
822 		if (bcrypt_pbkdf(passwd, pwlen, lock_salt, sizeof(lock_salt),
823 		    passwdhash, sizeof(passwdhash), LOCK_ROUNDS) < 0)
824 			fatal("bcrypt_pbkdf");
825 		if (timingsafe_bcmp(passwdhash, lock_pwhash, LOCK_SIZE) == 0) {
826 			debug("agent unlocked");
827 			locked = 0;
828 			fail_count = 0;
829 			explicit_bzero(lock_pwhash, sizeof(lock_pwhash));
830 			success = 1;
831 		} else {
832 			/* delay in 0.1s increments up to 10s */
833 			if (fail_count < 100)
834 				fail_count++;
835 			delay = 100000 * fail_count;
836 			debug("unlock failed, delaying %0.1lf seconds",
837 			    (double)delay/1000000);
838 			usleep(delay);
839 		}
840 		explicit_bzero(passwdhash, sizeof(passwdhash));
841 	} else if (!locked && lock) {
842 		debug("agent locked");
843 		locked = 1;
844 		arc4random_buf(lock_salt, sizeof(lock_salt));
845 		if (bcrypt_pbkdf(passwd, pwlen, lock_salt, sizeof(lock_salt),
846 		    lock_pwhash, sizeof(lock_pwhash), LOCK_ROUNDS) < 0)
847 			fatal("bcrypt_pbkdf");
848 		success = 1;
849 	}
850 	freezero(passwd, pwlen);
851 	send_status(e, success);
852 }
853 
854 static void
855 no_identities(SocketEntry *e)
856 {
857 	struct sshbuf *msg;
858 	int r;
859 
860 	if ((msg = sshbuf_new()) == NULL)
861 		fatal_f("sshbuf_new failed");
862 	if ((r = sshbuf_put_u8(msg, SSH2_AGENT_IDENTITIES_ANSWER)) != 0 ||
863 	    (r = sshbuf_put_u32(msg, 0)) != 0 ||
864 	    (r = sshbuf_put_stringb(e->output, msg)) != 0)
865 		fatal_fr(r, "compose");
866 	sshbuf_free(msg);
867 }
868 
869 #ifdef ENABLE_PKCS11
870 static void
871 process_add_smartcard_key(SocketEntry *e)
872 {
873 	char *provider = NULL, *pin = NULL, canonical_provider[PATH_MAX];
874 	char **comments = NULL;
875 	int r, i, count = 0, success = 0, confirm = 0;
876 	u_int seconds = 0;
877 	time_t death = 0;
878 	struct sshkey **keys = NULL, *k;
879 	Identity *id;
880 
881 	debug2_f("entering");
882 	if ((r = sshbuf_get_cstring(e->request, &provider, NULL)) != 0 ||
883 	    (r = sshbuf_get_cstring(e->request, &pin, NULL)) != 0) {
884 		error_fr(r, "parse");
885 		goto send;
886 	}
887 	if (parse_key_constraints(e->request, NULL, &death, &seconds, &confirm,
888 	    NULL) != 0) {
889 		error_f("failed to parse constraints");
890 		goto send;
891 	}
892 	if (realpath(provider, canonical_provider) == NULL) {
893 		verbose("failed PKCS#11 add of \"%.100s\": realpath: %s",
894 		    provider, strerror(errno));
895 		goto send;
896 	}
897 	if (match_pattern_list(canonical_provider, allowed_providers, 0) != 1) {
898 		verbose("refusing PKCS#11 add of \"%.100s\": "
899 		    "provider not allowed", canonical_provider);
900 		goto send;
901 	}
902 	debug_f("add %.100s", canonical_provider);
903 	if (lifetime && !death)
904 		death = monotime() + lifetime;
905 
906 	count = pkcs11_add_provider(canonical_provider, pin, &keys, &comments);
907 	for (i = 0; i < count; i++) {
908 		k = keys[i];
909 		if (lookup_identity(k) == NULL) {
910 			id = xcalloc(1, sizeof(Identity));
911 			id->key = k;
912 			keys[i] = NULL; /* transferred */
913 			id->provider = xstrdup(canonical_provider);
914 			if (*comments[i] != '\0') {
915 				id->comment = comments[i];
916 				comments[i] = NULL; /* transferred */
917 			} else {
918 				id->comment = xstrdup(canonical_provider);
919 			}
920 			id->death = death;
921 			id->confirm = confirm;
922 			TAILQ_INSERT_TAIL(&idtab->idlist, id, next);
923 			idtab->nentries++;
924 			success = 1;
925 		}
926 		/* XXX update constraints for existing keys */
927 		sshkey_free(keys[i]);
928 		free(comments[i]);
929 	}
930 send:
931 	free(pin);
932 	free(provider);
933 	free(keys);
934 	free(comments);
935 	send_status(e, success);
936 }
937 
938 static void
939 process_remove_smartcard_key(SocketEntry *e)
940 {
941 	char *provider = NULL, *pin = NULL, canonical_provider[PATH_MAX];
942 	int r, success = 0;
943 	Identity *id, *nxt;
944 
945 	debug2_f("entering");
946 	if ((r = sshbuf_get_cstring(e->request, &provider, NULL)) != 0 ||
947 	    (r = sshbuf_get_cstring(e->request, &pin, NULL)) != 0) {
948 		error_fr(r, "parse");
949 		goto send;
950 	}
951 	free(pin);
952 
953 	if (realpath(provider, canonical_provider) == NULL) {
954 		verbose("failed PKCS#11 add of \"%.100s\": realpath: %s",
955 		    provider, strerror(errno));
956 		goto send;
957 	}
958 
959 	debug_f("remove %.100s", canonical_provider);
960 	for (id = TAILQ_FIRST(&idtab->idlist); id; id = nxt) {
961 		nxt = TAILQ_NEXT(id, next);
962 		/* Skip file--based keys */
963 		if (id->provider == NULL)
964 			continue;
965 		if (!strcmp(canonical_provider, id->provider)) {
966 			TAILQ_REMOVE(&idtab->idlist, id, next);
967 			free_identity(id);
968 			idtab->nentries--;
969 		}
970 	}
971 	if (pkcs11_del_provider(canonical_provider) == 0)
972 		success = 1;
973 	else
974 		error_f("pkcs11_del_provider failed");
975 send:
976 	free(provider);
977 	send_status(e, success);
978 }
979 #endif /* ENABLE_PKCS11 */
980 
981 /*
982  * dispatch incoming message.
983  * returns 1 on success, 0 for incomplete messages or -1 on error.
984  */
985 static int
986 process_message(u_int socknum)
987 {
988 	u_int msg_len;
989 	u_char type;
990 	const u_char *cp;
991 	int r;
992 	SocketEntry *e;
993 
994 	if (socknum >= sockets_alloc)
995 		fatal_f("sock %u >= allocated %u", socknum, sockets_alloc);
996 	e = &sockets[socknum];
997 
998 	if (sshbuf_len(e->input) < 5)
999 		return 0;		/* Incomplete message header. */
1000 	cp = sshbuf_ptr(e->input);
1001 	msg_len = PEEK_U32(cp);
1002 	if (msg_len > AGENT_MAX_LEN) {
1003 		debug_f("socket %u (fd=%d) message too long %u > %u",
1004 		    socknum, e->fd, msg_len, AGENT_MAX_LEN);
1005 		return -1;
1006 	}
1007 	if (sshbuf_len(e->input) < msg_len + 4)
1008 		return 0;		/* Incomplete message body. */
1009 
1010 	/* move the current input to e->request */
1011 	sshbuf_reset(e->request);
1012 	if ((r = sshbuf_get_stringb(e->input, e->request)) != 0 ||
1013 	    (r = sshbuf_get_u8(e->request, &type)) != 0) {
1014 		if (r == SSH_ERR_MESSAGE_INCOMPLETE ||
1015 		    r == SSH_ERR_STRING_TOO_LARGE) {
1016 			error_fr(r, "parse");
1017 			return -1;
1018 		}
1019 		fatal_fr(r, "parse");
1020 	}
1021 
1022 	debug_f("socket %u (fd=%d) type %d", socknum, e->fd, type);
1023 
1024 	/* check whether agent is locked */
1025 	if (locked && type != SSH_AGENTC_UNLOCK) {
1026 		sshbuf_reset(e->request);
1027 		switch (type) {
1028 		case SSH2_AGENTC_REQUEST_IDENTITIES:
1029 			/* send empty lists */
1030 			no_identities(e);
1031 			break;
1032 		default:
1033 			/* send a fail message for all other request types */
1034 			send_status(e, 0);
1035 		}
1036 		return 1;
1037 	}
1038 
1039 	switch (type) {
1040 	case SSH_AGENTC_LOCK:
1041 	case SSH_AGENTC_UNLOCK:
1042 		process_lock_agent(e, type == SSH_AGENTC_LOCK);
1043 		break;
1044 	case SSH_AGENTC_REMOVE_ALL_RSA_IDENTITIES:
1045 		process_remove_all_identities(e); /* safe for !WITH_SSH1 */
1046 		break;
1047 	/* ssh2 */
1048 	case SSH2_AGENTC_SIGN_REQUEST:
1049 		process_sign_request2(e);
1050 		break;
1051 	case SSH2_AGENTC_REQUEST_IDENTITIES:
1052 		process_request_identities(e);
1053 		break;
1054 	case SSH2_AGENTC_ADD_IDENTITY:
1055 	case SSH2_AGENTC_ADD_ID_CONSTRAINED:
1056 		process_add_identity(e);
1057 		break;
1058 	case SSH2_AGENTC_REMOVE_IDENTITY:
1059 		process_remove_identity(e);
1060 		break;
1061 	case SSH2_AGENTC_REMOVE_ALL_IDENTITIES:
1062 		process_remove_all_identities(e);
1063 		break;
1064 #ifdef ENABLE_PKCS11
1065 	case SSH_AGENTC_ADD_SMARTCARD_KEY:
1066 	case SSH_AGENTC_ADD_SMARTCARD_KEY_CONSTRAINED:
1067 		process_add_smartcard_key(e);
1068 		break;
1069 	case SSH_AGENTC_REMOVE_SMARTCARD_KEY:
1070 		process_remove_smartcard_key(e);
1071 		break;
1072 #endif /* ENABLE_PKCS11 */
1073 	default:
1074 		/* Unknown message.  Respond with failure. */
1075 		error("Unknown message %d", type);
1076 		sshbuf_reset(e->request);
1077 		send_status(e, 0);
1078 		break;
1079 	}
1080 	return 1;
1081 }
1082 
1083 static void
1084 new_socket(sock_type type, int fd)
1085 {
1086 	u_int i, old_alloc, new_alloc;
1087 
1088 	debug_f("type = %s", type == AUTH_CONNECTION ? "CONNECTION" :
1089 	    (type == AUTH_SOCKET ? "SOCKET" : "UNKNOWN"));
1090 	if (type == AUTH_CONNECTION) {
1091 		debug("xcount %d -> %d", xcount, xcount + 1);
1092 		++xcount;
1093 	}
1094 	set_nonblock(fd);
1095 
1096 	if (fd > max_fd)
1097 		max_fd = fd;
1098 
1099 	for (i = 0; i < sockets_alloc; i++)
1100 		if (sockets[i].type == AUTH_UNUSED) {
1101 			sockets[i].fd = fd;
1102 			if ((sockets[i].input = sshbuf_new()) == NULL ||
1103 			    (sockets[i].output = sshbuf_new()) == NULL ||
1104 			    (sockets[i].request = sshbuf_new()) == NULL)
1105 				fatal_f("sshbuf_new failed");
1106 			sockets[i].type = type;
1107 			return;
1108 		}
1109 	old_alloc = sockets_alloc;
1110 	new_alloc = sockets_alloc + 10;
1111 	sockets = xrecallocarray(sockets, old_alloc, new_alloc,
1112 	    sizeof(sockets[0]));
1113 	for (i = old_alloc; i < new_alloc; i++)
1114 		sockets[i].type = AUTH_UNUSED;
1115 	sockets_alloc = new_alloc;
1116 	sockets[old_alloc].fd = fd;
1117 	if ((sockets[old_alloc].input = sshbuf_new()) == NULL ||
1118 	    (sockets[old_alloc].output = sshbuf_new()) == NULL ||
1119 	    (sockets[old_alloc].request = sshbuf_new()) == NULL)
1120 		fatal_f("sshbuf_new failed");
1121 	sockets[old_alloc].type = type;
1122 }
1123 
1124 static int
1125 handle_socket_read(u_int socknum)
1126 {
1127 	struct sockaddr_un sunaddr;
1128 	socklen_t slen;
1129 	uid_t euid;
1130 	gid_t egid;
1131 	int fd;
1132 
1133 	slen = sizeof(sunaddr);
1134 	fd = accept(sockets[socknum].fd, (struct sockaddr *)&sunaddr, &slen);
1135 	if (fd == -1) {
1136 		error("accept from AUTH_SOCKET: %s", strerror(errno));
1137 		return -1;
1138 	}
1139 	if (getpeereid(fd, &euid, &egid) == -1) {
1140 		error("getpeereid %d failed: %s", fd, strerror(errno));
1141 		close(fd);
1142 		return -1;
1143 	}
1144 	if ((euid != 0) && (getuid() != euid)) {
1145 		error("uid mismatch: peer euid %u != uid %u",
1146 		    (u_int) euid, (u_int) getuid());
1147 		close(fd);
1148 		return -1;
1149 	}
1150 	new_socket(AUTH_CONNECTION, fd);
1151 	return 0;
1152 }
1153 
1154 static int
1155 handle_conn_read(u_int socknum)
1156 {
1157 	char buf[AGENT_RBUF_LEN];
1158 	ssize_t len;
1159 	int r;
1160 
1161 	if ((len = read(sockets[socknum].fd, buf, sizeof(buf))) <= 0) {
1162 		if (len == -1) {
1163 			if (errno == EAGAIN || errno == EINTR)
1164 				return 0;
1165 			error_f("read error on socket %u (fd %d): %s",
1166 			    socknum, sockets[socknum].fd, strerror(errno));
1167 		}
1168 		return -1;
1169 	}
1170 	if ((r = sshbuf_put(sockets[socknum].input, buf, len)) != 0)
1171 		fatal_fr(r, "compose");
1172 	explicit_bzero(buf, sizeof(buf));
1173 	for (;;) {
1174 		if ((r = process_message(socknum)) == -1)
1175 			return -1;
1176 		else if (r == 0)
1177 			break;
1178 	}
1179 	return 0;
1180 }
1181 
1182 static int
1183 handle_conn_write(u_int socknum)
1184 {
1185 	ssize_t len;
1186 	int r;
1187 
1188 	if (sshbuf_len(sockets[socknum].output) == 0)
1189 		return 0; /* shouldn't happen */
1190 	if ((len = write(sockets[socknum].fd,
1191 	    sshbuf_ptr(sockets[socknum].output),
1192 	    sshbuf_len(sockets[socknum].output))) <= 0) {
1193 		if (len == -1) {
1194 			if (errno == EAGAIN || errno == EINTR)
1195 				return 0;
1196 			error_f("read error on socket %u (fd %d): %s",
1197 			    socknum, sockets[socknum].fd, strerror(errno));
1198 		}
1199 		return -1;
1200 	}
1201 	if ((r = sshbuf_consume(sockets[socknum].output, len)) != 0)
1202 		fatal_fr(r, "consume");
1203 	return 0;
1204 }
1205 
1206 static void
1207 after_poll(struct pollfd *pfd, size_t npfd, u_int maxfds)
1208 {
1209 	size_t i;
1210 	u_int socknum, activefds = npfd;
1211 
1212 	for (i = 0; i < npfd; i++) {
1213 		if (pfd[i].revents == 0)
1214 			continue;
1215 		/* Find sockets entry */
1216 		for (socknum = 0; socknum < sockets_alloc; socknum++) {
1217 			if (sockets[socknum].type != AUTH_SOCKET &&
1218 			    sockets[socknum].type != AUTH_CONNECTION)
1219 				continue;
1220 			if (pfd[i].fd == sockets[socknum].fd)
1221 				break;
1222 		}
1223 		if (socknum >= sockets_alloc) {
1224 			error_f("no socket for fd %d", pfd[i].fd);
1225 			continue;
1226 		}
1227 		/* Process events */
1228 		switch (sockets[socknum].type) {
1229 		case AUTH_SOCKET:
1230 			if ((pfd[i].revents & (POLLIN|POLLERR)) == 0)
1231 				break;
1232 			if (npfd > maxfds) {
1233 				debug3("out of fds (active %u >= limit %u); "
1234 				    "skipping accept", activefds, maxfds);
1235 				break;
1236 			}
1237 			if (handle_socket_read(socknum) == 0)
1238 				activefds++;
1239 			break;
1240 		case AUTH_CONNECTION:
1241 			if ((pfd[i].revents & (POLLIN|POLLERR)) != 0 &&
1242 			    handle_conn_read(socknum) != 0) {
1243 				goto close_sock;
1244 			}
1245 			if ((pfd[i].revents & (POLLOUT|POLLHUP)) != 0 &&
1246 			    handle_conn_write(socknum) != 0) {
1247  close_sock:
1248 				if (activefds == 0)
1249 					fatal("activefds == 0 at close_sock");
1250 				close_socket(&sockets[socknum]);
1251 				activefds--;
1252 				break;
1253 			}
1254 			break;
1255 		default:
1256 			break;
1257 		}
1258 	}
1259 }
1260 
1261 static int
1262 prepare_poll(struct pollfd **pfdp, size_t *npfdp, int *timeoutp, u_int maxfds)
1263 {
1264 	struct pollfd *pfd = *pfdp;
1265 	size_t i, j, npfd = 0;
1266 	time_t deadline;
1267 	int r;
1268 
1269 	/* Count active sockets */
1270 	for (i = 0; i < sockets_alloc; i++) {
1271 		switch (sockets[i].type) {
1272 		case AUTH_SOCKET:
1273 		case AUTH_CONNECTION:
1274 			npfd++;
1275 			break;
1276 		case AUTH_UNUSED:
1277 			break;
1278 		default:
1279 			fatal("Unknown socket type %d", sockets[i].type);
1280 			break;
1281 		}
1282 	}
1283 	if (npfd != *npfdp &&
1284 	    (pfd = recallocarray(pfd, *npfdp, npfd, sizeof(*pfd))) == NULL)
1285 		fatal_f("recallocarray failed");
1286 	*pfdp = pfd;
1287 	*npfdp = npfd;
1288 
1289 	for (i = j = 0; i < sockets_alloc; i++) {
1290 		switch (sockets[i].type) {
1291 		case AUTH_SOCKET:
1292 			if (npfd > maxfds) {
1293 				debug3("out of fds (active %zu >= limit %u); "
1294 				    "skipping arming listener", npfd, maxfds);
1295 				break;
1296 			}
1297 			pfd[j].fd = sockets[i].fd;
1298 			pfd[j].revents = 0;
1299 			pfd[j].events = POLLIN;
1300 			j++;
1301 			break;
1302 		case AUTH_CONNECTION:
1303 			pfd[j].fd = sockets[i].fd;
1304 			pfd[j].revents = 0;
1305 			/*
1306 			 * Only prepare to read if we can handle a full-size
1307 			 * input read buffer and enqueue a max size reply..
1308 			 */
1309 			if ((r = sshbuf_check_reserve(sockets[i].input,
1310 			    AGENT_RBUF_LEN)) == 0 &&
1311 			    (r = sshbuf_check_reserve(sockets[i].output,
1312 			    AGENT_MAX_LEN)) == 0)
1313 				pfd[j].events = POLLIN;
1314 			else if (r != SSH_ERR_NO_BUFFER_SPACE)
1315 				fatal_fr(r, "reserve");
1316 			if (sshbuf_len(sockets[i].output) > 0)
1317 				pfd[j].events |= POLLOUT;
1318 			j++;
1319 			break;
1320 		default:
1321 			break;
1322 		}
1323 	}
1324 	deadline = reaper();
1325 	if (parent_alive_interval != 0)
1326 		deadline = (deadline == 0) ? parent_alive_interval :
1327 		    MINIMUM(deadline, parent_alive_interval);
1328 	if (deadline == 0) {
1329 		*timeoutp = -1; /* INFTIM */
1330 	} else {
1331 		if (deadline > INT_MAX / 1000)
1332 			*timeoutp = INT_MAX / 1000;
1333 		else
1334 			*timeoutp = deadline * 1000;
1335 	}
1336 	return (1);
1337 }
1338 
1339 static void
1340 cleanup_socket(void)
1341 {
1342 	if (cleanup_pid != 0 && getpid() != cleanup_pid)
1343 		return;
1344 	debug_f("cleanup");
1345 	if (socket_name[0])
1346 		unlink(socket_name);
1347 	if (socket_dir[0])
1348 		rmdir(socket_dir);
1349 }
1350 
1351 void
1352 cleanup_exit(int i)
1353 {
1354 	cleanup_socket();
1355 	_exit(i);
1356 }
1357 
1358 /*ARGSUSED*/
1359 static void
1360 cleanup_handler(int sig)
1361 {
1362 	cleanup_socket();
1363 #ifdef ENABLE_PKCS11
1364 	pkcs11_terminate();
1365 #endif
1366 	_exit(2);
1367 }
1368 
1369 static void
1370 check_parent_exists(void)
1371 {
1372 	/*
1373 	 * If our parent has exited then getppid() will return (pid_t)1,
1374 	 * so testing for that should be safe.
1375 	 */
1376 	if (parent_pid != -1 && getppid() != parent_pid) {
1377 		/* printf("Parent has died - Authentication agent exiting.\n"); */
1378 		cleanup_socket();
1379 		_exit(2);
1380 	}
1381 }
1382 
1383 static void
1384 usage(void)
1385 {
1386 	fprintf(stderr,
1387 	    "usage: ssh-agent [-c | -s] [-Ddx] [-a bind_address] [-E fingerprint_hash]\n"
1388 	    "                 [-P allowed_providers] [-t life]\n"
1389 	    "       ssh-agent [-a bind_address] [-E fingerprint_hash] [-P allowed_providers]\n"
1390 	    "                 [-t life] command [arg ...]\n"
1391 	    "       ssh-agent [-c | -s] -k\n");
1392 	exit(1);
1393 }
1394 
1395 int
1396 main(int ac, char **av)
1397 {
1398 	int c_flag = 0, d_flag = 0, D_flag = 0, k_flag = 0, s_flag = 0;
1399 	int sock, ch, result, saved_errno;
1400 	char *shell, *format, *pidstr, *agentsocket = NULL;
1401 #ifdef HAVE_SETRLIMIT
1402 	struct rlimit rlim;
1403 #endif
1404 	extern int optind;
1405 	extern char *optarg;
1406 	pid_t pid;
1407 	char pidstrbuf[1 + 3 * sizeof pid];
1408 	size_t len;
1409 	mode_t prev_mask;
1410 	int timeout = -1; /* INFTIM */
1411 	struct pollfd *pfd = NULL;
1412 	size_t npfd = 0;
1413 	u_int maxfds;
1414 
1415 	/* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
1416 	sanitise_stdfd();
1417 
1418 	/* drop */
1419 	setegid(getgid());
1420 	setgid(getgid());
1421 	setuid(geteuid());
1422 
1423 	platform_disable_tracing(0);	/* strict=no */
1424 
1425 #ifdef RLIMIT_NOFILE
1426 	if (getrlimit(RLIMIT_NOFILE, &rlim) == -1)
1427 		fatal("%s: getrlimit: %s", __progname, strerror(errno));
1428 #endif
1429 
1430 	__progname = ssh_get_progname(av[0]);
1431 	seed_rng();
1432 
1433 	while ((ch = getopt(ac, av, "cDdksE:a:O:P:t:x")) != -1) {
1434 		switch (ch) {
1435 		case 'E':
1436 			fingerprint_hash = ssh_digest_alg_by_name(optarg);
1437 			if (fingerprint_hash == -1)
1438 				fatal("Invalid hash algorithm \"%s\"", optarg);
1439 			break;
1440 		case 'c':
1441 			if (s_flag)
1442 				usage();
1443 			c_flag++;
1444 			break;
1445 		case 'k':
1446 			k_flag++;
1447 			break;
1448 		case 'O':
1449 			if (strcmp(optarg, "no-restrict-websafe") == 0)
1450 				restrict_websafe  = 0;
1451 			else
1452 				fatal("Unknown -O option");
1453 			break;
1454 		case 'P':
1455 			if (allowed_providers != NULL)
1456 				fatal("-P option already specified");
1457 			allowed_providers = xstrdup(optarg);
1458 			break;
1459 		case 's':
1460 			if (c_flag)
1461 				usage();
1462 			s_flag++;
1463 			break;
1464 		case 'd':
1465 			if (d_flag || D_flag)
1466 				usage();
1467 			d_flag++;
1468 			break;
1469 		case 'D':
1470 			if (d_flag || D_flag)
1471 				usage();
1472 			D_flag++;
1473 			break;
1474 		case 'a':
1475 			agentsocket = optarg;
1476 			break;
1477 		case 't':
1478 			if ((lifetime = convtime(optarg)) == -1) {
1479 				fprintf(stderr, "Invalid lifetime\n");
1480 				usage();
1481 			}
1482 			break;
1483 		case 'x':
1484 			xcount = 0;
1485 			break;
1486 		default:
1487 			usage();
1488 		}
1489 	}
1490 	ac -= optind;
1491 	av += optind;
1492 
1493 	if (ac > 0 && (c_flag || k_flag || s_flag || d_flag || D_flag))
1494 		usage();
1495 
1496 	if (allowed_providers == NULL)
1497 		allowed_providers = xstrdup(DEFAULT_ALLOWED_PROVIDERS);
1498 
1499 	if (ac == 0 && !c_flag && !s_flag) {
1500 		shell = getenv("SHELL");
1501 		if (shell != NULL && (len = strlen(shell)) > 2 &&
1502 		    strncmp(shell + len - 3, "csh", 3) == 0)
1503 			c_flag = 1;
1504 	}
1505 	if (k_flag) {
1506 		const char *errstr = NULL;
1507 
1508 		pidstr = getenv(SSH_AGENTPID_ENV_NAME);
1509 		if (pidstr == NULL) {
1510 			fprintf(stderr, "%s not set, cannot kill agent\n",
1511 			    SSH_AGENTPID_ENV_NAME);
1512 			exit(1);
1513 		}
1514 		pid = (int)strtonum(pidstr, 2, INT_MAX, &errstr);
1515 		if (errstr) {
1516 			fprintf(stderr,
1517 			    "%s=\"%s\", which is not a good PID: %s\n",
1518 			    SSH_AGENTPID_ENV_NAME, pidstr, errstr);
1519 			exit(1);
1520 		}
1521 		if (kill(pid, SIGTERM) == -1) {
1522 			perror("kill");
1523 			exit(1);
1524 		}
1525 		format = c_flag ? "unsetenv %s;\n" : "unset %s;\n";
1526 		printf(format, SSH_AUTHSOCKET_ENV_NAME);
1527 		printf(format, SSH_AGENTPID_ENV_NAME);
1528 		printf("echo Agent pid %ld killed;\n", (long)pid);
1529 		exit(0);
1530 	}
1531 
1532 	/*
1533 	 * Minimum file descriptors:
1534 	 * stdio (3) + listener (1) + syslog (1 maybe) + connection (1) +
1535 	 * a few spare for libc / stack protectors / sanitisers, etc.
1536 	 */
1537 #define SSH_AGENT_MIN_FDS (3+1+1+1+4)
1538 	if (rlim.rlim_cur < SSH_AGENT_MIN_FDS)
1539 		fatal("%s: file descriptor rlimit %lld too low (minimum %u)",
1540 		    __progname, (long long)rlim.rlim_cur, SSH_AGENT_MIN_FDS);
1541 	maxfds = rlim.rlim_cur - SSH_AGENT_MIN_FDS;
1542 
1543 	parent_pid = getpid();
1544 
1545 	if (agentsocket == NULL) {
1546 		/* Create private directory for agent socket */
1547 		mktemp_proto(socket_dir, sizeof(socket_dir));
1548 		if (mkdtemp(socket_dir) == NULL) {
1549 			perror("mkdtemp: private socket dir");
1550 			exit(1);
1551 		}
1552 		snprintf(socket_name, sizeof socket_name, "%s/agent.%ld", socket_dir,
1553 		    (long)parent_pid);
1554 	} else {
1555 		/* Try to use specified agent socket */
1556 		socket_dir[0] = '\0';
1557 		strlcpy(socket_name, agentsocket, sizeof socket_name);
1558 	}
1559 
1560 	/*
1561 	 * Create socket early so it will exist before command gets run from
1562 	 * the parent.
1563 	 */
1564 	prev_mask = umask(0177);
1565 	sock = unix_listener(socket_name, SSH_LISTEN_BACKLOG, 0);
1566 	if (sock < 0) {
1567 		/* XXX - unix_listener() calls error() not perror() */
1568 		*socket_name = '\0'; /* Don't unlink any existing file */
1569 		cleanup_exit(1);
1570 	}
1571 	umask(prev_mask);
1572 
1573 	/*
1574 	 * Fork, and have the parent execute the command, if any, or present
1575 	 * the socket data.  The child continues as the authentication agent.
1576 	 */
1577 	if (D_flag || d_flag) {
1578 		log_init(__progname,
1579 		    d_flag ? SYSLOG_LEVEL_DEBUG3 : SYSLOG_LEVEL_INFO,
1580 		    SYSLOG_FACILITY_AUTH, 1);
1581 		format = c_flag ? "setenv %s %s;\n" : "%s=%s; export %s;\n";
1582 		printf(format, SSH_AUTHSOCKET_ENV_NAME, socket_name,
1583 		    SSH_AUTHSOCKET_ENV_NAME);
1584 		printf("echo Agent pid %ld;\n", (long)parent_pid);
1585 		fflush(stdout);
1586 		goto skip;
1587 	}
1588 	pid = fork();
1589 	if (pid == -1) {
1590 		perror("fork");
1591 		cleanup_exit(1);
1592 	}
1593 	if (pid != 0) {		/* Parent - execute the given command. */
1594 		close(sock);
1595 		snprintf(pidstrbuf, sizeof pidstrbuf, "%ld", (long)pid);
1596 		if (ac == 0) {
1597 			format = c_flag ? "setenv %s %s;\n" : "%s=%s; export %s;\n";
1598 			printf(format, SSH_AUTHSOCKET_ENV_NAME, socket_name,
1599 			    SSH_AUTHSOCKET_ENV_NAME);
1600 			printf(format, SSH_AGENTPID_ENV_NAME, pidstrbuf,
1601 			    SSH_AGENTPID_ENV_NAME);
1602 			printf("echo Agent pid %ld;\n", (long)pid);
1603 			exit(0);
1604 		}
1605 		if (setenv(SSH_AUTHSOCKET_ENV_NAME, socket_name, 1) == -1 ||
1606 		    setenv(SSH_AGENTPID_ENV_NAME, pidstrbuf, 1) == -1) {
1607 			perror("setenv");
1608 			exit(1);
1609 		}
1610 		execvp(av[0], av);
1611 		perror(av[0]);
1612 		exit(1);
1613 	}
1614 	/* child */
1615 	log_init(__progname, SYSLOG_LEVEL_INFO, SYSLOG_FACILITY_AUTH, 0);
1616 
1617 	if (setsid() == -1) {
1618 		error("setsid: %s", strerror(errno));
1619 		cleanup_exit(1);
1620 	}
1621 
1622 	(void)chdir("/");
1623 	if (stdfd_devnull(1, 1, 1) == -1)
1624 		error_f("stdfd_devnull failed");
1625 
1626 #ifdef HAVE_SETRLIMIT
1627 	/* deny core dumps, since memory contains unencrypted private keys */
1628 	rlim.rlim_cur = rlim.rlim_max = 0;
1629 	if (setrlimit(RLIMIT_CORE, &rlim) == -1) {
1630 		error("setrlimit RLIMIT_CORE: %s", strerror(errno));
1631 		cleanup_exit(1);
1632 	}
1633 #endif
1634 
1635 skip:
1636 
1637 	cleanup_pid = getpid();
1638 
1639 #ifdef ENABLE_PKCS11
1640 	pkcs11_init(0);
1641 #endif
1642 	new_socket(AUTH_SOCKET, sock);
1643 	if (ac > 0)
1644 		parent_alive_interval = 10;
1645 	idtab_init();
1646 	ssh_signal(SIGPIPE, SIG_IGN);
1647 	ssh_signal(SIGINT, (d_flag | D_flag) ? cleanup_handler : SIG_IGN);
1648 	ssh_signal(SIGHUP, cleanup_handler);
1649 	ssh_signal(SIGTERM, cleanup_handler);
1650 
1651 	if (pledge("stdio rpath cpath unix id proc exec", NULL) == -1)
1652 		fatal("%s: pledge: %s", __progname, strerror(errno));
1653 	platform_pledge_agent();
1654 
1655 	while (1) {
1656 		prepare_poll(&pfd, &npfd, &timeout, maxfds);
1657 		result = poll(pfd, npfd, timeout);
1658 		saved_errno = errno;
1659 		if (parent_alive_interval != 0)
1660 			check_parent_exists();
1661 		(void) reaper();	/* remove expired keys */
1662 		if (result == -1) {
1663 			if (saved_errno == EINTR)
1664 				continue;
1665 			fatal("poll: %s", strerror(saved_errno));
1666 		} else if (result > 0)
1667 			after_poll(pfd, npfd, maxfds);
1668 	}
1669 	/* NOTREACHED */
1670 }
1671