blob: c179ceaa2d388e9d4cd2aa255012107be1af84d9 [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 Miller9b481512002-09-12 10:43:29 +100037#include "openbsd-compat/sys-queue.h"
Damien Millerd5580922003-05-14 13:40:06 +100038RCSID("$OpenBSD: ssh-agent.c,v 1.109 2003/04/08 20:21:29 itojun Exp $");
Damien Millerec52d7c2002-01-22 23:52:17 +110039
Ben Lindstrom226cfa02001-01-22 05:34:40 +000040#include <openssl/evp.h>
41#include <openssl/md5.h>
Damien Millerd4a8b7e1999-10-27 13:42:43 +100042
43#include "ssh.h"
44#include "rsa.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100045#include "buffer.h"
46#include "bufaux.h"
47#include "xmalloc.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100048#include "getput.h"
Damien Miller994cf142000-07-21 10:19:44 +100049#include "key.h"
50#include "authfd.h"
Damien Miller62cee002000-09-23 17:15:56 +110051#include "compat.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000052#include "log.h"
Damien Miller6c711792003-01-24 11:36:23 +110053#include "readpass.h"
54#include "misc.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100055
Ben Lindstrom3f471632001-07-04 03:53:15 +000056#ifdef SMARTCARD
Ben Lindstrom3f471632001-07-04 03:53:15 +000057#include "scard.h"
Ben Lindstrombcc18082001-08-06 21:59:25 +000058#endif
Ben Lindstrom3f471632001-07-04 03:53:15 +000059
Ben Lindstrom65366a82001-12-06 16:32:47 +000060typedef enum {
61 AUTH_UNUSED,
62 AUTH_SOCKET,
63 AUTH_CONNECTION
64} sock_type;
65
Damien Miller95def091999-11-25 00:26:21 +110066typedef struct {
67 int fd;
Ben Lindstrom65366a82001-12-06 16:32:47 +000068 sock_type type;
Damien Miller95def091999-11-25 00:26:21 +110069 Buffer input;
70 Buffer output;
Ben Lindstrom21d1ed82002-06-06 21:48:57 +000071 Buffer request;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100072} SocketEntry;
73
Ben Lindstrom46c16222000-12-22 01:43:59 +000074u_int sockets_alloc = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100075SocketEntry *sockets = NULL;
76
Damien Miller1a534ae2002-01-22 23:26:13 +110077typedef struct identity {
78 TAILQ_ENTRY(identity) next;
Damien Millerad833b32000-08-23 10:46:23 +100079 Key *key;
Damien Miller95def091999-11-25 00:26:21 +110080 char *comment;
Ben Lindstrom61d328a2002-06-06 21:54:57 +000081 u_int death;
Damien Miller6c711792003-01-24 11:36:23 +110082 u_int confirm;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100083} Identity;
84
Damien Millerad833b32000-08-23 10:46:23 +100085typedef struct {
86 int nentries;
Damien Miller1a534ae2002-01-22 23:26:13 +110087 TAILQ_HEAD(idqueue, identity) idlist;
Damien Millerad833b32000-08-23 10:46:23 +100088} Idtab;
89
90/* private key table, one per protocol version */
91Idtab idtable[3];
Damien Millerd4a8b7e1999-10-27 13:42:43 +100092
93int max_fd = 0;
94
95/* pid of shell == parent of agent */
Damien Miller166fca82000-04-20 07:42:21 +100096pid_t parent_pid = -1;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100097
98/* pathname and directory for AUTH_SOCKET */
99char socket_name[1024];
100char socket_dir[1024];
101
Ben Lindstrom2f717042002-06-06 21:52:03 +0000102/* locking */
103int locked = 0;
104char *lock_passwd = NULL;
105
Damien Miller95def091999-11-25 00:26:21 +1100106#ifdef HAVE___PROGNAME
107extern char *__progname;
Ben Lindstrom49a79c02000-11-17 03:47:20 +0000108#else
109char *__progname;
110#endif
Damien Miller95def091999-11-25 00:26:21 +1100111
Damien Miller53d81482003-01-22 11:47:19 +1100112/* Default lifetime (0 == forever) */
113static int lifetime = 0;
114
Ben Lindstrombba81212001-06-25 05:01:22 +0000115static void
Damien Miller58f34862002-09-04 16:31:21 +1000116close_socket(SocketEntry *e)
117{
Damien Miller58f34862002-09-04 16:31:21 +1000118 close(e->fd);
119 e->fd = -1;
120 e->type = AUTH_UNUSED;
121 buffer_free(&e->input);
122 buffer_free(&e->output);
123 buffer_free(&e->request);
124}
125
126static void
Damien Millerad833b32000-08-23 10:46:23 +1000127idtab_init(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000128{
Damien Millerad833b32000-08-23 10:46:23 +1000129 int i;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000130
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +0000131 for (i = 0; i <=2; i++) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100132 TAILQ_INIT(&idtable[i].idlist);
Damien Millerad833b32000-08-23 10:46:23 +1000133 idtable[i].nentries = 0;
134 }
135}
136
137/* return private key table for requested protocol version */
Ben Lindstrombba81212001-06-25 05:01:22 +0000138static Idtab *
Damien Millerad833b32000-08-23 10:46:23 +1000139idtab_lookup(int version)
140{
141 if (version < 1 || version > 2)
142 fatal("internal error, bad protocol version %d", version);
143 return &idtable[version];
144}
145
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000146static void
147free_identity(Identity *id)
148{
149 key_free(id->key);
150 xfree(id->comment);
151 xfree(id);
152}
153
Damien Millerad833b32000-08-23 10:46:23 +1000154/* return matching private key for given public key */
Damien Miller1a534ae2002-01-22 23:26:13 +1100155static Identity *
156lookup_identity(Key *key, int version)
Damien Millerad833b32000-08-23 10:46:23 +1000157{
Damien Miller1a534ae2002-01-22 23:26:13 +1100158 Identity *id;
159
Damien Millerad833b32000-08-23 10:46:23 +1000160 Idtab *tab = idtab_lookup(version);
Damien Miller1a534ae2002-01-22 23:26:13 +1100161 TAILQ_FOREACH(id, &tab->idlist, next) {
162 if (key_equal(key, id->key))
163 return (id);
Damien Millerad833b32000-08-23 10:46:23 +1000164 }
Damien Miller1a534ae2002-01-22 23:26:13 +1100165 return (NULL);
166}
167
Damien Miller6c711792003-01-24 11:36:23 +1100168/* Check confirmation of keysign request */
169static int
170confirm_key(Identity *id)
171{
172 char *p, prompt[1024];
173 int ret = -1;
174
175 p = key_fingerprint(id->key, SSH_FP_MD5, SSH_FP_HEX);
176 snprintf(prompt, sizeof(prompt), "Allow use of key %s?\n"
177 "Key fingerprint %s.", id->comment, p);
178 xfree(p);
179 p = read_passphrase(prompt, RP_ALLOW_EOF);
180 if (p != NULL) {
181 /*
182 * Accept empty responses and responses consisting
183 * of the word "yes" as affirmative.
184 */
185 if (*p == '\0' || *p == '\n' || strcasecmp(p, "yes") == 0)
186 ret = 0;
187 xfree(p);
188 }
189 return (ret);
190}
191
Damien Millerad833b32000-08-23 10:46:23 +1000192/* send list of supported public keys to 'client' */
Ben Lindstrombba81212001-06-25 05:01:22 +0000193static void
Damien Millerad833b32000-08-23 10:46:23 +1000194process_request_identities(SocketEntry *e, int version)
195{
196 Idtab *tab = idtab_lookup(version);
Damien Miller1a534ae2002-01-22 23:26:13 +1100197 Identity *id;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000198 Buffer msg;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000199
Damien Miller95def091999-11-25 00:26:21 +1100200 buffer_init(&msg);
Damien Millerad833b32000-08-23 10:46:23 +1000201 buffer_put_char(&msg, (version == 1) ?
202 SSH_AGENT_RSA_IDENTITIES_ANSWER : SSH2_AGENT_IDENTITIES_ANSWER);
203 buffer_put_int(&msg, tab->nentries);
Damien Miller1a534ae2002-01-22 23:26:13 +1100204 TAILQ_FOREACH(id, &tab->idlist, next) {
Damien Miller0bc1bd82000-11-13 22:57:25 +1100205 if (id->key->type == KEY_RSA1) {
Damien Millerad833b32000-08-23 10:46:23 +1000206 buffer_put_int(&msg, BN_num_bits(id->key->rsa->n));
207 buffer_put_bignum(&msg, id->key->rsa->e);
208 buffer_put_bignum(&msg, id->key->rsa->n);
209 } else {
Ben Lindstrom46c16222000-12-22 01:43:59 +0000210 u_char *blob;
211 u_int blen;
Damien Miller0bc1bd82000-11-13 22:57:25 +1100212 key_to_blob(id->key, &blob, &blen);
Damien Millerad833b32000-08-23 10:46:23 +1000213 buffer_put_string(&msg, blob, blen);
214 xfree(blob);
215 }
216 buffer_put_cstring(&msg, id->comment);
Damien Miller95def091999-11-25 00:26:21 +1100217 }
218 buffer_put_int(&e->output, buffer_len(&msg));
219 buffer_append(&e->output, buffer_ptr(&msg), buffer_len(&msg));
220 buffer_free(&msg);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000221}
222
Damien Millerad833b32000-08-23 10:46:23 +1000223/* ssh1 only */
Ben Lindstrombba81212001-06-25 05:01:22 +0000224static void
Damien Millerad833b32000-08-23 10:46:23 +1000225process_authentication_challenge1(SocketEntry *e)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000226{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000227 u_char buf[32], mdbuf[16], session_id[16];
228 u_int response_type;
Damien Millerad833b32000-08-23 10:46:23 +1000229 BIGNUM *challenge;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000230 Identity *id;
Damien Millerad833b32000-08-23 10:46:23 +1000231 int i, len;
Damien Miller95def091999-11-25 00:26:21 +1100232 Buffer msg;
233 MD5_CTX md;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000234 Key *key;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000235
Damien Miller95def091999-11-25 00:26:21 +1100236 buffer_init(&msg);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100237 key = key_new(KEY_RSA1);
Damien Millerda755162002-01-22 23:09:22 +1100238 if ((challenge = BN_new()) == NULL)
239 fatal("process_authentication_challenge1: BN_new failed");
Damien Millerad833b32000-08-23 10:46:23 +1000240
Ben Lindstromc5a7f4f2002-06-25 23:19:13 +0000241 (void) buffer_get_int(&e->request); /* ignored */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000242 buffer_get_bignum(&e->request, key->rsa->e);
243 buffer_get_bignum(&e->request, key->rsa->n);
244 buffer_get_bignum(&e->request, challenge);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000245
Damien Millerad833b32000-08-23 10:46:23 +1000246 /* Only protocol 1.1 is supported */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000247 if (buffer_len(&e->request) == 0)
Damien Millerad833b32000-08-23 10:46:23 +1000248 goto failure;
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000249 buffer_get(&e->request, session_id, 16);
250 response_type = buffer_get_int(&e->request);
Damien Millerad833b32000-08-23 10:46:23 +1000251 if (response_type != 1)
252 goto failure;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000253
Damien Miller1a534ae2002-01-22 23:26:13 +1100254 id = lookup_identity(key, 1);
Damien Miller6c711792003-01-24 11:36:23 +1100255 if (id != NULL && (!id->confirm || confirm_key(id) == 0)) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100256 Key *private = id->key;
Damien Millerad833b32000-08-23 10:46:23 +1000257 /* Decrypt the challenge using the private key. */
Damien Miller7650bc62001-01-30 09:27:26 +1100258 if (rsa_private_decrypt(challenge, challenge, private->rsa) <= 0)
259 goto failure;
Damien Millerfd7c9111999-11-08 16:15:55 +1100260
Damien Millerad833b32000-08-23 10:46:23 +1000261 /* The response is MD5 of decrypted challenge plus session id. */
262 len = BN_num_bytes(challenge);
263 if (len <= 0 || len > 32) {
Damien Miller996acd22003-04-09 20:59:48 +1000264 logit("process_authentication_challenge: bad challenge length %d", len);
Damien Millerad833b32000-08-23 10:46:23 +1000265 goto failure;
Damien Miller95def091999-11-25 00:26:21 +1100266 }
Damien Millerad833b32000-08-23 10:46:23 +1000267 memset(buf, 0, 32);
268 BN_bn2bin(challenge, buf + 32 - len);
269 MD5_Init(&md);
270 MD5_Update(&md, buf, 32);
271 MD5_Update(&md, session_id, 16);
272 MD5_Final(mdbuf, &md);
273
274 /* Send the response. */
275 buffer_put_char(&msg, SSH_AGENT_RSA_RESPONSE);
276 for (i = 0; i < 16; i++)
277 buffer_put_char(&msg, mdbuf[i]);
278 goto send;
279 }
280
281failure:
282 /* Unknown identity or protocol error. Send failure. */
Damien Miller95def091999-11-25 00:26:21 +1100283 buffer_put_char(&msg, SSH_AGENT_FAILURE);
284send:
285 buffer_put_int(&e->output, buffer_len(&msg));
Damien Millerad833b32000-08-23 10:46:23 +1000286 buffer_append(&e->output, buffer_ptr(&msg), buffer_len(&msg));
287 key_free(key);
Damien Miller95def091999-11-25 00:26:21 +1100288 BN_clear_free(challenge);
Damien Millerad833b32000-08-23 10:46:23 +1000289 buffer_free(&msg);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000290}
291
Damien Millerad833b32000-08-23 10:46:23 +1000292/* ssh2 only */
Ben Lindstrombba81212001-06-25 05:01:22 +0000293static void
Damien Millerad833b32000-08-23 10:46:23 +1000294process_sign_request2(SocketEntry *e)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000295{
Ben Lindstrom46c16222000-12-22 01:43:59 +0000296 u_char *blob, *data, *signature = NULL;
297 u_int blen, dlen, slen = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000298 extern int datafellows;
299 int ok = -1, flags;
Damien Millerad833b32000-08-23 10:46:23 +1000300 Buffer msg;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000301 Key *key;
Damien Millerad833b32000-08-23 10:46:23 +1000302
303 datafellows = 0;
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000304
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000305 blob = buffer_get_string(&e->request, &blen);
306 data = buffer_get_string(&e->request, &dlen);
Damien Miller62cee002000-09-23 17:15:56 +1100307
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000308 flags = buffer_get_int(&e->request);
Damien Miller62cee002000-09-23 17:15:56 +1100309 if (flags & SSH_AGENT_OLD_SIGNATURE)
310 datafellows = SSH_BUG_SIGBLOB;
Damien Millerad833b32000-08-23 10:46:23 +1000311
Damien Miller0bc1bd82000-11-13 22:57:25 +1100312 key = key_from_blob(blob, blen);
Damien Millerad833b32000-08-23 10:46:23 +1000313 if (key != NULL) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100314 Identity *id = lookup_identity(key, 2);
Damien Miller6c711792003-01-24 11:36:23 +1100315 if (id != NULL && (!id->confirm || confirm_key(id) == 0))
Damien Miller1a534ae2002-01-22 23:26:13 +1100316 ok = key_sign(id->key, &signature, &slen, data, dlen);
Damien Millerad833b32000-08-23 10:46:23 +1000317 }
318 key_free(key);
319 buffer_init(&msg);
320 if (ok == 0) {
321 buffer_put_char(&msg, SSH2_AGENT_SIGN_RESPONSE);
322 buffer_put_string(&msg, signature, slen);
323 } else {
324 buffer_put_char(&msg, SSH_AGENT_FAILURE);
325 }
326 buffer_put_int(&e->output, buffer_len(&msg));
327 buffer_append(&e->output, buffer_ptr(&msg),
328 buffer_len(&msg));
329 buffer_free(&msg);
330 xfree(data);
331 xfree(blob);
332 if (signature != NULL)
333 xfree(signature);
334}
335
336/* shared */
Ben Lindstrombba81212001-06-25 05:01:22 +0000337static void
Damien Millerad833b32000-08-23 10:46:23 +1000338process_remove_identity(SocketEntry *e, int version)
339{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000340 u_int blen, bits;
341 int success = 0;
Damien Miller1a534ae2002-01-22 23:26:13 +1100342 Key *key = NULL;
Ben Lindstrom46c16222000-12-22 01:43:59 +0000343 u_char *blob;
Damien Miller7e8e8201999-11-16 13:37:16 +1100344
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +0000345 switch (version) {
Damien Millerad833b32000-08-23 10:46:23 +1000346 case 1:
Damien Miller0bc1bd82000-11-13 22:57:25 +1100347 key = key_new(KEY_RSA1);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000348 bits = buffer_get_int(&e->request);
349 buffer_get_bignum(&e->request, key->rsa->e);
350 buffer_get_bignum(&e->request, key->rsa->n);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000351
Damien Millerad833b32000-08-23 10:46:23 +1000352 if (bits != key_size(key))
Damien Miller996acd22003-04-09 20:59:48 +1000353 logit("Warning: identity keysize mismatch: actual %u, announced %u",
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000354 key_size(key), bits);
Damien Millerad833b32000-08-23 10:46:23 +1000355 break;
356 case 2:
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000357 blob = buffer_get_string(&e->request, &blen);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100358 key = key_from_blob(blob, blen);
Damien Millerad833b32000-08-23 10:46:23 +1000359 xfree(blob);
360 break;
361 }
362 if (key != NULL) {
Damien Miller1a534ae2002-01-22 23:26:13 +1100363 Identity *id = lookup_identity(key, version);
364 if (id != NULL) {
Damien Miller5428f641999-11-25 11:54:57 +1100365 /*
366 * We have this key. Free the old key. Since we
367 * don\'t want to leave empty slots in the middle of
Ben Lindstrom14920292000-11-21 21:24:55 +0000368 * the array, we actually free the key there and move
369 * all the entries between the empty slot and the end
370 * of the array.
Damien Miller5428f641999-11-25 11:54:57 +1100371 */
Damien Millerad833b32000-08-23 10:46:23 +1000372 Idtab *tab = idtab_lookup(version);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100373 if (tab->nentries < 1)
374 fatal("process_remove_identity: "
375 "internal error: tab->nentries %d",
376 tab->nentries);
Damien Miller1a534ae2002-01-22 23:26:13 +1100377 TAILQ_REMOVE(&tab->idlist, id, next);
378 free_identity(id);
Damien Millerad833b32000-08-23 10:46:23 +1000379 tab->nentries--;
380 success = 1;
Damien Miller95def091999-11-25 00:26:21 +1100381 }
Damien Millerad833b32000-08-23 10:46:23 +1000382 key_free(key);
383 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000384 buffer_put_int(&e->output, 1);
Damien Millerad833b32000-08-23 10:46:23 +1000385 buffer_put_char(&e->output,
386 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000387}
388
Ben Lindstrombba81212001-06-25 05:01:22 +0000389static void
Damien Millerad833b32000-08-23 10:46:23 +1000390process_remove_all_identities(SocketEntry *e, int version)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000391{
Damien Millerad833b32000-08-23 10:46:23 +1000392 Idtab *tab = idtab_lookup(version);
Damien Miller1a534ae2002-01-22 23:26:13 +1100393 Identity *id;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000394
Damien Miller95def091999-11-25 00:26:21 +1100395 /* Loop over all identities and clear the keys. */
Damien Miller1a534ae2002-01-22 23:26:13 +1100396 for (id = TAILQ_FIRST(&tab->idlist); id;
397 id = TAILQ_FIRST(&tab->idlist)) {
398 TAILQ_REMOVE(&tab->idlist, id, next);
399 free_identity(id);
Damien Miller95def091999-11-25 00:26:21 +1100400 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000401
Damien Miller95def091999-11-25 00:26:21 +1100402 /* Mark that there are no identities. */
Damien Millerad833b32000-08-23 10:46:23 +1000403 tab->nentries = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000404
405 /* Send success. */
406 buffer_put_int(&e->output, 1);
407 buffer_put_char(&e->output, SSH_AGENT_SUCCESS);
Damien Miller95def091999-11-25 00:26:21 +1100408}
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000409
Ben Lindstrombba81212001-06-25 05:01:22 +0000410static void
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000411reaper(void)
412{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000413 u_int now = time(NULL);
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000414 Identity *id, *nxt;
415 int version;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000416 Idtab *tab;
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000417
418 for (version = 1; version < 3; version++) {
419 tab = idtab_lookup(version);
420 for (id = TAILQ_FIRST(&tab->idlist); id; id = nxt) {
421 nxt = TAILQ_NEXT(id, next);
422 if (id->death != 0 && now >= id->death) {
423 TAILQ_REMOVE(&tab->idlist, id, next);
424 free_identity(id);
425 tab->nentries--;
426 }
427 }
428 }
429}
430
431static void
Damien Millerad833b32000-08-23 10:46:23 +1000432process_add_identity(SocketEntry *e, int version)
Damien Miller95def091999-11-25 00:26:21 +1100433{
Damien Millerad833b32000-08-23 10:46:23 +1000434 Idtab *tab = idtab_lookup(version);
Damien Miller6c711792003-01-24 11:36:23 +1100435 int type, success = 0, death = 0, confirm = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000436 char *type_name, *comment;
437 Key *k = NULL;
Damien Miller95def091999-11-25 00:26:21 +1100438
Damien Millerad833b32000-08-23 10:46:23 +1000439 switch (version) {
440 case 1:
Damien Miller0bc1bd82000-11-13 22:57:25 +1100441 k = key_new_private(KEY_RSA1);
Ben Lindstromc5a7f4f2002-06-25 23:19:13 +0000442 (void) buffer_get_int(&e->request); /* ignored */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000443 buffer_get_bignum(&e->request, k->rsa->n);
444 buffer_get_bignum(&e->request, k->rsa->e);
445 buffer_get_bignum(&e->request, k->rsa->d);
446 buffer_get_bignum(&e->request, k->rsa->iqmp);
Damien Miller95def091999-11-25 00:26:21 +1100447
Damien Millerad833b32000-08-23 10:46:23 +1000448 /* SSH and SSL have p and q swapped */
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000449 buffer_get_bignum(&e->request, k->rsa->q); /* p */
450 buffer_get_bignum(&e->request, k->rsa->p); /* q */
Damien Miller95def091999-11-25 00:26:21 +1100451
Damien Millerad833b32000-08-23 10:46:23 +1000452 /* Generate additional parameters */
Ben Lindstromf7297dd2001-07-04 05:02:23 +0000453 rsa_generate_additional_parameters(k->rsa);
Damien Millerad833b32000-08-23 10:46:23 +1000454 break;
455 case 2:
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000456 type_name = buffer_get_string(&e->request, NULL);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000457 type = key_type_from_name(type_name);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100458 xfree(type_name);
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +0000459 switch (type) {
Damien Miller0bc1bd82000-11-13 22:57:25 +1100460 case KEY_DSA:
461 k = key_new_private(type);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000462 buffer_get_bignum2(&e->request, k->dsa->p);
463 buffer_get_bignum2(&e->request, k->dsa->q);
464 buffer_get_bignum2(&e->request, k->dsa->g);
465 buffer_get_bignum2(&e->request, k->dsa->pub_key);
466 buffer_get_bignum2(&e->request, k->dsa->priv_key);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100467 break;
468 case KEY_RSA:
469 k = key_new_private(type);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000470 buffer_get_bignum2(&e->request, k->rsa->n);
471 buffer_get_bignum2(&e->request, k->rsa->e);
472 buffer_get_bignum2(&e->request, k->rsa->d);
473 buffer_get_bignum2(&e->request, k->rsa->iqmp);
474 buffer_get_bignum2(&e->request, k->rsa->p);
475 buffer_get_bignum2(&e->request, k->rsa->q);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100476
477 /* Generate additional parameters */
Ben Lindstromf7297dd2001-07-04 05:02:23 +0000478 rsa_generate_additional_parameters(k->rsa);
Damien Miller0bc1bd82000-11-13 22:57:25 +1100479 break;
480 default:
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000481 buffer_clear(&e->request);
Damien Millerad833b32000-08-23 10:46:23 +1000482 goto send;
Damien Miller95def091999-11-25 00:26:21 +1100483 }
Damien Millerad833b32000-08-23 10:46:23 +1000484 break;
485 }
Damien Millerc51d0732003-03-15 11:37:09 +1100486 /* enable blinding */
487 switch (k->type) {
488 case KEY_RSA:
489 case KEY_RSA1:
490 if (RSA_blinding_on(k->rsa, NULL) != 1) {
491 error("process_add_identity: RSA_blinding_on failed");
492 key_free(k);
493 goto send;
494 }
495 break;
496 }
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000497 comment = buffer_get_string(&e->request, NULL);
Damien Millerad833b32000-08-23 10:46:23 +1000498 if (k == NULL) {
499 xfree(comment);
500 goto send;
501 }
502 success = 1;
Ben Lindstrom4eb4c4e2002-06-21 00:04:48 +0000503 while (buffer_len(&e->request)) {
504 switch (buffer_get_char(&e->request)) {
Ben Lindstromc90f8a92002-06-21 00:06:54 +0000505 case SSH_AGENT_CONSTRAIN_LIFETIME:
Ben Lindstrom4eb4c4e2002-06-21 00:04:48 +0000506 death = time(NULL) + buffer_get_int(&e->request);
507 break;
Damien Miller6c711792003-01-24 11:36:23 +1100508 case SSH_AGENT_CONSTRAIN_CONFIRM:
509 confirm = 1;
510 break;
Ben Lindstrom4eb4c4e2002-06-21 00:04:48 +0000511 default:
512 break;
513 }
514 }
Damien Miller53d81482003-01-22 11:47:19 +1100515 if (lifetime && !death)
516 death = time(NULL) + lifetime;
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000517 if (lookup_identity(k, version) == NULL) {
518 Identity *id = xmalloc(sizeof(Identity));
519 id->key = k;
520 id->comment = comment;
521 id->death = death;
Damien Miller6c711792003-01-24 11:36:23 +1100522 id->confirm = confirm;
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000523 TAILQ_INSERT_TAIL(&tab->idlist, id, next);
524 /* Increment the number of identities. */
525 tab->nentries++;
526 } else {
527 key_free(k);
528 xfree(comment);
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000529 }
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000530send:
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000531 buffer_put_int(&e->output, 1);
532 buffer_put_char(&e->output,
533 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
534}
535
Ben Lindstrom2f717042002-06-06 21:52:03 +0000536/* XXX todo: encrypt sensitive data with passphrase */
537static void
538process_lock_agent(SocketEntry *e, int lock)
539{
Ben Lindstrom2f717042002-06-06 21:52:03 +0000540 int success = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000541 char *passwd;
Ben Lindstrom2f717042002-06-06 21:52:03 +0000542
543 passwd = buffer_get_string(&e->request, NULL);
544 if (locked && !lock && strcmp(passwd, lock_passwd) == 0) {
545 locked = 0;
546 memset(lock_passwd, 0, strlen(lock_passwd));
547 xfree(lock_passwd);
548 lock_passwd = NULL;
549 success = 1;
550 } else if (!locked && lock) {
551 locked = 1;
552 lock_passwd = xstrdup(passwd);
553 success = 1;
554 }
555 memset(passwd, 0, strlen(passwd));
556 xfree(passwd);
Ben Lindstromcb72e4f2002-06-21 00:41:51 +0000557
Ben Lindstrom2f717042002-06-06 21:52:03 +0000558 buffer_put_int(&e->output, 1);
559 buffer_put_char(&e->output,
560 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
Ben Lindstrom2f717042002-06-06 21:52:03 +0000561}
562
563static void
564no_identities(SocketEntry *e, u_int type)
565{
566 Buffer msg;
567
568 buffer_init(&msg);
569 buffer_put_char(&msg,
570 (type == SSH_AGENTC_REQUEST_RSA_IDENTITIES) ?
571 SSH_AGENT_RSA_IDENTITIES_ANSWER : SSH2_AGENT_IDENTITIES_ANSWER);
572 buffer_put_int(&msg, 0);
573 buffer_put_int(&e->output, buffer_len(&msg));
574 buffer_append(&e->output, buffer_ptr(&msg), buffer_len(&msg));
575 buffer_free(&msg);
576}
Ben Lindstrom3f471632001-07-04 03:53:15 +0000577
578#ifdef SMARTCARD
579static void
580process_add_smartcard_key (SocketEntry *e)
581{
Ben Lindstromba72d302002-03-22 03:51:06 +0000582 char *sc_reader_id = NULL, *pin;
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000583 int i, version, success = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000584 Key **keys, *k;
585 Identity *id;
586 Idtab *tab;
Damien Miller9f0f5c62001-12-21 14:45:46 +1100587
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000588 sc_reader_id = buffer_get_string(&e->request, NULL);
589 pin = buffer_get_string(&e->request, NULL);
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000590 keys = sc_get_keys(sc_reader_id, pin);
Ben Lindstromf7db3bb2001-08-06 21:35:51 +0000591 xfree(sc_reader_id);
Ben Lindstromba72d302002-03-22 03:51:06 +0000592 xfree(pin);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000593
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000594 if (keys == NULL || keys[0] == NULL) {
595 error("sc_get_keys failed");
Ben Lindstrom3f471632001-07-04 03:53:15 +0000596 goto send;
597 }
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000598 for (i = 0; keys[i] != NULL; i++) {
599 k = keys[i];
600 version = k->type == KEY_RSA1 ? 1 : 2;
601 tab = idtab_lookup(version);
602 if (lookup_identity(k, version) == NULL) {
603 id = xmalloc(sizeof(Identity));
604 id->key = k;
605 id->comment = xstrdup("smartcard key");
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000606 id->death = 0;
Damien Miller6c711792003-01-24 11:36:23 +1100607 id->confirm = 0;
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000608 TAILQ_INSERT_TAIL(&tab->idlist, id, next);
609 tab->nentries++;
610 success = 1;
611 } else {
612 key_free(k);
613 }
614 keys[i] = NULL;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000615 }
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000616 xfree(keys);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000617send:
618 buffer_put_int(&e->output, 1);
619 buffer_put_char(&e->output,
620 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
621}
622
623static void
624process_remove_smartcard_key(SocketEntry *e)
625{
Ben Lindstromba72d302002-03-22 03:51:06 +0000626 char *sc_reader_id = NULL, *pin;
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000627 int i, version, success = 0;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000628 Key **keys, *k = NULL;
629 Identity *id;
630 Idtab *tab;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000631
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000632 sc_reader_id = buffer_get_string(&e->request, NULL);
633 pin = buffer_get_string(&e->request, NULL);
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000634 keys = sc_get_keys(sc_reader_id, pin);
Ben Lindstromf7db3bb2001-08-06 21:35:51 +0000635 xfree(sc_reader_id);
Ben Lindstromba72d302002-03-22 03:51:06 +0000636 xfree(pin);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000637
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000638 if (keys == NULL || keys[0] == NULL) {
639 error("sc_get_keys failed");
640 goto send;
641 }
642 for (i = 0; keys[i] != NULL; i++) {
643 k = keys[i];
644 version = k->type == KEY_RSA1 ? 1 : 2;
645 if ((id = lookup_identity(k, version)) != NULL) {
646 tab = idtab_lookup(version);
Ben Lindstrom5a6abda2002-06-09 19:41:48 +0000647 TAILQ_REMOVE(&tab->idlist, id, next);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000648 tab->nentries--;
Damien Miller1a534ae2002-01-22 23:26:13 +1100649 free_identity(id);
Ben Lindstrom3f471632001-07-04 03:53:15 +0000650 success = 1;
651 }
652 key_free(k);
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000653 keys[i] = NULL;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000654 }
Ben Lindstrom0936a5b2002-03-26 03:17:42 +0000655 xfree(keys);
656send:
Ben Lindstrom3f471632001-07-04 03:53:15 +0000657 buffer_put_int(&e->output, 1);
658 buffer_put_char(&e->output,
659 success ? SSH_AGENT_SUCCESS : SSH_AGENT_FAILURE);
660}
Ben Lindstromffce1472001-08-06 21:57:31 +0000661#endif /* SMARTCARD */
Ben Lindstrom3f471632001-07-04 03:53:15 +0000662
Damien Millerad833b32000-08-23 10:46:23 +1000663/* dispatch incoming messages */
664
Ben Lindstrombba81212001-06-25 05:01:22 +0000665static void
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000666process_message(SocketEntry *e)
667{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000668 u_int msg_len, type;
Ben Lindstrom46c16222000-12-22 01:43:59 +0000669 u_char *cp;
Ben Lindstrom61d328a2002-06-06 21:54:57 +0000670
671 /* kill dead keys */
672 reaper();
673
Damien Miller95def091999-11-25 00:26:21 +1100674 if (buffer_len(&e->input) < 5)
675 return; /* Incomplete message. */
Damien Miller708d21c2002-01-22 23:18:15 +1100676 cp = buffer_ptr(&e->input);
Damien Miller95def091999-11-25 00:26:21 +1100677 msg_len = GET_32BIT(cp);
678 if (msg_len > 256 * 1024) {
Damien Miller58f34862002-09-04 16:31:21 +1000679 close_socket(e);
Damien Miller95def091999-11-25 00:26:21 +1100680 return;
681 }
682 if (buffer_len(&e->input) < msg_len + 4)
683 return;
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000684
685 /* move the current input to e->request */
Damien Miller95def091999-11-25 00:26:21 +1100686 buffer_consume(&e->input, 4);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000687 buffer_clear(&e->request);
688 buffer_append(&e->request, buffer_ptr(&e->input), msg_len);
689 buffer_consume(&e->input, msg_len);
690 type = buffer_get_char(&e->request);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000691
Ben Lindstrom2f717042002-06-06 21:52:03 +0000692 /* check wheter agent is locked */
693 if (locked && type != SSH_AGENTC_UNLOCK) {
694 buffer_clear(&e->request);
695 switch (type) {
696 case SSH_AGENTC_REQUEST_RSA_IDENTITIES:
697 case SSH2_AGENTC_REQUEST_IDENTITIES:
698 /* send empty lists */
699 no_identities(e, type);
700 break;
701 default:
702 /* send a fail message for all other request types */
703 buffer_put_int(&e->output, 1);
704 buffer_put_char(&e->output, SSH_AGENT_FAILURE);
705 }
706 return;
707 }
708
Ben Lindstrom3f471632001-07-04 03:53:15 +0000709 debug("type %d", type);
Damien Miller95def091999-11-25 00:26:21 +1100710 switch (type) {
Ben Lindstrom2f717042002-06-06 21:52:03 +0000711 case SSH_AGENTC_LOCK:
712 case SSH_AGENTC_UNLOCK:
713 process_lock_agent(e, type == SSH_AGENTC_LOCK);
714 break;
Damien Millerad833b32000-08-23 10:46:23 +1000715 /* ssh1 */
Damien Miller95def091999-11-25 00:26:21 +1100716 case SSH_AGENTC_RSA_CHALLENGE:
Damien Millerad833b32000-08-23 10:46:23 +1000717 process_authentication_challenge1(e);
718 break;
719 case SSH_AGENTC_REQUEST_RSA_IDENTITIES:
720 process_request_identities(e, 1);
Damien Miller95def091999-11-25 00:26:21 +1100721 break;
722 case SSH_AGENTC_ADD_RSA_IDENTITY:
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000723 case SSH_AGENTC_ADD_RSA_ID_CONSTRAINED:
Damien Millerad833b32000-08-23 10:46:23 +1000724 process_add_identity(e, 1);
Damien Miller95def091999-11-25 00:26:21 +1100725 break;
726 case SSH_AGENTC_REMOVE_RSA_IDENTITY:
Damien Millerad833b32000-08-23 10:46:23 +1000727 process_remove_identity(e, 1);
Damien Miller95def091999-11-25 00:26:21 +1100728 break;
729 case SSH_AGENTC_REMOVE_ALL_RSA_IDENTITIES:
Damien Millerad833b32000-08-23 10:46:23 +1000730 process_remove_all_identities(e, 1);
731 break;
732 /* ssh2 */
733 case SSH2_AGENTC_SIGN_REQUEST:
734 process_sign_request2(e);
735 break;
736 case SSH2_AGENTC_REQUEST_IDENTITIES:
737 process_request_identities(e, 2);
738 break;
739 case SSH2_AGENTC_ADD_IDENTITY:
Ben Lindstrom2b266b72002-06-21 00:08:39 +0000740 case SSH2_AGENTC_ADD_ID_CONSTRAINED:
Damien Millerad833b32000-08-23 10:46:23 +1000741 process_add_identity(e, 2);
742 break;
743 case SSH2_AGENTC_REMOVE_IDENTITY:
744 process_remove_identity(e, 2);
745 break;
746 case SSH2_AGENTC_REMOVE_ALL_IDENTITIES:
747 process_remove_all_identities(e, 2);
Damien Miller95def091999-11-25 00:26:21 +1100748 break;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000749#ifdef SMARTCARD
750 case SSH_AGENTC_ADD_SMARTCARD_KEY:
751 process_add_smartcard_key(e);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100752 break;
Ben Lindstrom3f471632001-07-04 03:53:15 +0000753 case SSH_AGENTC_REMOVE_SMARTCARD_KEY:
754 process_remove_smartcard_key(e);
Damien Miller9f0f5c62001-12-21 14:45:46 +1100755 break;
Ben Lindstromffce1472001-08-06 21:57:31 +0000756#endif /* SMARTCARD */
Damien Miller95def091999-11-25 00:26:21 +1100757 default:
758 /* Unknown message. Respond with failure. */
759 error("Unknown message %d", type);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000760 buffer_clear(&e->request);
Damien Miller95def091999-11-25 00:26:21 +1100761 buffer_put_int(&e->output, 1);
762 buffer_put_char(&e->output, SSH_AGENT_FAILURE);
763 break;
764 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000765}
766
Ben Lindstrombba81212001-06-25 05:01:22 +0000767static void
Ben Lindstrom65366a82001-12-06 16:32:47 +0000768new_socket(sock_type type, int fd)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000769{
Ben Lindstrom46c16222000-12-22 01:43:59 +0000770 u_int i, old_alloc;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000771
Damien Miller95def091999-11-25 00:26:21 +1100772 if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0)
773 error("fcntl O_NONBLOCK: %s", strerror(errno));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000774
Damien Miller95def091999-11-25 00:26:21 +1100775 if (fd > max_fd)
776 max_fd = fd;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000777
Damien Miller95def091999-11-25 00:26:21 +1100778 for (i = 0; i < sockets_alloc; i++)
779 if (sockets[i].type == AUTH_UNUSED) {
780 sockets[i].fd = fd;
781 sockets[i].type = type;
782 buffer_init(&sockets[i].input);
783 buffer_init(&sockets[i].output);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000784 buffer_init(&sockets[i].request);
Damien Miller95def091999-11-25 00:26:21 +1100785 return;
786 }
787 old_alloc = sockets_alloc;
788 sockets_alloc += 10;
789 if (sockets)
790 sockets = xrealloc(sockets, sockets_alloc * sizeof(sockets[0]));
791 else
792 sockets = xmalloc(sockets_alloc * sizeof(sockets[0]));
793 for (i = old_alloc; i < sockets_alloc; i++)
794 sockets[i].type = AUTH_UNUSED;
795 sockets[old_alloc].type = type;
796 sockets[old_alloc].fd = fd;
797 buffer_init(&sockets[old_alloc].input);
798 buffer_init(&sockets[old_alloc].output);
Ben Lindstrom21d1ed82002-06-06 21:48:57 +0000799 buffer_init(&sockets[old_alloc].request);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000800}
801
Ben Lindstrombba81212001-06-25 05:01:22 +0000802static int
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +0000803prepare_select(fd_set **fdrp, fd_set **fdwp, int *fdl, int *nallocp)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000804{
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000805 u_int i, sz;
806 int n = 0;
807
808 for (i = 0; i < sockets_alloc; i++) {
Damien Miller95def091999-11-25 00:26:21 +1100809 switch (sockets[i].type) {
810 case AUTH_SOCKET:
811 case AUTH_CONNECTION:
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000812 n = MAX(n, sockets[i].fd);
Damien Miller95def091999-11-25 00:26:21 +1100813 break;
814 case AUTH_UNUSED:
815 break;
816 default:
817 fatal("Unknown socket type %d", sockets[i].type);
818 break;
819 }
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000820 }
821
822 sz = howmany(n+1, NFDBITS) * sizeof(fd_mask);
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +0000823 if (*fdrp == NULL || sz > *nallocp) {
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000824 if (*fdrp)
Ben Lindstrom86ebcb62001-04-04 01:53:20 +0000825 xfree(*fdrp);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000826 if (*fdwp)
Ben Lindstrom86ebcb62001-04-04 01:53:20 +0000827 xfree(*fdwp);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000828 *fdrp = xmalloc(sz);
829 *fdwp = xmalloc(sz);
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +0000830 *nallocp = sz;
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000831 }
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +0000832 if (n < *fdl)
833 debug("XXX shrink: %d < %d", n, *fdl);
834 *fdl = n;
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000835 memset(*fdrp, 0, sz);
836 memset(*fdwp, 0, sz);
837
838 for (i = 0; i < sockets_alloc; i++) {
839 switch (sockets[i].type) {
840 case AUTH_SOCKET:
841 case AUTH_CONNECTION:
842 FD_SET(sockets[i].fd, *fdrp);
843 if (buffer_len(&sockets[i].output) > 0)
844 FD_SET(sockets[i].fd, *fdwp);
845 break;
846 default:
847 break;
848 }
849 }
850 return (1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000851}
852
Ben Lindstrombba81212001-06-25 05:01:22 +0000853static void
Damien Miller95def091999-11-25 00:26:21 +1100854after_select(fd_set *readset, fd_set *writeset)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000855{
Ben Lindstrom822b6342002-06-23 21:38:49 +0000856 struct sockaddr_un sunaddr;
Damien Miller7684ee12000-03-17 23:40:15 +1100857 socklen_t slen;
Damien Miller95def091999-11-25 00:26:21 +1100858 char buf[1024];
Ben Lindstrom822b6342002-06-23 21:38:49 +0000859 int len, sock;
860 u_int i;
Damien Miller538f1812002-09-12 09:51:10 +1000861 uid_t euid;
862 gid_t egid;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000863
Damien Miller95def091999-11-25 00:26:21 +1100864 for (i = 0; i < sockets_alloc; i++)
865 switch (sockets[i].type) {
866 case AUTH_UNUSED:
867 break;
868 case AUTH_SOCKET:
869 if (FD_ISSET(sockets[i].fd, readset)) {
Damien Miller7684ee12000-03-17 23:40:15 +1100870 slen = sizeof(sunaddr);
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000871 sock = accept(sockets[i].fd,
872 (struct sockaddr *) &sunaddr, &slen);
Damien Miller95def091999-11-25 00:26:21 +1100873 if (sock < 0) {
Damien Millerc653b892002-02-08 22:05:41 +1100874 error("accept from AUTH_SOCKET: %s",
875 strerror(errno));
Damien Miller95def091999-11-25 00:26:21 +1100876 break;
877 }
Damien Miller538f1812002-09-12 09:51:10 +1000878 if (getpeereid(sock, &euid, &egid) < 0) {
879 error("getpeereid %d failed: %s",
880 sock, strerror(errno));
881 close(sock);
882 break;
883 }
Damien Milleraf9de382002-10-03 11:54:35 +1000884 if ((euid != 0) && (getuid() != euid)) {
Damien Miller538f1812002-09-12 09:51:10 +1000885 error("uid mismatch: "
Damien Millerdb30b122002-09-19 11:46:58 +1000886 "peer euid %u != uid %u",
887 (u_int) euid, (u_int) getuid());
Damien Miller538f1812002-09-12 09:51:10 +1000888 close(sock);
889 break;
890 }
Damien Miller95def091999-11-25 00:26:21 +1100891 new_socket(AUTH_CONNECTION, sock);
892 }
893 break;
894 case AUTH_CONNECTION:
895 if (buffer_len(&sockets[i].output) > 0 &&
896 FD_ISSET(sockets[i].fd, writeset)) {
Ben Lindstromb3144e52001-03-06 03:31:34 +0000897 do {
898 len = write(sockets[i].fd,
899 buffer_ptr(&sockets[i].output),
900 buffer_len(&sockets[i].output));
901 if (len == -1 && (errno == EAGAIN ||
902 errno == EINTR))
903 continue;
904 break;
905 } while (1);
Damien Miller95def091999-11-25 00:26:21 +1100906 if (len <= 0) {
Damien Miller58f34862002-09-04 16:31:21 +1000907 close_socket(&sockets[i]);
Damien Miller95def091999-11-25 00:26:21 +1100908 break;
909 }
910 buffer_consume(&sockets[i].output, len);
911 }
912 if (FD_ISSET(sockets[i].fd, readset)) {
Ben Lindstromb3144e52001-03-06 03:31:34 +0000913 do {
914 len = read(sockets[i].fd, buf, sizeof(buf));
915 if (len == -1 && (errno == EAGAIN ||
916 errno == EINTR))
917 continue;
918 break;
919 } while (1);
Damien Miller95def091999-11-25 00:26:21 +1100920 if (len <= 0) {
Damien Miller58f34862002-09-04 16:31:21 +1000921 close_socket(&sockets[i]);
Damien Miller95def091999-11-25 00:26:21 +1100922 break;
923 }
924 buffer_append(&sockets[i].input, buf, len);
925 process_message(&sockets[i]);
926 }
927 break;
928 default:
929 fatal("Unknown type %d", sockets[i].type);
930 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000931}
932
Ben Lindstrombba81212001-06-25 05:01:22 +0000933static void
Damien Millerc653b892002-02-08 22:05:41 +1100934cleanup_socket(void *p)
Damien Miller792c5111999-10-29 09:47:09 +1000935{
Ben Lindstrom77808ab2001-01-26 05:10:34 +0000936 if (socket_name[0])
937 unlink(socket_name);
938 if (socket_dir[0])
939 rmdir(socket_dir);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000940}
941
Ben Lindstrombba81212001-06-25 05:01:22 +0000942static void
Damien Miller792c5111999-10-29 09:47:09 +1000943cleanup_exit(int i)
944{
Damien Millerc653b892002-02-08 22:05:41 +1100945 cleanup_socket(NULL);
Damien Miller95def091999-11-25 00:26:21 +1100946 exit(i);
Damien Miller792c5111999-10-29 09:47:09 +1000947}
948
Ben Lindstrombba81212001-06-25 05:01:22 +0000949static void
Ben Lindstrom77808ab2001-01-26 05:10:34 +0000950cleanup_handler(int sig)
951{
Damien Millerc653b892002-02-08 22:05:41 +1100952 cleanup_socket(NULL);
Ben Lindstrom77808ab2001-01-26 05:10:34 +0000953 _exit(2);
954}
955
Ben Lindstrombba81212001-06-25 05:01:22 +0000956static void
Ben Lindstrom0250da02001-07-22 20:44:00 +0000957check_parent_exists(int sig)
958{
959 int save_errno = errno;
960
961 if (parent_pid != -1 && kill(parent_pid, 0) < 0) {
962 /* printf("Parent has died - Authentication agent exiting.\n"); */
963 cleanup_handler(sig); /* safe */
964 }
965 signal(SIGALRM, check_parent_exists);
966 alarm(10);
967 errno = save_errno;
968}
969
970static void
Ben Lindstrom28072eb2001-02-10 23:13:41 +0000971usage(void)
Damien Miller792c5111999-10-29 09:47:09 +1000972{
Ben Lindstromddfb1e32001-08-06 22:06:35 +0000973 fprintf(stderr, "Usage: %s [options] [command [args ...]]\n",
Kevin Steves29265862001-01-24 14:06:28 +0000974 __progname);
Ben Lindstromddfb1e32001-08-06 22:06:35 +0000975 fprintf(stderr, "Options:\n");
976 fprintf(stderr, " -c Generate C-shell commands on stdout.\n");
977 fprintf(stderr, " -s Generate Bourne shell commands on stdout.\n");
978 fprintf(stderr, " -k Kill the current agent.\n");
979 fprintf(stderr, " -d Debug mode.\n");
Ben Lindstromb7788f32002-06-06 21:46:08 +0000980 fprintf(stderr, " -a socket Bind agent socket to given name.\n");
Damien Miller53d81482003-01-22 11:47:19 +1100981 fprintf(stderr, " -t life Default identity lifetime (seconds).\n");
Damien Miller95def091999-11-25 00:26:21 +1100982 exit(1);
Damien Miller792c5111999-10-29 09:47:09 +1000983}
984
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000985int
986main(int ac, char **av)
987{
Damien Miller6c711792003-01-24 11:36:23 +1100988 int c_flag = 0, d_flag = 0, k_flag = 0, s_flag = 0;
989 int sock, fd, ch, nalloc;
Ben Lindstrom822b6342002-06-23 21:38:49 +0000990 char *shell, *format, *pidstr, *agentsocket = NULL;
991 fd_set *readsetp = NULL, *writesetp = NULL;
Damien Miller95def091999-11-25 00:26:21 +1100992 struct sockaddr_un sunaddr;
Ben Lindstrom2c467a22000-12-27 04:57:41 +0000993#ifdef HAVE_SETRLIMIT
Ben Lindstromc72745a2000-12-02 19:03:54 +0000994 struct rlimit rlim;
Ben Lindstrom2c467a22000-12-27 04:57:41 +0000995#endif
Ben Lindstrom3524d692001-05-03 22:59:24 +0000996#ifdef HAVE_CYGWIN
997 int prev_mask;
998#endif
Damien Miller615f9392000-05-17 22:53:33 +1000999 extern int optind;
Ben Lindstrom883844d2002-06-23 00:20:50 +00001000 extern char *optarg;
Ben Lindstrom822b6342002-06-23 21:38:49 +00001001 pid_t pid;
1002 char pidstrbuf[1 + 3 * sizeof pid];
Kevin Stevesde41bc62000-12-14 00:04:08 +00001003
Damien Miller6cffb9a2002-09-04 16:20:26 +10001004 /* drop */
1005 setegid(getgid());
1006 setgid(getgid());
1007
Ben Lindstromd09fcf52001-03-29 00:29:54 +00001008 SSLeay_add_all_algorithms();
1009
Ben Lindstrom49a79c02000-11-17 03:47:20 +00001010 __progname = get_progname(av[0]);
Damien Millerf9b625c2000-07-09 22:42:32 +10001011 init_rng();
Damien Miller60bc5172001-03-19 09:38:15 +11001012 seed_rng();
Kevin Stevesef4eea92001-02-05 12:42:17 +00001013
Damien Miller53d81482003-01-22 11:47:19 +11001014 while ((ch = getopt(ac, av, "cdksa:t:")) != -1) {
Damien Miller95def091999-11-25 00:26:21 +11001015 switch (ch) {
1016 case 'c':
1017 if (s_flag)
1018 usage();
1019 c_flag++;
1020 break;
1021 case 'k':
1022 k_flag++;
1023 break;
1024 case 's':
1025 if (c_flag)
1026 usage();
1027 s_flag++;
1028 break;
Ben Lindstromd94580c2001-07-04 03:48:02 +00001029 case 'd':
1030 if (d_flag)
1031 usage();
1032 d_flag++;
1033 break;
Ben Lindstromb7788f32002-06-06 21:46:08 +00001034 case 'a':
1035 agentsocket = optarg;
1036 break;
Damien Miller53d81482003-01-22 11:47:19 +11001037 case 't':
1038 if ((lifetime = convtime(optarg)) == -1) {
1039 fprintf(stderr, "Invalid lifetime\n");
1040 usage();
1041 }
1042 break;
Damien Miller95def091999-11-25 00:26:21 +11001043 default:
1044 usage();
1045 }
Damien Miller792c5111999-10-29 09:47:09 +10001046 }
Damien Miller95def091999-11-25 00:26:21 +11001047 ac -= optind;
1048 av += optind;
Damien Miller792c5111999-10-29 09:47:09 +10001049
Ben Lindstromd94580c2001-07-04 03:48:02 +00001050 if (ac > 0 && (c_flag || k_flag || s_flag || d_flag))
Damien Miller95def091999-11-25 00:26:21 +11001051 usage();
Damien Miller792c5111999-10-29 09:47:09 +10001052
Ben Lindstromeecdf232002-04-02 21:03:51 +00001053 if (ac == 0 && !c_flag && !s_flag) {
Damien Miller95def091999-11-25 00:26:21 +11001054 shell = getenv("SHELL");
1055 if (shell != NULL && strncmp(shell + strlen(shell) - 3, "csh", 3) == 0)
1056 c_flag = 1;
Damien Miller792c5111999-10-29 09:47:09 +10001057 }
Damien Miller95def091999-11-25 00:26:21 +11001058 if (k_flag) {
1059 pidstr = getenv(SSH_AGENTPID_ENV_NAME);
1060 if (pidstr == NULL) {
1061 fprintf(stderr, "%s not set, cannot kill agent\n",
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001062 SSH_AGENTPID_ENV_NAME);
Damien Miller95def091999-11-25 00:26:21 +11001063 exit(1);
1064 }
1065 pid = atoi(pidstr);
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001066 if (pid < 1) {
Damien Miller95def091999-11-25 00:26:21 +11001067 fprintf(stderr, "%s=\"%s\", which is not a good PID\n",
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001068 SSH_AGENTPID_ENV_NAME, pidstr);
Damien Miller95def091999-11-25 00:26:21 +11001069 exit(1);
1070 }
1071 if (kill(pid, SIGTERM) == -1) {
1072 perror("kill");
1073 exit(1);
1074 }
1075 format = c_flag ? "unsetenv %s;\n" : "unset %s;\n";
1076 printf(format, SSH_AUTHSOCKET_ENV_NAME);
1077 printf(format, SSH_AGENTPID_ENV_NAME);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001078 printf("echo Agent pid %ld killed;\n", (long)pid);
Damien Miller95def091999-11-25 00:26:21 +11001079 exit(0);
Damien Miller792c5111999-10-29 09:47:09 +10001080 }
Damien Miller95def091999-11-25 00:26:21 +11001081 parent_pid = getpid();
1082
Ben Lindstromb7788f32002-06-06 21:46:08 +00001083 if (agentsocket == NULL) {
1084 /* Create private directory for agent socket */
1085 strlcpy(socket_dir, "/tmp/ssh-XXXXXXXX", sizeof socket_dir);
1086 if (mkdtemp(socket_dir) == NULL) {
1087 perror("mkdtemp: private socket dir");
1088 exit(1);
1089 }
Ben Lindstromce0f6342002-06-11 16:42:49 +00001090 snprintf(socket_name, sizeof socket_name, "%s/agent.%ld", socket_dir,
1091 (long)parent_pid);
Ben Lindstromb7788f32002-06-06 21:46:08 +00001092 } else {
1093 /* Try to use specified agent socket */
1094 socket_dir[0] = '\0';
1095 strlcpy(socket_name, agentsocket, sizeof socket_name);
Damien Miller792c5111999-10-29 09:47:09 +10001096 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001097
Damien Miller5428f641999-11-25 11:54:57 +11001098 /*
1099 * Create socket early so it will exist before command gets run from
1100 * the parent.
1101 */
Damien Miller95def091999-11-25 00:26:21 +11001102 sock = socket(AF_UNIX, SOCK_STREAM, 0);
1103 if (sock < 0) {
1104 perror("socket");
1105 cleanup_exit(1);
Damien Miller792c5111999-10-29 09:47:09 +10001106 }
Damien Miller95def091999-11-25 00:26:21 +11001107 memset(&sunaddr, 0, sizeof(sunaddr));
1108 sunaddr.sun_family = AF_UNIX;
1109 strlcpy(sunaddr.sun_path, socket_name, sizeof(sunaddr.sun_path));
Ben Lindstrom3524d692001-05-03 22:59:24 +00001110#ifdef HAVE_CYGWIN
1111 prev_mask = umask(0177);
1112#endif
Damien Miller95def091999-11-25 00:26:21 +11001113 if (bind(sock, (struct sockaddr *) & sunaddr, sizeof(sunaddr)) < 0) {
1114 perror("bind");
Ben Lindstrom3524d692001-05-03 22:59:24 +00001115#ifdef HAVE_CYGWIN
1116 umask(prev_mask);
1117#endif
Damien Miller95def091999-11-25 00:26:21 +11001118 cleanup_exit(1);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001119 }
Ben Lindstrom3524d692001-05-03 22:59:24 +00001120#ifdef HAVE_CYGWIN
1121 umask(prev_mask);
1122#endif
Damien Miller4efdfff2002-09-04 16:28:18 +10001123 if (listen(sock, 128) < 0) {
Damien Miller95def091999-11-25 00:26:21 +11001124 perror("listen");
1125 cleanup_exit(1);
1126 }
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001127
Damien Miller5428f641999-11-25 11:54:57 +11001128 /*
1129 * Fork, and have the parent execute the command, if any, or present
1130 * the socket data. The child continues as the authentication agent.
1131 */
Ben Lindstromd94580c2001-07-04 03:48:02 +00001132 if (d_flag) {
1133 log_init(__progname, SYSLOG_LEVEL_DEBUG1, SYSLOG_FACILITY_AUTH, 1);
1134 format = c_flag ? "setenv %s %s;\n" : "%s=%s; export %s;\n";
1135 printf(format, SSH_AUTHSOCKET_ENV_NAME, socket_name,
1136 SSH_AUTHSOCKET_ENV_NAME);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001137 printf("echo Agent pid %ld;\n", (long)parent_pid);
Ben Lindstromd94580c2001-07-04 03:48:02 +00001138 goto skip;
1139 }
Damien Miller95def091999-11-25 00:26:21 +11001140 pid = fork();
1141 if (pid == -1) {
1142 perror("fork");
Damien Millerc653b892002-02-08 22:05:41 +11001143 cleanup_exit(1);
Damien Miller95def091999-11-25 00:26:21 +11001144 }
1145 if (pid != 0) { /* Parent - execute the given command. */
1146 close(sock);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001147 snprintf(pidstrbuf, sizeof pidstrbuf, "%ld", (long)pid);
Damien Miller95def091999-11-25 00:26:21 +11001148 if (ac == 0) {
1149 format = c_flag ? "setenv %s %s;\n" : "%s=%s; export %s;\n";
1150 printf(format, SSH_AUTHSOCKET_ENV_NAME, socket_name,
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001151 SSH_AUTHSOCKET_ENV_NAME);
Damien Miller95def091999-11-25 00:26:21 +11001152 printf(format, SSH_AGENTPID_ENV_NAME, pidstrbuf,
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001153 SSH_AGENTPID_ENV_NAME);
Ben Lindstromce0f6342002-06-11 16:42:49 +00001154 printf("echo Agent pid %ld;\n", (long)pid);
Damien Miller95def091999-11-25 00:26:21 +11001155 exit(0);
1156 }
Damien Millere4340be2000-09-16 13:29:08 +11001157 if (setenv(SSH_AUTHSOCKET_ENV_NAME, socket_name, 1) == -1 ||
1158 setenv(SSH_AGENTPID_ENV_NAME, pidstrbuf, 1) == -1) {
1159 perror("setenv");
1160 exit(1);
1161 }
Damien Miller95def091999-11-25 00:26:21 +11001162 execvp(av[0], av);
1163 perror(av[0]);
1164 exit(1);
1165 }
Damien Millerc653b892002-02-08 22:05:41 +11001166 /* child */
1167 log_init(__progname, SYSLOG_LEVEL_INFO, SYSLOG_FACILITY_AUTH, 0);
Ben Lindstrom3fdf8762001-07-22 20:40:24 +00001168
1169 if (setsid() == -1) {
Damien Millerc653b892002-02-08 22:05:41 +11001170 error("setsid: %s", strerror(errno));
Ben Lindstrom3fdf8762001-07-22 20:40:24 +00001171 cleanup_exit(1);
1172 }
1173
1174 (void)chdir("/");
Damien Miller6c711792003-01-24 11:36:23 +11001175 if ((fd = open(_PATH_DEVNULL, O_RDWR, 0)) != -1) {
1176 /* XXX might close listen socket */
1177 (void)dup2(fd, STDIN_FILENO);
1178 (void)dup2(fd, STDOUT_FILENO);
1179 (void)dup2(fd, STDERR_FILENO);
1180 if (fd > 2)
1181 close(fd);
1182 }
Damien Miller95def091999-11-25 00:26:21 +11001183
Ben Lindstrom2c467a22000-12-27 04:57:41 +00001184#ifdef HAVE_SETRLIMIT
Ben Lindstromc72745a2000-12-02 19:03:54 +00001185 /* deny core dumps, since memory contains unencrypted private keys */
1186 rlim.rlim_cur = rlim.rlim_max = 0;
1187 if (setrlimit(RLIMIT_CORE, &rlim) < 0) {
Damien Millerc653b892002-02-08 22:05:41 +11001188 error("setrlimit RLIMIT_CORE: %s", strerror(errno));
Ben Lindstromc72745a2000-12-02 19:03:54 +00001189 cleanup_exit(1);
1190 }
Ben Lindstrom2c467a22000-12-27 04:57:41 +00001191#endif
Ben Lindstromd94580c2001-07-04 03:48:02 +00001192
1193skip:
Damien Millerc653b892002-02-08 22:05:41 +11001194 fatal_add_cleanup(cleanup_socket, NULL);
Damien Miller95def091999-11-25 00:26:21 +11001195 new_socket(AUTH_SOCKET, sock);
1196 if (ac > 0) {
1197 signal(SIGALRM, check_parent_exists);
1198 alarm(10);
1199 }
Damien Millerad833b32000-08-23 10:46:23 +10001200 idtab_init();
Damien Miller48bfa9c2001-07-14 12:12:55 +10001201 if (!d_flag)
Ben Lindstromd94580c2001-07-04 03:48:02 +00001202 signal(SIGINT, SIG_IGN);
Damien Miller48bfa9c2001-07-14 12:12:55 +10001203 signal(SIGPIPE, SIG_IGN);
Ben Lindstrom77808ab2001-01-26 05:10:34 +00001204 signal(SIGHUP, cleanup_handler);
1205 signal(SIGTERM, cleanup_handler);
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +00001206 nalloc = 0;
1207
Damien Miller95def091999-11-25 00:26:21 +11001208 while (1) {
Ben Lindstroma3d5a4c2001-07-18 15:58:08 +00001209 prepare_select(&readsetp, &writesetp, &max_fd, &nalloc);
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001210 if (select(max_fd + 1, readsetp, writesetp, NULL, NULL) < 0) {
Damien Miller95def091999-11-25 00:26:21 +11001211 if (errno == EINTR)
1212 continue;
Damien Millerc653b892002-02-08 22:05:41 +11001213 fatal("select: %s", strerror(errno));
Damien Miller95def091999-11-25 00:26:21 +11001214 }
Ben Lindstrom226cfa02001-01-22 05:34:40 +00001215 after_select(readsetp, writesetp);
Damien Miller95def091999-11-25 00:26:21 +11001216 }
1217 /* NOTREACHED */
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001218}