blob: 086c0d345e2a703067f2fa16778c5e52df44ef29 [file] [log] [blame]
Darren Tucker75456e82010-01-08 18:55:58 +11001/* $OpenBSD: ssh-keyscan.c,v 1.80 2009/12/25 19:40:21 stevesk Exp $ */
Ben Lindstromb6434ae2000-12-05 01:15:09 +00002/*
3 * Copyright 1995, 1996 by David Mazieres <dm@lcs.mit.edu>.
4 *
5 * Modification and redistribution in source and binary forms is
6 * permitted provided that due credit is given to the author and the
Ben Lindstroma238f6e2001-06-09 01:30:39 +00007 * OpenBSD project by leaving this copyright notice intact.
Ben Lindstromb6434ae2000-12-05 01:15:09 +00008 */
9
10#include "includes.h"
Damien Millercd4223c2006-03-15 11:22:47 +110011
Damien Miller9b481512002-09-12 10:43:29 +100012#include "openbsd-compat/sys-queue.h"
Damien Millercd4223c2006-03-15 11:22:47 +110013#include <sys/resource.h>
Damien Miller9aec9192006-08-05 10:57:45 +100014#ifdef HAVE_SYS_TIME_H
15# include <sys/time.h>
16#endif
Damien Millere3476ed2006-07-24 14:13:33 +100017
Darren Tucker46aa3e02006-09-02 15:32:40 +100018#include <netinet/in.h>
19#include <arpa/inet.h>
20
Damien Millere3476ed2006-07-24 14:13:33 +100021#include <openssl/bn.h>
22
Damien Millerb8fe89c2006-07-24 14:51:00 +100023#include <netdb.h>
Darren Tuckerdeecec92006-07-12 22:44:34 +100024#include <errno.h>
Damien Millerbe43ebf2006-07-24 13:51:51 +100025#include <setjmp.h>
Damien Millere7a1e5c2006-08-05 11:34:19 +100026#include <stdarg.h>
Damien Millera7a73ee2006-08-05 11:37:59 +100027#include <stdio.h>
Damien Millere7a1e5c2006-08-05 11:34:19 +100028#include <stdlib.h>
Damien Millerd7834352006-08-05 12:39:39 +100029#include <signal.h>
Damien Millere3476ed2006-07-24 14:13:33 +100030#include <string.h>
Damien Millere6b3b612006-07-24 14:01:23 +100031#include <unistd.h>
Ben Lindstromb6434ae2000-12-05 01:15:09 +000032
Ben Lindstromb6434ae2000-12-05 01:15:09 +000033#include "xmalloc.h"
34#include "ssh.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000035#include "ssh1.h"
Damien Millerd7834352006-08-05 12:39:39 +100036#include "buffer.h"
Ben Lindstromb6434ae2000-12-05 01:15:09 +000037#include "key.h"
Damien Millerd7834352006-08-05 12:39:39 +100038#include "cipher.h"
Ben Lindstrom325e70c2001-08-06 22:41:30 +000039#include "kex.h"
40#include "compat.h"
41#include "myproposal.h"
42#include "packet.h"
43#include "dispatch.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000044#include "log.h"
Ben Lindstromd20b8552001-03-05 07:01:18 +000045#include "atomicio.h"
Ben Lindstrom325e70c2001-08-06 22:41:30 +000046#include "misc.h"
Damien Millerdb7b8172005-03-01 21:48:03 +110047#include "hostfile.h"
Ben Lindstromb6434ae2000-12-05 01:15:09 +000048
Ben Lindstrom325e70c2001-08-06 22:41:30 +000049/* Flag indicating whether IPv4 or IPv6. This can be set on the command line.
50 Default value is AF_UNSPEC means both IPv4 and IPv6. */
Ben Lindstrom325e70c2001-08-06 22:41:30 +000051int IPv4or6 = AF_UNSPEC;
Ben Lindstromb6434ae2000-12-05 01:15:09 +000052
Ben Lindstrom325e70c2001-08-06 22:41:30 +000053int ssh_port = SSH_DEFAULT_PORT;
Kevin Steves76e7d9b2001-09-20 20:30:09 +000054
55#define KT_RSA1 1
56#define KT_DSA 2
57#define KT_RSA 4
58
Damien Millerbacb7fb2008-05-19 14:56:33 +100059int get_keytypes = KT_RSA; /* Get only RSA keys by default */
Ben Lindstromb6434ae2000-12-05 01:15:09 +000060
Damien Millerdb7b8172005-03-01 21:48:03 +110061int hash_hosts = 0; /* Hash hostname on output */
62
Ben Lindstromb6434ae2000-12-05 01:15:09 +000063#define MAXMAXFD 256
64
65/* The number of seconds after which to give up on a TCP connection */
66int timeout = 5;
67
68int maxfd;
Ben Lindstromd20b8552001-03-05 07:01:18 +000069#define MAXCON (maxfd - 10)
Ben Lindstromb6434ae2000-12-05 01:15:09 +000070
Darren Tucker34e314d2010-01-08 17:03:46 +110071/* The default routing domain */
72int scan_rdomain = -1;
73
Kevin Stevesec84dc12000-12-13 17:45:15 +000074extern char *__progname;
Ben Lindstromc1e04212001-03-05 07:04:38 +000075fd_set *read_wait;
Damien Miller07d86be2006-03-26 14:19:21 +110076size_t read_wait_nfdset;
Ben Lindstromb6434ae2000-12-05 01:15:09 +000077int ncon;
Ben Lindstrom325e70c2001-08-06 22:41:30 +000078int nonfatal_fatal = 0;
79jmp_buf kexjmp;
Ben Lindstrom520b55c2001-09-12 18:05:05 +000080Key *kexjmp_key;
Ben Lindstromb6434ae2000-12-05 01:15:09 +000081
82/*
83 * Keep a connection structure for each file descriptor. The state
84 * associated with file descriptor n is held in fdcon[n].
85 */
86typedef struct Connection {
Ben Lindstrom46c16222000-12-22 01:43:59 +000087 u_char c_status; /* State of connection on this file desc. */
Ben Lindstromb6434ae2000-12-05 01:15:09 +000088#define CS_UNUSED 0 /* File descriptor unused */
89#define CS_CON 1 /* Waiting to connect/read greeting */
90#define CS_SIZE 2 /* Waiting to read initial packet size */
91#define CS_KEYS 3 /* Waiting to read public key packet */
92 int c_fd; /* Quick lookup: c->c_fd == c - fdcon */
93 int c_plen; /* Packet length field for ssh packet */
94 int c_len; /* Total bytes which must be read. */
95 int c_off; /* Length of data read so far. */
Ben Lindstrom325e70c2001-08-06 22:41:30 +000096 int c_keytype; /* Only one of KT_RSA1, KT_DSA, or KT_RSA */
Ben Lindstromb6434ae2000-12-05 01:15:09 +000097 char *c_namebase; /* Address to free for c_name and c_namelist */
98 char *c_name; /* Hostname of connection for errors */
99 char *c_namelist; /* Pointer to other possible addresses */
100 char *c_output_name; /* Hostname of connection for output */
101 char *c_data; /* Data read from this fd */
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000102 Kex *c_kex; /* The key-exchange struct for ssh2 */
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000103 struct timeval c_tv; /* Time at which connection gets aborted */
104 TAILQ_ENTRY(Connection) c_link; /* List of connections in timeout order. */
105} con;
106
107TAILQ_HEAD(conlist, Connection) tq; /* Timeout Queue */
108con *fdcon;
109
110/*
111 * This is just a wrapper around fgets() to make it usable.
112 */
113
114/* Stress-test. Increase this later. */
115#define LINEBUF_SIZE 16
116
117typedef struct {
118 char *buf;
Ben Lindstrom46c16222000-12-22 01:43:59 +0000119 u_int size;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000120 int lineno;
121 const char *filename;
122 FILE *stream;
123 void (*errfun) (const char *,...);
124} Linebuf;
125
Ben Lindstrombba81212001-06-25 05:01:22 +0000126static Linebuf *
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000127Linebuf_alloc(const char *filename, void (*errfun) (const char *,...))
128{
129 Linebuf *lb;
130
131 if (!(lb = malloc(sizeof(*lb)))) {
132 if (errfun)
Ben Lindstrom04f9af72002-07-04 00:03:56 +0000133 (*errfun) ("linebuf (%s): malloc failed\n",
134 filename ? filename : "(stdin)");
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000135 return (NULL);
136 }
137 if (filename) {
138 lb->filename = filename;
139 if (!(lb->stream = fopen(filename, "r"))) {
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000140 xfree(lb);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000141 if (errfun)
142 (*errfun) ("%s: %s\n", filename, strerror(errno));
143 return (NULL);
144 }
145 } else {
146 lb->filename = "(stdin)";
147 lb->stream = stdin;
148 }
149
Damien Miller3bbaba62006-03-26 13:59:38 +1100150 if (!(lb->buf = malloc((lb->size = LINEBUF_SIZE)))) {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000151 if (errfun)
152 (*errfun) ("linebuf (%s): malloc failed\n", lb->filename);
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000153 xfree(lb);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000154 return (NULL);
155 }
156 lb->errfun = errfun;
157 lb->lineno = 0;
158 return (lb);
159}
160
Ben Lindstrombba81212001-06-25 05:01:22 +0000161static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000162Linebuf_free(Linebuf * lb)
163{
164 fclose(lb->stream);
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000165 xfree(lb->buf);
166 xfree(lb);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000167}
168
Ben Lindstrombba81212001-06-25 05:01:22 +0000169#if 0
170static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000171Linebuf_restart(Linebuf * lb)
172{
173 clearerr(lb->stream);
174 rewind(lb->stream);
175 lb->lineno = 0;
176}
177
Ben Lindstrombba81212001-06-25 05:01:22 +0000178static int
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000179Linebuf_lineno(Linebuf * lb)
180{
181 return (lb->lineno);
182}
Ben Lindstrombba81212001-06-25 05:01:22 +0000183#endif
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000184
Ben Lindstrombba81212001-06-25 05:01:22 +0000185static char *
Ben Lindstromc791beb2001-02-10 23:18:11 +0000186Linebuf_getline(Linebuf * lb)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000187{
Damien Millereccb9de2005-06-17 12:59:34 +1000188 size_t n = 0;
Ben Lindstrom965710f2002-07-07 22:17:22 +0000189 void *p;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000190
191 lb->lineno++;
192 for (;;) {
193 /* Read a line */
194 if (!fgets(&lb->buf[n], lb->size - n, lb->stream)) {
195 if (ferror(lb->stream) && lb->errfun)
Ben Lindstrom965710f2002-07-07 22:17:22 +0000196 (*lb->errfun)("%s: %s\n", lb->filename,
Ben Lindstromb0a4cd82001-03-05 04:54:49 +0000197 strerror(errno));
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000198 return (NULL);
199 }
200 n = strlen(lb->buf);
201
202 /* Return it or an error if it fits */
203 if (n > 0 && lb->buf[n - 1] == '\n') {
204 lb->buf[n - 1] = '\0';
205 return (lb->buf);
206 }
207 if (n != lb->size - 1) {
208 if (lb->errfun)
Ben Lindstrom965710f2002-07-07 22:17:22 +0000209 (*lb->errfun)("%s: skipping incomplete last line\n",
Ben Lindstromb0a4cd82001-03-05 04:54:49 +0000210 lb->filename);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000211 return (NULL);
212 }
213 /* Double the buffer if we need more space */
Ben Lindstrom965710f2002-07-07 22:17:22 +0000214 lb->size *= 2;
215 if ((p = realloc(lb->buf, lb->size)) == NULL) {
216 lb->size /= 2;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000217 if (lb->errfun)
Ben Lindstrom965710f2002-07-07 22:17:22 +0000218 (*lb->errfun)("linebuf (%s): realloc failed\n",
Ben Lindstromb0a4cd82001-03-05 04:54:49 +0000219 lb->filename);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000220 return (NULL);
221 }
Ben Lindstrom965710f2002-07-07 22:17:22 +0000222 lb->buf = p;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000223 }
224}
225
Ben Lindstrombba81212001-06-25 05:01:22 +0000226static int
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000227fdlim_get(int hard)
228{
Ben Lindstrom5adbad22000-12-27 07:06:21 +0000229#if defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000230 struct rlimit rlfd;
Ben Lindstromb0a4cd82001-03-05 04:54:49 +0000231
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000232 if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0)
233 return (-1);
234 if ((hard ? rlfd.rlim_max : rlfd.rlim_cur) == RLIM_INFINITY)
Damien Millere00074a2003-11-24 13:07:45 +1100235 return SSH_SYSFDMAX;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000236 else
237 return hard ? rlfd.rlim_max : rlfd.rlim_cur;
Ben Lindstrom2c467a22000-12-27 04:57:41 +0000238#else
Damien Millere00074a2003-11-24 13:07:45 +1100239 return SSH_SYSFDMAX;
Ben Lindstrom2c467a22000-12-27 04:57:41 +0000240#endif
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000241}
242
Ben Lindstrombba81212001-06-25 05:01:22 +0000243static int
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000244fdlim_set(int lim)
245{
Ben Lindstrom5adbad22000-12-27 07:06:21 +0000246#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000247 struct rlimit rlfd;
Ben Lindstrom2c467a22000-12-27 04:57:41 +0000248#endif
Ben Lindstrom5c98db52002-07-07 22:25:29 +0000249
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000250 if (lim <= 0)
251 return (-1);
Ben Lindstrom5adbad22000-12-27 07:06:21 +0000252#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000253 if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0)
254 return (-1);
255 rlfd.rlim_cur = lim;
256 if (setrlimit(RLIMIT_NOFILE, &rlfd) < 0)
257 return (-1);
Ben Lindstrom2c467a22000-12-27 04:57:41 +0000258#elif defined (HAVE_SETDTABLESIZE)
Kevin Steves28a7f262001-02-05 15:43:59 +0000259 setdtablesize(lim);
Ben Lindstrom2c467a22000-12-27 04:57:41 +0000260#endif
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000261 return (0);
262}
263
264/*
265 * This is an strsep function that returns a null field for adjacent
266 * separators. This is the same as the 4.4BSD strsep, but different from the
267 * one in the GNU libc.
268 */
Ben Lindstrombba81212001-06-25 05:01:22 +0000269static char *
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000270xstrsep(char **str, const char *delim)
271{
272 char *s, *e;
273
274 if (!**str)
275 return (NULL);
276
277 s = *str;
278 e = s + strcspn(s, delim);
279
280 if (*e != '\0')
281 *e++ = '\0';
282 *str = e;
283
284 return (s);
285}
286
287/*
288 * Get the next non-null token (like GNU strsep). Strsep() will return a
289 * null token for two adjacent separators, so we may have to loop.
290 */
Ben Lindstrombba81212001-06-25 05:01:22 +0000291static char *
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000292strnnsep(char **stringp, char *delim)
293{
294 char *tok;
295
296 do {
297 tok = xstrsep(stringp, delim);
298 } while (tok && *tok == '\0');
299 return (tok);
300}
301
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000302static Key *
303keygrab_ssh1(con *c)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000304{
305 static Key *rsa;
306 static Buffer msg;
307
308 if (rsa == NULL) {
309 buffer_init(&msg);
310 rsa = key_new(KEY_RSA1);
311 }
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000312 buffer_append(&msg, c->c_data, c->c_plen);
313 buffer_consume(&msg, 8 - (c->c_plen & 7)); /* padding */
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000314 if (buffer_get_char(&msg) != (int) SSH_SMSG_PUBLIC_KEY) {
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000315 error("%s: invalid packet type", c->c_name);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000316 buffer_clear(&msg);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000317 return NULL;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000318 }
319 buffer_consume(&msg, 8); /* cookie */
320
321 /* server key */
322 (void) buffer_get_int(&msg);
323 buffer_get_bignum(&msg, rsa->rsa->e);
324 buffer_get_bignum(&msg, rsa->rsa->n);
325
326 /* host key */
327 (void) buffer_get_int(&msg);
328 buffer_get_bignum(&msg, rsa->rsa->e);
329 buffer_get_bignum(&msg, rsa->rsa->n);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000330
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000331 buffer_clear(&msg);
332
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000333 return (rsa);
334}
335
336static int
337hostjump(Key *hostkey)
338{
Ben Lindstrom520b55c2001-09-12 18:05:05 +0000339 kexjmp_key = hostkey;
340 longjmp(kexjmp, 1);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000341}
342
343static int
344ssh2_capable(int remote_major, int remote_minor)
345{
346 switch (remote_major) {
347 case 1:
348 if (remote_minor == 99)
349 return 1;
350 break;
351 case 2:
352 return 1;
353 default:
354 break;
355 }
356 return 0;
357}
358
359static Key *
360keygrab_ssh2(con *c)
361{
362 int j;
363
364 packet_set_connection(c->c_fd, c->c_fd);
365 enable_compat20();
366 myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = c->c_keytype == KT_DSA?
367 "ssh-dss": "ssh-rsa";
368 c->c_kex = kex_setup(myproposal);
Damien Miller8e7fb332003-02-24 12:03:03 +1100369 c->c_kex->kex[KEX_DH_GRP1_SHA1] = kexdh_client;
Damien Millerf675fc42004-06-15 10:30:09 +1000370 c->c_kex->kex[KEX_DH_GRP14_SHA1] = kexdh_client;
Damien Miller8e7fb332003-02-24 12:03:03 +1100371 c->c_kex->kex[KEX_DH_GEX_SHA1] = kexgex_client;
Damien Millera63128d2006-03-15 12:08:28 +1100372 c->c_kex->kex[KEX_DH_GEX_SHA256] = kexgex_client;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000373 c->c_kex->verify_host_key = hostjump;
374
375 if (!(j = setjmp(kexjmp))) {
376 nonfatal_fatal = 1;
377 dispatch_run(DISPATCH_BLOCK, &c->c_kex->done, c->c_kex);
378 fprintf(stderr, "Impossible! dispatch_run() returned!\n");
379 exit(1);
380 }
381 nonfatal_fatal = 0;
382 xfree(c->c_kex);
383 c->c_kex = NULL;
384 packet_close();
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000385
Ben Lindstrom520b55c2001-09-12 18:05:05 +0000386 return j < 0? NULL : kexjmp_key;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000387}
388
389static void
390keyprint(con *c, Key *key)
391{
Damien Millerdb7b8172005-03-01 21:48:03 +1100392 char *host = c->c_output_name ? c->c_output_name : c->c_name;
393
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000394 if (!key)
395 return;
Damien Millerdb7b8172005-03-01 21:48:03 +1100396 if (hash_hosts && (host = host_hash(host, NULL, 0)) == NULL)
397 fatal("host_hash failed");
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000398
Damien Millerdb7b8172005-03-01 21:48:03 +1100399 fprintf(stdout, "%s ", host);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000400 key_write(key, stdout);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000401 fputs("\n", stdout);
402}
403
Ben Lindstrombba81212001-06-25 05:01:22 +0000404static int
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000405tcpconnect(char *host)
406{
407 struct addrinfo hints, *ai, *aitop;
408 char strport[NI_MAXSERV];
409 int gaierr, s = -1;
410
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000411 snprintf(strport, sizeof strport, "%d", ssh_port);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000412 memset(&hints, 0, sizeof(hints));
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000413 hints.ai_family = IPv4or6;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000414 hints.ai_socktype = SOCK_STREAM;
415 if ((gaierr = getaddrinfo(host, strport, &hints, &aitop)) != 0)
Darren Tucker4abde772007-12-29 02:43:51 +1100416 fatal("getaddrinfo %s: %s", host, ssh_gai_strerror(gaierr));
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000417 for (ai = aitop; ai; ai = ai->ai_next) {
Darren Tucker34e314d2010-01-08 17:03:46 +1100418 s = socket_rdomain(ai->ai_family, ai->ai_socktype,
419 ai->ai_protocol, scan_rdomain);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000420 if (s < 0) {
421 error("socket: %s", strerror(errno));
422 continue;
423 }
Damien Miller232711f2004-06-15 10:35:30 +1000424 if (set_nonblock(s) == -1)
425 fatal("%s: set_nonblock(%d)", __func__, s);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000426 if (connect(s, ai->ai_addr, ai->ai_addrlen) < 0 &&
427 errno != EINPROGRESS)
428 error("connect (`%s'): %s", host, strerror(errno));
429 else
430 break;
431 close(s);
432 s = -1;
433 }
434 freeaddrinfo(aitop);
435 return s;
436}
437
Ben Lindstrombba81212001-06-25 05:01:22 +0000438static int
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000439conalloc(char *iname, char *oname, int keytype)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000440{
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000441 char *namebase, *name, *namelist;
Ben Lindstrom965710f2002-07-07 22:17:22 +0000442 int s;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000443
444 namebase = namelist = xstrdup(iname);
445
446 do {
447 name = xstrsep(&namelist, ",");
448 if (!name) {
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000449 xfree(namebase);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000450 return (-1);
451 }
452 } while ((s = tcpconnect(name)) < 0);
453
454 if (s >= maxfd)
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000455 fatal("conalloc: fdno %d too high", s);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000456 if (fdcon[s].c_status)
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000457 fatal("conalloc: attempt to reuse fdno %d", s);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000458
459 fdcon[s].c_fd = s;
460 fdcon[s].c_status = CS_CON;
461 fdcon[s].c_namebase = namebase;
462 fdcon[s].c_name = name;
463 fdcon[s].c_namelist = namelist;
464 fdcon[s].c_output_name = xstrdup(oname);
465 fdcon[s].c_data = (char *) &fdcon[s].c_plen;
466 fdcon[s].c_len = 4;
467 fdcon[s].c_off = 0;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000468 fdcon[s].c_keytype = keytype;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000469 gettimeofday(&fdcon[s].c_tv, NULL);
470 fdcon[s].c_tv.tv_sec += timeout;
471 TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link);
Ben Lindstromc1e04212001-03-05 07:04:38 +0000472 FD_SET(s, read_wait);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000473 ncon++;
474 return (s);
475}
476
Ben Lindstrombba81212001-06-25 05:01:22 +0000477static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000478confree(int s)
479{
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000480 if (s >= maxfd || fdcon[s].c_status == CS_UNUSED)
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000481 fatal("confree: attempt to free bad fdno %d", s);
Ben Lindstromd20b8552001-03-05 07:01:18 +0000482 close(s);
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000483 xfree(fdcon[s].c_namebase);
484 xfree(fdcon[s].c_output_name);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000485 if (fdcon[s].c_status == CS_KEYS)
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000486 xfree(fdcon[s].c_data);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000487 fdcon[s].c_status = CS_UNUSED;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000488 fdcon[s].c_keytype = 0;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000489 TAILQ_REMOVE(&tq, &fdcon[s], c_link);
Ben Lindstromc1e04212001-03-05 07:04:38 +0000490 FD_CLR(s, read_wait);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000491 ncon--;
492}
493
Ben Lindstrombba81212001-06-25 05:01:22 +0000494static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000495contouch(int s)
496{
497 TAILQ_REMOVE(&tq, &fdcon[s], c_link);
498 gettimeofday(&fdcon[s].c_tv, NULL);
499 fdcon[s].c_tv.tv_sec += timeout;
500 TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link);
501}
502
Ben Lindstrombba81212001-06-25 05:01:22 +0000503static int
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000504conrecycle(int s)
505{
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000506 con *c = &fdcon[s];
Ben Lindstrom965710f2002-07-07 22:17:22 +0000507 int ret;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000508
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000509 ret = conalloc(c->c_namelist, c->c_output_name, c->c_keytype);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000510 confree(s);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000511 return (ret);
512}
513
Ben Lindstrombba81212001-06-25 05:01:22 +0000514static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000515congreet(int s)
516{
Damien Millereccb9de2005-06-17 12:59:34 +1000517 int n = 0, remote_major = 0, remote_minor = 0;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000518 char buf[256], *cp;
Damien Miller83c02ef2001-12-21 12:45:43 +1100519 char remote_version[sizeof buf];
Damien Millereccb9de2005-06-17 12:59:34 +1000520 size_t bufsiz;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000521 con *c = &fdcon[s];
522
Damien Miller4bbacb72005-11-05 15:12:28 +1100523 for (;;) {
524 memset(buf, '\0', sizeof(buf));
525 bufsiz = sizeof(buf);
526 cp = buf;
527 while (bufsiz-- &&
528 (n = atomicio(read, s, cp, 1)) == 1 && *cp != '\n') {
529 if (*cp == '\r')
530 *cp = '\n';
531 cp++;
532 }
533 if (n != 1 || strncmp(buf, "SSH-", 4) == 0)
534 break;
Ben Lindstromde8fc6f2001-08-06 22:43:50 +0000535 }
Ben Lindstrom6b28c352002-03-05 01:54:52 +0000536 if (n == 0) {
Damien Millerb253cc42005-05-26 12:23:44 +1000537 switch (errno) {
538 case EPIPE:
539 error("%s: Connection closed by remote host", c->c_name);
540 break;
541 case ECONNREFUSED:
542 break;
543 default:
544 error("read (%s): %s", c->c_name, strerror(errno));
545 break;
546 }
Ben Lindstrom6b28c352002-03-05 01:54:52 +0000547 conrecycle(s);
548 return;
549 }
Ben Lindstrom884a4ac2001-03-06 03:33:04 +0000550 if (*cp != '\n' && *cp != '\r') {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000551 error("%s: bad greeting", c->c_name);
552 confree(s);
553 return;
554 }
Ben Lindstrom884a4ac2001-03-06 03:33:04 +0000555 *cp = '\0';
Damien Miller83c02ef2001-12-21 12:45:43 +1100556 if (sscanf(buf, "SSH-%d.%d-%[^\n]\n",
557 &remote_major, &remote_minor, remote_version) == 3)
558 compat_datafellows(remote_version);
559 else
560 datafellows = 0;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000561 if (c->c_keytype != KT_RSA1) {
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000562 if (!ssh2_capable(remote_major, remote_minor)) {
563 debug("%s doesn't support ssh2", c->c_name);
564 confree(s);
565 return;
566 }
Damien Miller83c02ef2001-12-21 12:45:43 +1100567 } else if (remote_major != 1) {
568 debug("%s doesn't support ssh1", c->c_name);
569 confree(s);
570 return;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000571 }
Ben Lindstromde8fc6f2001-08-06 22:43:50 +0000572 fprintf(stderr, "# %s %s\n", c->c_name, chop(buf));
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000573 n = snprintf(buf, sizeof buf, "SSH-%d.%d-OpenSSH-keyscan\r\n",
574 c->c_keytype == KT_RSA1? PROTOCOL_MAJOR_1 : PROTOCOL_MAJOR_2,
575 c->c_keytype == KT_RSA1? PROTOCOL_MINOR_1 : PROTOCOL_MINOR_2);
Damien Millereccb9de2005-06-17 12:59:34 +1000576 if (n < 0 || (size_t)n >= sizeof(buf)) {
Damien Miller41bfc292005-05-26 12:07:32 +1000577 error("snprintf: buffer too small");
578 confree(s);
579 return;
580 }
Damien Millereccb9de2005-06-17 12:59:34 +1000581 if (atomicio(vwrite, s, buf, n) != (size_t)n) {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000582 error("write (%s): %s", c->c_name, strerror(errno));
583 confree(s);
584 return;
585 }
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000586 if (c->c_keytype != KT_RSA1) {
587 keyprint(c, keygrab_ssh2(c));
588 confree(s);
589 return;
590 }
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000591 c->c_status = CS_SIZE;
592 contouch(s);
593}
594
Ben Lindstrombba81212001-06-25 05:01:22 +0000595static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000596conread(int s)
597{
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000598 con *c = &fdcon[s];
Damien Millerb253cc42005-05-26 12:23:44 +1000599 size_t n;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000600
601 if (c->c_status == CS_CON) {
602 congreet(s);
603 return;
604 }
Darren Tuckerfe6649d2004-08-13 21:19:37 +1000605 n = atomicio(read, s, c->c_data + c->c_off, c->c_len - c->c_off);
Damien Millerb253cc42005-05-26 12:23:44 +1000606 if (n == 0) {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000607 error("read (%s): %s", c->c_name, strerror(errno));
608 confree(s);
609 return;
610 }
611 c->c_off += n;
612
613 if (c->c_off == c->c_len)
614 switch (c->c_status) {
615 case CS_SIZE:
616 c->c_plen = htonl(c->c_plen);
617 c->c_len = c->c_plen + 8 - (c->c_plen & 7);
618 c->c_off = 0;
619 c->c_data = xmalloc(c->c_len);
620 c->c_status = CS_KEYS;
621 break;
622 case CS_KEYS:
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000623 keyprint(c, keygrab_ssh1(c));
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000624 confree(s);
625 return;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000626 default:
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000627 fatal("conread: invalid status %d", c->c_status);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000628 break;
629 }
630
631 contouch(s);
632}
633
Ben Lindstrombba81212001-06-25 05:01:22 +0000634static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000635conloop(void)
636{
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000637 struct timeval seltime, now;
Ben Lindstrom965710f2002-07-07 22:17:22 +0000638 fd_set *r, *e;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000639 con *c;
Ben Lindstrom965710f2002-07-07 22:17:22 +0000640 int i;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000641
642 gettimeofday(&now, NULL);
Ben Lindstrom61c183b2002-06-21 00:09:54 +0000643 c = TAILQ_FIRST(&tq);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000644
Ben Lindstromd20b8552001-03-05 07:01:18 +0000645 if (c && (c->c_tv.tv_sec > now.tv_sec ||
646 (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec > now.tv_usec))) {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000647 seltime = c->c_tv;
648 seltime.tv_sec -= now.tv_sec;
649 seltime.tv_usec -= now.tv_usec;
Ben Lindstromc791beb2001-02-10 23:18:11 +0000650 if (seltime.tv_usec < 0) {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000651 seltime.tv_usec += 1000000;
652 seltime.tv_sec--;
653 }
654 } else
655 seltime.tv_sec = seltime.tv_usec = 0;
656
Damien Miller07d86be2006-03-26 14:19:21 +1100657 r = xcalloc(read_wait_nfdset, sizeof(fd_mask));
658 e = xcalloc(read_wait_nfdset, sizeof(fd_mask));
659 memcpy(r, read_wait, read_wait_nfdset * sizeof(fd_mask));
660 memcpy(e, read_wait, read_wait_nfdset * sizeof(fd_mask));
Ben Lindstromc1e04212001-03-05 07:04:38 +0000661
662 while (select(maxfd, r, NULL, e, &seltime) == -1 &&
Damien Millerd8968ad2008-07-04 23:10:49 +1000663 (errno == EAGAIN || errno == EINTR || errno == EWOULDBLOCK))
Ben Lindstromf9452512001-02-15 03:12:08 +0000664 ;
665
Ben Lindstromd20b8552001-03-05 07:01:18 +0000666 for (i = 0; i < maxfd; i++) {
Ben Lindstromc1e04212001-03-05 07:04:38 +0000667 if (FD_ISSET(i, e)) {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000668 error("%s: exception!", fdcon[i].c_name);
669 confree(i);
Ben Lindstromc1e04212001-03-05 07:04:38 +0000670 } else if (FD_ISSET(i, r))
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000671 conread(i);
Ben Lindstromd20b8552001-03-05 07:01:18 +0000672 }
Ben Lindstromc1e04212001-03-05 07:04:38 +0000673 xfree(r);
674 xfree(e);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000675
Ben Lindstrom61c183b2002-06-21 00:09:54 +0000676 c = TAILQ_FIRST(&tq);
Ben Lindstromd20b8552001-03-05 07:01:18 +0000677 while (c && (c->c_tv.tv_sec < now.tv_sec ||
678 (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec < now.tv_usec))) {
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000679 int s = c->c_fd;
Ben Lindstromd20b8552001-03-05 07:01:18 +0000680
Ben Lindstrom61c183b2002-06-21 00:09:54 +0000681 c = TAILQ_NEXT(c, c_link);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000682 conrecycle(s);
683 }
684}
685
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000686static void
687do_host(char *host)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000688{
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000689 char *name = strnnsep(&host, " \t\n");
690 int j;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000691
Ben Lindstromeaffb9d2001-12-06 16:28:19 +0000692 if (name == NULL)
693 return;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000694 for (j = KT_RSA1; j <= KT_RSA; j *= 2) {
695 if (get_keytypes & j) {
696 while (ncon >= MAXCON)
697 conloop();
698 conalloc(name, *host ? host : name, j);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000699 }
700 }
701}
702
Ben Lindstrom9c8edc92002-02-26 17:52:14 +0000703void
704fatal(const char *fmt,...)
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000705{
Ben Lindstrom9c8edc92002-02-26 17:52:14 +0000706 va_list args;
Ben Lindstrom965710f2002-07-07 22:17:22 +0000707
Ben Lindstrom9c8edc92002-02-26 17:52:14 +0000708 va_start(args, fmt);
709 do_log(SYSLOG_LEVEL_FATAL, fmt, args);
710 va_end(args);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000711 if (nonfatal_fatal)
712 longjmp(kexjmp, -1);
Ben Lindstrom9c8edc92002-02-26 17:52:14 +0000713 else
Darren Tucker3d326222003-09-22 21:11:20 +1000714 exit(255);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000715}
716
717static void
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000718usage(void)
719{
Damien Millerc1719f72008-11-03 19:27:07 +1100720 fprintf(stderr,
721 "usage: %s [-46Hv] [-f file] [-p port] [-T timeout] [-t type]\n"
Darren Tucker34e314d2010-01-08 17:03:46 +1100722 "\t\t [-V rdomain] [host | addrlist namelist] ...\n",
Ben Lindstromddfb1e32001-08-06 22:06:35 +0000723 __progname);
Ben Lindstromddfb1e32001-08-06 22:06:35 +0000724 exit(1);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000725}
726
727int
728main(int argc, char **argv)
729{
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000730 int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO;
731 int opt, fopt_count = 0;
732 char *tname;
Kevin Steves76e7d9b2001-09-20 20:30:09 +0000733
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000734 extern int optind;
735 extern char *optarg;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000736
Damien Miller59d3d5b2003-08-22 09:34:41 +1000737 __progname = ssh_get_progname(argv[0]);
Ben Lindstrom4e088e42001-10-10 20:45:43 +0000738 init_rng();
739 seed_rng();
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000740 TAILQ_INIT(&tq);
741
Darren Tuckerce321d82005-10-03 18:11:24 +1000742 /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
743 sanitise_stdfd();
744
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000745 if (argc <= 1)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000746 usage();
747
Darren Tucker34e314d2010-01-08 17:03:46 +1100748 while ((opt = getopt(argc, argv, "Hv46p:T:t:f:V:")) != -1) {
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000749 switch (opt) {
Damien Millerdb7b8172005-03-01 21:48:03 +1100750 case 'H':
751 hash_hosts = 1;
752 break;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000753 case 'p':
754 ssh_port = a2port(optarg);
Damien Miller3dc71ad2009-01-28 16:31:22 +1100755 if (ssh_port <= 0) {
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000756 fprintf(stderr, "Bad port '%s'\n", optarg);
757 exit(1);
758 }
759 break;
760 case 'T':
Ben Lindstromedd098b2002-07-04 00:07:13 +0000761 timeout = convtime(optarg);
762 if (timeout == -1 || timeout == 0) {
763 fprintf(stderr, "Bad timeout '%s'\n", optarg);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000764 usage();
Ben Lindstromedd098b2002-07-04 00:07:13 +0000765 }
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000766 break;
767 case 'v':
768 if (!debug_flag) {
769 debug_flag = 1;
770 log_level = SYSLOG_LEVEL_DEBUG1;
771 }
772 else if (log_level < SYSLOG_LEVEL_DEBUG3)
773 log_level++;
774 else
775 fatal("Too high debugging level.");
776 break;
Kevin Steves76e7d9b2001-09-20 20:30:09 +0000777 case 'f':
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000778 if (strcmp(optarg, "-") == 0)
779 optarg = NULL;
780 argv[fopt_count++] = optarg;
781 break;
782 case 't':
783 get_keytypes = 0;
784 tname = strtok(optarg, ",");
785 while (tname) {
786 int type = key_type_from_name(tname);
787 switch (type) {
788 case KEY_RSA1:
789 get_keytypes |= KT_RSA1;
790 break;
791 case KEY_DSA:
792 get_keytypes |= KT_DSA;
793 break;
794 case KEY_RSA:
795 get_keytypes |= KT_RSA;
796 break;
797 case KEY_UNSPEC:
Ben Lindstrom28c603b2001-12-06 16:45:10 +0000798 fatal("unknown key type %s", tname);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000799 }
800 tname = strtok(NULL, ",");
801 }
802 break;
803 case '4':
804 IPv4or6 = AF_INET;
805 break;
806 case '6':
807 IPv4or6 = AF_INET6;
808 break;
Darren Tucker34e314d2010-01-08 17:03:46 +1100809 case 'V':
Darren Tucker1f5e3dc2010-01-08 19:53:52 +1100810#ifdef USE_ROUTINGDOMAIN
Darren Tucker75456e82010-01-08 18:55:58 +1100811 scan_rdomain = a2rdomain(optarg);
812 if (scan_rdomain == -1) {
813 fprintf(stderr, "Bad rdomain '%s'\n", optarg);
814 exit(1);
815 }
Darren Tucker1f5e3dc2010-01-08 19:53:52 +1100816#else
817 fprintf(stderr, "RoutingDomain not supported on this "
818 "platform.\n");
819 exit(1);
820#endif
Darren Tucker34e314d2010-01-08 17:03:46 +1100821 break;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000822 case '?':
823 default:
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000824 usage();
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000825 }
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000826 }
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000827 if (optind == argc && !fopt_count)
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000828 usage();
829
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000830 log_init("ssh-keyscan", log_level, SYSLOG_FACILITY_USER, 1);
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000831
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000832 maxfd = fdlim_get(1);
833 if (maxfd < 0)
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000834 fatal("%s: fdlim_get: bad value", __progname);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000835 if (maxfd > MAXMAXFD)
836 maxfd = MAXMAXFD;
Ben Lindstromd20b8552001-03-05 07:01:18 +0000837 if (MAXCON <= 0)
Kevin Stevesfa72dda2000-12-15 18:39:12 +0000838 fatal("%s: not enough file descriptors", __progname);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000839 if (maxfd > fdlim_get(0))
840 fdlim_set(maxfd);
Damien Miller07d86be2006-03-26 14:19:21 +1100841 fdcon = xcalloc(maxfd, sizeof(con));
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000842
Damien Miller07d86be2006-03-26 14:19:21 +1100843 read_wait_nfdset = howmany(maxfd, NFDBITS);
844 read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask));
Ben Lindstromc1e04212001-03-05 07:04:38 +0000845
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000846 if (fopt_count) {
847 Linebuf *lb;
848 char *line;
849 int j;
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000850
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000851 for (j = 0; j < fopt_count; j++) {
852 lb = Linebuf_alloc(argv[j], error);
Ben Lindstrom78bbd9e2001-09-12 17:10:40 +0000853 if (!lb)
854 continue;
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000855 while ((line = Linebuf_getline(lb)) != NULL)
856 do_host(line);
857 Linebuf_free(lb);
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000858 }
Ben Lindstrom325e70c2001-08-06 22:41:30 +0000859 }
860
861 while (optind < argc)
862 do_host(argv[optind++]);
863
Ben Lindstromb6434ae2000-12-05 01:15:09 +0000864 while (ncon > 0)
865 conloop();
866
867 return (0);
868}