login_cap.c revision 1.12 1 1.12 itojun /* $NetBSD: login_cap.c,v 1.12 2003/09/08 16:19:40 itojun 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.12 itojun __RCSID("$NetBSD: login_cap.c,v 1.12 2003/09/08 16:19:40 itojun 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.9 itojun 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.1 mjl if (strcmp(lc->lc_class, LOGIN_DEFCLASS) == NULL &&
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.1 mjl char *res;
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.1 mjl char *res, *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.1 mjl char *res;
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.1 mjl char *res;
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.1 mjl { -1, 0, 0 }
429 1.1 mjl };
430 1.1 mjl
431 1.1 mjl static int
432 1.6 ad gsetrl(login_cap_t *lc, int what, char *name, int type)
433 1.1 mjl {
434 1.1 mjl struct rlimit rl;
435 1.1 mjl struct rlimit r;
436 1.1 mjl char name_cur[32];
437 1.1 mjl char name_max[32];
438 1.1 mjl
439 1.10 lukem _DIAGASSERT(name != NULL);
440 1.10 lukem
441 1.1 mjl sprintf(name_cur, "%s-cur", name);
442 1.1 mjl sprintf(name_max, "%s-max", name);
443 1.1 mjl
444 1.1 mjl if (getrlimit(what, &r)) {
445 1.1 mjl syslog(LOG_ERR, "getting resource limit: %m");
446 1.1 mjl return (-1);
447 1.1 mjl }
448 1.1 mjl
449 1.1 mjl #define RCUR r.rlim_cur
450 1.1 mjl #define RMAX r.rlim_max
451 1.1 mjl
452 1.1 mjl switch (type) {
453 1.7 ad case R_CTIME:
454 1.1 mjl RCUR = login_getcaptime(lc, name, RCUR, RCUR);
455 1.1 mjl RMAX = login_getcaptime(lc, name, RMAX, RMAX);
456 1.1 mjl rl.rlim_cur = login_getcaptime(lc, name_cur, RCUR, RCUR);
457 1.1 mjl rl.rlim_max = login_getcaptime(lc, name_max, RMAX, RMAX);
458 1.1 mjl break;
459 1.7 ad case R_CSIZE:
460 1.1 mjl RCUR = login_getcapsize(lc, name, RCUR, RCUR);
461 1.1 mjl RMAX = login_getcapsize(lc, name, RMAX, RMAX);
462 1.1 mjl rl.rlim_cur = login_getcapsize(lc, name_cur, RCUR, RCUR);
463 1.1 mjl rl.rlim_max = login_getcapsize(lc, name_max, RMAX, RMAX);
464 1.1 mjl break;
465 1.7 ad case R_CNUMB:
466 1.1 mjl RCUR = login_getcapnum(lc, name, RCUR, RCUR);
467 1.1 mjl RMAX = login_getcapnum(lc, name, RMAX, RMAX);
468 1.1 mjl rl.rlim_cur = login_getcapnum(lc, name_cur, RCUR, RCUR);
469 1.1 mjl rl.rlim_max = login_getcapnum(lc, name_max, RMAX, RMAX);
470 1.1 mjl break;
471 1.1 mjl default:
472 1.1 mjl return (-1);
473 1.1 mjl }
474 1.1 mjl
475 1.1 mjl if (setrlimit(what, &rl)) {
476 1.1 mjl syslog(LOG_ERR, "%s: setting resource limit %s: %m",
477 1.1 mjl lc->lc_class, name);
478 1.1 mjl return (-1);
479 1.1 mjl }
480 1.1 mjl #undef RCUR
481 1.1 mjl #undef RMAX
482 1.1 mjl return (0);
483 1.1 mjl }
484 1.1 mjl
485 1.4 mjl static int
486 1.6 ad setuserenv(login_cap_t *lc)
487 1.4 mjl {
488 1.4 mjl char *stop = ", \t";
489 1.4 mjl int i, count;
490 1.4 mjl char *ptr;
491 1.4 mjl char **res;
492 1.4 mjl char *str = login_getcapstr(lc, "setenv", NULL, NULL);
493 1.4 mjl
494 1.8 itojun if (str == NULL || *str == '\0')
495 1.4 mjl return 0;
496 1.4 mjl
497 1.4 mjl /* count the sub-strings */
498 1.4 mjl for (i = 1, ptr = str; *ptr; i++) {
499 1.4 mjl ptr += strcspn(ptr, stop);
500 1.4 mjl if (*ptr)
501 1.4 mjl ptr++;
502 1.8 itojun }
503 1.4 mjl
504 1.4 mjl /* allocate ptr array and string */
505 1.4 mjl count = i;
506 1.8 itojun res = malloc(count * sizeof(char *) + strlen(str) + 1);
507 1.4 mjl
508 1.8 itojun if (!res)
509 1.4 mjl return -1;
510 1.4 mjl
511 1.4 mjl ptr = (char *)res + count * sizeof(char *);
512 1.4 mjl strcpy(ptr, str);
513 1.4 mjl
514 1.4 mjl /* split string */
515 1.4 mjl for (i = 0; *ptr && i < count; i++) {
516 1.4 mjl res[i] = ptr;
517 1.4 mjl ptr += strcspn(ptr, stop);
518 1.4 mjl if (*ptr)
519 1.4 mjl *ptr++ = '\0';
520 1.8 itojun }
521 1.4 mjl
522 1.4 mjl res[i] = NULL;
523 1.4 mjl
524 1.4 mjl for (i = 0; i < count && res[i]; i++) {
525 1.4 mjl if (*res[i] != '\0') {
526 1.6 ad if ((ptr = strchr(res[i], '=')) != NULL)
527 1.4 mjl *ptr++ = '\0';
528 1.4 mjl else
529 1.4 mjl ptr = "";
530 1.4 mjl setenv(res[i], ptr, 1);
531 1.4 mjl }
532 1.4 mjl }
533 1.4 mjl
534 1.5 mjl free(res);
535 1.4 mjl return 0;
536 1.4 mjl }
537 1.4 mjl
538 1.1 mjl int
539 1.6 ad setclasscontext(char *class, u_int flags)
540 1.1 mjl {
541 1.1 mjl int ret;
542 1.1 mjl login_cap_t *lc;
543 1.1 mjl
544 1.1 mjl flags &= LOGIN_SETRESOURCES | LOGIN_SETPRIORITY | LOGIN_SETUMASK |
545 1.1 mjl LOGIN_SETPATH;
546 1.1 mjl
547 1.1 mjl lc = login_getclass(class);
548 1.1 mjl ret = lc ? setusercontext(lc, NULL, 0, flags) : -1;
549 1.1 mjl login_close(lc);
550 1.1 mjl return (ret);
551 1.1 mjl }
552 1.1 mjl
553 1.1 mjl int
554 1.6 ad setusercontext(login_cap_t *lc, struct passwd *pwd, uid_t uid, u_int flags)
555 1.1 mjl {
556 1.1 mjl login_cap_t *flc;
557 1.1 mjl quad_t p;
558 1.1 mjl int i;
559 1.1 mjl
560 1.1 mjl flc = NULL;
561 1.1 mjl
562 1.3 mjl if (!lc)
563 1.3 mjl flc = lc = login_getclass(pwd ? pwd->pw_class : NULL);
564 1.1 mjl
565 1.1 mjl /*
566 1.1 mjl * Without the pwd entry being passed we cannot set either
567 1.1 mjl * the group or the login. We could complain about it.
568 1.1 mjl */
569 1.1 mjl if (pwd == NULL)
570 1.1 mjl flags &= ~(LOGIN_SETGROUP|LOGIN_SETLOGIN);
571 1.1 mjl
572 1.1 mjl if (flags & LOGIN_SETRESOURCES)
573 1.1 mjl for (i = 0; r_list[i].name; ++i)
574 1.1 mjl if (gsetrl(lc, r_list[i].what, r_list[i].name,
575 1.1 mjl r_list[i].type))
576 1.1 mjl /* XXX - call syslog()? */;
577 1.1 mjl
578 1.1 mjl if (flags & LOGIN_SETPRIORITY) {
579 1.1 mjl p = login_getcapnum(lc, "priority", 0LL, 0LL);
580 1.1 mjl
581 1.1 mjl if (setpriority(PRIO_PROCESS, 0, (int)p) < 0)
582 1.1 mjl syslog(LOG_ERR, "%s: setpriority: %m", lc->lc_class);
583 1.1 mjl }
584 1.1 mjl
585 1.1 mjl if (flags & LOGIN_SETUMASK) {
586 1.1 mjl p = login_getcapnum(lc, "umask", (quad_t) LOGIN_DEFUMASK,
587 1.1 mjl (quad_t) LOGIN_DEFUMASK);
588 1.1 mjl umask((mode_t)p);
589 1.1 mjl }
590 1.1 mjl
591 1.1 mjl if (flags & LOGIN_SETGROUP) {
592 1.1 mjl if (setgid(pwd->pw_gid) < 0) {
593 1.1 mjl syslog(LOG_ERR, "setgid(%d): %m", pwd->pw_gid);
594 1.1 mjl login_close(flc);
595 1.1 mjl return (-1);
596 1.1 mjl }
597 1.1 mjl
598 1.1 mjl if (initgroups(pwd->pw_name, pwd->pw_gid) < 0) {
599 1.1 mjl syslog(LOG_ERR, "initgroups(%s,%d): %m",
600 1.1 mjl pwd->pw_name, pwd->pw_gid);
601 1.1 mjl login_close(flc);
602 1.1 mjl return (-1);
603 1.1 mjl }
604 1.1 mjl }
605 1.1 mjl
606 1.1 mjl if (flags & LOGIN_SETLOGIN)
607 1.1 mjl if (setlogin(pwd->pw_name) < 0) {
608 1.1 mjl syslog(LOG_ERR, "setlogin(%s) failure: %m",
609 1.1 mjl pwd->pw_name);
610 1.1 mjl login_close(flc);
611 1.1 mjl return (-1);
612 1.1 mjl }
613 1.1 mjl
614 1.1 mjl if (flags & LOGIN_SETUSER)
615 1.1 mjl if (setuid(uid) < 0) {
616 1.1 mjl syslog(LOG_ERR, "setuid(%d): %m", uid);
617 1.1 mjl login_close(flc);
618 1.1 mjl return (-1);
619 1.1 mjl }
620 1.4 mjl
621 1.4 mjl if (flags & LOGIN_SETENV)
622 1.4 mjl setuserenv(lc);
623 1.1 mjl
624 1.1 mjl if (flags & LOGIN_SETPATH)
625 1.1 mjl setuserpath(lc, pwd ? pwd->pw_dir : "");
626 1.1 mjl
627 1.1 mjl login_close(flc);
628 1.1 mjl return (0);
629 1.1 mjl }
630 1.1 mjl
631 1.1 mjl static void
632 1.6 ad setuserpath(login_cap_t *lc, char *home)
633 1.1 mjl {
634 1.1 mjl size_t hlen, plen;
635 1.1 mjl int cnt = 0;
636 1.1 mjl char *path;
637 1.1 mjl char *p, *q;
638 1.1 mjl
639 1.10 lukem _DIAGASSERT(home != NULL);
640 1.10 lukem
641 1.1 mjl hlen = strlen(home);
642 1.1 mjl
643 1.1 mjl p = path = login_getcapstr(lc, "path", NULL, NULL);
644 1.1 mjl if (p) {
645 1.1 mjl while (*p)
646 1.1 mjl if (*p++ == '~')
647 1.1 mjl ++cnt;
648 1.1 mjl plen = (p - path) + cnt * (hlen + 1) + 1;
649 1.1 mjl p = path;
650 1.1 mjl q = path = malloc(plen);
651 1.1 mjl if (q) {
652 1.1 mjl while (*p) {
653 1.1 mjl p += strspn(p, " \t");
654 1.1 mjl if (*p == '\0')
655 1.1 mjl break;
656 1.1 mjl plen = strcspn(p, " \t");
657 1.1 mjl if (hlen == 0 && *p == '~') {
658 1.1 mjl p += plen;
659 1.1 mjl continue;
660 1.1 mjl }
661 1.1 mjl if (q != path)
662 1.1 mjl *q++ = ':';
663 1.1 mjl if (*p == '~') {
664 1.1 mjl strcpy(q, home);
665 1.1 mjl q += hlen;
666 1.1 mjl ++p;
667 1.1 mjl --plen;
668 1.1 mjl }
669 1.1 mjl memcpy(q, p, plen);
670 1.1 mjl p += plen;
671 1.1 mjl q += plen;
672 1.1 mjl }
673 1.1 mjl *q = '\0';
674 1.1 mjl } else
675 1.1 mjl path = _PATH_DEFPATH;
676 1.1 mjl } else
677 1.1 mjl path = _PATH_DEFPATH;
678 1.1 mjl if (setenv("PATH", path, 1))
679 1.1 mjl warn("could not set PATH");
680 1.1 mjl }
681 1.1 mjl
682 1.1 mjl /*
683 1.1 mjl * Convert an expression of the following forms
684 1.1 mjl * 1) A number.
685 1.1 mjl * 2) A number followed by a b (mult by 512).
686 1.1 mjl * 3) A number followed by a k (mult by 1024).
687 1.1 mjl * 5) A number followed by a m (mult by 1024 * 1024).
688 1.1 mjl * 6) A number followed by a g (mult by 1024 * 1024 * 1024).
689 1.1 mjl * 7) A number followed by a t (mult by 1024 * 1024 * 1024 * 1024).
690 1.1 mjl * 8) Two or more numbers (with/without k,b,m,g, or t).
691 1.11 wiz * separated by x (also * for backwards compatibility), specifying
692 1.1 mjl * the product of the indicated values.
693 1.1 mjl */
694 1.6 ad static u_quad_t
695 1.6 ad strtosize(char *str, char **endptr, int radix)
696 1.1 mjl {
697 1.1 mjl u_quad_t num, num2;
698 1.1 mjl char *expr, *expr2;
699 1.1 mjl
700 1.10 lukem _DIAGASSERT(str != NULL);
701 1.10 lukem /* endptr may be NULL */
702 1.10 lukem
703 1.1 mjl errno = 0;
704 1.1 mjl num = strtouq(str, &expr, radix);
705 1.1 mjl if (errno || expr == str) {
706 1.1 mjl if (endptr)
707 1.1 mjl *endptr = expr;
708 1.1 mjl return (num);
709 1.1 mjl }
710 1.1 mjl
711 1.1 mjl switch(*expr) {
712 1.1 mjl case 'b': case 'B':
713 1.1 mjl num = multiply(num, (u_quad_t)512);
714 1.1 mjl ++expr;
715 1.1 mjl break;
716 1.1 mjl case 'k': case 'K':
717 1.1 mjl num = multiply(num, (u_quad_t)1024);
718 1.1 mjl ++expr;
719 1.1 mjl break;
720 1.1 mjl case 'm': case 'M':
721 1.1 mjl num = multiply(num, (u_quad_t)1024 * 1024);
722 1.1 mjl ++expr;
723 1.1 mjl break;
724 1.1 mjl case 'g': case 'G':
725 1.1 mjl num = multiply(num, (u_quad_t)1024 * 1024 * 1024);
726 1.1 mjl ++expr;
727 1.1 mjl break;
728 1.1 mjl case 't': case 'T':
729 1.1 mjl num = multiply(num, (u_quad_t)1024 * 1024);
730 1.1 mjl num = multiply(num, (u_quad_t)1024 * 1024);
731 1.1 mjl ++expr;
732 1.1 mjl break;
733 1.1 mjl }
734 1.1 mjl
735 1.1 mjl if (errno)
736 1.1 mjl goto erange;
737 1.1 mjl
738 1.1 mjl switch(*expr) {
739 1.1 mjl case '*': /* Backward compatible. */
740 1.1 mjl case 'x':
741 1.1 mjl num2 = strtosize(expr+1, &expr2, radix);
742 1.1 mjl if (errno) {
743 1.1 mjl expr = expr2;
744 1.1 mjl goto erange;
745 1.1 mjl }
746 1.1 mjl
747 1.1 mjl if (expr2 == expr + 1) {
748 1.1 mjl if (endptr)
749 1.1 mjl *endptr = expr;
750 1.1 mjl return (num);
751 1.1 mjl }
752 1.1 mjl expr = expr2;
753 1.1 mjl num = multiply(num, num2);
754 1.1 mjl if (errno)
755 1.1 mjl goto erange;
756 1.1 mjl break;
757 1.1 mjl }
758 1.1 mjl if (endptr)
759 1.1 mjl *endptr = expr;
760 1.1 mjl return (num);
761 1.1 mjl erange:
762 1.1 mjl if (endptr)
763 1.1 mjl *endptr = expr;
764 1.1 mjl errno = ERANGE;
765 1.1 mjl return (UQUAD_MAX);
766 1.1 mjl }
767 1.1 mjl
768 1.6 ad static u_quad_t
769 1.6 ad strtolimit(char *str, char **endptr, int radix)
770 1.1 mjl {
771 1.10 lukem
772 1.10 lukem _DIAGASSERT(str != NULL);
773 1.10 lukem /* endptr may be NULL */
774 1.10 lukem
775 1.5 mjl if (isinfinite(str)) {
776 1.1 mjl if (endptr)
777 1.1 mjl *endptr = str + strlen(str);
778 1.1 mjl return ((u_quad_t)RLIM_INFINITY);
779 1.1 mjl }
780 1.1 mjl return (strtosize(str, endptr, radix));
781 1.5 mjl }
782 1.5 mjl
783 1.5 mjl static int
784 1.5 mjl isinfinite(const char *s)
785 1.5 mjl {
786 1.5 mjl static const char *infs[] = {
787 1.5 mjl "infinity",
788 1.5 mjl "inf",
789 1.5 mjl "unlimited",
790 1.5 mjl "unlimit",
791 1.5 mjl NULL
792 1.5 mjl };
793 1.5 mjl const char **i;
794 1.10 lukem
795 1.10 lukem _DIAGASSERT(s != NULL);
796 1.5 mjl
797 1.8 itojun for (i = infs; *i; i++) {
798 1.5 mjl if (!strcasecmp(s, *i))
799 1.5 mjl return 1;
800 1.5 mjl }
801 1.5 mjl return 0;
802 1.1 mjl }
803 1.1 mjl
804 1.1 mjl static u_quad_t
805 1.6 ad multiply(u_quad_t n1, u_quad_t n2)
806 1.1 mjl {
807 1.1 mjl static int bpw = 0;
808 1.1 mjl u_quad_t m;
809 1.1 mjl u_quad_t r;
810 1.1 mjl int b1, b2;
811 1.1 mjl
812 1.1 mjl /*
813 1.1 mjl * Get rid of the simple cases
814 1.1 mjl */
815 1.1 mjl if (n1 == 0 || n2 == 0)
816 1.1 mjl return (0);
817 1.1 mjl if (n1 == 1)
818 1.1 mjl return (n2);
819 1.1 mjl if (n2 == 1)
820 1.1 mjl return (n1);
821 1.1 mjl
822 1.1 mjl /*
823 1.1 mjl * sizeof() returns number of bytes needed for storage.
824 1.1 mjl * This may be different from the actual number of useful bits.
825 1.1 mjl */
826 1.1 mjl if (!bpw) {
827 1.1 mjl bpw = sizeof(u_quad_t) * 8;
828 1.1 mjl while (((u_quad_t)1 << (bpw-1)) == 0)
829 1.1 mjl --bpw;
830 1.1 mjl }
831 1.1 mjl
832 1.1 mjl /*
833 1.1 mjl * First check the magnitude of each number. If the sum of the
834 1.1 mjl * magnatude is way to high, reject the number. (If this test
835 1.1 mjl * is not done then the first multiply below may overflow.)
836 1.1 mjl */
837 1.1 mjl for (b1 = bpw; (((u_quad_t)1 << (b1-1)) & n1) == 0; --b1)
838 1.1 mjl ;
839 1.1 mjl for (b2 = bpw; (((u_quad_t)1 << (b2-1)) & n2) == 0; --b2)
840 1.1 mjl ;
841 1.1 mjl if (b1 + b2 - 2 > bpw) {
842 1.1 mjl errno = ERANGE;
843 1.1 mjl return (UQUAD_MAX);
844 1.1 mjl }
845 1.1 mjl
846 1.1 mjl /*
847 1.1 mjl * Decompose the multiplication to be:
848 1.1 mjl * h1 = n1 & ~1
849 1.1 mjl * h2 = n2 & ~1
850 1.1 mjl * l1 = n1 & 1
851 1.1 mjl * l2 = n2 & 1
852 1.1 mjl * (h1 + l1) * (h2 + l2)
853 1.1 mjl * (h1 * h2) + (h1 * l2) + (l1 * h2) + (l1 * l2)
854 1.1 mjl *
855 1.1 mjl * Since h1 && h2 do not have the low bit set, we can then say:
856 1.1 mjl *
857 1.1 mjl * (h1>>1 * h2>>1 * 4) + ...
858 1.1 mjl *
859 1.1 mjl * So if (h1>>1 * h2>>1) > (1<<(bpw - 2)) then the result will
860 1.1 mjl * overflow.
861 1.1 mjl *
862 1.1 mjl * Finally, if MAX - ((h1 * l2) + (l1 * h2) + (l1 * l2)) < (h1*h2)
863 1.1 mjl * then adding in residual amout will cause an overflow.
864 1.1 mjl */
865 1.1 mjl
866 1.1 mjl m = (n1 >> 1) * (n2 >> 1);
867 1.1 mjl
868 1.1 mjl if (m >= ((u_quad_t)1 << (bpw-2))) {
869 1.1 mjl errno = ERANGE;
870 1.1 mjl return (UQUAD_MAX);
871 1.1 mjl }
872 1.1 mjl
873 1.1 mjl m *= 4;
874 1.1 mjl
875 1.1 mjl r = (n1 & n2 & 1)
876 1.1 mjl + (n2 & 1) * (n1 & ~(u_quad_t)1)
877 1.1 mjl + (n1 & 1) * (n2 & ~(u_quad_t)1);
878 1.1 mjl
879 1.1 mjl if ((u_quad_t)(m + r) < m) {
880 1.1 mjl errno = ERANGE;
881 1.1 mjl return (UQUAD_MAX);
882 1.1 mjl }
883 1.1 mjl m += r;
884 1.1 mjl
885 1.1 mjl return (m);
886 1.1 mjl }
887