blob: b45087629cfe646d4ad828ea357c84bfd47f829c [file] [log] [blame]
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001/*
Damien Miller95def091999-11-25 00:26:21 +11002 * Author: Tatu Ylonen <ylo@cs.hut.fi>
3 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
4 * All rights reserved
Damien Miller95def091999-11-25 00:26:21 +11005 * The authentication agent program.
Damien Millerad833b32000-08-23 10:46:23 +10006 *
Damien Millere4340be2000-09-16 13:29:08 +11007 * As far as I am concerned, the code I have written for this software
8 * can be used freely for any purpose. Any derived versions of this
9 * software must be clearly marked as such, and if the derived work is
10 * incompatible with the protocol description in the RFC file, it must be
11 * called by a name other than "ssh" or "Secure Shell".
12 *
Ben Lindstrom44697232001-07-04 03:32:30 +000013 * Copyright (c) 2000, 2001 Markus Friedl. All rights reserved.
Damien Millere4340be2000-09-16 13:29:08 +110014 *
15 * Redistribution and use in source and binary forms, with or without
16 * modification, are permitted provided that the following conditions
17 * are met:
18 * 1. Redistributions of source code must retain the above copyright
19 * notice, this list of conditions and the following disclaimer.
20 * 2. Redistributions in binary form must reproduce the above copyright
21 * notice, this list of conditions and the following disclaimer in the
22 * documentation and/or other materials provided with the distribution.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
25 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
26 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
27 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
28 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
29 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
33 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Damien Miller95def091999-11-25 00:26:21 +110034 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +100035
36#include "includes.h"
Damien Miller6ff3cad2006-03-15 11:52:09 +110037RCSID("$OpenBSD: ssh-agent.c,v 1.129 2006/02/20 17:02:44 stevesk Exp $");
Damien Miller574c41f2006-03-15 11:40:10 +110038
39#include <sys/types.h>
40#ifdef HAVE_SYS_UN_H
41# include <sys/un.h>
42#endif
Damien Miller9b481512002-09-12 10:43:29 +100043#include "openbsd-compat/sys-queue.h"
Damien Millercd4223c2006-03-15 11:22:47 +110044#include <sys/resource.h>
Damien Miller03e20032006-03-15 11:16:59 +110045#ifdef HAVE_PATHS_H
Damien Millera9263d02006-03-15 11:18:26 +110046# include <paths.h>
Damien Miller03e20032006-03-15 11:16:59 +110047#endif
Damien Miller6ff3cad2006-03-15 11:52:09 +110048#include <signal.h>
49
Ben Lindstrom226cfa02001-01-22 05:34:40 +000050#include <openssl/evp.h>
51#include <openssl/md5.h>
Damien Millerd4a8b7e1999-10-27 13:42:43 +100052
53#include "ssh.h"
54#include "rsa.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100055#include "buffer.h"
56#include "bufaux.h"
57#include "xmalloc.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100058#include "getput.h"
Damien Miller994cf142000-07-21 10:19:44 +100059#include "key.h"
60#include "authfd.h"
Damien Miller62cee002000-09-23 17:15:56 +110061#include "compat.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000062#include "log.h"
Damien Miller6c711792003-01-24 11:36:23 +110063#include "misc.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100064
Ben Lindstrom3f471632001-07-04 03:53:15 +000065#ifdef SMARTCARD
Ben Lindstrom3f471632001-07-04 03:53:15 +000066#include "scard.h"
Ben Lindstrombcc18082001-08-06 21:59:25 +000067#endif
Ben Lindstrom3f471632001-07-04 03:53:15 +000068
Damien Miller6c4914a2004-03-03 11:08:59 +110069#if defined(HAVE_SYS_PRCTL_H)
70#include <sys/prctl.h> /* For prctl() and PR_SET_DUMPABLE */
71#endif
72
Ben Lindstrom65366a82001-12-06 16:32:47 +000073typedef enum {
74 AUTH_UNUSED,
75 AUTH_SOCKET,
76 AUTH_CONNECTION
77} sock_type;
78
Damien Miller95def091999-11-25 00:26:21 +110079typedef struct {
80 int fd;
Ben Lindstrom65366a82001-12-06 16:32:47 +000081 sock_type type;
Damien Miller95def091999-11-25 00:26:21 +110082 Buffer input;
83 Buffer output;
Ben Lindstrom21d1ed82002-06-06 21:48:57 +000084 Buffer request;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100085} SocketEntry;
86
Ben Lindstrom46c16222000-12-22 01:43:59 +000087u_int sockets_alloc = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100088SocketEntry *sockets = NULL;
89
Damien Miller1a534ae2002-01-22 23:26:13 +110090typedef struct identity {
91 TAILQ_ENTRY(identity) next;
Damien Millerad833b32000-08-23 10:46:23 +100092 Key *key;
Damien Miller95def091999-11-25 00:26:21 +110093 char *comment;
Ben Lindstrom61d328a2002-06-06 21:54:57 +000094 u_int death;
Damien Miller6c711792003-01-24 11:36:23 +110095 u_int confirm;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100096} Identity;
97
Damien Millerad833b32000-08-23 10:46:23 +100098typedef struct {
99 int nentries;
Damien Miller1a534ae2002-01-22 23:26:13 +1100100 TAILQ_HEAD(idqueue, identity) idlist;
Damien Millerad833b32000-08-23 10:46:23 +1000101} Idtab;
102
103/* private key table, one per protocol version */
104Idtab idtable[3];
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000105
106int max_fd = 0;
107
108/* pid of shell == parent of agent */
Damien Miller166fca82000-04-20 07:42:21 +1000109pid_t parent_pid = -1;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000110
111/* pathname and directory for AUTH_SOCKET */
112char socket_name[1024];
113char socket_dir[1024];
114
Ben Lindstrom2f717042002-06-06 21:52:03 +0000115/* locking */
116int locked = 0;
117char *lock_passwd = NULL;
118
Damien Miller95def091999-11-25 00:26:21 +1100119extern char *__progname;
Damien Miller95def091999-11-25 00:26:21 +1100120
Damien Miller53d81482003-01-22 11:47:19 +1100121/* Default lifetime (0 == forever) */
122static int lifetime = 0;
123
Ben Lindstrombba81212001-06-25 05:01:22 +0000124static void
Damien Miller58f34862002-09-04 16:31:21 +1000125close_socket(SocketEntry *e)
126{
Damien Miller58f34862002-09-04 16:31:21 +1000127 close(e->fd);
128 e->fd = -1;
129 e->type = AUTH_UNUSED;
130 buffer_free(&e->input);
131 buffer_free(&e->output);
132 buffer_free(&e->request);
133}
134
135static void
Damien Millerad833b32000-08-23 10:46:23 +1000136idtab_init(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000137{
Damien Millerad833b32000-08-23 10:46:23 +1000138 int i;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000139
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +0000140 for (i = 0; i <=2; i++) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100141 TAILQ_INIT(&idtable[i].idlist);
Damien Millerad833b32000-08-23 10:46:23 +1000142 idtable[i].nentries = 0;
143 }
144}
145
146/* return private key table for requested protocol version */
Ben Lindstrombba81212001-06-25 05:01:22 +0000147static Idtab *
Damien Millerad833b32000-08-23 10:46:23 +1000148idtab_lookup(int version)
149{
150 if (version < 1 || version > 2)
151 fatal("internal error, bad protocol version %d", version);
152 return &idtable[version];
153}
154
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000155static void
156free_identity(Identity *id)
157{
158 key_free(id->key);
159 xfree(id->comment);
160 xfree(id);
161}
162
Damien Millerad833b32000-08-23 10:46:23 +1000163/* return matching private key for given public key */
Damien Miller1a534ae2002-01-22 23:26:13 +1100164static Identity *
165lookup_identity(Key *key, int version)
Damien Millerad833b32000-08-23 10:46:23 +1000166{
Damien Miller1a534ae2002-01-22 23:26:13 +1100167 Identity *id;
168
Damien Millerad833b32000-08-23 10:46:23 +1000169 Idtab *tab = idtab_lookup(version);
Damien Miller1a534ae2002-01-22 23:26:13 +1100170 TAILQ_FOREACH(id, &tab->idlist, next) {
171 if (key_equal(key, id->key))
172 return (id);
Damien Millerad833b32000-08-23 10:46:23 +1000173 }
Damien Miller1a534ae2002-01-22 23:26:13 +1100174 return (NULL);
175}
176
Damien Miller6c711792003-01-24 11:36:23 +1100177/* Check confirmation of keysign request */
178static int
179confirm_key(Identity *id)
180{
Darren Tuckerce327b62004-11-05 20:38:03 +1100181 char *p;
Damien Miller6c711792003-01-24 11:36:23 +1100182 int ret = -1;
183
184 p = key_fingerprint(id->key, SSH_FP_MD5, SSH_FP_HEX);
Darren Tuckerce327b62004-11-05 20:38:03 +1100185 if (ask_permission("Allow use of key %s?\nKey fingerprint %s.",
186 id->comment, p))
187 ret = 0;
Damien Miller6c711792003-01-24 11:36:23 +1100188 xfree(p);
Darren Tuckerce327b62004-11-05 20:38:03 +1100189
Damien Miller6c711792003-01-24 11:36:23 +1100190 return (ret);
191}
192
Damien Millerad833b32000-08-23 10:46:23 +1000193/* send list of supported public keys to 'client' */
Ben Lindstrombba81212001-06-25 05:01:22 +0000194static void
Damien Millerad833b32000-08-23 10:46:23 +1000195process_request_identities(SocketEntry *e, int version)
196{
197 Idtab *tab = idtab_lookup(version);
Damien Miller1a534ae2002-01-22 23:26:13 +1100198 Identity *id;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000199 Buffer msg;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000200
Damien Miller95def091999-11-25 00:26:21 +1100201 buffer_init(&msg);
Damien Millerad833b32000-08-23 10:46:23 +1000202 buffer_put_char(&msg, (version == 1) ?
203 SSH_AGENT_RSA_IDENTITIES_ANSWER : SSH2_AGENT_IDENTITIES_ANSWER);
204 buffer_put_int(&msg, tab->nentries);
Damien Miller1a534ae2002-01-22 23:26:13 +1100205 TAILQ_FOREACH(id, &tab->idlist, next) {
Damien Miller0bc1bd82000-11-13 22:57:25 +1100206 if (id->key->type == KEY_RSA1) {
Damien Millerad833b32000-08-23 10:46:23 +1000207 buffer_put_int(&msg, BN_num_bits(id->key->rsa->n));
208 buffer_put_bignum(&msg, id->key->rsa->e);
209 buffer_put_bignum(&msg, id->key->rsa->n);
210 } else {
Ben Lindstrom46c16222000-12-22 01:43:59 +0000211 u_char *blob;
212 u_int blen;
Damien Miller0bc1bd82000-11-13 22:57:25 +1100213 key_to_blob(id->key, &blob, &blen);
Damien Millerad833b32000-08-23 10:46:23 +1000214 buffer_put_string(&msg, blob, blen);
215 xfree(blob);
216 }
217 buffer_put_cstring(&msg, id->comment);
Damien Miller95def091999-11-25 00:26:21 +1100218 }
219 buffer_put_int(&e->output, buffer_len(&msg));
220 buffer_append(&e->output, buffer_ptr(&msg), buffer_len(&msg));
221 buffer_free(&msg);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000222}
223
Damien Millerad833b32000-08-23 10:46:23 +1000224/* ssh1 only */
Ben Lindstrombba81212001-06-25 05:01:22 +0000225static void
Damien Millerad833b32000-08-23 10:46:23 +1000226process_authentication_challenge1(SocketEntry *e)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000227{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000228 u_char buf[32], mdbuf[16], session_id[16];
229 u_int response_type;
Damien Millerad833b32000-08-23 10:46:23 +1000230 BIGNUM *challenge;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000231 Identity *id;
Damien Millerad833b32000-08-23 10:46:23 +1000232 int i, len;
Damien Miller95def091999-11-25 00:26:21 +1100233 Buffer msg;
234 MD5_CTX md;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000235 Key *key;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000236
Damien Miller95def091999-11-25 00:26:21 +1100237 buffer_init(&msg);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100238 key = key_new(KEY_RSA1);
Damien Millerda755162002-01-22 23:09:22 +1100239 if ((challenge = BN_new()) == NULL)
240 fatal("process_authentication_challenge1: BN_new failed");
Damien Millerad833b32000-08-23 10:46:23 +1000241
Ben Lindstromc5a7f4f2002-06-25 23:19:13 +0000242 (void) buffer_get_int(&e->request); /* ignored */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000243 buffer_get_bignum(&e->request, key->rsa->e);
244 buffer_get_bignum(&e->request, key->rsa->n);
245 buffer_get_bignum(&e->request, challenge);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000246
Damien Millerad833b32000-08-23 10:46:23 +1000247 /* Only protocol 1.1 is supported */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000248 if (buffer_len(&e->request) == 0)
Damien Millerad833b32000-08-23 10:46:23 +1000249 goto failure;
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000250 buffer_get(&e->request, session_id, 16);
251 response_type = buffer_get_int(&e->request);
Damien Millerad833b32000-08-23 10:46:23 +1000252 if (response_type != 1)
253 goto failure;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000254
Damien Miller1a534ae2002-01-22 23:26:13 +1100255 id = lookup_identity(key, 1);
Damien Miller6c711792003-01-24 11:36:23 +1100256 if (id != NULL && (!id->confirm || confirm_key(id) == 0)) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100257 Key *private = id->key;
Damien Millerad833b32000-08-23 10:46:23 +1000258 /* Decrypt the challenge using the private key. */
Damien Miller7650bc62001-01-30 09:27:26 +1100259 if (rsa_private_decrypt(challenge, challenge, private->rsa) <= 0)
260 goto failure;
Damien Millerfd7c9111999-11-08 16:15:55 +1100261
Damien Millerad833b32000-08-23 10:46:23 +1000262 /* The response is MD5 of decrypted challenge plus session id. */
263 len = BN_num_bytes(challenge);
264 if (len <= 0 || len > 32) {
Damien Miller996acd22003-04-09 20:59:48 +1000265 logit("process_authentication_challenge: bad challenge length %d", len);
Damien Millerad833b32000-08-23 10:46:23 +1000266 goto failure;
Damien Miller95def091999-11-25 00:26:21 +1100267 }
Damien Millerad833b32000-08-23 10:46:23 +1000268 memset(buf, 0, 32);
269 BN_bn2bin(challenge, buf + 32 - len);
270 MD5_Init(&md);
271 MD5_Update(&md, buf, 32);
272 MD5_Update(&md, session_id, 16);
273 MD5_Final(mdbuf, &md);
274
275 /* Send the response. */
276 buffer_put_char(&msg, SSH_AGENT_RSA_RESPONSE);
277 for (i = 0; i < 16; i++)
278 buffer_put_char(&msg, mdbuf[i]);
279 goto send;
280 }
281
282failure:
283 /* Unknown identity or protocol error. Send failure. */
Damien Miller95def091999-11-25 00:26:21 +1100284 buffer_put_char(&msg, SSH_AGENT_FAILURE);
285send:
286 buffer_put_int(&e->output, buffer_len(&msg));
Damien Millerad833b32000-08-23 10:46:23 +1000287 buffer_append(&e->output, buffer_ptr(&msg), buffer_len(&msg));
288 key_free(key);
Damien Miller95def091999-11-25 00:26:21 +1100289 BN_clear_free(challenge);
Damien Millerad833b32000-08-23 10:46:23 +1000290 buffer_free(&msg);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000291}
292
Damien Millerad833b32000-08-23 10:46:23 +1000293/* ssh2 only */
Ben Lindstrombba81212001-06-25 05:01:22 +0000294static void
Damien Millerad833b32000-08-23 10:46:23 +1000295process_sign_request2(SocketEntry *e)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000296{
Ben Lindstrom46c16222000-12-22 01:43:59 +0000297 u_char *blob, *data, *signature = NULL;
298 u_int blen, dlen, slen = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000299 extern int datafellows;
300 int ok = -1, flags;
Damien Millerad833b32000-08-23 10:46:23 +1000301 Buffer msg;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000302 Key *key;
Damien Millerad833b32000-08-23 10:46:23 +1000303
304 datafellows = 0;
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000305
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000306 blob = buffer_get_string(&e->request, &blen);
307 data = buffer_get_string(&e->request, &dlen);
Damien Miller62cee002000-09-23 17:15:56 +1100308
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000309 flags = buffer_get_int(&e->request);
Damien Miller62cee002000-09-23 17:15:56 +1100310 if (flags & SSH_AGENT_OLD_SIGNATURE)
311 datafellows = SSH_BUG_SIGBLOB;
Damien Millerad833b32000-08-23 10:46:23 +1000312
Damien Miller0bc1bd82000-11-13 22:57:25 +1100313 key = key_from_blob(blob, blen);
Damien Millerad833b32000-08-23 10:46:23 +1000314 if (key != NULL) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100315 Identity *id = lookup_identity(key, 2);
Damien Miller6c711792003-01-24 11:36:23 +1100316 if (id != NULL && (!id->confirm || confirm_key(id) == 0))
Damien Miller1a534ae2002-01-22 23:26:13 +1100317 ok = key_sign(id->key, &signature, &slen, data, dlen);
Damien Millerad833b32000-08-23 10:46:23 +1000318 }
319 key_free(key);
320 buffer_init(&msg);
321 if (ok == 0) {
322 buffer_put_char(&msg, SSH2_AGENT_SIGN_RESPONSE);
323 buffer_put_string(&msg, signature, slen);
324 } else {
325 buffer_put_char(&msg, SSH_AGENT_FAILURE);
326 }
327 buffer_put_int(&e->output, buffer_len(&msg));
328 buffer_append(&e->output, buffer_ptr(&msg),
329 buffer_len(&msg));
330 buffer_free(&msg);
331 xfree(data);
332 xfree(blob);
333 if (signature != NULL)
334 xfree(signature);
335}
336
337/* shared */
Ben Lindstrombba81212001-06-25 05:01:22 +0000338static void
Damien Millerad833b32000-08-23 10:46:23 +1000339process_remove_identity(SocketEntry *e, int version)
340{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000341 u_int blen, bits;
342 int success = 0;
Damien Miller1a534ae2002-01-22 23:26:13 +1100343 Key *key = NULL;
Ben Lindstrom46c16222000-12-22 01:43:59 +0000344 u_char *blob;
Damien Miller7e8e8201999-11-16 13:37:16 +1100345
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +0000346 switch (version) {
Damien Millerad833b32000-08-23 10:46:23 +1000347 case 1:
Damien Miller0bc1bd82000-11-13 22:57:25 +1100348 key = key_new(KEY_RSA1);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000349 bits = buffer_get_int(&e->request);
350 buffer_get_bignum(&e->request, key->rsa->e);
351 buffer_get_bignum(&e->request, key->rsa->n);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000352
Damien Millerad833b32000-08-23 10:46:23 +1000353 if (bits != key_size(key))
Damien Miller996acd22003-04-09 20:59:48 +1000354 logit("Warning: identity keysize mismatch: actual %u, announced %u",
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000355 key_size(key), bits);
Damien Millerad833b32000-08-23 10:46:23 +1000356 break;
357 case 2:
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000358 blob = buffer_get_string(&e->request, &blen);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100359 key = key_from_blob(blob, blen);
Damien Millerad833b32000-08-23 10:46:23 +1000360 xfree(blob);
361 break;
362 }
363 if (key != NULL) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100364 Identity *id = lookup_identity(key, version);
365 if (id != NULL) {
Damien Miller5428f641999-11-25 11:54:57 +1100366 /*
367 * We have this key. Free the old key. Since we
Damien Miller788f2122005-11-05 15:14:59 +1100368 * don't want to leave empty slots in the middle of
Ben Lindstrom14920292000-11-21 21:24:55 +0000369 * the array, we actually free the key there and move
370 * all the entries between the empty slot and the end
371 * of the array.
Damien Miller5428f641999-11-25 11:54:57 +1100372 */
Damien Millerad833b32000-08-23 10:46:23 +1000373 Idtab *tab = idtab_lookup(version);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100374 if (tab->nentries < 1)
375 fatal("process_remove_identity: "
376 "internal error: tab->nentries %d",
377 tab->nentries);
Damien Miller1a534ae2002-01-22 23:26:13 +1100378 TAILQ_REMOVE(&tab->idlist, id, next);
379 free_identity(id);
Damien Millerad833b32000-08-23 10:46:23 +1000380 tab->nentries--;
381 success = 1;
Damien Miller95def091999-11-25 00:26:21 +1100382 }
Damien Millerad833b32000-08-23 10:46:23 +1000383 key_free(key);
384 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000385 buffer_put_int(&e->output, 1);
Damien Millerad833b32000-08-23 10:46:23 +1000386 buffer_put_char(&e->output,
387 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000388}
389
Ben Lindstrombba81212001-06-25 05:01:22 +0000390static void
Damien Millerad833b32000-08-23 10:46:23 +1000391process_remove_all_identities(SocketEntry *e, int version)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000392{
Damien Millerad833b32000-08-23 10:46:23 +1000393 Idtab *tab = idtab_lookup(version);
Damien Miller1a534ae2002-01-22 23:26:13 +1100394 Identity *id;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000395
Damien Miller95def091999-11-25 00:26:21 +1100396 /* Loop over all identities and clear the keys. */
Damien Miller1a534ae2002-01-22 23:26:13 +1100397 for (id = TAILQ_FIRST(&tab->idlist); id;
398 id = TAILQ_FIRST(&tab->idlist)) {
399 TAILQ_REMOVE(&tab->idlist, id, next);
400 free_identity(id);
Damien Miller95def091999-11-25 00:26:21 +1100401 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000402
Damien Miller95def091999-11-25 00:26:21 +1100403 /* Mark that there are no identities. */
Damien Millerad833b32000-08-23 10:46:23 +1000404 tab->nentries = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000405
406 /* Send success. */
407 buffer_put_int(&e->output, 1);
408 buffer_put_char(&e->output, SSH_AGENT_SUCCESS);
Damien Miller95def091999-11-25 00:26:21 +1100409}
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000410
Ben Lindstrombba81212001-06-25 05:01:22 +0000411static void
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000412reaper(void)
413{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000414 u_int now = time(NULL);
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000415 Identity *id, *nxt;
416 int version;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000417 Idtab *tab;
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000418
419 for (version = 1; version < 3; version++) {
420 tab = idtab_lookup(version);
421 for (id = TAILQ_FIRST(&tab->idlist); id; id = nxt) {
422 nxt = TAILQ_NEXT(id, next);
423 if (id->death != 0 && now >= id->death) {
424 TAILQ_REMOVE(&tab->idlist, id, next);
425 free_identity(id);
426 tab->nentries--;
427 }
428 }
429 }
430}
431
432static void
Damien Millerad833b32000-08-23 10:46:23 +1000433process_add_identity(SocketEntry *e, int version)
Damien Miller95def091999-11-25 00:26:21 +1100434{
Damien Millerad833b32000-08-23 10:46:23 +1000435 Idtab *tab = idtab_lookup(version);
Damien Miller6c711792003-01-24 11:36:23 +1100436 int type, success = 0, death = 0, confirm = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000437 char *type_name, *comment;
438 Key *k = NULL;
Damien Miller95def091999-11-25 00:26:21 +1100439
Damien Millerad833b32000-08-23 10:46:23 +1000440 switch (version) {
441 case 1:
Damien Miller0bc1bd82000-11-13 22:57:25 +1100442 k = key_new_private(KEY_RSA1);
Ben Lindstromc5a7f4f2002-06-25 23:19:13 +0000443 (void) buffer_get_int(&e->request); /* ignored */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000444 buffer_get_bignum(&e->request, k->rsa->n);
445 buffer_get_bignum(&e->request, k->rsa->e);
446 buffer_get_bignum(&e->request, k->rsa->d);
447 buffer_get_bignum(&e->request, k->rsa->iqmp);
Damien Miller95def091999-11-25 00:26:21 +1100448
Damien Millerad833b32000-08-23 10:46:23 +1000449 /* SSH and SSL have p and q swapped */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000450 buffer_get_bignum(&e->request, k->rsa->q); /* p */
451 buffer_get_bignum(&e->request, k->rsa->p); /* q */
Damien Miller95def091999-11-25 00:26:21 +1100452
Damien Millerad833b32000-08-23 10:46:23 +1000453 /* Generate additional parameters */
Ben Lindstromf7297dd2001-07-04 05:02:23 +0000454 rsa_generate_additional_parameters(k->rsa);
Damien Millerad833b32000-08-23 10:46:23 +1000455 break;
456 case 2:
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000457 type_name = buffer_get_string(&e->request, NULL);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000458 type = key_type_from_name(type_name);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100459 xfree(type_name);
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +0000460 switch (type) {
Damien Miller0bc1bd82000-11-13 22:57:25 +1100461 case KEY_DSA:
462 k = key_new_private(type);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000463 buffer_get_bignum2(&e->request, k->dsa->p);
464 buffer_get_bignum2(&e->request, k->dsa->q);
465 buffer_get_bignum2(&e->request, k->dsa->g);
466 buffer_get_bignum2(&e->request, k->dsa->pub_key);
467 buffer_get_bignum2(&e->request, k->dsa->priv_key);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100468 break;
469 case KEY_RSA:
470 k = key_new_private(type);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000471 buffer_get_bignum2(&e->request, k->rsa->n);
472 buffer_get_bignum2(&e->request, k->rsa->e);
473 buffer_get_bignum2(&e->request, k->rsa->d);
474 buffer_get_bignum2(&e->request, k->rsa->iqmp);
475 buffer_get_bignum2(&e->request, k->rsa->p);
476 buffer_get_bignum2(&e->request, k->rsa->q);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100477
478 /* Generate additional parameters */
Ben Lindstromf7297dd2001-07-04 05:02:23 +0000479 rsa_generate_additional_parameters(k->rsa);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100480 break;
481 default:
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000482 buffer_clear(&e->request);
Damien Millerad833b32000-08-23 10:46:23 +1000483 goto send;
Damien Miller95def091999-11-25 00:26:21 +1100484 }
Damien Millerad833b32000-08-23 10:46:23 +1000485 break;
486 }
Damien Millerc51d0732003-03-15 11:37:09 +1100487 /* enable blinding */
488 switch (k->type) {
489 case KEY_RSA:
490 case KEY_RSA1:
491 if (RSA_blinding_on(k->rsa, NULL) != 1) {
492 error("process_add_identity: RSA_blinding_on failed");
493 key_free(k);
494 goto send;
495 }
496 break;
497 }
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000498 comment = buffer_get_string(&e->request, NULL);
Damien Millerad833b32000-08-23 10:46:23 +1000499 if (k == NULL) {
500 xfree(comment);
501 goto send;
502 }
503 success = 1;
Ben Lindstrom4eb4c4e2002-06-21 00:04:48 +0000504 while (buffer_len(&e->request)) {
505 switch (buffer_get_char(&e->request)) {
Ben Lindstromc90f8a92002-06-21 00:06:54 +0000506 case SSH_AGENT_CONSTRAIN_LIFETIME:
Ben Lindstrom4eb4c4e2002-06-21 00:04:48 +0000507 death = time(NULL) + buffer_get_int(&e->request);
508 break;
Damien Miller6c711792003-01-24 11:36:23 +1100509 case SSH_AGENT_CONSTRAIN_CONFIRM:
510 confirm = 1;
511 break;
Ben Lindstrom4eb4c4e2002-06-21 00:04:48 +0000512 default:
513 break;
514 }
515 }
Damien Miller53d81482003-01-22 11:47:19 +1100516 if (lifetime && !death)
517 death = time(NULL) + lifetime;
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000518 if (lookup_identity(k, version) == NULL) {
519 Identity *id = xmalloc(sizeof(Identity));
520 id->key = k;
521 id->comment = comment;
522 id->death = death;
Damien Miller6c711792003-01-24 11:36:23 +1100523 id->confirm = confirm;
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000524 TAILQ_INSERT_TAIL(&tab->idlist, id, next);
525 /* Increment the number of identities. */
526 tab->nentries++;
527 } else {
528 key_free(k);
529 xfree(comment);
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000530 }
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000531send:
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000532 buffer_put_int(&e->output, 1);
533 buffer_put_char(&e->output,
534 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
535}
536
Ben Lindstrom2f717042002-06-06 21:52:03 +0000537/* XXX todo: encrypt sensitive data with passphrase */
538static void
539process_lock_agent(SocketEntry *e, int lock)
540{
Ben Lindstrom2f717042002-06-06 21:52:03 +0000541 int success = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000542 char *passwd;
Ben Lindstrom2f717042002-06-06 21:52:03 +0000543
544 passwd = buffer_get_string(&e->request, NULL);
545 if (locked && !lock && strcmp(passwd, lock_passwd) == 0) {
546 locked = 0;
547 memset(lock_passwd, 0, strlen(lock_passwd));
548 xfree(lock_passwd);
549 lock_passwd = NULL;
550 success = 1;
551 } else if (!locked && lock) {
552 locked = 1;
553 lock_passwd = xstrdup(passwd);
554 success = 1;
555 }
556 memset(passwd, 0, strlen(passwd));
557 xfree(passwd);
Ben Lindstromcb72e4f2002-06-21 00:41:51 +0000558
Ben Lindstrom2f717042002-06-06 21:52:03 +0000559 buffer_put_int(&e->output, 1);
560 buffer_put_char(&e->output,
561 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
Ben Lindstrom2f717042002-06-06 21:52:03 +0000562}
563
564static void
565no_identities(SocketEntry *e, u_int type)
566{
567 Buffer msg;
568
569 buffer_init(&msg);
570 buffer_put_char(&msg,
571 (type == SSH_AGENTC_REQUEST_RSA_IDENTITIES) ?
572 SSH_AGENT_RSA_IDENTITIES_ANSWER : SSH2_AGENT_IDENTITIES_ANSWER);
573 buffer_put_int(&msg, 0);
574 buffer_put_int(&e->output, buffer_len(&msg));
575 buffer_append(&e->output, buffer_ptr(&msg), buffer_len(&msg));
576 buffer_free(&msg);
577}
Ben Lindstrom3f471632001-07-04 03:53:15 +0000578
579#ifdef SMARTCARD
580static void
581process_add_smartcard_key (SocketEntry *e)
582{
Ben Lindstromba72d302002-03-22 03:51:06 +0000583 char *sc_reader_id = NULL, *pin;
Damien Millerd94f20d2003-06-11 22:06:33 +1000584 int i, version, success = 0, death = 0, confirm = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000585 Key **keys, *k;
586 Identity *id;
587 Idtab *tab;
Damien Miller9f0f5c62001-12-21 14:45:46 +1100588
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000589 sc_reader_id = buffer_get_string(&e->request, NULL);
590 pin = buffer_get_string(&e->request, NULL);
Damien Millerd94f20d2003-06-11 22:06:33 +1000591
592 while (buffer_len(&e->request)) {
593 switch (buffer_get_char(&e->request)) {
594 case SSH_AGENT_CONSTRAIN_LIFETIME:
595 death = time(NULL) + buffer_get_int(&e->request);
596 break;
597 case SSH_AGENT_CONSTRAIN_CONFIRM:
598 confirm = 1;
599 break;
600 default:
601 break;
602 }
603 }
604 if (lifetime && !death)
605 death = time(NULL) + lifetime;
606
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000607 keys = sc_get_keys(sc_reader_id, pin);
Ben Lindstromf7db3bb2001-08-06 21:35:51 +0000608 xfree(sc_reader_id);
Ben Lindstromba72d302002-03-22 03:51:06 +0000609 xfree(pin);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000610
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000611 if (keys == NULL || keys[0] == NULL) {
612 error("sc_get_keys failed");
Ben Lindstrom3f471632001-07-04 03:53:15 +0000613 goto send;
614 }
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000615 for (i = 0; keys[i] != NULL; i++) {
616 k = keys[i];
617 version = k->type == KEY_RSA1 ? 1 : 2;
618 tab = idtab_lookup(version);
619 if (lookup_identity(k, version) == NULL) {
620 id = xmalloc(sizeof(Identity));
621 id->key = k;
Damien Miller56a0bb02003-06-18 20:28:40 +1000622 id->comment = sc_get_key_label(k);
Damien Millerd94f20d2003-06-11 22:06:33 +1000623 id->death = death;
624 id->confirm = confirm;
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000625 TAILQ_INSERT_TAIL(&tab->idlist, id, next);
626 tab->nentries++;
627 success = 1;
628 } else {
629 key_free(k);
630 }
631 keys[i] = NULL;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000632 }
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000633 xfree(keys);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000634send:
635 buffer_put_int(&e->output, 1);
636 buffer_put_char(&e->output,
637 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
638}
639
640static void
641process_remove_smartcard_key(SocketEntry *e)
642{
Ben Lindstromba72d302002-03-22 03:51:06 +0000643 char *sc_reader_id = NULL, *pin;
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000644 int i, version, success = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000645 Key **keys, *k = NULL;
646 Identity *id;
647 Idtab *tab;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000648
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000649 sc_reader_id = buffer_get_string(&e->request, NULL);
650 pin = buffer_get_string(&e->request, NULL);
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000651 keys = sc_get_keys(sc_reader_id, pin);
Ben Lindstromf7db3bb2001-08-06 21:35:51 +0000652 xfree(sc_reader_id);
Ben Lindstromba72d302002-03-22 03:51:06 +0000653 xfree(pin);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000654
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000655 if (keys == NULL || keys[0] == NULL) {
656 error("sc_get_keys failed");
657 goto send;
658 }
659 for (i = 0; keys[i] != NULL; i++) {
660 k = keys[i];
661 version = k->type == KEY_RSA1 ? 1 : 2;
662 if ((id = lookup_identity(k, version)) != NULL) {
663 tab = idtab_lookup(version);
Ben Lindstrom5a6abda2002-06-09 19:41:48 +0000664 TAILQ_REMOVE(&tab->idlist, id, next);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000665 tab->nentries--;
Damien Miller1a534ae2002-01-22 23:26:13 +1100666 free_identity(id);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000667 success = 1;
668 }
669 key_free(k);
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000670 keys[i] = NULL;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000671 }
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000672 xfree(keys);
673send:
Ben Lindstrom3f471632001-07-04 03:53:15 +0000674 buffer_put_int(&e->output, 1);
675 buffer_put_char(&e->output,
676 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
677}
Ben Lindstromffce1472001-08-06 21:57:31 +0000678#endif /* SMARTCARD */
Ben Lindstrom3f471632001-07-04 03:53:15 +0000679
Damien Millerad833b32000-08-23 10:46:23 +1000680/* dispatch incoming messages */
681
Ben Lindstrombba81212001-06-25 05:01:22 +0000682static void
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000683process_message(SocketEntry *e)
684{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000685 u_int msg_len, type;
Ben Lindstrom46c16222000-12-22 01:43:59 +0000686 u_char *cp;
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000687
688 /* kill dead keys */
689 reaper();
690
Damien Miller95def091999-11-25 00:26:21 +1100691 if (buffer_len(&e->input) < 5)
692 return; /* Incomplete message. */
Damien Miller708d21c2002-01-22 23:18:15 +1100693 cp = buffer_ptr(&e->input);
Damien Miller95def091999-11-25 00:26:21 +1100694 msg_len = GET_32BIT(cp);
695 if (msg_len > 256 * 1024) {
Damien Miller58f34862002-09-04 16:31:21 +1000696 close_socket(e);
Damien Miller95def091999-11-25 00:26:21 +1100697 return;
698 }
699 if (buffer_len(&e->input) < msg_len + 4)
700 return;
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000701
702 /* move the current input to e->request */
Damien Miller95def091999-11-25 00:26:21 +1100703 buffer_consume(&e->input, 4);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000704 buffer_clear(&e->request);
705 buffer_append(&e->request, buffer_ptr(&e->input), msg_len);
706 buffer_consume(&e->input, msg_len);
707 type = buffer_get_char(&e->request);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000708
Ben Lindstrom2f717042002-06-06 21:52:03 +0000709 /* check wheter agent is locked */
710 if (locked && type != SSH_AGENTC_UNLOCK) {
711 buffer_clear(&e->request);
712 switch (type) {
713 case SSH_AGENTC_REQUEST_RSA_IDENTITIES:
714 case SSH2_AGENTC_REQUEST_IDENTITIES:
715 /* send empty lists */
716 no_identities(e, type);
717 break;
718 default:
719 /* send a fail message for all other request types */
720 buffer_put_int(&e->output, 1);
721 buffer_put_char(&e->output, SSH_AGENT_FAILURE);
722 }
723 return;
724 }
725
Ben Lindstrom3f471632001-07-04 03:53:15 +0000726 debug("type %d", type);
Damien Miller95def091999-11-25 00:26:21 +1100727 switch (type) {
Ben Lindstrom2f717042002-06-06 21:52:03 +0000728 case SSH_AGENTC_LOCK:
729 case SSH_AGENTC_UNLOCK:
730 process_lock_agent(e, type == SSH_AGENTC_LOCK);
731 break;
Damien Millerad833b32000-08-23 10:46:23 +1000732 /* ssh1 */
Damien Miller95def091999-11-25 00:26:21 +1100733 case SSH_AGENTC_RSA_CHALLENGE:
Damien Millerad833b32000-08-23 10:46:23 +1000734 process_authentication_challenge1(e);
735 break;
736 case SSH_AGENTC_REQUEST_RSA_IDENTITIES:
737 process_request_identities(e, 1);
Damien Miller95def091999-11-25 00:26:21 +1100738 break;
739 case SSH_AGENTC_ADD_RSA_IDENTITY:
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000740 case SSH_AGENTC_ADD_RSA_ID_CONSTRAINED:
Damien Millerad833b32000-08-23 10:46:23 +1000741 process_add_identity(e, 1);
Damien Miller95def091999-11-25 00:26:21 +1100742 break;
743 case SSH_AGENTC_REMOVE_RSA_IDENTITY:
Damien Millerad833b32000-08-23 10:46:23 +1000744 process_remove_identity(e, 1);
Damien Miller95def091999-11-25 00:26:21 +1100745 break;
746 case SSH_AGENTC_REMOVE_ALL_RSA_IDENTITIES:
Damien Millerad833b32000-08-23 10:46:23 +1000747 process_remove_all_identities(e, 1);
748 break;
749 /* ssh2 */
750 case SSH2_AGENTC_SIGN_REQUEST:
751 process_sign_request2(e);
752 break;
753 case SSH2_AGENTC_REQUEST_IDENTITIES:
754 process_request_identities(e, 2);
755 break;
756 case SSH2_AGENTC_ADD_IDENTITY:
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000757 case SSH2_AGENTC_ADD_ID_CONSTRAINED:
Damien Millerad833b32000-08-23 10:46:23 +1000758 process_add_identity(e, 2);
759 break;
760 case SSH2_AGENTC_REMOVE_IDENTITY:
761 process_remove_identity(e, 2);
762 break;
763 case SSH2_AGENTC_REMOVE_ALL_IDENTITIES:
764 process_remove_all_identities(e, 2);
Damien Miller95def091999-11-25 00:26:21 +1100765 break;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000766#ifdef SMARTCARD
767 case SSH_AGENTC_ADD_SMARTCARD_KEY:
Damien Millerd94f20d2003-06-11 22:06:33 +1000768 case SSH_AGENTC_ADD_SMARTCARD_KEY_CONSTRAINED:
Ben Lindstrom3f471632001-07-04 03:53:15 +0000769 process_add_smartcard_key(e);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100770 break;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000771 case SSH_AGENTC_REMOVE_SMARTCARD_KEY:
772 process_remove_smartcard_key(e);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100773 break;
Ben Lindstromffce1472001-08-06 21:57:31 +0000774#endif /* SMARTCARD */
Damien Miller95def091999-11-25 00:26:21 +1100775 default:
776 /* Unknown message. Respond with failure. */
777 error("Unknown message %d", type);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000778 buffer_clear(&e->request);
Damien Miller95def091999-11-25 00:26:21 +1100779 buffer_put_int(&e->output, 1);
780 buffer_put_char(&e->output, SSH_AGENT_FAILURE);
781 break;
782 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000783}
784
Ben Lindstrombba81212001-06-25 05:01:22 +0000785static void
Ben Lindstrom65366a82001-12-06 16:32:47 +0000786new_socket(sock_type type, int fd)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000787{
Darren Tuckerfb16b242003-09-22 21:04:23 +1000788 u_int i, old_alloc, new_alloc;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000789
Damien Miller232711f2004-06-15 10:35:30 +1000790 set_nonblock(fd);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000791
Damien Miller95def091999-11-25 00:26:21 +1100792 if (fd > max_fd)
793 max_fd = fd;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000794
Damien Miller95def091999-11-25 00:26:21 +1100795 for (i = 0; i < sockets_alloc; i++)
796 if (sockets[i].type == AUTH_UNUSED) {
797 sockets[i].fd = fd;
Damien Miller95def091999-11-25 00:26:21 +1100798 buffer_init(&sockets[i].input);
799 buffer_init(&sockets[i].output);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000800 buffer_init(&sockets[i].request);
Darren Tuckerfb16b242003-09-22 21:04:23 +1000801 sockets[i].type = type;
Damien Miller95def091999-11-25 00:26:21 +1100802 return;
803 }
804 old_alloc = sockets_alloc;
Darren Tuckerfb16b242003-09-22 21:04:23 +1000805 new_alloc = sockets_alloc + 10;
Damien Miller95def091999-11-25 00:26:21 +1100806 if (sockets)
Darren Tuckerfb16b242003-09-22 21:04:23 +1000807 sockets = xrealloc(sockets, new_alloc * sizeof(sockets[0]));
Damien Miller95def091999-11-25 00:26:21 +1100808 else
Darren Tuckerfb16b242003-09-22 21:04:23 +1000809 sockets = xmalloc(new_alloc * sizeof(sockets[0]));
810 for (i = old_alloc; i < new_alloc; i++)
Damien Miller95def091999-11-25 00:26:21 +1100811 sockets[i].type = AUTH_UNUSED;
Darren Tuckerfb16b242003-09-22 21:04:23 +1000812 sockets_alloc = new_alloc;
Damien Miller95def091999-11-25 00:26:21 +1100813 sockets[old_alloc].fd = fd;
814 buffer_init(&sockets[old_alloc].input);
815 buffer_init(&sockets[old_alloc].output);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000816 buffer_init(&sockets[old_alloc].request);
Darren Tuckerfb16b242003-09-22 21:04:23 +1000817 sockets[old_alloc].type = type;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000818}
819
Ben Lindstrombba81212001-06-25 05:01:22 +0000820static int
Darren Tuckerc7a6fc42004-08-13 21:18:00 +1000821prepare_select(fd_set **fdrp, fd_set **fdwp, int *fdl, u_int *nallocp)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000822{
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000823 u_int i, sz;
824 int n = 0;
825
826 for (i = 0; i < sockets_alloc; i++) {
Damien Miller95def091999-11-25 00:26:21 +1100827 switch (sockets[i].type) {
828 case AUTH_SOCKET:
829 case AUTH_CONNECTION:
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000830 n = MAX(n, sockets[i].fd);
Damien Miller95def091999-11-25 00:26:21 +1100831 break;
832 case AUTH_UNUSED:
833 break;
834 default:
835 fatal("Unknown socket type %d", sockets[i].type);
836 break;
837 }
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000838 }
839
840 sz = howmany(n+1, NFDBITS) * sizeof(fd_mask);
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +0000841 if (*fdrp == NULL || sz > *nallocp) {
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000842 if (*fdrp)
Ben Lindstrom86ebcb62001-04-04 01:53:20 +0000843 xfree(*fdrp);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000844 if (*fdwp)
Ben Lindstrom86ebcb62001-04-04 01:53:20 +0000845 xfree(*fdwp);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000846 *fdrp = xmalloc(sz);
847 *fdwp = xmalloc(sz);
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +0000848 *nallocp = sz;
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000849 }
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +0000850 if (n < *fdl)
851 debug("XXX shrink: %d < %d", n, *fdl);
852 *fdl = n;
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000853 memset(*fdrp, 0, sz);
854 memset(*fdwp, 0, sz);
855
856 for (i = 0; i < sockets_alloc; i++) {
857 switch (sockets[i].type) {
858 case AUTH_SOCKET:
859 case AUTH_CONNECTION:
860 FD_SET(sockets[i].fd, *fdrp);
861 if (buffer_len(&sockets[i].output) > 0)
862 FD_SET(sockets[i].fd, *fdwp);
863 break;
864 default:
865 break;
866 }
867 }
868 return (1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000869}
870
Ben Lindstrombba81212001-06-25 05:01:22 +0000871static void
Damien Miller95def091999-11-25 00:26:21 +1100872after_select(fd_set *readset, fd_set *writeset)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000873{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000874 struct sockaddr_un sunaddr;
Damien Miller7684ee12000-03-17 23:40:15 +1100875 socklen_t slen;
Damien Miller95def091999-11-25 00:26:21 +1100876 char buf[1024];
Ben Lindstrom822b6342002-06-23 21:38:49 +0000877 int len, sock;
878 u_int i;
Damien Miller538f1812002-09-12 09:51:10 +1000879 uid_t euid;
880 gid_t egid;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000881
Damien Miller95def091999-11-25 00:26:21 +1100882 for (i = 0; i < sockets_alloc; i++)
883 switch (sockets[i].type) {
884 case AUTH_UNUSED:
885 break;
886 case AUTH_SOCKET:
887 if (FD_ISSET(sockets[i].fd, readset)) {
Damien Miller7684ee12000-03-17 23:40:15 +1100888 slen = sizeof(sunaddr);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000889 sock = accept(sockets[i].fd,
890 (struct sockaddr *) &sunaddr, &slen);
Damien Miller95def091999-11-25 00:26:21 +1100891 if (sock < 0) {
Damien Millerc653b892002-02-08 22:05:41 +1100892 error("accept from AUTH_SOCKET: %s",
893 strerror(errno));
Damien Miller95def091999-11-25 00:26:21 +1100894 break;
895 }
Damien Miller538f1812002-09-12 09:51:10 +1000896 if (getpeereid(sock, &euid, &egid) < 0) {
897 error("getpeereid %d failed: %s",
898 sock, strerror(errno));
899 close(sock);
900 break;
901 }
Damien Milleraf9de382002-10-03 11:54:35 +1000902 if ((euid != 0) && (getuid() != euid)) {
Damien Miller538f1812002-09-12 09:51:10 +1000903 error("uid mismatch: "
Damien Millerdb30b122002-09-19 11:46:58 +1000904 "peer euid %u != uid %u",
905 (u_int) euid, (u_int) getuid());
Damien Miller538f1812002-09-12 09:51:10 +1000906 close(sock);
907 break;
908 }
Damien Miller95def091999-11-25 00:26:21 +1100909 new_socket(AUTH_CONNECTION, sock);
910 }
911 break;
912 case AUTH_CONNECTION:
913 if (buffer_len(&sockets[i].output) > 0 &&
914 FD_ISSET(sockets[i].fd, writeset)) {
Ben Lindstromb3144e52001-03-06 03:31:34 +0000915 do {
916 len = write(sockets[i].fd,
917 buffer_ptr(&sockets[i].output),
918 buffer_len(&sockets[i].output));
919 if (len == -1 && (errno == EAGAIN ||
920 errno == EINTR))
921 continue;
922 break;
923 } while (1);
Damien Miller95def091999-11-25 00:26:21 +1100924 if (len <= 0) {
Damien Miller58f34862002-09-04 16:31:21 +1000925 close_socket(&sockets[i]);
Damien Miller95def091999-11-25 00:26:21 +1100926 break;
927 }
928 buffer_consume(&sockets[i].output, len);
929 }
930 if (FD_ISSET(sockets[i].fd, readset)) {
Ben Lindstromb3144e52001-03-06 03:31:34 +0000931 do {
932 len = read(sockets[i].fd, buf, sizeof(buf));
933 if (len == -1 && (errno == EAGAIN ||
934 errno == EINTR))
935 continue;
936 break;
937 } while (1);
Damien Miller95def091999-11-25 00:26:21 +1100938 if (len <= 0) {
Damien Miller58f34862002-09-04 16:31:21 +1000939 close_socket(&sockets[i]);
Damien Miller95def091999-11-25 00:26:21 +1100940 break;
941 }
942 buffer_append(&sockets[i].input, buf, len);
943 process_message(&sockets[i]);
944 }
945 break;
946 default:
947 fatal("Unknown type %d", sockets[i].type);
948 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000949}
950
Ben Lindstrombba81212001-06-25 05:01:22 +0000951static void
Darren Tucker6fa8abd2003-09-22 21:10:21 +1000952cleanup_socket(void)
Damien Miller792c5111999-10-29 09:47:09 +1000953{
Ben Lindstrom77808ab2001-01-26 05:10:34 +0000954 if (socket_name[0])
955 unlink(socket_name);
956 if (socket_dir[0])
957 rmdir(socket_dir);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000958}
959
Darren Tucker3e33cec2003-10-02 16:12:36 +1000960void
Damien Miller792c5111999-10-29 09:47:09 +1000961cleanup_exit(int i)
962{
Darren Tucker6fa8abd2003-09-22 21:10:21 +1000963 cleanup_socket();
964 _exit(i);
Damien Miller792c5111999-10-29 09:47:09 +1000965}
966
Ben Lindstrombba81212001-06-25 05:01:22 +0000967static void
Ben Lindstrom77808ab2001-01-26 05:10:34 +0000968cleanup_handler(int sig)
969{
Darren Tucker6fa8abd2003-09-22 21:10:21 +1000970 cleanup_socket();
Ben Lindstrom77808ab2001-01-26 05:10:34 +0000971 _exit(2);
972}
973
Ben Lindstrombba81212001-06-25 05:01:22 +0000974static void
Ben Lindstrom0250da02001-07-22 20:44:00 +0000975check_parent_exists(int sig)
976{
977 int save_errno = errno;
978
979 if (parent_pid != -1 && kill(parent_pid, 0) < 0) {
980 /* printf("Parent has died - Authentication agent exiting.\n"); */
981 cleanup_handler(sig); /* safe */
982 }
Damien Miller0cbb9de2003-06-04 22:56:15 +1000983 mysignal(SIGALRM, check_parent_exists);
Ben Lindstrom0250da02001-07-22 20:44:00 +0000984 alarm(10);
985 errno = save_errno;
986}
987
988static void
Ben Lindstrom28072eb2001-02-10 23:13:41 +0000989usage(void)
Damien Miller792c5111999-10-29 09:47:09 +1000990{
Ben Lindstromddfb1e32001-08-06 22:06:35 +0000991 fprintf(stderr, "Usage: %s [options] [command [args ...]]\n",
Kevin Steves29265862001-01-24 14:06:28 +0000992 __progname);
Ben Lindstromddfb1e32001-08-06 22:06:35 +0000993 fprintf(stderr, "Options:\n");
994 fprintf(stderr, " -c Generate C-shell commands on stdout.\n");
995 fprintf(stderr, " -s Generate Bourne shell commands on stdout.\n");
996 fprintf(stderr, " -k Kill the current agent.\n");
997 fprintf(stderr, " -d Debug mode.\n");
Ben Lindstromb7788f32002-06-06 21:46:08 +0000998 fprintf(stderr, " -a socket Bind agent socket to given name.\n");
Damien Miller53d81482003-01-22 11:47:19 +1100999 fprintf(stderr, " -t life Default identity lifetime (seconds).\n");
Damien Miller95def091999-11-25 00:26:21 +11001000 exit(1);
Damien Miller792c5111999-10-29 09:47:09 +10001001}
1002
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001003int
1004main(int ac, char **av)
1005{
Damien Miller6c711792003-01-24 11:36:23 +11001006 int c_flag = 0, d_flag = 0, k_flag = 0, s_flag = 0;
Darren Tuckerc7a6fc42004-08-13 21:18:00 +10001007 int sock, fd, ch;
1008 u_int nalloc;
Ben Lindstrom822b6342002-06-23 21:38:49 +00001009 char *shell, *format, *pidstr, *agentsocket = NULL;
1010 fd_set *readsetp = NULL, *writesetp = NULL;
Damien Miller95def091999-11-25 00:26:21 +11001011 struct sockaddr_un sunaddr;
Ben Lindstrom2c467a22000-12-27 04:57:41 +00001012#ifdef HAVE_SETRLIMIT
Ben Lindstromc72745a2000-12-02 19:03:54 +00001013 struct rlimit rlim;
Ben Lindstrom2c467a22000-12-27 04:57:41 +00001014#endif
Ben Lindstrom3524d692001-05-03 22:59:24 +00001015 int prev_mask;
Damien Miller615f9392000-05-17 22:53:33 +10001016 extern int optind;
Ben Lindstrom883844d2002-06-23 00:20:50 +00001017 extern char *optarg;
Ben Lindstrom822b6342002-06-23 21:38:49 +00001018 pid_t pid;
1019 char pidstrbuf[1 + 3 * sizeof pid];
Kevin Stevesde41bc62000-12-14 00:04:08 +00001020
Darren Tuckerce321d82005-10-03 18:11:24 +10001021 /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
1022 sanitise_stdfd();
1023
Damien Miller6cffb9a2002-09-04 16:20:26 +10001024 /* drop */
1025 setegid(getgid());
1026 setgid(getgid());
1027
Damien Miller6c4914a2004-03-03 11:08:59 +11001028#if defined(HAVE_PRCTL) && defined(PR_SET_DUMPABLE)
1029 /* Disable ptrace on Linux without sgid bit */
1030 prctl(PR_SET_DUMPABLE, 0);
1031#endif
1032
Ben Lindstromd09fcf52001-03-29 00:29:54 +00001033 SSLeay_add_all_algorithms();
1034
Damien Miller59d3d5b2003-08-22 09:34:41 +10001035 __progname = ssh_get_progname(av[0]);
Damien Millerf9b625c2000-07-09 22:42:32 +10001036 init_rng();
Damien Miller60bc5172001-03-19 09:38:15 +11001037 seed_rng();
Kevin Stevesef4eea92001-02-05 12:42:17 +00001038
Damien Miller53d81482003-01-22 11:47:19 +11001039 while ((ch = getopt(ac, av, "cdksa:t:")) != -1) {
Damien Miller95def091999-11-25 00:26:21 +11001040 switch (ch) {
1041 case 'c':
1042 if (s_flag)
1043 usage();
1044 c_flag++;
1045 break;
1046 case 'k':
1047 k_flag++;
1048 break;
1049 case 's':
1050 if (c_flag)
1051 usage();
1052 s_flag++;
1053 break;
Ben Lindstromd94580c2001-07-04 03:48:02 +00001054 case 'd':
1055 if (d_flag)
1056 usage();
1057 d_flag++;
1058 break;
Ben Lindstromb7788f32002-06-06 21:46:08 +00001059 case 'a':
1060 agentsocket = optarg;
1061 break;
Damien Miller53d81482003-01-22 11:47:19 +11001062 case 't':
1063 if ((lifetime = convtime(optarg)) == -1) {
1064 fprintf(stderr, "Invalid lifetime\n");
1065 usage();
1066 }
1067 break;
Damien Miller95def091999-11-25 00:26:21 +11001068 default:
1069 usage();
1070 }
Damien Miller792c5111999-10-29 09:47:09 +10001071 }
Damien Miller95def091999-11-25 00:26:21 +11001072 ac -= optind;
1073 av += optind;
Damien Miller792c5111999-10-29 09:47:09 +10001074
Ben Lindstromd94580c2001-07-04 03:48:02 +00001075 if (ac > 0 && (c_flag || k_flag || s_flag || d_flag))
Damien Miller95def091999-11-25 00:26:21 +11001076 usage();
Damien Miller792c5111999-10-29 09:47:09 +10001077
Ben Lindstromeecdf232002-04-02 21:03:51 +00001078 if (ac == 0 && !c_flag && !s_flag) {
Damien Miller95def091999-11-25 00:26:21 +11001079 shell = getenv("SHELL");
1080 if (shell != NULL && strncmp(shell + strlen(shell) - 3, "csh", 3) == 0)
1081 c_flag = 1;
Damien Miller792c5111999-10-29 09:47:09 +10001082 }
Damien Miller95def091999-11-25 00:26:21 +11001083 if (k_flag) {
1084 pidstr = getenv(SSH_AGENTPID_ENV_NAME);
1085 if (pidstr == NULL) {
1086 fprintf(stderr, "%s not set, cannot kill agent\n",
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001087 SSH_AGENTPID_ENV_NAME);
Damien Miller95def091999-11-25 00:26:21 +11001088 exit(1);
1089 }
1090 pid = atoi(pidstr);
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001091 if (pid < 1) {
Damien Miller95def091999-11-25 00:26:21 +11001092 fprintf(stderr, "%s=\"%s\", which is not a good PID\n",
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001093 SSH_AGENTPID_ENV_NAME, pidstr);
Damien Miller95def091999-11-25 00:26:21 +11001094 exit(1);
1095 }
1096 if (kill(pid, SIGTERM) == -1) {
1097 perror("kill");
1098 exit(1);
1099 }
1100 format = c_flag ? "unsetenv %s;\n" : "unset %s;\n";
1101 printf(format, SSH_AUTHSOCKET_ENV_NAME);
1102 printf(format, SSH_AGENTPID_ENV_NAME);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001103 printf("echo Agent pid %ld killed;\n", (long)pid);
Damien Miller95def091999-11-25 00:26:21 +11001104 exit(0);
Damien Miller792c5111999-10-29 09:47:09 +10001105 }
Damien Miller95def091999-11-25 00:26:21 +11001106 parent_pid = getpid();
1107
Ben Lindstromb7788f32002-06-06 21:46:08 +00001108 if (agentsocket == NULL) {
1109 /* Create private directory for agent socket */
Darren Tucker072a7b12003-10-15 16:10:25 +10001110 strlcpy(socket_dir, "/tmp/ssh-XXXXXXXXXX", sizeof socket_dir);
Ben Lindstromb7788f32002-06-06 21:46:08 +00001111 if (mkdtemp(socket_dir) == NULL) {
1112 perror("mkdtemp: private socket dir");
1113 exit(1);
1114 }
Ben Lindstromce0f6342002-06-11 16:42:49 +00001115 snprintf(socket_name, sizeof socket_name, "%s/agent.%ld", socket_dir,
1116 (long)parent_pid);
Ben Lindstromb7788f32002-06-06 21:46:08 +00001117 } else {
1118 /* Try to use specified agent socket */
1119 socket_dir[0] = '\0';
1120 strlcpy(socket_name, agentsocket, sizeof socket_name);
Damien Miller792c5111999-10-29 09:47:09 +10001121 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001122
Damien Miller5428f641999-11-25 11:54:57 +11001123 /*
1124 * Create socket early so it will exist before command gets run from
1125 * the parent.
1126 */
Damien Miller95def091999-11-25 00:26:21 +11001127 sock = socket(AF_UNIX, SOCK_STREAM, 0);
1128 if (sock < 0) {
1129 perror("socket");
Darren Tucker1dee8682004-11-05 20:26:49 +11001130 *socket_name = '\0'; /* Don't unlink any existing file */
Damien Miller95def091999-11-25 00:26:21 +11001131 cleanup_exit(1);
Damien Miller792c5111999-10-29 09:47:09 +10001132 }
Damien Miller95def091999-11-25 00:26:21 +11001133 memset(&sunaddr, 0, sizeof(sunaddr));
1134 sunaddr.sun_family = AF_UNIX;
1135 strlcpy(sunaddr.sun_path, socket_name, sizeof(sunaddr.sun_path));
Ben Lindstrom3524d692001-05-03 22:59:24 +00001136 prev_mask = umask(0177);
Damien Miller95def091999-11-25 00:26:21 +11001137 if (bind(sock, (struct sockaddr *) & sunaddr, sizeof(sunaddr)) < 0) {
1138 perror("bind");
Darren Tucker1dee8682004-11-05 20:26:49 +11001139 *socket_name = '\0'; /* Don't unlink any existing file */
Ben Lindstrom3524d692001-05-03 22:59:24 +00001140 umask(prev_mask);
Damien Miller95def091999-11-25 00:26:21 +11001141 cleanup_exit(1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001142 }
Ben Lindstrom3524d692001-05-03 22:59:24 +00001143 umask(prev_mask);
Darren Tucker3175eb92003-12-09 19:15:11 +11001144 if (listen(sock, SSH_LISTEN_BACKLOG) < 0) {
Damien Miller95def091999-11-25 00:26:21 +11001145 perror("listen");
1146 cleanup_exit(1);
1147 }
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001148
Damien Miller5428f641999-11-25 11:54:57 +11001149 /*
1150 * Fork, and have the parent execute the command, if any, or present
1151 * the socket data. The child continues as the authentication agent.
1152 */
Ben Lindstromd94580c2001-07-04 03:48:02 +00001153 if (d_flag) {
1154 log_init(__progname, SYSLOG_LEVEL_DEBUG1, SYSLOG_FACILITY_AUTH, 1);
1155 format = c_flag ? "setenv %s %s;\n" : "%s=%s; export %s;\n";
1156 printf(format, SSH_AUTHSOCKET_ENV_NAME, socket_name,
1157 SSH_AUTHSOCKET_ENV_NAME);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001158 printf("echo Agent pid %ld;\n", (long)parent_pid);
Ben Lindstromd94580c2001-07-04 03:48:02 +00001159 goto skip;
1160 }
Damien Miller95def091999-11-25 00:26:21 +11001161 pid = fork();
1162 if (pid == -1) {
1163 perror("fork");
Damien Millerc653b892002-02-08 22:05:41 +11001164 cleanup_exit(1);
Damien Miller95def091999-11-25 00:26:21 +11001165 }
1166 if (pid != 0) { /* Parent - execute the given command. */
1167 close(sock);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001168 snprintf(pidstrbuf, sizeof pidstrbuf, "%ld", (long)pid);
Damien Miller95def091999-11-25 00:26:21 +11001169 if (ac == 0) {
1170 format = c_flag ? "setenv %s %s;\n" : "%s=%s; export %s;\n";
1171 printf(format, SSH_AUTHSOCKET_ENV_NAME, socket_name,
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001172 SSH_AUTHSOCKET_ENV_NAME);
Damien Miller95def091999-11-25 00:26:21 +11001173 printf(format, SSH_AGENTPID_ENV_NAME, pidstrbuf,
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001174 SSH_AGENTPID_ENV_NAME);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001175 printf("echo Agent pid %ld;\n", (long)pid);
Damien Miller95def091999-11-25 00:26:21 +11001176 exit(0);
1177 }
Damien Millere4340be2000-09-16 13:29:08 +11001178 if (setenv(SSH_AUTHSOCKET_ENV_NAME, socket_name, 1) == -1 ||
1179 setenv(SSH_AGENTPID_ENV_NAME, pidstrbuf, 1) == -1) {
1180 perror("setenv");
1181 exit(1);
1182 }
Damien Miller95def091999-11-25 00:26:21 +11001183 execvp(av[0], av);
1184 perror(av[0]);
1185 exit(1);
1186 }
Damien Millerc653b892002-02-08 22:05:41 +11001187 /* child */
1188 log_init(__progname, SYSLOG_LEVEL_INFO, SYSLOG_FACILITY_AUTH, 0);
Ben Lindstrom3fdf8762001-07-22 20:40:24 +00001189
1190 if (setsid() == -1) {
Damien Millerc653b892002-02-08 22:05:41 +11001191 error("setsid: %s", strerror(errno));
Ben Lindstrom3fdf8762001-07-22 20:40:24 +00001192 cleanup_exit(1);
1193 }
1194
1195 (void)chdir("/");
Damien Miller6c711792003-01-24 11:36:23 +11001196 if ((fd = open(_PATH_DEVNULL, O_RDWR, 0)) != -1) {
1197 /* XXX might close listen socket */
1198 (void)dup2(fd, STDIN_FILENO);
1199 (void)dup2(fd, STDOUT_FILENO);
1200 (void)dup2(fd, STDERR_FILENO);
1201 if (fd > 2)
1202 close(fd);
1203 }
Damien Miller95def091999-11-25 00:26:21 +11001204
Ben Lindstrom2c467a22000-12-27 04:57:41 +00001205#ifdef HAVE_SETRLIMIT
Ben Lindstromc72745a2000-12-02 19:03:54 +00001206 /* deny core dumps, since memory contains unencrypted private keys */
1207 rlim.rlim_cur = rlim.rlim_max = 0;
1208 if (setrlimit(RLIMIT_CORE, &rlim) < 0) {
Damien Millerc653b892002-02-08 22:05:41 +11001209 error("setrlimit RLIMIT_CORE: %s", strerror(errno));
Ben Lindstromc72745a2000-12-02 19:03:54 +00001210 cleanup_exit(1);
1211 }
Ben Lindstrom2c467a22000-12-27 04:57:41 +00001212#endif
Ben Lindstromd94580c2001-07-04 03:48:02 +00001213
1214skip:
Damien Miller95def091999-11-25 00:26:21 +11001215 new_socket(AUTH_SOCKET, sock);
1216 if (ac > 0) {
Damien Miller0cbb9de2003-06-04 22:56:15 +10001217 mysignal(SIGALRM, check_parent_exists);
Damien Miller95def091999-11-25 00:26:21 +11001218 alarm(10);
1219 }
Damien Millerad833b32000-08-23 10:46:23 +10001220 idtab_init();
Damien Miller48bfa9c2001-07-14 12:12:55 +10001221 if (!d_flag)
Ben Lindstromd94580c2001-07-04 03:48:02 +00001222 signal(SIGINT, SIG_IGN);
Damien Miller48bfa9c2001-07-14 12:12:55 +10001223 signal(SIGPIPE, SIG_IGN);
Ben Lindstrom77808ab2001-01-26 05:10:34 +00001224 signal(SIGHUP, cleanup_handler);
1225 signal(SIGTERM, cleanup_handler);
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +00001226 nalloc = 0;
1227
Damien Miller95def091999-11-25 00:26:21 +11001228 while (1) {
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +00001229 prepare_select(&readsetp, &writesetp, &max_fd, &nalloc);
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001230 if (select(max_fd + 1, readsetp, writesetp, NULL, NULL) < 0) {
Damien Miller95def091999-11-25 00:26:21 +11001231 if (errno == EINTR)
1232 continue;
Damien Millerc653b892002-02-08 22:05:41 +11001233 fatal("select: %s", strerror(errno));
Damien Miller95def091999-11-25 00:26:21 +11001234 }
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001235 after_select(readsetp, writesetp);
Damien Miller95def091999-11-25 00:26:21 +11001236 }
1237 /* NOTREACHED */
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001238}