blob: 1f320353ef16a9803ba4dd53f9ba722b491c42ce [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"
Darren Tucker6db8f932003-11-03 20:07:14 +110026RCSID("$OpenBSD: misc.c,v 1.23 2003/10/28 09:08:06 markus 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 Miller61c51502000-08-18 14:01:04 +100049void
50set_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));
57 return;
58 }
Damien Miller69b69aa2000-10-28 14:19:58 +110059 if (val & O_NONBLOCK) {
Ben Lindstromc93e84c2001-05-12 00:08:37 +000060 debug2("fd %d is O_NONBLOCK", fd);
Damien Miller61c51502000-08-18 14:01:04 +100061 return;
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;
65 if (fcntl(fd, F_SETFL, val) == -1)
Ben Lindstrom84fcb312002-03-05 01:48:09 +000066 debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
67 fd, strerror(errno));
Damien Miller61c51502000-08-18 14:01:04 +100068}
69
Ben Lindstromc93e84c2001-05-12 00:08:37 +000070void
71unset_nonblock(int fd)
72{
73 int val;
74
75 val = fcntl(fd, F_GETFL, 0);
76 if (val < 0) {
77 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
78 return;
79 }
80 if (!(val & O_NONBLOCK)) {
81 debug2("fd %d is not O_NONBLOCK", fd);
82 return;
83 }
Ben Lindstrom352b1c22001-06-21 03:04:37 +000084 debug("fd %d clearing O_NONBLOCK", fd);
Ben Lindstromc93e84c2001-05-12 00:08:37 +000085 val &= ~O_NONBLOCK;
86 if (fcntl(fd, F_SETFL, val) == -1)
Ben Lindstrom84fcb312002-03-05 01:48:09 +000087 debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
88 fd, strerror(errno));
Ben Lindstromc93e84c2001-05-12 00:08:37 +000089}
90
Damien Miller398e1cf2002-02-05 11:52:13 +110091/* disable nagle on socket */
92void
93set_nodelay(int fd)
94{
Ben Lindstrome86de512002-03-05 01:28:14 +000095 int opt;
96 socklen_t optlen;
Damien Miller398e1cf2002-02-05 11:52:13 +110097
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +000098 optlen = sizeof opt;
99 if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) {
Darren Tucker6db8f932003-11-03 20:07:14 +1100100 debug("getsockopt TCP_NODELAY: %.100s", strerror(errno));
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000101 return;
102 }
103 if (opt == 1) {
104 debug2("fd %d is TCP_NODELAY", fd);
105 return;
106 }
107 opt = 1;
Ben Lindstrom1d568f92002-12-23 02:44:36 +0000108 debug2("fd %d setting TCP_NODELAY", fd);
Ben Lindstrom1ebd7a52002-02-26 18:12:51 +0000109 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1)
Damien Miller398e1cf2002-02-05 11:52:13 +1100110 error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
111}
112
Damien Miller61c51502000-08-18 14:01:04 +1000113/* Characters considered whitespace in strsep calls. */
114#define WHITESPACE " \t\r\n"
115
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000116/* return next token in configuration line */
Damien Miller61c51502000-08-18 14:01:04 +1000117char *
118strdelim(char **s)
119{
120 char *old;
121 int wspace = 0;
122
123 if (*s == NULL)
124 return NULL;
125
126 old = *s;
127
128 *s = strpbrk(*s, WHITESPACE "=");
129 if (*s == NULL)
130 return (old);
131
132 /* Allow only one '=' to be skipped */
133 if (*s[0] == '=')
134 wspace = 1;
135 *s[0] = '\0';
136
137 *s += strspn(*s + 1, WHITESPACE) + 1;
138 if (*s[0] == '=' && !wspace)
139 *s += strspn(*s + 1, WHITESPACE) + 1;
140
141 return (old);
142}
Kevin Stevesb6e773a2001-02-04 13:20:36 +0000143
Ben Lindstrom086cf212001-03-05 05:56:40 +0000144struct passwd *
145pwcopy(struct passwd *pw)
146{
147 struct passwd *copy = xmalloc(sizeof(*copy));
Ben Lindstrom40304422001-03-05 06:22:01 +0000148
Ben Lindstrom086cf212001-03-05 05:56:40 +0000149 memset(copy, 0, sizeof(*copy));
150 copy->pw_name = xstrdup(pw->pw_name);
151 copy->pw_passwd = xstrdup(pw->pw_passwd);
Ben Lindstrom40304422001-03-05 06:22:01 +0000152 copy->pw_gecos = xstrdup(pw->pw_gecos);
Ben Lindstrom086cf212001-03-05 05:56:40 +0000153 copy->pw_uid = pw->pw_uid;
154 copy->pw_gid = pw->pw_gid;
Kevin Steves82456952001-06-22 21:14:18 +0000155#ifdef HAVE_PW_EXPIRE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000156 copy->pw_expire = pw->pw_expire;
Kevin Steves82456952001-06-22 21:14:18 +0000157#endif
158#ifdef HAVE_PW_CHANGE_IN_PASSWD
Ben Lindstrom3af4d462001-06-21 03:11:27 +0000159 copy->pw_change = pw->pw_change;
Kevin Steves82456952001-06-22 21:14:18 +0000160#endif
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000161#ifdef HAVE_PW_CLASS_IN_PASSWD
Ben Lindstrom086cf212001-03-05 05:56:40 +0000162 copy->pw_class = xstrdup(pw->pw_class);
Ben Lindstrom0f68db42001-03-05 07:57:09 +0000163#endif
Ben Lindstrom086cf212001-03-05 05:56:40 +0000164 copy->pw_dir = xstrdup(pw->pw_dir);
165 copy->pw_shell = xstrdup(pw->pw_shell);
166 return copy;
167}
168
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000169/*
170 * Convert ASCII string to TCP/IP port number.
171 * Port must be >0 and <=65535.
172 * Return 0 if invalid.
173 */
174int
175a2port(const char *s)
Ben Lindstrom19066a12001-04-12 23:39:26 +0000176{
177 long port;
178 char *endp;
179
180 errno = 0;
181 port = strtol(s, &endp, 0);
182 if (s == endp || *endp != '\0' ||
183 (errno == ERANGE && (port == LONG_MIN || port == LONG_MAX)) ||
184 port <= 0 || port > 65535)
185 return 0;
186
187 return port;
188}
189
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000190#define SECONDS 1
191#define MINUTES (SECONDS * 60)
192#define HOURS (MINUTES * 60)
193#define DAYS (HOURS * 24)
194#define WEEKS (DAYS * 7)
195
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000196/*
197 * Convert a time string into seconds; format is
198 * a sequence of:
199 * time[qualifier]
200 *
201 * Valid time qualifiers are:
202 * <none> seconds
203 * s|S seconds
204 * m|M minutes
205 * h|H hours
206 * d|D days
207 * w|W weeks
208 *
209 * Examples:
210 * 90m 90 minutes
211 * 1h30m 90 minutes
212 * 2d 2 days
213 * 1w 1 week
214 *
215 * Return -1 if time string is invalid.
216 */
217long
218convtime(const char *s)
Ben Lindstrom1bda4c82001-06-05 19:59:08 +0000219{
220 long total, secs;
221 const char *p;
222 char *endp;
223
224 errno = 0;
225 total = 0;
226 p = s;
227
228 if (p == NULL || *p == '\0')
229 return -1;
230
231 while (*p) {
232 secs = strtol(p, &endp, 10);
233 if (p == endp ||
234 (errno == ERANGE && (secs == LONG_MIN || secs == LONG_MAX)) ||
235 secs < 0)
236 return -1;
237
238 switch (*endp++) {
239 case '\0':
240 endp--;
241 case 's':
242 case 'S':
243 break;
244 case 'm':
245 case 'M':
246 secs *= MINUTES;
247 break;
248 case 'h':
249 case 'H':
250 secs *= HOURS;
251 break;
252 case 'd':
253 case 'D':
254 secs *= DAYS;
255 break;
256 case 'w':
257 case 'W':
258 secs *= WEEKS;
259 break;
260 default:
261 return -1;
262 }
263 total += secs;
264 if (total < 0)
265 return -1;
266 p = endp;
267 }
268
269 return total;
270}
271
Ben Lindstrom4529b702001-05-03 23:39:53 +0000272char *
273cleanhostname(char *host)
274{
275 if (*host == '[' && host[strlen(host) - 1] == ']') {
276 host[strlen(host) - 1] = '\0';
277 return (host + 1);
278 } else
279 return host;
280}
281
282char *
283colon(char *cp)
284{
285 int flag = 0;
286
287 if (*cp == ':') /* Leading colon is part of file name. */
288 return (0);
289 if (*cp == '[')
290 flag = 1;
291
292 for (; *cp; ++cp) {
293 if (*cp == '@' && *(cp+1) == '[')
294 flag = 1;
295 if (*cp == ']' && *(cp+1) == ':' && flag)
296 return (cp+1);
297 if (*cp == ':' && !flag)
298 return (cp);
299 if (*cp == '/')
300 return (0);
301 }
302 return (0);
303}
304
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000305/* function to assist building execv() arguments */
Ben Lindstrom387c4722001-05-08 20:27:25 +0000306void
307addargs(arglist *args, char *fmt, ...)
308{
309 va_list ap;
310 char buf[1024];
Darren Tuckerfb16b242003-09-22 21:04:23 +1000311 int nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000312
313 va_start(ap, fmt);
314 vsnprintf(buf, sizeof(buf), fmt, ap);
315 va_end(ap);
316
Darren Tuckerfb16b242003-09-22 21:04:23 +1000317 nalloc = args->nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000318 if (args->list == NULL) {
Darren Tuckerfb16b242003-09-22 21:04:23 +1000319 nalloc = 32;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000320 args->num = 0;
Darren Tuckerfb16b242003-09-22 21:04:23 +1000321 } else if (args->num+2 >= nalloc)
322 nalloc *= 2;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000323
Darren Tuckerfb16b242003-09-22 21:04:23 +1000324 args->list = xrealloc(args->list, nalloc * sizeof(char *));
325 args->nalloc = nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000326 args->list[args->num++] = xstrdup(buf);
327 args->list[args->num] = NULL;
328}