1*d93a896eSDag-Erling Smørgrav /* $OpenBSD: ssh-keyscan.c,v 1.109 2017/03/10 04:26:06 djm Exp $ */ 21e8db6e2SBrian Feldman /* 31e8db6e2SBrian Feldman * Copyright 1995, 1996 by David Mazieres <dm@lcs.mit.edu>. 41e8db6e2SBrian Feldman * 51e8db6e2SBrian Feldman * Modification and redistribution in source and binary forms is 61e8db6e2SBrian Feldman * permitted provided that due credit is given to the author and the 7ae1f160dSDag-Erling Smørgrav * OpenBSD project by leaving this copyright notice intact. 81e8db6e2SBrian Feldman */ 91e8db6e2SBrian Feldman 101e8db6e2SBrian Feldman #include "includes.h" 111e8db6e2SBrian Feldman 12bc5531deSDag-Erling Smørgrav #include <sys/types.h> 13f388f5efSDag-Erling Smørgrav #include "openbsd-compat/sys-queue.h" 14333ee039SDag-Erling Smørgrav #include <sys/resource.h> 15333ee039SDag-Erling Smørgrav #ifdef HAVE_SYS_TIME_H 16333ee039SDag-Erling Smørgrav # include <sys/time.h> 17333ee039SDag-Erling Smørgrav #endif 18333ee039SDag-Erling Smørgrav 19333ee039SDag-Erling Smørgrav #include <netinet/in.h> 20333ee039SDag-Erling Smørgrav #include <arpa/inet.h> 211e8db6e2SBrian Feldman 221e8db6e2SBrian Feldman #include <openssl/bn.h> 231e8db6e2SBrian Feldman 24333ee039SDag-Erling Smørgrav #include <netdb.h> 25333ee039SDag-Erling Smørgrav #include <errno.h> 26333ee039SDag-Erling Smørgrav #include <stdarg.h> 27333ee039SDag-Erling Smørgrav #include <stdio.h> 28333ee039SDag-Erling Smørgrav #include <stdlib.h> 29333ee039SDag-Erling Smørgrav #include <signal.h> 30333ee039SDag-Erling Smørgrav #include <string.h> 31333ee039SDag-Erling Smørgrav #include <unistd.h> 32333ee039SDag-Erling Smørgrav 331e8db6e2SBrian Feldman #include "xmalloc.h" 341e8db6e2SBrian Feldman #include "ssh.h" 351e8db6e2SBrian Feldman #include "ssh1.h" 36bc5531deSDag-Erling Smørgrav #include "sshbuf.h" 37bc5531deSDag-Erling Smørgrav #include "sshkey.h" 38333ee039SDag-Erling Smørgrav #include "cipher.h" 39ae1f160dSDag-Erling Smørgrav #include "kex.h" 40ae1f160dSDag-Erling Smørgrav #include "compat.h" 41ae1f160dSDag-Erling Smørgrav #include "myproposal.h" 42ae1f160dSDag-Erling Smørgrav #include "packet.h" 43ae1f160dSDag-Erling Smørgrav #include "dispatch.h" 441e8db6e2SBrian Feldman #include "log.h" 451e8db6e2SBrian Feldman #include "atomicio.h" 46ae1f160dSDag-Erling Smørgrav #include "misc.h" 47aa49c926SDag-Erling Smørgrav #include "hostfile.h" 48bc5531deSDag-Erling Smørgrav #include "ssherr.h" 49bc5531deSDag-Erling Smørgrav #include "ssh_api.h" 501e8db6e2SBrian Feldman 51989dd127SDag-Erling Smørgrav /* Flag indicating whether IPv4 or IPv6. This can be set on the command line. 52989dd127SDag-Erling Smørgrav Default value is AF_UNSPEC means both IPv4 and IPv6. */ 53989dd127SDag-Erling Smørgrav int IPv4or6 = AF_UNSPEC; 541e8db6e2SBrian Feldman 55ae1f160dSDag-Erling Smørgrav int ssh_port = SSH_DEFAULT_PORT; 56ae1f160dSDag-Erling Smørgrav 57ae1f160dSDag-Erling Smørgrav #define KT_RSA1 1 58ae1f160dSDag-Erling Smørgrav #define KT_DSA 2 59ae1f160dSDag-Erling Smørgrav #define KT_RSA 4 604a421b63SDag-Erling Smørgrav #define KT_ECDSA 8 61f7167e0eSDag-Erling Smørgrav #define KT_ED25519 16 62ae1f160dSDag-Erling Smørgrav 63acc1a9efSDag-Erling Smørgrav int get_cert = 0; 64a0ee8cc6SDag-Erling Smørgrav int get_keytypes = KT_RSA|KT_ECDSA|KT_ED25519; 651e8db6e2SBrian Feldman 66aa49c926SDag-Erling Smørgrav int hash_hosts = 0; /* Hash hostname on output */ 67aa49c926SDag-Erling Smørgrav 681e8db6e2SBrian Feldman #define MAXMAXFD 256 691e8db6e2SBrian Feldman 701e8db6e2SBrian Feldman /* The number of seconds after which to give up on a TCP connection */ 711e8db6e2SBrian Feldman int timeout = 5; 721e8db6e2SBrian Feldman 731e8db6e2SBrian Feldman int maxfd; 741e8db6e2SBrian Feldman #define MAXCON (maxfd - 10) 751e8db6e2SBrian Feldman 761e8db6e2SBrian Feldman extern char *__progname; 771e8db6e2SBrian Feldman fd_set *read_wait; 78333ee039SDag-Erling Smørgrav size_t read_wait_nfdset; 791e8db6e2SBrian Feldman int ncon; 80bc5531deSDag-Erling Smørgrav 81bc5531deSDag-Erling Smørgrav struct ssh *active_state = NULL; /* XXX needed for linking */ 821e8db6e2SBrian Feldman 831e8db6e2SBrian Feldman /* 841e8db6e2SBrian Feldman * Keep a connection structure for each file descriptor. The state 851e8db6e2SBrian Feldman * associated with file descriptor n is held in fdcon[n]. 861e8db6e2SBrian Feldman */ 871e8db6e2SBrian Feldman typedef struct Connection { 881e8db6e2SBrian Feldman u_char c_status; /* State of connection on this file desc. */ 891e8db6e2SBrian Feldman #define CS_UNUSED 0 /* File descriptor unused */ 901e8db6e2SBrian Feldman #define CS_CON 1 /* Waiting to connect/read greeting */ 911e8db6e2SBrian Feldman #define CS_SIZE 2 /* Waiting to read initial packet size */ 921e8db6e2SBrian Feldman #define CS_KEYS 3 /* Waiting to read public key packet */ 931e8db6e2SBrian Feldman int c_fd; /* Quick lookup: c->c_fd == c - fdcon */ 941e8db6e2SBrian Feldman int c_plen; /* Packet length field for ssh packet */ 951e8db6e2SBrian Feldman int c_len; /* Total bytes which must be read. */ 961e8db6e2SBrian Feldman int c_off; /* Length of data read so far. */ 97ae1f160dSDag-Erling Smørgrav int c_keytype; /* Only one of KT_RSA1, KT_DSA, or KT_RSA */ 98557f75e5SDag-Erling Smørgrav sig_atomic_t c_done; /* SSH2 done */ 991e8db6e2SBrian Feldman char *c_namebase; /* Address to free for c_name and c_namelist */ 1001e8db6e2SBrian Feldman char *c_name; /* Hostname of connection for errors */ 1011e8db6e2SBrian Feldman char *c_namelist; /* Pointer to other possible addresses */ 1021e8db6e2SBrian Feldman char *c_output_name; /* Hostname of connection for output */ 1031e8db6e2SBrian Feldman char *c_data; /* Data read from this fd */ 104bc5531deSDag-Erling Smørgrav struct ssh *c_ssh; /* SSH-connection */ 1051e8db6e2SBrian Feldman struct timeval c_tv; /* Time at which connection gets aborted */ 1061e8db6e2SBrian Feldman TAILQ_ENTRY(Connection) c_link; /* List of connections in timeout order. */ 1071e8db6e2SBrian Feldman } con; 1081e8db6e2SBrian Feldman 1091e8db6e2SBrian Feldman TAILQ_HEAD(conlist, Connection) tq; /* Timeout Queue */ 1101e8db6e2SBrian Feldman con *fdcon; 1111e8db6e2SBrian Feldman 112bc5531deSDag-Erling Smørgrav static void keyprint(con *c, struct sshkey *key); 113bc5531deSDag-Erling Smørgrav 114ae1f160dSDag-Erling Smørgrav static int 1151e8db6e2SBrian Feldman fdlim_get(int hard) 1161e8db6e2SBrian Feldman { 117989dd127SDag-Erling Smørgrav #if defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE) 1181e8db6e2SBrian Feldman struct rlimit rlfd; 1191e8db6e2SBrian Feldman 1201e8db6e2SBrian Feldman if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0) 1211e8db6e2SBrian Feldman return (-1); 1221e8db6e2SBrian Feldman if ((hard ? rlfd.rlim_max : rlfd.rlim_cur) == RLIM_INFINITY) 1231ec0d754SDag-Erling Smørgrav return SSH_SYSFDMAX; 1241e8db6e2SBrian Feldman else 1251e8db6e2SBrian Feldman return hard ? rlfd.rlim_max : rlfd.rlim_cur; 126989dd127SDag-Erling Smørgrav #else 1271ec0d754SDag-Erling Smørgrav return SSH_SYSFDMAX; 128989dd127SDag-Erling Smørgrav #endif 1291e8db6e2SBrian Feldman } 1301e8db6e2SBrian Feldman 131ae1f160dSDag-Erling Smørgrav static int 1321e8db6e2SBrian Feldman fdlim_set(int lim) 1331e8db6e2SBrian Feldman { 134989dd127SDag-Erling Smørgrav #if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE) 1351e8db6e2SBrian Feldman struct rlimit rlfd; 136989dd127SDag-Erling Smørgrav #endif 137f388f5efSDag-Erling Smørgrav 1381e8db6e2SBrian Feldman if (lim <= 0) 1391e8db6e2SBrian Feldman return (-1); 140989dd127SDag-Erling Smørgrav #if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE) 1411e8db6e2SBrian Feldman if (getrlimit(RLIMIT_NOFILE, &rlfd) < 0) 1421e8db6e2SBrian Feldman return (-1); 1431e8db6e2SBrian Feldman rlfd.rlim_cur = lim; 1441e8db6e2SBrian Feldman if (setrlimit(RLIMIT_NOFILE, &rlfd) < 0) 1451e8db6e2SBrian Feldman return (-1); 146989dd127SDag-Erling Smørgrav #elif defined (HAVE_SETDTABLESIZE) 147989dd127SDag-Erling Smørgrav setdtablesize(lim); 148989dd127SDag-Erling Smørgrav #endif 1491e8db6e2SBrian Feldman return (0); 1501e8db6e2SBrian Feldman } 1511e8db6e2SBrian Feldman 1521e8db6e2SBrian Feldman /* 1531e8db6e2SBrian Feldman * This is an strsep function that returns a null field for adjacent 1541e8db6e2SBrian Feldman * separators. This is the same as the 4.4BSD strsep, but different from the 1551e8db6e2SBrian Feldman * one in the GNU libc. 1561e8db6e2SBrian Feldman */ 157ae1f160dSDag-Erling Smørgrav static char * 1581e8db6e2SBrian Feldman xstrsep(char **str, const char *delim) 1591e8db6e2SBrian Feldman { 1601e8db6e2SBrian Feldman char *s, *e; 1611e8db6e2SBrian Feldman 1621e8db6e2SBrian Feldman if (!**str) 1631e8db6e2SBrian Feldman return (NULL); 1641e8db6e2SBrian Feldman 1651e8db6e2SBrian Feldman s = *str; 1661e8db6e2SBrian Feldman e = s + strcspn(s, delim); 1671e8db6e2SBrian Feldman 1681e8db6e2SBrian Feldman if (*e != '\0') 1691e8db6e2SBrian Feldman *e++ = '\0'; 1701e8db6e2SBrian Feldman *str = e; 1711e8db6e2SBrian Feldman 1721e8db6e2SBrian Feldman return (s); 1731e8db6e2SBrian Feldman } 1741e8db6e2SBrian Feldman 1751e8db6e2SBrian Feldman /* 1761e8db6e2SBrian Feldman * Get the next non-null token (like GNU strsep). Strsep() will return a 1771e8db6e2SBrian Feldman * null token for two adjacent separators, so we may have to loop. 1781e8db6e2SBrian Feldman */ 179ae1f160dSDag-Erling Smørgrav static char * 1801e8db6e2SBrian Feldman strnnsep(char **stringp, char *delim) 1811e8db6e2SBrian Feldman { 1821e8db6e2SBrian Feldman char *tok; 1831e8db6e2SBrian Feldman 1841e8db6e2SBrian Feldman do { 1851e8db6e2SBrian Feldman tok = xstrsep(stringp, delim); 1861e8db6e2SBrian Feldman } while (tok && *tok == '\0'); 1871e8db6e2SBrian Feldman return (tok); 1881e8db6e2SBrian Feldman } 1891e8db6e2SBrian Feldman 190a0ee8cc6SDag-Erling Smørgrav #ifdef WITH_SSH1 191bc5531deSDag-Erling Smørgrav static struct sshkey * 192ae1f160dSDag-Erling Smørgrav keygrab_ssh1(con *c) 1931e8db6e2SBrian Feldman { 194bc5531deSDag-Erling Smørgrav static struct sshkey *rsa; 195bc5531deSDag-Erling Smørgrav static struct sshbuf *msg; 196bc5531deSDag-Erling Smørgrav int r; 197bc5531deSDag-Erling Smørgrav u_char type; 1981e8db6e2SBrian Feldman 1991e8db6e2SBrian Feldman if (rsa == NULL) { 200bc5531deSDag-Erling Smørgrav if ((rsa = sshkey_new(KEY_RSA1)) == NULL) { 201bc5531deSDag-Erling Smørgrav error("%s: sshkey_new failed", __func__); 202ae1f160dSDag-Erling Smørgrav return NULL; 2031e8db6e2SBrian Feldman } 204bc5531deSDag-Erling Smørgrav if ((msg = sshbuf_new()) == NULL) 205bc5531deSDag-Erling Smørgrav fatal("%s: sshbuf_new failed", __func__); 206bc5531deSDag-Erling Smørgrav } 207bc5531deSDag-Erling Smørgrav if ((r = sshbuf_put(msg, c->c_data, c->c_plen)) != 0 || 208bc5531deSDag-Erling Smørgrav (r = sshbuf_consume(msg, 8 - (c->c_plen & 7))) != 0 || /* padding */ 209bc5531deSDag-Erling Smørgrav (r = sshbuf_get_u8(msg, &type)) != 0) 210bc5531deSDag-Erling Smørgrav goto buf_err; 211bc5531deSDag-Erling Smørgrav if (type != (int) SSH_SMSG_PUBLIC_KEY) { 212bc5531deSDag-Erling Smørgrav error("%s: invalid packet type", c->c_name); 213bc5531deSDag-Erling Smørgrav sshbuf_reset(msg); 214bc5531deSDag-Erling Smørgrav return NULL; 215bc5531deSDag-Erling Smørgrav } 216bc5531deSDag-Erling Smørgrav if ((r = sshbuf_consume(msg, 8)) != 0 || /* cookie */ 2171e8db6e2SBrian Feldman /* server key */ 218bc5531deSDag-Erling Smørgrav (r = sshbuf_get_u32(msg, NULL)) != 0 || 219bc5531deSDag-Erling Smørgrav (r = sshbuf_get_bignum1(msg, NULL)) != 0 || 220bc5531deSDag-Erling Smørgrav (r = sshbuf_get_bignum1(msg, NULL)) != 0 || 2211e8db6e2SBrian Feldman /* host key */ 222bc5531deSDag-Erling Smørgrav (r = sshbuf_get_u32(msg, NULL)) != 0 || 223bc5531deSDag-Erling Smørgrav (r = sshbuf_get_bignum1(msg, rsa->rsa->e)) != 0 || 224bc5531deSDag-Erling Smørgrav (r = sshbuf_get_bignum1(msg, rsa->rsa->n)) != 0) { 225bc5531deSDag-Erling Smørgrav buf_err: 226bc5531deSDag-Erling Smørgrav error("%s: buffer error: %s", __func__, ssh_err(r)); 227bc5531deSDag-Erling Smørgrav sshbuf_reset(msg); 228bc5531deSDag-Erling Smørgrav return NULL; 229bc5531deSDag-Erling Smørgrav } 230ae1f160dSDag-Erling Smørgrav 231bc5531deSDag-Erling Smørgrav sshbuf_reset(msg); 2321e8db6e2SBrian Feldman 233ae1f160dSDag-Erling Smørgrav return (rsa); 234ae1f160dSDag-Erling Smørgrav } 235a0ee8cc6SDag-Erling Smørgrav #endif 236ae1f160dSDag-Erling Smørgrav 237ae1f160dSDag-Erling Smørgrav static int 238bc5531deSDag-Erling Smørgrav key_print_wrapper(struct sshkey *hostkey, struct ssh *ssh) 239ae1f160dSDag-Erling Smørgrav { 240bc5531deSDag-Erling Smørgrav con *c; 241bc5531deSDag-Erling Smørgrav 242bc5531deSDag-Erling Smørgrav if ((c = ssh_get_app_data(ssh)) != NULL) 243bc5531deSDag-Erling Smørgrav keyprint(c, hostkey); 244bc5531deSDag-Erling Smørgrav /* always abort key exchange */ 245bc5531deSDag-Erling Smørgrav return -1; 246ae1f160dSDag-Erling Smørgrav } 247ae1f160dSDag-Erling Smørgrav 248ae1f160dSDag-Erling Smørgrav static int 249ae1f160dSDag-Erling Smørgrav ssh2_capable(int remote_major, int remote_minor) 250ae1f160dSDag-Erling Smørgrav { 251ae1f160dSDag-Erling Smørgrav switch (remote_major) { 252ae1f160dSDag-Erling Smørgrav case 1: 253ae1f160dSDag-Erling Smørgrav if (remote_minor == 99) 254ae1f160dSDag-Erling Smørgrav return 1; 255ae1f160dSDag-Erling Smørgrav break; 256ae1f160dSDag-Erling Smørgrav case 2: 257ae1f160dSDag-Erling Smørgrav return 1; 258ae1f160dSDag-Erling Smørgrav default: 259ae1f160dSDag-Erling Smørgrav break; 260ae1f160dSDag-Erling Smørgrav } 261ae1f160dSDag-Erling Smørgrav return 0; 262ae1f160dSDag-Erling Smørgrav } 263ae1f160dSDag-Erling Smørgrav 264bc5531deSDag-Erling Smørgrav static void 265ae1f160dSDag-Erling Smørgrav keygrab_ssh2(con *c) 266ae1f160dSDag-Erling Smørgrav { 267a0ee8cc6SDag-Erling Smørgrav char *myproposal[PROPOSAL_MAX] = { KEX_CLIENT }; 268bc5531deSDag-Erling Smørgrav int r; 269ae1f160dSDag-Erling Smørgrav 270ae1f160dSDag-Erling Smørgrav enable_compat20(); 271acc1a9efSDag-Erling Smørgrav switch (c->c_keytype) { 272acc1a9efSDag-Erling Smørgrav case KT_DSA: 273acc1a9efSDag-Erling Smørgrav myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? 274acc1a9efSDag-Erling Smørgrav "ssh-dss-cert-v01@openssh.com" : "ssh-dss"; 275acc1a9efSDag-Erling Smørgrav break; 276acc1a9efSDag-Erling Smørgrav case KT_RSA: 277acc1a9efSDag-Erling Smørgrav myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? 278acc1a9efSDag-Erling Smørgrav "ssh-rsa-cert-v01@openssh.com" : "ssh-rsa"; 279acc1a9efSDag-Erling Smørgrav break; 280acc1a9efSDag-Erling Smørgrav case KT_ED25519: 281acc1a9efSDag-Erling Smørgrav myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? 282acc1a9efSDag-Erling Smørgrav "ssh-ed25519-cert-v01@openssh.com" : "ssh-ed25519"; 283acc1a9efSDag-Erling Smørgrav break; 284acc1a9efSDag-Erling Smørgrav case KT_ECDSA: 285acc1a9efSDag-Erling Smørgrav myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ? 286acc1a9efSDag-Erling Smørgrav "ecdsa-sha2-nistp256-cert-v01@openssh.com," 287acc1a9efSDag-Erling Smørgrav "ecdsa-sha2-nistp384-cert-v01@openssh.com," 288acc1a9efSDag-Erling Smørgrav "ecdsa-sha2-nistp521-cert-v01@openssh.com" : 289acc1a9efSDag-Erling Smørgrav "ecdsa-sha2-nistp256," 290acc1a9efSDag-Erling Smørgrav "ecdsa-sha2-nistp384," 291acc1a9efSDag-Erling Smørgrav "ecdsa-sha2-nistp521"; 292acc1a9efSDag-Erling Smørgrav break; 293acc1a9efSDag-Erling Smørgrav default: 294acc1a9efSDag-Erling Smørgrav fatal("unknown key type %d", c->c_keytype); 295acc1a9efSDag-Erling Smørgrav break; 296acc1a9efSDag-Erling Smørgrav } 297bc5531deSDag-Erling Smørgrav if ((r = kex_setup(c->c_ssh, myproposal)) != 0) { 298bc5531deSDag-Erling Smørgrav free(c->c_ssh); 299bc5531deSDag-Erling Smørgrav fprintf(stderr, "kex_setup: %s\n", ssh_err(r)); 300ae1f160dSDag-Erling Smørgrav exit(1); 301ae1f160dSDag-Erling Smørgrav } 302bc5531deSDag-Erling Smørgrav #ifdef WITH_OPENSSL 303bc5531deSDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_DH_GRP1_SHA1] = kexdh_client; 304bc5531deSDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_DH_GRP14_SHA1] = kexdh_client; 305076ad2f8SDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_DH_GRP14_SHA256] = kexdh_client; 306076ad2f8SDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_DH_GRP16_SHA512] = kexdh_client; 307076ad2f8SDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_DH_GRP18_SHA512] = kexdh_client; 308bc5531deSDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_DH_GEX_SHA1] = kexgex_client; 309bc5531deSDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_DH_GEX_SHA256] = kexgex_client; 310bc5531deSDag-Erling Smørgrav # ifdef OPENSSL_HAS_ECC 311bc5531deSDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_ECDH_SHA2] = kexecdh_client; 312bc5531deSDag-Erling Smørgrav # endif 313bc5531deSDag-Erling Smørgrav #endif 314bc5531deSDag-Erling Smørgrav c->c_ssh->kex->kex[KEX_C25519_SHA256] = kexc25519_client; 315bc5531deSDag-Erling Smørgrav ssh_set_verify_host_key_callback(c->c_ssh, key_print_wrapper); 316bc5531deSDag-Erling Smørgrav /* 317bc5531deSDag-Erling Smørgrav * do the key-exchange until an error occurs or until 318bc5531deSDag-Erling Smørgrav * the key_print_wrapper() callback sets c_done. 319bc5531deSDag-Erling Smørgrav */ 320bc5531deSDag-Erling Smørgrav ssh_dispatch_run(c->c_ssh, DISPATCH_BLOCK, &c->c_done, c->c_ssh); 321ae1f160dSDag-Erling Smørgrav } 322ae1f160dSDag-Erling Smørgrav 323ae1f160dSDag-Erling Smørgrav static void 324*d93a896eSDag-Erling Smørgrav keyprint_one(const char *host, struct sshkey *key) 325ae1f160dSDag-Erling Smørgrav { 326acc1a9efSDag-Erling Smørgrav char *hostport; 327*d93a896eSDag-Erling Smørgrav const char *known_host, *hashed; 328ae1f160dSDag-Erling Smørgrav 329557f75e5SDag-Erling Smørgrav hostport = put_host_port(host, ssh_port); 330*d93a896eSDag-Erling Smørgrav lowercase(hostport); 331*d93a896eSDag-Erling Smørgrav if (hash_hosts && (hashed = host_hash(host, NULL, 0)) == NULL) 332*d93a896eSDag-Erling Smørgrav fatal("host_hash failed"); 333*d93a896eSDag-Erling Smørgrav known_host = hash_hosts ? hashed : hostport; 334acc1a9efSDag-Erling Smørgrav if (!get_cert) 335*d93a896eSDag-Erling Smørgrav fprintf(stdout, "%s ", known_host); 336bc5531deSDag-Erling Smørgrav sshkey_write(key, stdout); 3371e8db6e2SBrian Feldman fputs("\n", stdout); 338557f75e5SDag-Erling Smørgrav free(hostport); 3391e8db6e2SBrian Feldman } 3401e8db6e2SBrian Feldman 341acc1a9efSDag-Erling Smørgrav static void 342acc1a9efSDag-Erling Smørgrav keyprint(con *c, struct sshkey *key) 343acc1a9efSDag-Erling Smørgrav { 344acc1a9efSDag-Erling Smørgrav char *hosts = c->c_output_name ? c->c_output_name : c->c_name; 345acc1a9efSDag-Erling Smørgrav char *host, *ohosts; 346acc1a9efSDag-Erling Smørgrav 347acc1a9efSDag-Erling Smørgrav if (key == NULL) 348acc1a9efSDag-Erling Smørgrav return; 349acc1a9efSDag-Erling Smørgrav if (get_cert || (!hash_hosts && ssh_port == SSH_DEFAULT_PORT)) { 350acc1a9efSDag-Erling Smørgrav keyprint_one(hosts, key); 351acc1a9efSDag-Erling Smørgrav return; 352acc1a9efSDag-Erling Smørgrav } 353acc1a9efSDag-Erling Smørgrav ohosts = hosts = xstrdup(hosts); 354acc1a9efSDag-Erling Smørgrav while ((host = strsep(&hosts, ",")) != NULL) 355acc1a9efSDag-Erling Smørgrav keyprint_one(host, key); 356acc1a9efSDag-Erling Smørgrav free(ohosts); 357acc1a9efSDag-Erling Smørgrav } 358acc1a9efSDag-Erling Smørgrav 359ae1f160dSDag-Erling Smørgrav static int 3601e8db6e2SBrian Feldman tcpconnect(char *host) 3611e8db6e2SBrian Feldman { 3621e8db6e2SBrian Feldman struct addrinfo hints, *ai, *aitop; 3631e8db6e2SBrian Feldman char strport[NI_MAXSERV]; 3641e8db6e2SBrian Feldman int gaierr, s = -1; 3651e8db6e2SBrian Feldman 366ae1f160dSDag-Erling Smørgrav snprintf(strport, sizeof strport, "%d", ssh_port); 3671e8db6e2SBrian Feldman memset(&hints, 0, sizeof(hints)); 368ae1f160dSDag-Erling Smørgrav hints.ai_family = IPv4or6; 3691e8db6e2SBrian Feldman hints.ai_socktype = SOCK_STREAM; 370bc5531deSDag-Erling Smørgrav if ((gaierr = getaddrinfo(host, strport, &hints, &aitop)) != 0) { 371bc5531deSDag-Erling Smørgrav error("getaddrinfo %s: %s", host, ssh_gai_strerror(gaierr)); 372bc5531deSDag-Erling Smørgrav return -1; 373bc5531deSDag-Erling Smørgrav } 3741e8db6e2SBrian Feldman for (ai = aitop; ai; ai = ai->ai_next) { 375cf2b5f3bSDag-Erling Smørgrav s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); 3761e8db6e2SBrian Feldman if (s < 0) { 3771e8db6e2SBrian Feldman error("socket: %s", strerror(errno)); 3781e8db6e2SBrian Feldman continue; 3791e8db6e2SBrian Feldman } 38021e764dfSDag-Erling Smørgrav if (set_nonblock(s) == -1) 38121e764dfSDag-Erling Smørgrav fatal("%s: set_nonblock(%d)", __func__, s); 3821e8db6e2SBrian Feldman if (connect(s, ai->ai_addr, ai->ai_addrlen) < 0 && 3831e8db6e2SBrian Feldman errno != EINPROGRESS) 3841e8db6e2SBrian Feldman error("connect (`%s'): %s", host, strerror(errno)); 3851e8db6e2SBrian Feldman else 3861e8db6e2SBrian Feldman break; 3871e8db6e2SBrian Feldman close(s); 3881e8db6e2SBrian Feldman s = -1; 3891e8db6e2SBrian Feldman } 3901e8db6e2SBrian Feldman freeaddrinfo(aitop); 3911e8db6e2SBrian Feldman return s; 3921e8db6e2SBrian Feldman } 3931e8db6e2SBrian Feldman 394ae1f160dSDag-Erling Smørgrav static int 395ae1f160dSDag-Erling Smørgrav conalloc(char *iname, char *oname, int keytype) 3961e8db6e2SBrian Feldman { 3971e8db6e2SBrian Feldman char *namebase, *name, *namelist; 398f388f5efSDag-Erling Smørgrav int s; 3991e8db6e2SBrian Feldman 4001e8db6e2SBrian Feldman namebase = namelist = xstrdup(iname); 4011e8db6e2SBrian Feldman 4021e8db6e2SBrian Feldman do { 4031e8db6e2SBrian Feldman name = xstrsep(&namelist, ","); 4041e8db6e2SBrian Feldman if (!name) { 405e4a9863fSDag-Erling Smørgrav free(namebase); 4061e8db6e2SBrian Feldman return (-1); 4071e8db6e2SBrian Feldman } 4081e8db6e2SBrian Feldman } while ((s = tcpconnect(name)) < 0); 4091e8db6e2SBrian Feldman 4101e8db6e2SBrian Feldman if (s >= maxfd) 4111e8db6e2SBrian Feldman fatal("conalloc: fdno %d too high", s); 4121e8db6e2SBrian Feldman if (fdcon[s].c_status) 4131e8db6e2SBrian Feldman fatal("conalloc: attempt to reuse fdno %d", s); 4141e8db6e2SBrian Feldman 415acc1a9efSDag-Erling Smørgrav debug3("%s: oname %s kt %d", __func__, oname, keytype); 4161e8db6e2SBrian Feldman fdcon[s].c_fd = s; 4171e8db6e2SBrian Feldman fdcon[s].c_status = CS_CON; 4181e8db6e2SBrian Feldman fdcon[s].c_namebase = namebase; 4191e8db6e2SBrian Feldman fdcon[s].c_name = name; 4201e8db6e2SBrian Feldman fdcon[s].c_namelist = namelist; 4211e8db6e2SBrian Feldman fdcon[s].c_output_name = xstrdup(oname); 4221e8db6e2SBrian Feldman fdcon[s].c_data = (char *) &fdcon[s].c_plen; 4231e8db6e2SBrian Feldman fdcon[s].c_len = 4; 4241e8db6e2SBrian Feldman fdcon[s].c_off = 0; 425ae1f160dSDag-Erling Smørgrav fdcon[s].c_keytype = keytype; 4261e8db6e2SBrian Feldman gettimeofday(&fdcon[s].c_tv, NULL); 4271e8db6e2SBrian Feldman fdcon[s].c_tv.tv_sec += timeout; 4281e8db6e2SBrian Feldman TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link); 4291e8db6e2SBrian Feldman FD_SET(s, read_wait); 4301e8db6e2SBrian Feldman ncon++; 4311e8db6e2SBrian Feldman return (s); 4321e8db6e2SBrian Feldman } 4331e8db6e2SBrian Feldman 434ae1f160dSDag-Erling Smørgrav static void 4351e8db6e2SBrian Feldman confree(int s) 4361e8db6e2SBrian Feldman { 4371e8db6e2SBrian Feldman if (s >= maxfd || fdcon[s].c_status == CS_UNUSED) 4381e8db6e2SBrian Feldman fatal("confree: attempt to free bad fdno %d", s); 4391e8db6e2SBrian Feldman close(s); 440e4a9863fSDag-Erling Smørgrav free(fdcon[s].c_namebase); 441e4a9863fSDag-Erling Smørgrav free(fdcon[s].c_output_name); 4421e8db6e2SBrian Feldman if (fdcon[s].c_status == CS_KEYS) 443e4a9863fSDag-Erling Smørgrav free(fdcon[s].c_data); 4441e8db6e2SBrian Feldman fdcon[s].c_status = CS_UNUSED; 445ae1f160dSDag-Erling Smørgrav fdcon[s].c_keytype = 0; 446bc5531deSDag-Erling Smørgrav if (fdcon[s].c_ssh) { 447bc5531deSDag-Erling Smørgrav ssh_packet_close(fdcon[s].c_ssh); 448bc5531deSDag-Erling Smørgrav free(fdcon[s].c_ssh); 449bc5531deSDag-Erling Smørgrav fdcon[s].c_ssh = NULL; 450bc5531deSDag-Erling Smørgrav } 4511e8db6e2SBrian Feldman TAILQ_REMOVE(&tq, &fdcon[s], c_link); 4521e8db6e2SBrian Feldman FD_CLR(s, read_wait); 4531e8db6e2SBrian Feldman ncon--; 4541e8db6e2SBrian Feldman } 4551e8db6e2SBrian Feldman 456ae1f160dSDag-Erling Smørgrav static void 4571e8db6e2SBrian Feldman contouch(int s) 4581e8db6e2SBrian Feldman { 4591e8db6e2SBrian Feldman TAILQ_REMOVE(&tq, &fdcon[s], c_link); 4601e8db6e2SBrian Feldman gettimeofday(&fdcon[s].c_tv, NULL); 4611e8db6e2SBrian Feldman fdcon[s].c_tv.tv_sec += timeout; 4621e8db6e2SBrian Feldman TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link); 4631e8db6e2SBrian Feldman } 4641e8db6e2SBrian Feldman 465ae1f160dSDag-Erling Smørgrav static int 4661e8db6e2SBrian Feldman conrecycle(int s) 4671e8db6e2SBrian Feldman { 4681e8db6e2SBrian Feldman con *c = &fdcon[s]; 469f388f5efSDag-Erling Smørgrav int ret; 4701e8db6e2SBrian Feldman 471ae1f160dSDag-Erling Smørgrav ret = conalloc(c->c_namelist, c->c_output_name, c->c_keytype); 4721e8db6e2SBrian Feldman confree(s); 4731e8db6e2SBrian Feldman return (ret); 4741e8db6e2SBrian Feldman } 4751e8db6e2SBrian Feldman 476ae1f160dSDag-Erling Smørgrav static void 4771e8db6e2SBrian Feldman congreet(int s) 4781e8db6e2SBrian Feldman { 479d4ecd108SDag-Erling Smørgrav int n = 0, remote_major = 0, remote_minor = 0; 480ae1f160dSDag-Erling Smørgrav char buf[256], *cp; 481ae1f160dSDag-Erling Smørgrav char remote_version[sizeof buf]; 4821e8db6e2SBrian Feldman size_t bufsiz; 4831e8db6e2SBrian Feldman con *c = &fdcon[s]; 4841e8db6e2SBrian Feldman 485b74df5b2SDag-Erling Smørgrav for (;;) { 486b74df5b2SDag-Erling Smørgrav memset(buf, '\0', sizeof(buf)); 4871e8db6e2SBrian Feldman bufsiz = sizeof(buf); 4881e8db6e2SBrian Feldman cp = buf; 489b74df5b2SDag-Erling Smørgrav while (bufsiz-- && 490b74df5b2SDag-Erling Smørgrav (n = atomicio(read, s, cp, 1)) == 1 && *cp != '\n') { 491ae1f160dSDag-Erling Smørgrav if (*cp == '\r') 492ae1f160dSDag-Erling Smørgrav *cp = '\n'; 4931e8db6e2SBrian Feldman cp++; 494ae1f160dSDag-Erling Smørgrav } 495b74df5b2SDag-Erling Smørgrav if (n != 1 || strncmp(buf, "SSH-", 4) == 0) 496b74df5b2SDag-Erling Smørgrav break; 497b74df5b2SDag-Erling Smørgrav } 498ae1f160dSDag-Erling Smørgrav if (n == 0) { 499d4ecd108SDag-Erling Smørgrav switch (errno) { 500d4ecd108SDag-Erling Smørgrav case EPIPE: 501ae1f160dSDag-Erling Smørgrav error("%s: Connection closed by remote host", c->c_name); 502d4ecd108SDag-Erling Smørgrav break; 503d4ecd108SDag-Erling Smørgrav case ECONNREFUSED: 504d4ecd108SDag-Erling Smørgrav break; 505d4ecd108SDag-Erling Smørgrav default: 506d4ecd108SDag-Erling Smørgrav error("read (%s): %s", c->c_name, strerror(errno)); 507d4ecd108SDag-Erling Smørgrav break; 508d4ecd108SDag-Erling Smørgrav } 509ae1f160dSDag-Erling Smørgrav conrecycle(s); 510ae1f160dSDag-Erling Smørgrav return; 511ae1f160dSDag-Erling Smørgrav } 5121e8db6e2SBrian Feldman if (*cp != '\n' && *cp != '\r') { 5131e8db6e2SBrian Feldman error("%s: bad greeting", c->c_name); 5141e8db6e2SBrian Feldman confree(s); 5151e8db6e2SBrian Feldman return; 5161e8db6e2SBrian Feldman } 5171e8db6e2SBrian Feldman *cp = '\0'; 518bc5531deSDag-Erling Smørgrav if ((c->c_ssh = ssh_packet_set_connection(NULL, s, s)) == NULL) 519bc5531deSDag-Erling Smørgrav fatal("ssh_packet_set_connection failed"); 520bc5531deSDag-Erling Smørgrav ssh_packet_set_timeout(c->c_ssh, timeout, 1); 521bc5531deSDag-Erling Smørgrav ssh_set_app_data(c->c_ssh, c); /* back link */ 522ae1f160dSDag-Erling Smørgrav if (sscanf(buf, "SSH-%d.%d-%[^\n]\n", 523ae1f160dSDag-Erling Smørgrav &remote_major, &remote_minor, remote_version) == 3) 524bc5531deSDag-Erling Smørgrav c->c_ssh->compat = compat_datafellows(remote_version); 525ae1f160dSDag-Erling Smørgrav else 526bc5531deSDag-Erling Smørgrav c->c_ssh->compat = 0; 527ae1f160dSDag-Erling Smørgrav if (c->c_keytype != KT_RSA1) { 528ae1f160dSDag-Erling Smørgrav if (!ssh2_capable(remote_major, remote_minor)) { 529ae1f160dSDag-Erling Smørgrav debug("%s doesn't support ssh2", c->c_name); 530ae1f160dSDag-Erling Smørgrav confree(s); 531ae1f160dSDag-Erling Smørgrav return; 532ae1f160dSDag-Erling Smørgrav } 533ae1f160dSDag-Erling Smørgrav } else if (remote_major != 1) { 534ae1f160dSDag-Erling Smørgrav debug("%s doesn't support ssh1", c->c_name); 535ae1f160dSDag-Erling Smørgrav confree(s); 536ae1f160dSDag-Erling Smørgrav return; 537ae1f160dSDag-Erling Smørgrav } 538557f75e5SDag-Erling Smørgrav fprintf(stderr, "# %s:%d %s\n", c->c_name, ssh_port, chop(buf)); 539ae1f160dSDag-Erling Smørgrav n = snprintf(buf, sizeof buf, "SSH-%d.%d-OpenSSH-keyscan\r\n", 540ae1f160dSDag-Erling Smørgrav c->c_keytype == KT_RSA1? PROTOCOL_MAJOR_1 : PROTOCOL_MAJOR_2, 541ae1f160dSDag-Erling Smørgrav c->c_keytype == KT_RSA1? PROTOCOL_MINOR_1 : PROTOCOL_MINOR_2); 542d4ecd108SDag-Erling Smørgrav if (n < 0 || (size_t)n >= sizeof(buf)) { 543d4ecd108SDag-Erling Smørgrav error("snprintf: buffer too small"); 544d4ecd108SDag-Erling Smørgrav confree(s); 545d4ecd108SDag-Erling Smørgrav return; 546d4ecd108SDag-Erling Smørgrav } 547d4ecd108SDag-Erling Smørgrav if (atomicio(vwrite, s, buf, n) != (size_t)n) { 5481e8db6e2SBrian Feldman error("write (%s): %s", c->c_name, strerror(errno)); 5491e8db6e2SBrian Feldman confree(s); 5501e8db6e2SBrian Feldman return; 5511e8db6e2SBrian Feldman } 552ae1f160dSDag-Erling Smørgrav if (c->c_keytype != KT_RSA1) { 553bc5531deSDag-Erling Smørgrav keygrab_ssh2(c); 554ae1f160dSDag-Erling Smørgrav confree(s); 555ae1f160dSDag-Erling Smørgrav return; 556ae1f160dSDag-Erling Smørgrav } 5571e8db6e2SBrian Feldman c->c_status = CS_SIZE; 5581e8db6e2SBrian Feldman contouch(s); 5591e8db6e2SBrian Feldman } 5601e8db6e2SBrian Feldman 561ae1f160dSDag-Erling Smørgrav static void 5621e8db6e2SBrian Feldman conread(int s) 5631e8db6e2SBrian Feldman { 5641e8db6e2SBrian Feldman con *c = &fdcon[s]; 565d4ecd108SDag-Erling Smørgrav size_t n; 5661e8db6e2SBrian Feldman 5671e8db6e2SBrian Feldman if (c->c_status == CS_CON) { 5681e8db6e2SBrian Feldman congreet(s); 5691e8db6e2SBrian Feldman return; 5701e8db6e2SBrian Feldman } 57121e764dfSDag-Erling Smørgrav n = atomicio(read, s, c->c_data + c->c_off, c->c_len - c->c_off); 572d4ecd108SDag-Erling Smørgrav if (n == 0) { 5731e8db6e2SBrian Feldman error("read (%s): %s", c->c_name, strerror(errno)); 5741e8db6e2SBrian Feldman confree(s); 5751e8db6e2SBrian Feldman return; 5761e8db6e2SBrian Feldman } 5771e8db6e2SBrian Feldman c->c_off += n; 5781e8db6e2SBrian Feldman 5791e8db6e2SBrian Feldman if (c->c_off == c->c_len) 5801e8db6e2SBrian Feldman switch (c->c_status) { 5811e8db6e2SBrian Feldman case CS_SIZE: 5821e8db6e2SBrian Feldman c->c_plen = htonl(c->c_plen); 5831e8db6e2SBrian Feldman c->c_len = c->c_plen + 8 - (c->c_plen & 7); 5841e8db6e2SBrian Feldman c->c_off = 0; 5851e8db6e2SBrian Feldman c->c_data = xmalloc(c->c_len); 5861e8db6e2SBrian Feldman c->c_status = CS_KEYS; 5871e8db6e2SBrian Feldman break; 588a0ee8cc6SDag-Erling Smørgrav #ifdef WITH_SSH1 5891e8db6e2SBrian Feldman case CS_KEYS: 590ae1f160dSDag-Erling Smørgrav keyprint(c, keygrab_ssh1(c)); 5911e8db6e2SBrian Feldman confree(s); 5921e8db6e2SBrian Feldman return; 593a0ee8cc6SDag-Erling Smørgrav #endif 5941e8db6e2SBrian Feldman default: 5951e8db6e2SBrian Feldman fatal("conread: invalid status %d", c->c_status); 5961e8db6e2SBrian Feldman break; 5971e8db6e2SBrian Feldman } 5981e8db6e2SBrian Feldman 5991e8db6e2SBrian Feldman contouch(s); 6001e8db6e2SBrian Feldman } 6011e8db6e2SBrian Feldman 602ae1f160dSDag-Erling Smørgrav static void 6031e8db6e2SBrian Feldman conloop(void) 6041e8db6e2SBrian Feldman { 6051e8db6e2SBrian Feldman struct timeval seltime, now; 606f388f5efSDag-Erling Smørgrav fd_set *r, *e; 6071e8db6e2SBrian Feldman con *c; 608f388f5efSDag-Erling Smørgrav int i; 6091e8db6e2SBrian Feldman 6101e8db6e2SBrian Feldman gettimeofday(&now, NULL); 611545d5ecaSDag-Erling Smørgrav c = TAILQ_FIRST(&tq); 6121e8db6e2SBrian Feldman 6131e8db6e2SBrian Feldman if (c && (c->c_tv.tv_sec > now.tv_sec || 6141e8db6e2SBrian Feldman (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec > now.tv_usec))) { 6151e8db6e2SBrian Feldman seltime = c->c_tv; 6161e8db6e2SBrian Feldman seltime.tv_sec -= now.tv_sec; 6171e8db6e2SBrian Feldman seltime.tv_usec -= now.tv_usec; 6181e8db6e2SBrian Feldman if (seltime.tv_usec < 0) { 6191e8db6e2SBrian Feldman seltime.tv_usec += 1000000; 6201e8db6e2SBrian Feldman seltime.tv_sec--; 6211e8db6e2SBrian Feldman } 6221e8db6e2SBrian Feldman } else 623e146993eSDag-Erling Smørgrav timerclear(&seltime); 6241e8db6e2SBrian Feldman 625333ee039SDag-Erling Smørgrav r = xcalloc(read_wait_nfdset, sizeof(fd_mask)); 626333ee039SDag-Erling Smørgrav e = xcalloc(read_wait_nfdset, sizeof(fd_mask)); 627333ee039SDag-Erling Smørgrav memcpy(r, read_wait, read_wait_nfdset * sizeof(fd_mask)); 628333ee039SDag-Erling Smørgrav memcpy(e, read_wait, read_wait_nfdset * sizeof(fd_mask)); 6291e8db6e2SBrian Feldman 6301e8db6e2SBrian Feldman while (select(maxfd, r, NULL, e, &seltime) == -1 && 631d4af9e69SDag-Erling Smørgrav (errno == EAGAIN || errno == EINTR || errno == EWOULDBLOCK)) 6321e8db6e2SBrian Feldman ; 6331e8db6e2SBrian Feldman 6341e8db6e2SBrian Feldman for (i = 0; i < maxfd; i++) { 6351e8db6e2SBrian Feldman if (FD_ISSET(i, e)) { 6361e8db6e2SBrian Feldman error("%s: exception!", fdcon[i].c_name); 6371e8db6e2SBrian Feldman confree(i); 6381e8db6e2SBrian Feldman } else if (FD_ISSET(i, r)) 6391e8db6e2SBrian Feldman conread(i); 6401e8db6e2SBrian Feldman } 641e4a9863fSDag-Erling Smørgrav free(r); 642e4a9863fSDag-Erling Smørgrav free(e); 6431e8db6e2SBrian Feldman 644545d5ecaSDag-Erling Smørgrav c = TAILQ_FIRST(&tq); 6451e8db6e2SBrian Feldman while (c && (c->c_tv.tv_sec < now.tv_sec || 6461e8db6e2SBrian Feldman (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec < now.tv_usec))) { 6471e8db6e2SBrian Feldman int s = c->c_fd; 6481e8db6e2SBrian Feldman 649545d5ecaSDag-Erling Smørgrav c = TAILQ_NEXT(c, c_link); 6501e8db6e2SBrian Feldman conrecycle(s); 6511e8db6e2SBrian Feldman } 6521e8db6e2SBrian Feldman } 6531e8db6e2SBrian Feldman 654ae1f160dSDag-Erling Smørgrav static void 655ae1f160dSDag-Erling Smørgrav do_host(char *host) 6561e8db6e2SBrian Feldman { 657ae1f160dSDag-Erling Smørgrav char *name = strnnsep(&host, " \t\n"); 658ae1f160dSDag-Erling Smørgrav int j; 6591e8db6e2SBrian Feldman 660ae1f160dSDag-Erling Smørgrav if (name == NULL) 661ae1f160dSDag-Erling Smørgrav return; 662f7167e0eSDag-Erling Smørgrav for (j = KT_RSA1; j <= KT_ED25519; j *= 2) { 663ae1f160dSDag-Erling Smørgrav if (get_keytypes & j) { 664ae1f160dSDag-Erling Smørgrav while (ncon >= MAXCON) 665ae1f160dSDag-Erling Smørgrav conloop(); 666ae1f160dSDag-Erling Smørgrav conalloc(name, *host ? host : name, j); 6671e8db6e2SBrian Feldman } 6681e8db6e2SBrian Feldman } 6691e8db6e2SBrian Feldman } 6701e8db6e2SBrian Feldman 6711e8db6e2SBrian Feldman void 672ae1f160dSDag-Erling Smørgrav fatal(const char *fmt,...) 673ae1f160dSDag-Erling Smørgrav { 674ae1f160dSDag-Erling Smørgrav va_list args; 675f388f5efSDag-Erling Smørgrav 676ae1f160dSDag-Erling Smørgrav va_start(args, fmt); 677ae1f160dSDag-Erling Smørgrav do_log(SYSLOG_LEVEL_FATAL, fmt, args); 678ae1f160dSDag-Erling Smørgrav va_end(args); 6791ec0d754SDag-Erling Smørgrav exit(255); 680ae1f160dSDag-Erling Smørgrav } 681ae1f160dSDag-Erling Smørgrav 682ae1f160dSDag-Erling Smørgrav static void 6831e8db6e2SBrian Feldman usage(void) 6841e8db6e2SBrian Feldman { 685cce7d346SDag-Erling Smørgrav fprintf(stderr, 686acc1a9efSDag-Erling Smørgrav "usage: %s [-46cHv] [-f file] [-p port] [-T timeout] [-t type]\n" 687cce7d346SDag-Erling Smørgrav "\t\t [host | addrlist namelist] ...\n", 688ae1f160dSDag-Erling Smørgrav __progname); 689ae1f160dSDag-Erling Smørgrav exit(1); 6901e8db6e2SBrian Feldman } 6911e8db6e2SBrian Feldman 6921e8db6e2SBrian Feldman int 6931e8db6e2SBrian Feldman main(int argc, char **argv) 6941e8db6e2SBrian Feldman { 695ae1f160dSDag-Erling Smørgrav int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO; 696e2f6069cSDag-Erling Smørgrav int opt, fopt_count = 0, j; 697e2f6069cSDag-Erling Smørgrav char *tname, *cp, line[NI_MAXHOST]; 698e2f6069cSDag-Erling Smørgrav FILE *fp; 699e2f6069cSDag-Erling Smørgrav u_long linenum; 700ae1f160dSDag-Erling Smørgrav 701ae1f160dSDag-Erling Smørgrav extern int optind; 702ae1f160dSDag-Erling Smørgrav extern char *optarg; 7031e8db6e2SBrian Feldman 704acc1a9efSDag-Erling Smørgrav ssh_malloc_init(); /* must be called before any mallocs */ 705cf2b5f3bSDag-Erling Smørgrav __progname = ssh_get_progname(argv[0]); 706989dd127SDag-Erling Smørgrav seed_rng(); 7071e8db6e2SBrian Feldman TAILQ_INIT(&tq); 7081e8db6e2SBrian Feldman 709b74df5b2SDag-Erling Smørgrav /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */ 710b74df5b2SDag-Erling Smørgrav sanitise_stdfd(); 711b74df5b2SDag-Erling Smørgrav 712ae1f160dSDag-Erling Smørgrav if (argc <= 1) 7131e8db6e2SBrian Feldman usage(); 7141e8db6e2SBrian Feldman 715acc1a9efSDag-Erling Smørgrav while ((opt = getopt(argc, argv, "cHv46p:T:t:f:")) != -1) { 716ae1f160dSDag-Erling Smørgrav switch (opt) { 717aa49c926SDag-Erling Smørgrav case 'H': 718aa49c926SDag-Erling Smørgrav hash_hosts = 1; 719aa49c926SDag-Erling Smørgrav break; 720acc1a9efSDag-Erling Smørgrav case 'c': 721acc1a9efSDag-Erling Smørgrav get_cert = 1; 722acc1a9efSDag-Erling Smørgrav break; 723ae1f160dSDag-Erling Smørgrav case 'p': 724ae1f160dSDag-Erling Smørgrav ssh_port = a2port(optarg); 725cce7d346SDag-Erling Smørgrav if (ssh_port <= 0) { 726ae1f160dSDag-Erling Smørgrav fprintf(stderr, "Bad port '%s'\n", optarg); 727ae1f160dSDag-Erling Smørgrav exit(1); 7281e8db6e2SBrian Feldman } 729ae1f160dSDag-Erling Smørgrav break; 730ae1f160dSDag-Erling Smørgrav case 'T': 731f388f5efSDag-Erling Smørgrav timeout = convtime(optarg); 732f388f5efSDag-Erling Smørgrav if (timeout == -1 || timeout == 0) { 733f388f5efSDag-Erling Smørgrav fprintf(stderr, "Bad timeout '%s'\n", optarg); 7341e8db6e2SBrian Feldman usage(); 735f388f5efSDag-Erling Smørgrav } 736ae1f160dSDag-Erling Smørgrav break; 737ae1f160dSDag-Erling Smørgrav case 'v': 738ae1f160dSDag-Erling Smørgrav if (!debug_flag) { 739ae1f160dSDag-Erling Smørgrav debug_flag = 1; 740ae1f160dSDag-Erling Smørgrav log_level = SYSLOG_LEVEL_DEBUG1; 7411e8db6e2SBrian Feldman } 742ae1f160dSDag-Erling Smørgrav else if (log_level < SYSLOG_LEVEL_DEBUG3) 743ae1f160dSDag-Erling Smørgrav log_level++; 744ae1f160dSDag-Erling Smørgrav else 745ae1f160dSDag-Erling Smørgrav fatal("Too high debugging level."); 746ae1f160dSDag-Erling Smørgrav break; 747ae1f160dSDag-Erling Smørgrav case 'f': 748ae1f160dSDag-Erling Smørgrav if (strcmp(optarg, "-") == 0) 749ae1f160dSDag-Erling Smørgrav optarg = NULL; 750ae1f160dSDag-Erling Smørgrav argv[fopt_count++] = optarg; 751ae1f160dSDag-Erling Smørgrav break; 752ae1f160dSDag-Erling Smørgrav case 't': 753ae1f160dSDag-Erling Smørgrav get_keytypes = 0; 754ae1f160dSDag-Erling Smørgrav tname = strtok(optarg, ","); 755ae1f160dSDag-Erling Smørgrav while (tname) { 756bc5531deSDag-Erling Smørgrav int type = sshkey_type_from_name(tname); 757*d93a896eSDag-Erling Smørgrav 758ae1f160dSDag-Erling Smørgrav switch (type) { 759*d93a896eSDag-Erling Smørgrav #ifdef WITH_SSH1 760ae1f160dSDag-Erling Smørgrav case KEY_RSA1: 761ae1f160dSDag-Erling Smørgrav get_keytypes |= KT_RSA1; 762ae1f160dSDag-Erling Smørgrav break; 763*d93a896eSDag-Erling Smørgrav #endif 764ae1f160dSDag-Erling Smørgrav case KEY_DSA: 765ae1f160dSDag-Erling Smørgrav get_keytypes |= KT_DSA; 766ae1f160dSDag-Erling Smørgrav break; 7674a421b63SDag-Erling Smørgrav case KEY_ECDSA: 7684a421b63SDag-Erling Smørgrav get_keytypes |= KT_ECDSA; 7694a421b63SDag-Erling Smørgrav break; 770ae1f160dSDag-Erling Smørgrav case KEY_RSA: 771ae1f160dSDag-Erling Smørgrav get_keytypes |= KT_RSA; 772ae1f160dSDag-Erling Smørgrav break; 773f7167e0eSDag-Erling Smørgrav case KEY_ED25519: 774f7167e0eSDag-Erling Smørgrav get_keytypes |= KT_ED25519; 775f7167e0eSDag-Erling Smørgrav break; 776ae1f160dSDag-Erling Smørgrav case KEY_UNSPEC: 777*d93a896eSDag-Erling Smørgrav default: 778*d93a896eSDag-Erling Smørgrav fatal("Unknown key type \"%s\"", tname); 779ae1f160dSDag-Erling Smørgrav } 780ae1f160dSDag-Erling Smørgrav tname = strtok(NULL, ","); 781ae1f160dSDag-Erling Smørgrav } 782ae1f160dSDag-Erling Smørgrav break; 783ae1f160dSDag-Erling Smørgrav case '4': 784ae1f160dSDag-Erling Smørgrav IPv4or6 = AF_INET; 785ae1f160dSDag-Erling Smørgrav break; 786ae1f160dSDag-Erling Smørgrav case '6': 787ae1f160dSDag-Erling Smørgrav IPv4or6 = AF_INET6; 788ae1f160dSDag-Erling Smørgrav break; 789ae1f160dSDag-Erling Smørgrav case '?': 790ae1f160dSDag-Erling Smørgrav default: 7911e8db6e2SBrian Feldman usage(); 792ae1f160dSDag-Erling Smørgrav } 793ae1f160dSDag-Erling Smørgrav } 794ae1f160dSDag-Erling Smørgrav if (optind == argc && !fopt_count) 795ae1f160dSDag-Erling Smørgrav usage(); 796ae1f160dSDag-Erling Smørgrav 797ae1f160dSDag-Erling Smørgrav log_init("ssh-keyscan", log_level, SYSLOG_FACILITY_USER, 1); 7981e8db6e2SBrian Feldman 7991e8db6e2SBrian Feldman maxfd = fdlim_get(1); 8001e8db6e2SBrian Feldman if (maxfd < 0) 8011e8db6e2SBrian Feldman fatal("%s: fdlim_get: bad value", __progname); 8021e8db6e2SBrian Feldman if (maxfd > MAXMAXFD) 8031e8db6e2SBrian Feldman maxfd = MAXMAXFD; 8041e8db6e2SBrian Feldman if (MAXCON <= 0) 8051e8db6e2SBrian Feldman fatal("%s: not enough file descriptors", __progname); 8061e8db6e2SBrian Feldman if (maxfd > fdlim_get(0)) 8071e8db6e2SBrian Feldman fdlim_set(maxfd); 808333ee039SDag-Erling Smørgrav fdcon = xcalloc(maxfd, sizeof(con)); 8091e8db6e2SBrian Feldman 810333ee039SDag-Erling Smørgrav read_wait_nfdset = howmany(maxfd, NFDBITS); 811333ee039SDag-Erling Smørgrav read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask)); 8121e8db6e2SBrian Feldman 813ae1f160dSDag-Erling Smørgrav for (j = 0; j < fopt_count; j++) { 814e2f6069cSDag-Erling Smørgrav if (argv[j] == NULL) 815e2f6069cSDag-Erling Smørgrav fp = stdin; 816e2f6069cSDag-Erling Smørgrav else if ((fp = fopen(argv[j], "r")) == NULL) 817e2f6069cSDag-Erling Smørgrav fatal("%s: %s: %s", __progname, argv[j], 818e2f6069cSDag-Erling Smørgrav strerror(errno)); 819e2f6069cSDag-Erling Smørgrav linenum = 0; 820e2f6069cSDag-Erling Smørgrav 821e2f6069cSDag-Erling Smørgrav while (read_keyfile_line(fp, 822e2f6069cSDag-Erling Smørgrav argv[j] == NULL ? "(stdin)" : argv[j], line, sizeof(line), 823e2f6069cSDag-Erling Smørgrav &linenum) != -1) { 824e2f6069cSDag-Erling Smørgrav /* Chomp off trailing whitespace and comments */ 825e2f6069cSDag-Erling Smørgrav if ((cp = strchr(line, '#')) == NULL) 826e2f6069cSDag-Erling Smørgrav cp = line + strlen(line) - 1; 827e2f6069cSDag-Erling Smørgrav while (cp >= line) { 828e2f6069cSDag-Erling Smørgrav if (*cp == ' ' || *cp == '\t' || 829e2f6069cSDag-Erling Smørgrav *cp == '\n' || *cp == '#') 830e2f6069cSDag-Erling Smørgrav *cp-- = '\0'; 831e2f6069cSDag-Erling Smørgrav else 832e2f6069cSDag-Erling Smørgrav break; 8331e8db6e2SBrian Feldman } 834e2f6069cSDag-Erling Smørgrav 835e2f6069cSDag-Erling Smørgrav /* Skip empty lines */ 836e2f6069cSDag-Erling Smørgrav if (*line == '\0') 837e2f6069cSDag-Erling Smørgrav continue; 838e2f6069cSDag-Erling Smørgrav 839e2f6069cSDag-Erling Smørgrav do_host(line); 840e2f6069cSDag-Erling Smørgrav } 841e2f6069cSDag-Erling Smørgrav 842e2f6069cSDag-Erling Smørgrav if (ferror(fp)) 843e2f6069cSDag-Erling Smørgrav fatal("%s: %s: %s", __progname, argv[j], 844e2f6069cSDag-Erling Smørgrav strerror(errno)); 845e2f6069cSDag-Erling Smørgrav 846e2f6069cSDag-Erling Smørgrav fclose(fp); 847ae1f160dSDag-Erling Smørgrav } 848ae1f160dSDag-Erling Smørgrav 849ae1f160dSDag-Erling Smørgrav while (optind < argc) 850ae1f160dSDag-Erling Smørgrav do_host(argv[optind++]); 851ae1f160dSDag-Erling Smørgrav 8521e8db6e2SBrian Feldman while (ncon > 0) 8531e8db6e2SBrian Feldman conloop(); 8541e8db6e2SBrian Feldman 8551e8db6e2SBrian Feldman return (0); 8561e8db6e2SBrian Feldman } 857