ssh-keyscan.c revision 255767
1255767Sdes/* $OpenBSD: ssh-keyscan.c,v 1.87 2013/05/17 00:13:14 djm Exp $ */ 276259Sgreen/* 376259Sgreen * Copyright 1995, 1996 by David Mazieres <dm@lcs.mit.edu>. 476259Sgreen * 576259Sgreen * Modification and redistribution in source and binary forms is 676259Sgreen * permitted provided that due credit is given to the author and the 792555Sdes * OpenBSD project by leaving this copyright notice intact. 876259Sgreen */ 976259Sgreen 1076259Sgreen#include "includes.h" 11162856Sdes 12106130Sdes#include "openbsd-compat/sys-queue.h" 13162856Sdes#include <sys/resource.h> 14162856Sdes#ifdef HAVE_SYS_TIME_H 15162856Sdes# include <sys/time.h> 16162856Sdes#endif 1776259Sgreen 18162856Sdes#include <netinet/in.h> 19162856Sdes#include <arpa/inet.h> 20162856Sdes 2176259Sgreen#include <openssl/bn.h> 2276259Sgreen 23162856Sdes#include <netdb.h> 24162856Sdes#include <errno.h> 2592555Sdes#include <setjmp.h> 26162856Sdes#include <stdarg.h> 27162856Sdes#include <stdio.h> 28162856Sdes#include <stdlib.h> 29162856Sdes#include <signal.h> 30162856Sdes#include <string.h> 31162856Sdes#include <unistd.h> 32162856Sdes 3376259Sgreen#include "xmalloc.h" 3476259Sgreen#include "ssh.h" 3576259Sgreen#include "ssh1.h" 36162856Sdes#include "buffer.h" 3776259Sgreen#include "key.h" 38162856Sdes#include "cipher.h" 3992555Sdes#include "kex.h" 4092555Sdes#include "compat.h" 4192555Sdes#include "myproposal.h" 4292555Sdes#include "packet.h" 4392555Sdes#include "dispatch.h" 4476259Sgreen#include "log.h" 4576259Sgreen#include "atomicio.h" 4692555Sdes#include "misc.h" 47147005Sdes#include "hostfile.h" 4876259Sgreen 4998941Sdes/* Flag indicating whether IPv4 or IPv6. This can be set on the command line. 5098941Sdes Default value is AF_UNSPEC means both IPv4 and IPv6. */ 5198941Sdesint IPv4or6 = AF_UNSPEC; 5276259Sgreen 5392555Sdesint ssh_port = SSH_DEFAULT_PORT; 5476259Sgreen 55221420Sdes#define KT_RSA1 1 56221420Sdes#define KT_DSA 2 57221420Sdes#define KT_RSA 4 58221420Sdes#define KT_ECDSA 8 5992555Sdes 60237568Sdelphijint get_keytypes = KT_RSA|KT_ECDSA;/* Get RSA and ECDSA keys by default */ 6192555Sdes 62147005Sdesint hash_hosts = 0; /* Hash hostname on output */ 63147005Sdes 6476259Sgreen#define MAXMAXFD 256 6576259Sgreen 6676259Sgreen/* The number of seconds after which to give up on a TCP connection */ 6776259Sgreenint timeout = 5; 6876259Sgreen 6976259Sgreenint maxfd; 7076259Sgreen#define MAXCON (maxfd - 10) 7176259Sgreen 7276259Sgreenextern char *__progname; 7376259Sgreenfd_set *read_wait; 74162856Sdessize_t read_wait_nfdset; 7576259Sgreenint ncon; 7692555Sdesint nonfatal_fatal = 0; 7792555Sdesjmp_buf kexjmp; 7892555SdesKey *kexjmp_key; 7976259Sgreen 8076259Sgreen/* 8176259Sgreen * Keep a connection structure for each file descriptor. The state 8276259Sgreen * associated with file descriptor n is held in fdcon[n]. 8376259Sgreen */ 8476259Sgreentypedef struct Connection { 8576259Sgreen u_char c_status; /* State of connection on this file desc. */ 8676259Sgreen#define CS_UNUSED 0 /* File descriptor unused */ 8776259Sgreen#define CS_CON 1 /* Waiting to connect/read greeting */ 8876259Sgreen#define CS_SIZE 2 /* Waiting to read initial packet size */ 8976259Sgreen#define CS_KEYS 3 /* Waiting to read public key packet */ 9076259Sgreen int c_fd; /* Quick lookup: c->c_fd == c - fdcon */ 9176259Sgreen int c_plen; /* Packet length field for ssh packet */ 9276259Sgreen int c_len; /* Total bytes which must be read. */ 9376259Sgreen int c_off; /* Length of data read so far. */ 9492555Sdes int c_keytype; /* Only one of KT_RSA1, KT_DSA, or KT_RSA */ 9576259Sgreen char *c_namebase; /* Address to free for c_name and c_namelist */ 9676259Sgreen char *c_name; /* Hostname of connection for errors */ 9776259Sgreen char *c_namelist; /* Pointer to other possible addresses */ 9876259Sgreen char *c_output_name; /* Hostname of connection for output */ 9976259Sgreen char *c_data; /* Data read from this fd */ 10092555Sdes Kex *c_kex; /* The key-exchange struct for ssh2 */ 10176259Sgreen struct timeval c_tv; /* Time at which connection gets aborted */ 10276259Sgreen TAILQ_ENTRY(Connection) c_link; /* List of connections in timeout order. */ 10376259Sgreen} con; 10476259Sgreen 10576259SgreenTAILQ_HEAD(conlist, Connection) tq; /* Timeout Queue */ 10676259Sgreencon *fdcon; 10776259Sgreen 10892555Sdesstatic int 10976259Sgreenfdlim_get(int hard) 11076259Sgreen{ 11198941Sdes#if defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE) 11276259Sgreen struct rlimit rlfd; 11376259Sgreen 11476259Sgreen if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0) 11576259Sgreen return (-1); 11676259Sgreen if ((hard ? rlfd.rlim_max : rlfd.rlim_cur) == RLIM_INFINITY) 117126277Sdes return SSH_SYSFDMAX; 11876259Sgreen else 11976259Sgreen return hard ? rlfd.rlim_max : rlfd.rlim_cur; 12098941Sdes#else 121126277Sdes return SSH_SYSFDMAX; 12298941Sdes#endif 12376259Sgreen} 12476259Sgreen 12592555Sdesstatic int 12676259Sgreenfdlim_set(int lim) 12776259Sgreen{ 12898941Sdes#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE) 12976259Sgreen struct rlimit rlfd; 13098941Sdes#endif 131106130Sdes 13276259Sgreen if (lim <= 0) 13376259Sgreen return (-1); 13498941Sdes#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE) 13576259Sgreen if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0) 13676259Sgreen return (-1); 13776259Sgreen rlfd.rlim_cur = lim; 13876259Sgreen if (setrlimit(RLIMIT_NOFILE, &rlfd) < 0) 13976259Sgreen return (-1); 14098941Sdes#elif defined (HAVE_SETDTABLESIZE) 14198941Sdes setdtablesize(lim); 14298941Sdes#endif 14376259Sgreen return (0); 14476259Sgreen} 14576259Sgreen 14676259Sgreen/* 14776259Sgreen * This is an strsep function that returns a null field for adjacent 14876259Sgreen * separators. This is the same as the 4.4BSD strsep, but different from the 14976259Sgreen * one in the GNU libc. 15076259Sgreen */ 15192555Sdesstatic char * 15276259Sgreenxstrsep(char **str, const char *delim) 15376259Sgreen{ 15476259Sgreen char *s, *e; 15576259Sgreen 15676259Sgreen if (!**str) 15776259Sgreen return (NULL); 15876259Sgreen 15976259Sgreen s = *str; 16076259Sgreen e = s + strcspn(s, delim); 16176259Sgreen 16276259Sgreen if (*e != '\0') 16376259Sgreen *e++ = '\0'; 16476259Sgreen *str = e; 16576259Sgreen 16676259Sgreen return (s); 16776259Sgreen} 16876259Sgreen 16976259Sgreen/* 17076259Sgreen * Get the next non-null token (like GNU strsep). Strsep() will return a 17176259Sgreen * null token for two adjacent separators, so we may have to loop. 17276259Sgreen */ 17392555Sdesstatic char * 17476259Sgreenstrnnsep(char **stringp, char *delim) 17576259Sgreen{ 17676259Sgreen char *tok; 17776259Sgreen 17876259Sgreen do { 17976259Sgreen tok = xstrsep(stringp, delim); 18076259Sgreen } while (tok && *tok == '\0'); 18176259Sgreen return (tok); 18276259Sgreen} 18376259Sgreen 18492555Sdesstatic Key * 18592555Sdeskeygrab_ssh1(con *c) 18676259Sgreen{ 18776259Sgreen static Key *rsa; 18876259Sgreen static Buffer msg; 18976259Sgreen 19076259Sgreen if (rsa == NULL) { 19176259Sgreen buffer_init(&msg); 19276259Sgreen rsa = key_new(KEY_RSA1); 19376259Sgreen } 19492555Sdes buffer_append(&msg, c->c_data, c->c_plen); 19592555Sdes buffer_consume(&msg, 8 - (c->c_plen & 7)); /* padding */ 19676259Sgreen if (buffer_get_char(&msg) != (int) SSH_SMSG_PUBLIC_KEY) { 19792555Sdes error("%s: invalid packet type", c->c_name); 19876259Sgreen buffer_clear(&msg); 19992555Sdes return NULL; 20076259Sgreen } 20176259Sgreen buffer_consume(&msg, 8); /* cookie */ 20276259Sgreen 20376259Sgreen /* server key */ 20476259Sgreen (void) buffer_get_int(&msg); 20576259Sgreen buffer_get_bignum(&msg, rsa->rsa->e); 20676259Sgreen buffer_get_bignum(&msg, rsa->rsa->n); 20776259Sgreen 20876259Sgreen /* host key */ 20976259Sgreen (void) buffer_get_int(&msg); 21076259Sgreen buffer_get_bignum(&msg, rsa->rsa->e); 21176259Sgreen buffer_get_bignum(&msg, rsa->rsa->n); 21292555Sdes 21376259Sgreen buffer_clear(&msg); 21476259Sgreen 21592555Sdes return (rsa); 21692555Sdes} 21792555Sdes 21892555Sdesstatic int 21992555Sdeshostjump(Key *hostkey) 22092555Sdes{ 22192555Sdes kexjmp_key = hostkey; 22292555Sdes longjmp(kexjmp, 1); 22392555Sdes} 22492555Sdes 22592555Sdesstatic int 22692555Sdesssh2_capable(int remote_major, int remote_minor) 22792555Sdes{ 22892555Sdes switch (remote_major) { 22992555Sdes case 1: 23092555Sdes if (remote_minor == 99) 23192555Sdes return 1; 23292555Sdes break; 23392555Sdes case 2: 23492555Sdes return 1; 23592555Sdes default: 23692555Sdes break; 23792555Sdes } 23892555Sdes return 0; 23992555Sdes} 24092555Sdes 24192555Sdesstatic Key * 24292555Sdeskeygrab_ssh2(con *c) 24392555Sdes{ 24492555Sdes int j; 24592555Sdes 24692555Sdes packet_set_connection(c->c_fd, c->c_fd); 24792555Sdes enable_compat20(); 24892555Sdes myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = c->c_keytype == KT_DSA? 249221420Sdes "ssh-dss" : (c->c_keytype == KT_RSA ? "ssh-rsa" : 250221420Sdes "ecdsa-sha2-nistp256,ecdsa-sha2-nistp384,ecdsa-sha2-nistp521"); 25192555Sdes c->c_kex = kex_setup(myproposal); 252113911Sdes c->c_kex->kex[KEX_DH_GRP1_SHA1] = kexdh_client; 253137019Sdes c->c_kex->kex[KEX_DH_GRP14_SHA1] = kexdh_client; 254113911Sdes c->c_kex->kex[KEX_DH_GEX_SHA1] = kexgex_client; 255162856Sdes c->c_kex->kex[KEX_DH_GEX_SHA256] = kexgex_client; 256221420Sdes c->c_kex->kex[KEX_ECDH_SHA2] = kexecdh_client; 25792555Sdes c->c_kex->verify_host_key = hostjump; 25892555Sdes 25992555Sdes if (!(j = setjmp(kexjmp))) { 26092555Sdes nonfatal_fatal = 1; 26192555Sdes dispatch_run(DISPATCH_BLOCK, &c->c_kex->done, c->c_kex); 26292555Sdes fprintf(stderr, "Impossible! dispatch_run() returned!\n"); 26392555Sdes exit(1); 26492555Sdes } 26592555Sdes nonfatal_fatal = 0; 266255767Sdes free(c->c_kex); 26792555Sdes c->c_kex = NULL; 26892555Sdes packet_close(); 26992555Sdes 27092555Sdes return j < 0? NULL : kexjmp_key; 27192555Sdes} 27292555Sdes 27392555Sdesstatic void 27492555Sdeskeyprint(con *c, Key *key) 27592555Sdes{ 276147005Sdes char *host = c->c_output_name ? c->c_output_name : c->c_name; 277147005Sdes 27892555Sdes if (!key) 27992555Sdes return; 280147005Sdes if (hash_hosts && (host = host_hash(host, NULL, 0)) == NULL) 281147005Sdes fatal("host_hash failed"); 28292555Sdes 283147005Sdes fprintf(stdout, "%s ", host); 28492555Sdes key_write(key, stdout); 28576259Sgreen fputs("\n", stdout); 28676259Sgreen} 28776259Sgreen 28892555Sdesstatic int 28976259Sgreentcpconnect(char *host) 29076259Sgreen{ 29176259Sgreen struct addrinfo hints, *ai, *aitop; 29276259Sgreen char strport[NI_MAXSERV]; 29376259Sgreen int gaierr, s = -1; 29476259Sgreen 29592555Sdes snprintf(strport, sizeof strport, "%d", ssh_port); 29676259Sgreen memset(&hints, 0, sizeof(hints)); 29792555Sdes hints.ai_family = IPv4or6; 29876259Sgreen hints.ai_socktype = SOCK_STREAM; 29976259Sgreen if ((gaierr = getaddrinfo(host, strport, &hints, &aitop)) != 0) 300181111Sdes fatal("getaddrinfo %s: %s", host, ssh_gai_strerror(gaierr)); 30176259Sgreen for (ai = aitop; ai; ai = ai->ai_next) { 302124211Sdes s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); 30376259Sgreen if (s < 0) { 30476259Sgreen error("socket: %s", strerror(errno)); 30576259Sgreen continue; 30676259Sgreen } 307137019Sdes if (set_nonblock(s) == -1) 308137019Sdes fatal("%s: set_nonblock(%d)", __func__, s); 30976259Sgreen if (connect(s, ai->ai_addr, ai->ai_addrlen) < 0 && 31076259Sgreen errno != EINPROGRESS) 31176259Sgreen error("connect (`%s'): %s", host, strerror(errno)); 31276259Sgreen else 31376259Sgreen break; 31476259Sgreen close(s); 31576259Sgreen s = -1; 31676259Sgreen } 31776259Sgreen freeaddrinfo(aitop); 31876259Sgreen return s; 31976259Sgreen} 32076259Sgreen 32192555Sdesstatic int 32292555Sdesconalloc(char *iname, char *oname, int keytype) 32376259Sgreen{ 324106130Sdes char *namebase, *name, *namelist; 32576259Sgreen int s; 32676259Sgreen 32776259Sgreen namebase = namelist = xstrdup(iname); 32876259Sgreen 32976259Sgreen do { 33076259Sgreen name = xstrsep(&namelist, ","); 33176259Sgreen if (!name) { 332255767Sdes free(namebase); 33376259Sgreen return (-1); 33476259Sgreen } 33576259Sgreen } while ((s = tcpconnect(name)) < 0); 33676259Sgreen 33776259Sgreen if (s >= maxfd) 33876259Sgreen fatal("conalloc: fdno %d too high", s); 33976259Sgreen if (fdcon[s].c_status) 34076259Sgreen fatal("conalloc: attempt to reuse fdno %d", s); 34176259Sgreen 34276259Sgreen fdcon[s].c_fd = s; 34376259Sgreen fdcon[s].c_status = CS_CON; 34476259Sgreen fdcon[s].c_namebase = namebase; 34576259Sgreen fdcon[s].c_name = name; 34676259Sgreen fdcon[s].c_namelist = namelist; 34776259Sgreen fdcon[s].c_output_name = xstrdup(oname); 34876259Sgreen fdcon[s].c_data = (char *) &fdcon[s].c_plen; 34976259Sgreen fdcon[s].c_len = 4; 35076259Sgreen fdcon[s].c_off = 0; 35192555Sdes fdcon[s].c_keytype = keytype; 35276259Sgreen gettimeofday(&fdcon[s].c_tv, NULL); 35376259Sgreen fdcon[s].c_tv.tv_sec += timeout; 35476259Sgreen TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link); 35576259Sgreen FD_SET(s, read_wait); 35676259Sgreen ncon++; 35776259Sgreen return (s); 35876259Sgreen} 35976259Sgreen 36092555Sdesstatic void 36176259Sgreenconfree(int s) 36276259Sgreen{ 36376259Sgreen if (s >= maxfd || fdcon[s].c_status == CS_UNUSED) 36476259Sgreen fatal("confree: attempt to free bad fdno %d", s); 36576259Sgreen close(s); 366255767Sdes free(fdcon[s].c_namebase); 367255767Sdes free(fdcon[s].c_output_name); 36876259Sgreen if (fdcon[s].c_status == CS_KEYS) 369255767Sdes free(fdcon[s].c_data); 37076259Sgreen fdcon[s].c_status = CS_UNUSED; 37192555Sdes fdcon[s].c_keytype = 0; 37276259Sgreen TAILQ_REMOVE(&tq, &fdcon[s], c_link); 37376259Sgreen FD_CLR(s, read_wait); 37476259Sgreen ncon--; 37576259Sgreen} 37676259Sgreen 37792555Sdesstatic void 37876259Sgreencontouch(int s) 37976259Sgreen{ 38076259Sgreen TAILQ_REMOVE(&tq, &fdcon[s], c_link); 38176259Sgreen gettimeofday(&fdcon[s].c_tv, NULL); 38276259Sgreen fdcon[s].c_tv.tv_sec += timeout; 38376259Sgreen TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link); 38476259Sgreen} 38576259Sgreen 38692555Sdesstatic int 38776259Sgreenconrecycle(int s) 38876259Sgreen{ 389106130Sdes con *c = &fdcon[s]; 39076259Sgreen int ret; 39176259Sgreen 39292555Sdes ret = conalloc(c->c_namelist, c->c_output_name, c->c_keytype); 39376259Sgreen confree(s); 39476259Sgreen return (ret); 39576259Sgreen} 39676259Sgreen 39792555Sdesstatic void 39876259Sgreencongreet(int s) 39976259Sgreen{ 400149753Sdes int n = 0, remote_major = 0, remote_minor = 0; 40192555Sdes char buf[256], *cp; 40292555Sdes char remote_version[sizeof buf]; 40376259Sgreen size_t bufsiz; 40476259Sgreen con *c = &fdcon[s]; 40576259Sgreen 406157019Sdes for (;;) { 407157019Sdes memset(buf, '\0', sizeof(buf)); 408157019Sdes bufsiz = sizeof(buf); 409157019Sdes cp = buf; 410157019Sdes while (bufsiz-- && 411157019Sdes (n = atomicio(read, s, cp, 1)) == 1 && *cp != '\n') { 412157019Sdes if (*cp == '\r') 413157019Sdes *cp = '\n'; 414157019Sdes cp++; 415157019Sdes } 416157019Sdes if (n != 1 || strncmp(buf, "SSH-", 4) == 0) 417157019Sdes break; 41892555Sdes } 419149753Sdes if (n == 0) { 420149753Sdes switch (errno) { 421149753Sdes case EPIPE: 422149753Sdes error("%s: Connection closed by remote host", c->c_name); 423149753Sdes break; 424149753Sdes case ECONNREFUSED: 425149753Sdes break; 426149753Sdes default: 42776259Sgreen error("read (%s): %s", c->c_name, strerror(errno)); 428149753Sdes break; 429149753Sdes } 43076259Sgreen conrecycle(s); 43176259Sgreen return; 43276259Sgreen } 43376259Sgreen if (*cp != '\n' && *cp != '\r') { 43476259Sgreen error("%s: bad greeting", c->c_name); 43576259Sgreen confree(s); 43676259Sgreen return; 43776259Sgreen } 43876259Sgreen *cp = '\0'; 43992555Sdes if (sscanf(buf, "SSH-%d.%d-%[^\n]\n", 44092555Sdes &remote_major, &remote_minor, remote_version) == 3) 44192555Sdes compat_datafellows(remote_version); 44292555Sdes else 44392555Sdes datafellows = 0; 44492555Sdes if (c->c_keytype != KT_RSA1) { 44592555Sdes if (!ssh2_capable(remote_major, remote_minor)) { 44692555Sdes debug("%s doesn't support ssh2", c->c_name); 44792555Sdes confree(s); 44892555Sdes return; 44992555Sdes } 45092555Sdes } else if (remote_major != 1) { 45192555Sdes debug("%s doesn't support ssh1", c->c_name); 45292555Sdes confree(s); 45392555Sdes return; 45492555Sdes } 45592555Sdes fprintf(stderr, "# %s %s\n", c->c_name, chop(buf)); 45692555Sdes n = snprintf(buf, sizeof buf, "SSH-%d.%d-OpenSSH-keyscan\r\n", 45792555Sdes c->c_keytype == KT_RSA1? PROTOCOL_MAJOR_1 : PROTOCOL_MAJOR_2, 45892555Sdes c->c_keytype == KT_RSA1? PROTOCOL_MINOR_1 : PROTOCOL_MINOR_2); 459149753Sdes if (n < 0 || (size_t)n >= sizeof(buf)) { 460149753Sdes error("snprintf: buffer too small"); 461149753Sdes confree(s); 462149753Sdes return; 463149753Sdes } 464149753Sdes if (atomicio(vwrite, s, buf, n) != (size_t)n) { 46576259Sgreen error("write (%s): %s", c->c_name, strerror(errno)); 46676259Sgreen confree(s); 46776259Sgreen return; 46876259Sgreen } 46992555Sdes if (c->c_keytype != KT_RSA1) { 47092555Sdes keyprint(c, keygrab_ssh2(c)); 47192555Sdes confree(s); 47292555Sdes return; 47392555Sdes } 47476259Sgreen c->c_status = CS_SIZE; 47576259Sgreen contouch(s); 47676259Sgreen} 47776259Sgreen 47892555Sdesstatic void 47976259Sgreenconread(int s) 48076259Sgreen{ 481106130Sdes con *c = &fdcon[s]; 482149753Sdes size_t n; 48376259Sgreen 48476259Sgreen if (c->c_status == CS_CON) { 48576259Sgreen congreet(s); 48676259Sgreen return; 48776259Sgreen } 488137019Sdes n = atomicio(read, s, c->c_data + c->c_off, c->c_len - c->c_off); 489149753Sdes if (n == 0) { 49076259Sgreen error("read (%s): %s", c->c_name, strerror(errno)); 49176259Sgreen confree(s); 49276259Sgreen return; 49376259Sgreen } 49476259Sgreen c->c_off += n; 49576259Sgreen 49676259Sgreen if (c->c_off == c->c_len) 49776259Sgreen switch (c->c_status) { 49876259Sgreen case CS_SIZE: 49976259Sgreen c->c_plen = htonl(c->c_plen); 50076259Sgreen c->c_len = c->c_plen + 8 - (c->c_plen & 7); 50176259Sgreen c->c_off = 0; 50276259Sgreen c->c_data = xmalloc(c->c_len); 50376259Sgreen c->c_status = CS_KEYS; 50476259Sgreen break; 50576259Sgreen case CS_KEYS: 50692555Sdes keyprint(c, keygrab_ssh1(c)); 50776259Sgreen confree(s); 50876259Sgreen return; 50976259Sgreen default: 51076259Sgreen fatal("conread: invalid status %d", c->c_status); 51176259Sgreen break; 51276259Sgreen } 51376259Sgreen 51476259Sgreen contouch(s); 51576259Sgreen} 51676259Sgreen 51792555Sdesstatic void 51876259Sgreenconloop(void) 51976259Sgreen{ 520106130Sdes struct timeval seltime, now; 52176259Sgreen fd_set *r, *e; 522106130Sdes con *c; 52376259Sgreen int i; 52476259Sgreen 52576259Sgreen gettimeofday(&now, NULL); 52698675Sdes c = TAILQ_FIRST(&tq); 52776259Sgreen 52876259Sgreen if (c && (c->c_tv.tv_sec > now.tv_sec || 52976259Sgreen (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec > now.tv_usec))) { 53076259Sgreen seltime = c->c_tv; 53176259Sgreen seltime.tv_sec -= now.tv_sec; 53276259Sgreen seltime.tv_usec -= now.tv_usec; 53376259Sgreen if (seltime.tv_usec < 0) { 53476259Sgreen seltime.tv_usec += 1000000; 53576259Sgreen seltime.tv_sec--; 53676259Sgreen } 53776259Sgreen } else 538226046Sdes timerclear(&seltime); 53976259Sgreen 540162856Sdes r = xcalloc(read_wait_nfdset, sizeof(fd_mask)); 541162856Sdes e = xcalloc(read_wait_nfdset, sizeof(fd_mask)); 542162856Sdes memcpy(r, read_wait, read_wait_nfdset * sizeof(fd_mask)); 543162856Sdes memcpy(e, read_wait, read_wait_nfdset * sizeof(fd_mask)); 54476259Sgreen 54576259Sgreen while (select(maxfd, r, NULL, e, &seltime) == -1 && 546181111Sdes (errno == EAGAIN || errno == EINTR || errno == EWOULDBLOCK)) 54776259Sgreen ; 54876259Sgreen 54976259Sgreen for (i = 0; i < maxfd; i++) { 55076259Sgreen if (FD_ISSET(i, e)) { 55176259Sgreen error("%s: exception!", fdcon[i].c_name); 55276259Sgreen confree(i); 55376259Sgreen } else if (FD_ISSET(i, r)) 55476259Sgreen conread(i); 55576259Sgreen } 556255767Sdes free(r); 557255767Sdes free(e); 55876259Sgreen 55998675Sdes c = TAILQ_FIRST(&tq); 56076259Sgreen while (c && (c->c_tv.tv_sec < now.tv_sec || 56176259Sgreen (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec < now.tv_usec))) { 56276259Sgreen int s = c->c_fd; 56376259Sgreen 56498675Sdes c = TAILQ_NEXT(c, c_link); 56576259Sgreen conrecycle(s); 56676259Sgreen } 56776259Sgreen} 56876259Sgreen 56992555Sdesstatic void 57092555Sdesdo_host(char *host) 57176259Sgreen{ 57292555Sdes char *name = strnnsep(&host, " \t\n"); 57392555Sdes int j; 57476259Sgreen 57592555Sdes if (name == NULL) 57692555Sdes return; 577221420Sdes for (j = KT_RSA1; j <= KT_ECDSA; j *= 2) { 57892555Sdes if (get_keytypes & j) { 57992555Sdes while (ncon >= MAXCON) 58092555Sdes conloop(); 58192555Sdes conalloc(name, *host ? host : name, j); 58276259Sgreen } 58376259Sgreen } 58476259Sgreen} 58576259Sgreen 58676259Sgreenvoid 58792555Sdesfatal(const char *fmt,...) 58892555Sdes{ 58992555Sdes va_list args; 590106130Sdes 59192555Sdes va_start(args, fmt); 59292555Sdes do_log(SYSLOG_LEVEL_FATAL, fmt, args); 59392555Sdes va_end(args); 59492555Sdes if (nonfatal_fatal) 59592555Sdes longjmp(kexjmp, -1); 59692555Sdes else 597126277Sdes exit(255); 59892555Sdes} 59992555Sdes 60092555Sdesstatic void 60176259Sgreenusage(void) 60276259Sgreen{ 603192595Sdes fprintf(stderr, 604192595Sdes "usage: %s [-46Hv] [-f file] [-p port] [-T timeout] [-t type]\n" 605192595Sdes "\t\t [host | addrlist namelist] ...\n", 60692555Sdes __progname); 60792555Sdes exit(1); 60876259Sgreen} 60976259Sgreen 61076259Sgreenint 61176259Sgreenmain(int argc, char **argv) 61276259Sgreen{ 61392555Sdes int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO; 614215116Sdes int opt, fopt_count = 0, j; 615215116Sdes char *tname, *cp, line[NI_MAXHOST]; 616215116Sdes FILE *fp; 617215116Sdes u_long linenum; 61876259Sgreen 61992555Sdes extern int optind; 62092555Sdes extern char *optarg; 62192555Sdes 622124211Sdes __progname = ssh_get_progname(argv[0]); 62398941Sdes seed_rng(); 62476259Sgreen TAILQ_INIT(&tq); 62576259Sgreen 626157019Sdes /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */ 627157019Sdes sanitise_stdfd(); 628157019Sdes 62992555Sdes if (argc <= 1) 63076259Sgreen usage(); 63176259Sgreen 632147005Sdes while ((opt = getopt(argc, argv, "Hv46p:T:t:f:")) != -1) { 63392555Sdes switch (opt) { 634147005Sdes case 'H': 635147005Sdes hash_hosts = 1; 636147005Sdes break; 63792555Sdes case 'p': 63892555Sdes ssh_port = a2port(optarg); 639192595Sdes if (ssh_port <= 0) { 64092555Sdes fprintf(stderr, "Bad port '%s'\n", optarg); 64192555Sdes exit(1); 64292555Sdes } 64392555Sdes break; 64492555Sdes case 'T': 645106130Sdes timeout = convtime(optarg); 646106130Sdes if (timeout == -1 || timeout == 0) { 647106130Sdes fprintf(stderr, "Bad timeout '%s'\n", optarg); 64876259Sgreen usage(); 649106130Sdes } 65092555Sdes break; 65192555Sdes case 'v': 65292555Sdes if (!debug_flag) { 65392555Sdes debug_flag = 1; 65492555Sdes log_level = SYSLOG_LEVEL_DEBUG1; 65592555Sdes } 65692555Sdes else if (log_level < SYSLOG_LEVEL_DEBUG3) 65792555Sdes log_level++; 65892555Sdes else 65992555Sdes fatal("Too high debugging level."); 66092555Sdes break; 66192555Sdes case 'f': 66292555Sdes if (strcmp(optarg, "-") == 0) 66392555Sdes optarg = NULL; 66492555Sdes argv[fopt_count++] = optarg; 66592555Sdes break; 66692555Sdes case 't': 66792555Sdes get_keytypes = 0; 66892555Sdes tname = strtok(optarg, ","); 66992555Sdes while (tname) { 67092555Sdes int type = key_type_from_name(tname); 67192555Sdes switch (type) { 67292555Sdes case KEY_RSA1: 67392555Sdes get_keytypes |= KT_RSA1; 67492555Sdes break; 67592555Sdes case KEY_DSA: 67692555Sdes get_keytypes |= KT_DSA; 67792555Sdes break; 678221420Sdes case KEY_ECDSA: 679221420Sdes get_keytypes |= KT_ECDSA; 680221420Sdes break; 68192555Sdes case KEY_RSA: 68292555Sdes get_keytypes |= KT_RSA; 68392555Sdes break; 68492555Sdes case KEY_UNSPEC: 68592555Sdes fatal("unknown key type %s", tname); 68692555Sdes } 68792555Sdes tname = strtok(NULL, ","); 68892555Sdes } 68992555Sdes break; 69092555Sdes case '4': 69192555Sdes IPv4or6 = AF_INET; 69292555Sdes break; 69392555Sdes case '6': 69492555Sdes IPv4or6 = AF_INET6; 69592555Sdes break; 69692555Sdes case '?': 69792555Sdes default: 69892555Sdes usage(); 69976259Sgreen } 70076259Sgreen } 70192555Sdes if (optind == argc && !fopt_count) 70276259Sgreen usage(); 70376259Sgreen 70492555Sdes log_init("ssh-keyscan", log_level, SYSLOG_FACILITY_USER, 1); 70592555Sdes 70676259Sgreen maxfd = fdlim_get(1); 70776259Sgreen if (maxfd < 0) 70876259Sgreen fatal("%s: fdlim_get: bad value", __progname); 70976259Sgreen if (maxfd > MAXMAXFD) 71076259Sgreen maxfd = MAXMAXFD; 71176259Sgreen if (MAXCON <= 0) 71276259Sgreen fatal("%s: not enough file descriptors", __progname); 71376259Sgreen if (maxfd > fdlim_get(0)) 71476259Sgreen fdlim_set(maxfd); 715162856Sdes fdcon = xcalloc(maxfd, sizeof(con)); 71676259Sgreen 717162856Sdes read_wait_nfdset = howmany(maxfd, NFDBITS); 718162856Sdes read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask)); 71976259Sgreen 720215116Sdes for (j = 0; j < fopt_count; j++) { 721215116Sdes if (argv[j] == NULL) 722215116Sdes fp = stdin; 723215116Sdes else if ((fp = fopen(argv[j], "r")) == NULL) 724215116Sdes fatal("%s: %s: %s", __progname, argv[j], 725215116Sdes strerror(errno)); 726215116Sdes linenum = 0; 72776259Sgreen 728215116Sdes while (read_keyfile_line(fp, 729215116Sdes argv[j] == NULL ? "(stdin)" : argv[j], line, sizeof(line), 730215116Sdes &linenum) != -1) { 731215116Sdes /* Chomp off trailing whitespace and comments */ 732215116Sdes if ((cp = strchr(line, '#')) == NULL) 733215116Sdes cp = line + strlen(line) - 1; 734215116Sdes while (cp >= line) { 735215116Sdes if (*cp == ' ' || *cp == '\t' || 736215116Sdes *cp == '\n' || *cp == '#') 737215116Sdes *cp-- = '\0'; 738215116Sdes else 739215116Sdes break; 740215116Sdes } 741215116Sdes 742215116Sdes /* Skip empty lines */ 743215116Sdes if (*line == '\0') 74492555Sdes continue; 745215116Sdes 746215116Sdes do_host(line); 74776259Sgreen } 748215116Sdes 749215116Sdes if (ferror(fp)) 750215116Sdes fatal("%s: %s: %s", __progname, argv[j], 751215116Sdes strerror(errno)); 752215116Sdes 753215116Sdes fclose(fp); 75492555Sdes } 75592555Sdes 75692555Sdes while (optind < argc) 75792555Sdes do_host(argv[optind++]); 75892555Sdes 75976259Sgreen while (ncon > 0) 76076259Sgreen conloop(); 76176259Sgreen 76276259Sgreen return (0); 76376259Sgreen} 764