blob: 8cb411ccc92b633d5f0c48f5910b96a927e01316 [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 Tuckerc7a6fc42004-08-13 21:18:00 +100026RCSID("$OpenBSD: misc.c,v 1.25 2004/08/11 21:43:05 avsm 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
Ben Lindstrom4529b702001-05-03 23:39:53 +0000278char *
279cleanhostname(char *host)
280{
281 if (*host == '[' && host[strlen(host) - 1] == ']') {
282 host[strlen(host) - 1] = '\0';
283 return (host + 1);
284 } else
285 return host;
286}
287
288char *
289colon(char *cp)
290{
291 int flag = 0;
292
293 if (*cp == ':') /* Leading colon is part of file name. */
294 return (0);
295 if (*cp == '[')
296 flag = 1;
297
298 for (; *cp; ++cp) {
299 if (*cp == '@' && *(cp+1) == '[')
300 flag = 1;
301 if (*cp == ']' && *(cp+1) == ':' && flag)
302 return (cp+1);
303 if (*cp == ':' && !flag)
304 return (cp);
305 if (*cp == '/')
306 return (0);
307 }
308 return (0);
309}
310
Ben Lindstrom4cc240d2001-07-04 04:46:56 +0000311/* function to assist building execv() arguments */
Ben Lindstrom387c4722001-05-08 20:27:25 +0000312void
313addargs(arglist *args, char *fmt, ...)
314{
315 va_list ap;
316 char buf[1024];
Darren Tuckerc7a6fc42004-08-13 21:18:00 +1000317 u_int nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000318
319 va_start(ap, fmt);
320 vsnprintf(buf, sizeof(buf), fmt, ap);
321 va_end(ap);
322
Darren Tuckerfb16b242003-09-22 21:04:23 +1000323 nalloc = args->nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000324 if (args->list == NULL) {
Darren Tuckerfb16b242003-09-22 21:04:23 +1000325 nalloc = 32;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000326 args->num = 0;
Darren Tuckerfb16b242003-09-22 21:04:23 +1000327 } else if (args->num+2 >= nalloc)
328 nalloc *= 2;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000329
Darren Tuckerfb16b242003-09-22 21:04:23 +1000330 args->list = xrealloc(args->list, nalloc * sizeof(char *));
331 args->nalloc = nalloc;
Ben Lindstrom387c4722001-05-08 20:27:25 +0000332 args->list[args->num++] = xstrdup(buf);
333 args->list[args->num] = NULL;
334}