You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
856 lines
19 KiB
856 lines
19 KiB
/* $OpenBSD: ssh-keyscan.c,v 1.109 2017/03/10 04:26:06 djm Exp $ */ |
|
/* |
|
* Copyright 1995, 1996 by David Mazieres <dm@lcs.mit.edu>. |
|
* |
|
* Modification and redistribution in source and binary forms is |
|
* permitted provided that due credit is given to the author and the |
|
* OpenBSD project by leaving this copyright notice intact. |
|
*/ |
|
|
|
#include "includes.h" |
|
|
|
#include <sys/types.h> |
|
#include "openbsd-compat/sys-queue.h" |
|
#include <sys/resource.h> |
|
#ifdef HAVE_SYS_TIME_H |
|
# include <sys/time.h> |
|
#endif |
|
|
|
#include <netinet/in.h> |
|
#include <arpa/inet.h> |
|
|
|
#include <openssl/bn.h> |
|
|
|
#include <netdb.h> |
|
#include <errno.h> |
|
#include <stdarg.h> |
|
#include <stdio.h> |
|
#include <stdlib.h> |
|
#include <signal.h> |
|
#include <string.h> |
|
#include <unistd.h> |
|
|
|
#include "xmalloc.h" |
|
#include "ssh.h" |
|
#include "ssh1.h" |
|
#include "sshbuf.h" |
|
#include "sshkey.h" |
|
#include "cipher.h" |
|
#include "kex.h" |
|
#include "compat.h" |
|
#include "myproposal.h" |
|
#include "packet.h" |
|
#include "dispatch.h" |
|
#include "log.h" |
|
#include "atomicio.h" |
|
#include "misc.h" |
|
#include "hostfile.h" |
|
#include "ssherr.h" |
|
#include "ssh_api.h" |
|
|
|
/* Flag indicating whether IPv4 or IPv6. This can be set on the command line. |
|
Default value is AF_UNSPEC means both IPv4 and IPv6. */ |
|
int IPv4or6 = AF_UNSPEC; |
|
|
|
int ssh_port = SSH_DEFAULT_PORT; |
|
|
|
#define KT_RSA1 1 |
|
#define KT_DSA 2 |
|
#define KT_RSA 4 |
|
#define KT_ECDSA 8 |
|
#define KT_ED25519 16 |
|
|
|
int get_cert = 0; |
|
int get_keytypes = KT_RSA|KT_ECDSA|KT_ED25519; |
|
|
|
int hash_hosts = 0; /* Hash hostname on output */ |
|
|
|
#define MAXMAXFD 256 |
|
|
|
/* The number of seconds after which to give up on a TCP connection */ |
|
int timeout = 5; |
|
|
|
int maxfd; |
|
#define MAXCON (maxfd - 10) |
|
|
|
extern char *__progname; |
|
fd_set *read_wait; |
|
size_t read_wait_nfdset; |
|
int ncon; |
|
|
|
struct ssh *active_state = NULL; /* XXX needed for linking */ |
|
|
|
/* |
|
* Keep a connection structure for each file descriptor. The state |
|
* associated with file descriptor n is held in fdcon[n]. |
|
*/ |
|
typedef struct Connection { |
|
u_char c_status; /* State of connection on this file desc. */ |
|
#define CS_UNUSED 0 /* File descriptor unused */ |
|
#define CS_CON 1 /* Waiting to connect/read greeting */ |
|
#define CS_SIZE 2 /* Waiting to read initial packet size */ |
|
#define CS_KEYS 3 /* Waiting to read public key packet */ |
|
int c_fd; /* Quick lookup: c->c_fd == c - fdcon */ |
|
int c_plen; /* Packet length field for ssh packet */ |
|
int c_len; /* Total bytes which must be read. */ |
|
int c_off; /* Length of data read so far. */ |
|
int c_keytype; /* Only one of KT_RSA1, KT_DSA, or KT_RSA */ |
|
sig_atomic_t c_done; /* SSH2 done */ |
|
char *c_namebase; /* Address to free for c_name and c_namelist */ |
|
char *c_name; /* Hostname of connection for errors */ |
|
char *c_namelist; /* Pointer to other possible addresses */ |
|
char *c_output_name; /* Hostname of connection for output */ |
|
char *c_data; /* Data read from this fd */ |
|
struct ssh *c_ssh; /* SSH-connection */ |
|
struct timeval c_tv; /* Time at which connection gets aborted */ |
|
TAILQ_ENTRY(Connection) c_link; /* List of connections in timeout order. */ |
|
} con; |
|
|
|
TAILQ_HEAD(conlist, Connection) tq; /* Timeout Queue */ |
|
con *fdcon; |
|
|
|
static void keyprint(con *c, struct sshkey *key); |
|
|
|
static int |
|
fdlim_get(int hard) |
|
{ |
|
#if defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE) |
|
struct rlimit rlfd; |
|
|
|
if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0) |
|
return (-1); |
|
if ((hard ? rlfd.rlim_max : rlfd.rlim_cur) == RLIM_INFINITY) |
|
return SSH_SYSFDMAX; |
|
else |
|
return hard ? rlfd.rlim_max : rlfd.rlim_cur; |
|
#else |
|
return SSH_SYSFDMAX; |
|
#endif |
|
} |
|
|
|
static int |
|
fdlim_set(int lim) |
|
{ |
|
#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE) |
|
struct rlimit rlfd; |
|
#endif |
|
|
|
if (lim <= 0) |
|
return (-1); |
|
#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE) |
|
if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0) |
|
return (-1); |
|
rlfd.rlim_cur = lim; |
|
if (setrlimit(RLIMIT_NOFILE, &rlfd) < 0) |
|
return (-1); |
|
#elif defined (HAVE_SETDTABLESIZE) |
|
setdtablesize(lim); |
|
#endif |
|
return (0); |
|
} |
|
|
|
/* |
|
* This is an strsep function that returns a null field for adjacent |
|
* separators. This is the same as the 4.4BSD strsep, but different from the |
|
* one in the GNU libc. |
|
*/ |
|
static char * |
|
xstrsep(char **str, const char *delim) |
|
{ |
|
char *s, *e; |
|
|
|
if (!**str) |
|
return (NULL); |
|
|
|
s = *str; |
|
e = s + strcspn(s, delim); |
|
|
|
if (*e != '\0') |
|
*e++ = '\0'; |
|
*str = e; |
|
|
|
return (s); |
|
} |
|
|
|
/* |
|
* Get the next non-null token (like GNU strsep). Strsep() will return a |
|
* null token for two adjacent separators, so we may have to loop. |
|
*/ |
|
static char * |
|
strnnsep(char **stringp, char *delim) |
|
{ |
|
char *tok; |
|
|
|
do { |
|
tok = xstrsep(stringp, delim); |
|
} while (tok && *tok == '\0'); |
|
return (tok); |
|
} |
|
|
|
#ifdef WITH_SSH1 |
|
static struct sshkey * |
|
keygrab_ssh1(con *c) |
|
{ |
|
static struct sshkey *rsa; |
|
static struct sshbuf *msg; |
|
int r; |
|
u_char type; |
|
|
|
if (rsa == NULL) { |
|
if ((rsa = sshkey_new(KEY_RSA1)) == NULL) { |
|
error("%s: sshkey_new failed", __func__); |
|
return NULL; |
|
} |
|
if ((msg = sshbuf_new()) == NULL) |
|
fatal("%s: sshbuf_new failed", __func__); |
|
} |
|
if ((r = sshbuf_put(msg, c->c_data, c->c_plen)) != 0 || |
|
(r = sshbuf_consume(msg, 8 - (c->c_plen & 7))) != 0 || /* padding */ |
|
(r = sshbuf_get_u8(msg, &type)) != 0) |
|
goto buf_err; |
|
if (type != (int) SSH_SMSG_PUBLIC_KEY) { |
|
error("%s: invalid packet type", c->c_name); |
|
sshbuf_reset(msg); |
|
return NULL; |
|
} |
|
if ((r = sshbuf_consume(msg, 8)) != 0 || /* cookie */ |
|
/* server key */ |
|
(r = sshbuf_get_u32(msg, NULL)) != 0 || |
|
(r = sshbuf_get_bignum1(msg, NULL)) != 0 || |
|
(r = sshbuf_get_bignum1(msg, NULL)) != 0 || |
|
/* host key */ |
|
(r = sshbuf_get_u32(msg, NULL)) != 0 || |
|
(r = sshbuf_get_bignum1(msg, rsa->rsa->e)) != 0 || |
|
(r = sshbuf_get_bignum1(msg, rsa->rsa->n)) != 0) { |
|
buf_err: |
|
error("%s: buffer error: %s", __func__, ssh_err(r)); |
|
sshbuf_reset(msg); |
|
return NULL; |
|
} |
|
|
|
sshbuf_reset(msg); |
|
|
|
return (rsa); |
|
} |
|
#endif |
|
|
|
static int |
|
key_print_wrapper(struct sshkey *hostkey, struct ssh *ssh) |
|
{ |
|
con *c; |
|
|
|
if ((c = ssh_get_app_data(ssh)) != NULL) |
|
keyprint(c, hostkey); |
|
/* always abort key exchange */ |
|
return -1; |
|
} |
|
|
|
static int |
|
ssh2_capable(int remote_major, int remote_minor) |
|
{ |
|
switch (remote_major) { |
|
case 1: |
|
if (remote_minor == 99) |
|
return 1; |
|
break; |
|
case 2: |
|
return 1; |
|
default: |
|
break; |
|
} |
|
return 0; |
|
} |
|
|
|
static void |
|
keygrab_ssh2(con *c) |
|
{ |
|
char *myproposal[PROPOSAL_MAX] = { KEX_CLIENT }; |
|
int r; |
|
|
|
enable_compat20(); |
|
switch (c->c_keytype) { |
|
case KT_DSA: |
|
myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? |
|
"ssh-dss-cert-v01@openssh.com" : "ssh-dss"; |
|
break; |
|
case KT_RSA: |
|
myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? |
|
"ssh-rsa-cert-v01@openssh.com" : "ssh-rsa"; |
|
break; |
|
case KT_ED25519: |
|
myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? |
|
"ssh-ed25519-cert-v01@openssh.com" : "ssh-ed25519"; |
|
break; |
|
case KT_ECDSA: |
|
myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? |
|
"ecdsa-sha2-nistp256-cert-v01@openssh.com," |
|
"ecdsa-sha2-nistp384-cert-v01@openssh.com," |
|
"ecdsa-sha2-nistp521-cert-v01@openssh.com" : |
|
"ecdsa-sha2-nistp256," |
|
"ecdsa-sha2-nistp384," |
|
"ecdsa-sha2-nistp521"; |
|
break; |
|
default: |
|
fatal("unknown key type %d", c->c_keytype); |
|
break; |
|
} |
|
if ((r = kex_setup(c->c_ssh, myproposal)) != 0) { |
|
free(c->c_ssh); |
|
fprintf(stderr, "kex_setup: %s\n", ssh_err(r)); |
|
exit(1); |
|
} |
|
#ifdef WITH_OPENSSL |
|
c->c_ssh->kex->kex[KEX_DH_GRP1_SHA1] = kexdh_client; |
|
c->c_ssh->kex->kex[KEX_DH_GRP14_SHA1] = kexdh_client; |
|
c->c_ssh->kex->kex[KEX_DH_GRP14_SHA256] = kexdh_client; |
|
c->c_ssh->kex->kex[KEX_DH_GRP16_SHA512] = kexdh_client; |
|
c->c_ssh->kex->kex[KEX_DH_GRP18_SHA512] = kexdh_client; |
|
c->c_ssh->kex->kex[KEX_DH_GEX_SHA1] = kexgex_client; |
|
c->c_ssh->kex->kex[KEX_DH_GEX_SHA256] = kexgex_client; |
|
# ifdef OPENSSL_HAS_ECC |
|
c->c_ssh->kex->kex[KEX_ECDH_SHA2] = kexecdh_client; |
|
# endif |
|
#endif |
|
c->c_ssh->kex->kex[KEX_C25519_SHA256] = kexc25519_client; |
|
ssh_set_verify_host_key_callback(c->c_ssh, key_print_wrapper); |
|
/* |
|
* do the key-exchange until an error occurs or until |
|
* the key_print_wrapper() callback sets c_done. |
|
*/ |
|
ssh_dispatch_run(c->c_ssh, DISPATCH_BLOCK, &c->c_done, c->c_ssh); |
|
} |
|
|
|
static void |
|
keyprint_one(const char *host, struct sshkey *key) |
|
{ |
|
char *hostport; |
|
const char *known_host, *hashed; |
|
|
|
hostport = put_host_port(host, ssh_port); |
|
lowercase(hostport); |
|
if (hash_hosts && (hashed = host_hash(host, NULL, 0)) == NULL) |
|
fatal("host_hash failed"); |
|
known_host = hash_hosts ? hashed : hostport; |
|
if (!get_cert) |
|
fprintf(stdout, "%s ", known_host); |
|
sshkey_write(key, stdout); |
|
fputs("\n", stdout); |
|
free(hostport); |
|
} |
|
|
|
static void |
|
keyprint(con *c, struct sshkey *key) |
|
{ |
|
char *hosts = c->c_output_name ? c->c_output_name : c->c_name; |
|
char *host, *ohosts; |
|
|
|
if (key == NULL) |
|
return; |
|
if (get_cert || (!hash_hosts && ssh_port == SSH_DEFAULT_PORT)) { |
|
keyprint_one(hosts, key); |
|
return; |
|
} |
|
ohosts = hosts = xstrdup(hosts); |
|
while ((host = strsep(&hosts, ",")) != NULL) |
|
keyprint_one(host, key); |
|
free(ohosts); |
|
} |
|
|
|
static int |
|
tcpconnect(char *host) |
|
{ |
|
struct addrinfo hints, *ai, *aitop; |
|
char strport[NI_MAXSERV]; |
|
int gaierr, s = -1; |
|
|
|
snprintf(strport, sizeof strport, "%d", ssh_port); |
|
memset(&hints, 0, sizeof(hints)); |
|
hints.ai_family = IPv4or6; |
|
hints.ai_socktype = SOCK_STREAM; |
|
if ((gaierr = getaddrinfo(host, strport, &hints, &aitop)) != 0) { |
|
error("getaddrinfo %s: %s", host, ssh_gai_strerror(gaierr)); |
|
return -1; |
|
} |
|
for (ai = aitop; ai; ai = ai->ai_next) { |
|
s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); |
|
if (s < 0) { |
|
error("socket: %s", strerror(errno)); |
|
continue; |
|
} |
|
if (set_nonblock(s) == -1) |
|
fatal("%s: set_nonblock(%d)", __func__, s); |
|
if (connect(s, ai->ai_addr, ai->ai_addrlen) < 0 && |
|
errno != EINPROGRESS) |
|
error("connect (`%s'): %s", host, strerror(errno)); |
|
else |
|
break; |
|
close(s); |
|
s = -1; |
|
} |
|
freeaddrinfo(aitop); |
|
return s; |
|
} |
|
|
|
static int |
|
conalloc(char *iname, char *oname, int keytype) |
|
{ |
|
char *namebase, *name, *namelist; |
|
int s; |
|
|
|
namebase = namelist = xstrdup(iname); |
|
|
|
do { |
|
name = xstrsep(&namelist, ","); |
|
if (!name) { |
|
free(namebase); |
|
return (-1); |
|
} |
|
} while ((s = tcpconnect(name)) < 0); |
|
|
|
if (s >= maxfd) |
|
fatal("conalloc: fdno %d too high", s); |
|
if (fdcon[s].c_status) |
|
fatal("conalloc: attempt to reuse fdno %d", s); |
|
|
|
debug3("%s: oname %s kt %d", __func__, oname, keytype); |
|
fdcon[s].c_fd = s; |
|
fdcon[s].c_status = CS_CON; |
|
fdcon[s].c_namebase = namebase; |
|
fdcon[s].c_name = name; |
|
fdcon[s].c_namelist = namelist; |
|
fdcon[s].c_output_name = xstrdup(oname); |
|
fdcon[s].c_data = (char *) &fdcon[s].c_plen; |
|
fdcon[s].c_len = 4; |
|
fdcon[s].c_off = 0; |
|
fdcon[s].c_keytype = keytype; |
|
gettimeofday(&fdcon[s].c_tv, NULL); |
|
fdcon[s].c_tv.tv_sec += timeout; |
|
TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link); |
|
FD_SET(s, read_wait); |
|
ncon++; |
|
return (s); |
|
} |
|
|
|
static void |
|
confree(int s) |
|
{ |
|
if (s >= maxfd || fdcon[s].c_status == CS_UNUSED) |
|
fatal("confree: attempt to free bad fdno %d", s); |
|
close(s); |
|
free(fdcon[s].c_namebase); |
|
free(fdcon[s].c_output_name); |
|
if (fdcon[s].c_status == CS_KEYS) |
|
free(fdcon[s].c_data); |
|
fdcon[s].c_status = CS_UNUSED; |
|
fdcon[s].c_keytype = 0; |
|
if (fdcon[s].c_ssh) { |
|
ssh_packet_close(fdcon[s].c_ssh); |
|
free(fdcon[s].c_ssh); |
|
fdcon[s].c_ssh = NULL; |
|
} |
|
TAILQ_REMOVE(&tq, &fdcon[s], c_link); |
|
FD_CLR(s, read_wait); |
|
ncon--; |
|
} |
|
|
|
static void |
|
contouch(int s) |
|
{ |
|
TAILQ_REMOVE(&tq, &fdcon[s], c_link); |
|
gettimeofday(&fdcon[s].c_tv, NULL); |
|
fdcon[s].c_tv.tv_sec += timeout; |
|
TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link); |
|
} |
|
|
|
static int |
|
conrecycle(int s) |
|
{ |
|
con *c = &fdcon[s]; |
|
int ret; |
|
|
|
ret = conalloc(c->c_namelist, c->c_output_name, c->c_keytype); |
|
confree(s); |
|
return (ret); |
|
} |
|
|
|
static void |
|
congreet(int s) |
|
{ |
|
int n = 0, remote_major = 0, remote_minor = 0; |
|
char buf[256], *cp; |
|
char remote_version[sizeof buf]; |
|
size_t bufsiz; |
|
con *c = &fdcon[s]; |
|
|
|
for (;;) { |
|
memset(buf, '\0', sizeof(buf)); |
|
bufsiz = sizeof(buf); |
|
cp = buf; |
|
while (bufsiz-- && |
|
(n = atomicio(read, s, cp, 1)) == 1 && *cp != '\n') { |
|
if (*cp == '\r') |
|
*cp = '\n'; |
|
cp++; |
|
} |
|
if (n != 1 || strncmp(buf, "SSH-", 4) == 0) |
|
break; |
|
} |
|
if (n == 0) { |
|
switch (errno) { |
|
case EPIPE: |
|
error("%s: Connection closed by remote host", c->c_name); |
|
break; |
|
case ECONNREFUSED: |
|
break; |
|
default: |
|
error("read (%s): %s", c->c_name, strerror(errno)); |
|
break; |
|
} |
|
conrecycle(s); |
|
return; |
|
} |
|
if (*cp != '\n' && *cp != '\r') { |
|
error("%s: bad greeting", c->c_name); |
|
confree(s); |
|
return; |
|
} |
|
*cp = '\0'; |
|
if ((c->c_ssh = ssh_packet_set_connection(NULL, s, s)) == NULL) |
|
fatal("ssh_packet_set_connection failed"); |
|
ssh_packet_set_timeout(c->c_ssh, timeout, 1); |
|
ssh_set_app_data(c->c_ssh, c); /* back link */ |
|
if (sscanf(buf, "SSH-%d.%d-%[^\n]\n", |
|
&remote_major, &remote_minor, remote_version) == 3) |
|
c->c_ssh->compat = compat_datafellows(remote_version); |
|
else |
|
c->c_ssh->compat = 0; |
|
if (c->c_keytype != KT_RSA1) { |
|
if (!ssh2_capable(remote_major, remote_minor)) { |
|
debug("%s doesn't support ssh2", c->c_name); |
|
confree(s); |
|
return; |
|
} |
|
} else if (remote_major != 1) { |
|
debug("%s doesn't support ssh1", c->c_name); |
|
confree(s); |
|
return; |
|
} |
|
fprintf(stderr, "# %s:%d %s\n", c->c_name, ssh_port, chop(buf)); |
|
n = snprintf(buf, sizeof buf, "SSH-%d.%d-OpenSSH-keyscan\r\n", |
|
c->c_keytype == KT_RSA1? PROTOCOL_MAJOR_1 : PROTOCOL_MAJOR_2, |
|
c->c_keytype == KT_RSA1? PROTOCOL_MINOR_1 : PROTOCOL_MINOR_2); |
|
if (n < 0 || (size_t)n >= sizeof(buf)) { |
|
error("snprintf: buffer too small"); |
|
confree(s); |
|
return; |
|
} |
|
if (atomicio(vwrite, s, buf, n) != (size_t)n) { |
|
error("write (%s): %s", c->c_name, strerror(errno)); |
|
confree(s); |
|
return; |
|
} |
|
if (c->c_keytype != KT_RSA1) { |
|
keygrab_ssh2(c); |
|
confree(s); |
|
return; |
|
} |
|
c->c_status = CS_SIZE; |
|
contouch(s); |
|
} |
|
|
|
static void |
|
conread(int s) |
|
{ |
|
con *c = &fdcon[s]; |
|
size_t n; |
|
|
|
if (c->c_status == CS_CON) { |
|
congreet(s); |
|
return; |
|
} |
|
n = atomicio(read, s, c->c_data + c->c_off, c->c_len - c->c_off); |
|
if (n == 0) { |
|
error("read (%s): %s", c->c_name, strerror(errno)); |
|
confree(s); |
|
return; |
|
} |
|
c->c_off += n; |
|
|
|
if (c->c_off == c->c_len) |
|
switch (c->c_status) { |
|
case CS_SIZE: |
|
c->c_plen = htonl(c->c_plen); |
|
c->c_len = c->c_plen + 8 - (c->c_plen & 7); |
|
c->c_off = 0; |
|
c->c_data = xmalloc(c->c_len); |
|
c->c_status = CS_KEYS; |
|
break; |
|
#ifdef WITH_SSH1 |
|
case CS_KEYS: |
|
keyprint(c, keygrab_ssh1(c)); |
|
confree(s); |
|
return; |
|
#endif |
|
default: |
|
fatal("conread: invalid status %d", c->c_status); |
|
break; |
|
} |
|
|
|
contouch(s); |
|
} |
|
|
|
static void |
|
conloop(void) |
|
{ |
|
struct timeval seltime, now; |
|
fd_set *r, *e; |
|
con *c; |
|
int i; |
|
|
|
gettimeofday(&now, NULL); |
|
c = TAILQ_FIRST(&tq); |
|
|
|
if (c && (c->c_tv.tv_sec > now.tv_sec || |
|
(c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec > now.tv_usec))) { |
|
seltime = c->c_tv; |
|
seltime.tv_sec -= now.tv_sec; |
|
seltime.tv_usec -= now.tv_usec; |
|
if (seltime.tv_usec < 0) { |
|
seltime.tv_usec += 1000000; |
|
seltime.tv_sec--; |
|
} |
|
} else |
|
timerclear(&seltime); |
|
|
|
r = xcalloc(read_wait_nfdset, sizeof(fd_mask)); |
|
e = xcalloc(read_wait_nfdset, sizeof(fd_mask)); |
|
memcpy(r, read_wait, read_wait_nfdset * sizeof(fd_mask)); |
|
memcpy(e, read_wait, read_wait_nfdset * sizeof(fd_mask)); |
|
|
|
while (select(maxfd, r, NULL, e, &seltime) == -1 && |
|
(errno == EAGAIN || errno == EINTR || errno == EWOULDBLOCK)) |
|
; |
|
|
|
for (i = 0; i < maxfd; i++) { |
|
if (FD_ISSET(i, e)) { |
|
error("%s: exception!", fdcon[i].c_name); |
|
confree(i); |
|
} else if (FD_ISSET(i, r)) |
|
conread(i); |
|
} |
|
free(r); |
|
free(e); |
|
|
|
c = TAILQ_FIRST(&tq); |
|
while (c && (c->c_tv.tv_sec < now.tv_sec || |
|
(c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec < now.tv_usec))) { |
|
int s = c->c_fd; |
|
|
|
c = TAILQ_NEXT(c, c_link); |
|
conrecycle(s); |
|
} |
|
} |
|
|
|
static void |
|
do_host(char *host) |
|
{ |
|
char *name = strnnsep(&host, " \t\n"); |
|
int j; |
|
|
|
if (name == NULL) |
|
return; |
|
for (j = KT_RSA1; j <= KT_ED25519; j *= 2) { |
|
if (get_keytypes & j) { |
|
while (ncon >= MAXCON) |
|
conloop(); |
|
conalloc(name, *host ? host : name, j); |
|
} |
|
} |
|
} |
|
|
|
void |
|
fatal(const char *fmt,...) |
|
{ |
|
va_list args; |
|
|
|
va_start(args, fmt); |
|
do_log(SYSLOG_LEVEL_FATAL, fmt, args); |
|
va_end(args); |
|
exit(255); |
|
} |
|
|
|
static void |
|
usage(void) |
|
{ |
|
fprintf(stderr, |
|
"usage: %s [-46cHv] [-f file] [-p port] [-T timeout] [-t type]\n" |
|
"\t\t [host | addrlist namelist] ...\n", |
|
__progname); |
|
exit(1); |
|
} |
|
|
|
int |
|
main(int argc, char **argv) |
|
{ |
|
int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO; |
|
int opt, fopt_count = 0, j; |
|
char *tname, *cp, line[NI_MAXHOST]; |
|
FILE *fp; |
|
u_long linenum; |
|
|
|
extern int optind; |
|
extern char *optarg; |
|
|
|
ssh_malloc_init(); /* must be called before any mallocs */ |
|
__progname = ssh_get_progname(argv[0]); |
|
seed_rng(); |
|
TAILQ_INIT(&tq); |
|
|
|
/* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */ |
|
sanitise_stdfd(); |
|
|
|
if (argc <= 1) |
|
usage(); |
|
|
|
while ((opt = getopt(argc, argv, "cHv46p:T:t:f:")) != -1) { |
|
switch (opt) { |
|
case 'H': |
|
hash_hosts = 1; |
|
break; |
|
case 'c': |
|
get_cert = 1; |
|
break; |
|
case 'p': |
|
ssh_port = a2port(optarg); |
|
if (ssh_port <= 0) { |
|
fprintf(stderr, "Bad port '%s'\n", optarg); |
|
exit(1); |
|
} |
|
break; |
|
case 'T': |
|
timeout = convtime(optarg); |
|
if (timeout == -1 || timeout == 0) { |
|
fprintf(stderr, "Bad timeout '%s'\n", optarg); |
|
usage(); |
|
} |
|
break; |
|
case 'v': |
|
if (!debug_flag) { |
|
debug_flag = 1; |
|
log_level = SYSLOG_LEVEL_DEBUG1; |
|
} |
|
else if (log_level < SYSLOG_LEVEL_DEBUG3) |
|
log_level++; |
|
else |
|
fatal("Too high debugging level."); |
|
break; |
|
case 'f': |
|
if (strcmp(optarg, "-") == 0) |
|
optarg = NULL; |
|
argv[fopt_count++] = optarg; |
|
break; |
|
case 't': |
|
get_keytypes = 0; |
|
tname = strtok(optarg, ","); |
|
while (tname) { |
|
int type = sshkey_type_from_name(tname); |
|
|
|
switch (type) { |
|
#ifdef WITH_SSH1 |
|
case KEY_RSA1: |
|
get_keytypes |= KT_RSA1; |
|
break; |
|
#endif |
|
case KEY_DSA: |
|
get_keytypes |= KT_DSA; |
|
break; |
|
case KEY_ECDSA: |
|
get_keytypes |= KT_ECDSA; |
|
break; |
|
case KEY_RSA: |
|
get_keytypes |= KT_RSA; |
|
break; |
|
case KEY_ED25519: |
|
get_keytypes |= KT_ED25519; |
|
break; |
|
case KEY_UNSPEC: |
|
default: |
|
fatal("Unknown key type \"%s\"", tname); |
|
} |
|
tname = strtok(NULL, ","); |
|
} |
|
break; |
|
case '4': |
|
IPv4or6 = AF_INET; |
|
break; |
|
case '6': |
|
IPv4or6 = AF_INET6; |
|
break; |
|
case '?': |
|
default: |
|
usage(); |
|
} |
|
} |
|
if (optind == argc && !fopt_count) |
|
usage(); |
|
|
|
log_init("ssh-keyscan", log_level, SYSLOG_FACILITY_USER, 1); |
|
|
|
maxfd = fdlim_get(1); |
|
if (maxfd < 0) |
|
fatal("%s: fdlim_get: bad value", __progname); |
|
if (maxfd > MAXMAXFD) |
|
maxfd = MAXMAXFD; |
|
if (MAXCON <= 0) |
|
fatal("%s: not enough file descriptors", __progname); |
|
if (maxfd > fdlim_get(0)) |
|
fdlim_set(maxfd); |
|
fdcon = xcalloc(maxfd, sizeof(con)); |
|
|
|
read_wait_nfdset = howmany(maxfd, NFDBITS); |
|
read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask)); |
|
|
|
for (j = 0; j < fopt_count; j++) { |
|
if (argv[j] == NULL) |
|
fp = stdin; |
|
else if ((fp = fopen(argv[j], "r")) == NULL) |
|
fatal("%s: %s: %s", __progname, argv[j], |
|
strerror(errno)); |
|
linenum = 0; |
|
|
|
while (read_keyfile_line(fp, |
|
argv[j] == NULL ? "(stdin)" : argv[j], line, sizeof(line), |
|
&linenum) != -1) { |
|
/* Chomp off trailing whitespace and comments */ |
|
if ((cp = strchr(line, '#')) == NULL) |
|
cp = line + strlen(line) - 1; |
|
while (cp >= line) { |
|
if (*cp == ' ' || *cp == '\t' || |
|
*cp == '\n' || *cp == '#') |
|
*cp-- = '\0'; |
|
else |
|
break; |
|
} |
|
|
|
/* Skip empty lines */ |
|
if (*line == '\0') |
|
continue; |
|
|
|
do_host(line); |
|
} |
|
|
|
if (ferror(fp)) |
|
fatal("%s: %s: %s", __progname, argv[j], |
|
strerror(errno)); |
|
|
|
fclose(fp); |
|
} |
|
|
|
while (optind < argc) |
|
do_host(argv[optind++]); |
|
|
|
while (ncon > 0) |
|
conloop(); |
|
|
|
return (0); |
|
}
|
|
|