blob: f0f1fd8416e7981f969f42dda448e74e16bd154e [file] [log] [blame]
Darren Tucker34e314d2010-01-08 17:03:46 +11001/* $OpenBSD: misc.c,v 1.72 2009/10/28 16:38:18 reyk 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 Millere6b3b612006-07-24 14:01:23 +100038#include <unistd.h>
Darren Tucker5d196262006-07-12 22:15:16 +100039
Damien Miller8ec8c3e2006-07-10 20:35:38 +100040#include <netinet/in.h>
Damien Miller3a4051e2006-03-15 11:19:42 +110041#include <netinet/tcp.h>
Damien Miller8ec8c3e2006-07-10 20:35:38 +100042
Darren Tucker39972492006-07-12 22:22:46 +100043#include <errno.h>
Damien Miller57cf6382006-07-10 21:13:46 +100044#include <fcntl.h>
Darren Tucker4abde772007-12-29 02:43:51 +110045#include <netdb.h>
Damien Miller03e20032006-03-15 11:16:59 +110046#ifdef HAVE_PATHS_H
Damien Millera9263d02006-03-15 11:18:26 +110047# include <paths.h>
Damien Miller9f2abc42006-07-10 20:53:08 +100048#include <pwd.h>
Damien Miller03e20032006-03-15 11:16:59 +110049#endif
Damien Miller3beb8522006-01-02 23:40:10 +110050#ifdef SSH_TUN_OPENBSD
51#include <net/if.h>
52#endif
Damien Miller61c51502000-08-18 14:01:04 +100053
Damien Millerd7834352006-08-05 12:39:39 +100054#include "xmalloc.h"
Kevin Stevesb6e773a2001-02-04 13:20:36 +000055#include "misc.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000056#include "log.h"
Darren Tuckerda345532006-07-10 23:04:19 +100057#include "ssh.h"
Damien Miller61c51502000-08-18 14:01:04 +100058
Ben Lindstrom4cc240d2001-07-04 04:46:56 +000059/* remove newline at end of string */
Damien Miller61c51502000-08-18 14:01:04 +100060char *
61chop(char *s)
62{
63 char *t = s;
64 while (*t) {
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +000065 if (*t == '\n' || *t == '\r') {
Damien Miller61c51502000-08-18 14:01:04 +100066 *t = '\0';
67 return s;
68 }
69 t++;
70 }
71 return s;
72
73}
74
Ben Lindstrom4cc240d2001-07-04 04:46:56 +000075/* set/unset filedescriptor to non-blocking */
Damien Miller232711f2004-06-15 10:35:30 +100076int
Damien Miller61c51502000-08-18 14:01:04 +100077set_nonblock(int fd)
78{
79 int val;
Ben Lindstromc93e84c2001-05-12 00:08:37 +000080
Damien Miller61c51502000-08-18 14:01:04 +100081 val = fcntl(fd, F_GETFL, 0);
82 if (val < 0) {
83 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +100084 return (-1);
Damien Miller61c51502000-08-18 14:01:04 +100085 }
Damien Miller69b69aa2000-10-28 14:19:58 +110086 if (val & O_NONBLOCK) {
Damien Miller232711f2004-06-15 10:35:30 +100087 debug3("fd %d is O_NONBLOCK", fd);
88 return (0);
Damien Miller69b69aa2000-10-28 14:19:58 +110089 }
Damien Milleref095ce2003-05-14 13:41:39 +100090 debug2("fd %d setting O_NONBLOCK", fd);
Damien Miller61c51502000-08-18 14:01:04 +100091 val |= O_NONBLOCK;
Damien Miller232711f2004-06-15 10:35:30 +100092 if (fcntl(fd, F_SETFL, val) == -1) {
93 debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s", fd,
94 strerror(errno));
95 return (-1);
96 }
97 return (0);
Damien Miller61c51502000-08-18 14:01:04 +100098}
99
Damien Miller232711f2004-06-15 10:35:30 +1000100int
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000101unset_nonblock(int fd)
102{
103 int val;
104
105 val = fcntl(fd, F_GETFL, 0);
106 if (val < 0) {
107 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +1000108 return (-1);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000109 }
110 if (!(val & O_NONBLOCK)) {
Damien Miller232711f2004-06-15 10:35:30 +1000111 debug3("fd %d is not O_NONBLOCK", fd);
112 return (0);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000113 }
Ben Lindstrom352b1c22001-06-21 03:04:37 +0000114 debug("fd %d clearing O_NONBLOCK", fd);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000115 val &= ~O_NONBLOCK;
Damien Miller232711f2004-06-15 10:35:30 +1000116 if (fcntl(fd, F_SETFL, val) == -1) {
117 debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s",
Ben Lindstrom84fcb312002-03-05 01:48:09 +0000118 fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +1000119 return (-1);
120 }
121 return (0);
Ben Lindstromc93e84c2001-05-12 00:08:37 +0000122}
123
Darren Tucker4abde772007-12-29 02:43:51 +1100124const char *
125ssh_gai_strerror(int gaierr)
126{
Darren Tucker912428c2008-01-01 20:33:35 +1100127 if (gaierr == EAI_SYSTEM)
128 return strerror(errno);
129 return gai_strerror(gaierr);
Darren Tucker4abde772007-12-29 02:43:51 +1100130}
131
Damien Miller398e1cf2002-02-05 11:52:13 +1100132/* disable nagle on socket */
133void
134set_nodelay(int fd)
135{
Ben Lindstrome86de512002-03-05 01:28:14 +0000136 int opt;
137 socklen_t optlen;
Damien Miller398e1cf2002-02-05 11:52:13 +1100138
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000139 optlen = sizeof opt;
140 if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) {
Darren Tucker6db8f932003-11-03 20:07:14 +1100141 debug("getsockopt TCP_NODELAY: %.100s", strerror(errno));
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000142 return;
143 }
144 if (opt == 1) {
145 debug2("fd %d is TCP_NODELAY", fd);
146 return;
147 }
148 opt = 1;
Ben Lindstrom1d568f92002-12-23 02:44:36 +0000149 debug2("fd %d setting TCP_NODELAY", fd);
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000150 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1)
Damien Miller398e1cf2002-02-05 11:52:13 +1100151 error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
152}
153
Darren Tucker34e314d2010-01-08 17:03:46 +1100154/* open a socket in the specified routing domain */
155int
156socket_rdomain(int domain, int type, int protocol, int rdomain)
157{
158 int sock, ipproto = IPPROTO_IP;
159
160 if ((sock = socket(domain, type, protocol)) == -1)
161 return (-1);
162
163 if (rdomain == -1)
164 return (sock);
165
166 switch (domain) {
167 case AF_INET6:
168 ipproto = IPPROTO_IPV6;
169 /* FALLTHROUGH */
170 case AF_INET:
171 debug2("socket %d af %d setting rdomain %d",
172 sock, domain, rdomain);
173 if (setsockopt(sock, ipproto, SO_RDOMAIN, &rdomain,
174 sizeof(rdomain)) == -1) {
175 debug("setsockopt SO_RDOMAIN: %.100s",
176 strerror(errno));
177 close(sock);
178 return (-1);
179 }
180 break;
181 default:
182 debug("socket %d af %d does not support rdomain %d",
183 sock, domain, rdomain);
184 close(sock);
185 return (-1);
186 }
187
188 return (sock);
189}
190
Damien Miller61c51502000-08-18 14:01:04 +1000191/* Characters considered whitespace in strsep calls. */
192#define WHITESPACE " \t\r\n"
Damien Miller306d1182006-03-15 12:05:59 +1100193#define QUOTE "\""
Damien Miller61c51502000-08-18 14:01:04 +1000194
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000195/* return next token in configuration line */
Damien Miller61c51502000-08-18 14:01:04 +1000196char *
197strdelim(char **s)
198{
199 char *old;
200 int wspace = 0;
201
202 if (*s == NULL)
203 return NULL;
204
205 old = *s;
206
Damien Miller306d1182006-03-15 12:05:59 +1100207 *s = strpbrk(*s, WHITESPACE QUOTE "=");
Damien Miller61c51502000-08-18 14:01:04 +1000208 if (*s == NULL)
209 return (old);
210
Damien Miller306d1182006-03-15 12:05:59 +1100211 if (*s[0] == '\"') {
212 memmove(*s, *s + 1, strlen(*s)); /* move nul too */
213 /* Find matching quote */
214 if ((*s = strpbrk(*s, QUOTE)) == NULL) {
215 return (NULL); /* no matching quote */
216 } else {
217 *s[0] = '\0';
218 return (old);
219 }
220 }
221
Damien Miller61c51502000-08-18 14:01:04 +1000222 /* Allow only one '=' to be skipped */
223 if (*s[0] == '=')
224 wspace = 1;
225 *s[0] = '\0';
226
Damien Miller306d1182006-03-15 12:05:59 +1100227 /* Skip any extra whitespace after first token */
Damien Miller61c51502000-08-18 14:01:04 +1000228 *s += strspn(*s + 1, WHITESPACE) + 1;
229 if (*s[0] == '=' && !wspace)
230 *s += strspn(*s + 1, WHITESPACE) + 1;
231
232 return (old);
233}
Kevin Stevesb6e773a2001-02-04 13:20:36 +0000234
Ben Lindstrom086cf212001-03-05 05:56:40 +0000235struct passwd *
236pwcopy(struct passwd *pw)
237{
Damien Miller07d86be2006-03-26 14:19:21 +1100238 struct passwd *copy = xcalloc(1, sizeof(*copy));
Ben Lindstrom40304422001-03-05 06:22:01 +0000239
Ben Lindstrom086cf212001-03-05 05:56:40 +0000240 copy->pw_name = xstrdup(pw->pw_name);
241 copy->pw_passwd = xstrdup(pw->pw_passwd);
Ben Lindstrom40304422001-03-05 06:22:01 +0000242 copy->pw_gecos = xstrdup(pw->pw_gecos);
Ben Lindstrom086cf212001-03-05 05:56:40 +0000243 copy->pw_uid = pw->pw_uid;
244 copy->pw_gid = pw->pw_gid;
Kevin Steves82456952001-06-22 21:14:18 +0000245#ifdef HAVE_PW_EXPIRE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000246 copy->pw_expire = pw->pw_expire;
Kevin Steves82456952001-06-22 21:14:18 +0000247#endif
248#ifdef HAVE_PW_CHANGE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000249 copy->pw_change = pw->pw_change;
Kevin Steves82456952001-06-22 21:14:18 +0000250#endif
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000251#ifdef HAVE_PW_CLASS_IN_PASSWD
Ben Lindstrom086cf212001-03-05 05:56:40 +0000252 copy->pw_class = xstrdup(pw->pw_class);
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000253#endif
Ben Lindstrom086cf212001-03-05 05:56:40 +0000254 copy->pw_dir = xstrdup(pw->pw_dir);
255 copy->pw_shell = xstrdup(pw->pw_shell);
256 return copy;
257}
258
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000259/*
260 * Convert ASCII string to TCP/IP port number.
Damien Miller3dc71ad2009-01-28 16:31:22 +1100261 * Port must be >=0 and <=65535.
262 * Return -1 if invalid.
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000263 */
264int
265a2port(const char *s)
Ben Lindstrom19066a12001-04-12 23:39:26 +0000266{
Damien Miller3dc71ad2009-01-28 16:31:22 +1100267 long long port;
268 const char *errstr;
Ben Lindstrom19066a12001-04-12 23:39:26 +0000269
Damien Miller3dc71ad2009-01-28 16:31:22 +1100270 port = strtonum(s, 0, 65535, &errstr);
271 if (errstr != NULL)
272 return -1;
273 return (int)port;
Ben Lindstrom19066a12001-04-12 23:39:26 +0000274}
275
Damien Millerd27b9472005-12-13 19:29:02 +1100276int
277a2tun(const char *s, int *remote)
278{
279 const char *errstr = NULL;
280 char *sp, *ep;
281 int tun;
282
283 if (remote != NULL) {
Damien Miller7b58e802005-12-13 19:33:19 +1100284 *remote = SSH_TUNID_ANY;
Damien Millerd27b9472005-12-13 19:29:02 +1100285 sp = xstrdup(s);
286 if ((ep = strchr(sp, ':')) == NULL) {
287 xfree(sp);
288 return (a2tun(s, NULL));
289 }
290 ep[0] = '\0'; ep++;
291 *remote = a2tun(ep, NULL);
292 tun = a2tun(sp, NULL);
293 xfree(sp);
Damien Miller7b58e802005-12-13 19:33:19 +1100294 return (*remote == SSH_TUNID_ERR ? *remote : tun);
Damien Millerd27b9472005-12-13 19:29:02 +1100295 }
296
297 if (strcasecmp(s, "any") == 0)
Damien Miller7b58e802005-12-13 19:33:19 +1100298 return (SSH_TUNID_ANY);
Damien Millerd27b9472005-12-13 19:29:02 +1100299
Damien Miller7b58e802005-12-13 19:33:19 +1100300 tun = strtonum(s, 0, SSH_TUNID_MAX, &errstr);
301 if (errstr != NULL)
302 return (SSH_TUNID_ERR);
Damien Millerd27b9472005-12-13 19:29:02 +1100303
304 return (tun);
305}
306
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000307#define SECONDS 1
308#define MINUTES (SECONDS * 60)
309#define HOURS (MINUTES * 60)
310#define DAYS (HOURS * 24)
311#define WEEKS (DAYS * 7)
312
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000313/*
314 * Convert a time string into seconds; format is
315 * a sequence of:
316 * time[qualifier]
317 *
318 * Valid time qualifiers are:
319 * <none> seconds
320 * s|S seconds
321 * m|M minutes
322 * h|H hours
323 * d|D days
324 * w|W weeks
325 *
326 * Examples:
327 * 90m 90 minutes
328 * 1h30m 90 minutes
329 * 2d 2 days
330 * 1w 1 week
331 *
332 * Return -1 if time string is invalid.
333 */
334long
335convtime(const char *s)
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000336{
337 long total, secs;
338 const char *p;
339 char *endp;
340
341 errno = 0;
342 total = 0;
343 p = s;
344
345 if (p == NULL || *p == '\0')
346 return -1;
347
348 while (*p) {
349 secs = strtol(p, &endp, 10);
350 if (p == endp ||
351 (errno == ERANGE && (secs == LONG_MIN || secs == LONG_MAX)) ||
352 secs < 0)
353 return -1;
354
355 switch (*endp++) {
356 case '\0':
357 endp--;
Damien Miller69b72032006-03-26 14:02:35 +1100358 break;
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000359 case 's':
360 case 'S':
361 break;
362 case 'm':
363 case 'M':
364 secs *= MINUTES;
365 break;
366 case 'h':
367 case 'H':
368 secs *= HOURS;
369 break;
370 case 'd':
371 case 'D':
372 secs *= DAYS;
373 break;
374 case 'w':
375 case 'W':
376 secs *= WEEKS;
377 break;
378 default:
379 return -1;
380 }
381 total += secs;
382 if (total < 0)
383 return -1;
384 p = endp;
385 }
386
387 return total;
388}
389
Damien Millerf91ee4c2005-03-01 21:24:33 +1100390/*
Darren Tuckerda345532006-07-10 23:04:19 +1000391 * Returns a standardized host+port identifier string.
392 * Caller must free returned string.
393 */
394char *
395put_host_port(const char *host, u_short port)
396{
397 char *hoststr;
398
399 if (port == 0 || port == SSH_DEFAULT_PORT)
400 return(xstrdup(host));
401 if (asprintf(&hoststr, "[%s]:%d", host, (int)port) < 0)
402 fatal("put_host_port: asprintf: %s", strerror(errno));
403 debug3("put_host_port: %s", hoststr);
404 return hoststr;
405}
406
407/*
Damien Millerf91ee4c2005-03-01 21:24:33 +1100408 * Search for next delimiter between hostnames/addresses and ports.
409 * Argument may be modified (for termination).
410 * Returns *cp if parsing succeeds.
411 * *cp is set to the start of the next delimiter, if one was found.
412 * If this is the last field, *cp is set to NULL.
413 */
414char *
415hpdelim(char **cp)
416{
417 char *s, *old;
418
419 if (cp == NULL || *cp == NULL)
420 return NULL;
421
422 old = s = *cp;
423 if (*s == '[') {
424 if ((s = strchr(s, ']')) == NULL)
425 return NULL;
426 else
427 s++;
428 } else if ((s = strpbrk(s, ":/")) == NULL)
429 s = *cp + strlen(*cp); /* skip to end (see first case below) */
430
431 switch (*s) {
432 case '\0':
433 *cp = NULL; /* no more fields*/
434 break;
Darren Tucker47eede72005-03-14 23:08:12 +1100435
Damien Millerf91ee4c2005-03-01 21:24:33 +1100436 case ':':
437 case '/':
438 *s = '\0'; /* terminate */
439 *cp = s + 1;
440 break;
Darren Tucker47eede72005-03-14 23:08:12 +1100441
Damien Millerf91ee4c2005-03-01 21:24:33 +1100442 default:
443 return NULL;
444 }
445
446 return old;
447}
448
Ben Lindstrom4529b702001-05-03 23:39:53 +0000449char *
450cleanhostname(char *host)
451{
452 if (*host == '[' && host[strlen(host) - 1] == ']') {
453 host[strlen(host) - 1] = '\0';
454 return (host + 1);
455 } else
456 return host;
457}
458
459char *
460colon(char *cp)
461{
462 int flag = 0;
463
464 if (*cp == ':') /* Leading colon is part of file name. */
465 return (0);
466 if (*cp == '[')
467 flag = 1;
468
469 for (; *cp; ++cp) {
470 if (*cp == '@' && *(cp+1) == '[')
471 flag = 1;
472 if (*cp == ']' && *(cp+1) == ':' && flag)
473 return (cp+1);
474 if (*cp == ':' && !flag)
475 return (cp);
476 if (*cp == '/')
477 return (0);
478 }
479 return (0);
480}
481
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000482/* function to assist building execv() arguments */
Ben Lindstrom387c4722001-05-08 20:27:25 +0000483void
484addargs(arglist *args, char *fmt, ...)
485{
486 va_list ap;
Damien Miller3eec6b72006-01-31 21:49:27 +1100487 char *cp;
Darren Tuckerc7a6fc42004-08-13 21:18:00 +1000488 u_int nalloc;
Damien Miller3eec6b72006-01-31 21:49:27 +1100489 int r;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000490
491 va_start(ap, fmt);
Damien Miller3eec6b72006-01-31 21:49:27 +1100492 r = vasprintf(&cp, fmt, ap);
Ben Lindstrom387c4722001-05-08 20:27:25 +0000493 va_end(ap);
Damien Miller3eec6b72006-01-31 21:49:27 +1100494 if (r == -1)
495 fatal("addargs: argument too long");
Ben Lindstrom387c4722001-05-08 20:27:25 +0000496
Darren Tuckerfb16b242003-09-22 21:04:23 +1000497 nalloc = args->nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000498 if (args->list == NULL) {
Darren Tuckerfb16b242003-09-22 21:04:23 +1000499 nalloc = 32;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000500 args->num = 0;
Darren Tuckerfb16b242003-09-22 21:04:23 +1000501 } else if (args->num+2 >= nalloc)
502 nalloc *= 2;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000503
Damien Miller36812092006-03-26 14:22:47 +1100504 args->list = xrealloc(args->list, nalloc, sizeof(char *));
Darren Tuckerfb16b242003-09-22 21:04:23 +1000505 args->nalloc = nalloc;
Damien Miller3eec6b72006-01-31 21:49:27 +1100506 args->list[args->num++] = cp;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000507 args->list[args->num] = NULL;
508}
Darren Tucker22cc7412004-12-06 22:47:41 +1100509
Damien Miller3eec6b72006-01-31 21:49:27 +1100510void
511replacearg(arglist *args, u_int which, char *fmt, ...)
512{
513 va_list ap;
514 char *cp;
515 int r;
516
517 va_start(ap, fmt);
518 r = vasprintf(&cp, fmt, ap);
519 va_end(ap);
520 if (r == -1)
521 fatal("replacearg: argument too long");
522
523 if (which >= args->num)
524 fatal("replacearg: tried to replace invalid arg %d >= %d",
525 which, args->num);
526 xfree(args->list[which]);
527 args->list[which] = cp;
528}
529
530void
531freeargs(arglist *args)
532{
533 u_int i;
534
535 if (args->list != NULL) {
536 for (i = 0; i < args->num; i++)
537 xfree(args->list[i]);
538 xfree(args->list);
539 args->nalloc = args->num = 0;
540 args->list = NULL;
541 }
542}
543
Darren Tucker22cc7412004-12-06 22:47:41 +1100544/*
Damien Miller5fd38c02005-05-26 12:02:14 +1000545 * Expands tildes in the file name. Returns data allocated by xmalloc.
546 * Warning: this calls getpw*.
547 */
548char *
549tilde_expand_filename(const char *filename, uid_t uid)
550{
551 const char *path;
552 char user[128], ret[MAXPATHLEN];
553 struct passwd *pw;
Damien Millereccb9de2005-06-17 12:59:34 +1000554 u_int len, slash;
Damien Miller5fd38c02005-05-26 12:02:14 +1000555
556 if (*filename != '~')
557 return (xstrdup(filename));
558 filename++;
559
560 path = strchr(filename, '/');
561 if (path != NULL && path > filename) { /* ~user/path */
Damien Millereccb9de2005-06-17 12:59:34 +1000562 slash = path - filename;
563 if (slash > sizeof(user) - 1)
Damien Miller5fd38c02005-05-26 12:02:14 +1000564 fatal("tilde_expand_filename: ~username too long");
Damien Millereccb9de2005-06-17 12:59:34 +1000565 memcpy(user, filename, slash);
566 user[slash] = '\0';
Damien Miller5fd38c02005-05-26 12:02:14 +1000567 if ((pw = getpwnam(user)) == NULL)
568 fatal("tilde_expand_filename: No such user %s", user);
569 } else if ((pw = getpwuid(uid)) == NULL) /* ~/path */
Darren Tucker7517b5b2008-06-13 14:48:59 +1000570 fatal("tilde_expand_filename: No such uid %ld", (long)uid);
Damien Miller5fd38c02005-05-26 12:02:14 +1000571
572 if (strlcpy(ret, pw->pw_dir, sizeof(ret)) >= sizeof(ret))
573 fatal("tilde_expand_filename: Path too long");
574
575 /* Make sure directory has a trailing '/' */
576 len = strlen(pw->pw_dir);
577 if ((len == 0 || pw->pw_dir[len - 1] != '/') &&
578 strlcat(ret, "/", sizeof(ret)) >= sizeof(ret))
579 fatal("tilde_expand_filename: Path too long");
580
581 /* Skip leading '/' from specified path */
582 if (path != NULL)
583 filename = path + 1;
584 if (strlcat(ret, filename, sizeof(ret)) >= sizeof(ret))
585 fatal("tilde_expand_filename: Path too long");
586
587 return (xstrdup(ret));
588}
589
590/*
Damien Miller6476cad2005-06-16 13:18:34 +1000591 * Expand a string with a set of %[char] escapes. A number of escapes may be
592 * specified as (char *escape_chars, char *replacement) pairs. The list must
Darren Tuckerbee73d52005-07-14 17:05:02 +1000593 * be terminated by a NULL escape_char. Returns replaced string in memory
Damien Miller6476cad2005-06-16 13:18:34 +1000594 * allocated by xmalloc.
595 */
596char *
597percent_expand(const char *string, ...)
598{
599#define EXPAND_MAX_KEYS 16
600 struct {
601 const char *key;
602 const char *repl;
603 } keys[EXPAND_MAX_KEYS];
Damien Millereccb9de2005-06-17 12:59:34 +1000604 u_int num_keys, i, j;
Damien Miller6476cad2005-06-16 13:18:34 +1000605 char buf[4096];
606 va_list ap;
607
608 /* Gather keys */
609 va_start(ap, string);
610 for (num_keys = 0; num_keys < EXPAND_MAX_KEYS; num_keys++) {
611 keys[num_keys].key = va_arg(ap, char *);
612 if (keys[num_keys].key == NULL)
613 break;
614 keys[num_keys].repl = va_arg(ap, char *);
615 if (keys[num_keys].repl == NULL)
616 fatal("percent_expand: NULL replacement");
617 }
618 va_end(ap);
619
620 if (num_keys >= EXPAND_MAX_KEYS)
621 fatal("percent_expand: too many keys");
622
623 /* Expand string */
624 *buf = '\0';
625 for (i = 0; *string != '\0'; string++) {
626 if (*string != '%') {
627 append:
628 buf[i++] = *string;
629 if (i >= sizeof(buf))
630 fatal("percent_expand: string too long");
631 buf[i] = '\0';
632 continue;
633 }
634 string++;
635 if (*string == '%')
636 goto append;
637 for (j = 0; j < num_keys; j++) {
638 if (strchr(keys[j].key, *string) != NULL) {
639 i = strlcat(buf, keys[j].repl, sizeof(buf));
640 if (i >= sizeof(buf))
641 fatal("percent_expand: string too long");
642 break;
643 }
644 }
645 if (j >= num_keys)
646 fatal("percent_expand: unknown key %%%c", *string);
647 }
648 return (xstrdup(buf));
649#undef EXPAND_MAX_KEYS
650}
651
652/*
Darren Tucker22cc7412004-12-06 22:47:41 +1100653 * Read an entire line from a public key file into a static buffer, discarding
654 * lines that exceed the buffer size. Returns 0 on success, -1 on failure.
655 */
656int
657read_keyfile_line(FILE *f, const char *filename, char *buf, size_t bufsz,
Darren Tuckerf0f90982004-12-11 13:39:50 +1100658 u_long *lineno)
Darren Tucker22cc7412004-12-06 22:47:41 +1100659{
660 while (fgets(buf, bufsz, f) != NULL) {
Damien Miller3ca8b772007-01-05 16:24:47 +1100661 if (buf[0] == '\0')
662 continue;
Darren Tucker22cc7412004-12-06 22:47:41 +1100663 (*lineno)++;
664 if (buf[strlen(buf) - 1] == '\n' || feof(f)) {
665 return 0;
666 } else {
Darren Tuckerf0f90982004-12-11 13:39:50 +1100667 debug("%s: %s line %lu exceeds size limit", __func__,
668 filename, *lineno);
Darren Tucker22cc7412004-12-06 22:47:41 +1100669 /* discard remainder of line */
Darren Tucker47eede72005-03-14 23:08:12 +1100670 while (fgetc(f) != '\n' && !feof(f))
Darren Tucker22cc7412004-12-06 22:47:41 +1100671 ; /* nothing */
672 }
673 }
674 return -1;
675}
Damien Miller13390022005-07-06 09:44:19 +1000676
Damien Millerd27b9472005-12-13 19:29:02 +1100677int
Damien Miller7b58e802005-12-13 19:33:19 +1100678tun_open(int tun, int mode)
Damien Millerd27b9472005-12-13 19:29:02 +1100679{
Damien Miller62a31c92005-12-13 20:44:13 +1100680#if defined(CUSTOM_SYS_TUN_OPEN)
681 return (sys_tun_open(tun, mode));
Damien Miller2dcddbf2006-01-01 19:47:05 +1100682#elif defined(SSH_TUN_OPENBSD)
Damien Miller7b58e802005-12-13 19:33:19 +1100683 struct ifreq ifr;
Damien Millerd27b9472005-12-13 19:29:02 +1100684 char name[100];
Damien Miller7b58e802005-12-13 19:33:19 +1100685 int fd = -1, sock;
Damien Millerd27b9472005-12-13 19:29:02 +1100686
Damien Miller7b58e802005-12-13 19:33:19 +1100687 /* Open the tunnel device */
688 if (tun <= SSH_TUNID_MAX) {
Damien Millerd27b9472005-12-13 19:29:02 +1100689 snprintf(name, sizeof(name), "/dev/tun%d", tun);
Damien Miller7b58e802005-12-13 19:33:19 +1100690 fd = open(name, O_RDWR);
691 } else if (tun == SSH_TUNID_ANY) {
692 for (tun = 100; tun >= 0; tun--) {
693 snprintf(name, sizeof(name), "/dev/tun%d", tun);
694 if ((fd = open(name, O_RDWR)) >= 0)
695 break;
Damien Millerd27b9472005-12-13 19:29:02 +1100696 }
697 } else {
Damien Millera210d522006-01-02 23:40:30 +1100698 debug("%s: invalid tunnel %u", __func__, tun);
Damien Miller7b58e802005-12-13 19:33:19 +1100699 return (-1);
Damien Millerd27b9472005-12-13 19:29:02 +1100700 }
Damien Miller7b58e802005-12-13 19:33:19 +1100701
702 if (fd < 0) {
703 debug("%s: %s open failed: %s", __func__, name, strerror(errno));
704 return (-1);
705 }
706
707 debug("%s: %s mode %d fd %d", __func__, name, mode, fd);
708
709 /* Set the tunnel device operation mode */
710 snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "tun%d", tun);
711 if ((sock = socket(PF_UNIX, SOCK_STREAM, 0)) == -1)
712 goto failed;
713
714 if (ioctl(sock, SIOCGIFFLAGS, &ifr) == -1)
715 goto failed;
Damien Millera1d9a182006-01-02 23:41:21 +1100716
717 /* Set interface mode */
718 ifr.ifr_flags &= ~IFF_UP;
719 if (mode == SSH_TUNMODE_ETHERNET)
Damien Miller7b58e802005-12-13 19:33:19 +1100720 ifr.ifr_flags |= IFF_LINK0;
Damien Millera1d9a182006-01-02 23:41:21 +1100721 else
722 ifr.ifr_flags &= ~IFF_LINK0;
723 if (ioctl(sock, SIOCSIFFLAGS, &ifr) == -1)
724 goto failed;
725
726 /* Bring interface up */
Damien Miller7b58e802005-12-13 19:33:19 +1100727 ifr.ifr_flags |= IFF_UP;
728 if (ioctl(sock, SIOCSIFFLAGS, &ifr) == -1)
729 goto failed;
730
731 close(sock);
732 return (fd);
733
734 failed:
735 if (fd >= 0)
736 close(fd);
737 if (sock >= 0)
738 close(sock);
739 debug("%s: failed to set %s mode %d: %s", __func__, name,
740 mode, strerror(errno));
Damien Millerd27b9472005-12-13 19:29:02 +1100741 return (-1);
Damien Miller62a31c92005-12-13 20:44:13 +1100742#else
743 error("Tunnel interfaces are not supported on this platform");
744 return (-1);
745#endif
Damien Millerd27b9472005-12-13 19:29:02 +1100746}
747
Darren Tuckerce321d82005-10-03 18:11:24 +1000748void
749sanitise_stdfd(void)
750{
Damien Miller72c5b7d2006-01-06 14:50:44 +1100751 int nullfd, dupfd;
Darren Tuckerce321d82005-10-03 18:11:24 +1000752
Damien Miller72c5b7d2006-01-06 14:50:44 +1100753 if ((nullfd = dupfd = open(_PATH_DEVNULL, O_RDWR)) == -1) {
Damien Miller9eab9562009-02-22 08:47:02 +1100754 fprintf(stderr, "Couldn't open /dev/null: %s\n",
755 strerror(errno));
Darren Tuckerce321d82005-10-03 18:11:24 +1000756 exit(1);
757 }
Damien Miller72c5b7d2006-01-06 14:50:44 +1100758 while (++dupfd <= 2) {
759 /* Only clobber closed fds */
760 if (fcntl(dupfd, F_GETFL, 0) >= 0)
761 continue;
762 if (dup2(nullfd, dupfd) == -1) {
Damien Miller9eab9562009-02-22 08:47:02 +1100763 fprintf(stderr, "dup2: %s\n", strerror(errno));
Darren Tuckerce321d82005-10-03 18:11:24 +1000764 exit(1);
765 }
Darren Tuckerce321d82005-10-03 18:11:24 +1000766 }
767 if (nullfd > 2)
768 close(nullfd);
769}
770
Damien Miller13390022005-07-06 09:44:19 +1000771char *
Damien Miller3f941882006-03-31 23:13:02 +1100772tohex(const void *vp, size_t l)
Damien Miller13390022005-07-06 09:44:19 +1000773{
Damien Miller3f941882006-03-31 23:13:02 +1100774 const u_char *p = (const u_char *)vp;
Damien Miller13390022005-07-06 09:44:19 +1000775 char b[3], *r;
Damien Miller3f941882006-03-31 23:13:02 +1100776 size_t i, hl;
777
778 if (l > 65536)
779 return xstrdup("tohex: length > 65536");
Damien Miller13390022005-07-06 09:44:19 +1000780
781 hl = l * 2 + 1;
Damien Miller07d86be2006-03-26 14:19:21 +1100782 r = xcalloc(1, hl);
Damien Miller13390022005-07-06 09:44:19 +1000783 for (i = 0; i < l; i++) {
Damien Miller3f941882006-03-31 23:13:02 +1100784 snprintf(b, sizeof(b), "%02x", p[i]);
Damien Miller13390022005-07-06 09:44:19 +1000785 strlcat(r, b, hl);
786 }
787 return (r);
788}
789
Damien Miller3f941882006-03-31 23:13:02 +1100790u_int64_t
791get_u64(const void *vp)
792{
793 const u_char *p = (const u_char *)vp;
794 u_int64_t v;
795
796 v = (u_int64_t)p[0] << 56;
797 v |= (u_int64_t)p[1] << 48;
798 v |= (u_int64_t)p[2] << 40;
799 v |= (u_int64_t)p[3] << 32;
800 v |= (u_int64_t)p[4] << 24;
801 v |= (u_int64_t)p[5] << 16;
802 v |= (u_int64_t)p[6] << 8;
803 v |= (u_int64_t)p[7];
804
805 return (v);
806}
807
808u_int32_t
809get_u32(const void *vp)
810{
811 const u_char *p = (const u_char *)vp;
812 u_int32_t v;
813
814 v = (u_int32_t)p[0] << 24;
815 v |= (u_int32_t)p[1] << 16;
816 v |= (u_int32_t)p[2] << 8;
817 v |= (u_int32_t)p[3];
818
819 return (v);
820}
821
822u_int16_t
823get_u16(const void *vp)
824{
825 const u_char *p = (const u_char *)vp;
826 u_int16_t v;
827
828 v = (u_int16_t)p[0] << 8;
829 v |= (u_int16_t)p[1];
830
831 return (v);
832}
833
834void
835put_u64(void *vp, u_int64_t v)
836{
837 u_char *p = (u_char *)vp;
838
839 p[0] = (u_char)(v >> 56) & 0xff;
840 p[1] = (u_char)(v >> 48) & 0xff;
841 p[2] = (u_char)(v >> 40) & 0xff;
842 p[3] = (u_char)(v >> 32) & 0xff;
843 p[4] = (u_char)(v >> 24) & 0xff;
844 p[5] = (u_char)(v >> 16) & 0xff;
845 p[6] = (u_char)(v >> 8) & 0xff;
846 p[7] = (u_char)v & 0xff;
847}
848
849void
850put_u32(void *vp, u_int32_t v)
851{
852 u_char *p = (u_char *)vp;
853
854 p[0] = (u_char)(v >> 24) & 0xff;
855 p[1] = (u_char)(v >> 16) & 0xff;
856 p[2] = (u_char)(v >> 8) & 0xff;
857 p[3] = (u_char)v & 0xff;
858}
859
860
861void
862put_u16(void *vp, u_int16_t v)
863{
864 u_char *p = (u_char *)vp;
865
866 p[0] = (u_char)(v >> 8) & 0xff;
867 p[1] = (u_char)v & 0xff;
868}
Darren Tucker3fc464e2008-06-13 06:42:45 +1000869
870void
871ms_subtract_diff(struct timeval *start, int *ms)
872{
873 struct timeval diff, finish;
874
875 gettimeofday(&finish, NULL);
876 timersub(&finish, start, &diff);
877 *ms -= (diff.tv_sec * 1000) + (diff.tv_usec / 1000);
878}
879
880void
881ms_to_timeval(struct timeval *tv, int ms)
882{
883 if (ms < 0)
884 ms = 0;
885 tv->tv_sec = ms / 1000;
886 tv->tv_usec = (ms % 1000) * 1000;
887}
888
Damien Miller04ee0f82009-11-18 17:48:30 +1100889void
890sock_set_v6only(int s)
891{
892#ifdef IPV6_V6ONLY
893 int on = 1;
894
895 debug3("%s: set socket %d IPV6_V6ONLY", __func__, s);
896 if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on)) == -1)
897 error("setsockopt IPV6_V6ONLY: %s", strerror(errno));
898#endif
899}