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