blob: 919b04e6b7a99bc2133f0230ef0dff9c71990c5b [file] [log] [blame]
Damien Miller2cd62932010-12-01 11:50:35 +11001/* $OpenBSD: misc.c,v 1.84 2010/11/21 01:01:13 djm Exp $ */
Damien Millere4340be2000-09-16 13:29:08 +11002/*
3 * Copyright (c) 2000 Markus Friedl. All rights reserved.
Damien Miller3f941882006-03-31 23:13:02 +11004 * Copyright (c) 2005,2006 Damien Miller. All rights reserved.
Damien Millere4340be2000-09-16 13:29:08 +11005 *
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
Damien Miller61c51502000-08-18 14:01:04 +100027#include "includes.h"
Damien Miller17e91c02006-03-15 11:28:34 +110028
Damien Miller9f2abc42006-07-10 20:53:08 +100029#include <sys/types.h>
Damien Millerd7834352006-08-05 12:39:39 +100030#include <sys/ioctl.h>
Damien Miller8ec8c3e2006-07-10 20:35:38 +100031#include <sys/socket.h>
Damien Miller8dbffe72006-08-05 11:02:17 +100032#include <sys/param.h>
Damien Miller8ec8c3e2006-07-10 20:35:38 +100033
Darren Tucker5d196262006-07-12 22:15:16 +100034#include <stdarg.h>
Damien Millera7a73ee2006-08-05 11:37:59 +100035#include <stdio.h>
Damien Millere7a1e5c2006-08-05 11:34:19 +100036#include <stdlib.h>
Damien Millere3476ed2006-07-24 14:13:33 +100037#include <string.h>
Damien Miller1708cb72011-01-13 12:21:34 +110038#include <time.h>
Damien Millere6b3b612006-07-24 14:01:23 +100039#include <unistd.h>
Darren Tucker5d196262006-07-12 22:15:16 +100040
Damien Miller8ec8c3e2006-07-10 20:35:38 +100041#include <netinet/in.h>
Damien Miller0dac6fb2010-11-20 15:19:38 +110042#include <netinet/in_systm.h>
43#include <netinet/ip.h>
Damien Miller3a4051e2006-03-15 11:19:42 +110044#include <netinet/tcp.h>
Damien Miller8ec8c3e2006-07-10 20:35:38 +100045
Darren Tucker39972492006-07-12 22:22:46 +100046#include <errno.h>
Damien Miller57cf6382006-07-10 21:13:46 +100047#include <fcntl.h>
Darren Tucker4abde772007-12-29 02:43:51 +110048#include <netdb.h>
Damien Miller03e20032006-03-15 11:16:59 +110049#ifdef HAVE_PATHS_H
Damien Millera9263d02006-03-15 11:18:26 +110050# include <paths.h>
Damien Miller9f2abc42006-07-10 20:53:08 +100051#include <pwd.h>
Damien Miller03e20032006-03-15 11:16:59 +110052#endif
Damien Miller3beb8522006-01-02 23:40:10 +110053#ifdef SSH_TUN_OPENBSD
54#include <net/if.h>
55#endif
Damien Miller61c51502000-08-18 14:01:04 +100056
Damien Millerd7834352006-08-05 12:39:39 +100057#include "xmalloc.h"
Kevin Stevesb6e773a2001-02-04 13:20:36 +000058#include "misc.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000059#include "log.h"
Darren Tuckerda345532006-07-10 23:04:19 +100060#include "ssh.h"
Damien Miller61c51502000-08-18 14:01:04 +100061
Ben Lindstrom4cc240d2001-07-04 04:46:56 +000062/* remove newline at end of string */
Damien Miller61c51502000-08-18 14:01:04 +100063char *
64chop(char *s)
65{
66 char *t = s;
67 while (*t) {
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +000068 if (*t == '\n' || *t == '\r') {
Damien Miller61c51502000-08-18 14:01:04 +100069 *t = '\0';
70 return s;
71 }
72 t++;
73 }
74 return s;
75
76}
77
Ben Lindstrom4cc240d2001-07-04 04:46:56 +000078/* set/unset filedescriptor to non-blocking */
Damien Miller232711f2004-06-15 10:35:30 +100079int
Damien Miller61c51502000-08-18 14:01:04 +100080set_nonblock(int fd)
81{
82 int val;
Ben Lindstromc93e84c2001-05-12 00:08:37 +000083
Damien Miller61c51502000-08-18 14:01:04 +100084 val = fcntl(fd, F_GETFL, 0);
85 if (val < 0) {
86 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +100087 return (-1);
Damien Miller61c51502000-08-18 14:01:04 +100088 }
Damien Miller69b69aa2000-10-28 14:19:58 +110089 if (val & O_NONBLOCK) {
Damien Miller232711f2004-06-15 10:35:30 +100090 debug3("fd %d is O_NONBLOCK", fd);
91 return (0);
Damien Miller69b69aa2000-10-28 14:19:58 +110092 }
Damien Milleref095ce2003-05-14 13:41:39 +100093 debug2("fd %d setting O_NONBLOCK", fd);
Damien Miller61c51502000-08-18 14:01:04 +100094 val |= O_NONBLOCK;
Damien Miller232711f2004-06-15 10:35:30 +100095 if (fcntl(fd, F_SETFL, val) == -1) {
96 debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s", fd,
97 strerror(errno));
98 return (-1);
99 }
100 return (0);
Damien Miller61c51502000-08-18 14:01:04 +1000101}
102
Damien Miller232711f2004-06-15 10:35:30 +1000103int
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000104unset_nonblock(int fd)
105{
106 int val;
107
108 val = fcntl(fd, F_GETFL, 0);
109 if (val < 0) {
110 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +1000111 return (-1);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000112 }
113 if (!(val & O_NONBLOCK)) {
Damien Miller232711f2004-06-15 10:35:30 +1000114 debug3("fd %d is not O_NONBLOCK", fd);
115 return (0);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000116 }
Ben Lindstrom352b1c22001-06-21 03:04:37 +0000117 debug("fd %d clearing O_NONBLOCK", fd);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000118 val &= ~O_NONBLOCK;
Damien Miller232711f2004-06-15 10:35:30 +1000119 if (fcntl(fd, F_SETFL, val) == -1) {
120 debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s",
Ben Lindstrom84fcb312002-03-05 01:48:09 +0000121 fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +1000122 return (-1);
123 }
124 return (0);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000125}
126
Darren Tucker4abde772007-12-29 02:43:51 +1100127const char *
128ssh_gai_strerror(int gaierr)
129{
Darren Tucker912428c2008-01-01 20:33:35 +1100130 if (gaierr == EAI_SYSTEM)
131 return strerror(errno);
132 return gai_strerror(gaierr);
Darren Tucker4abde772007-12-29 02:43:51 +1100133}
134
Damien Miller398e1cf2002-02-05 11:52:13 +1100135/* disable nagle on socket */
136void
137set_nodelay(int fd)
138{
Ben Lindstrome86de512002-03-05 01:28:14 +0000139 int opt;
140 socklen_t optlen;
Damien Miller398e1cf2002-02-05 11:52:13 +1100141
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000142 optlen = sizeof opt;
143 if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) {
Darren Tucker6db8f932003-11-03 20:07:14 +1100144 debug("getsockopt TCP_NODELAY: %.100s", strerror(errno));
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000145 return;
146 }
147 if (opt == 1) {
148 debug2("fd %d is TCP_NODELAY", fd);
149 return;
150 }
151 opt = 1;
Ben Lindstrom1d568f92002-12-23 02:44:36 +0000152 debug2("fd %d setting TCP_NODELAY", fd);
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000153 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1)
Damien Miller398e1cf2002-02-05 11:52:13 +1100154 error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
155}
156
Damien Miller61c51502000-08-18 14:01:04 +1000157/* Characters considered whitespace in strsep calls. */
158#define WHITESPACE " \t\r\n"
Damien Miller306d1182006-03-15 12:05:59 +1100159#define QUOTE "\""
Damien Miller61c51502000-08-18 14:01:04 +1000160
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000161/* return next token in configuration line */
Damien Miller61c51502000-08-18 14:01:04 +1000162char *
163strdelim(char **s)
164{
165 char *old;
166 int wspace = 0;
167
168 if (*s == NULL)
169 return NULL;
170
171 old = *s;
172
Damien Miller306d1182006-03-15 12:05:59 +1100173 *s = strpbrk(*s, WHITESPACE QUOTE "=");
Damien Miller61c51502000-08-18 14:01:04 +1000174 if (*s == NULL)
175 return (old);
176
Damien Miller306d1182006-03-15 12:05:59 +1100177 if (*s[0] == '\"') {
178 memmove(*s, *s + 1, strlen(*s)); /* move nul too */
179 /* Find matching quote */
180 if ((*s = strpbrk(*s, QUOTE)) == NULL) {
181 return (NULL); /* no matching quote */
182 } else {
183 *s[0] = '\0';
Damien Miller9308fc72010-07-16 13:56:01 +1000184 *s += strspn(*s + 1, WHITESPACE) + 1;
Damien Miller306d1182006-03-15 12:05:59 +1100185 return (old);
186 }
187 }
188
Damien Miller61c51502000-08-18 14:01:04 +1000189 /* Allow only one '=' to be skipped */
190 if (*s[0] == '=')
191 wspace = 1;
192 *s[0] = '\0';
193
Damien Miller306d1182006-03-15 12:05:59 +1100194 /* Skip any extra whitespace after first token */
Damien Miller61c51502000-08-18 14:01:04 +1000195 *s += strspn(*s + 1, WHITESPACE) + 1;
196 if (*s[0] == '=' && !wspace)
197 *s += strspn(*s + 1, WHITESPACE) + 1;
198
199 return (old);
200}
Kevin Stevesb6e773a2001-02-04 13:20:36 +0000201
Ben Lindstrom086cf212001-03-05 05:56:40 +0000202struct passwd *
203pwcopy(struct passwd *pw)
204{
Damien Miller07d86be2006-03-26 14:19:21 +1100205 struct passwd *copy = xcalloc(1, sizeof(*copy));
Ben Lindstrom40304422001-03-05 06:22:01 +0000206
Ben Lindstrom086cf212001-03-05 05:56:40 +0000207 copy->pw_name = xstrdup(pw->pw_name);
208 copy->pw_passwd = xstrdup(pw->pw_passwd);
Ben Lindstrom40304422001-03-05 06:22:01 +0000209 copy->pw_gecos = xstrdup(pw->pw_gecos);
Ben Lindstrom086cf212001-03-05 05:56:40 +0000210 copy->pw_uid = pw->pw_uid;
211 copy->pw_gid = pw->pw_gid;
Kevin Steves82456952001-06-22 21:14:18 +0000212#ifdef HAVE_PW_EXPIRE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000213 copy->pw_expire = pw->pw_expire;
Kevin Steves82456952001-06-22 21:14:18 +0000214#endif
215#ifdef HAVE_PW_CHANGE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000216 copy->pw_change = pw->pw_change;
Kevin Steves82456952001-06-22 21:14:18 +0000217#endif
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000218#ifdef HAVE_PW_CLASS_IN_PASSWD
Ben Lindstrom086cf212001-03-05 05:56:40 +0000219 copy->pw_class = xstrdup(pw->pw_class);
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000220#endif
Ben Lindstrom086cf212001-03-05 05:56:40 +0000221 copy->pw_dir = xstrdup(pw->pw_dir);
222 copy->pw_shell = xstrdup(pw->pw_shell);
223 return copy;
224}
225
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000226/*
227 * Convert ASCII string to TCP/IP port number.
Damien Miller3dc71ad2009-01-28 16:31:22 +1100228 * Port must be >=0 and <=65535.
229 * Return -1 if invalid.
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000230 */
231int
232a2port(const char *s)
Ben Lindstrom19066a12001-04-12 23:39:26 +0000233{
Damien Miller3dc71ad2009-01-28 16:31:22 +1100234 long long port;
235 const char *errstr;
Ben Lindstrom19066a12001-04-12 23:39:26 +0000236
Damien Miller3dc71ad2009-01-28 16:31:22 +1100237 port = strtonum(s, 0, 65535, &errstr);
238 if (errstr != NULL)
239 return -1;
240 return (int)port;
Ben Lindstrom19066a12001-04-12 23:39:26 +0000241}
242
Damien Millerd27b9472005-12-13 19:29:02 +1100243int
244a2tun(const char *s, int *remote)
245{
246 const char *errstr = NULL;
247 char *sp, *ep;
248 int tun;
249
250 if (remote != NULL) {
Damien Miller7b58e802005-12-13 19:33:19 +1100251 *remote = SSH_TUNID_ANY;
Damien Millerd27b9472005-12-13 19:29:02 +1100252 sp = xstrdup(s);
253 if ((ep = strchr(sp, ':')) == NULL) {
254 xfree(sp);
255 return (a2tun(s, NULL));
256 }
257 ep[0] = '\0'; ep++;
258 *remote = a2tun(ep, NULL);
259 tun = a2tun(sp, NULL);
260 xfree(sp);
Damien Miller7b58e802005-12-13 19:33:19 +1100261 return (*remote == SSH_TUNID_ERR ? *remote : tun);
Damien Millerd27b9472005-12-13 19:29:02 +1100262 }
263
264 if (strcasecmp(s, "any") == 0)
Damien Miller7b58e802005-12-13 19:33:19 +1100265 return (SSH_TUNID_ANY);
Damien Millerd27b9472005-12-13 19:29:02 +1100266
Damien Miller7b58e802005-12-13 19:33:19 +1100267 tun = strtonum(s, 0, SSH_TUNID_MAX, &errstr);
268 if (errstr != NULL)
269 return (SSH_TUNID_ERR);
Damien Millerd27b9472005-12-13 19:29:02 +1100270
271 return (tun);
272}
273
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000274#define SECONDS 1
275#define MINUTES (SECONDS * 60)
276#define HOURS (MINUTES * 60)
277#define DAYS (HOURS * 24)
278#define WEEKS (DAYS * 7)
279
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000280/*
281 * Convert a time string into seconds; format is
282 * a sequence of:
283 * time[qualifier]
284 *
285 * Valid time qualifiers are:
286 * <none> seconds
287 * s|S seconds
288 * m|M minutes
289 * h|H hours
290 * d|D days
291 * w|W weeks
292 *
293 * Examples:
294 * 90m 90 minutes
295 * 1h30m 90 minutes
296 * 2d 2 days
297 * 1w 1 week
298 *
299 * Return -1 if time string is invalid.
300 */
301long
302convtime(const char *s)
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000303{
304 long total, secs;
305 const char *p;
306 char *endp;
307
308 errno = 0;
309 total = 0;
310 p = s;
311
312 if (p == NULL || *p == '\0')
313 return -1;
314
315 while (*p) {
316 secs = strtol(p, &endp, 10);
317 if (p == endp ||
318 (errno == ERANGE && (secs == LONG_MIN || secs == LONG_MAX)) ||
319 secs < 0)
320 return -1;
321
322 switch (*endp++) {
323 case '\0':
324 endp--;
Damien Miller69b72032006-03-26 14:02:35 +1100325 break;
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000326 case 's':
327 case 'S':
328 break;
329 case 'm':
330 case 'M':
331 secs *= MINUTES;
332 break;
333 case 'h':
334 case 'H':
335 secs *= HOURS;
336 break;
337 case 'd':
338 case 'D':
339 secs *= DAYS;
340 break;
341 case 'w':
342 case 'W':
343 secs *= WEEKS;
344 break;
345 default:
346 return -1;
347 }
348 total += secs;
349 if (total < 0)
350 return -1;
351 p = endp;
352 }
353
354 return total;
355}
356
Damien Millerf91ee4c2005-03-01 21:24:33 +1100357/*
Darren Tuckerda345532006-07-10 23:04:19 +1000358 * Returns a standardized host+port identifier string.
359 * Caller must free returned string.
360 */
361char *
362put_host_port(const char *host, u_short port)
363{
364 char *hoststr;
365
366 if (port == 0 || port == SSH_DEFAULT_PORT)
367 return(xstrdup(host));
368 if (asprintf(&hoststr, "[%s]:%d", host, (int)port) < 0)
369 fatal("put_host_port: asprintf: %s", strerror(errno));
370 debug3("put_host_port: %s", hoststr);
371 return hoststr;
372}
373
374/*
Damien Millerf91ee4c2005-03-01 21:24:33 +1100375 * Search for next delimiter between hostnames/addresses and ports.
376 * Argument may be modified (for termination).
377 * Returns *cp if parsing succeeds.
378 * *cp is set to the start of the next delimiter, if one was found.
379 * If this is the last field, *cp is set to NULL.
380 */
381char *
382hpdelim(char **cp)
383{
384 char *s, *old;
385
386 if (cp == NULL || *cp == NULL)
387 return NULL;
388
389 old = s = *cp;
390 if (*s == '[') {
391 if ((s = strchr(s, ']')) == NULL)
392 return NULL;
393 else
394 s++;
395 } else if ((s = strpbrk(s, ":/")) == NULL)
396 s = *cp + strlen(*cp); /* skip to end (see first case below) */
397
398 switch (*s) {
399 case '\0':
400 *cp = NULL; /* no more fields*/
401 break;
Darren Tucker47eede72005-03-14 23:08:12 +1100402
Damien Millerf91ee4c2005-03-01 21:24:33 +1100403 case ':':
404 case '/':
405 *s = '\0'; /* terminate */
406 *cp = s + 1;
407 break;
Darren Tucker47eede72005-03-14 23:08:12 +1100408
Damien Millerf91ee4c2005-03-01 21:24:33 +1100409 default:
410 return NULL;
411 }
412
413 return old;
414}
415
Ben Lindstrom4529b702001-05-03 23:39:53 +0000416char *
417cleanhostname(char *host)
418{
419 if (*host == '[' && host[strlen(host) - 1] == ']') {
420 host[strlen(host) - 1] = '\0';
421 return (host + 1);
422 } else
423 return host;
424}
425
426char *
427colon(char *cp)
428{
429 int flag = 0;
430
431 if (*cp == ':') /* Leading colon is part of file name. */
Damien Miller2e774462010-06-26 09:30:47 +1000432 return NULL;
Ben Lindstrom4529b702001-05-03 23:39:53 +0000433 if (*cp == '[')
434 flag = 1;
435
436 for (; *cp; ++cp) {
437 if (*cp == '@' && *(cp+1) == '[')
438 flag = 1;
439 if (*cp == ']' && *(cp+1) == ':' && flag)
440 return (cp+1);
441 if (*cp == ':' && !flag)
442 return (cp);
443 if (*cp == '/')
Damien Miller2e774462010-06-26 09:30:47 +1000444 return NULL;
Ben Lindstrom4529b702001-05-03 23:39:53 +0000445 }
Damien Miller2e774462010-06-26 09:30:47 +1000446 return NULL;
Ben Lindstrom4529b702001-05-03 23:39:53 +0000447}
448
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000449/* function to assist building execv() arguments */
Ben Lindstrom387c4722001-05-08 20:27:25 +0000450void
451addargs(arglist *args, char *fmt, ...)
452{
453 va_list ap;
Damien Miller3eec6b72006-01-31 21:49:27 +1100454 char *cp;
Darren Tuckerc7a6fc42004-08-13 21:18:00 +1000455 u_int nalloc;
Damien Miller3eec6b72006-01-31 21:49:27 +1100456 int r;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000457
458 va_start(ap, fmt);
Damien Miller3eec6b72006-01-31 21:49:27 +1100459 r = vasprintf(&cp, fmt, ap);
Ben Lindstrom387c4722001-05-08 20:27:25 +0000460 va_end(ap);
Damien Miller3eec6b72006-01-31 21:49:27 +1100461 if (r == -1)
462 fatal("addargs: argument too long");
Ben Lindstrom387c4722001-05-08 20:27:25 +0000463
Darren Tuckerfb16b242003-09-22 21:04:23 +1000464 nalloc = args->nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000465 if (args->list == NULL) {
Darren Tuckerfb16b242003-09-22 21:04:23 +1000466 nalloc = 32;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000467 args->num = 0;
Darren Tuckerfb16b242003-09-22 21:04:23 +1000468 } else if (args->num+2 >= nalloc)
469 nalloc *= 2;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000470
Damien Miller36812092006-03-26 14:22:47 +1100471 args->list = xrealloc(args->list, nalloc, sizeof(char *));
Darren Tuckerfb16b242003-09-22 21:04:23 +1000472 args->nalloc = nalloc;
Damien Miller3eec6b72006-01-31 21:49:27 +1100473 args->list[args->num++] = cp;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000474 args->list[args->num] = NULL;
475}
Darren Tucker22cc7412004-12-06 22:47:41 +1100476
Damien Miller3eec6b72006-01-31 21:49:27 +1100477void
478replacearg(arglist *args, u_int which, char *fmt, ...)
479{
480 va_list ap;
481 char *cp;
482 int r;
483
484 va_start(ap, fmt);
485 r = vasprintf(&cp, fmt, ap);
486 va_end(ap);
487 if (r == -1)
488 fatal("replacearg: argument too long");
489
490 if (which >= args->num)
491 fatal("replacearg: tried to replace invalid arg %d >= %d",
492 which, args->num);
493 xfree(args->list[which]);
494 args->list[which] = cp;
495}
496
497void
498freeargs(arglist *args)
499{
500 u_int i;
501
502 if (args->list != NULL) {
503 for (i = 0; i < args->num; i++)
504 xfree(args->list[i]);
505 xfree(args->list);
506 args->nalloc = args->num = 0;
507 args->list = NULL;
508 }
509}
510
Darren Tucker22cc7412004-12-06 22:47:41 +1100511/*
Damien Miller5fd38c02005-05-26 12:02:14 +1000512 * Expands tildes in the file name. Returns data allocated by xmalloc.
513 * Warning: this calls getpw*.
514 */
515char *
516tilde_expand_filename(const char *filename, uid_t uid)
517{
518 const char *path;
519 char user[128], ret[MAXPATHLEN];
520 struct passwd *pw;
Damien Millereccb9de2005-06-17 12:59:34 +1000521 u_int len, slash;
Damien Miller5fd38c02005-05-26 12:02:14 +1000522
523 if (*filename != '~')
524 return (xstrdup(filename));
525 filename++;
526
527 path = strchr(filename, '/');
528 if (path != NULL && path > filename) { /* ~user/path */
Damien Millereccb9de2005-06-17 12:59:34 +1000529 slash = path - filename;
530 if (slash > sizeof(user) - 1)
Damien Miller5fd38c02005-05-26 12:02:14 +1000531 fatal("tilde_expand_filename: ~username too long");
Damien Millereccb9de2005-06-17 12:59:34 +1000532 memcpy(user, filename, slash);
533 user[slash] = '\0';
Damien Miller5fd38c02005-05-26 12:02:14 +1000534 if ((pw = getpwnam(user)) == NULL)
535 fatal("tilde_expand_filename: No such user %s", user);
536 } else if ((pw = getpwuid(uid)) == NULL) /* ~/path */
Darren Tucker7517b5b2008-06-13 14:48:59 +1000537 fatal("tilde_expand_filename: No such uid %ld", (long)uid);
Damien Miller5fd38c02005-05-26 12:02:14 +1000538
539 if (strlcpy(ret, pw->pw_dir, sizeof(ret)) >= sizeof(ret))
540 fatal("tilde_expand_filename: Path too long");
541
542 /* Make sure directory has a trailing '/' */
543 len = strlen(pw->pw_dir);
544 if ((len == 0 || pw->pw_dir[len - 1] != '/') &&
545 strlcat(ret, "/", sizeof(ret)) >= sizeof(ret))
546 fatal("tilde_expand_filename: Path too long");
547
548 /* Skip leading '/' from specified path */
549 if (path != NULL)
550 filename = path + 1;
551 if (strlcat(ret, filename, sizeof(ret)) >= sizeof(ret))
552 fatal("tilde_expand_filename: Path too long");
553
554 return (xstrdup(ret));
555}
556
557/*
Damien Miller6476cad2005-06-16 13:18:34 +1000558 * Expand a string with a set of %[char] escapes. A number of escapes may be
559 * specified as (char *escape_chars, char *replacement) pairs. The list must
Darren Tuckerbee73d52005-07-14 17:05:02 +1000560 * be terminated by a NULL escape_char. Returns replaced string in memory
Damien Miller6476cad2005-06-16 13:18:34 +1000561 * allocated by xmalloc.
562 */
563char *
564percent_expand(const char *string, ...)
565{
566#define EXPAND_MAX_KEYS 16
Darren Tucker70d87692010-01-08 18:49:16 +1100567 u_int num_keys, i, j;
Damien Miller6476cad2005-06-16 13:18:34 +1000568 struct {
569 const char *key;
570 const char *repl;
571 } keys[EXPAND_MAX_KEYS];
Damien Miller6476cad2005-06-16 13:18:34 +1000572 char buf[4096];
573 va_list ap;
574
575 /* Gather keys */
576 va_start(ap, string);
577 for (num_keys = 0; num_keys < EXPAND_MAX_KEYS; num_keys++) {
578 keys[num_keys].key = va_arg(ap, char *);
579 if (keys[num_keys].key == NULL)
580 break;
581 keys[num_keys].repl = va_arg(ap, char *);
582 if (keys[num_keys].repl == NULL)
Darren Tucker70d87692010-01-08 18:49:16 +1100583 fatal("%s: NULL replacement", __func__);
Damien Miller6476cad2005-06-16 13:18:34 +1000584 }
Darren Tucker70d87692010-01-08 18:49:16 +1100585 if (num_keys == EXPAND_MAX_KEYS && va_arg(ap, char *) != NULL)
586 fatal("%s: too many keys", __func__);
Damien Miller6476cad2005-06-16 13:18:34 +1000587 va_end(ap);
588
Damien Miller6476cad2005-06-16 13:18:34 +1000589 /* Expand string */
590 *buf = '\0';
591 for (i = 0; *string != '\0'; string++) {
592 if (*string != '%') {
593 append:
594 buf[i++] = *string;
595 if (i >= sizeof(buf))
Darren Tucker70d87692010-01-08 18:49:16 +1100596 fatal("%s: string too long", __func__);
Damien Miller6476cad2005-06-16 13:18:34 +1000597 buf[i] = '\0';
598 continue;
599 }
600 string++;
Darren Tucker70d87692010-01-08 18:49:16 +1100601 /* %% case */
Damien Miller6476cad2005-06-16 13:18:34 +1000602 if (*string == '%')
603 goto append;
604 for (j = 0; j < num_keys; j++) {
605 if (strchr(keys[j].key, *string) != NULL) {
606 i = strlcat(buf, keys[j].repl, sizeof(buf));
607 if (i >= sizeof(buf))
Darren Tucker70d87692010-01-08 18:49:16 +1100608 fatal("%s: string too long", __func__);
Damien Miller6476cad2005-06-16 13:18:34 +1000609 break;
610 }
611 }
612 if (j >= num_keys)
Darren Tucker70d87692010-01-08 18:49:16 +1100613 fatal("%s: unknown key %%%c", __func__, *string);
Damien Miller6476cad2005-06-16 13:18:34 +1000614 }
615 return (xstrdup(buf));
616#undef EXPAND_MAX_KEYS
617}
618
619/*
Darren Tucker22cc7412004-12-06 22:47:41 +1100620 * Read an entire line from a public key file into a static buffer, discarding
621 * lines that exceed the buffer size. Returns 0 on success, -1 on failure.
622 */
623int
624read_keyfile_line(FILE *f, const char *filename, char *buf, size_t bufsz,
Darren Tuckerf0f90982004-12-11 13:39:50 +1100625 u_long *lineno)
Darren Tucker22cc7412004-12-06 22:47:41 +1100626{
627 while (fgets(buf, bufsz, f) != NULL) {
Damien Miller3ca8b772007-01-05 16:24:47 +1100628 if (buf[0] == '\0')
629 continue;
Darren Tucker22cc7412004-12-06 22:47:41 +1100630 (*lineno)++;
631 if (buf[strlen(buf) - 1] == '\n' || feof(f)) {
632 return 0;
633 } else {
Darren Tuckerf0f90982004-12-11 13:39:50 +1100634 debug("%s: %s line %lu exceeds size limit", __func__,
635 filename, *lineno);
Darren Tucker22cc7412004-12-06 22:47:41 +1100636 /* discard remainder of line */
Darren Tucker47eede72005-03-14 23:08:12 +1100637 while (fgetc(f) != '\n' && !feof(f))
Darren Tucker22cc7412004-12-06 22:47:41 +1100638 ; /* nothing */
639 }
640 }
641 return -1;
642}
Damien Miller13390022005-07-06 09:44:19 +1000643
Damien Millerd27b9472005-12-13 19:29:02 +1100644int
Damien Miller7b58e802005-12-13 19:33:19 +1100645tun_open(int tun, int mode)
Damien Millerd27b9472005-12-13 19:29:02 +1100646{
Damien Miller62a31c92005-12-13 20:44:13 +1100647#if defined(CUSTOM_SYS_TUN_OPEN)
648 return (sys_tun_open(tun, mode));
Damien Miller2dcddbf2006-01-01 19:47:05 +1100649#elif defined(SSH_TUN_OPENBSD)
Damien Miller7b58e802005-12-13 19:33:19 +1100650 struct ifreq ifr;
Damien Millerd27b9472005-12-13 19:29:02 +1100651 char name[100];
Damien Miller7b58e802005-12-13 19:33:19 +1100652 int fd = -1, sock;
Damien Millerd27b9472005-12-13 19:29:02 +1100653
Damien Miller7b58e802005-12-13 19:33:19 +1100654 /* Open the tunnel device */
655 if (tun <= SSH_TUNID_MAX) {
Damien Millerd27b9472005-12-13 19:29:02 +1100656 snprintf(name, sizeof(name), "/dev/tun%d", tun);
Damien Miller7b58e802005-12-13 19:33:19 +1100657 fd = open(name, O_RDWR);
658 } else if (tun == SSH_TUNID_ANY) {
659 for (tun = 100; tun >= 0; tun--) {
660 snprintf(name, sizeof(name), "/dev/tun%d", tun);
661 if ((fd = open(name, O_RDWR)) >= 0)
662 break;
Damien Millerd27b9472005-12-13 19:29:02 +1100663 }
664 } else {
Damien Millera210d522006-01-02 23:40:30 +1100665 debug("%s: invalid tunnel %u", __func__, tun);
Damien Miller7b58e802005-12-13 19:33:19 +1100666 return (-1);
Damien Millerd27b9472005-12-13 19:29:02 +1100667 }
Damien Miller7b58e802005-12-13 19:33:19 +1100668
669 if (fd < 0) {
670 debug("%s: %s open failed: %s", __func__, name, strerror(errno));
671 return (-1);
672 }
673
674 debug("%s: %s mode %d fd %d", __func__, name, mode, fd);
675
676 /* Set the tunnel device operation mode */
677 snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "tun%d", tun);
678 if ((sock = socket(PF_UNIX, SOCK_STREAM, 0)) == -1)
679 goto failed;
680
681 if (ioctl(sock, SIOCGIFFLAGS, &ifr) == -1)
682 goto failed;
Damien Millera1d9a182006-01-02 23:41:21 +1100683
684 /* Set interface mode */
685 ifr.ifr_flags &= ~IFF_UP;
686 if (mode == SSH_TUNMODE_ETHERNET)
Damien Miller7b58e802005-12-13 19:33:19 +1100687 ifr.ifr_flags |= IFF_LINK0;
Damien Millera1d9a182006-01-02 23:41:21 +1100688 else
689 ifr.ifr_flags &= ~IFF_LINK0;
690 if (ioctl(sock, SIOCSIFFLAGS, &ifr) == -1)
691 goto failed;
692
693 /* Bring interface up */
Damien Miller7b58e802005-12-13 19:33:19 +1100694 ifr.ifr_flags |= IFF_UP;
695 if (ioctl(sock, SIOCSIFFLAGS, &ifr) == -1)
696 goto failed;
697
698 close(sock);
699 return (fd);
700
701 failed:
702 if (fd >= 0)
703 close(fd);
704 if (sock >= 0)
705 close(sock);
706 debug("%s: failed to set %s mode %d: %s", __func__, name,
707 mode, strerror(errno));
Damien Millerd27b9472005-12-13 19:29:02 +1100708 return (-1);
Damien Miller62a31c92005-12-13 20:44:13 +1100709#else
710 error("Tunnel interfaces are not supported on this platform");
711 return (-1);
712#endif
Damien Millerd27b9472005-12-13 19:29:02 +1100713}
714
Darren Tuckerce321d82005-10-03 18:11:24 +1000715void
716sanitise_stdfd(void)
717{
Damien Miller72c5b7d2006-01-06 14:50:44 +1100718 int nullfd, dupfd;
Darren Tuckerce321d82005-10-03 18:11:24 +1000719
Damien Miller72c5b7d2006-01-06 14:50:44 +1100720 if ((nullfd = dupfd = open(_PATH_DEVNULL, O_RDWR)) == -1) {
Damien Miller9eab9562009-02-22 08:47:02 +1100721 fprintf(stderr, "Couldn't open /dev/null: %s\n",
722 strerror(errno));
Darren Tuckerce321d82005-10-03 18:11:24 +1000723 exit(1);
724 }
Damien Miller72c5b7d2006-01-06 14:50:44 +1100725 while (++dupfd <= 2) {
726 /* Only clobber closed fds */
727 if (fcntl(dupfd, F_GETFL, 0) >= 0)
728 continue;
729 if (dup2(nullfd, dupfd) == -1) {
Damien Miller9eab9562009-02-22 08:47:02 +1100730 fprintf(stderr, "dup2: %s\n", strerror(errno));
Darren Tuckerce321d82005-10-03 18:11:24 +1000731 exit(1);
732 }
Darren Tuckerce321d82005-10-03 18:11:24 +1000733 }
734 if (nullfd > 2)
735 close(nullfd);
736}
737
Damien Miller13390022005-07-06 09:44:19 +1000738char *
Damien Miller3f941882006-03-31 23:13:02 +1100739tohex(const void *vp, size_t l)
Damien Miller13390022005-07-06 09:44:19 +1000740{
Damien Miller3f941882006-03-31 23:13:02 +1100741 const u_char *p = (const u_char *)vp;
Damien Miller13390022005-07-06 09:44:19 +1000742 char b[3], *r;
Damien Miller3f941882006-03-31 23:13:02 +1100743 size_t i, hl;
744
745 if (l > 65536)
746 return xstrdup("tohex: length > 65536");
Damien Miller13390022005-07-06 09:44:19 +1000747
748 hl = l * 2 + 1;
Damien Miller07d86be2006-03-26 14:19:21 +1100749 r = xcalloc(1, hl);
Damien Miller13390022005-07-06 09:44:19 +1000750 for (i = 0; i < l; i++) {
Damien Miller3f941882006-03-31 23:13:02 +1100751 snprintf(b, sizeof(b), "%02x", p[i]);
Damien Miller13390022005-07-06 09:44:19 +1000752 strlcat(r, b, hl);
753 }
754 return (r);
755}
756
Damien Miller3f941882006-03-31 23:13:02 +1100757u_int64_t
758get_u64(const void *vp)
759{
760 const u_char *p = (const u_char *)vp;
761 u_int64_t v;
762
763 v = (u_int64_t)p[0] << 56;
764 v |= (u_int64_t)p[1] << 48;
765 v |= (u_int64_t)p[2] << 40;
766 v |= (u_int64_t)p[3] << 32;
767 v |= (u_int64_t)p[4] << 24;
768 v |= (u_int64_t)p[5] << 16;
769 v |= (u_int64_t)p[6] << 8;
770 v |= (u_int64_t)p[7];
771
772 return (v);
773}
774
775u_int32_t
776get_u32(const void *vp)
777{
778 const u_char *p = (const u_char *)vp;
779 u_int32_t v;
780
781 v = (u_int32_t)p[0] << 24;
782 v |= (u_int32_t)p[1] << 16;
783 v |= (u_int32_t)p[2] << 8;
784 v |= (u_int32_t)p[3];
785
786 return (v);
787}
788
789u_int16_t
790get_u16(const void *vp)
791{
792 const u_char *p = (const u_char *)vp;
793 u_int16_t v;
794
795 v = (u_int16_t)p[0] << 8;
796 v |= (u_int16_t)p[1];
797
798 return (v);
799}
800
801void
802put_u64(void *vp, u_int64_t v)
803{
804 u_char *p = (u_char *)vp;
805
806 p[0] = (u_char)(v >> 56) & 0xff;
807 p[1] = (u_char)(v >> 48) & 0xff;
808 p[2] = (u_char)(v >> 40) & 0xff;
809 p[3] = (u_char)(v >> 32) & 0xff;
810 p[4] = (u_char)(v >> 24) & 0xff;
811 p[5] = (u_char)(v >> 16) & 0xff;
812 p[6] = (u_char)(v >> 8) & 0xff;
813 p[7] = (u_char)v & 0xff;
814}
815
816void
817put_u32(void *vp, u_int32_t v)
818{
819 u_char *p = (u_char *)vp;
820
821 p[0] = (u_char)(v >> 24) & 0xff;
822 p[1] = (u_char)(v >> 16) & 0xff;
823 p[2] = (u_char)(v >> 8) & 0xff;
824 p[3] = (u_char)v & 0xff;
825}
826
827
828void
829put_u16(void *vp, u_int16_t v)
830{
831 u_char *p = (u_char *)vp;
832
833 p[0] = (u_char)(v >> 8) & 0xff;
834 p[1] = (u_char)v & 0xff;
835}
Darren Tucker3fc464e2008-06-13 06:42:45 +1000836
837void
838ms_subtract_diff(struct timeval *start, int *ms)
839{
840 struct timeval diff, finish;
841
842 gettimeofday(&finish, NULL);
843 timersub(&finish, start, &diff);
844 *ms -= (diff.tv_sec * 1000) + (diff.tv_usec / 1000);
845}
846
847void
848ms_to_timeval(struct timeval *tv, int ms)
849{
850 if (ms < 0)
851 ms = 0;
852 tv->tv_sec = ms / 1000;
853 tv->tv_usec = (ms % 1000) * 1000;
854}
855
Damien Miller65e42f82010-09-24 22:15:11 +1000856void
857bandwidth_limit_init(struct bwlimit *bw, u_int64_t kbps, size_t buflen)
858{
859 bw->buflen = buflen;
860 bw->rate = kbps;
861 bw->thresh = bw->rate;
862 bw->lamt = 0;
863 timerclear(&bw->bwstart);
864 timerclear(&bw->bwend);
865}
866
867/* Callback from read/write loop to insert bandwidth-limiting delays */
868void
869bandwidth_limit(struct bwlimit *bw, size_t read_len)
870{
871 u_int64_t waitlen;
872 struct timespec ts, rm;
873
874 if (!timerisset(&bw->bwstart)) {
875 gettimeofday(&bw->bwstart, NULL);
876 return;
877 }
878
879 bw->lamt += read_len;
880 if (bw->lamt < bw->thresh)
881 return;
882
883 gettimeofday(&bw->bwend, NULL);
884 timersub(&bw->bwend, &bw->bwstart, &bw->bwend);
885 if (!timerisset(&bw->bwend))
886 return;
887
888 bw->lamt *= 8;
889 waitlen = (double)1000000L * bw->lamt / bw->rate;
890
891 bw->bwstart.tv_sec = waitlen / 1000000L;
892 bw->bwstart.tv_usec = waitlen % 1000000L;
893
894 if (timercmp(&bw->bwstart, &bw->bwend, >)) {
895 timersub(&bw->bwstart, &bw->bwend, &bw->bwend);
896
897 /* Adjust the wait time */
898 if (bw->bwend.tv_sec) {
899 bw->thresh /= 2;
900 if (bw->thresh < bw->buflen / 4)
901 bw->thresh = bw->buflen / 4;
902 } else if (bw->bwend.tv_usec < 10000) {
903 bw->thresh *= 2;
904 if (bw->thresh > bw->buflen * 8)
905 bw->thresh = bw->buflen * 8;
906 }
907
908 TIMEVAL_TO_TIMESPEC(&bw->bwend, &ts);
909 while (nanosleep(&ts, &rm) == -1) {
910 if (errno != EINTR)
911 break;
912 ts = rm;
913 }
914 }
915
916 bw->lamt = 0;
917 gettimeofday(&bw->bwstart, NULL);
918}
Damien Miller0dac6fb2010-11-20 15:19:38 +1100919
Damien Miller2cd62932010-12-01 11:50:35 +1100920/* Make a template filename for mk[sd]temp() */
921void
922mktemp_proto(char *s, size_t len)
923{
924 const char *tmpdir;
925 int r;
926
927 if ((tmpdir = getenv("TMPDIR")) != NULL) {
928 r = snprintf(s, len, "%s/ssh-XXXXXXXXXXXX", tmpdir);
929 if (r > 0 && (size_t)r < len)
930 return;
931 }
932 r = snprintf(s, len, "/tmp/ssh-XXXXXXXXXXXX");
933 if (r < 0 || (size_t)r >= len)
934 fatal("%s: template string too short", __func__);
935}
936
Damien Miller0dac6fb2010-11-20 15:19:38 +1100937static const struct {
938 const char *name;
939 int value;
940} ipqos[] = {
941 { "af11", IPTOS_DSCP_AF11 },
942 { "af12", IPTOS_DSCP_AF12 },
943 { "af13", IPTOS_DSCP_AF13 },
944 { "af14", IPTOS_DSCP_AF21 },
945 { "af22", IPTOS_DSCP_AF22 },
946 { "af23", IPTOS_DSCP_AF23 },
947 { "af31", IPTOS_DSCP_AF31 },
948 { "af32", IPTOS_DSCP_AF32 },
949 { "af33", IPTOS_DSCP_AF33 },
950 { "af41", IPTOS_DSCP_AF41 },
951 { "af42", IPTOS_DSCP_AF42 },
952 { "af43", IPTOS_DSCP_AF43 },
953 { "cs0", IPTOS_DSCP_CS0 },
954 { "cs1", IPTOS_DSCP_CS1 },
955 { "cs2", IPTOS_DSCP_CS2 },
956 { "cs3", IPTOS_DSCP_CS3 },
957 { "cs4", IPTOS_DSCP_CS4 },
958 { "cs5", IPTOS_DSCP_CS5 },
959 { "cs6", IPTOS_DSCP_CS6 },
960 { "cs7", IPTOS_DSCP_CS7 },
961 { "ef", IPTOS_DSCP_EF },
962 { "lowdelay", IPTOS_LOWDELAY },
963 { "throughput", IPTOS_THROUGHPUT },
964 { "reliability", IPTOS_RELIABILITY },
965 { NULL, -1 }
966};
967
968int
969parse_ipqos(const char *cp)
970{
971 u_int i;
972 char *ep;
973 long val;
974
975 if (cp == NULL)
976 return -1;
977 for (i = 0; ipqos[i].name != NULL; i++) {
978 if (strcasecmp(cp, ipqos[i].name) == 0)
979 return ipqos[i].value;
980 }
981 /* Try parsing as an integer */
982 val = strtol(cp, &ep, 0);
983 if (*cp == '\0' || *ep != '\0' || val < 0 || val > 255)
984 return -1;
985 return val;
986}
987
Damien Miller04ee0f82009-11-18 17:48:30 +1100988void
989sock_set_v6only(int s)
990{
991#ifdef IPV6_V6ONLY
992 int on = 1;
993
994 debug3("%s: set socket %d IPV6_V6ONLY", __func__, s);
995 if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on)) == -1)
996 error("setsockopt IPV6_V6ONLY: %s", strerror(errno));
997#endif
998}