2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
4 * Copyright (c) 2011 The FreeBSD Foundation
6 * This software was developed by Edward Tomasz Napierala under sponsorship
7 * from the FreeBSD Foundation.
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 * Processes may set login class name using setloginclass(2). This
35 * is usually done through call to setusercontext(3), by programs
36 * such as login(1), based on information from master.passwd(5). Kernel
37 * uses this information to enforce per-class resource limits. Current
38 * login class can be determined using id(1). Login class is inherited
39 * from the parent process during fork(2). If not set, it defaults
42 * Code in this file implements setloginclass(2) and getloginclass(2)
43 * system calls, and maintains class name storage and retrieval.
46 #include <sys/cdefs.h>
47 __FBSDID("$FreeBSD$");
49 #include <sys/param.h>
50 #include <sys/eventhandler.h>
51 #include <sys/kernel.h>
53 #include <sys/loginclass.h>
54 #include <sys/malloc.h>
55 #include <sys/types.h>
58 #include <sys/queue.h>
59 #include <sys/racct.h>
61 #include <sys/refcount.h>
62 #include <sys/rwlock.h>
63 #include <sys/sysproto.h>
64 #include <sys/systm.h>
66 static MALLOC_DEFINE(M_LOGINCLASS, "loginclass", "loginclass structures");
68 LIST_HEAD(, loginclass) loginclasses;
71 * Lock protecting loginclasses list.
73 static struct rwlock loginclasses_lock;
74 RW_SYSINIT(loginclasses_init, &loginclasses_lock, "loginclasses lock");
77 loginclass_hold(struct loginclass *lc)
80 refcount_acquire(&lc->lc_refcount);
84 loginclass_free(struct loginclass *lc)
87 if (refcount_release_if_not_last(&lc->lc_refcount))
90 rw_wlock(&loginclasses_lock);
91 if (!refcount_release(&lc->lc_refcount)) {
92 rw_wunlock(&loginclasses_lock);
96 racct_destroy(&lc->lc_racct);
97 LIST_REMOVE(lc, lc_next);
98 rw_wunlock(&loginclasses_lock);
100 free(lc, M_LOGINCLASS);
104 * Look up a loginclass struct for the parameter name.
105 * loginclasses_lock must be locked.
106 * Increase refcount on loginclass struct returned.
108 static struct loginclass *
109 loginclass_lookup(const char *name)
111 struct loginclass *lc;
113 rw_assert(&loginclasses_lock, RA_LOCKED);
114 LIST_FOREACH(lc, &loginclasses, lc_next)
115 if (strcmp(name, lc->lc_name) == 0) {
124 * Return loginclass structure with a corresponding name. Not
125 * performance critical, as it's used mainly by setloginclass(2),
126 * which happens once per login session. Caller has to use
127 * loginclass_free() on the returned value when it's no longer
131 loginclass_find(const char *name)
133 struct loginclass *lc, *new_lc;
135 if (name[0] == '\0' || strlen(name) >= MAXLOGNAME)
138 lc = curthread->td_ucred->cr_loginclass;
139 if (strcmp(name, lc->lc_name) == 0) {
144 rw_rlock(&loginclasses_lock);
145 lc = loginclass_lookup(name);
146 rw_runlock(&loginclasses_lock);
150 new_lc = malloc(sizeof(*new_lc), M_LOGINCLASS, M_ZERO | M_WAITOK);
151 racct_create(&new_lc->lc_racct);
152 refcount_init(&new_lc->lc_refcount, 1);
153 strcpy(new_lc->lc_name, name);
155 rw_wlock(&loginclasses_lock);
157 * There's a chance someone created our loginclass while we
158 * were in malloc and not holding the lock, so we have to
159 * make sure we don't insert a duplicate loginclass.
161 if ((lc = loginclass_lookup(name)) == NULL) {
162 LIST_INSERT_HEAD(&loginclasses, new_lc, lc_next);
163 rw_wunlock(&loginclasses_lock);
166 rw_wunlock(&loginclasses_lock);
167 racct_destroy(&new_lc->lc_racct);
168 free(new_lc, M_LOGINCLASS);
175 * Get login class name.
177 #ifndef _SYS_SYSPROTO_H_
178 struct getloginclass_args {
185 sys_getloginclass(struct thread *td, struct getloginclass_args *uap)
187 struct loginclass *lc;
190 lc = td->td_ucred->cr_loginclass;
191 lcnamelen = strlen(lc->lc_name) + 1;
192 if (lcnamelen > uap->namelen)
194 return (copyout(lc->lc_name, uap->namebuf, lcnamelen));
198 * Set login class name.
200 #ifndef _SYS_SYSPROTO_H_
201 struct setloginclass_args {
207 sys_setloginclass(struct thread *td, struct setloginclass_args *uap)
209 struct proc *p = td->td_proc;
211 char lcname[MAXLOGNAME];
212 struct loginclass *newlc;
213 struct ucred *newcred, *oldcred;
215 error = priv_check(td, PRIV_PROC_SETLOGINCLASS);
218 error = copyinstr(uap->namebuf, lcname, sizeof(lcname), NULL);
222 newlc = loginclass_find(lcname);
228 oldcred = crcopysafe(p, newcred);
229 newcred->cr_loginclass = newlc;
230 proc_set_cred(p, newcred);
232 racct_proc_ucred_changed(p, oldcred, newcred);
237 rctl_proc_ucred_changed(p, newcred);
240 loginclass_free(oldcred->cr_loginclass);
247 loginclass_racct_foreach(void (*callback)(struct racct *racct,
248 void *arg2, void *arg3), void (*pre)(void), void (*post)(void),
249 void *arg2, void *arg3)
251 struct loginclass *lc;
253 rw_rlock(&loginclasses_lock);
256 LIST_FOREACH(lc, &loginclasses, lc_next)
257 (callback)(lc->lc_racct, arg2, arg3);
260 rw_runlock(&loginclasses_lock);