xref: /openssh-portable/misc.c (revision e2c0a21a)
1 /* $OpenBSD: misc.c,v 1.143 2019/11/22 06:50:30 dtucker Exp $ */
2 /*
3  * Copyright (c) 2000 Markus Friedl.  All rights reserved.
4  * Copyright (c) 2005,2006 Damien Miller.  All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26 
27 #include "includes.h"
28 
29 #include <sys/types.h>
30 #include <sys/ioctl.h>
31 #include <sys/socket.h>
32 #include <sys/stat.h>
33 #include <sys/time.h>
34 #include <sys/wait.h>
35 #include <sys/un.h>
36 
37 #include <limits.h>
38 #ifdef HAVE_LIBGEN_H
39 # include <libgen.h>
40 #endif
41 #ifdef HAVE_POLL_H
42 #include <poll.h>
43 #endif
44 #include <signal.h>
45 #include <stdarg.h>
46 #include <stdio.h>
47 #include <stdlib.h>
48 #include <string.h>
49 #include <time.h>
50 #include <unistd.h>
51 
52 #include <netinet/in.h>
53 #include <netinet/in_systm.h>
54 #include <netinet/ip.h>
55 #include <netinet/tcp.h>
56 #include <arpa/inet.h>
57 
58 #include <ctype.h>
59 #include <errno.h>
60 #include <fcntl.h>
61 #include <netdb.h>
62 #ifdef HAVE_PATHS_H
63 # include <paths.h>
64 #include <pwd.h>
65 #endif
66 #ifdef SSH_TUN_OPENBSD
67 #include <net/if.h>
68 #endif
69 
70 #include "xmalloc.h"
71 #include "misc.h"
72 #include "log.h"
73 #include "ssh.h"
74 #include "sshbuf.h"
75 #include "ssherr.h"
76 #include "platform.h"
77 
78 /* remove newline at end of string */
79 char *
chop(char * s)80 chop(char *s)
81 {
82 	char *t = s;
83 	while (*t) {
84 		if (*t == '\n' || *t == '\r') {
85 			*t = '\0';
86 			return s;
87 		}
88 		t++;
89 	}
90 	return s;
91 
92 }
93 
94 /* set/unset filedescriptor to non-blocking */
95 int
set_nonblock(int fd)96 set_nonblock(int fd)
97 {
98 	int val;
99 
100 	val = fcntl(fd, F_GETFL);
101 	if (val == -1) {
102 		error("fcntl(%d, F_GETFL): %s", fd, strerror(errno));
103 		return (-1);
104 	}
105 	if (val & O_NONBLOCK) {
106 		debug3("fd %d is O_NONBLOCK", fd);
107 		return (0);
108 	}
109 	debug2("fd %d setting O_NONBLOCK", fd);
110 	val |= O_NONBLOCK;
111 	if (fcntl(fd, F_SETFL, val) == -1) {
112 		debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s", fd,
113 		    strerror(errno));
114 		return (-1);
115 	}
116 	return (0);
117 }
118 
119 int
unset_nonblock(int fd)120 unset_nonblock(int fd)
121 {
122 	int val;
123 
124 	val = fcntl(fd, F_GETFL);
125 	if (val == -1) {
126 		error("fcntl(%d, F_GETFL): %s", fd, strerror(errno));
127 		return (-1);
128 	}
129 	if (!(val & O_NONBLOCK)) {
130 		debug3("fd %d is not O_NONBLOCK", fd);
131 		return (0);
132 	}
133 	debug("fd %d clearing O_NONBLOCK", fd);
134 	val &= ~O_NONBLOCK;
135 	if (fcntl(fd, F_SETFL, val) == -1) {
136 		debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s",
137 		    fd, strerror(errno));
138 		return (-1);
139 	}
140 	return (0);
141 }
142 
143 const char *
ssh_gai_strerror(int gaierr)144 ssh_gai_strerror(int gaierr)
145 {
146 	if (gaierr == EAI_SYSTEM && errno != 0)
147 		return strerror(errno);
148 	return gai_strerror(gaierr);
149 }
150 
151 /* disable nagle on socket */
152 void
set_nodelay(int fd)153 set_nodelay(int fd)
154 {
155 	int opt;
156 	socklen_t optlen;
157 
158 	optlen = sizeof opt;
159 	if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) {
160 		debug("getsockopt TCP_NODELAY: %.100s", strerror(errno));
161 		return;
162 	}
163 	if (opt == 1) {
164 		debug2("fd %d is TCP_NODELAY", fd);
165 		return;
166 	}
167 	opt = 1;
168 	debug2("fd %d setting TCP_NODELAY", fd);
169 	if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1)
170 		error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
171 }
172 
173 /* Allow local port reuse in TIME_WAIT */
174 int
set_reuseaddr(int fd)175 set_reuseaddr(int fd)
176 {
177 	int on = 1;
178 
179 	if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1) {
180 		error("setsockopt SO_REUSEADDR fd %d: %s", fd, strerror(errno));
181 		return -1;
182 	}
183 	return 0;
184 }
185 
186 /* Get/set routing domain */
187 char *
get_rdomain(int fd)188 get_rdomain(int fd)
189 {
190 #if defined(HAVE_SYS_GET_RDOMAIN)
191 	return sys_get_rdomain(fd);
192 #elif defined(__OpenBSD__)
193 	int rtable;
194 	char *ret;
195 	socklen_t len = sizeof(rtable);
196 
197 	if (getsockopt(fd, SOL_SOCKET, SO_RTABLE, &rtable, &len) == -1) {
198 		error("Failed to get routing domain for fd %d: %s",
199 		    fd, strerror(errno));
200 		return NULL;
201 	}
202 	xasprintf(&ret, "%d", rtable);
203 	return ret;
204 #else /* defined(__OpenBSD__) */
205 	return NULL;
206 #endif
207 }
208 
209 int
set_rdomain(int fd,const char * name)210 set_rdomain(int fd, const char *name)
211 {
212 #if defined(HAVE_SYS_SET_RDOMAIN)
213 	return sys_set_rdomain(fd, name);
214 #elif defined(__OpenBSD__)
215 	int rtable;
216 	const char *errstr;
217 
218 	if (name == NULL)
219 		return 0; /* default table */
220 
221 	rtable = (int)strtonum(name, 0, 255, &errstr);
222 	if (errstr != NULL) {
223 		/* Shouldn't happen */
224 		error("Invalid routing domain \"%s\": %s", name, errstr);
225 		return -1;
226 	}
227 	if (setsockopt(fd, SOL_SOCKET, SO_RTABLE,
228 	    &rtable, sizeof(rtable)) == -1) {
229 		error("Failed to set routing domain %d on fd %d: %s",
230 		    rtable, fd, strerror(errno));
231 		return -1;
232 	}
233 	return 0;
234 #else /* defined(__OpenBSD__) */
235 	error("Setting routing domain is not supported on this platform");
236 	return -1;
237 #endif
238 }
239 
240 /*
241  * Wait up to *timeoutp milliseconds for events on fd. Updates
242  * *timeoutp with time remaining.
243  * Returns 0 if fd ready or -1 on timeout or error (see errno).
244  */
245 static int
waitfd(int fd,int * timeoutp,short events)246 waitfd(int fd, int *timeoutp, short events)
247 {
248 	struct pollfd pfd;
249 	struct timeval t_start;
250 	int oerrno, r;
251 
252 	monotime_tv(&t_start);
253 	pfd.fd = fd;
254 	pfd.events = events;
255 	for (; *timeoutp >= 0;) {
256 		r = poll(&pfd, 1, *timeoutp);
257 		oerrno = errno;
258 		ms_subtract_diff(&t_start, timeoutp);
259 		errno = oerrno;
260 		if (r > 0)
261 			return 0;
262 		else if (r == -1 && errno != EAGAIN)
263 			return -1;
264 		else if (r == 0)
265 			break;
266 	}
267 	/* timeout */
268 	errno = ETIMEDOUT;
269 	return -1;
270 }
271 
272 /*
273  * Wait up to *timeoutp milliseconds for fd to be readable. Updates
274  * *timeoutp with time remaining.
275  * Returns 0 if fd ready or -1 on timeout or error (see errno).
276  */
277 int
waitrfd(int fd,int * timeoutp)278 waitrfd(int fd, int *timeoutp) {
279 	return waitfd(fd, timeoutp, POLLIN);
280 }
281 
282 /*
283  * Attempt a non-blocking connect(2) to the specified address, waiting up to
284  * *timeoutp milliseconds for the connection to complete. If the timeout is
285  * <=0, then wait indefinitely.
286  *
287  * Returns 0 on success or -1 on failure.
288  */
289 int
timeout_connect(int sockfd,const struct sockaddr * serv_addr,socklen_t addrlen,int * timeoutp)290 timeout_connect(int sockfd, const struct sockaddr *serv_addr,
291     socklen_t addrlen, int *timeoutp)
292 {
293 	int optval = 0;
294 	socklen_t optlen = sizeof(optval);
295 
296 	/* No timeout: just do a blocking connect() */
297 	if (timeoutp == NULL || *timeoutp <= 0)
298 		return connect(sockfd, serv_addr, addrlen);
299 
300 	set_nonblock(sockfd);
301 	if (connect(sockfd, serv_addr, addrlen) == 0) {
302 		/* Succeeded already? */
303 		unset_nonblock(sockfd);
304 		return 0;
305 	} else if (errno != EINPROGRESS)
306 		return -1;
307 
308 	if (waitfd(sockfd, timeoutp, POLLIN | POLLOUT) == -1)
309 		return -1;
310 
311 	/* Completed or failed */
312 	if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &optval, &optlen) == -1) {
313 		debug("getsockopt: %s", strerror(errno));
314 		return -1;
315 	}
316 	if (optval != 0) {
317 		errno = optval;
318 		return -1;
319 	}
320 	unset_nonblock(sockfd);
321 	return 0;
322 }
323 
324 /* Characters considered whitespace in strsep calls. */
325 #define WHITESPACE " \t\r\n"
326 #define QUOTE	"\""
327 
328 /* return next token in configuration line */
329 static char *
strdelim_internal(char ** s,int split_equals)330 strdelim_internal(char **s, int split_equals)
331 {
332 	char *old;
333 	int wspace = 0;
334 
335 	if (*s == NULL)
336 		return NULL;
337 
338 	old = *s;
339 
340 	*s = strpbrk(*s,
341 	    split_equals ? WHITESPACE QUOTE "=" : WHITESPACE QUOTE);
342 	if (*s == NULL)
343 		return (old);
344 
345 	if (*s[0] == '\"') {
346 		memmove(*s, *s + 1, strlen(*s)); /* move nul too */
347 		/* Find matching quote */
348 		if ((*s = strpbrk(*s, QUOTE)) == NULL) {
349 			return (NULL);		/* no matching quote */
350 		} else {
351 			*s[0] = '\0';
352 			*s += strspn(*s + 1, WHITESPACE) + 1;
353 			return (old);
354 		}
355 	}
356 
357 	/* Allow only one '=' to be skipped */
358 	if (split_equals && *s[0] == '=')
359 		wspace = 1;
360 	*s[0] = '\0';
361 
362 	/* Skip any extra whitespace after first token */
363 	*s += strspn(*s + 1, WHITESPACE) + 1;
364 	if (split_equals && *s[0] == '=' && !wspace)
365 		*s += strspn(*s + 1, WHITESPACE) + 1;
366 
367 	return (old);
368 }
369 
370 /*
371  * Return next token in configuration line; splts on whitespace or a
372  * single '=' character.
373  */
374 char *
strdelim(char ** s)375 strdelim(char **s)
376 {
377 	return strdelim_internal(s, 1);
378 }
379 
380 /*
381  * Return next token in configuration line; splts on whitespace only.
382  */
383 char *
strdelimw(char ** s)384 strdelimw(char **s)
385 {
386 	return strdelim_internal(s, 0);
387 }
388 
389 struct passwd *
pwcopy(struct passwd * pw)390 pwcopy(struct passwd *pw)
391 {
392 	struct passwd *copy = xcalloc(1, sizeof(*copy));
393 
394 	copy->pw_name = xstrdup(pw->pw_name);
395 	copy->pw_passwd = xstrdup(pw->pw_passwd);
396 #ifdef HAVE_STRUCT_PASSWD_PW_GECOS
397 	copy->pw_gecos = xstrdup(pw->pw_gecos);
398 #endif
399 	copy->pw_uid = pw->pw_uid;
400 	copy->pw_gid = pw->pw_gid;
401 #ifdef HAVE_STRUCT_PASSWD_PW_EXPIRE
402 	copy->pw_expire = pw->pw_expire;
403 #endif
404 #ifdef HAVE_STRUCT_PASSWD_PW_CHANGE
405 	copy->pw_change = pw->pw_change;
406 #endif
407 #ifdef HAVE_STRUCT_PASSWD_PW_CLASS
408 	copy->pw_class = xstrdup(pw->pw_class);
409 #endif
410 	copy->pw_dir = xstrdup(pw->pw_dir);
411 	copy->pw_shell = xstrdup(pw->pw_shell);
412 	return copy;
413 }
414 
415 /*
416  * Convert ASCII string to TCP/IP port number.
417  * Port must be >=0 and <=65535.
418  * Return -1 if invalid.
419  */
420 int
a2port(const char * s)421 a2port(const char *s)
422 {
423 	struct servent *se;
424 	long long port;
425 	const char *errstr;
426 
427 	port = strtonum(s, 0, 65535, &errstr);
428 	if (errstr == NULL)
429 		return (int)port;
430 	if ((se = getservbyname(s, "tcp")) != NULL)
431 		return ntohs(se->s_port);
432 	return -1;
433 }
434 
435 int
a2tun(const char * s,int * remote)436 a2tun(const char *s, int *remote)
437 {
438 	const char *errstr = NULL;
439 	char *sp, *ep;
440 	int tun;
441 
442 	if (remote != NULL) {
443 		*remote = SSH_TUNID_ANY;
444 		sp = xstrdup(s);
445 		if ((ep = strchr(sp, ':')) == NULL) {
446 			free(sp);
447 			return (a2tun(s, NULL));
448 		}
449 		ep[0] = '\0'; ep++;
450 		*remote = a2tun(ep, NULL);
451 		tun = a2tun(sp, NULL);
452 		free(sp);
453 		return (*remote == SSH_TUNID_ERR ? *remote : tun);
454 	}
455 
456 	if (strcasecmp(s, "any") == 0)
457 		return (SSH_TUNID_ANY);
458 
459 	tun = strtonum(s, 0, SSH_TUNID_MAX, &errstr);
460 	if (errstr != NULL)
461 		return (SSH_TUNID_ERR);
462 
463 	return (tun);
464 }
465 
466 #define SECONDS		1
467 #define MINUTES		(SECONDS * 60)
468 #define HOURS		(MINUTES * 60)
469 #define DAYS		(HOURS * 24)
470 #define WEEKS		(DAYS * 7)
471 
472 /*
473  * Convert a time string into seconds; format is
474  * a sequence of:
475  *      time[qualifier]
476  *
477  * Valid time qualifiers are:
478  *      <none>  seconds
479  *      s|S     seconds
480  *      m|M     minutes
481  *      h|H     hours
482  *      d|D     days
483  *      w|W     weeks
484  *
485  * Examples:
486  *      90m     90 minutes
487  *      1h30m   90 minutes
488  *      2d      2 days
489  *      1w      1 week
490  *
491  * Return -1 if time string is invalid.
492  */
493 long
convtime(const char * s)494 convtime(const char *s)
495 {
496 	long total, secs, multiplier = 1;
497 	const char *p;
498 	char *endp;
499 
500 	errno = 0;
501 	total = 0;
502 	p = s;
503 
504 	if (p == NULL || *p == '\0')
505 		return -1;
506 
507 	while (*p) {
508 		secs = strtol(p, &endp, 10);
509 		if (p == endp ||
510 		    (errno == ERANGE && (secs == LONG_MIN || secs == LONG_MAX)) ||
511 		    secs < 0)
512 			return -1;
513 
514 		switch (*endp++) {
515 		case '\0':
516 			endp--;
517 			break;
518 		case 's':
519 		case 'S':
520 			break;
521 		case 'm':
522 		case 'M':
523 			multiplier = MINUTES;
524 			break;
525 		case 'h':
526 		case 'H':
527 			multiplier = HOURS;
528 			break;
529 		case 'd':
530 		case 'D':
531 			multiplier = DAYS;
532 			break;
533 		case 'w':
534 		case 'W':
535 			multiplier = WEEKS;
536 			break;
537 		default:
538 			return -1;
539 		}
540 		if (secs >= LONG_MAX / multiplier)
541 			return -1;
542 		secs *= multiplier;
543 		if  (total >= LONG_MAX - secs)
544 			return -1;
545 		total += secs;
546 		if (total < 0)
547 			return -1;
548 		p = endp;
549 	}
550 
551 	return total;
552 }
553 
554 /*
555  * Returns a standardized host+port identifier string.
556  * Caller must free returned string.
557  */
558 char *
put_host_port(const char * host,u_short port)559 put_host_port(const char *host, u_short port)
560 {
561 	char *hoststr;
562 
563 	if (port == 0 || port == SSH_DEFAULT_PORT)
564 		return(xstrdup(host));
565 	if (asprintf(&hoststr, "[%s]:%d", host, (int)port) == -1)
566 		fatal("put_host_port: asprintf: %s", strerror(errno));
567 	debug3("put_host_port: %s", hoststr);
568 	return hoststr;
569 }
570 
571 /*
572  * Search for next delimiter between hostnames/addresses and ports.
573  * Argument may be modified (for termination).
574  * Returns *cp if parsing succeeds.
575  * *cp is set to the start of the next field, if one was found.
576  * The delimiter char, if present, is stored in delim.
577  * If this is the last field, *cp is set to NULL.
578  */
579 char *
hpdelim2(char ** cp,char * delim)580 hpdelim2(char **cp, char *delim)
581 {
582 	char *s, *old;
583 
584 	if (cp == NULL || *cp == NULL)
585 		return NULL;
586 
587 	old = s = *cp;
588 	if (*s == '[') {
589 		if ((s = strchr(s, ']')) == NULL)
590 			return NULL;
591 		else
592 			s++;
593 	} else if ((s = strpbrk(s, ":/")) == NULL)
594 		s = *cp + strlen(*cp); /* skip to end (see first case below) */
595 
596 	switch (*s) {
597 	case '\0':
598 		*cp = NULL;	/* no more fields*/
599 		break;
600 
601 	case ':':
602 	case '/':
603 		if (delim != NULL)
604 			*delim = *s;
605 		*s = '\0';	/* terminate */
606 		*cp = s + 1;
607 		break;
608 
609 	default:
610 		return NULL;
611 	}
612 
613 	return old;
614 }
615 
616 char *
hpdelim(char ** cp)617 hpdelim(char **cp)
618 {
619 	return hpdelim2(cp, NULL);
620 }
621 
622 char *
cleanhostname(char * host)623 cleanhostname(char *host)
624 {
625 	if (*host == '[' && host[strlen(host) - 1] == ']') {
626 		host[strlen(host) - 1] = '\0';
627 		return (host + 1);
628 	} else
629 		return host;
630 }
631 
632 char *
colon(char * cp)633 colon(char *cp)
634 {
635 	int flag = 0;
636 
637 	if (*cp == ':')		/* Leading colon is part of file name. */
638 		return NULL;
639 	if (*cp == '[')
640 		flag = 1;
641 
642 	for (; *cp; ++cp) {
643 		if (*cp == '@' && *(cp+1) == '[')
644 			flag = 1;
645 		if (*cp == ']' && *(cp+1) == ':' && flag)
646 			return (cp+1);
647 		if (*cp == ':' && !flag)
648 			return (cp);
649 		if (*cp == '/')
650 			return NULL;
651 	}
652 	return NULL;
653 }
654 
655 /*
656  * Parse a [user@]host:[path] string.
657  * Caller must free returned user, host and path.
658  * Any of the pointer return arguments may be NULL (useful for syntax checking).
659  * If user was not specified then *userp will be set to NULL.
660  * If host was not specified then *hostp will be set to NULL.
661  * If path was not specified then *pathp will be set to ".".
662  * Returns 0 on success, -1 on failure.
663  */
664 int
parse_user_host_path(const char * s,char ** userp,char ** hostp,char ** pathp)665 parse_user_host_path(const char *s, char **userp, char **hostp, char **pathp)
666 {
667 	char *user = NULL, *host = NULL, *path = NULL;
668 	char *sdup, *tmp;
669 	int ret = -1;
670 
671 	if (userp != NULL)
672 		*userp = NULL;
673 	if (hostp != NULL)
674 		*hostp = NULL;
675 	if (pathp != NULL)
676 		*pathp = NULL;
677 
678 	sdup = xstrdup(s);
679 
680 	/* Check for remote syntax: [user@]host:[path] */
681 	if ((tmp = colon(sdup)) == NULL)
682 		goto out;
683 
684 	/* Extract optional path */
685 	*tmp++ = '\0';
686 	if (*tmp == '\0')
687 		tmp = ".";
688 	path = xstrdup(tmp);
689 
690 	/* Extract optional user and mandatory host */
691 	tmp = strrchr(sdup, '@');
692 	if (tmp != NULL) {
693 		*tmp++ = '\0';
694 		host = xstrdup(cleanhostname(tmp));
695 		if (*sdup != '\0')
696 			user = xstrdup(sdup);
697 	} else {
698 		host = xstrdup(cleanhostname(sdup));
699 		user = NULL;
700 	}
701 
702 	/* Success */
703 	if (userp != NULL) {
704 		*userp = user;
705 		user = NULL;
706 	}
707 	if (hostp != NULL) {
708 		*hostp = host;
709 		host = NULL;
710 	}
711 	if (pathp != NULL) {
712 		*pathp = path;
713 		path = NULL;
714 	}
715 	ret = 0;
716 out:
717 	free(sdup);
718 	free(user);
719 	free(host);
720 	free(path);
721 	return ret;
722 }
723 
724 /*
725  * Parse a [user@]host[:port] string.
726  * Caller must free returned user and host.
727  * Any of the pointer return arguments may be NULL (useful for syntax checking).
728  * If user was not specified then *userp will be set to NULL.
729  * If port was not specified then *portp will be -1.
730  * Returns 0 on success, -1 on failure.
731  */
732 int
parse_user_host_port(const char * s,char ** userp,char ** hostp,int * portp)733 parse_user_host_port(const char *s, char **userp, char **hostp, int *portp)
734 {
735 	char *sdup, *cp, *tmp;
736 	char *user = NULL, *host = NULL;
737 	int port = -1, ret = -1;
738 
739 	if (userp != NULL)
740 		*userp = NULL;
741 	if (hostp != NULL)
742 		*hostp = NULL;
743 	if (portp != NULL)
744 		*portp = -1;
745 
746 	if ((sdup = tmp = strdup(s)) == NULL)
747 		return -1;
748 	/* Extract optional username */
749 	if ((cp = strrchr(tmp, '@')) != NULL) {
750 		*cp = '\0';
751 		if (*tmp == '\0')
752 			goto out;
753 		if ((user = strdup(tmp)) == NULL)
754 			goto out;
755 		tmp = cp + 1;
756 	}
757 	/* Extract mandatory hostname */
758 	if ((cp = hpdelim(&tmp)) == NULL || *cp == '\0')
759 		goto out;
760 	host = xstrdup(cleanhostname(cp));
761 	/* Convert and verify optional port */
762 	if (tmp != NULL && *tmp != '\0') {
763 		if ((port = a2port(tmp)) <= 0)
764 			goto out;
765 	}
766 	/* Success */
767 	if (userp != NULL) {
768 		*userp = user;
769 		user = NULL;
770 	}
771 	if (hostp != NULL) {
772 		*hostp = host;
773 		host = NULL;
774 	}
775 	if (portp != NULL)
776 		*portp = port;
777 	ret = 0;
778  out:
779 	free(sdup);
780 	free(user);
781 	free(host);
782 	return ret;
783 }
784 
785 /*
786  * Converts a two-byte hex string to decimal.
787  * Returns the decimal value or -1 for invalid input.
788  */
789 static int
hexchar(const char * s)790 hexchar(const char *s)
791 {
792 	unsigned char result[2];
793 	int i;
794 
795 	for (i = 0; i < 2; i++) {
796 		if (s[i] >= '0' && s[i] <= '9')
797 			result[i] = (unsigned char)(s[i] - '0');
798 		else if (s[i] >= 'a' && s[i] <= 'f')
799 			result[i] = (unsigned char)(s[i] - 'a') + 10;
800 		else if (s[i] >= 'A' && s[i] <= 'F')
801 			result[i] = (unsigned char)(s[i] - 'A') + 10;
802 		else
803 			return -1;
804 	}
805 	return (result[0] << 4) | result[1];
806 }
807 
808 /*
809  * Decode an url-encoded string.
810  * Returns a newly allocated string on success or NULL on failure.
811  */
812 static char *
urldecode(const char * src)813 urldecode(const char *src)
814 {
815 	char *ret, *dst;
816 	int ch;
817 
818 	ret = xmalloc(strlen(src) + 1);
819 	for (dst = ret; *src != '\0'; src++) {
820 		switch (*src) {
821 		case '+':
822 			*dst++ = ' ';
823 			break;
824 		case '%':
825 			if (!isxdigit((unsigned char)src[1]) ||
826 			    !isxdigit((unsigned char)src[2]) ||
827 			    (ch = hexchar(src + 1)) == -1) {
828 				free(ret);
829 				return NULL;
830 			}
831 			*dst++ = ch;
832 			src += 2;
833 			break;
834 		default:
835 			*dst++ = *src;
836 			break;
837 		}
838 	}
839 	*dst = '\0';
840 
841 	return ret;
842 }
843 
844 /*
845  * Parse an (scp|ssh|sftp)://[user@]host[:port][/path] URI.
846  * See https://tools.ietf.org/html/draft-ietf-secsh-scp-sftp-ssh-uri-04
847  * Either user or path may be url-encoded (but not host or port).
848  * Caller must free returned user, host and path.
849  * Any of the pointer return arguments may be NULL (useful for syntax checking)
850  * but the scheme must always be specified.
851  * If user was not specified then *userp will be set to NULL.
852  * If port was not specified then *portp will be -1.
853  * If path was not specified then *pathp will be set to NULL.
854  * Returns 0 on success, 1 if non-uri/wrong scheme, -1 on error/invalid uri.
855  */
856 int
parse_uri(const char * scheme,const char * uri,char ** userp,char ** hostp,int * portp,char ** pathp)857 parse_uri(const char *scheme, const char *uri, char **userp, char **hostp,
858     int *portp, char **pathp)
859 {
860 	char *uridup, *cp, *tmp, ch;
861 	char *user = NULL, *host = NULL, *path = NULL;
862 	int port = -1, ret = -1;
863 	size_t len;
864 
865 	len = strlen(scheme);
866 	if (strncmp(uri, scheme, len) != 0 || strncmp(uri + len, "://", 3) != 0)
867 		return 1;
868 	uri += len + 3;
869 
870 	if (userp != NULL)
871 		*userp = NULL;
872 	if (hostp != NULL)
873 		*hostp = NULL;
874 	if (portp != NULL)
875 		*portp = -1;
876 	if (pathp != NULL)
877 		*pathp = NULL;
878 
879 	uridup = tmp = xstrdup(uri);
880 
881 	/* Extract optional ssh-info (username + connection params) */
882 	if ((cp = strchr(tmp, '@')) != NULL) {
883 		char *delim;
884 
885 		*cp = '\0';
886 		/* Extract username and connection params */
887 		if ((delim = strchr(tmp, ';')) != NULL) {
888 			/* Just ignore connection params for now */
889 			*delim = '\0';
890 		}
891 		if (*tmp == '\0') {
892 			/* Empty username */
893 			goto out;
894 		}
895 		if ((user = urldecode(tmp)) == NULL)
896 			goto out;
897 		tmp = cp + 1;
898 	}
899 
900 	/* Extract mandatory hostname */
901 	if ((cp = hpdelim2(&tmp, &ch)) == NULL || *cp == '\0')
902 		goto out;
903 	host = xstrdup(cleanhostname(cp));
904 	if (!valid_domain(host, 0, NULL))
905 		goto out;
906 
907 	if (tmp != NULL && *tmp != '\0') {
908 		if (ch == ':') {
909 			/* Convert and verify port. */
910 			if ((cp = strchr(tmp, '/')) != NULL)
911 				*cp = '\0';
912 			if ((port = a2port(tmp)) <= 0)
913 				goto out;
914 			tmp = cp ? cp + 1 : NULL;
915 		}
916 		if (tmp != NULL && *tmp != '\0') {
917 			/* Extract optional path */
918 			if ((path = urldecode(tmp)) == NULL)
919 				goto out;
920 		}
921 	}
922 
923 	/* Success */
924 	if (userp != NULL) {
925 		*userp = user;
926 		user = NULL;
927 	}
928 	if (hostp != NULL) {
929 		*hostp = host;
930 		host = NULL;
931 	}
932 	if (portp != NULL)
933 		*portp = port;
934 	if (pathp != NULL) {
935 		*pathp = path;
936 		path = NULL;
937 	}
938 	ret = 0;
939  out:
940 	free(uridup);
941 	free(user);
942 	free(host);
943 	free(path);
944 	return ret;
945 }
946 
947 /* function to assist building execv() arguments */
948 void
addargs(arglist * args,char * fmt,...)949 addargs(arglist *args, char *fmt, ...)
950 {
951 	va_list ap;
952 	char *cp;
953 	u_int nalloc;
954 	int r;
955 
956 	va_start(ap, fmt);
957 	r = vasprintf(&cp, fmt, ap);
958 	va_end(ap);
959 	if (r == -1)
960 		fatal("addargs: argument too long");
961 
962 	nalloc = args->nalloc;
963 	if (args->list == NULL) {
964 		nalloc = 32;
965 		args->num = 0;
966 	} else if (args->num+2 >= nalloc)
967 		nalloc *= 2;
968 
969 	args->list = xrecallocarray(args->list, args->nalloc, nalloc, sizeof(char *));
970 	args->nalloc = nalloc;
971 	args->list[args->num++] = cp;
972 	args->list[args->num] = NULL;
973 }
974 
975 void
replacearg(arglist * args,u_int which,char * fmt,...)976 replacearg(arglist *args, u_int which, char *fmt, ...)
977 {
978 	va_list ap;
979 	char *cp;
980 	int r;
981 
982 	va_start(ap, fmt);
983 	r = vasprintf(&cp, fmt, ap);
984 	va_end(ap);
985 	if (r == -1)
986 		fatal("replacearg: argument too long");
987 
988 	if (which >= args->num)
989 		fatal("replacearg: tried to replace invalid arg %d >= %d",
990 		    which, args->num);
991 	free(args->list[which]);
992 	args->list[which] = cp;
993 }
994 
995 void
freeargs(arglist * args)996 freeargs(arglist *args)
997 {
998 	u_int i;
999 
1000 	if (args->list != NULL) {
1001 		for (i = 0; i < args->num; i++)
1002 			free(args->list[i]);
1003 		free(args->list);
1004 		args->nalloc = args->num = 0;
1005 		args->list = NULL;
1006 	}
1007 }
1008 
1009 /*
1010  * Expands tildes in the file name.  Returns data allocated by xmalloc.
1011  * Warning: this calls getpw*.
1012  */
1013 char *
tilde_expand_filename(const char * filename,uid_t uid)1014 tilde_expand_filename(const char *filename, uid_t uid)
1015 {
1016 	const char *path, *sep;
1017 	char user[128], *ret;
1018 	struct passwd *pw;
1019 	u_int len, slash;
1020 
1021 	if (*filename != '~')
1022 		return (xstrdup(filename));
1023 	filename++;
1024 
1025 	path = strchr(filename, '/');
1026 	if (path != NULL && path > filename) {		/* ~user/path */
1027 		slash = path - filename;
1028 		if (slash > sizeof(user) - 1)
1029 			fatal("tilde_expand_filename: ~username too long");
1030 		memcpy(user, filename, slash);
1031 		user[slash] = '\0';
1032 		if ((pw = getpwnam(user)) == NULL)
1033 			fatal("tilde_expand_filename: No such user %s", user);
1034 	} else if ((pw = getpwuid(uid)) == NULL)	/* ~/path */
1035 		fatal("tilde_expand_filename: No such uid %ld", (long)uid);
1036 
1037 	/* Make sure directory has a trailing '/' */
1038 	len = strlen(pw->pw_dir);
1039 	if (len == 0 || pw->pw_dir[len - 1] != '/')
1040 		sep = "/";
1041 	else
1042 		sep = "";
1043 
1044 	/* Skip leading '/' from specified path */
1045 	if (path != NULL)
1046 		filename = path + 1;
1047 
1048 	if (xasprintf(&ret, "%s%s%s", pw->pw_dir, sep, filename) >= PATH_MAX)
1049 		fatal("tilde_expand_filename: Path too long");
1050 
1051 	return (ret);
1052 }
1053 
1054 /*
1055  * Expand a string with a set of %[char] escapes. A number of escapes may be
1056  * specified as (char *escape_chars, char *replacement) pairs. The list must
1057  * be terminated by a NULL escape_char. Returns replaced string in memory
1058  * allocated by xmalloc.
1059  */
1060 char *
percent_expand(const char * string,...)1061 percent_expand(const char *string, ...)
1062 {
1063 #define EXPAND_MAX_KEYS	16
1064 	u_int num_keys, i;
1065 	struct {
1066 		const char *key;
1067 		const char *repl;
1068 	} keys[EXPAND_MAX_KEYS];
1069 	struct sshbuf *buf;
1070 	va_list ap;
1071 	int r;
1072 	char *ret;
1073 
1074 	if ((buf = sshbuf_new()) == NULL)
1075 		fatal("%s: sshbuf_new failed", __func__);
1076 
1077 	/* Gather keys */
1078 	va_start(ap, string);
1079 	for (num_keys = 0; num_keys < EXPAND_MAX_KEYS; num_keys++) {
1080 		keys[num_keys].key = va_arg(ap, char *);
1081 		if (keys[num_keys].key == NULL)
1082 			break;
1083 		keys[num_keys].repl = va_arg(ap, char *);
1084 		if (keys[num_keys].repl == NULL)
1085 			fatal("%s: NULL replacement", __func__);
1086 	}
1087 	if (num_keys == EXPAND_MAX_KEYS && va_arg(ap, char *) != NULL)
1088 		fatal("%s: too many keys", __func__);
1089 	va_end(ap);
1090 
1091 	/* Expand string */
1092 	for (i = 0; *string != '\0'; string++) {
1093 		if (*string != '%') {
1094  append:
1095 			if ((r = sshbuf_put_u8(buf, *string)) != 0) {
1096 				fatal("%s: sshbuf_put_u8: %s",
1097 				    __func__, ssh_err(r));
1098 			}
1099 			continue;
1100 		}
1101 		string++;
1102 		/* %% case */
1103 		if (*string == '%')
1104 			goto append;
1105 		if (*string == '\0')
1106 			fatal("%s: invalid format", __func__);
1107 		for (i = 0; i < num_keys; i++) {
1108 			if (strchr(keys[i].key, *string) != NULL) {
1109 				if ((r = sshbuf_put(buf, keys[i].repl,
1110 				    strlen(keys[i].repl))) != 0) {
1111 					fatal("%s: sshbuf_put: %s",
1112 					    __func__, ssh_err(r));
1113 				}
1114 				break;
1115 			}
1116 		}
1117 		if (i >= num_keys)
1118 			fatal("%s: unknown key %%%c", __func__, *string);
1119 	}
1120 	if ((ret = sshbuf_dup_string(buf)) == NULL)
1121 		fatal("%s: sshbuf_dup_string failed", __func__);
1122 	sshbuf_free(buf);
1123 	return ret;
1124 #undef EXPAND_MAX_KEYS
1125 }
1126 
1127 int
tun_open(int tun,int mode,char ** ifname)1128 tun_open(int tun, int mode, char **ifname)
1129 {
1130 #if defined(CUSTOM_SYS_TUN_OPEN)
1131 	return (sys_tun_open(tun, mode, ifname));
1132 #elif defined(SSH_TUN_OPENBSD)
1133 	struct ifreq ifr;
1134 	char name[100];
1135 	int fd = -1, sock;
1136 	const char *tunbase = "tun";
1137 
1138 	if (ifname != NULL)
1139 		*ifname = NULL;
1140 
1141 	if (mode == SSH_TUNMODE_ETHERNET)
1142 		tunbase = "tap";
1143 
1144 	/* Open the tunnel device */
1145 	if (tun <= SSH_TUNID_MAX) {
1146 		snprintf(name, sizeof(name), "/dev/%s%d", tunbase, tun);
1147 		fd = open(name, O_RDWR);
1148 	} else if (tun == SSH_TUNID_ANY) {
1149 		for (tun = 100; tun >= 0; tun--) {
1150 			snprintf(name, sizeof(name), "/dev/%s%d",
1151 			    tunbase, tun);
1152 			if ((fd = open(name, O_RDWR)) >= 0)
1153 				break;
1154 		}
1155 	} else {
1156 		debug("%s: invalid tunnel %u", __func__, tun);
1157 		return -1;
1158 	}
1159 
1160 	if (fd == -1) {
1161 		debug("%s: %s open: %s", __func__, name, strerror(errno));
1162 		return -1;
1163 	}
1164 
1165 	debug("%s: %s mode %d fd %d", __func__, name, mode, fd);
1166 
1167 	/* Bring interface up if it is not already */
1168 	snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "%s%d", tunbase, tun);
1169 	if ((sock = socket(PF_UNIX, SOCK_STREAM, 0)) == -1)
1170 		goto failed;
1171 
1172 	if (ioctl(sock, SIOCGIFFLAGS, &ifr) == -1) {
1173 		debug("%s: get interface %s flags: %s", __func__,
1174 		    ifr.ifr_name, strerror(errno));
1175 		goto failed;
1176 	}
1177 
1178 	if (!(ifr.ifr_flags & IFF_UP)) {
1179 		ifr.ifr_flags |= IFF_UP;
1180 		if (ioctl(sock, SIOCSIFFLAGS, &ifr) == -1) {
1181 			debug("%s: activate interface %s: %s", __func__,
1182 			    ifr.ifr_name, strerror(errno));
1183 			goto failed;
1184 		}
1185 	}
1186 
1187 	if (ifname != NULL)
1188 		*ifname = xstrdup(ifr.ifr_name);
1189 
1190 	close(sock);
1191 	return fd;
1192 
1193  failed:
1194 	if (fd >= 0)
1195 		close(fd);
1196 	if (sock >= 0)
1197 		close(sock);
1198 	return -1;
1199 #else
1200 	error("Tunnel interfaces are not supported on this platform");
1201 	return (-1);
1202 #endif
1203 }
1204 
1205 void
sanitise_stdfd(void)1206 sanitise_stdfd(void)
1207 {
1208 	int nullfd, dupfd;
1209 
1210 	if ((nullfd = dupfd = open(_PATH_DEVNULL, O_RDWR)) == -1) {
1211 		fprintf(stderr, "Couldn't open /dev/null: %s\n",
1212 		    strerror(errno));
1213 		exit(1);
1214 	}
1215 	while (++dupfd <= STDERR_FILENO) {
1216 		/* Only populate closed fds. */
1217 		if (fcntl(dupfd, F_GETFL) == -1 && errno == EBADF) {
1218 			if (dup2(nullfd, dupfd) == -1) {
1219 				fprintf(stderr, "dup2: %s\n", strerror(errno));
1220 				exit(1);
1221 			}
1222 		}
1223 	}
1224 	if (nullfd > STDERR_FILENO)
1225 		close(nullfd);
1226 }
1227 
1228 char *
tohex(const void * vp,size_t l)1229 tohex(const void *vp, size_t l)
1230 {
1231 	const u_char *p = (const u_char *)vp;
1232 	char b[3], *r;
1233 	size_t i, hl;
1234 
1235 	if (l > 65536)
1236 		return xstrdup("tohex: length > 65536");
1237 
1238 	hl = l * 2 + 1;
1239 	r = xcalloc(1, hl);
1240 	for (i = 0; i < l; i++) {
1241 		snprintf(b, sizeof(b), "%02x", p[i]);
1242 		strlcat(r, b, hl);
1243 	}
1244 	return (r);
1245 }
1246 
1247 u_int64_t
get_u64(const void * vp)1248 get_u64(const void *vp)
1249 {
1250 	const u_char *p = (const u_char *)vp;
1251 	u_int64_t v;
1252 
1253 	v  = (u_int64_t)p[0] << 56;
1254 	v |= (u_int64_t)p[1] << 48;
1255 	v |= (u_int64_t)p[2] << 40;
1256 	v |= (u_int64_t)p[3] << 32;
1257 	v |= (u_int64_t)p[4] << 24;
1258 	v |= (u_int64_t)p[5] << 16;
1259 	v |= (u_int64_t)p[6] << 8;
1260 	v |= (u_int64_t)p[7];
1261 
1262 	return (v);
1263 }
1264 
1265 u_int32_t
get_u32(const void * vp)1266 get_u32(const void *vp)
1267 {
1268 	const u_char *p = (const u_char *)vp;
1269 	u_int32_t v;
1270 
1271 	v  = (u_int32_t)p[0] << 24;
1272 	v |= (u_int32_t)p[1] << 16;
1273 	v |= (u_int32_t)p[2] << 8;
1274 	v |= (u_int32_t)p[3];
1275 
1276 	return (v);
1277 }
1278 
1279 u_int32_t
get_u32_le(const void * vp)1280 get_u32_le(const void *vp)
1281 {
1282 	const u_char *p = (const u_char *)vp;
1283 	u_int32_t v;
1284 
1285 	v  = (u_int32_t)p[0];
1286 	v |= (u_int32_t)p[1] << 8;
1287 	v |= (u_int32_t)p[2] << 16;
1288 	v |= (u_int32_t)p[3] << 24;
1289 
1290 	return (v);
1291 }
1292 
1293 u_int16_t
get_u16(const void * vp)1294 get_u16(const void *vp)
1295 {
1296 	const u_char *p = (const u_char *)vp;
1297 	u_int16_t v;
1298 
1299 	v  = (u_int16_t)p[0] << 8;
1300 	v |= (u_int16_t)p[1];
1301 
1302 	return (v);
1303 }
1304 
1305 void
put_u64(void * vp,u_int64_t v)1306 put_u64(void *vp, u_int64_t v)
1307 {
1308 	u_char *p = (u_char *)vp;
1309 
1310 	p[0] = (u_char)(v >> 56) & 0xff;
1311 	p[1] = (u_char)(v >> 48) & 0xff;
1312 	p[2] = (u_char)(v >> 40) & 0xff;
1313 	p[3] = (u_char)(v >> 32) & 0xff;
1314 	p[4] = (u_char)(v >> 24) & 0xff;
1315 	p[5] = (u_char)(v >> 16) & 0xff;
1316 	p[6] = (u_char)(v >> 8) & 0xff;
1317 	p[7] = (u_char)v & 0xff;
1318 }
1319 
1320 void
put_u32(void * vp,u_int32_t v)1321 put_u32(void *vp, u_int32_t v)
1322 {
1323 	u_char *p = (u_char *)vp;
1324 
1325 	p[0] = (u_char)(v >> 24) & 0xff;
1326 	p[1] = (u_char)(v >> 16) & 0xff;
1327 	p[2] = (u_char)(v >> 8) & 0xff;
1328 	p[3] = (u_char)v & 0xff;
1329 }
1330 
1331 void
put_u32_le(void * vp,u_int32_t v)1332 put_u32_le(void *vp, u_int32_t v)
1333 {
1334 	u_char *p = (u_char *)vp;
1335 
1336 	p[0] = (u_char)v & 0xff;
1337 	p[1] = (u_char)(v >> 8) & 0xff;
1338 	p[2] = (u_char)(v >> 16) & 0xff;
1339 	p[3] = (u_char)(v >> 24) & 0xff;
1340 }
1341 
1342 void
put_u16(void * vp,u_int16_t v)1343 put_u16(void *vp, u_int16_t v)
1344 {
1345 	u_char *p = (u_char *)vp;
1346 
1347 	p[0] = (u_char)(v >> 8) & 0xff;
1348 	p[1] = (u_char)v & 0xff;
1349 }
1350 
1351 void
ms_subtract_diff(struct timeval * start,int * ms)1352 ms_subtract_diff(struct timeval *start, int *ms)
1353 {
1354 	struct timeval diff, finish;
1355 
1356 	monotime_tv(&finish);
1357 	timersub(&finish, start, &diff);
1358 	*ms -= (diff.tv_sec * 1000) + (diff.tv_usec / 1000);
1359 }
1360 
1361 void
ms_to_timeval(struct timeval * tv,int ms)1362 ms_to_timeval(struct timeval *tv, int ms)
1363 {
1364 	if (ms < 0)
1365 		ms = 0;
1366 	tv->tv_sec = ms / 1000;
1367 	tv->tv_usec = (ms % 1000) * 1000;
1368 }
1369 
1370 void
monotime_ts(struct timespec * ts)1371 monotime_ts(struct timespec *ts)
1372 {
1373 	struct timeval tv;
1374 #if defined(HAVE_CLOCK_GETTIME) && (defined(CLOCK_BOOTTIME) || \
1375     defined(CLOCK_MONOTONIC) || defined(CLOCK_REALTIME))
1376 	static int gettime_failed = 0;
1377 
1378 	if (!gettime_failed) {
1379 # ifdef CLOCK_BOOTTIME
1380 		if (clock_gettime(CLOCK_BOOTTIME, ts) == 0)
1381 			return;
1382 # endif /* CLOCK_BOOTTIME */
1383 # ifdef CLOCK_MONOTONIC
1384 		if (clock_gettime(CLOCK_MONOTONIC, ts) == 0)
1385 			return;
1386 # endif /* CLOCK_MONOTONIC */
1387 # ifdef CLOCK_REALTIME
1388 		/* Not monotonic, but we're almost out of options here. */
1389 		if (clock_gettime(CLOCK_REALTIME, ts) == 0)
1390 			return;
1391 # endif /* CLOCK_REALTIME */
1392 		debug3("clock_gettime: %s", strerror(errno));
1393 		gettime_failed = 1;
1394 	}
1395 #endif /* HAVE_CLOCK_GETTIME && (BOOTTIME || MONOTONIC || REALTIME) */
1396 	gettimeofday(&tv, NULL);
1397 	ts->tv_sec = tv.tv_sec;
1398 	ts->tv_nsec = (long)tv.tv_usec * 1000;
1399 }
1400 
1401 void
monotime_tv(struct timeval * tv)1402 monotime_tv(struct timeval *tv)
1403 {
1404 	struct timespec ts;
1405 
1406 	monotime_ts(&ts);
1407 	tv->tv_sec = ts.tv_sec;
1408 	tv->tv_usec = ts.tv_nsec / 1000;
1409 }
1410 
1411 time_t
monotime(void)1412 monotime(void)
1413 {
1414 	struct timespec ts;
1415 
1416 	monotime_ts(&ts);
1417 	return ts.tv_sec;
1418 }
1419 
1420 double
monotime_double(void)1421 monotime_double(void)
1422 {
1423 	struct timespec ts;
1424 
1425 	monotime_ts(&ts);
1426 	return ts.tv_sec + ((double)ts.tv_nsec / 1000000000);
1427 }
1428 
1429 void
bandwidth_limit_init(struct bwlimit * bw,u_int64_t kbps,size_t buflen)1430 bandwidth_limit_init(struct bwlimit *bw, u_int64_t kbps, size_t buflen)
1431 {
1432 	bw->buflen = buflen;
1433 	bw->rate = kbps;
1434 	bw->thresh = buflen;
1435 	bw->lamt = 0;
1436 	timerclear(&bw->bwstart);
1437 	timerclear(&bw->bwend);
1438 }
1439 
1440 /* Callback from read/write loop to insert bandwidth-limiting delays */
1441 void
bandwidth_limit(struct bwlimit * bw,size_t read_len)1442 bandwidth_limit(struct bwlimit *bw, size_t read_len)
1443 {
1444 	u_int64_t waitlen;
1445 	struct timespec ts, rm;
1446 
1447 	bw->lamt += read_len;
1448 	if (!timerisset(&bw->bwstart)) {
1449 		monotime_tv(&bw->bwstart);
1450 		return;
1451 	}
1452 	if (bw->lamt < bw->thresh)
1453 		return;
1454 
1455 	monotime_tv(&bw->bwend);
1456 	timersub(&bw->bwend, &bw->bwstart, &bw->bwend);
1457 	if (!timerisset(&bw->bwend))
1458 		return;
1459 
1460 	bw->lamt *= 8;
1461 	waitlen = (double)1000000L * bw->lamt / bw->rate;
1462 
1463 	bw->bwstart.tv_sec = waitlen / 1000000L;
1464 	bw->bwstart.tv_usec = waitlen % 1000000L;
1465 
1466 	if (timercmp(&bw->bwstart, &bw->bwend, >)) {
1467 		timersub(&bw->bwstart, &bw->bwend, &bw->bwend);
1468 
1469 		/* Adjust the wait time */
1470 		if (bw->bwend.tv_sec) {
1471 			bw->thresh /= 2;
1472 			if (bw->thresh < bw->buflen / 4)
1473 				bw->thresh = bw->buflen / 4;
1474 		} else if (bw->bwend.tv_usec < 10000) {
1475 			bw->thresh *= 2;
1476 			if (bw->thresh > bw->buflen * 8)
1477 				bw->thresh = bw->buflen * 8;
1478 		}
1479 
1480 		TIMEVAL_TO_TIMESPEC(&bw->bwend, &ts);
1481 		while (nanosleep(&ts, &rm) == -1) {
1482 			if (errno != EINTR)
1483 				break;
1484 			ts = rm;
1485 		}
1486 	}
1487 
1488 	bw->lamt = 0;
1489 	monotime_tv(&bw->bwstart);
1490 }
1491 
1492 /* Make a template filename for mk[sd]temp() */
1493 void
mktemp_proto(char * s,size_t len)1494 mktemp_proto(char *s, size_t len)
1495 {
1496 	const char *tmpdir;
1497 	int r;
1498 
1499 	if ((tmpdir = getenv("TMPDIR")) != NULL) {
1500 		r = snprintf(s, len, "%s/ssh-XXXXXXXXXXXX", tmpdir);
1501 		if (r > 0 && (size_t)r < len)
1502 			return;
1503 	}
1504 	r = snprintf(s, len, "/tmp/ssh-XXXXXXXXXXXX");
1505 	if (r < 0 || (size_t)r >= len)
1506 		fatal("%s: template string too short", __func__);
1507 }
1508 
1509 static const struct {
1510 	const char *name;
1511 	int value;
1512 } ipqos[] = {
1513 	{ "none", INT_MAX },		/* can't use 0 here; that's CS0 */
1514 	{ "af11", IPTOS_DSCP_AF11 },
1515 	{ "af12", IPTOS_DSCP_AF12 },
1516 	{ "af13", IPTOS_DSCP_AF13 },
1517 	{ "af21", IPTOS_DSCP_AF21 },
1518 	{ "af22", IPTOS_DSCP_AF22 },
1519 	{ "af23", IPTOS_DSCP_AF23 },
1520 	{ "af31", IPTOS_DSCP_AF31 },
1521 	{ "af32", IPTOS_DSCP_AF32 },
1522 	{ "af33", IPTOS_DSCP_AF33 },
1523 	{ "af41", IPTOS_DSCP_AF41 },
1524 	{ "af42", IPTOS_DSCP_AF42 },
1525 	{ "af43", IPTOS_DSCP_AF43 },
1526 	{ "cs0", IPTOS_DSCP_CS0 },
1527 	{ "cs1", IPTOS_DSCP_CS1 },
1528 	{ "cs2", IPTOS_DSCP_CS2 },
1529 	{ "cs3", IPTOS_DSCP_CS3 },
1530 	{ "cs4", IPTOS_DSCP_CS4 },
1531 	{ "cs5", IPTOS_DSCP_CS5 },
1532 	{ "cs6", IPTOS_DSCP_CS6 },
1533 	{ "cs7", IPTOS_DSCP_CS7 },
1534 	{ "ef", IPTOS_DSCP_EF },
1535 	{ "lowdelay", IPTOS_LOWDELAY },
1536 	{ "throughput", IPTOS_THROUGHPUT },
1537 	{ "reliability", IPTOS_RELIABILITY },
1538 	{ NULL, -1 }
1539 };
1540 
1541 int
parse_ipqos(const char * cp)1542 parse_ipqos(const char *cp)
1543 {
1544 	u_int i;
1545 	char *ep;
1546 	long val;
1547 
1548 	if (cp == NULL)
1549 		return -1;
1550 	for (i = 0; ipqos[i].name != NULL; i++) {
1551 		if (strcasecmp(cp, ipqos[i].name) == 0)
1552 			return ipqos[i].value;
1553 	}
1554 	/* Try parsing as an integer */
1555 	val = strtol(cp, &ep, 0);
1556 	if (*cp == '\0' || *ep != '\0' || val < 0 || val > 255)
1557 		return -1;
1558 	return val;
1559 }
1560 
1561 const char *
iptos2str(int iptos)1562 iptos2str(int iptos)
1563 {
1564 	int i;
1565 	static char iptos_str[sizeof "0xff"];
1566 
1567 	for (i = 0; ipqos[i].name != NULL; i++) {
1568 		if (ipqos[i].value == iptos)
1569 			return ipqos[i].name;
1570 	}
1571 	snprintf(iptos_str, sizeof iptos_str, "0x%02x", iptos);
1572 	return iptos_str;
1573 }
1574 
1575 void
lowercase(char * s)1576 lowercase(char *s)
1577 {
1578 	for (; *s; s++)
1579 		*s = tolower((u_char)*s);
1580 }
1581 
1582 int
unix_listener(const char * path,int backlog,int unlink_first)1583 unix_listener(const char *path, int backlog, int unlink_first)
1584 {
1585 	struct sockaddr_un sunaddr;
1586 	int saved_errno, sock;
1587 
1588 	memset(&sunaddr, 0, sizeof(sunaddr));
1589 	sunaddr.sun_family = AF_UNIX;
1590 	if (strlcpy(sunaddr.sun_path, path,
1591 	    sizeof(sunaddr.sun_path)) >= sizeof(sunaddr.sun_path)) {
1592 		error("%s: path \"%s\" too long for Unix domain socket",
1593 		    __func__, path);
1594 		errno = ENAMETOOLONG;
1595 		return -1;
1596 	}
1597 
1598 	sock = socket(PF_UNIX, SOCK_STREAM, 0);
1599 	if (sock == -1) {
1600 		saved_errno = errno;
1601 		error("%s: socket: %.100s", __func__, strerror(errno));
1602 		errno = saved_errno;
1603 		return -1;
1604 	}
1605 	if (unlink_first == 1) {
1606 		if (unlink(path) != 0 && errno != ENOENT)
1607 			error("unlink(%s): %.100s", path, strerror(errno));
1608 	}
1609 	if (bind(sock, (struct sockaddr *)&sunaddr, sizeof(sunaddr)) == -1) {
1610 		saved_errno = errno;
1611 		error("%s: cannot bind to path %s: %s",
1612 		    __func__, path, strerror(errno));
1613 		close(sock);
1614 		errno = saved_errno;
1615 		return -1;
1616 	}
1617 	if (listen(sock, backlog) == -1) {
1618 		saved_errno = errno;
1619 		error("%s: cannot listen on path %s: %s",
1620 		    __func__, path, strerror(errno));
1621 		close(sock);
1622 		unlink(path);
1623 		errno = saved_errno;
1624 		return -1;
1625 	}
1626 	return sock;
1627 }
1628 
1629 void
sock_set_v6only(int s)1630 sock_set_v6only(int s)
1631 {
1632 #if defined(IPV6_V6ONLY) && !defined(__OpenBSD__)
1633 	int on = 1;
1634 
1635 	debug3("%s: set socket %d IPV6_V6ONLY", __func__, s);
1636 	if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on)) == -1)
1637 		error("setsockopt IPV6_V6ONLY: %s", strerror(errno));
1638 #endif
1639 }
1640 
1641 /*
1642  * Compares two strings that maybe be NULL. Returns non-zero if strings
1643  * are both NULL or are identical, returns zero otherwise.
1644  */
1645 static int
strcmp_maybe_null(const char * a,const char * b)1646 strcmp_maybe_null(const char *a, const char *b)
1647 {
1648 	if ((a == NULL && b != NULL) || (a != NULL && b == NULL))
1649 		return 0;
1650 	if (a != NULL && strcmp(a, b) != 0)
1651 		return 0;
1652 	return 1;
1653 }
1654 
1655 /*
1656  * Compare two forwards, returning non-zero if they are identical or
1657  * zero otherwise.
1658  */
1659 int
forward_equals(const struct Forward * a,const struct Forward * b)1660 forward_equals(const struct Forward *a, const struct Forward *b)
1661 {
1662 	if (strcmp_maybe_null(a->listen_host, b->listen_host) == 0)
1663 		return 0;
1664 	if (a->listen_port != b->listen_port)
1665 		return 0;
1666 	if (strcmp_maybe_null(a->listen_path, b->listen_path) == 0)
1667 		return 0;
1668 	if (strcmp_maybe_null(a->connect_host, b->connect_host) == 0)
1669 		return 0;
1670 	if (a->connect_port != b->connect_port)
1671 		return 0;
1672 	if (strcmp_maybe_null(a->connect_path, b->connect_path) == 0)
1673 		return 0;
1674 	/* allocated_port and handle are not checked */
1675 	return 1;
1676 }
1677 
1678 /* returns 1 if process is already daemonized, 0 otherwise */
1679 int
daemonized(void)1680 daemonized(void)
1681 {
1682 	int fd;
1683 
1684 	if ((fd = open(_PATH_TTY, O_RDONLY | O_NOCTTY)) >= 0) {
1685 		close(fd);
1686 		return 0;	/* have controlling terminal */
1687 	}
1688 	if (getppid() != 1)
1689 		return 0;	/* parent is not init */
1690 	if (getsid(0) != getpid())
1691 		return 0;	/* not session leader */
1692 	debug3("already daemonized");
1693 	return 1;
1694 }
1695 
1696 
1697 /*
1698  * Splits 's' into an argument vector. Handles quoted string and basic
1699  * escape characters (\\, \", \'). Caller must free the argument vector
1700  * and its members.
1701  */
1702 int
argv_split(const char * s,int * argcp,char *** argvp)1703 argv_split(const char *s, int *argcp, char ***argvp)
1704 {
1705 	int r = SSH_ERR_INTERNAL_ERROR;
1706 	int argc = 0, quote, i, j;
1707 	char *arg, **argv = xcalloc(1, sizeof(*argv));
1708 
1709 	*argvp = NULL;
1710 	*argcp = 0;
1711 
1712 	for (i = 0; s[i] != '\0'; i++) {
1713 		/* Skip leading whitespace */
1714 		if (s[i] == ' ' || s[i] == '\t')
1715 			continue;
1716 
1717 		/* Start of a token */
1718 		quote = 0;
1719 		if (s[i] == '\\' &&
1720 		    (s[i + 1] == '\'' || s[i + 1] == '\"' || s[i + 1] == '\\'))
1721 			i++;
1722 		else if (s[i] == '\'' || s[i] == '"')
1723 			quote = s[i++];
1724 
1725 		argv = xreallocarray(argv, (argc + 2), sizeof(*argv));
1726 		arg = argv[argc++] = xcalloc(1, strlen(s + i) + 1);
1727 		argv[argc] = NULL;
1728 
1729 		/* Copy the token in, removing escapes */
1730 		for (j = 0; s[i] != '\0'; i++) {
1731 			if (s[i] == '\\') {
1732 				if (s[i + 1] == '\'' ||
1733 				    s[i + 1] == '\"' ||
1734 				    s[i + 1] == '\\') {
1735 					i++; /* Skip '\' */
1736 					arg[j++] = s[i];
1737 				} else {
1738 					/* Unrecognised escape */
1739 					arg[j++] = s[i];
1740 				}
1741 			} else if (quote == 0 && (s[i] == ' ' || s[i] == '\t'))
1742 				break; /* done */
1743 			else if (quote != 0 && s[i] == quote)
1744 				break; /* done */
1745 			else
1746 				arg[j++] = s[i];
1747 		}
1748 		if (s[i] == '\0') {
1749 			if (quote != 0) {
1750 				/* Ran out of string looking for close quote */
1751 				r = SSH_ERR_INVALID_FORMAT;
1752 				goto out;
1753 			}
1754 			break;
1755 		}
1756 	}
1757 	/* Success */
1758 	*argcp = argc;
1759 	*argvp = argv;
1760 	argc = 0;
1761 	argv = NULL;
1762 	r = 0;
1763  out:
1764 	if (argc != 0 && argv != NULL) {
1765 		for (i = 0; i < argc; i++)
1766 			free(argv[i]);
1767 		free(argv);
1768 	}
1769 	return r;
1770 }
1771 
1772 /*
1773  * Reassemble an argument vector into a string, quoting and escaping as
1774  * necessary. Caller must free returned string.
1775  */
1776 char *
argv_assemble(int argc,char ** argv)1777 argv_assemble(int argc, char **argv)
1778 {
1779 	int i, j, ws, r;
1780 	char c, *ret;
1781 	struct sshbuf *buf, *arg;
1782 
1783 	if ((buf = sshbuf_new()) == NULL || (arg = sshbuf_new()) == NULL)
1784 		fatal("%s: sshbuf_new failed", __func__);
1785 
1786 	for (i = 0; i < argc; i++) {
1787 		ws = 0;
1788 		sshbuf_reset(arg);
1789 		for (j = 0; argv[i][j] != '\0'; j++) {
1790 			r = 0;
1791 			c = argv[i][j];
1792 			switch (c) {
1793 			case ' ':
1794 			case '\t':
1795 				ws = 1;
1796 				r = sshbuf_put_u8(arg, c);
1797 				break;
1798 			case '\\':
1799 			case '\'':
1800 			case '"':
1801 				if ((r = sshbuf_put_u8(arg, '\\')) != 0)
1802 					break;
1803 				/* FALLTHROUGH */
1804 			default:
1805 				r = sshbuf_put_u8(arg, c);
1806 				break;
1807 			}
1808 			if (r != 0)
1809 				fatal("%s: sshbuf_put_u8: %s",
1810 				    __func__, ssh_err(r));
1811 		}
1812 		if ((i != 0 && (r = sshbuf_put_u8(buf, ' ')) != 0) ||
1813 		    (ws != 0 && (r = sshbuf_put_u8(buf, '"')) != 0) ||
1814 		    (r = sshbuf_putb(buf, arg)) != 0 ||
1815 		    (ws != 0 && (r = sshbuf_put_u8(buf, '"')) != 0))
1816 			fatal("%s: buffer error: %s", __func__, ssh_err(r));
1817 	}
1818 	if ((ret = malloc(sshbuf_len(buf) + 1)) == NULL)
1819 		fatal("%s: malloc failed", __func__);
1820 	memcpy(ret, sshbuf_ptr(buf), sshbuf_len(buf));
1821 	ret[sshbuf_len(buf)] = '\0';
1822 	sshbuf_free(buf);
1823 	sshbuf_free(arg);
1824 	return ret;
1825 }
1826 
1827 /* Returns 0 if pid exited cleanly, non-zero otherwise */
1828 int
exited_cleanly(pid_t pid,const char * tag,const char * cmd,int quiet)1829 exited_cleanly(pid_t pid, const char *tag, const char *cmd, int quiet)
1830 {
1831 	int status;
1832 
1833 	while (waitpid(pid, &status, 0) == -1) {
1834 		if (errno != EINTR) {
1835 			error("%s: waitpid: %s", tag, strerror(errno));
1836 			return -1;
1837 		}
1838 	}
1839 	if (WIFSIGNALED(status)) {
1840 		error("%s %s exited on signal %d", tag, cmd, WTERMSIG(status));
1841 		return -1;
1842 	} else if (WEXITSTATUS(status) != 0) {
1843 		do_log2(quiet ? SYSLOG_LEVEL_DEBUG1 : SYSLOG_LEVEL_INFO,
1844 		    "%s %s failed, status %d", tag, cmd, WEXITSTATUS(status));
1845 		return -1;
1846 	}
1847 	return 0;
1848 }
1849 
1850 /*
1851  * Check a given path for security. This is defined as all components
1852  * of the path to the file must be owned by either the owner of
1853  * of the file or root and no directories must be group or world writable.
1854  *
1855  * XXX Should any specific check be done for sym links ?
1856  *
1857  * Takes a file name, its stat information (preferably from fstat() to
1858  * avoid races), the uid of the expected owner, their home directory and an
1859  * error buffer plus max size as arguments.
1860  *
1861  * Returns 0 on success and -1 on failure
1862  */
1863 int
safe_path(const char * name,struct stat * stp,const char * pw_dir,uid_t uid,char * err,size_t errlen)1864 safe_path(const char *name, struct stat *stp, const char *pw_dir,
1865     uid_t uid, char *err, size_t errlen)
1866 {
1867 	char buf[PATH_MAX], homedir[PATH_MAX];
1868 	char *cp;
1869 	int comparehome = 0;
1870 	struct stat st;
1871 
1872 	if (realpath(name, buf) == NULL) {
1873 		snprintf(err, errlen, "realpath %s failed: %s", name,
1874 		    strerror(errno));
1875 		return -1;
1876 	}
1877 	if (pw_dir != NULL && realpath(pw_dir, homedir) != NULL)
1878 		comparehome = 1;
1879 
1880 	if (!S_ISREG(stp->st_mode)) {
1881 		snprintf(err, errlen, "%s is not a regular file", buf);
1882 		return -1;
1883 	}
1884 	if ((!platform_sys_dir_uid(stp->st_uid) && stp->st_uid != uid) ||
1885 	    (stp->st_mode & 022) != 0) {
1886 		snprintf(err, errlen, "bad ownership or modes for file %s",
1887 		    buf);
1888 		return -1;
1889 	}
1890 
1891 	/* for each component of the canonical path, walking upwards */
1892 	for (;;) {
1893 		if ((cp = dirname(buf)) == NULL) {
1894 			snprintf(err, errlen, "dirname() failed");
1895 			return -1;
1896 		}
1897 		strlcpy(buf, cp, sizeof(buf));
1898 
1899 		if (stat(buf, &st) == -1 ||
1900 		    (!platform_sys_dir_uid(st.st_uid) && st.st_uid != uid) ||
1901 		    (st.st_mode & 022) != 0) {
1902 			snprintf(err, errlen,
1903 			    "bad ownership or modes for directory %s", buf);
1904 			return -1;
1905 		}
1906 
1907 		/* If are past the homedir then we can stop */
1908 		if (comparehome && strcmp(homedir, buf) == 0)
1909 			break;
1910 
1911 		/*
1912 		 * dirname should always complete with a "/" path,
1913 		 * but we can be paranoid and check for "." too
1914 		 */
1915 		if ((strcmp("/", buf) == 0) || (strcmp(".", buf) == 0))
1916 			break;
1917 	}
1918 	return 0;
1919 }
1920 
1921 /*
1922  * Version of safe_path() that accepts an open file descriptor to
1923  * avoid races.
1924  *
1925  * Returns 0 on success and -1 on failure
1926  */
1927 int
safe_path_fd(int fd,const char * file,struct passwd * pw,char * err,size_t errlen)1928 safe_path_fd(int fd, const char *file, struct passwd *pw,
1929     char *err, size_t errlen)
1930 {
1931 	struct stat st;
1932 
1933 	/* check the open file to avoid races */
1934 	if (fstat(fd, &st) == -1) {
1935 		snprintf(err, errlen, "cannot stat file %s: %s",
1936 		    file, strerror(errno));
1937 		return -1;
1938 	}
1939 	return safe_path(file, &st, pw->pw_dir, pw->pw_uid, err, errlen);
1940 }
1941 
1942 /*
1943  * Sets the value of the given variable in the environment.  If the variable
1944  * already exists, its value is overridden.
1945  */
1946 void
child_set_env(char *** envp,u_int * envsizep,const char * name,const char * value)1947 child_set_env(char ***envp, u_int *envsizep, const char *name,
1948 	const char *value)
1949 {
1950 	char **env;
1951 	u_int envsize;
1952 	u_int i, namelen;
1953 
1954 	if (strchr(name, '=') != NULL) {
1955 		error("Invalid environment variable \"%.100s\"", name);
1956 		return;
1957 	}
1958 
1959 	/*
1960 	 * If we're passed an uninitialized list, allocate a single null
1961 	 * entry before continuing.
1962 	 */
1963 	if (*envp == NULL && *envsizep == 0) {
1964 		*envp = xmalloc(sizeof(char *));
1965 		*envp[0] = NULL;
1966 		*envsizep = 1;
1967 	}
1968 
1969 	/*
1970 	 * Find the slot where the value should be stored.  If the variable
1971 	 * already exists, we reuse the slot; otherwise we append a new slot
1972 	 * at the end of the array, expanding if necessary.
1973 	 */
1974 	env = *envp;
1975 	namelen = strlen(name);
1976 	for (i = 0; env[i]; i++)
1977 		if (strncmp(env[i], name, namelen) == 0 && env[i][namelen] == '=')
1978 			break;
1979 	if (env[i]) {
1980 		/* Reuse the slot. */
1981 		free(env[i]);
1982 	} else {
1983 		/* New variable.  Expand if necessary. */
1984 		envsize = *envsizep;
1985 		if (i >= envsize - 1) {
1986 			if (envsize >= 1000)
1987 				fatal("child_set_env: too many env vars");
1988 			envsize += 50;
1989 			env = (*envp) = xreallocarray(env, envsize, sizeof(char *));
1990 			*envsizep = envsize;
1991 		}
1992 		/* Need to set the NULL pointer at end of array beyond the new slot. */
1993 		env[i + 1] = NULL;
1994 	}
1995 
1996 	/* Allocate space and format the variable in the appropriate slot. */
1997 	/* XXX xasprintf */
1998 	env[i] = xmalloc(strlen(name) + 1 + strlen(value) + 1);
1999 	snprintf(env[i], strlen(name) + 1 + strlen(value) + 1, "%s=%s", name, value);
2000 }
2001 
2002 /*
2003  * Check and optionally lowercase a domain name, also removes trailing '.'
2004  * Returns 1 on success and 0 on failure, storing an error message in errstr.
2005  */
2006 int
valid_domain(char * name,int makelower,const char ** errstr)2007 valid_domain(char *name, int makelower, const char **errstr)
2008 {
2009 	size_t i, l = strlen(name);
2010 	u_char c, last = '\0';
2011 	static char errbuf[256];
2012 
2013 	if (l == 0) {
2014 		strlcpy(errbuf, "empty domain name", sizeof(errbuf));
2015 		goto bad;
2016 	}
2017 	if (!isalpha((u_char)name[0]) && !isdigit((u_char)name[0])) {
2018 		snprintf(errbuf, sizeof(errbuf), "domain name \"%.100s\" "
2019 		    "starts with invalid character", name);
2020 		goto bad;
2021 	}
2022 	for (i = 0; i < l; i++) {
2023 		c = tolower((u_char)name[i]);
2024 		if (makelower)
2025 			name[i] = (char)c;
2026 		if (last == '.' && c == '.') {
2027 			snprintf(errbuf, sizeof(errbuf), "domain name "
2028 			    "\"%.100s\" contains consecutive separators", name);
2029 			goto bad;
2030 		}
2031 		if (c != '.' && c != '-' && !isalnum(c) &&
2032 		    c != '_') /* technically invalid, but common */ {
2033 			snprintf(errbuf, sizeof(errbuf), "domain name "
2034 			    "\"%.100s\" contains invalid characters", name);
2035 			goto bad;
2036 		}
2037 		last = c;
2038 	}
2039 	if (name[l - 1] == '.')
2040 		name[l - 1] = '\0';
2041 	if (errstr != NULL)
2042 		*errstr = NULL;
2043 	return 1;
2044 bad:
2045 	if (errstr != NULL)
2046 		*errstr = errbuf;
2047 	return 0;
2048 }
2049 
2050 /*
2051  * Verify that a environment variable name (not including initial '$') is
2052  * valid; consisting of one or more alphanumeric or underscore characters only.
2053  * Returns 1 on valid, 0 otherwise.
2054  */
2055 int
valid_env_name(const char * name)2056 valid_env_name(const char *name)
2057 {
2058 	const char *cp;
2059 
2060 	if (name[0] == '\0')
2061 		return 0;
2062 	for (cp = name; *cp != '\0'; cp++) {
2063 		if (!isalnum((u_char)*cp) && *cp != '_')
2064 			return 0;
2065 	}
2066 	return 1;
2067 }
2068 
2069 const char *
atoi_err(const char * nptr,int * val)2070 atoi_err(const char *nptr, int *val)
2071 {
2072 	const char *errstr = NULL;
2073 	long long num;
2074 
2075 	if (nptr == NULL || *nptr == '\0')
2076 		return "missing";
2077 	num = strtonum(nptr, 0, INT_MAX, &errstr);
2078 	if (errstr == NULL)
2079 		*val = (int)num;
2080 	return errstr;
2081 }
2082 
2083 int
parse_absolute_time(const char * s,uint64_t * tp)2084 parse_absolute_time(const char *s, uint64_t *tp)
2085 {
2086 	struct tm tm;
2087 	time_t tt;
2088 	char buf[32], *fmt;
2089 
2090 	*tp = 0;
2091 
2092 	/*
2093 	 * POSIX strptime says "The application shall ensure that there
2094 	 * is white-space or other non-alphanumeric characters between
2095 	 * any two conversion specifications" so arrange things this way.
2096 	 */
2097 	switch (strlen(s)) {
2098 	case 8: /* YYYYMMDD */
2099 		fmt = "%Y-%m-%d";
2100 		snprintf(buf, sizeof(buf), "%.4s-%.2s-%.2s", s, s + 4, s + 6);
2101 		break;
2102 	case 12: /* YYYYMMDDHHMM */
2103 		fmt = "%Y-%m-%dT%H:%M";
2104 		snprintf(buf, sizeof(buf), "%.4s-%.2s-%.2sT%.2s:%.2s",
2105 		    s, s + 4, s + 6, s + 8, s + 10);
2106 		break;
2107 	case 14: /* YYYYMMDDHHMMSS */
2108 		fmt = "%Y-%m-%dT%H:%M:%S";
2109 		snprintf(buf, sizeof(buf), "%.4s-%.2s-%.2sT%.2s:%.2s:%.2s",
2110 		    s, s + 4, s + 6, s + 8, s + 10, s + 12);
2111 		break;
2112 	default:
2113 		return SSH_ERR_INVALID_FORMAT;
2114 	}
2115 
2116 	memset(&tm, 0, sizeof(tm));
2117 	if (strptime(buf, fmt, &tm) == NULL)
2118 		return SSH_ERR_INVALID_FORMAT;
2119 	if ((tt = mktime(&tm)) < 0)
2120 		return SSH_ERR_INVALID_FORMAT;
2121 	/* success */
2122 	*tp = (uint64_t)tt;
2123 	return 0;
2124 }
2125 
2126 void
format_absolute_time(uint64_t t,char * buf,size_t len)2127 format_absolute_time(uint64_t t, char *buf, size_t len)
2128 {
2129 	time_t tt = t > INT_MAX ? INT_MAX : t; /* XXX revisit in 2038 :P */
2130 	struct tm tm;
2131 
2132 	localtime_r(&tt, &tm);
2133 	strftime(buf, len, "%Y-%m-%dT%H:%M:%S", &tm);
2134 }
2135 
2136 /* check if path is absolute */
2137 int
path_absolute(const char * path)2138 path_absolute(const char *path)
2139 {
2140 	return (*path == '/') ? 1 : 0;
2141 }
2142 
2143 void
skip_space(char ** cpp)2144 skip_space(char **cpp)
2145 {
2146 	char *cp;
2147 
2148 	for (cp = *cpp; *cp == ' ' || *cp == '\t'; cp++)
2149 		;
2150 	*cpp = cp;
2151 }
2152 
2153 /* authorized_key-style options parsing helpers */
2154 
2155 /*
2156  * Match flag 'opt' in *optsp, and if allow_negate is set then also match
2157  * 'no-opt'. Returns -1 if option not matched, 1 if option matches or 0
2158  * if negated option matches.
2159  * If the option or negated option matches, then *optsp is updated to
2160  * point to the first character after the option.
2161  */
2162 int
opt_flag(const char * opt,int allow_negate,const char ** optsp)2163 opt_flag(const char *opt, int allow_negate, const char **optsp)
2164 {
2165 	size_t opt_len = strlen(opt);
2166 	const char *opts = *optsp;
2167 	int negate = 0;
2168 
2169 	if (allow_negate && strncasecmp(opts, "no-", 3) == 0) {
2170 		opts += 3;
2171 		negate = 1;
2172 	}
2173 	if (strncasecmp(opts, opt, opt_len) == 0) {
2174 		*optsp = opts + opt_len;
2175 		return negate ? 0 : 1;
2176 	}
2177 	return -1;
2178 }
2179 
2180 char *
opt_dequote(const char ** sp,const char ** errstrp)2181 opt_dequote(const char **sp, const char **errstrp)
2182 {
2183 	const char *s = *sp;
2184 	char *ret;
2185 	size_t i;
2186 
2187 	*errstrp = NULL;
2188 	if (*s != '"') {
2189 		*errstrp = "missing start quote";
2190 		return NULL;
2191 	}
2192 	s++;
2193 	if ((ret = malloc(strlen((s)) + 1)) == NULL) {
2194 		*errstrp = "memory allocation failed";
2195 		return NULL;
2196 	}
2197 	for (i = 0; *s != '\0' && *s != '"';) {
2198 		if (s[0] == '\\' && s[1] == '"')
2199 			s++;
2200 		ret[i++] = *s++;
2201 	}
2202 	if (*s == '\0') {
2203 		*errstrp = "missing end quote";
2204 		free(ret);
2205 		return NULL;
2206 	}
2207 	ret[i] = '\0';
2208 	s++;
2209 	*sp = s;
2210 	return ret;
2211 }
2212 
2213 int
opt_match(const char ** opts,const char * term)2214 opt_match(const char **opts, const char *term)
2215 {
2216 	if (strncasecmp((*opts), term, strlen(term)) == 0 &&
2217 	    (*opts)[strlen(term)] == '=') {
2218 		*opts += strlen(term) + 1;
2219 		return 1;
2220 	}
2221 	return 0;
2222 }
2223 
2224