blob: 2e366f81b270055cd99a291cfa9795f6d59fa898 [file] [log] [blame]
Damien Millere4340be2000-09-16 13:29:08 +11001/*
2 * Copyright (c) 2000 Markus Friedl. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
14 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
15 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
16 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
17 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
18 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
22 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 */
24
Damien Miller61c51502000-08-18 14:01:04 +100025#include "includes.h"
Damien Millerf91ee4c2005-03-01 21:24:33 +110026RCSID("$OpenBSD: misc.c,v 1.28 2005/03/01 10:09:52 djm Exp $");
Damien Miller61c51502000-08-18 14:01:04 +100027
Kevin Stevesb6e773a2001-02-04 13:20:36 +000028#include "misc.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000029#include "log.h"
Ben Lindstrom06909012001-03-05 06:09:31 +000030#include "xmalloc.h"
Damien Miller61c51502000-08-18 14:01:04 +100031
Ben Lindstrom4cc240d2001-07-04 04:46:56 +000032/* remove newline at end of string */
Damien Miller61c51502000-08-18 14:01:04 +100033char *
34chop(char *s)
35{
36 char *t = s;
37 while (*t) {
Ben Lindstrom1c37c6a2001-12-06 18:00:18 +000038 if (*t == '\n' || *t == '\r') {
Damien Miller61c51502000-08-18 14:01:04 +100039 *t = '\0';
40 return s;
41 }
42 t++;
43 }
44 return s;
45
46}
47
Ben Lindstrom4cc240d2001-07-04 04:46:56 +000048/* set/unset filedescriptor to non-blocking */
Damien Miller232711f2004-06-15 10:35:30 +100049int
Damien Miller61c51502000-08-18 14:01:04 +100050set_nonblock(int fd)
51{
52 int val;
Ben Lindstromc93e84c2001-05-12 00:08:37 +000053
Damien Miller61c51502000-08-18 14:01:04 +100054 val = fcntl(fd, F_GETFL, 0);
55 if (val < 0) {
56 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +100057 return (-1);
Damien Miller61c51502000-08-18 14:01:04 +100058 }
Damien Miller69b69aa2000-10-28 14:19:58 +110059 if (val & O_NONBLOCK) {
Damien Miller232711f2004-06-15 10:35:30 +100060 debug3("fd %d is O_NONBLOCK", fd);
61 return (0);
Damien Miller69b69aa2000-10-28 14:19:58 +110062 }
Damien Milleref095ce2003-05-14 13:41:39 +100063 debug2("fd %d setting O_NONBLOCK", fd);
Damien Miller61c51502000-08-18 14:01:04 +100064 val |= O_NONBLOCK;
Damien Miller232711f2004-06-15 10:35:30 +100065 if (fcntl(fd, F_SETFL, val) == -1) {
66 debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s", fd,
67 strerror(errno));
68 return (-1);
69 }
70 return (0);
Damien Miller61c51502000-08-18 14:01:04 +100071}
72
Damien Miller232711f2004-06-15 10:35:30 +100073int
Ben Lindstromc93e84c2001-05-12 00:08:37 +000074unset_nonblock(int fd)
75{
76 int val;
77
78 val = fcntl(fd, F_GETFL, 0);
79 if (val < 0) {
80 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +100081 return (-1);
Ben Lindstromc93e84c2001-05-12 00:08:37 +000082 }
83 if (!(val & O_NONBLOCK)) {
Damien Miller232711f2004-06-15 10:35:30 +100084 debug3("fd %d is not O_NONBLOCK", fd);
85 return (0);
Ben Lindstromc93e84c2001-05-12 00:08:37 +000086 }
Ben Lindstrom352b1c22001-06-21 03:04:37 +000087 debug("fd %d clearing O_NONBLOCK", fd);
Ben Lindstromc93e84c2001-05-12 00:08:37 +000088 val &= ~O_NONBLOCK;
Damien Miller232711f2004-06-15 10:35:30 +100089 if (fcntl(fd, F_SETFL, val) == -1) {
90 debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s",
Ben Lindstrom84fcb312002-03-05 01:48:09 +000091 fd, strerror(errno));
Damien Miller232711f2004-06-15 10:35:30 +100092 return (-1);
93 }
94 return (0);
Ben Lindstromc93e84c2001-05-12 00:08:37 +000095}
96
Damien Miller398e1cf2002-02-05 11:52:13 +110097/* disable nagle on socket */
98void
99set_nodelay(int fd)
100{
Ben Lindstrome86de512002-03-05 01:28:14 +0000101 int opt;
102 socklen_t optlen;
Damien Miller398e1cf2002-02-05 11:52:13 +1100103
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000104 optlen = sizeof opt;
105 if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) {
Darren Tucker6db8f932003-11-03 20:07:14 +1100106 debug("getsockopt TCP_NODELAY: %.100s", strerror(errno));
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000107 return;
108 }
109 if (opt == 1) {
110 debug2("fd %d is TCP_NODELAY", fd);
111 return;
112 }
113 opt = 1;
Ben Lindstrom1d568f92002-12-23 02:44:36 +0000114 debug2("fd %d setting TCP_NODELAY", fd);
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000115 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1)
Damien Miller398e1cf2002-02-05 11:52:13 +1100116 error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
117}
118
Damien Miller61c51502000-08-18 14:01:04 +1000119/* Characters considered whitespace in strsep calls. */
120#define WHITESPACE " \t\r\n"
121
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000122/* return next token in configuration line */
Damien Miller61c51502000-08-18 14:01:04 +1000123char *
124strdelim(char **s)
125{
126 char *old;
127 int wspace = 0;
128
129 if (*s == NULL)
130 return NULL;
131
132 old = *s;
133
134 *s = strpbrk(*s, WHITESPACE "=");
135 if (*s == NULL)
136 return (old);
137
138 /* Allow only one '=' to be skipped */
139 if (*s[0] == '=')
140 wspace = 1;
141 *s[0] = '\0';
142
143 *s += strspn(*s + 1, WHITESPACE) + 1;
144 if (*s[0] == '=' && !wspace)
145 *s += strspn(*s + 1, WHITESPACE) + 1;
146
147 return (old);
148}
Kevin Stevesb6e773a2001-02-04 13:20:36 +0000149
Ben Lindstrom086cf212001-03-05 05:56:40 +0000150struct passwd *
151pwcopy(struct passwd *pw)
152{
153 struct passwd *copy = xmalloc(sizeof(*copy));
Ben Lindstrom40304422001-03-05 06:22:01 +0000154
Ben Lindstrom086cf212001-03-05 05:56:40 +0000155 memset(copy, 0, sizeof(*copy));
156 copy->pw_name = xstrdup(pw->pw_name);
157 copy->pw_passwd = xstrdup(pw->pw_passwd);
Ben Lindstrom40304422001-03-05 06:22:01 +0000158 copy->pw_gecos = xstrdup(pw->pw_gecos);
Ben Lindstrom086cf212001-03-05 05:56:40 +0000159 copy->pw_uid = pw->pw_uid;
160 copy->pw_gid = pw->pw_gid;
Kevin Steves82456952001-06-22 21:14:18 +0000161#ifdef HAVE_PW_EXPIRE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000162 copy->pw_expire = pw->pw_expire;
Kevin Steves82456952001-06-22 21:14:18 +0000163#endif
164#ifdef HAVE_PW_CHANGE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000165 copy->pw_change = pw->pw_change;
Kevin Steves82456952001-06-22 21:14:18 +0000166#endif
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000167#ifdef HAVE_PW_CLASS_IN_PASSWD
Ben Lindstrom086cf212001-03-05 05:56:40 +0000168 copy->pw_class = xstrdup(pw->pw_class);
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000169#endif
Ben Lindstrom086cf212001-03-05 05:56:40 +0000170 copy->pw_dir = xstrdup(pw->pw_dir);
171 copy->pw_shell = xstrdup(pw->pw_shell);
172 return copy;
173}
174
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000175/*
176 * Convert ASCII string to TCP/IP port number.
177 * Port must be >0 and <=65535.
178 * Return 0 if invalid.
179 */
180int
181a2port(const char *s)
Ben Lindstrom19066a12001-04-12 23:39:26 +0000182{
183 long port;
184 char *endp;
185
186 errno = 0;
187 port = strtol(s, &endp, 0);
188 if (s == endp || *endp != '\0' ||
189 (errno == ERANGE && (port == LONG_MIN || port == LONG_MAX)) ||
190 port <= 0 || port > 65535)
191 return 0;
192
193 return port;
194}
195
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000196#define SECONDS 1
197#define MINUTES (SECONDS * 60)
198#define HOURS (MINUTES * 60)
199#define DAYS (HOURS * 24)
200#define WEEKS (DAYS * 7)
201
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000202/*
203 * Convert a time string into seconds; format is
204 * a sequence of:
205 * time[qualifier]
206 *
207 * Valid time qualifiers are:
208 * <none> seconds
209 * s|S seconds
210 * m|M minutes
211 * h|H hours
212 * d|D days
213 * w|W weeks
214 *
215 * Examples:
216 * 90m 90 minutes
217 * 1h30m 90 minutes
218 * 2d 2 days
219 * 1w 1 week
220 *
221 * Return -1 if time string is invalid.
222 */
223long
224convtime(const char *s)
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000225{
226 long total, secs;
227 const char *p;
228 char *endp;
229
230 errno = 0;
231 total = 0;
232 p = s;
233
234 if (p == NULL || *p == '\0')
235 return -1;
236
237 while (*p) {
238 secs = strtol(p, &endp, 10);
239 if (p == endp ||
240 (errno == ERANGE && (secs == LONG_MIN || secs == LONG_MAX)) ||
241 secs < 0)
242 return -1;
243
244 switch (*endp++) {
245 case '\0':
246 endp--;
247 case 's':
248 case 'S':
249 break;
250 case 'm':
251 case 'M':
252 secs *= MINUTES;
253 break;
254 case 'h':
255 case 'H':
256 secs *= HOURS;
257 break;
258 case 'd':
259 case 'D':
260 secs *= DAYS;
261 break;
262 case 'w':
263 case 'W':
264 secs *= WEEKS;
265 break;
266 default:
267 return -1;
268 }
269 total += secs;
270 if (total < 0)
271 return -1;
272 p = endp;
273 }
274
275 return total;
276}
277
Damien Millerf91ee4c2005-03-01 21:24:33 +1100278/*
279 * Search for next delimiter between hostnames/addresses and ports.
280 * Argument may be modified (for termination).
281 * Returns *cp if parsing succeeds.
282 * *cp is set to the start of the next delimiter, if one was found.
283 * If this is the last field, *cp is set to NULL.
284 */
285char *
286hpdelim(char **cp)
287{
288 char *s, *old;
289
290 if (cp == NULL || *cp == NULL)
291 return NULL;
292
293 old = s = *cp;
294 if (*s == '[') {
295 if ((s = strchr(s, ']')) == NULL)
296 return NULL;
297 else
298 s++;
299 } else if ((s = strpbrk(s, ":/")) == NULL)
300 s = *cp + strlen(*cp); /* skip to end (see first case below) */
301
302 switch (*s) {
303 case '\0':
304 *cp = NULL; /* no more fields*/
305 break;
306
307 case ':':
308 case '/':
309 *s = '\0'; /* terminate */
310 *cp = s + 1;
311 break;
312
313 default:
314 return NULL;
315 }
316
317 return old;
318}
319
Ben Lindstrom4529b702001-05-03 23:39:53 +0000320char *
321cleanhostname(char *host)
322{
323 if (*host == '[' && host[strlen(host) - 1] == ']') {
324 host[strlen(host) - 1] = '\0';
325 return (host + 1);
326 } else
327 return host;
328}
329
330char *
331colon(char *cp)
332{
333 int flag = 0;
334
335 if (*cp == ':') /* Leading colon is part of file name. */
336 return (0);
337 if (*cp == '[')
338 flag = 1;
339
340 for (; *cp; ++cp) {
341 if (*cp == '@' && *(cp+1) == '[')
342 flag = 1;
343 if (*cp == ']' && *(cp+1) == ':' && flag)
344 return (cp+1);
345 if (*cp == ':' && !flag)
346 return (cp);
347 if (*cp == '/')
348 return (0);
349 }
350 return (0);
351}
352
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000353/* function to assist building execv() arguments */
Ben Lindstrom387c4722001-05-08 20:27:25 +0000354void
355addargs(arglist *args, char *fmt, ...)
356{
357 va_list ap;
358 char buf[1024];
Darren Tuckerc7a6fc42004-08-13 21:18:00 +1000359 u_int nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000360
361 va_start(ap, fmt);
362 vsnprintf(buf, sizeof(buf), fmt, ap);
363 va_end(ap);
364
Darren Tuckerfb16b242003-09-22 21:04:23 +1000365 nalloc = args->nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000366 if (args->list == NULL) {
Darren Tuckerfb16b242003-09-22 21:04:23 +1000367 nalloc = 32;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000368 args->num = 0;
Darren Tuckerfb16b242003-09-22 21:04:23 +1000369 } else if (args->num+2 >= nalloc)
370 nalloc *= 2;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000371
Darren Tuckerfb16b242003-09-22 21:04:23 +1000372 args->list = xrealloc(args->list, nalloc * sizeof(char *));
373 args->nalloc = nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000374 args->list[args->num++] = xstrdup(buf);
375 args->list[args->num] = NULL;
376}
Darren Tucker22cc7412004-12-06 22:47:41 +1100377
378/*
379 * Read an entire line from a public key file into a static buffer, discarding
380 * lines that exceed the buffer size. Returns 0 on success, -1 on failure.
381 */
382int
383read_keyfile_line(FILE *f, const char *filename, char *buf, size_t bufsz,
Darren Tuckerf0f90982004-12-11 13:39:50 +1100384 u_long *lineno)
Darren Tucker22cc7412004-12-06 22:47:41 +1100385{
386 while (fgets(buf, bufsz, f) != NULL) {
387 (*lineno)++;
388 if (buf[strlen(buf) - 1] == '\n' || feof(f)) {
389 return 0;
390 } else {
Darren Tuckerf0f90982004-12-11 13:39:50 +1100391 debug("%s: %s line %lu exceeds size limit", __func__,
392 filename, *lineno);
Darren Tucker22cc7412004-12-06 22:47:41 +1100393 /* discard remainder of line */
394 while(fgetc(f) != '\n' && !feof(f))
395 ; /* nothing */
396 }
397 }
398 return -1;
399}