xref: /openssh-portable/ssh-keyscan.c (revision 9b6e30b9)
1 /* $OpenBSD: ssh-keyscan.c,v 1.131 2019/12/15 19:47:10 djm Exp $ */
2 /*
3  * Copyright 1995, 1996 by David Mazieres <dm@lcs.mit.edu>.
4  *
5  * Modification and redistribution in source and binary forms is
6  * permitted provided that due credit is given to the author and the
7  * OpenBSD project by leaving this copyright notice intact.
8  */
9 
10 #include "includes.h"
11 
12 #include <sys/types.h>
13 #include "openbsd-compat/sys-queue.h"
14 #include <sys/resource.h>
15 #ifdef HAVE_SYS_TIME_H
16 # include <sys/time.h>
17 #endif
18 
19 #include <netinet/in.h>
20 #include <arpa/inet.h>
21 
22 #ifdef WITH_OPENSSL
23 #include <openssl/bn.h>
24 #endif
25 
26 #include <netdb.h>
27 #include <errno.h>
28 #include <stdarg.h>
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <signal.h>
32 #include <string.h>
33 #include <unistd.h>
34 
35 #include "xmalloc.h"
36 #include "ssh.h"
37 #include "sshbuf.h"
38 #include "sshkey.h"
39 #include "cipher.h"
40 #include "kex.h"
41 #include "compat.h"
42 #include "myproposal.h"
43 #include "packet.h"
44 #include "dispatch.h"
45 #include "log.h"
46 #include "atomicio.h"
47 #include "misc.h"
48 #include "hostfile.h"
49 #include "ssherr.h"
50 #include "ssh_api.h"
51 #include "dns.h"
52 
53 /* Flag indicating whether IPv4 or IPv6.  This can be set on the command line.
54    Default value is AF_UNSPEC means both IPv4 and IPv6. */
55 int IPv4or6 = AF_UNSPEC;
56 
57 int ssh_port = SSH_DEFAULT_PORT;
58 
59 #define KT_DSA		(1)
60 #define KT_RSA		(1<<1)
61 #define KT_ECDSA	(1<<2)
62 #define KT_ED25519	(1<<3)
63 #define KT_XMSS		(1<<4)
64 #define KT_ECDSA_SK	(1<<5)
65 #define KT_ED25519_SK	(1<<6)
66 
67 #define KT_MIN		KT_DSA
68 #define KT_MAX		KT_ED25519_SK
69 
70 int get_cert = 0;
71 int get_keytypes = KT_RSA|KT_ECDSA|KT_ED25519|KT_ECDSA_SK|KT_ED25519_SK;
72 
73 int hash_hosts = 0;		/* Hash hostname on output */
74 
75 int print_sshfp = 0;		/* Print SSHFP records instead of known_hosts */
76 
77 int found_one = 0;		/* Successfully found a key */
78 
79 #define MAXMAXFD 256
80 
81 /* The number of seconds after which to give up on a TCP connection */
82 int timeout = 5;
83 
84 int maxfd;
85 #define MAXCON (maxfd - 10)
86 
87 extern char *__progname;
88 fd_set *read_wait;
89 size_t read_wait_nfdset;
90 int ncon;
91 
92 /*
93  * Keep a connection structure for each file descriptor.  The state
94  * associated with file descriptor n is held in fdcon[n].
95  */
96 typedef struct Connection {
97 	u_char c_status;	/* State of connection on this file desc. */
98 #define CS_UNUSED 0		/* File descriptor unused */
99 #define CS_CON 1		/* Waiting to connect/read greeting */
100 #define CS_SIZE 2		/* Waiting to read initial packet size */
101 #define CS_KEYS 3		/* Waiting to read public key packet */
102 	int c_fd;		/* Quick lookup: c->c_fd == c - fdcon */
103 	int c_plen;		/* Packet length field for ssh packet */
104 	int c_len;		/* Total bytes which must be read. */
105 	int c_off;		/* Length of data read so far. */
106 	int c_keytype;		/* Only one of KT_* */
107 	sig_atomic_t c_done;	/* SSH2 done */
108 	char *c_namebase;	/* Address to free for c_name and c_namelist */
109 	char *c_name;		/* Hostname of connection for errors */
110 	char *c_namelist;	/* Pointer to other possible addresses */
111 	char *c_output_name;	/* Hostname of connection for output */
112 	char *c_data;		/* Data read from this fd */
113 	struct ssh *c_ssh;	/* SSH-connection */
114 	struct timeval c_tv;	/* Time at which connection gets aborted */
115 	TAILQ_ENTRY(Connection) c_link;	/* List of connections in timeout order. */
116 } con;
117 
118 TAILQ_HEAD(conlist, Connection) tq;	/* Timeout Queue */
119 con *fdcon;
120 
121 static void keyprint(con *c, struct sshkey *key);
122 
123 static int
fdlim_get(int hard)124 fdlim_get(int hard)
125 {
126 #if defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE)
127 	struct rlimit rlfd;
128 
129 	if (getrlimit(RLIMIT_NOFILE, &rlfd) == -1)
130 		return (-1);
131 	if ((hard ? rlfd.rlim_max : rlfd.rlim_cur) == RLIM_INFINITY)
132 		return SSH_SYSFDMAX;
133 	else
134 		return hard ? rlfd.rlim_max : rlfd.rlim_cur;
135 #else
136 	return SSH_SYSFDMAX;
137 #endif
138 }
139 
140 static int
fdlim_set(int lim)141 fdlim_set(int lim)
142 {
143 #if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
144 	struct rlimit rlfd;
145 #endif
146 
147 	if (lim <= 0)
148 		return (-1);
149 #if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
150 	if (getrlimit(RLIMIT_NOFILE, &rlfd) == -1)
151 		return (-1);
152 	rlfd.rlim_cur = lim;
153 	if (setrlimit(RLIMIT_NOFILE, &rlfd) == -1)
154 		return (-1);
155 #elif defined (HAVE_SETDTABLESIZE)
156 	setdtablesize(lim);
157 #endif
158 	return (0);
159 }
160 
161 /*
162  * This is an strsep function that returns a null field for adjacent
163  * separators.  This is the same as the 4.4BSD strsep, but different from the
164  * one in the GNU libc.
165  */
166 static char *
xstrsep(char ** str,const char * delim)167 xstrsep(char **str, const char *delim)
168 {
169 	char *s, *e;
170 
171 	if (!**str)
172 		return (NULL);
173 
174 	s = *str;
175 	e = s + strcspn(s, delim);
176 
177 	if (*e != '\0')
178 		*e++ = '\0';
179 	*str = e;
180 
181 	return (s);
182 }
183 
184 /*
185  * Get the next non-null token (like GNU strsep).  Strsep() will return a
186  * null token for two adjacent separators, so we may have to loop.
187  */
188 static char *
strnnsep(char ** stringp,char * delim)189 strnnsep(char **stringp, char *delim)
190 {
191 	char *tok;
192 
193 	do {
194 		tok = xstrsep(stringp, delim);
195 	} while (tok && *tok == '\0');
196 	return (tok);
197 }
198 
199 
200 static int
key_print_wrapper(struct sshkey * hostkey,struct ssh * ssh)201 key_print_wrapper(struct sshkey *hostkey, struct ssh *ssh)
202 {
203 	con *c;
204 
205 	if ((c = ssh_get_app_data(ssh)) != NULL)
206 		keyprint(c, hostkey);
207 	/* always abort key exchange */
208 	return -1;
209 }
210 
211 static int
ssh2_capable(int remote_major,int remote_minor)212 ssh2_capable(int remote_major, int remote_minor)
213 {
214 	switch (remote_major) {
215 	case 1:
216 		if (remote_minor == 99)
217 			return 1;
218 		break;
219 	case 2:
220 		return 1;
221 	default:
222 		break;
223 	}
224 	return 0;
225 }
226 
227 static void
keygrab_ssh2(con * c)228 keygrab_ssh2(con *c)
229 {
230 	char *myproposal[PROPOSAL_MAX] = { KEX_CLIENT };
231 	int r;
232 
233 	switch (c->c_keytype) {
234 	case KT_DSA:
235 		myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ?
236 		    "ssh-dss-cert-v01@openssh.com" : "ssh-dss";
237 		break;
238 	case KT_RSA:
239 		myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ?
240 		    "rsa-sha2-512-cert-v01@openssh.com,"
241 		    "rsa-sha2-256-cert-v01@openssh.com,"
242 		    "ssh-rsa-cert-v01@openssh.com" :
243 		    "rsa-sha2-512,"
244 		    "rsa-sha2-256,"
245 		    "ssh-rsa";
246 		break;
247 	case KT_ED25519:
248 		myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ?
249 		    "ssh-ed25519-cert-v01@openssh.com" : "ssh-ed25519";
250 		break;
251 	case KT_XMSS:
252 		myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ?
253 		    "ssh-xmss-cert-v01@openssh.com" : "ssh-xmss@openssh.com";
254 		break;
255 	case KT_ECDSA:
256 		myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ?
257 		    "ecdsa-sha2-nistp256-cert-v01@openssh.com,"
258 		    "ecdsa-sha2-nistp384-cert-v01@openssh.com,"
259 		    "ecdsa-sha2-nistp521-cert-v01@openssh.com" :
260 		    "ecdsa-sha2-nistp256,"
261 		    "ecdsa-sha2-nistp384,"
262 		    "ecdsa-sha2-nistp521";
263 		break;
264 	case KT_ECDSA_SK:
265 		myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ?
266 		    "sk-ecdsa-sha2-nistp256-cert-v01@openssh.com" :
267 		    "sk-ecdsa-sha2-nistp256@openssh.com";
268 		break;
269 	case KT_ED25519_SK:
270 		myproposal[PROPOSAL_SERVER_HOST_KEY_ALGS] = get_cert ?
271 		    "sk-ssh-ed25519-cert-v01@openssh.com" :
272 		    "sk-ssh-ed25519@openssh.com";
273 		break;
274 	default:
275 		fatal("unknown key type %d", c->c_keytype);
276 		break;
277 	}
278 	if ((r = kex_setup(c->c_ssh, myproposal)) != 0) {
279 		free(c->c_ssh);
280 		fprintf(stderr, "kex_setup: %s\n", ssh_err(r));
281 		exit(1);
282 	}
283 #ifdef WITH_OPENSSL
284 	c->c_ssh->kex->kex[KEX_DH_GRP1_SHA1] = kex_gen_client;
285 	c->c_ssh->kex->kex[KEX_DH_GRP14_SHA1] = kex_gen_client;
286 	c->c_ssh->kex->kex[KEX_DH_GRP14_SHA256] = kex_gen_client;
287 	c->c_ssh->kex->kex[KEX_DH_GRP16_SHA512] = kex_gen_client;
288 	c->c_ssh->kex->kex[KEX_DH_GRP18_SHA512] = kex_gen_client;
289 	c->c_ssh->kex->kex[KEX_DH_GEX_SHA1] = kexgex_client;
290 	c->c_ssh->kex->kex[KEX_DH_GEX_SHA256] = kexgex_client;
291 # ifdef OPENSSL_HAS_ECC
292 	c->c_ssh->kex->kex[KEX_ECDH_SHA2] = kex_gen_client;
293 # endif
294 #endif
295 	c->c_ssh->kex->kex[KEX_C25519_SHA256] = kex_gen_client;
296 	c->c_ssh->kex->kex[KEX_KEM_SNTRUP4591761X25519_SHA512] = kex_gen_client;
297 	ssh_set_verify_host_key_callback(c->c_ssh, key_print_wrapper);
298 	/*
299 	 * do the key-exchange until an error occurs or until
300 	 * the key_print_wrapper() callback sets c_done.
301 	 */
302 	ssh_dispatch_run(c->c_ssh, DISPATCH_BLOCK, &c->c_done);
303 }
304 
305 static void
keyprint_one(const char * host,struct sshkey * key)306 keyprint_one(const char *host, struct sshkey *key)
307 {
308 	char *hostport;
309 	const char *known_host, *hashed;
310 
311 	found_one = 1;
312 
313 	if (print_sshfp) {
314 		export_dns_rr(host, key, stdout, 0);
315 		return;
316 	}
317 
318 	hostport = put_host_port(host, ssh_port);
319 	lowercase(hostport);
320 	if (hash_hosts && (hashed = host_hash(host, NULL, 0)) == NULL)
321 		fatal("host_hash failed");
322 	known_host = hash_hosts ? hashed : hostport;
323 	if (!get_cert)
324 		fprintf(stdout, "%s ", known_host);
325 	sshkey_write(key, stdout);
326 	fputs("\n", stdout);
327 	free(hostport);
328 }
329 
330 static void
keyprint(con * c,struct sshkey * key)331 keyprint(con *c, struct sshkey *key)
332 {
333 	char *hosts = c->c_output_name ? c->c_output_name : c->c_name;
334 	char *host, *ohosts;
335 
336 	if (key == NULL)
337 		return;
338 	if (get_cert || (!hash_hosts && ssh_port == SSH_DEFAULT_PORT)) {
339 		keyprint_one(hosts, key);
340 		return;
341 	}
342 	ohosts = hosts = xstrdup(hosts);
343 	while ((host = strsep(&hosts, ",")) != NULL)
344 		keyprint_one(host, key);
345 	free(ohosts);
346 }
347 
348 static int
tcpconnect(char * host)349 tcpconnect(char *host)
350 {
351 	struct addrinfo hints, *ai, *aitop;
352 	char strport[NI_MAXSERV];
353 	int gaierr, s = -1;
354 
355 	snprintf(strport, sizeof strport, "%d", ssh_port);
356 	memset(&hints, 0, sizeof(hints));
357 	hints.ai_family = IPv4or6;
358 	hints.ai_socktype = SOCK_STREAM;
359 	if ((gaierr = getaddrinfo(host, strport, &hints, &aitop)) != 0) {
360 		error("getaddrinfo %s: %s", host, ssh_gai_strerror(gaierr));
361 		return -1;
362 	}
363 	for (ai = aitop; ai; ai = ai->ai_next) {
364 		s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
365 		if (s == -1) {
366 			error("socket: %s", strerror(errno));
367 			continue;
368 		}
369 		if (set_nonblock(s) == -1)
370 			fatal("%s: set_nonblock(%d)", __func__, s);
371 		if (connect(s, ai->ai_addr, ai->ai_addrlen) == -1 &&
372 		    errno != EINPROGRESS)
373 			error("connect (`%s'): %s", host, strerror(errno));
374 		else
375 			break;
376 		close(s);
377 		s = -1;
378 	}
379 	freeaddrinfo(aitop);
380 	return s;
381 }
382 
383 static int
conalloc(char * iname,char * oname,int keytype)384 conalloc(char *iname, char *oname, int keytype)
385 {
386 	char *namebase, *name, *namelist;
387 	int s;
388 
389 	namebase = namelist = xstrdup(iname);
390 
391 	do {
392 		name = xstrsep(&namelist, ",");
393 		if (!name) {
394 			free(namebase);
395 			return (-1);
396 		}
397 	} while ((s = tcpconnect(name)) < 0);
398 
399 	if (s >= maxfd)
400 		fatal("conalloc: fdno %d too high", s);
401 	if (fdcon[s].c_status)
402 		fatal("conalloc: attempt to reuse fdno %d", s);
403 
404 	debug3("%s: oname %s kt %d", __func__, oname, keytype);
405 	fdcon[s].c_fd = s;
406 	fdcon[s].c_status = CS_CON;
407 	fdcon[s].c_namebase = namebase;
408 	fdcon[s].c_name = name;
409 	fdcon[s].c_namelist = namelist;
410 	fdcon[s].c_output_name = xstrdup(oname);
411 	fdcon[s].c_data = (char *) &fdcon[s].c_plen;
412 	fdcon[s].c_len = 4;
413 	fdcon[s].c_off = 0;
414 	fdcon[s].c_keytype = keytype;
415 	monotime_tv(&fdcon[s].c_tv);
416 	fdcon[s].c_tv.tv_sec += timeout;
417 	TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link);
418 	FD_SET(s, read_wait);
419 	ncon++;
420 	return (s);
421 }
422 
423 static void
confree(int s)424 confree(int s)
425 {
426 	if (s >= maxfd || fdcon[s].c_status == CS_UNUSED)
427 		fatal("confree: attempt to free bad fdno %d", s);
428 	free(fdcon[s].c_namebase);
429 	free(fdcon[s].c_output_name);
430 	if (fdcon[s].c_status == CS_KEYS)
431 		free(fdcon[s].c_data);
432 	fdcon[s].c_status = CS_UNUSED;
433 	fdcon[s].c_keytype = 0;
434 	if (fdcon[s].c_ssh) {
435 		ssh_packet_close(fdcon[s].c_ssh);
436 		free(fdcon[s].c_ssh);
437 		fdcon[s].c_ssh = NULL;
438 	} else
439 		close(s);
440 	TAILQ_REMOVE(&tq, &fdcon[s], c_link);
441 	FD_CLR(s, read_wait);
442 	ncon--;
443 }
444 
445 static void
contouch(int s)446 contouch(int s)
447 {
448 	TAILQ_REMOVE(&tq, &fdcon[s], c_link);
449 	monotime_tv(&fdcon[s].c_tv);
450 	fdcon[s].c_tv.tv_sec += timeout;
451 	TAILQ_INSERT_TAIL(&tq, &fdcon[s], c_link);
452 }
453 
454 static int
conrecycle(int s)455 conrecycle(int s)
456 {
457 	con *c = &fdcon[s];
458 	int ret;
459 
460 	ret = conalloc(c->c_namelist, c->c_output_name, c->c_keytype);
461 	confree(s);
462 	return (ret);
463 }
464 
465 static void
congreet(int s)466 congreet(int s)
467 {
468 	int n = 0, remote_major = 0, remote_minor = 0;
469 	char buf[256], *cp;
470 	char remote_version[sizeof buf];
471 	size_t bufsiz;
472 	con *c = &fdcon[s];
473 
474 	/* send client banner */
475 	n = snprintf(buf, sizeof buf, "SSH-%d.%d-OpenSSH-keyscan\r\n",
476 	    PROTOCOL_MAJOR_2, PROTOCOL_MINOR_2);
477 	if (n < 0 || (size_t)n >= sizeof(buf)) {
478 		error("snprintf: buffer too small");
479 		confree(s);
480 		return;
481 	}
482 	if (atomicio(vwrite, s, buf, n) != (size_t)n) {
483 		error("write (%s): %s", c->c_name, strerror(errno));
484 		confree(s);
485 		return;
486 	}
487 
488 	for (;;) {
489 		memset(buf, '\0', sizeof(buf));
490 		bufsiz = sizeof(buf);
491 		cp = buf;
492 		while (bufsiz-- &&
493 		    (n = atomicio(read, s, cp, 1)) == 1 && *cp != '\n') {
494 			if (*cp == '\r')
495 				*cp = '\n';
496 			cp++;
497 		}
498 		if (n != 1 || strncmp(buf, "SSH-", 4) == 0)
499 			break;
500 	}
501 	if (n == 0) {
502 		switch (errno) {
503 		case EPIPE:
504 			error("%s: Connection closed by remote host", c->c_name);
505 			break;
506 		case ECONNREFUSED:
507 			break;
508 		default:
509 			error("read (%s): %s", c->c_name, strerror(errno));
510 			break;
511 		}
512 		conrecycle(s);
513 		return;
514 	}
515 	if (*cp != '\n' && *cp != '\r') {
516 		error("%s: bad greeting", c->c_name);
517 		confree(s);
518 		return;
519 	}
520 	*cp = '\0';
521 	if ((c->c_ssh = ssh_packet_set_connection(NULL, s, s)) == NULL)
522 		fatal("ssh_packet_set_connection failed");
523 	ssh_packet_set_timeout(c->c_ssh, timeout, 1);
524 	ssh_set_app_data(c->c_ssh, c);	/* back link */
525 	if (sscanf(buf, "SSH-%d.%d-%[^\n]\n",
526 	    &remote_major, &remote_minor, remote_version) == 3)
527 		c->c_ssh->compat = compat_datafellows(remote_version);
528 	else
529 		c->c_ssh->compat = 0;
530 	if (!ssh2_capable(remote_major, remote_minor)) {
531 		debug("%s doesn't support ssh2", c->c_name);
532 		confree(s);
533 		return;
534 	}
535 	fprintf(stderr, "%c %s:%d %s\n", print_sshfp ? ';' : '#',
536 	    c->c_name, ssh_port, chop(buf));
537 	keygrab_ssh2(c);
538 	confree(s);
539 }
540 
541 static void
conread(int s)542 conread(int s)
543 {
544 	con *c = &fdcon[s];
545 	size_t n;
546 
547 	if (c->c_status == CS_CON) {
548 		congreet(s);
549 		return;
550 	}
551 	n = atomicio(read, s, c->c_data + c->c_off, c->c_len - c->c_off);
552 	if (n == 0) {
553 		error("read (%s): %s", c->c_name, strerror(errno));
554 		confree(s);
555 		return;
556 	}
557 	c->c_off += n;
558 
559 	if (c->c_off == c->c_len)
560 		switch (c->c_status) {
561 		case CS_SIZE:
562 			c->c_plen = htonl(c->c_plen);
563 			c->c_len = c->c_plen + 8 - (c->c_plen & 7);
564 			c->c_off = 0;
565 			c->c_data = xmalloc(c->c_len);
566 			c->c_status = CS_KEYS;
567 			break;
568 		default:
569 			fatal("conread: invalid status %d", c->c_status);
570 			break;
571 		}
572 
573 	contouch(s);
574 }
575 
576 static void
conloop(void)577 conloop(void)
578 {
579 	struct timeval seltime, now;
580 	fd_set *r, *e;
581 	con *c;
582 	int i;
583 
584 	monotime_tv(&now);
585 	c = TAILQ_FIRST(&tq);
586 
587 	if (c && (c->c_tv.tv_sec > now.tv_sec ||
588 	    (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec > now.tv_usec))) {
589 		seltime = c->c_tv;
590 		seltime.tv_sec -= now.tv_sec;
591 		seltime.tv_usec -= now.tv_usec;
592 		if (seltime.tv_usec < 0) {
593 			seltime.tv_usec += 1000000;
594 			seltime.tv_sec--;
595 		}
596 	} else
597 		timerclear(&seltime);
598 
599 	r = xcalloc(read_wait_nfdset, sizeof(fd_mask));
600 	e = xcalloc(read_wait_nfdset, sizeof(fd_mask));
601 	memcpy(r, read_wait, read_wait_nfdset * sizeof(fd_mask));
602 	memcpy(e, read_wait, read_wait_nfdset * sizeof(fd_mask));
603 
604 	while (select(maxfd, r, NULL, e, &seltime) == -1 &&
605 	    (errno == EAGAIN || errno == EINTR || errno == EWOULDBLOCK))
606 		;
607 
608 	for (i = 0; i < maxfd; i++) {
609 		if (FD_ISSET(i, e)) {
610 			error("%s: exception!", fdcon[i].c_name);
611 			confree(i);
612 		} else if (FD_ISSET(i, r))
613 			conread(i);
614 	}
615 	free(r);
616 	free(e);
617 
618 	c = TAILQ_FIRST(&tq);
619 	while (c && (c->c_tv.tv_sec < now.tv_sec ||
620 	    (c->c_tv.tv_sec == now.tv_sec && c->c_tv.tv_usec < now.tv_usec))) {
621 		int s = c->c_fd;
622 
623 		c = TAILQ_NEXT(c, c_link);
624 		conrecycle(s);
625 	}
626 }
627 
628 static void
do_host(char * host)629 do_host(char *host)
630 {
631 	char *name = strnnsep(&host, " \t\n");
632 	int j;
633 
634 	if (name == NULL)
635 		return;
636 	for (j = KT_MIN; j <= KT_MAX; j *= 2) {
637 		if (get_keytypes & j) {
638 			while (ncon >= MAXCON)
639 				conloop();
640 			conalloc(name, *host ? host : name, j);
641 		}
642 	}
643 }
644 
645 void
fatal(const char * fmt,...)646 fatal(const char *fmt,...)
647 {
648 	va_list args;
649 
650 	va_start(args, fmt);
651 	do_log(SYSLOG_LEVEL_FATAL, fmt, args);
652 	va_end(args);
653 	exit(255);
654 }
655 
656 static void
usage(void)657 usage(void)
658 {
659 	fprintf(stderr,
660 	    "usage: %s [-46cDHv] [-f file] [-p port] [-T timeout] [-t type]\n"
661 	    "\t\t   [host | addrlist namelist]\n",
662 	    __progname);
663 	exit(1);
664 }
665 
666 int
main(int argc,char ** argv)667 main(int argc, char **argv)
668 {
669 	int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO;
670 	int opt, fopt_count = 0, j;
671 	char *tname, *cp, *line = NULL;
672 	size_t linesize = 0;
673 	FILE *fp;
674 
675 	extern int optind;
676 	extern char *optarg;
677 
678 	__progname = ssh_get_progname(argv[0]);
679 	seed_rng();
680 	TAILQ_INIT(&tq);
681 
682 	/* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
683 	sanitise_stdfd();
684 
685 	if (argc <= 1)
686 		usage();
687 
688 	while ((opt = getopt(argc, argv, "cDHv46p:T:t:f:")) != -1) {
689 		switch (opt) {
690 		case 'H':
691 			hash_hosts = 1;
692 			break;
693 		case 'c':
694 			get_cert = 1;
695 			break;
696 		case 'D':
697 			print_sshfp = 1;
698 			break;
699 		case 'p':
700 			ssh_port = a2port(optarg);
701 			if (ssh_port <= 0) {
702 				fprintf(stderr, "Bad port '%s'\n", optarg);
703 				exit(1);
704 			}
705 			break;
706 		case 'T':
707 			timeout = convtime(optarg);
708 			if (timeout == -1 || timeout == 0) {
709 				fprintf(stderr, "Bad timeout '%s'\n", optarg);
710 				usage();
711 			}
712 			break;
713 		case 'v':
714 			if (!debug_flag) {
715 				debug_flag = 1;
716 				log_level = SYSLOG_LEVEL_DEBUG1;
717 			}
718 			else if (log_level < SYSLOG_LEVEL_DEBUG3)
719 				log_level++;
720 			else
721 				fatal("Too high debugging level.");
722 			break;
723 		case 'f':
724 			if (strcmp(optarg, "-") == 0)
725 				optarg = NULL;
726 			argv[fopt_count++] = optarg;
727 			break;
728 		case 't':
729 			get_keytypes = 0;
730 			tname = strtok(optarg, ",");
731 			while (tname) {
732 				int type = sshkey_type_from_name(tname);
733 
734 				switch (type) {
735 				case KEY_DSA:
736 					get_keytypes |= KT_DSA;
737 					break;
738 				case KEY_ECDSA:
739 					get_keytypes |= KT_ECDSA;
740 					break;
741 				case KEY_RSA:
742 					get_keytypes |= KT_RSA;
743 					break;
744 				case KEY_ED25519:
745 					get_keytypes |= KT_ED25519;
746 					break;
747 				case KEY_XMSS:
748 					get_keytypes |= KT_XMSS;
749 					break;
750 				case KEY_ED25519_SK:
751 					get_keytypes |= KT_ED25519_SK;
752 					break;
753 				case KEY_ECDSA_SK:
754 					get_keytypes |= KT_ECDSA_SK;
755 					break;
756 				case KEY_UNSPEC:
757 				default:
758 					fatal("Unknown key type \"%s\"", tname);
759 				}
760 				tname = strtok(NULL, ",");
761 			}
762 			break;
763 		case '4':
764 			IPv4or6 = AF_INET;
765 			break;
766 		case '6':
767 			IPv4or6 = AF_INET6;
768 			break;
769 		case '?':
770 		default:
771 			usage();
772 		}
773 	}
774 	if (optind == argc && !fopt_count)
775 		usage();
776 
777 	log_init("ssh-keyscan", log_level, SYSLOG_FACILITY_USER, 1);
778 
779 	maxfd = fdlim_get(1);
780 	if (maxfd < 0)
781 		fatal("%s: fdlim_get: bad value", __progname);
782 	if (maxfd > MAXMAXFD)
783 		maxfd = MAXMAXFD;
784 	if (MAXCON <= 0)
785 		fatal("%s: not enough file descriptors", __progname);
786 	if (maxfd > fdlim_get(0))
787 		fdlim_set(maxfd);
788 	fdcon = xcalloc(maxfd, sizeof(con));
789 
790 	read_wait_nfdset = howmany(maxfd, NFDBITS);
791 	read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask));
792 
793 	for (j = 0; j < fopt_count; j++) {
794 		if (argv[j] == NULL)
795 			fp = stdin;
796 		else if ((fp = fopen(argv[j], "r")) == NULL)
797 			fatal("%s: %s: %s", __progname, argv[j],
798 			    strerror(errno));
799 
800 		while (getline(&line, &linesize, fp) != -1) {
801 			/* Chomp off trailing whitespace and comments */
802 			if ((cp = strchr(line, '#')) == NULL)
803 				cp = line + strlen(line) - 1;
804 			while (cp >= line) {
805 				if (*cp == ' ' || *cp == '\t' ||
806 				    *cp == '\n' || *cp == '#')
807 					*cp-- = '\0';
808 				else
809 					break;
810 			}
811 
812 			/* Skip empty lines */
813 			if (*line == '\0')
814 				continue;
815 
816 			do_host(line);
817 		}
818 
819 		if (ferror(fp))
820 			fatal("%s: %s: %s", __progname, argv[j],
821 			    strerror(errno));
822 
823 		fclose(fp);
824 	}
825 	free(line);
826 
827 	while (optind < argc)
828 		do_host(argv[optind++]);
829 
830 	while (ncon > 0)
831 		conloop();
832 
833 	return found_one ? 0 : 1;
834 }
835