| /* $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); |
| } |