1 /* 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 25 #include "includes.h" 26 RCSID("$OpenBSD: misc.c,v 1.25 2004/08/11 21:43:05 avsm Exp $"); 27 28 #include "misc.h" 29 #include "log.h" 30 #include "xmalloc.h" 31 32 /* remove newline at end of string */ 33 char * 34 chop(char *s) 35 { 36 char *t = s; 37 while (*t) { 38 if (*t == '\n' || *t == '\r') { 39 *t = '\0'; 40 return s; 41 } 42 t++; 43 } 44 return s; 45 46 } 47 48 /* set/unset filedescriptor to non-blocking */ 49 int 50 set_nonblock(int fd) 51 { 52 int val; 53 54 val = fcntl(fd, F_GETFL, 0); 55 if (val < 0) { 56 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno)); 57 return (-1); 58 } 59 if (val & O_NONBLOCK) { 60 debug3("fd %d is O_NONBLOCK", fd); 61 return (0); 62 } 63 debug2("fd %d setting O_NONBLOCK", fd); 64 val |= O_NONBLOCK; 65 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); 71 } 72 73 int 74 unset_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)); 81 return (-1); 82 } 83 if (!(val & O_NONBLOCK)) { 84 debug3("fd %d is not O_NONBLOCK", fd); 85 return (0); 86 } 87 debug("fd %d clearing O_NONBLOCK", fd); 88 val &= ~O_NONBLOCK; 89 if (fcntl(fd, F_SETFL, val) == -1) { 90 debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s", 91 fd, strerror(errno)); 92 return (-1); 93 } 94 return (0); 95 } 96 97 /* disable nagle on socket */ 98 void 99 set_nodelay(int fd) 100 { 101 int opt; 102 socklen_t optlen; 103 104 optlen = sizeof opt; 105 if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) { 106 debug("getsockopt TCP_NODELAY: %.100s", strerror(errno)); 107 return; 108 } 109 if (opt == 1) { 110 debug2("fd %d is TCP_NODELAY", fd); 111 return; 112 } 113 opt = 1; 114 debug2("fd %d setting TCP_NODELAY", fd); 115 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1) 116 error("setsockopt TCP_NODELAY: %.100s", strerror(errno)); 117 } 118 119 /* Characters considered whitespace in strsep calls. */ 120 #define WHITESPACE " \t\r\n" 121 122 /* return next token in configuration line */ 123 char * 124 strdelim(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 } 149 150 struct passwd * 151 pwcopy(struct passwd *pw) 152 { 153 struct passwd *copy = xmalloc(sizeof(*copy)); 154 155 memset(copy, 0, sizeof(*copy)); 156 copy->pw_name = xstrdup(pw->pw_name); 157 copy->pw_passwd = xstrdup(pw->pw_passwd); 158 copy->pw_gecos = xstrdup(pw->pw_gecos); 159 copy->pw_uid = pw->pw_uid; 160 copy->pw_gid = pw->pw_gid; 161 copy->pw_expire = pw->pw_expire; 162 copy->pw_change = pw->pw_change; 163 copy->pw_class = xstrdup(pw->pw_class); 164 copy->pw_dir = xstrdup(pw->pw_dir); 165 copy->pw_shell = xstrdup(pw->pw_shell); 166 return copy; 167 } 168 169 /* 170 * Convert ASCII string to TCP/IP port number. 171 * Port must be >0 and <=65535. 172 * Return 0 if invalid. 173 */ 174 int 175 a2port(const char *s) 176 { 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 190 #define SECONDS 1 191 #define MINUTES (SECONDS * 60) 192 #define HOURS (MINUTES * 60) 193 #define DAYS (HOURS * 24) 194 #define WEEKS (DAYS * 7) 195 196 /* 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 */ 217 long 218 convtime(const char *s) 219 { 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 272 char * 273 cleanhostname(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 282 char * 283 colon(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 305 /* function to assist building execv() arguments */ 306 void 307 addargs(arglist *args, char *fmt, ...) 308 { 309 va_list ap; 310 char buf[1024]; 311 u_int nalloc; 312 313 va_start(ap, fmt); 314 vsnprintf(buf, sizeof(buf), fmt, ap); 315 va_end(ap); 316 317 nalloc = args->nalloc; 318 if (args->list == NULL) { 319 nalloc = 32; 320 args->num = 0; 321 } else if (args->num+2 >= nalloc) 322 nalloc *= 2; 323 324 args->list = xrealloc(args->list, nalloc * sizeof(char *)); 325 args->nalloc = nalloc; 326 args->list[args->num++] = xstrdup(buf); 327 args->list[args->num] = NULL; 328 } 329