]> andersk Git - openssh.git/blame - ssh-keyscan.c
- (dtucker) [sftp.c] Expand ifdef for libedit to cover complete_is_remote
[openssh.git] / ssh-keyscan.c
CommitLineData
9e622dcd 1/* $OpenBSD: ssh-keyscan.c,v 1.80 2009/12/25 19:40:21 stevesk Exp $ */
f6fdbddf 2/*
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
aa144206 7 * OpenBSD project by leaving this copyright notice intact.
f6fdbddf 8 */
9
10#include "includes.h"
876faccd 11
9fd2a215 12#include "openbsd-compat/sys-queue.h"
876faccd 13#include <sys/resource.h>
e264ac72 14#ifdef HAVE_SYS_TIME_H
15# include <sys/time.h>
16#endif
00146caa 17
5e837c7b 18#include <netinet/in.h>
19#include <arpa/inet.h>
20
00146caa 21#include <openssl/bn.h>
22
28cb0a43 23#include <netdb.h>
403a29ac 24#include <errno.h>
c8dfff33 25#include <setjmp.h>
ffa517a8 26#include <stdarg.h>
cf851879 27#include <stdio.h>
ffa517a8 28#include <stdlib.h>
31652869 29#include <signal.h>
00146caa 30#include <string.h>
5188ba17 31#include <unistd.h>
f6fdbddf 32
f6fdbddf 33#include "xmalloc.h"
34#include "ssh.h"
42f11eb2 35#include "ssh1.h"
31652869 36#include "buffer.h"
f6fdbddf 37#include "key.h"
31652869 38#include "cipher.h"
5061072f 39#include "kex.h"
40#include "compat.h"
41#include "myproposal.h"
42#include "packet.h"
43#include "dispatch.h"
42f11eb2 44#include "log.h"
2671b47f 45#include "atomicio.h"
5061072f 46#include "misc.h"
90a8ae9f 47#include "hostfile.h"
f6fdbddf 48
5061072f 49/* 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. */
5061072f 51int IPv4or6 = AF_UNSPEC;
f6fdbddf 52
5061072f 53int ssh_port = SSH_DEFAULT_PORT;
a204806b 54
55#define KT_RSA1 1
56#define KT_DSA 2
57#define KT_RSA 4
58
c47ff7a6 59int get_keytypes = KT_RSA; /* Get only RSA keys by default */
f6fdbddf 60
90a8ae9f 61int hash_hosts = 0; /* Hash hostname on output */
62
f6fdbddf 63#define MAXMAXFD 256
64
65/* The number of seconds after which to give up on a TCP connection */
66int timeout = 5;
67
68int maxfd;
2671b47f 69#define MAXCON (maxfd - 10)
f6fdbddf 70
fe7dba42 71/* The default routing domain */
72int scan_rdomain = -1;
73
1fe6a48f 74extern char *__progname;
ff7fee59 75fd_set *read_wait;
52e3daed 76size_t read_wait_nfdset;
f6fdbddf 77int ncon;
5061072f 78int nonfatal_fatal = 0;
79jmp_buf kexjmp;
4278ff63 80Key *kexjmp_key;
f6fdbddf 81
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 {
1e3b8b07 87 u_char c_status; /* State of connection on this file desc. */
f6fdbddf 88#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. */
5061072f 96 int c_keytype; /* Only one of KT_RSA1, KT_DSA, or KT_RSA */
f6fdbddf 97 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 */
5061072f 102 Kex *c_kex; /* The key-exchange struct for ssh2 */
f6fdbddf 103 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;
1e3b8b07 119 u_int size;
f6fdbddf 120 int lineno;
121 const char *filename;
122 FILE *stream;
123 void (*errfun) (const char *,...);
124} Linebuf;
125
396c147e 126static Linebuf *
f6fdbddf 127Linebuf_alloc(const char *filename, void (*errfun) (const char *,...))
128{
129 Linebuf *lb;
130
131 if (!(lb = malloc(sizeof(*lb)))) {
132 if (errfun)
f11fe301 133 (*errfun) ("linebuf (%s): malloc failed\n",
134 filename ? filename : "(stdin)");
f6fdbddf 135 return (NULL);
136 }
137 if (filename) {
138 lb->filename = filename;
139 if (!(lb->stream = fopen(filename, "r"))) {
b5c334cc 140 xfree(lb);
f6fdbddf 141 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
6e56dc92 150 if (!(lb->buf = malloc((lb->size = LINEBUF_SIZE)))) {
f6fdbddf 151 if (errfun)
152 (*errfun) ("linebuf (%s): malloc failed\n", lb->filename);
b5c334cc 153 xfree(lb);
f6fdbddf 154 return (NULL);
155 }
156 lb->errfun = errfun;
157 lb->lineno = 0;
158 return (lb);
159}
160
396c147e 161static void
f6fdbddf 162Linebuf_free(Linebuf * lb)
163{
164 fclose(lb->stream);
b5c334cc 165 xfree(lb->buf);
166 xfree(lb);
f6fdbddf 167}
168
396c147e 169#if 0
170static void
f6fdbddf 171Linebuf_restart(Linebuf * lb)
172{
173 clearerr(lb->stream);
174 rewind(lb->stream);
175 lb->lineno = 0;
176}
177
396c147e 178static int
f6fdbddf 179Linebuf_lineno(Linebuf * lb)
180{
181 return (lb->lineno);
182}
396c147e 183#endif
f6fdbddf 184
396c147e 185static char *
5be2ec5e 186Linebuf_getline(Linebuf * lb)
f6fdbddf 187{
2ceb8101 188 size_t n = 0;
50d2fbbc 189 void *p;
f6fdbddf 190
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)
50d2fbbc 196 (*lb->errfun)("%s: %s\n", lb->filename,
778f6940 197 strerror(errno));
f6fdbddf 198 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)
50d2fbbc 209 (*lb->errfun)("%s: skipping incomplete last line\n",
778f6940 210 lb->filename);
f6fdbddf 211 return (NULL);
212 }
213 /* Double the buffer if we need more space */
50d2fbbc 214 lb->size *= 2;
215 if ((p = realloc(lb->buf, lb->size)) == NULL) {
216 lb->size /= 2;
f6fdbddf 217 if (lb->errfun)
50d2fbbc 218 (*lb->errfun)("linebuf (%s): realloc failed\n",
778f6940 219 lb->filename);
f6fdbddf 220 return (NULL);
221 }
50d2fbbc 222 lb->buf = p;
f6fdbddf 223 }
224}
225
396c147e 226static int
f6fdbddf 227fdlim_get(int hard)
228{
f318b98b 229#if defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE)
f6fdbddf 230 struct rlimit rlfd;
778f6940 231
f6fdbddf 232 if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0)
233 return (-1);
234 if ((hard ? rlfd.rlim_max : rlfd.rlim_cur) == RLIM_INFINITY)
341c3efe 235 return SSH_SYSFDMAX;
f6fdbddf 236 else
237 return hard ? rlfd.rlim_max : rlfd.rlim_cur;
b03bd394 238#else
341c3efe 239 return SSH_SYSFDMAX;
b03bd394 240#endif
f6fdbddf 241}
242
396c147e 243static int
f6fdbddf 244fdlim_set(int lim)
245{
f318b98b 246#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
f6fdbddf 247 struct rlimit rlfd;
b03bd394 248#endif
41545cb6 249
f6fdbddf 250 if (lim <= 0)
251 return (-1);
f318b98b 252#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
f6fdbddf 253 if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0)
254 return (-1);
255 rlfd.rlim_cur = lim;
256 if (setrlimit(RLIMIT_NOFILE, &rlfd) < 0)
257 return (-1);
b03bd394 258#elif defined (HAVE_SETDTABLESIZE)
b705bea0 259 setdtablesize(lim);
b03bd394 260#endif
f6fdbddf 261 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 */
396c147e 269static char *
f6fdbddf 270xstrsep(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 */
396c147e 291static char *
f6fdbddf 292strnnsep(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
5061072f 302static Key *
303keygrab_ssh1(con *c)
f6fdbddf 304{
305 static Key *rsa;
306 static Buffer msg;
307
308 if (rsa == NULL) {
309 buffer_init(&msg);
310 rsa = key_new(KEY_RSA1);
311 }
5061072f 312 buffer_append(&msg, c->c_data, c->c_plen);
313 buffer_consume(&msg, 8 - (c->c_plen & 7)); /* padding */
f6fdbddf 314 if (buffer_get_char(&msg) != (int) SSH_SMSG_PUBLIC_KEY) {
5061072f 315 error("%s: invalid packet type", c->c_name);
f6fdbddf 316 buffer_clear(&msg);
5061072f 317 return NULL;
f6fdbddf 318 }
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);
5061072f 330
f6fdbddf 331 buffer_clear(&msg);
332
5061072f 333 return (rsa);
334}
335
336static int
337hostjump(Key *hostkey)
338{
4278ff63 339 kexjmp_key = hostkey;
340 longjmp(kexjmp, 1);
5061072f 341}
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);
98a58eda 369 c->c_kex->kex[KEX_DH_GRP1_SHA1] = kexdh_client;
41e5bd9a 370 c->c_kex->kex[KEX_DH_GRP14_SHA1] = kexdh_client;
98a58eda 371 c->c_kex->kex[KEX_DH_GEX_SHA1] = kexgex_client;
2ff8003a 372 c->c_kex->kex[KEX_DH_GEX_SHA256] = kexgex_client;
5061072f 373 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();
5061072f 385
4278ff63 386 return j < 0? NULL : kexjmp_key;
5061072f 387}
388
389static void
390keyprint(con *c, Key *key)
391{
90a8ae9f 392 char *host = c->c_output_name ? c->c_output_name : c->c_name;
393
5061072f 394 if (!key)
395 return;
90a8ae9f 396 if (hash_hosts && (host = host_hash(host, NULL, 0)) == NULL)
397 fatal("host_hash failed");
5061072f 398
90a8ae9f 399 fprintf(stdout, "%s ", host);
5061072f 400 key_write(key, stdout);
f6fdbddf 401 fputs("\n", stdout);
402}
403
396c147e 404static int
f6fdbddf 405tcpconnect(char *host)
406{
407 struct addrinfo hints, *ai, *aitop;
408 char strport[NI_MAXSERV];
409 int gaierr, s = -1;
410
5061072f 411 snprintf(strport, sizeof strport, "%d", ssh_port);
f6fdbddf 412 memset(&hints, 0, sizeof(hints));
5061072f 413 hints.ai_family = IPv4or6;
f6fdbddf 414 hints.ai_socktype = SOCK_STREAM;
415 if ((gaierr = getaddrinfo(host, strport, &hints, &aitop)) != 0)
bb4626fe 416 fatal("getaddrinfo %s: %s", host, ssh_gai_strerror(gaierr));
f6fdbddf 417 for (ai = aitop; ai; ai = ai->ai_next) {
fe7dba42 418 s = socket_rdomain(ai->ai_family, ai->ai_socktype,
419 ai->ai_protocol, scan_rdomain);
f6fdbddf 420 if (s < 0) {
421 error("socket: %s", strerror(errno));
422 continue;
423 }
170694d7 424 if (set_nonblock(s) == -1)
425 fatal("%s: set_nonblock(%d)", __func__, s);
f6fdbddf 426 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
396c147e 438static int
5061072f 439conalloc(char *iname, char *oname, int keytype)
f6fdbddf 440{
f6fdbddf 441 char *namebase, *name, *namelist;
50d2fbbc 442 int s;
f6fdbddf 443
444 namebase = namelist = xstrdup(iname);
445
446 do {
447 name = xstrsep(&namelist, ",");
448 if (!name) {
b5c334cc 449 xfree(namebase);
f6fdbddf 450 return (-1);
451 }
452 } while ((s = tcpconnect(name)) < 0);
453
454 if (s >= maxfd)
227e8e86 455 fatal("conalloc: fdno %d too high", s);
f6fdbddf 456 if (fdcon[s].c_status)
227e8e86 457 fatal("conalloc: attempt to reuse fdno %d", s);
f6fdbddf 458
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;
5061072f 468 fdcon[s].c_keytype = keytype;
f6fdbddf 469 gettimeofday(&fdcon[s].c_tv, NULL);
470 fdcon[s].c_tv.tv_sec += timeout;
471 TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link);
ff7fee59 472 FD_SET(s, read_wait);
f6fdbddf 473 ncon++;
474 return (s);
475}
476
396c147e 477static void
f6fdbddf 478confree(int s)
479{
f6fdbddf 480 if (s >= maxfd || fdcon[s].c_status == CS_UNUSED)
227e8e86 481 fatal("confree: attempt to free bad fdno %d", s);
2671b47f 482 close(s);
b5c334cc 483 xfree(fdcon[s].c_namebase);
484 xfree(fdcon[s].c_output_name);
f6fdbddf 485 if (fdcon[s].c_status == CS_KEYS)
b5c334cc 486 xfree(fdcon[s].c_data);
f6fdbddf 487 fdcon[s].c_status = CS_UNUSED;
5061072f 488 fdcon[s].c_keytype = 0;
f6fdbddf 489 TAILQ_REMOVE(&tq, &fdcon[s], c_link);
ff7fee59 490 FD_CLR(s, read_wait);
f6fdbddf 491 ncon--;
492}
493
396c147e 494static void
f6fdbddf 495contouch(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
396c147e 503static int
f6fdbddf 504conrecycle(int s)
505{
f6fdbddf 506 con *c = &fdcon[s];
50d2fbbc 507 int ret;
f6fdbddf 508
5061072f 509 ret = conalloc(c->c_namelist, c->c_output_name, c->c_keytype);
f6fdbddf 510 confree(s);
f6fdbddf 511 return (ret);
512}
513
396c147e 514static void
f6fdbddf 515congreet(int s)
516{
2ceb8101 517 int n = 0, remote_major = 0, remote_minor = 0;
5061072f 518 char buf[256], *cp;
cdc95d6e 519 char remote_version[sizeof buf];
2ceb8101 520 size_t bufsiz;
f6fdbddf 521 con *c = &fdcon[s];
522
2995db03 523 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;
578954b1 535 }
27452401 536 if (n == 0) {
05624c18 537 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 }
27452401 547 conrecycle(s);
548 return;
549 }
c16c7f20 550 if (*cp != '\n' && *cp != '\r') {
f6fdbddf 551 error("%s: bad greeting", c->c_name);
552 confree(s);
553 return;
554 }
c16c7f20 555 *cp = '\0';
cdc95d6e 556 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;
5061072f 561 if (c->c_keytype != KT_RSA1) {
5061072f 562 if (!ssh2_capable(remote_major, remote_minor)) {
563 debug("%s doesn't support ssh2", c->c_name);
564 confree(s);
565 return;
566 }
cdc95d6e 567 } else if (remote_major != 1) {
568 debug("%s doesn't support ssh1", c->c_name);
569 confree(s);
570 return;
5061072f 571 }
578954b1 572 fprintf(stderr, "# %s %s\n", c->c_name, chop(buf));
5061072f 573 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);
2ceb8101 576 if (n < 0 || (size_t)n >= sizeof(buf)) {
4b826d75 577 error("snprintf: buffer too small");
578 confree(s);
579 return;
580 }
2ceb8101 581 if (atomicio(vwrite, s, buf, n) != (size_t)n) {
f6fdbddf 582 error("write (%s): %s", c->c_name, strerror(errno));
583 confree(s);
584 return;
585 }
5061072f 586 if (c->c_keytype != KT_RSA1) {
587 keyprint(c, keygrab_ssh2(c));
588 confree(s);
589 return;
590 }
f6fdbddf 591 c->c_status = CS_SIZE;
592 contouch(s);
593}
594
396c147e 595static void
f6fdbddf 596conread(int s)
597{
f6fdbddf 598 con *c = &fdcon[s];
05624c18 599 size_t n;
f6fdbddf 600
601 if (c->c_status == CS_CON) {
602 congreet(s);
603 return;
604 }
f581b6e8 605 n = atomicio(read, s, c->c_data + c->c_off, c->c_len - c->c_off);
05624c18 606 if (n == 0) {
f6fdbddf 607 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:
5061072f 623 keyprint(c, keygrab_ssh1(c));
f6fdbddf 624 confree(s);
625 return;
f6fdbddf 626 default:
227e8e86 627 fatal("conread: invalid status %d", c->c_status);
f6fdbddf 628 break;
629 }
630
631 contouch(s);
632}
633
396c147e 634static void
f6fdbddf 635conloop(void)
636{
f6fdbddf 637 struct timeval seltime, now;
50d2fbbc 638 fd_set *r, *e;
f6fdbddf 639 con *c;
50d2fbbc 640 int i;
f6fdbddf 641
642 gettimeofday(&now, NULL);
48ba8dc8 643 c = TAILQ_FIRST(&tq);
f6fdbddf 644
2671b47f 645 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))) {
f6fdbddf 647 seltime = c->c_tv;
648 seltime.tv_sec -= now.tv_sec;
649 seltime.tv_usec -= now.tv_usec;
5be2ec5e 650 if (seltime.tv_usec < 0) {
f6fdbddf 651 seltime.tv_usec += 1000000;
652 seltime.tv_sec--;
653 }
654 } else
655 seltime.tv_sec = seltime.tv_usec = 0;
656
52e3daed 657 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));
ff7fee59 661
662 while (select(maxfd, r, NULL, e, &seltime) == -1 &&
5a0c8771 663 (errno == EAGAIN || errno == EINTR || errno == EWOULDBLOCK))
fd193ca4 664 ;
665
2671b47f 666 for (i = 0; i < maxfd; i++) {
ff7fee59 667 if (FD_ISSET(i, e)) {
f6fdbddf 668 error("%s: exception!", fdcon[i].c_name);
669 confree(i);
ff7fee59 670 } else if (FD_ISSET(i, r))
f6fdbddf 671 conread(i);
2671b47f 672 }
ff7fee59 673 xfree(r);
674 xfree(e);
f6fdbddf 675
48ba8dc8 676 c = TAILQ_FIRST(&tq);
2671b47f 677 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))) {
f6fdbddf 679 int s = c->c_fd;
2671b47f 680
48ba8dc8 681 c = TAILQ_NEXT(c, c_link);
f6fdbddf 682 conrecycle(s);
683 }
684}
685
5061072f 686static void
687do_host(char *host)
f6fdbddf 688{
5061072f 689 char *name = strnnsep(&host, " \t\n");
690 int j;
691
0408c978 692 if (name == NULL)
693 return;
5061072f 694 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);
f6fdbddf 699 }
700 }
701}
702
2bae80e9 703void
704fatal(const char *fmt,...)
5061072f 705{
2bae80e9 706 va_list args;
50d2fbbc 707
2bae80e9 708 va_start(args, fmt);
709 do_log(SYSLOG_LEVEL_FATAL, fmt, args);
710 va_end(args);
5061072f 711 if (nonfatal_fatal)
712 longjmp(kexjmp, -1);
2bae80e9 713 else
eec6d341 714 exit(255);
5061072f 715}
716
396c147e 717static void
f6fdbddf 718usage(void)
719{
9995aaa3 720 fprintf(stderr,
721 "usage: %s [-46Hv] [-f file] [-p port] [-T timeout] [-t type]\n"
fe7dba42 722 "\t\t [-V rdomain] [host | addrlist namelist] ...\n",
33e766d2 723 __progname);
33e766d2 724 exit(1);
f6fdbddf 725}
726
727int
728main(int argc, char **argv)
729{
5061072f 730 int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO;
731 int opt, fopt_count = 0;
732 char *tname;
a204806b 733
5061072f 734 extern int optind;
735 extern char *optarg;
f6fdbddf 736
fda04d7d 737 __progname = ssh_get_progname(argv[0]);
8782141f 738 init_rng();
739 seed_rng();
f6fdbddf 740 TAILQ_INIT(&tq);
741
fd6168c1 742 /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
743 sanitise_stdfd();
744
5061072f 745 if (argc <= 1)
f6fdbddf 746 usage();
747
fe7dba42 748 while ((opt = getopt(argc, argv, "Hv46p:T:t:f:V:")) != -1) {
5061072f 749 switch (opt) {
90a8ae9f 750 case 'H':
751 hash_hosts = 1;
752 break;
5061072f 753 case 'p':
754 ssh_port = a2port(optarg);
5134115d 755 if (ssh_port <= 0) {
5061072f 756 fprintf(stderr, "Bad port '%s'\n", optarg);
757 exit(1);
758 }
759 break;
760 case 'T':
c8f94200 761 timeout = convtime(optarg);
762 if (timeout == -1 || timeout == 0) {
763 fprintf(stderr, "Bad timeout '%s'\n", optarg);
f6fdbddf 764 usage();
c8f94200 765 }
5061072f 766 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;
a204806b 777 case 'f':
5061072f 778 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:
f48e63c8 798 fatal("unknown key type %s", tname);
5061072f 799 }
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;
fe7dba42 809 case 'V':
85fee9c0 810#ifdef USE_ROUTINGDOMAIN
9e622dcd 811 scan_rdomain = a2rdomain(optarg);
812 if (scan_rdomain == -1) {
813 fprintf(stderr, "Bad rdomain '%s'\n", optarg);
814 exit(1);
815 }
85fee9c0 816#else
817 fprintf(stderr, "RoutingDomain not supported on this "
818 "platform.\n");
819 exit(1);
820#endif
fe7dba42 821 break;
5061072f 822 case '?':
823 default:
f6fdbddf 824 usage();
5061072f 825 }
f6fdbddf 826 }
5061072f 827 if (optind == argc && !fopt_count)
f6fdbddf 828 usage();
829
5061072f 830 log_init("ssh-keyscan", log_level, SYSLOG_FACILITY_USER, 1);
5061072f 831
f6fdbddf 832 maxfd = fdlim_get(1);
833 if (maxfd < 0)
227e8e86 834 fatal("%s: fdlim_get: bad value", __progname);
f6fdbddf 835 if (maxfd > MAXMAXFD)
836 maxfd = MAXMAXFD;
2671b47f 837 if (MAXCON <= 0)
227e8e86 838 fatal("%s: not enough file descriptors", __progname);
f6fdbddf 839 if (maxfd > fdlim_get(0))
840 fdlim_set(maxfd);
52e3daed 841 fdcon = xcalloc(maxfd, sizeof(con));
f6fdbddf 842
52e3daed 843 read_wait_nfdset = howmany(maxfd, NFDBITS);
844 read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask));
ff7fee59 845
5061072f 846 if (fopt_count) {
847 Linebuf *lb;
848 char *line;
849 int j;
850
851 for (j = 0; j < fopt_count; j++) {
852 lb = Linebuf_alloc(argv[j], error);
b7f79e7a 853 if (!lb)
854 continue;
5061072f 855 while ((line = Linebuf_getline(lb)) != NULL)
856 do_host(line);
857 Linebuf_free(lb);
f6fdbddf 858 }
5061072f 859 }
860
861 while (optind < argc)
862 do_host(argv[optind++]);
863
f6fdbddf 864 while (ncon > 0)
865 conloop();
866
867 return (0);
868}
This page took 0.56875 seconds and 5 git commands to generate.