login_cap.c revision 1.27 1 /* $NetBSD: login_cap.c,v 1.27 2007/02/04 08:19:26 elad Exp $ */
2
3 /*-
4 * Copyright (c) 1995,1997 Berkeley Software Design, Inc. All rights reserved.
5 *
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 * 3. All advertising materials mentioning features or use of this software
15 * must display the following acknowledgement:
16 * This product includes software developed by Berkeley Software Design,
17 * Inc.
18 * 4. The name of Berkeley Software Design, Inc. may not be used to endorse
19 * or promote products derived from this software without specific prior
20 * written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY BERKELEY SOFTWARE DESIGN, INC. ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL BERKELEY SOFTWARE DESIGN, INC. BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 *
34 * BSDI login_cap.c,v 2.13 1998/02/07 03:17:05 prb Exp
35 */
36
37 #include <sys/cdefs.h>
38 #if defined(LIBC_SCCS) && !defined(lint)
39 __RCSID("$NetBSD: login_cap.c,v 1.27 2007/02/04 08:19:26 elad Exp $");
40 #endif /* LIBC_SCCS and not lint */
41
42 #include <sys/types.h>
43 #include <sys/stat.h>
44 #include <sys/time.h>
45 #include <sys/resource.h>
46 #include <sys/param.h>
47
48 #include <assert.h>
49 #include <ctype.h>
50 #include <err.h>
51 #include <errno.h>
52 #include <fcntl.h>
53 #include <limits.h>
54 #include <login_cap.h>
55 #include <paths.h>
56 #include <pwd.h>
57 #include <stdio.h>
58 #include <stdlib.h>
59 #include <string.h>
60 #include <syslog.h>
61 #include <unistd.h>
62 #include <util.h>
63
64 static u_quad_t multiply(u_quad_t, u_quad_t);
65 static u_quad_t strtolimit(const char *, char **, int);
66 static u_quad_t strtosize(const char *, char **, int);
67 static int gsetrl(login_cap_t *, int, const char *, int type);
68 static int isinfinite(const char *);
69 static int envset(void *, const char *, const char *, int);
70
71 login_cap_t *
72 login_getclass(const char *class)
73 {
74 const char *classfiles[2];
75 login_cap_t *lc;
76 int res;
77
78 /* class may be NULL */
79
80 if (secure_path(_PATH_LOGIN_CONF) == 0) {
81 classfiles[0] = _PATH_LOGIN_CONF;
82 classfiles[1] = NULL;
83 } else {
84 classfiles[0] = NULL;
85 }
86
87 if ((lc = malloc(sizeof(login_cap_t))) == NULL) {
88 syslog(LOG_ERR, "%s:%d malloc: %m", __FILE__, __LINE__);
89 return (0);
90 }
91
92 lc->lc_cap = 0;
93 lc->lc_style = 0;
94
95 if (class == NULL || class[0] == '\0')
96 class = LOGIN_DEFCLASS;
97
98 if ((lc->lc_class = strdup(class)) == NULL) {
99 syslog(LOG_ERR, "%s:%d strdup: %m", __FILE__, __LINE__);
100 free(lc);
101 return (0);
102 }
103
104 /*
105 * Not having a login.conf file is not an error condition.
106 * The individual routines deal reasonably with missing
107 * capabilities and use default values.
108 */
109 if (classfiles[0] == NULL)
110 return(lc);
111
112 if ((res = cgetent(&lc->lc_cap, classfiles, lc->lc_class)) != 0) {
113 lc->lc_cap = 0;
114 switch (res) {
115 case 1:
116 syslog(LOG_ERR, "%s: couldn't resolve 'tc'",
117 lc->lc_class);
118 break;
119 case -1:
120 if (strcmp(lc->lc_class, LOGIN_DEFCLASS) == 0)
121 return (lc);
122 syslog(LOG_ERR, "%s: unknown class", lc->lc_class);
123 break;
124 case -2:
125 syslog(LOG_ERR, "%s: getting class information: %m",
126 lc->lc_class);
127 break;
128 case -3:
129 syslog(LOG_ERR, "%s: 'tc' reference loop",
130 lc->lc_class);
131 break;
132 default:
133 syslog(LOG_ERR, "%s: unexpected cgetent error",
134 lc->lc_class);
135 break;
136 }
137 free(lc->lc_class);
138 free(lc);
139 return (0);
140 }
141 return (lc);
142 }
143
144 login_cap_t *
145 login_getpwclass(const struct passwd *pwd)
146 {
147
148 /* pwd may be NULL */
149
150 return login_getclass(pwd ? pwd->pw_class : NULL);
151 }
152
153 char *
154 login_getcapstr(login_cap_t *lc, const char *cap, char *def, char *e)
155 {
156 char *res = NULL;
157 int status;
158
159 errno = 0;
160
161 _DIAGASSERT(cap != NULL);
162
163 if (!lc || !lc->lc_cap)
164 return (def);
165
166 switch (status = cgetstr(lc->lc_cap, cap, &res)) {
167 case -1:
168 if (res)
169 free(res);
170 return (def);
171 case -2:
172 syslog(LOG_ERR, "%s: getting capability %s: %m",
173 lc->lc_class, cap);
174 if (res)
175 free(res);
176 return (e);
177 default:
178 if (status >= 0)
179 return (res);
180 syslog(LOG_ERR, "%s: unexpected error with capability %s",
181 lc->lc_class, cap);
182 if (res)
183 free(res);
184 return (e);
185 }
186 }
187
188 quad_t
189 login_getcaptime(login_cap_t *lc, const char *cap, quad_t def, quad_t e)
190 {
191 char *ep;
192 char *res = NULL, *sres;
193 int status;
194 quad_t q, r;
195
196 _DIAGASSERT(cap != NULL);
197
198 errno = 0;
199 if (!lc || !lc->lc_cap)
200 return (def);
201
202 switch (status = cgetstr(lc->lc_cap, cap, &res)) {
203 case -1:
204 if (res)
205 free(res);
206 return (def);
207 case -2:
208 syslog(LOG_ERR, "%s: getting capability %s: %m",
209 lc->lc_class, cap);
210 errno = ERANGE;
211 if (res)
212 free(res);
213 return (e);
214 default:
215 if (status >= 0)
216 break;
217 syslog(LOG_ERR, "%s: unexpected error with capability %s",
218 lc->lc_class, cap);
219 errno = ERANGE;
220 if (res)
221 free(res);
222 return (e);
223 }
224
225 if (isinfinite(res))
226 return (RLIM_INFINITY);
227
228 errno = 0;
229
230 q = 0;
231 sres = res;
232 while (*res) {
233 r = strtoq(res, &ep, 0);
234 if (!ep || ep == res ||
235 ((r == QUAD_MIN || r == QUAD_MAX) && errno == ERANGE)) {
236 invalid:
237 syslog(LOG_ERR, "%s:%s=%s: invalid time",
238 lc->lc_class, cap, sres);
239 errno = ERANGE;
240 free(sres);
241 return (e);
242 }
243 switch (*ep++) {
244 case '\0':
245 --ep;
246 break;
247 case 's': case 'S':
248 break;
249 case 'm': case 'M':
250 r *= 60;
251 break;
252 case 'h': case 'H':
253 r *= 60 * 60;
254 break;
255 case 'd': case 'D':
256 r *= 60 * 60 * 24;
257 break;
258 case 'w': case 'W':
259 r *= 60 * 60 * 24 * 7;
260 break;
261 case 'y': case 'Y': /* Pretty absurd */
262 r *= 60 * 60 * 24 * 365;
263 break;
264 default:
265 goto invalid;
266 }
267 res = ep;
268 q += r;
269 }
270 free(sres);
271 return (q);
272 }
273
274 quad_t
275 login_getcapnum(login_cap_t *lc, const char *cap, quad_t def, quad_t e)
276 {
277 char *ep;
278 char *res = NULL;
279 int status;
280 quad_t q;
281
282 _DIAGASSERT(cap != NULL);
283
284 errno = 0;
285 if (!lc || !lc->lc_cap)
286 return (def);
287
288 switch (status = cgetstr(lc->lc_cap, cap, &res)) {
289 case -1:
290 if (res)
291 free(res);
292 return (def);
293 case -2:
294 syslog(LOG_ERR, "%s: getting capability %s: %m",
295 lc->lc_class, cap);
296 errno = ERANGE;
297 if (res)
298 free(res);
299 return (e);
300 default:
301 if (status >= 0)
302 break;
303 syslog(LOG_ERR, "%s: unexpected error with capability %s",
304 lc->lc_class, cap);
305 errno = ERANGE;
306 if (res)
307 free(res);
308 return (e);
309 }
310
311 if (isinfinite(res))
312 return (RLIM_INFINITY);
313
314 errno = 0;
315 q = strtoq(res, &ep, 0);
316 if (!ep || ep == res || ep[0] ||
317 ((q == QUAD_MIN || q == QUAD_MAX) && errno == ERANGE)) {
318 syslog(LOG_ERR, "%s:%s=%s: invalid number",
319 lc->lc_class, cap, res);
320 errno = ERANGE;
321 free(res);
322 return (e);
323 }
324 free(res);
325 return (q);
326 }
327
328 quad_t
329 login_getcapsize(login_cap_t *lc, const char *cap, quad_t def, quad_t e)
330 {
331 char *ep;
332 char *res = NULL;
333 int status;
334 quad_t q;
335
336 _DIAGASSERT(cap != NULL);
337
338 errno = 0;
339
340 if (!lc || !lc->lc_cap)
341 return (def);
342
343 switch (status = cgetstr(lc->lc_cap, cap, &res)) {
344 case -1:
345 if (res)
346 free(res);
347 return (def);
348 case -2:
349 syslog(LOG_ERR, "%s: getting capability %s: %m",
350 lc->lc_class, cap);
351 errno = ERANGE;
352 if (res)
353 free(res);
354 return (e);
355 default:
356 if (status >= 0)
357 break;
358 syslog(LOG_ERR, "%s: unexpected error with capability %s",
359 lc->lc_class, cap);
360 errno = ERANGE;
361 if (res)
362 free(res);
363 return (e);
364 }
365
366 errno = 0;
367 q = strtolimit(res, &ep, 0);
368 if (!ep || ep == res || (ep[0] && ep[1]) ||
369 ((q == QUAD_MIN || q == QUAD_MAX) && errno == ERANGE)) {
370 syslog(LOG_ERR, "%s:%s=%s: invalid size",
371 lc->lc_class, cap, res);
372 errno = ERANGE;
373 free(res);
374 return (e);
375 }
376 free(res);
377 return (q);
378 }
379
380 int
381 login_getcapbool(login_cap_t *lc, const char *cap, u_int def)
382 {
383
384 _DIAGASSERT(cap != NULL);
385
386 if (!lc || !lc->lc_cap)
387 return (def);
388
389 return (cgetcap(lc->lc_cap, cap, ':') != NULL);
390 }
391
392 void
393 login_close(login_cap_t *lc)
394 {
395
396 if (lc) {
397 if (lc->lc_class)
398 free(lc->lc_class);
399 if (lc->lc_cap)
400 free(lc->lc_cap);
401 if (lc->lc_style)
402 free(lc->lc_style);
403 free(lc);
404 }
405 }
406
407 #define R_CTIME 1
408 #define R_CSIZE 2
409 #define R_CNUMB 3
410
411 static struct {
412 int what;
413 int type;
414 const char *name;
415 } r_list[] = {
416 { RLIMIT_CPU, R_CTIME, "cputime", },
417 { RLIMIT_FSIZE, R_CSIZE, "filesize", },
418 { RLIMIT_DATA, R_CSIZE, "datasize", },
419 { RLIMIT_STACK, R_CSIZE, "stacksize", },
420 { RLIMIT_RSS, R_CSIZE, "memoryuse", },
421 { RLIMIT_MEMLOCK, R_CSIZE, "memorylocked", },
422 { RLIMIT_NPROC, R_CNUMB, "maxproc", },
423 { RLIMIT_NOFILE, R_CNUMB, "openfiles", },
424 { RLIMIT_CORE, R_CSIZE, "coredumpsize", },
425 { RLIMIT_SBSIZE, R_CSIZE, "sbsize", },
426 { -1, 0, 0 }
427 };
428
429 static int
430 gsetrl(login_cap_t *lc, int what, const char *name, int type)
431 {
432 struct rlimit rl;
433 struct rlimit r;
434 char name_cur[32];
435 char name_max[32];
436
437 _DIAGASSERT(name != NULL);
438
439 (void)snprintf(name_cur, sizeof(name_cur), "%s-cur", name);
440 (void)snprintf(name_max, sizeof(name_max), "%s-max", name);
441
442 if (getrlimit(what, &r)) {
443 syslog(LOG_ERR, "getting resource limit: %m");
444 return (-1);
445 }
446
447 #define RCUR r.rlim_cur
448 #define RMAX r.rlim_max
449
450 switch (type) {
451 case R_CTIME:
452 RCUR = login_getcaptime(lc, name, RCUR, RCUR);
453 RMAX = login_getcaptime(lc, name, RMAX, RMAX);
454 rl.rlim_cur = login_getcaptime(lc, name_cur, RCUR, RCUR);
455 rl.rlim_max = login_getcaptime(lc, name_max, RMAX, RMAX);
456 break;
457 case R_CSIZE:
458 RCUR = login_getcapsize(lc, name, RCUR, RCUR);
459 RMAX = login_getcapsize(lc, name, RMAX, RMAX);
460 rl.rlim_cur = login_getcapsize(lc, name_cur, RCUR, RCUR);
461 rl.rlim_max = login_getcapsize(lc, name_max, RMAX, RMAX);
462 break;
463 case R_CNUMB:
464 RCUR = login_getcapnum(lc, name, RCUR, RCUR);
465 RMAX = login_getcapnum(lc, name, RMAX, RMAX);
466 rl.rlim_cur = login_getcapnum(lc, name_cur, RCUR, RCUR);
467 rl.rlim_max = login_getcapnum(lc, name_max, RMAX, RMAX);
468 break;
469 default:
470 syslog(LOG_ERR, "%s: invalid type %d setting resource limit %s",
471 lc->lc_class, type, name);
472 return (-1);
473 }
474
475 if (setrlimit(what, &rl)) {
476 syslog(LOG_ERR, "%s: setting resource limit %s: %m",
477 lc->lc_class, name);
478 return (-1);
479 }
480 #undef RCUR
481 #undef RMAX
482 return (0);
483 }
484
485 static int
486 /*ARGSUSED*/
487 envset(void *envp __unused, const char *name, const char *value, int overwrite)
488 {
489 return setenv(name, value, overwrite);
490 }
491
492 int
493 setuserenv(login_cap_t *lc, envfunc_t senv, void *envp)
494 {
495 const char *stop = ", \t";
496 size_t i, count;
497 char *ptr;
498 char **res;
499 char *str = login_getcapstr(lc, "setenv", NULL, NULL);
500
501 if (str == NULL || *str == '\0')
502 return 0;
503
504 /*
505 * count the sub-strings, this may over-count since we don't
506 * account for escaped delimiters.
507 */
508 for (i = 1, ptr = str; *ptr; i++) {
509 ptr += strcspn(ptr, stop);
510 if (*ptr)
511 ptr++;
512 }
513
514 /* allocate ptr array and string */
515 count = i;
516 res = malloc(count * sizeof(char *) + strlen(str) + 1);
517
518 if (!res)
519 return -1;
520
521 ptr = (char *)(void *)&res[count];
522 (void)strcpy(ptr, str);
523
524 /* split string */
525 for (i = 0; (res[i] = stresep(&ptr, stop, '\\')) != NULL; )
526 if (*res[i])
527 i++;
528
529 count = i;
530
531 for (i = 0; i < count; i++) {
532 if ((ptr = strchr(res[i], '=')) != NULL)
533 *ptr++ = '\0';
534 else
535 ptr = NULL;
536 (void)(*senv)(envp, res[i], ptr ? ptr : "", 1);
537 }
538
539 free(res);
540 return 0;
541 }
542
543 int
544 setclasscontext(const char *class, u_int flags)
545 {
546 int ret;
547 login_cap_t *lc;
548
549 flags &= LOGIN_SETRESOURCES | LOGIN_SETPRIORITY | LOGIN_SETUMASK |
550 LOGIN_SETPATH;
551
552 lc = login_getclass(class);
553 ret = lc ? setusercontext(lc, NULL, 0, flags) : -1;
554 login_close(lc);
555 return (ret);
556 }
557
558 int
559 setusercontext(login_cap_t *lc, struct passwd *pwd, uid_t uid, u_int flags)
560 {
561 char per_user_tmp[MAXPATHLEN + 1];
562 login_cap_t *flc;
563 quad_t p;
564 int i;
565
566 flc = NULL;
567
568 if (!lc)
569 flc = lc = login_getclass(pwd ? pwd->pw_class : NULL);
570
571 /*
572 * Without the pwd entry being passed we cannot set either
573 * the group or the login. We could complain about it.
574 */
575 if (pwd == NULL)
576 flags &= ~(LOGIN_SETGROUP|LOGIN_SETLOGIN);
577
578 if (flags & LOGIN_SETRESOURCES)
579 for (i = 0; r_list[i].name; ++i)
580 (void)gsetrl(lc, r_list[i].what, r_list[i].name,
581 r_list[i].type);
582
583 if (flags & LOGIN_SETPRIORITY) {
584 p = login_getcapnum(lc, "priority", (quad_t)0, (quad_t)0);
585
586 if (setpriority(PRIO_PROCESS, 0, (int)p) < 0)
587 syslog(LOG_ERR, "%s: setpriority: %m", lc->lc_class);
588 }
589
590 if (flags & LOGIN_SETUMASK) {
591 p = login_getcapnum(lc, "umask", (quad_t) LOGIN_DEFUMASK,
592 (quad_t) LOGIN_DEFUMASK);
593 umask((mode_t)p);
594 }
595
596 if (flags & LOGIN_SETGROUP) {
597 if (setgid(pwd->pw_gid) < 0) {
598 syslog(LOG_ERR, "setgid(%d): %m", pwd->pw_gid);
599 login_close(flc);
600 return (-1);
601 }
602
603 if (initgroups(pwd->pw_name, pwd->pw_gid) < 0) {
604 syslog(LOG_ERR, "initgroups(%s,%d): %m",
605 pwd->pw_name, pwd->pw_gid);
606 login_close(flc);
607 return (-1);
608 }
609 }
610
611 /* Create per-user temporary directories if needed. */
612 if (readlink("/tmp", per_user_tmp, sizeof(per_user_tmp)) != -1) {
613 static const char atuid[] = "/@uid";
614 char *lp;
615
616 /* Check if it's magic symlink. */
617 lp = strstr(per_user_tmp, atuid);
618 if (lp != NULL && *(lp + (sizeof(atuid) - 1)) == '\0') {
619 lp++;
620
621 if ((sizeof(per_user_tmp) - (lp - per_user_tmp)) < 64) {
622 syslog(LOG_ERR, "real temporary path too long");
623 login_close(flc);
624 return (-1);
625 }
626 (void)sprintf(lp, "/%u", pwd->pw_uid); /* safe */
627 if (mkdir(per_user_tmp, S_IRWXU) != -1) {
628 (void)chown(per_user_tmp, pwd->pw_uid,
629 pwd->pw_gid);
630 } else {
631 syslog(LOG_ERR, "can't create `%s' directory",
632 per_user_tmp);
633 }
634 }
635 }
636 errno = 0;
637
638 if (flags & LOGIN_SETLOGIN)
639 if (setlogin(pwd->pw_name) < 0) {
640 syslog(LOG_ERR, "setlogin(%s) failure: %m",
641 pwd->pw_name);
642 login_close(flc);
643 return (-1);
644 }
645
646 if (flags & LOGIN_SETUSER)
647 if (setuid(uid) < 0) {
648 syslog(LOG_ERR, "setuid(%d): %m", uid);
649 login_close(flc);
650 return (-1);
651 }
652
653 if (flags & LOGIN_SETENV)
654 setuserenv(lc, envset, NULL);
655
656 if (flags & LOGIN_SETPATH)
657 setuserpath(lc, pwd ? pwd->pw_dir : "", envset, NULL);
658
659 login_close(flc);
660 return (0);
661 }
662
663 void
664 setuserpath(login_cap_t *lc, const char *home, envfunc_t senv, void *envp)
665 {
666 size_t hlen, plen;
667 int cnt = 0;
668 char *path;
669 const char *cpath;
670 char *p, *q;
671
672 _DIAGASSERT(home != NULL);
673
674 hlen = strlen(home);
675
676 p = path = login_getcapstr(lc, "path", NULL, NULL);
677 if (p) {
678 while (*p)
679 if (*p++ == '~')
680 ++cnt;
681 plen = (p - path) + cnt * (hlen + 1) + 1;
682 p = path;
683 q = path = malloc(plen);
684 if (q) {
685 while (*p) {
686 p += strspn(p, " \t");
687 if (*p == '\0')
688 break;
689 plen = strcspn(p, " \t");
690 if (hlen == 0 && *p == '~') {
691 p += plen;
692 continue;
693 }
694 if (q != path)
695 *q++ = ':';
696 if (*p == '~') {
697 strcpy(q, home);
698 q += hlen;
699 ++p;
700 --plen;
701 }
702 memcpy(q, p, plen);
703 p += plen;
704 q += plen;
705 }
706 *q = '\0';
707 cpath = path;
708 } else
709 cpath = _PATH_DEFPATH;
710 } else
711 cpath = _PATH_DEFPATH;
712 if ((*senv)(envp, "PATH", cpath, 1))
713 warn("could not set PATH");
714 }
715
716 /*
717 * Convert an expression of the following forms
718 * 1) A number.
719 * 2) A number followed by a b (mult by 512).
720 * 3) A number followed by a k (mult by 1024).
721 * 5) A number followed by a m (mult by 1024 * 1024).
722 * 6) A number followed by a g (mult by 1024 * 1024 * 1024).
723 * 7) A number followed by a t (mult by 1024 * 1024 * 1024 * 1024).
724 * 8) Two or more numbers (with/without k,b,m,g, or t).
725 * separated by x (also * for backwards compatibility), specifying
726 * the product of the indicated values.
727 */
728 static u_quad_t
729 strtosize(const char *str, char **endptr, int radix)
730 {
731 u_quad_t num, num2;
732 char *expr, *expr2;
733
734 _DIAGASSERT(str != NULL);
735 /* endptr may be NULL */
736
737 errno = 0;
738 num = strtouq(str, &expr, radix);
739 if (errno || expr == str) {
740 if (endptr)
741 *endptr = expr;
742 return (num);
743 }
744
745 switch(*expr) {
746 case 'b': case 'B':
747 num = multiply(num, (u_quad_t)512);
748 ++expr;
749 break;
750 case 'k': case 'K':
751 num = multiply(num, (u_quad_t)1024);
752 ++expr;
753 break;
754 case 'm': case 'M':
755 num = multiply(num, (u_quad_t)1024 * 1024);
756 ++expr;
757 break;
758 case 'g': case 'G':
759 num = multiply(num, (u_quad_t)1024 * 1024 * 1024);
760 ++expr;
761 break;
762 case 't': case 'T':
763 num = multiply(num, (u_quad_t)1024 * 1024);
764 num = multiply(num, (u_quad_t)1024 * 1024);
765 ++expr;
766 break;
767 }
768
769 if (errno)
770 goto erange;
771
772 switch(*expr) {
773 case '*': /* Backward compatible. */
774 case 'x':
775 num2 = strtosize(expr+1, &expr2, radix);
776 if (errno) {
777 expr = expr2;
778 goto erange;
779 }
780
781 if (expr2 == expr + 1) {
782 if (endptr)
783 *endptr = expr;
784 return (num);
785 }
786 expr = expr2;
787 num = multiply(num, num2);
788 if (errno)
789 goto erange;
790 break;
791 }
792 if (endptr)
793 *endptr = expr;
794 return (num);
795 erange:
796 if (endptr)
797 *endptr = expr;
798 errno = ERANGE;
799 return (UQUAD_MAX);
800 }
801
802 static u_quad_t
803 strtolimit(const char *str, char **endptr, int radix)
804 {
805
806 _DIAGASSERT(str != NULL);
807 /* endptr may be NULL */
808
809 if (isinfinite(str)) {
810 if (endptr)
811 *endptr = (char *)__UNCONST(str) + strlen(str);
812 return ((u_quad_t)RLIM_INFINITY);
813 }
814 return (strtosize(str, endptr, radix));
815 }
816
817 static int
818 isinfinite(const char *s)
819 {
820 static const char *infs[] = {
821 "infinity",
822 "inf",
823 "unlimited",
824 "unlimit",
825 NULL
826 };
827 const char **i;
828
829 _DIAGASSERT(s != NULL);
830
831 for (i = infs; *i; i++) {
832 if (!strcasecmp(s, *i))
833 return 1;
834 }
835 return 0;
836 }
837
838 static u_quad_t
839 multiply(u_quad_t n1, u_quad_t n2)
840 {
841 static int bpw = 0;
842 u_quad_t m;
843 u_quad_t r;
844 int b1, b2;
845
846 /*
847 * Get rid of the simple cases
848 */
849 if (n1 == 0 || n2 == 0)
850 return (0);
851 if (n1 == 1)
852 return (n2);
853 if (n2 == 1)
854 return (n1);
855
856 /*
857 * sizeof() returns number of bytes needed for storage.
858 * This may be different from the actual number of useful bits.
859 */
860 if (!bpw) {
861 bpw = sizeof(u_quad_t) * 8;
862 while (((u_quad_t)1 << (bpw-1)) == 0)
863 --bpw;
864 }
865
866 /*
867 * First check the magnitude of each number. If the sum of the
868 * magnatude is way to high, reject the number. (If this test
869 * is not done then the first multiply below may overflow.)
870 */
871 for (b1 = bpw; (((u_quad_t)1 << (b1-1)) & n1) == 0; --b1)
872 ;
873 for (b2 = bpw; (((u_quad_t)1 << (b2-1)) & n2) == 0; --b2)
874 ;
875 if (b1 + b2 - 2 > bpw) {
876 errno = ERANGE;
877 return (UQUAD_MAX);
878 }
879
880 /*
881 * Decompose the multiplication to be:
882 * h1 = n1 & ~1
883 * h2 = n2 & ~1
884 * l1 = n1 & 1
885 * l2 = n2 & 1
886 * (h1 + l1) * (h2 + l2)
887 * (h1 * h2) + (h1 * l2) + (l1 * h2) + (l1 * l2)
888 *
889 * Since h1 && h2 do not have the low bit set, we can then say:
890 *
891 * (h1>>1 * h2>>1 * 4) + ...
892 *
893 * So if (h1>>1 * h2>>1) > (1<<(bpw - 2)) then the result will
894 * overflow.
895 *
896 * Finally, if MAX - ((h1 * l2) + (l1 * h2) + (l1 * l2)) < (h1*h2)
897 * then adding in residual amout will cause an overflow.
898 */
899
900 m = (n1 >> 1) * (n2 >> 1);
901
902 if (m >= ((u_quad_t)1 << (bpw-2))) {
903 errno = ERANGE;
904 return (UQUAD_MAX);
905 }
906
907 m *= 4;
908
909 r = (n1 & n2 & 1)
910 + (n2 & 1) * (n1 & ~(u_quad_t)1)
911 + (n1 & 1) * (n2 & ~(u_quad_t)1);
912
913 if ((u_quad_t)(m + r) < m) {
914 errno = ERANGE;
915 return (UQUAD_MAX);
916 }
917 m += r;
918
919 return (m);
920 }
921