xref: /freebsd/sys/kern/kern_rctl.c (revision 36af98697d92141dcd4f4cca56da38a426ccb089)
1ec125fbbSEdward Tomasz Napierala /*-
2ec125fbbSEdward Tomasz Napierala  * Copyright (c) 2010 The FreeBSD Foundation
3ec125fbbSEdward Tomasz Napierala  * All rights reserved.
4ec125fbbSEdward Tomasz Napierala  *
5ec125fbbSEdward Tomasz Napierala  * This software was developed by Edward Tomasz Napierala under sponsorship
6ec125fbbSEdward Tomasz Napierala  * from the FreeBSD Foundation.
7ec125fbbSEdward Tomasz Napierala  *
8ec125fbbSEdward Tomasz Napierala  * Redistribution and use in source and binary forms, with or without
9ec125fbbSEdward Tomasz Napierala  * modification, are permitted provided that the following conditions
10ec125fbbSEdward Tomasz Napierala  * are met:
11ec125fbbSEdward Tomasz Napierala  * 1. Redistributions of source code must retain the above copyright
12ec125fbbSEdward Tomasz Napierala  *    notice, this list of conditions and the following disclaimer.
13ec125fbbSEdward Tomasz Napierala  * 2. Redistributions in binary form must reproduce the above copyright
14ec125fbbSEdward Tomasz Napierala  *    notice, this list of conditions and the following disclaimer in the
15ec125fbbSEdward Tomasz Napierala  *    documentation and/or other materials provided with the distribution.
16ec125fbbSEdward Tomasz Napierala  *
17ec125fbbSEdward Tomasz Napierala  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18ec125fbbSEdward Tomasz Napierala  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19ec125fbbSEdward Tomasz Napierala  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20ec125fbbSEdward Tomasz Napierala  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21ec125fbbSEdward Tomasz Napierala  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22ec125fbbSEdward Tomasz Napierala  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23ec125fbbSEdward Tomasz Napierala  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24ec125fbbSEdward Tomasz Napierala  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25ec125fbbSEdward Tomasz Napierala  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26ec125fbbSEdward Tomasz Napierala  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27ec125fbbSEdward Tomasz Napierala  * SUCH DAMAGE.
28ec125fbbSEdward Tomasz Napierala  *
29ec125fbbSEdward Tomasz Napierala  * $FreeBSD$
30ec125fbbSEdward Tomasz Napierala  */
31ec125fbbSEdward Tomasz Napierala 
32ec125fbbSEdward Tomasz Napierala #include <sys/cdefs.h>
33ec125fbbSEdward Tomasz Napierala __FBSDID("$FreeBSD$");
34ec125fbbSEdward Tomasz Napierala 
35ec125fbbSEdward Tomasz Napierala #include <sys/param.h>
36ec125fbbSEdward Tomasz Napierala #include <sys/bus.h>
37ec125fbbSEdward Tomasz Napierala #include <sys/malloc.h>
38ec125fbbSEdward Tomasz Napierala #include <sys/queue.h>
39ec125fbbSEdward Tomasz Napierala #include <sys/refcount.h>
40ec125fbbSEdward Tomasz Napierala #include <sys/jail.h>
41ec125fbbSEdward Tomasz Napierala #include <sys/kernel.h>
42ec125fbbSEdward Tomasz Napierala #include <sys/limits.h>
43ec125fbbSEdward Tomasz Napierala #include <sys/loginclass.h>
44ec125fbbSEdward Tomasz Napierala #include <sys/priv.h>
45ec125fbbSEdward Tomasz Napierala #include <sys/proc.h>
46ec125fbbSEdward Tomasz Napierala #include <sys/racct.h>
47ec125fbbSEdward Tomasz Napierala #include <sys/rctl.h>
48ec125fbbSEdward Tomasz Napierala #include <sys/resourcevar.h>
49ec125fbbSEdward Tomasz Napierala #include <sys/sx.h>
50ec125fbbSEdward Tomasz Napierala #include <sys/sysent.h>
51ec125fbbSEdward Tomasz Napierala #include <sys/sysproto.h>
52ec125fbbSEdward Tomasz Napierala #include <sys/systm.h>
53ec125fbbSEdward Tomasz Napierala #include <sys/types.h>
54ec125fbbSEdward Tomasz Napierala #include <sys/eventhandler.h>
55ec125fbbSEdward Tomasz Napierala #include <sys/lock.h>
56ec125fbbSEdward Tomasz Napierala #include <sys/mutex.h>
57ec125fbbSEdward Tomasz Napierala #include <sys/rwlock.h>
58ec125fbbSEdward Tomasz Napierala #include <sys/sbuf.h>
59ec125fbbSEdward Tomasz Napierala #include <sys/taskqueue.h>
60ec125fbbSEdward Tomasz Napierala #include <sys/tree.h>
61ec125fbbSEdward Tomasz Napierala #include <vm/uma.h>
62ec125fbbSEdward Tomasz Napierala 
63ec125fbbSEdward Tomasz Napierala #ifdef RCTL
64ec125fbbSEdward Tomasz Napierala #ifndef RACCT
65ec125fbbSEdward Tomasz Napierala #error "The RCTL option requires the RACCT option"
66ec125fbbSEdward Tomasz Napierala #endif
67ec125fbbSEdward Tomasz Napierala 
68ec125fbbSEdward Tomasz Napierala FEATURE(rctl, "Resource Limits");
69ec125fbbSEdward Tomasz Napierala 
70ec125fbbSEdward Tomasz Napierala #define	HRF_DEFAULT		0
71ec125fbbSEdward Tomasz Napierala #define	HRF_DONT_INHERIT	1
72ec125fbbSEdward Tomasz Napierala #define	HRF_DONT_ACCUMULATE	2
73ec125fbbSEdward Tomasz Napierala 
74ec125fbbSEdward Tomasz Napierala /* Default buffer size for rctl_get_rules(2). */
75ec125fbbSEdward Tomasz Napierala #define	RCTL_DEFAULT_BUFSIZE	4096
76786813aaSEdward Tomasz Napierala #define	RCTL_MAX_INBUFLEN	4096
77ec125fbbSEdward Tomasz Napierala #define	RCTL_LOG_BUFSIZE	128
78ec125fbbSEdward Tomasz Napierala 
79*36af9869SEdward Tomasz Napierala #define	RCTL_PCPU_SHIFT		(10 * 1000000)
80*36af9869SEdward Tomasz Napierala 
81ec125fbbSEdward Tomasz Napierala /*
82ec125fbbSEdward Tomasz Napierala  * 'rctl_rule_link' connects a rule with every racct it's related to.
83ec125fbbSEdward Tomasz Napierala  * For example, rule 'user:X:openfiles:deny=N/process' is linked
84ec125fbbSEdward Tomasz Napierala  * with uidinfo for user X, and to each process of that user.
85ec125fbbSEdward Tomasz Napierala  */
86ec125fbbSEdward Tomasz Napierala struct rctl_rule_link {
87ec125fbbSEdward Tomasz Napierala 	LIST_ENTRY(rctl_rule_link)	rrl_next;
88ec125fbbSEdward Tomasz Napierala 	struct rctl_rule		*rrl_rule;
89ec125fbbSEdward Tomasz Napierala 	int				rrl_exceeded;
90ec125fbbSEdward Tomasz Napierala };
91ec125fbbSEdward Tomasz Napierala 
92ec125fbbSEdward Tomasz Napierala struct dict {
93ec125fbbSEdward Tomasz Napierala 	const char	*d_name;
94ec125fbbSEdward Tomasz Napierala 	int		d_value;
95ec125fbbSEdward Tomasz Napierala };
96ec125fbbSEdward Tomasz Napierala 
97ec125fbbSEdward Tomasz Napierala static struct dict subjectnames[] = {
98ec125fbbSEdward Tomasz Napierala 	{ "process", RCTL_SUBJECT_TYPE_PROCESS },
99ec125fbbSEdward Tomasz Napierala 	{ "user", RCTL_SUBJECT_TYPE_USER },
100ec125fbbSEdward Tomasz Napierala 	{ "loginclass", RCTL_SUBJECT_TYPE_LOGINCLASS },
101ec125fbbSEdward Tomasz Napierala 	{ "jail", RCTL_SUBJECT_TYPE_JAIL },
102ec125fbbSEdward Tomasz Napierala 	{ NULL, -1 }};
103ec125fbbSEdward Tomasz Napierala 
104ec125fbbSEdward Tomasz Napierala static struct dict resourcenames[] = {
10585a2f1b4SEdward Tomasz Napierala 	{ "cputime", RACCT_CPU },
10685a2f1b4SEdward Tomasz Napierala 	{ "datasize", RACCT_DATA },
10785a2f1b4SEdward Tomasz Napierala 	{ "stacksize", RACCT_STACK },
10885a2f1b4SEdward Tomasz Napierala 	{ "coredumpsize", RACCT_CORE },
10985a2f1b4SEdward Tomasz Napierala 	{ "memoryuse", RACCT_RSS },
11085a2f1b4SEdward Tomasz Napierala 	{ "memorylocked", RACCT_MEMLOCK },
11185a2f1b4SEdward Tomasz Napierala 	{ "maxproc", RACCT_NPROC },
11285a2f1b4SEdward Tomasz Napierala 	{ "openfiles", RACCT_NOFILE },
11385a2f1b4SEdward Tomasz Napierala 	{ "vmemoryuse", RACCT_VMEM },
11485a2f1b4SEdward Tomasz Napierala 	{ "pseudoterminals", RACCT_NPTS },
11585a2f1b4SEdward Tomasz Napierala 	{ "swapuse", RACCT_SWAP },
116ec125fbbSEdward Tomasz Napierala 	{ "nthr", RACCT_NTHR },
117ec125fbbSEdward Tomasz Napierala 	{ "msgqqueued", RACCT_MSGQQUEUED },
118ec125fbbSEdward Tomasz Napierala 	{ "msgqsize", RACCT_MSGQSIZE },
119ec125fbbSEdward Tomasz Napierala 	{ "nmsgq", RACCT_NMSGQ },
120ec125fbbSEdward Tomasz Napierala 	{ "nsem", RACCT_NSEM },
121ec125fbbSEdward Tomasz Napierala 	{ "nsemop", RACCT_NSEMOP },
122ec125fbbSEdward Tomasz Napierala 	{ "nshm", RACCT_NSHM },
123ec125fbbSEdward Tomasz Napierala 	{ "shmsize", RACCT_SHMSIZE },
124ec125fbbSEdward Tomasz Napierala 	{ "wallclock", RACCT_WALLCLOCK },
125*36af9869SEdward Tomasz Napierala 	{ "pcpu", RACCT_PCTCPU },
126ec125fbbSEdward Tomasz Napierala 	{ NULL, -1 }};
127ec125fbbSEdward Tomasz Napierala 
128ec125fbbSEdward Tomasz Napierala static struct dict actionnames[] = {
129ec125fbbSEdward Tomasz Napierala 	{ "sighup", RCTL_ACTION_SIGHUP },
130ec125fbbSEdward Tomasz Napierala 	{ "sigint", RCTL_ACTION_SIGINT },
131ec125fbbSEdward Tomasz Napierala 	{ "sigquit", RCTL_ACTION_SIGQUIT },
132ec125fbbSEdward Tomasz Napierala 	{ "sigill", RCTL_ACTION_SIGILL },
133ec125fbbSEdward Tomasz Napierala 	{ "sigtrap", RCTL_ACTION_SIGTRAP },
134ec125fbbSEdward Tomasz Napierala 	{ "sigabrt", RCTL_ACTION_SIGABRT },
135ec125fbbSEdward Tomasz Napierala 	{ "sigemt", RCTL_ACTION_SIGEMT },
136ec125fbbSEdward Tomasz Napierala 	{ "sigfpe", RCTL_ACTION_SIGFPE },
137ec125fbbSEdward Tomasz Napierala 	{ "sigkill", RCTL_ACTION_SIGKILL },
138ec125fbbSEdward Tomasz Napierala 	{ "sigbus", RCTL_ACTION_SIGBUS },
139ec125fbbSEdward Tomasz Napierala 	{ "sigsegv", RCTL_ACTION_SIGSEGV },
140ec125fbbSEdward Tomasz Napierala 	{ "sigsys", RCTL_ACTION_SIGSYS },
141ec125fbbSEdward Tomasz Napierala 	{ "sigpipe", RCTL_ACTION_SIGPIPE },
142ec125fbbSEdward Tomasz Napierala 	{ "sigalrm", RCTL_ACTION_SIGALRM },
143ec125fbbSEdward Tomasz Napierala 	{ "sigterm", RCTL_ACTION_SIGTERM },
144ec125fbbSEdward Tomasz Napierala 	{ "sigurg", RCTL_ACTION_SIGURG },
145ec125fbbSEdward Tomasz Napierala 	{ "sigstop", RCTL_ACTION_SIGSTOP },
146ec125fbbSEdward Tomasz Napierala 	{ "sigtstp", RCTL_ACTION_SIGTSTP },
147ec125fbbSEdward Tomasz Napierala 	{ "sigchld", RCTL_ACTION_SIGCHLD },
148ec125fbbSEdward Tomasz Napierala 	{ "sigttin", RCTL_ACTION_SIGTTIN },
149ec125fbbSEdward Tomasz Napierala 	{ "sigttou", RCTL_ACTION_SIGTTOU },
150ec125fbbSEdward Tomasz Napierala 	{ "sigio", RCTL_ACTION_SIGIO },
151ec125fbbSEdward Tomasz Napierala 	{ "sigxcpu", RCTL_ACTION_SIGXCPU },
152ec125fbbSEdward Tomasz Napierala 	{ "sigxfsz", RCTL_ACTION_SIGXFSZ },
153ec125fbbSEdward Tomasz Napierala 	{ "sigvtalrm", RCTL_ACTION_SIGVTALRM },
154ec125fbbSEdward Tomasz Napierala 	{ "sigprof", RCTL_ACTION_SIGPROF },
155ec125fbbSEdward Tomasz Napierala 	{ "sigwinch", RCTL_ACTION_SIGWINCH },
156ec125fbbSEdward Tomasz Napierala 	{ "siginfo", RCTL_ACTION_SIGINFO },
157ec125fbbSEdward Tomasz Napierala 	{ "sigusr1", RCTL_ACTION_SIGUSR1 },
158ec125fbbSEdward Tomasz Napierala 	{ "sigusr2", RCTL_ACTION_SIGUSR2 },
159ec125fbbSEdward Tomasz Napierala 	{ "sigthr", RCTL_ACTION_SIGTHR },
160ec125fbbSEdward Tomasz Napierala 	{ "deny", RCTL_ACTION_DENY },
161ec125fbbSEdward Tomasz Napierala 	{ "log", RCTL_ACTION_LOG },
162ec125fbbSEdward Tomasz Napierala 	{ "devctl", RCTL_ACTION_DEVCTL },
163ec125fbbSEdward Tomasz Napierala 	{ NULL, -1 }};
164ec125fbbSEdward Tomasz Napierala 
165ec125fbbSEdward Tomasz Napierala static void rctl_init(void);
166ec125fbbSEdward Tomasz Napierala SYSINIT(rctl, SI_SUB_RACCT, SI_ORDER_FIRST, rctl_init, NULL);
167ec125fbbSEdward Tomasz Napierala 
168ec125fbbSEdward Tomasz Napierala static uma_zone_t rctl_rule_link_zone;
169ec125fbbSEdward Tomasz Napierala static uma_zone_t rctl_rule_zone;
170ec125fbbSEdward Tomasz Napierala static struct rwlock rctl_lock;
171ec125fbbSEdward Tomasz Napierala RW_SYSINIT(rctl_lock, &rctl_lock, "RCTL lock");
172ec125fbbSEdward Tomasz Napierala 
173ec125fbbSEdward Tomasz Napierala static int rctl_rule_fully_specified(const struct rctl_rule *rule);
174ec125fbbSEdward Tomasz Napierala static void rctl_rule_to_sbuf(struct sbuf *sb, const struct rctl_rule *rule);
175ec125fbbSEdward Tomasz Napierala 
176d745c852SEd Schouten static MALLOC_DEFINE(M_RCTL, "rctl", "Resource Limits");
177ec125fbbSEdward Tomasz Napierala 
178ec125fbbSEdward Tomasz Napierala static const char *
179ec125fbbSEdward Tomasz Napierala rctl_subject_type_name(int subject)
180ec125fbbSEdward Tomasz Napierala {
181ec125fbbSEdward Tomasz Napierala 	int i;
182ec125fbbSEdward Tomasz Napierala 
183ec125fbbSEdward Tomasz Napierala 	for (i = 0; subjectnames[i].d_name != NULL; i++) {
184ec125fbbSEdward Tomasz Napierala 		if (subjectnames[i].d_value == subject)
185ec125fbbSEdward Tomasz Napierala 			return (subjectnames[i].d_name);
186ec125fbbSEdward Tomasz Napierala 	}
187ec125fbbSEdward Tomasz Napierala 
188ec125fbbSEdward Tomasz Napierala 	panic("rctl_subject_type_name: unknown subject type %d", subject);
189ec125fbbSEdward Tomasz Napierala }
190ec125fbbSEdward Tomasz Napierala 
191ec125fbbSEdward Tomasz Napierala static const char *
192ec125fbbSEdward Tomasz Napierala rctl_action_name(int action)
193ec125fbbSEdward Tomasz Napierala {
194ec125fbbSEdward Tomasz Napierala 	int i;
195ec125fbbSEdward Tomasz Napierala 
196ec125fbbSEdward Tomasz Napierala 	for (i = 0; actionnames[i].d_name != NULL; i++) {
197ec125fbbSEdward Tomasz Napierala 		if (actionnames[i].d_value == action)
198ec125fbbSEdward Tomasz Napierala 			return (actionnames[i].d_name);
199ec125fbbSEdward Tomasz Napierala 	}
200ec125fbbSEdward Tomasz Napierala 
201ec125fbbSEdward Tomasz Napierala 	panic("rctl_action_name: unknown action %d", action);
202ec125fbbSEdward Tomasz Napierala }
203ec125fbbSEdward Tomasz Napierala 
204ec125fbbSEdward Tomasz Napierala const char *
205ec125fbbSEdward Tomasz Napierala rctl_resource_name(int resource)
206ec125fbbSEdward Tomasz Napierala {
207ec125fbbSEdward Tomasz Napierala 	int i;
208ec125fbbSEdward Tomasz Napierala 
209ec125fbbSEdward Tomasz Napierala 	for (i = 0; resourcenames[i].d_name != NULL; i++) {
210ec125fbbSEdward Tomasz Napierala 		if (resourcenames[i].d_value == resource)
211ec125fbbSEdward Tomasz Napierala 			return (resourcenames[i].d_name);
212ec125fbbSEdward Tomasz Napierala 	}
213ec125fbbSEdward Tomasz Napierala 
214ec125fbbSEdward Tomasz Napierala 	panic("rctl_resource_name: unknown resource %d", resource);
215ec125fbbSEdward Tomasz Napierala }
216ec125fbbSEdward Tomasz Napierala 
217ec125fbbSEdward Tomasz Napierala /*
218ec125fbbSEdward Tomasz Napierala  * Return the amount of resource that can be allocated by 'p' before
219ec125fbbSEdward Tomasz Napierala  * hitting 'rule'.
220ec125fbbSEdward Tomasz Napierala  */
221ec125fbbSEdward Tomasz Napierala static int64_t
222ec125fbbSEdward Tomasz Napierala rctl_available_resource(const struct proc *p, const struct rctl_rule *rule)
223ec125fbbSEdward Tomasz Napierala {
224ec125fbbSEdward Tomasz Napierala 	int resource;
225ec125fbbSEdward Tomasz Napierala 	int64_t available = INT64_MAX;
226ec125fbbSEdward Tomasz Napierala 	struct ucred *cred = p->p_ucred;
227ec125fbbSEdward Tomasz Napierala 
228ec125fbbSEdward Tomasz Napierala 	rw_assert(&rctl_lock, RA_LOCKED);
229ec125fbbSEdward Tomasz Napierala 
230ec125fbbSEdward Tomasz Napierala 	resource = rule->rr_resource;
231ec125fbbSEdward Tomasz Napierala 	switch (rule->rr_per) {
232ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_PROCESS:
233ec125fbbSEdward Tomasz Napierala 		available = rule->rr_amount -
234ec125fbbSEdward Tomasz Napierala 		    p->p_racct->r_resources[resource];
235ec125fbbSEdward Tomasz Napierala 		break;
236ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_USER:
237ec125fbbSEdward Tomasz Napierala 		available = rule->rr_amount -
238ec125fbbSEdward Tomasz Napierala 		    cred->cr_ruidinfo->ui_racct->r_resources[resource];
239ec125fbbSEdward Tomasz Napierala 		break;
240ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_LOGINCLASS:
241ec125fbbSEdward Tomasz Napierala 		available = rule->rr_amount -
242ec125fbbSEdward Tomasz Napierala 		    cred->cr_loginclass->lc_racct->r_resources[resource];
243ec125fbbSEdward Tomasz Napierala 		break;
244ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_JAIL:
245ec125fbbSEdward Tomasz Napierala 		available = rule->rr_amount -
246a7ad07bfSEdward Tomasz Napierala 		    cred->cr_prison->pr_prison_racct->prr_racct->
247a7ad07bfSEdward Tomasz Napierala 		        r_resources[resource];
248ec125fbbSEdward Tomasz Napierala 		break;
249ec125fbbSEdward Tomasz Napierala 	default:
250ec125fbbSEdward Tomasz Napierala 		panic("rctl_compute_available: unknown per %d",
251ec125fbbSEdward Tomasz Napierala 		    rule->rr_per);
252ec125fbbSEdward Tomasz Napierala 	}
253ec125fbbSEdward Tomasz Napierala 
254ec125fbbSEdward Tomasz Napierala 	return (available);
255ec125fbbSEdward Tomasz Napierala }
256ec125fbbSEdward Tomasz Napierala 
257ec125fbbSEdward Tomasz Napierala /*
258ec125fbbSEdward Tomasz Napierala  * Return non-zero if allocating 'amount' by proc 'p' would exceed
259ec125fbbSEdward Tomasz Napierala  * resource limit specified by 'rule'.
260ec125fbbSEdward Tomasz Napierala  */
261ec125fbbSEdward Tomasz Napierala static int
262ec125fbbSEdward Tomasz Napierala rctl_would_exceed(const struct proc *p, const struct rctl_rule *rule,
263ec125fbbSEdward Tomasz Napierala     int64_t amount)
264ec125fbbSEdward Tomasz Napierala {
265ec125fbbSEdward Tomasz Napierala 	int64_t available;
266ec125fbbSEdward Tomasz Napierala 
267ec125fbbSEdward Tomasz Napierala 	rw_assert(&rctl_lock, RA_LOCKED);
268ec125fbbSEdward Tomasz Napierala 
269ec125fbbSEdward Tomasz Napierala 	available = rctl_available_resource(p, rule);
270ec125fbbSEdward Tomasz Napierala 	if (available >= amount)
271ec125fbbSEdward Tomasz Napierala 		return (0);
272ec125fbbSEdward Tomasz Napierala 
273ec125fbbSEdward Tomasz Napierala 	return (1);
274ec125fbbSEdward Tomasz Napierala }
275ec125fbbSEdward Tomasz Napierala 
276ec125fbbSEdward Tomasz Napierala /*
277*36af9869SEdward Tomasz Napierala  * Special version of rctl_available() function for the %cpu resource.
278*36af9869SEdward Tomasz Napierala  * We slightly cheat here and return less than we normally would.
279*36af9869SEdward Tomasz Napierala  */
280*36af9869SEdward Tomasz Napierala int64_t
281*36af9869SEdward Tomasz Napierala rctl_pcpu_available(const struct proc *p) {
282*36af9869SEdward Tomasz Napierala 	struct rctl_rule *rule;
283*36af9869SEdward Tomasz Napierala 	struct rctl_rule_link *link;
284*36af9869SEdward Tomasz Napierala 	int64_t available, minavailable, limit;
285*36af9869SEdward Tomasz Napierala 
286*36af9869SEdward Tomasz Napierala 	minavailable = INT64_MAX;
287*36af9869SEdward Tomasz Napierala 	limit = 0;
288*36af9869SEdward Tomasz Napierala 
289*36af9869SEdward Tomasz Napierala 	rw_rlock(&rctl_lock);
290*36af9869SEdward Tomasz Napierala 
291*36af9869SEdward Tomasz Napierala 	LIST_FOREACH(link, &p->p_racct->r_rule_links, rrl_next) {
292*36af9869SEdward Tomasz Napierala 		rule = link->rrl_rule;
293*36af9869SEdward Tomasz Napierala 		if (rule->rr_resource != RACCT_PCTCPU)
294*36af9869SEdward Tomasz Napierala 			continue;
295*36af9869SEdward Tomasz Napierala 		if (rule->rr_action != RCTL_ACTION_DENY)
296*36af9869SEdward Tomasz Napierala 			continue;
297*36af9869SEdward Tomasz Napierala 		available = rctl_available_resource(p, rule);
298*36af9869SEdward Tomasz Napierala 		if (available < minavailable) {
299*36af9869SEdward Tomasz Napierala 			minavailable = available;
300*36af9869SEdward Tomasz Napierala 			limit = rule->rr_amount;
301*36af9869SEdward Tomasz Napierala 		}
302*36af9869SEdward Tomasz Napierala 	}
303*36af9869SEdward Tomasz Napierala 
304*36af9869SEdward Tomasz Napierala 	rw_runlock(&rctl_lock);
305*36af9869SEdward Tomasz Napierala 
306*36af9869SEdward Tomasz Napierala 	/*
307*36af9869SEdward Tomasz Napierala 	 * Return slightly less than actual value of the available
308*36af9869SEdward Tomasz Napierala 	 * %cpu resource.  This makes %cpu throttling more agressive
309*36af9869SEdward Tomasz Napierala 	 * and lets us act sooner than the limits are already exceeded.
310*36af9869SEdward Tomasz Napierala 	 */
311*36af9869SEdward Tomasz Napierala 	if (limit != 0) {
312*36af9869SEdward Tomasz Napierala 		if (limit > 2 * RCTL_PCPU_SHIFT)
313*36af9869SEdward Tomasz Napierala 			minavailable -= RCTL_PCPU_SHIFT;
314*36af9869SEdward Tomasz Napierala 		else
315*36af9869SEdward Tomasz Napierala 			minavailable -= (limit / 2);
316*36af9869SEdward Tomasz Napierala 	}
317*36af9869SEdward Tomasz Napierala 
318*36af9869SEdward Tomasz Napierala 	return (minavailable);
319*36af9869SEdward Tomasz Napierala }
320*36af9869SEdward Tomasz Napierala 
321*36af9869SEdward Tomasz Napierala /*
322ec125fbbSEdward Tomasz Napierala  * Check whether the proc 'p' can allocate 'amount' of 'resource' in addition
323ec125fbbSEdward Tomasz Napierala  * to what it keeps allocated now.  Returns non-zero if the allocation should
324ec125fbbSEdward Tomasz Napierala  * be denied, 0 otherwise.
325ec125fbbSEdward Tomasz Napierala  */
326ec125fbbSEdward Tomasz Napierala int
327ec125fbbSEdward Tomasz Napierala rctl_enforce(struct proc *p, int resource, uint64_t amount)
328ec125fbbSEdward Tomasz Napierala {
329ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
330ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
331ec125fbbSEdward Tomasz Napierala 	struct sbuf sb;
332ec125fbbSEdward Tomasz Napierala 	int should_deny = 0;
333ec125fbbSEdward Tomasz Napierala 	char *buf;
334ec125fbbSEdward Tomasz Napierala 	static int curtime = 0;
335ec125fbbSEdward Tomasz Napierala 	static struct timeval lasttime;
336ec125fbbSEdward Tomasz Napierala 
337ec125fbbSEdward Tomasz Napierala 	rw_rlock(&rctl_lock);
338ec125fbbSEdward Tomasz Napierala 
339ec125fbbSEdward Tomasz Napierala 	/*
340ec125fbbSEdward Tomasz Napierala 	 * There may be more than one matching rule; go through all of them.
341ec125fbbSEdward Tomasz Napierala 	 * Denial should be done last, after logging and sending signals.
342ec125fbbSEdward Tomasz Napierala 	 */
343ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &p->p_racct->r_rule_links, rrl_next) {
344ec125fbbSEdward Tomasz Napierala 		rule = link->rrl_rule;
345ec125fbbSEdward Tomasz Napierala 		if (rule->rr_resource != resource)
346ec125fbbSEdward Tomasz Napierala 			continue;
347ec125fbbSEdward Tomasz Napierala 		if (!rctl_would_exceed(p, rule, amount)) {
348ec125fbbSEdward Tomasz Napierala 			link->rrl_exceeded = 0;
349ec125fbbSEdward Tomasz Napierala 			continue;
350ec125fbbSEdward Tomasz Napierala 		}
351ec125fbbSEdward Tomasz Napierala 
352ec125fbbSEdward Tomasz Napierala 		switch (rule->rr_action) {
353ec125fbbSEdward Tomasz Napierala 		case RCTL_ACTION_DENY:
354ec125fbbSEdward Tomasz Napierala 			should_deny = 1;
355ec125fbbSEdward Tomasz Napierala 			continue;
356ec125fbbSEdward Tomasz Napierala 		case RCTL_ACTION_LOG:
357ec125fbbSEdward Tomasz Napierala 			/*
358ec125fbbSEdward Tomasz Napierala 			 * If rrl_exceeded != 0, it means we've already
359ec125fbbSEdward Tomasz Napierala 			 * logged a warning for this process.
360ec125fbbSEdward Tomasz Napierala 			 */
361ec125fbbSEdward Tomasz Napierala 			if (link->rrl_exceeded != 0)
362ec125fbbSEdward Tomasz Napierala 				continue;
363ec125fbbSEdward Tomasz Napierala 
36472a401d9SEdward Tomasz Napierala 			/*
36572a401d9SEdward Tomasz Napierala 			 * If the process state is not fully initialized yet,
36672a401d9SEdward Tomasz Napierala 			 * we can't access most of the required fields, e.g.
36772a401d9SEdward Tomasz Napierala 			 * p->p_comm.  This happens when called from fork1().
36872a401d9SEdward Tomasz Napierala 			 * Ignore this rule for now; it will be processed just
36972a401d9SEdward Tomasz Napierala 			 * after fork, when called from racct_proc_fork_done().
37072a401d9SEdward Tomasz Napierala 			 */
37172a401d9SEdward Tomasz Napierala 			if (p->p_state != PRS_NORMAL)
37272a401d9SEdward Tomasz Napierala 				continue;
37372a401d9SEdward Tomasz Napierala 
374ec125fbbSEdward Tomasz Napierala 			if (!ppsratecheck(&lasttime, &curtime, 10))
375ec125fbbSEdward Tomasz Napierala 				continue;
376ec125fbbSEdward Tomasz Napierala 
377ec125fbbSEdward Tomasz Napierala 			buf = malloc(RCTL_LOG_BUFSIZE, M_RCTL, M_NOWAIT);
378ec125fbbSEdward Tomasz Napierala 			if (buf == NULL) {
379ec125fbbSEdward Tomasz Napierala 				printf("rctl_enforce: out of memory\n");
380ec125fbbSEdward Tomasz Napierala 				continue;
381ec125fbbSEdward Tomasz Napierala 			}
382ec125fbbSEdward Tomasz Napierala 			sbuf_new(&sb, buf, RCTL_LOG_BUFSIZE, SBUF_FIXEDLEN);
383ec125fbbSEdward Tomasz Napierala 			rctl_rule_to_sbuf(&sb, rule);
384ec125fbbSEdward Tomasz Napierala 			sbuf_finish(&sb);
385ec125fbbSEdward Tomasz Napierala 			printf("rctl: rule \"%s\" matched by pid %d "
386ec125fbbSEdward Tomasz Napierala 			    "(%s), uid %d, jail %s\n", sbuf_data(&sb),
387ec125fbbSEdward Tomasz Napierala 			    p->p_pid, p->p_comm, p->p_ucred->cr_uid,
388a7ad07bfSEdward Tomasz Napierala 			    p->p_ucred->cr_prison->pr_prison_racct->prr_name);
389ec125fbbSEdward Tomasz Napierala 			sbuf_delete(&sb);
390ec125fbbSEdward Tomasz Napierala 			free(buf, M_RCTL);
391ec125fbbSEdward Tomasz Napierala 			link->rrl_exceeded = 1;
392ec125fbbSEdward Tomasz Napierala 			continue;
393ec125fbbSEdward Tomasz Napierala 		case RCTL_ACTION_DEVCTL:
394ec125fbbSEdward Tomasz Napierala 			if (link->rrl_exceeded != 0)
395ec125fbbSEdward Tomasz Napierala 				continue;
396ec125fbbSEdward Tomasz Napierala 
39772a401d9SEdward Tomasz Napierala 			if (p->p_state != PRS_NORMAL)
39872a401d9SEdward Tomasz Napierala 				continue;
39972a401d9SEdward Tomasz Napierala 
400ec125fbbSEdward Tomasz Napierala 			buf = malloc(RCTL_LOG_BUFSIZE, M_RCTL, M_NOWAIT);
401ec125fbbSEdward Tomasz Napierala 			if (buf == NULL) {
402ec125fbbSEdward Tomasz Napierala 				printf("rctl_enforce: out of memory\n");
403ec125fbbSEdward Tomasz Napierala 				continue;
404ec125fbbSEdward Tomasz Napierala 			}
405ec125fbbSEdward Tomasz Napierala 			sbuf_new(&sb, buf, RCTL_LOG_BUFSIZE, SBUF_FIXEDLEN);
406ec125fbbSEdward Tomasz Napierala 			sbuf_printf(&sb, "rule=");
407ec125fbbSEdward Tomasz Napierala 			rctl_rule_to_sbuf(&sb, rule);
408ec125fbbSEdward Tomasz Napierala 			sbuf_printf(&sb, " pid=%d ruid=%d jail=%s",
409ec125fbbSEdward Tomasz Napierala 			    p->p_pid, p->p_ucred->cr_ruid,
410a7ad07bfSEdward Tomasz Napierala 			    p->p_ucred->cr_prison->pr_prison_racct->prr_name);
411ec125fbbSEdward Tomasz Napierala 			sbuf_finish(&sb);
412ec125fbbSEdward Tomasz Napierala 			devctl_notify_f("RCTL", "rule", "matched",
413ec125fbbSEdward Tomasz Napierala 			    sbuf_data(&sb), M_NOWAIT);
414ec125fbbSEdward Tomasz Napierala 			sbuf_delete(&sb);
415ec125fbbSEdward Tomasz Napierala 			free(buf, M_RCTL);
416ec125fbbSEdward Tomasz Napierala 			link->rrl_exceeded = 1;
417ec125fbbSEdward Tomasz Napierala 			continue;
418ec125fbbSEdward Tomasz Napierala 		default:
419ec125fbbSEdward Tomasz Napierala 			if (link->rrl_exceeded != 0)
420ec125fbbSEdward Tomasz Napierala 				continue;
421ec125fbbSEdward Tomasz Napierala 
42272a401d9SEdward Tomasz Napierala 			if (p->p_state != PRS_NORMAL)
42372a401d9SEdward Tomasz Napierala 				continue;
42472a401d9SEdward Tomasz Napierala 
425ec125fbbSEdward Tomasz Napierala 			KASSERT(rule->rr_action > 0 &&
426ec125fbbSEdward Tomasz Napierala 			    rule->rr_action <= RCTL_ACTION_SIGNAL_MAX,
427ec125fbbSEdward Tomasz Napierala 			    ("rctl_enforce: unknown action %d",
428ec125fbbSEdward Tomasz Napierala 			     rule->rr_action));
429ec125fbbSEdward Tomasz Napierala 
430ec125fbbSEdward Tomasz Napierala 			/*
431ec125fbbSEdward Tomasz Napierala 			 * We're using the fact that RCTL_ACTION_SIG* values
432ec125fbbSEdward Tomasz Napierala 			 * are equal to their counterparts from sys/signal.h.
433ec125fbbSEdward Tomasz Napierala 			 */
4348451d0ddSKip Macy 			kern_psignal(p, rule->rr_action);
435ec125fbbSEdward Tomasz Napierala 			link->rrl_exceeded = 1;
436ec125fbbSEdward Tomasz Napierala 			continue;
437ec125fbbSEdward Tomasz Napierala 		}
438ec125fbbSEdward Tomasz Napierala 	}
439ec125fbbSEdward Tomasz Napierala 
440ec125fbbSEdward Tomasz Napierala 	rw_runlock(&rctl_lock);
441ec125fbbSEdward Tomasz Napierala 
442ec125fbbSEdward Tomasz Napierala 	if (should_deny) {
443ec125fbbSEdward Tomasz Napierala 		/*
444ec125fbbSEdward Tomasz Napierala 		 * Return fake error code; the caller should change it
445ec125fbbSEdward Tomasz Napierala 		 * into one proper for the situation - EFSIZ, ENOMEM etc.
446ec125fbbSEdward Tomasz Napierala 		 */
447ec125fbbSEdward Tomasz Napierala 		return (EDOOFUS);
448ec125fbbSEdward Tomasz Napierala 	}
449ec125fbbSEdward Tomasz Napierala 
450ec125fbbSEdward Tomasz Napierala 	return (0);
451ec125fbbSEdward Tomasz Napierala }
452ec125fbbSEdward Tomasz Napierala 
453ec125fbbSEdward Tomasz Napierala uint64_t
454ec125fbbSEdward Tomasz Napierala rctl_get_limit(struct proc *p, int resource)
455ec125fbbSEdward Tomasz Napierala {
456ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
457ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
458ec125fbbSEdward Tomasz Napierala 	uint64_t amount = UINT64_MAX;
459ec125fbbSEdward Tomasz Napierala 
460ec125fbbSEdward Tomasz Napierala 	rw_rlock(&rctl_lock);
461ec125fbbSEdward Tomasz Napierala 
462ec125fbbSEdward Tomasz Napierala 	/*
463ec125fbbSEdward Tomasz Napierala 	 * There may be more than one matching rule; go through all of them.
464ec125fbbSEdward Tomasz Napierala 	 * Denial should be done last, after logging and sending signals.
465ec125fbbSEdward Tomasz Napierala 	 */
466ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &p->p_racct->r_rule_links, rrl_next) {
467ec125fbbSEdward Tomasz Napierala 		rule = link->rrl_rule;
468ec125fbbSEdward Tomasz Napierala 		if (rule->rr_resource != resource)
469ec125fbbSEdward Tomasz Napierala 			continue;
470ec125fbbSEdward Tomasz Napierala 		if (rule->rr_action != RCTL_ACTION_DENY)
471ec125fbbSEdward Tomasz Napierala 			continue;
472ec125fbbSEdward Tomasz Napierala 		if (rule->rr_amount < amount)
473ec125fbbSEdward Tomasz Napierala 			amount = rule->rr_amount;
474ec125fbbSEdward Tomasz Napierala 	}
475ec125fbbSEdward Tomasz Napierala 
476ec125fbbSEdward Tomasz Napierala 	rw_runlock(&rctl_lock);
477ec125fbbSEdward Tomasz Napierala 
478ec125fbbSEdward Tomasz Napierala 	return (amount);
479ec125fbbSEdward Tomasz Napierala }
480ec125fbbSEdward Tomasz Napierala 
481ec125fbbSEdward Tomasz Napierala uint64_t
482ec125fbbSEdward Tomasz Napierala rctl_get_available(struct proc *p, int resource)
483ec125fbbSEdward Tomasz Napierala {
484ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
485ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
486ec125fbbSEdward Tomasz Napierala 	int64_t available, minavailable, allocated;
487ec125fbbSEdward Tomasz Napierala 
488ec125fbbSEdward Tomasz Napierala 	minavailable = INT64_MAX;
489ec125fbbSEdward Tomasz Napierala 
490ec125fbbSEdward Tomasz Napierala 	rw_rlock(&rctl_lock);
491ec125fbbSEdward Tomasz Napierala 
492ec125fbbSEdward Tomasz Napierala 	/*
493ec125fbbSEdward Tomasz Napierala 	 * There may be more than one matching rule; go through all of them.
494ec125fbbSEdward Tomasz Napierala 	 * Denial should be done last, after logging and sending signals.
495ec125fbbSEdward Tomasz Napierala 	 */
496ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &p->p_racct->r_rule_links, rrl_next) {
497ec125fbbSEdward Tomasz Napierala 		rule = link->rrl_rule;
498ec125fbbSEdward Tomasz Napierala 		if (rule->rr_resource != resource)
499ec125fbbSEdward Tomasz Napierala 			continue;
500ec125fbbSEdward Tomasz Napierala 		if (rule->rr_action != RCTL_ACTION_DENY)
501ec125fbbSEdward Tomasz Napierala 			continue;
502ec125fbbSEdward Tomasz Napierala 		available = rctl_available_resource(p, rule);
503ec125fbbSEdward Tomasz Napierala 		if (available < minavailable)
504ec125fbbSEdward Tomasz Napierala 			minavailable = available;
505ec125fbbSEdward Tomasz Napierala 	}
506ec125fbbSEdward Tomasz Napierala 
507ec125fbbSEdward Tomasz Napierala 	rw_runlock(&rctl_lock);
508ec125fbbSEdward Tomasz Napierala 
509ec125fbbSEdward Tomasz Napierala 	/*
510ec125fbbSEdward Tomasz Napierala 	 * XXX: Think about this _hard_.
511ec125fbbSEdward Tomasz Napierala 	 */
512ec125fbbSEdward Tomasz Napierala 	allocated = p->p_racct->r_resources[resource];
513ec125fbbSEdward Tomasz Napierala 	if (minavailable < INT64_MAX - allocated)
514ec125fbbSEdward Tomasz Napierala 		minavailable += allocated;
515ec125fbbSEdward Tomasz Napierala 	if (minavailable < 0)
516ec125fbbSEdward Tomasz Napierala 		minavailable = 0;
517ec125fbbSEdward Tomasz Napierala 	return (minavailable);
518ec125fbbSEdward Tomasz Napierala }
519ec125fbbSEdward Tomasz Napierala 
520ec125fbbSEdward Tomasz Napierala static int
521ec125fbbSEdward Tomasz Napierala rctl_rule_matches(const struct rctl_rule *rule, const struct rctl_rule *filter)
522ec125fbbSEdward Tomasz Napierala {
523ec125fbbSEdward Tomasz Napierala 
524ec125fbbSEdward Tomasz Napierala 	if (filter->rr_subject_type != RCTL_SUBJECT_TYPE_UNDEFINED) {
525ec125fbbSEdward Tomasz Napierala 		if (rule->rr_subject_type != filter->rr_subject_type)
526ec125fbbSEdward Tomasz Napierala 			return (0);
527ec125fbbSEdward Tomasz Napierala 
528ec125fbbSEdward Tomasz Napierala 		switch (filter->rr_subject_type) {
529ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_PROCESS:
530ec125fbbSEdward Tomasz Napierala 			if (filter->rr_subject.rs_proc != NULL &&
531ec125fbbSEdward Tomasz Napierala 			    rule->rr_subject.rs_proc !=
532ec125fbbSEdward Tomasz Napierala 			    filter->rr_subject.rs_proc)
533ec125fbbSEdward Tomasz Napierala 				return (0);
534ec125fbbSEdward Tomasz Napierala 			break;
535ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_USER:
536ec125fbbSEdward Tomasz Napierala 			if (filter->rr_subject.rs_uip != NULL &&
537ec125fbbSEdward Tomasz Napierala 			    rule->rr_subject.rs_uip !=
538ec125fbbSEdward Tomasz Napierala 			    filter->rr_subject.rs_uip)
539ec125fbbSEdward Tomasz Napierala 				return (0);
540ec125fbbSEdward Tomasz Napierala 			break;
541ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_LOGINCLASS:
542415896e3SEdward Tomasz Napierala 			if (filter->rr_subject.rs_loginclass != NULL &&
543415896e3SEdward Tomasz Napierala 			    rule->rr_subject.rs_loginclass !=
544415896e3SEdward Tomasz Napierala 			    filter->rr_subject.rs_loginclass)
545ec125fbbSEdward Tomasz Napierala 				return (0);
546ec125fbbSEdward Tomasz Napierala 			break;
547ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_JAIL:
548a7ad07bfSEdward Tomasz Napierala 			if (filter->rr_subject.rs_prison_racct != NULL &&
549a7ad07bfSEdward Tomasz Napierala 			    rule->rr_subject.rs_prison_racct !=
550a7ad07bfSEdward Tomasz Napierala 			    filter->rr_subject.rs_prison_racct)
551ec125fbbSEdward Tomasz Napierala 				return (0);
552ec125fbbSEdward Tomasz Napierala 			break;
553ec125fbbSEdward Tomasz Napierala 		default:
554ec125fbbSEdward Tomasz Napierala 			panic("rctl_rule_matches: unknown subject type %d",
555ec125fbbSEdward Tomasz Napierala 			    filter->rr_subject_type);
556ec125fbbSEdward Tomasz Napierala 		}
557ec125fbbSEdward Tomasz Napierala 	}
558ec125fbbSEdward Tomasz Napierala 
559ec125fbbSEdward Tomasz Napierala 	if (filter->rr_resource != RACCT_UNDEFINED) {
560ec125fbbSEdward Tomasz Napierala 		if (rule->rr_resource != filter->rr_resource)
561ec125fbbSEdward Tomasz Napierala 			return (0);
562ec125fbbSEdward Tomasz Napierala 	}
563ec125fbbSEdward Tomasz Napierala 
564ec125fbbSEdward Tomasz Napierala 	if (filter->rr_action != RCTL_ACTION_UNDEFINED) {
565ec125fbbSEdward Tomasz Napierala 		if (rule->rr_action != filter->rr_action)
566ec125fbbSEdward Tomasz Napierala 			return (0);
567ec125fbbSEdward Tomasz Napierala 	}
568ec125fbbSEdward Tomasz Napierala 
569ec125fbbSEdward Tomasz Napierala 	if (filter->rr_amount != RCTL_AMOUNT_UNDEFINED) {
570ec125fbbSEdward Tomasz Napierala 		if (rule->rr_amount != filter->rr_amount)
571ec125fbbSEdward Tomasz Napierala 			return (0);
572ec125fbbSEdward Tomasz Napierala 	}
573ec125fbbSEdward Tomasz Napierala 
574ec125fbbSEdward Tomasz Napierala 	if (filter->rr_per != RCTL_SUBJECT_TYPE_UNDEFINED) {
575ec125fbbSEdward Tomasz Napierala 		if (rule->rr_per != filter->rr_per)
576ec125fbbSEdward Tomasz Napierala 			return (0);
577ec125fbbSEdward Tomasz Napierala 	}
578ec125fbbSEdward Tomasz Napierala 
579ec125fbbSEdward Tomasz Napierala 	return (1);
580ec125fbbSEdward Tomasz Napierala }
581ec125fbbSEdward Tomasz Napierala 
582ec125fbbSEdward Tomasz Napierala static int
583ec125fbbSEdward Tomasz Napierala str2value(const char *str, int *value, struct dict *table)
584ec125fbbSEdward Tomasz Napierala {
585ec125fbbSEdward Tomasz Napierala 	int i;
586ec125fbbSEdward Tomasz Napierala 
587ec125fbbSEdward Tomasz Napierala 	if (value == NULL)
588ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
589ec125fbbSEdward Tomasz Napierala 
590ec125fbbSEdward Tomasz Napierala 	for (i = 0; table[i].d_name != NULL; i++) {
591ec125fbbSEdward Tomasz Napierala 		if (strcasecmp(table[i].d_name, str) == 0) {
592ec125fbbSEdward Tomasz Napierala 			*value =  table[i].d_value;
593ec125fbbSEdward Tomasz Napierala 			return (0);
594ec125fbbSEdward Tomasz Napierala 		}
595ec125fbbSEdward Tomasz Napierala 	}
596ec125fbbSEdward Tomasz Napierala 
597ec125fbbSEdward Tomasz Napierala 	return (EINVAL);
598ec125fbbSEdward Tomasz Napierala }
599ec125fbbSEdward Tomasz Napierala 
600ec125fbbSEdward Tomasz Napierala static int
601ec125fbbSEdward Tomasz Napierala str2id(const char *str, id_t *value)
602ec125fbbSEdward Tomasz Napierala {
603ec125fbbSEdward Tomasz Napierala 	char *end;
604ec125fbbSEdward Tomasz Napierala 
605ec125fbbSEdward Tomasz Napierala 	if (str == NULL)
606ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
607ec125fbbSEdward Tomasz Napierala 
608ec125fbbSEdward Tomasz Napierala 	*value = strtoul(str, &end, 10);
609ec125fbbSEdward Tomasz Napierala 	if ((size_t)(end - str) != strlen(str))
610ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
611ec125fbbSEdward Tomasz Napierala 
612ec125fbbSEdward Tomasz Napierala 	return (0);
613ec125fbbSEdward Tomasz Napierala }
614ec125fbbSEdward Tomasz Napierala 
615ec125fbbSEdward Tomasz Napierala static int
616ec125fbbSEdward Tomasz Napierala str2int64(const char *str, int64_t *value)
617ec125fbbSEdward Tomasz Napierala {
618ec125fbbSEdward Tomasz Napierala 	char *end;
619ec125fbbSEdward Tomasz Napierala 
620ec125fbbSEdward Tomasz Napierala 	if (str == NULL)
621ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
622ec125fbbSEdward Tomasz Napierala 
623ec125fbbSEdward Tomasz Napierala 	*value = strtoul(str, &end, 10);
624ec125fbbSEdward Tomasz Napierala 	if ((size_t)(end - str) != strlen(str))
625ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
626ec125fbbSEdward Tomasz Napierala 
627ec125fbbSEdward Tomasz Napierala 	return (0);
628ec125fbbSEdward Tomasz Napierala }
629ec125fbbSEdward Tomasz Napierala 
630ec125fbbSEdward Tomasz Napierala /*
631ec125fbbSEdward Tomasz Napierala  * Connect the rule to the racct, increasing refcount for the rule.
632ec125fbbSEdward Tomasz Napierala  */
633ec125fbbSEdward Tomasz Napierala static void
634ec125fbbSEdward Tomasz Napierala rctl_racct_add_rule(struct racct *racct, struct rctl_rule *rule)
635ec125fbbSEdward Tomasz Napierala {
636ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
637ec125fbbSEdward Tomasz Napierala 
638ec125fbbSEdward Tomasz Napierala 	KASSERT(rctl_rule_fully_specified(rule), ("rule not fully specified"));
639ec125fbbSEdward Tomasz Napierala 
640ec125fbbSEdward Tomasz Napierala 	rctl_rule_acquire(rule);
641ec125fbbSEdward Tomasz Napierala 	link = uma_zalloc(rctl_rule_link_zone, M_WAITOK);
642ec125fbbSEdward Tomasz Napierala 	link->rrl_rule = rule;
643ec125fbbSEdward Tomasz Napierala 	link->rrl_exceeded = 0;
644ec125fbbSEdward Tomasz Napierala 
645ec125fbbSEdward Tomasz Napierala 	rw_wlock(&rctl_lock);
646ec125fbbSEdward Tomasz Napierala 	LIST_INSERT_HEAD(&racct->r_rule_links, link, rrl_next);
647ec125fbbSEdward Tomasz Napierala 	rw_wunlock(&rctl_lock);
648ec125fbbSEdward Tomasz Napierala }
649ec125fbbSEdward Tomasz Napierala 
650ec125fbbSEdward Tomasz Napierala static int
651ec125fbbSEdward Tomasz Napierala rctl_racct_add_rule_locked(struct racct *racct, struct rctl_rule *rule)
652ec125fbbSEdward Tomasz Napierala {
653ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
654ec125fbbSEdward Tomasz Napierala 
655ec125fbbSEdward Tomasz Napierala 	KASSERT(rctl_rule_fully_specified(rule), ("rule not fully specified"));
656ec125fbbSEdward Tomasz Napierala 	rw_assert(&rctl_lock, RA_WLOCKED);
657ec125fbbSEdward Tomasz Napierala 
658ec125fbbSEdward Tomasz Napierala 	link = uma_zalloc(rctl_rule_link_zone, M_NOWAIT);
659ec125fbbSEdward Tomasz Napierala 	if (link == NULL)
660ec125fbbSEdward Tomasz Napierala 		return (ENOMEM);
661ec125fbbSEdward Tomasz Napierala 	rctl_rule_acquire(rule);
662ec125fbbSEdward Tomasz Napierala 	link->rrl_rule = rule;
663ec125fbbSEdward Tomasz Napierala 	link->rrl_exceeded = 0;
664ec125fbbSEdward Tomasz Napierala 
665ec125fbbSEdward Tomasz Napierala 	LIST_INSERT_HEAD(&racct->r_rule_links, link, rrl_next);
666ec125fbbSEdward Tomasz Napierala 	return (0);
667ec125fbbSEdward Tomasz Napierala }
668ec125fbbSEdward Tomasz Napierala 
669ec125fbbSEdward Tomasz Napierala /*
670ec125fbbSEdward Tomasz Napierala  * Remove limits for a rules matching the filter and release
671ec125fbbSEdward Tomasz Napierala  * the refcounts for the rules, possibly freeing them.  Returns
672ec125fbbSEdward Tomasz Napierala  * the number of limit structures removed.
673ec125fbbSEdward Tomasz Napierala  */
674ec125fbbSEdward Tomasz Napierala static int
675ec125fbbSEdward Tomasz Napierala rctl_racct_remove_rules(struct racct *racct,
676ec125fbbSEdward Tomasz Napierala     const struct rctl_rule *filter)
677ec125fbbSEdward Tomasz Napierala {
678ec125fbbSEdward Tomasz Napierala 	int removed = 0;
679ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link, *linktmp;
680ec125fbbSEdward Tomasz Napierala 
681ec125fbbSEdward Tomasz Napierala 	rw_assert(&rctl_lock, RA_WLOCKED);
682ec125fbbSEdward Tomasz Napierala 
683ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH_SAFE(link, &racct->r_rule_links, rrl_next, linktmp) {
684ec125fbbSEdward Tomasz Napierala 		if (!rctl_rule_matches(link->rrl_rule, filter))
685ec125fbbSEdward Tomasz Napierala 			continue;
686ec125fbbSEdward Tomasz Napierala 
687ec125fbbSEdward Tomasz Napierala 		LIST_REMOVE(link, rrl_next);
688ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(link->rrl_rule);
689ec125fbbSEdward Tomasz Napierala 		uma_zfree(rctl_rule_link_zone, link);
690ec125fbbSEdward Tomasz Napierala 		removed++;
691ec125fbbSEdward Tomasz Napierala 	}
692ec125fbbSEdward Tomasz Napierala 	return (removed);
693ec125fbbSEdward Tomasz Napierala }
694ec125fbbSEdward Tomasz Napierala 
695ec125fbbSEdward Tomasz Napierala static void
696ec125fbbSEdward Tomasz Napierala rctl_rule_acquire_subject(struct rctl_rule *rule)
697ec125fbbSEdward Tomasz Napierala {
698ec125fbbSEdward Tomasz Napierala 
699ec125fbbSEdward Tomasz Napierala 	switch (rule->rr_subject_type) {
700ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_UNDEFINED:
701ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_PROCESS:
702a7ad07bfSEdward Tomasz Napierala 		break;
703ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_JAIL:
704a7ad07bfSEdward Tomasz Napierala 		if (rule->rr_subject.rs_prison_racct != NULL)
705a7ad07bfSEdward Tomasz Napierala 			prison_racct_hold(rule->rr_subject.rs_prison_racct);
706ec125fbbSEdward Tomasz Napierala 		break;
707ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_USER:
708ec125fbbSEdward Tomasz Napierala 		if (rule->rr_subject.rs_uip != NULL)
709ec125fbbSEdward Tomasz Napierala 			uihold(rule->rr_subject.rs_uip);
710ec125fbbSEdward Tomasz Napierala 		break;
711ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_LOGINCLASS:
712415896e3SEdward Tomasz Napierala 		if (rule->rr_subject.rs_loginclass != NULL)
713415896e3SEdward Tomasz Napierala 			loginclass_hold(rule->rr_subject.rs_loginclass);
714ec125fbbSEdward Tomasz Napierala 		break;
715ec125fbbSEdward Tomasz Napierala 	default:
716ec125fbbSEdward Tomasz Napierala 		panic("rctl_rule_acquire_subject: unknown subject type %d",
717ec125fbbSEdward Tomasz Napierala 		    rule->rr_subject_type);
718ec125fbbSEdward Tomasz Napierala 	}
719ec125fbbSEdward Tomasz Napierala }
720ec125fbbSEdward Tomasz Napierala 
721ec125fbbSEdward Tomasz Napierala static void
722ec125fbbSEdward Tomasz Napierala rctl_rule_release_subject(struct rctl_rule *rule)
723ec125fbbSEdward Tomasz Napierala {
724ec125fbbSEdward Tomasz Napierala 
725ec125fbbSEdward Tomasz Napierala 	switch (rule->rr_subject_type) {
726ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_UNDEFINED:
727ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_PROCESS:
728a7ad07bfSEdward Tomasz Napierala 		break;
729ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_JAIL:
730a7ad07bfSEdward Tomasz Napierala 		if (rule->rr_subject.rs_prison_racct != NULL)
731a7ad07bfSEdward Tomasz Napierala 			prison_racct_free(rule->rr_subject.rs_prison_racct);
732ec125fbbSEdward Tomasz Napierala 		break;
733ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_USER:
734ec125fbbSEdward Tomasz Napierala 		if (rule->rr_subject.rs_uip != NULL)
735ec125fbbSEdward Tomasz Napierala 			uifree(rule->rr_subject.rs_uip);
736ec125fbbSEdward Tomasz Napierala 		break;
737ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_LOGINCLASS:
738415896e3SEdward Tomasz Napierala 		if (rule->rr_subject.rs_loginclass != NULL)
739415896e3SEdward Tomasz Napierala 			loginclass_free(rule->rr_subject.rs_loginclass);
740ec125fbbSEdward Tomasz Napierala 		break;
741ec125fbbSEdward Tomasz Napierala 	default:
742ec125fbbSEdward Tomasz Napierala 		panic("rctl_rule_release_subject: unknown subject type %d",
743ec125fbbSEdward Tomasz Napierala 		    rule->rr_subject_type);
744ec125fbbSEdward Tomasz Napierala 	}
745ec125fbbSEdward Tomasz Napierala }
746ec125fbbSEdward Tomasz Napierala 
747ec125fbbSEdward Tomasz Napierala struct rctl_rule *
748ec125fbbSEdward Tomasz Napierala rctl_rule_alloc(int flags)
749ec125fbbSEdward Tomasz Napierala {
750ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
751ec125fbbSEdward Tomasz Napierala 
752ec125fbbSEdward Tomasz Napierala 	rule = uma_zalloc(rctl_rule_zone, flags);
753ec125fbbSEdward Tomasz Napierala 	if (rule == NULL)
754ec125fbbSEdward Tomasz Napierala 		return (NULL);
755ec125fbbSEdward Tomasz Napierala 	rule->rr_subject_type = RCTL_SUBJECT_TYPE_UNDEFINED;
756ec125fbbSEdward Tomasz Napierala 	rule->rr_subject.rs_proc = NULL;
757ec125fbbSEdward Tomasz Napierala 	rule->rr_subject.rs_uip = NULL;
758415896e3SEdward Tomasz Napierala 	rule->rr_subject.rs_loginclass = NULL;
759a7ad07bfSEdward Tomasz Napierala 	rule->rr_subject.rs_prison_racct = NULL;
760ec125fbbSEdward Tomasz Napierala 	rule->rr_per = RCTL_SUBJECT_TYPE_UNDEFINED;
761ec125fbbSEdward Tomasz Napierala 	rule->rr_resource = RACCT_UNDEFINED;
762ec125fbbSEdward Tomasz Napierala 	rule->rr_action = RCTL_ACTION_UNDEFINED;
763ec125fbbSEdward Tomasz Napierala 	rule->rr_amount = RCTL_AMOUNT_UNDEFINED;
764ec125fbbSEdward Tomasz Napierala 	refcount_init(&rule->rr_refcount, 1);
765ec125fbbSEdward Tomasz Napierala 
766ec125fbbSEdward Tomasz Napierala 	return (rule);
767ec125fbbSEdward Tomasz Napierala }
768ec125fbbSEdward Tomasz Napierala 
769ec125fbbSEdward Tomasz Napierala struct rctl_rule *
770ec125fbbSEdward Tomasz Napierala rctl_rule_duplicate(const struct rctl_rule *rule, int flags)
771ec125fbbSEdward Tomasz Napierala {
772ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *copy;
773ec125fbbSEdward Tomasz Napierala 
774ec125fbbSEdward Tomasz Napierala 	copy = uma_zalloc(rctl_rule_zone, flags);
775ec125fbbSEdward Tomasz Napierala 	if (copy == NULL)
776ec125fbbSEdward Tomasz Napierala 		return (NULL);
777ec125fbbSEdward Tomasz Napierala 	copy->rr_subject_type = rule->rr_subject_type;
778ec125fbbSEdward Tomasz Napierala 	copy->rr_subject.rs_proc = rule->rr_subject.rs_proc;
779ec125fbbSEdward Tomasz Napierala 	copy->rr_subject.rs_uip = rule->rr_subject.rs_uip;
780415896e3SEdward Tomasz Napierala 	copy->rr_subject.rs_loginclass = rule->rr_subject.rs_loginclass;
781a7ad07bfSEdward Tomasz Napierala 	copy->rr_subject.rs_prison_racct = rule->rr_subject.rs_prison_racct;
782ec125fbbSEdward Tomasz Napierala 	copy->rr_per = rule->rr_per;
783ec125fbbSEdward Tomasz Napierala 	copy->rr_resource = rule->rr_resource;
784ec125fbbSEdward Tomasz Napierala 	copy->rr_action = rule->rr_action;
785ec125fbbSEdward Tomasz Napierala 	copy->rr_amount = rule->rr_amount;
786ec125fbbSEdward Tomasz Napierala 	refcount_init(&copy->rr_refcount, 1);
787ec125fbbSEdward Tomasz Napierala 	rctl_rule_acquire_subject(copy);
788ec125fbbSEdward Tomasz Napierala 
789ec125fbbSEdward Tomasz Napierala 	return (copy);
790ec125fbbSEdward Tomasz Napierala }
791ec125fbbSEdward Tomasz Napierala 
792ec125fbbSEdward Tomasz Napierala void
793ec125fbbSEdward Tomasz Napierala rctl_rule_acquire(struct rctl_rule *rule)
794ec125fbbSEdward Tomasz Napierala {
795ec125fbbSEdward Tomasz Napierala 
796ec125fbbSEdward Tomasz Napierala 	KASSERT(rule->rr_refcount > 0, ("rule->rr_refcount <= 0"));
797ec125fbbSEdward Tomasz Napierala 
798ec125fbbSEdward Tomasz Napierala 	refcount_acquire(&rule->rr_refcount);
799ec125fbbSEdward Tomasz Napierala }
800ec125fbbSEdward Tomasz Napierala 
801ec125fbbSEdward Tomasz Napierala static void
802ec125fbbSEdward Tomasz Napierala rctl_rule_free(void *context, int pending)
803ec125fbbSEdward Tomasz Napierala {
804ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
805ec125fbbSEdward Tomasz Napierala 
806ec125fbbSEdward Tomasz Napierala 	rule = (struct rctl_rule *)context;
807ec125fbbSEdward Tomasz Napierala 
808ec125fbbSEdward Tomasz Napierala 	KASSERT(rule->rr_refcount == 0, ("rule->rr_refcount != 0"));
809ec125fbbSEdward Tomasz Napierala 
810ec125fbbSEdward Tomasz Napierala 	/*
811ec125fbbSEdward Tomasz Napierala 	 * We don't need locking here; rule is guaranteed to be inaccessible.
812ec125fbbSEdward Tomasz Napierala 	 */
813ec125fbbSEdward Tomasz Napierala 
814ec125fbbSEdward Tomasz Napierala 	rctl_rule_release_subject(rule);
815ec125fbbSEdward Tomasz Napierala 	uma_zfree(rctl_rule_zone, rule);
816ec125fbbSEdward Tomasz Napierala }
817ec125fbbSEdward Tomasz Napierala 
818ec125fbbSEdward Tomasz Napierala void
819ec125fbbSEdward Tomasz Napierala rctl_rule_release(struct rctl_rule *rule)
820ec125fbbSEdward Tomasz Napierala {
821ec125fbbSEdward Tomasz Napierala 
822ec125fbbSEdward Tomasz Napierala 	KASSERT(rule->rr_refcount > 0, ("rule->rr_refcount <= 0"));
823ec125fbbSEdward Tomasz Napierala 
824ec125fbbSEdward Tomasz Napierala 	if (refcount_release(&rule->rr_refcount)) {
825ec125fbbSEdward Tomasz Napierala 		/*
826ec125fbbSEdward Tomasz Napierala 		 * rctl_rule_release() is often called when iterating
827ec125fbbSEdward Tomasz Napierala 		 * over all the uidinfo structures in the system,
828ec125fbbSEdward Tomasz Napierala 		 * holding uihashtbl_lock.  Since rctl_rule_free()
829ec125fbbSEdward Tomasz Napierala 		 * might end up calling uifree(), this would lead
830ec125fbbSEdward Tomasz Napierala 		 * to lock recursion.  Use taskqueue to avoid this.
831ec125fbbSEdward Tomasz Napierala 		 */
832ec125fbbSEdward Tomasz Napierala 		TASK_INIT(&rule->rr_task, 0, rctl_rule_free, rule);
833ec125fbbSEdward Tomasz Napierala 		taskqueue_enqueue(taskqueue_thread, &rule->rr_task);
834ec125fbbSEdward Tomasz Napierala 	}
835ec125fbbSEdward Tomasz Napierala }
836ec125fbbSEdward Tomasz Napierala 
837ec125fbbSEdward Tomasz Napierala static int
838ec125fbbSEdward Tomasz Napierala rctl_rule_fully_specified(const struct rctl_rule *rule)
839ec125fbbSEdward Tomasz Napierala {
840ec125fbbSEdward Tomasz Napierala 
841ec125fbbSEdward Tomasz Napierala 	switch (rule->rr_subject_type) {
842ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_UNDEFINED:
843ec125fbbSEdward Tomasz Napierala 		return (0);
844ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_PROCESS:
845ec125fbbSEdward Tomasz Napierala 		if (rule->rr_subject.rs_proc == NULL)
846ec125fbbSEdward Tomasz Napierala 			return (0);
847ec125fbbSEdward Tomasz Napierala 		break;
848ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_USER:
849ec125fbbSEdward Tomasz Napierala 		if (rule->rr_subject.rs_uip == NULL)
850ec125fbbSEdward Tomasz Napierala 			return (0);
851ec125fbbSEdward Tomasz Napierala 		break;
852ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_LOGINCLASS:
853415896e3SEdward Tomasz Napierala 		if (rule->rr_subject.rs_loginclass == NULL)
854ec125fbbSEdward Tomasz Napierala 			return (0);
855ec125fbbSEdward Tomasz Napierala 		break;
856ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_JAIL:
857a7ad07bfSEdward Tomasz Napierala 		if (rule->rr_subject.rs_prison_racct == NULL)
858ec125fbbSEdward Tomasz Napierala 			return (0);
859ec125fbbSEdward Tomasz Napierala 		break;
860ec125fbbSEdward Tomasz Napierala 	default:
861ec125fbbSEdward Tomasz Napierala 		panic("rctl_rule_fully_specified: unknown subject type %d",
862ec125fbbSEdward Tomasz Napierala 		    rule->rr_subject_type);
863ec125fbbSEdward Tomasz Napierala 	}
864ec125fbbSEdward Tomasz Napierala 	if (rule->rr_resource == RACCT_UNDEFINED)
865ec125fbbSEdward Tomasz Napierala 		return (0);
866ec125fbbSEdward Tomasz Napierala 	if (rule->rr_action == RCTL_ACTION_UNDEFINED)
867ec125fbbSEdward Tomasz Napierala 		return (0);
868ec125fbbSEdward Tomasz Napierala 	if (rule->rr_amount == RCTL_AMOUNT_UNDEFINED)
869ec125fbbSEdward Tomasz Napierala 		return (0);
870ec125fbbSEdward Tomasz Napierala 	if (rule->rr_per == RCTL_SUBJECT_TYPE_UNDEFINED)
871ec125fbbSEdward Tomasz Napierala 		return (0);
872ec125fbbSEdward Tomasz Napierala 
873ec125fbbSEdward Tomasz Napierala 	return (1);
874ec125fbbSEdward Tomasz Napierala }
875ec125fbbSEdward Tomasz Napierala 
876ec125fbbSEdward Tomasz Napierala static int
877ec125fbbSEdward Tomasz Napierala rctl_string_to_rule(char *rulestr, struct rctl_rule **rulep)
878ec125fbbSEdward Tomasz Napierala {
879ec125fbbSEdward Tomasz Napierala 	int error = 0;
880ec125fbbSEdward Tomasz Napierala 	char *subjectstr, *subject_idstr, *resourcestr, *actionstr,
881ec125fbbSEdward Tomasz Napierala 	     *amountstr, *perstr;
882ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
883ec125fbbSEdward Tomasz Napierala 	id_t id;
884ec125fbbSEdward Tomasz Napierala 
885ec125fbbSEdward Tomasz Napierala 	rule = rctl_rule_alloc(M_WAITOK);
886ec125fbbSEdward Tomasz Napierala 
887ec125fbbSEdward Tomasz Napierala 	subjectstr = strsep(&rulestr, ":");
888ec125fbbSEdward Tomasz Napierala 	subject_idstr = strsep(&rulestr, ":");
889ec125fbbSEdward Tomasz Napierala 	resourcestr = strsep(&rulestr, ":");
890ec125fbbSEdward Tomasz Napierala 	actionstr = strsep(&rulestr, "=/");
891ec125fbbSEdward Tomasz Napierala 	amountstr = strsep(&rulestr, "/");
892ec125fbbSEdward Tomasz Napierala 	perstr = rulestr;
893ec125fbbSEdward Tomasz Napierala 
894ec125fbbSEdward Tomasz Napierala 	if (subjectstr == NULL || subjectstr[0] == '\0')
895ec125fbbSEdward Tomasz Napierala 		rule->rr_subject_type = RCTL_SUBJECT_TYPE_UNDEFINED;
896ec125fbbSEdward Tomasz Napierala 	else {
897ec125fbbSEdward Tomasz Napierala 		error = str2value(subjectstr, &rule->rr_subject_type, subjectnames);
898ec125fbbSEdward Tomasz Napierala 		if (error != 0)
899ec125fbbSEdward Tomasz Napierala 			goto out;
900ec125fbbSEdward Tomasz Napierala 	}
901ec125fbbSEdward Tomasz Napierala 
902ec125fbbSEdward Tomasz Napierala 	if (subject_idstr == NULL || subject_idstr[0] == '\0') {
903ec125fbbSEdward Tomasz Napierala 		rule->rr_subject.rs_proc = NULL;
904ec125fbbSEdward Tomasz Napierala 		rule->rr_subject.rs_uip = NULL;
905415896e3SEdward Tomasz Napierala 		rule->rr_subject.rs_loginclass = NULL;
906a7ad07bfSEdward Tomasz Napierala 		rule->rr_subject.rs_prison_racct = NULL;
907ec125fbbSEdward Tomasz Napierala 	} else {
908ec125fbbSEdward Tomasz Napierala 		switch (rule->rr_subject_type) {
909ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_UNDEFINED:
910ec125fbbSEdward Tomasz Napierala 			error = EINVAL;
911ec125fbbSEdward Tomasz Napierala 			goto out;
912ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_PROCESS:
913ec125fbbSEdward Tomasz Napierala 			error = str2id(subject_idstr, &id);
914ec125fbbSEdward Tomasz Napierala 			if (error != 0)
915ec125fbbSEdward Tomasz Napierala 				goto out;
916ec125fbbSEdward Tomasz Napierala 			sx_assert(&allproc_lock, SA_LOCKED);
917ec125fbbSEdward Tomasz Napierala 			rule->rr_subject.rs_proc = pfind(id);
918ec125fbbSEdward Tomasz Napierala 			if (rule->rr_subject.rs_proc == NULL) {
919ec125fbbSEdward Tomasz Napierala 				error = ESRCH;
920ec125fbbSEdward Tomasz Napierala 				goto out;
921ec125fbbSEdward Tomasz Napierala 			}
922ec125fbbSEdward Tomasz Napierala 			PROC_UNLOCK(rule->rr_subject.rs_proc);
923ec125fbbSEdward Tomasz Napierala 			break;
924ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_USER:
925ec125fbbSEdward Tomasz Napierala 			error = str2id(subject_idstr, &id);
926ec125fbbSEdward Tomasz Napierala 			if (error != 0)
927ec125fbbSEdward Tomasz Napierala 				goto out;
928ec125fbbSEdward Tomasz Napierala 			rule->rr_subject.rs_uip = uifind(id);
929ec125fbbSEdward Tomasz Napierala 			break;
930ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_LOGINCLASS:
931415896e3SEdward Tomasz Napierala 			rule->rr_subject.rs_loginclass =
932ec125fbbSEdward Tomasz Napierala 			    loginclass_find(subject_idstr);
933415896e3SEdward Tomasz Napierala 			if (rule->rr_subject.rs_loginclass == NULL) {
934ec125fbbSEdward Tomasz Napierala 				error = ENAMETOOLONG;
935ec125fbbSEdward Tomasz Napierala 				goto out;
936ec125fbbSEdward Tomasz Napierala 			}
937ec125fbbSEdward Tomasz Napierala 			break;
938ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_JAIL:
939a7ad07bfSEdward Tomasz Napierala 			rule->rr_subject.rs_prison_racct =
940a7ad07bfSEdward Tomasz Napierala 			    prison_racct_find(subject_idstr);
941a7ad07bfSEdward Tomasz Napierala 			if (rule->rr_subject.rs_prison_racct == NULL) {
942a7ad07bfSEdward Tomasz Napierala 				error = ENAMETOOLONG;
943ec125fbbSEdward Tomasz Napierala 				goto out;
944ec125fbbSEdward Tomasz Napierala 			}
945ec125fbbSEdward Tomasz Napierala 			break;
946ec125fbbSEdward Tomasz Napierala                default:
947ec125fbbSEdward Tomasz Napierala                        panic("rctl_string_to_rule: unknown subject type %d",
948ec125fbbSEdward Tomasz Napierala                            rule->rr_subject_type);
949ec125fbbSEdward Tomasz Napierala                }
950ec125fbbSEdward Tomasz Napierala 	}
951ec125fbbSEdward Tomasz Napierala 
952ec125fbbSEdward Tomasz Napierala 	if (resourcestr == NULL || resourcestr[0] == '\0')
953ec125fbbSEdward Tomasz Napierala 		rule->rr_resource = RACCT_UNDEFINED;
954ec125fbbSEdward Tomasz Napierala 	else {
955ec125fbbSEdward Tomasz Napierala 		error = str2value(resourcestr, &rule->rr_resource,
956ec125fbbSEdward Tomasz Napierala 		    resourcenames);
957ec125fbbSEdward Tomasz Napierala 		if (error != 0)
958ec125fbbSEdward Tomasz Napierala 			goto out;
959ec125fbbSEdward Tomasz Napierala 	}
960ec125fbbSEdward Tomasz Napierala 
961ec125fbbSEdward Tomasz Napierala 	if (actionstr == NULL || actionstr[0] == '\0')
962ec125fbbSEdward Tomasz Napierala 		rule->rr_action = RCTL_ACTION_UNDEFINED;
963ec125fbbSEdward Tomasz Napierala 	else {
964ec125fbbSEdward Tomasz Napierala 		error = str2value(actionstr, &rule->rr_action, actionnames);
965ec125fbbSEdward Tomasz Napierala 		if (error != 0)
966ec125fbbSEdward Tomasz Napierala 			goto out;
967ec125fbbSEdward Tomasz Napierala 	}
968ec125fbbSEdward Tomasz Napierala 
969ec125fbbSEdward Tomasz Napierala 	if (amountstr == NULL || amountstr[0] == '\0')
970ec125fbbSEdward Tomasz Napierala 		rule->rr_amount = RCTL_AMOUNT_UNDEFINED;
971ec125fbbSEdward Tomasz Napierala 	else {
972ec125fbbSEdward Tomasz Napierala 		error = str2int64(amountstr, &rule->rr_amount);
973ec125fbbSEdward Tomasz Napierala 		if (error != 0)
974ec125fbbSEdward Tomasz Napierala 			goto out;
97585a2f1b4SEdward Tomasz Napierala 		if (RACCT_IS_IN_MILLIONS(rule->rr_resource))
976cff08ec0SEdward Tomasz Napierala 			rule->rr_amount *= 1000000;
977ec125fbbSEdward Tomasz Napierala 	}
978ec125fbbSEdward Tomasz Napierala 
979ec125fbbSEdward Tomasz Napierala 	if (perstr == NULL || perstr[0] == '\0')
980ec125fbbSEdward Tomasz Napierala 		rule->rr_per = RCTL_SUBJECT_TYPE_UNDEFINED;
981ec125fbbSEdward Tomasz Napierala 	else {
982ec125fbbSEdward Tomasz Napierala 		error = str2value(perstr, &rule->rr_per, subjectnames);
983ec125fbbSEdward Tomasz Napierala 		if (error != 0)
984ec125fbbSEdward Tomasz Napierala 			goto out;
985ec125fbbSEdward Tomasz Napierala 	}
986ec125fbbSEdward Tomasz Napierala 
987ec125fbbSEdward Tomasz Napierala out:
988ec125fbbSEdward Tomasz Napierala 	if (error == 0)
989ec125fbbSEdward Tomasz Napierala 		*rulep = rule;
990ec125fbbSEdward Tomasz Napierala 	else
991ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(rule);
992ec125fbbSEdward Tomasz Napierala 
993ec125fbbSEdward Tomasz Napierala 	return (error);
994ec125fbbSEdward Tomasz Napierala }
995ec125fbbSEdward Tomasz Napierala 
996ec125fbbSEdward Tomasz Napierala /*
997ec125fbbSEdward Tomasz Napierala  * Link a rule with all the subjects it applies to.
998ec125fbbSEdward Tomasz Napierala  */
999ec125fbbSEdward Tomasz Napierala int
1000ec125fbbSEdward Tomasz Napierala rctl_rule_add(struct rctl_rule *rule)
1001ec125fbbSEdward Tomasz Napierala {
1002ec125fbbSEdward Tomasz Napierala 	struct proc *p;
1003ec125fbbSEdward Tomasz Napierala 	struct ucred *cred;
1004ec125fbbSEdward Tomasz Napierala 	struct uidinfo *uip;
1005ec125fbbSEdward Tomasz Napierala 	struct prison *pr;
1006a7ad07bfSEdward Tomasz Napierala 	struct prison_racct *prr;
1007ec125fbbSEdward Tomasz Napierala 	struct loginclass *lc;
1008ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule2;
1009ec125fbbSEdward Tomasz Napierala 	int match;
1010ec125fbbSEdward Tomasz Napierala 
1011ec125fbbSEdward Tomasz Napierala 	KASSERT(rctl_rule_fully_specified(rule), ("rule not fully specified"));
1012ec125fbbSEdward Tomasz Napierala 
1013ec125fbbSEdward Tomasz Napierala 	/*
1014ec125fbbSEdward Tomasz Napierala 	 * Some rules just don't make sense.  Note that the one below
10154fe84775SEdward Tomasz Napierala 	 * cannot be rewritten using RACCT_IS_DENIABLE(); the RACCT_PCTCPU,
1016ec125fbbSEdward Tomasz Napierala 	 * for example, is not deniable in the racct sense, but the
1017ec125fbbSEdward Tomasz Napierala 	 * limit is enforced in a different way, so "deny" rules for %CPU
1018ec125fbbSEdward Tomasz Napierala 	 * do make sense.
1019ec125fbbSEdward Tomasz Napierala 	 */
1020ec125fbbSEdward Tomasz Napierala 	if (rule->rr_action == RCTL_ACTION_DENY &&
1021ec125fbbSEdward Tomasz Napierala 	    (rule->rr_resource == RACCT_CPU ||
1022ec125fbbSEdward Tomasz Napierala 	    rule->rr_resource == RACCT_WALLCLOCK))
1023ec125fbbSEdward Tomasz Napierala 		return (EOPNOTSUPP);
1024ec125fbbSEdward Tomasz Napierala 
1025ec125fbbSEdward Tomasz Napierala 	if (rule->rr_per == RCTL_SUBJECT_TYPE_PROCESS &&
10264fe84775SEdward Tomasz Napierala 	    RACCT_IS_SLOPPY(rule->rr_resource))
1027ec125fbbSEdward Tomasz Napierala 		return (EOPNOTSUPP);
1028ec125fbbSEdward Tomasz Napierala 
1029ec125fbbSEdward Tomasz Napierala 	/*
1030ec125fbbSEdward Tomasz Napierala 	 * Make sure there are no duplicated rules.  Also, for the "deny"
1031ec125fbbSEdward Tomasz Napierala 	 * rules, remove ones differing only by "amount".
1032ec125fbbSEdward Tomasz Napierala 	 */
1033ec125fbbSEdward Tomasz Napierala 	if (rule->rr_action == RCTL_ACTION_DENY) {
1034ec125fbbSEdward Tomasz Napierala 		rule2 = rctl_rule_duplicate(rule, M_WAITOK);
1035ec125fbbSEdward Tomasz Napierala 		rule2->rr_amount = RCTL_AMOUNT_UNDEFINED;
1036ec125fbbSEdward Tomasz Napierala 		rctl_rule_remove(rule2);
1037ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(rule2);
1038ec125fbbSEdward Tomasz Napierala 	} else
1039ec125fbbSEdward Tomasz Napierala 		rctl_rule_remove(rule);
1040ec125fbbSEdward Tomasz Napierala 
1041ec125fbbSEdward Tomasz Napierala 	switch (rule->rr_subject_type) {
1042ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_PROCESS:
1043ec125fbbSEdward Tomasz Napierala 		p = rule->rr_subject.rs_proc;
1044ec125fbbSEdward Tomasz Napierala 		KASSERT(p != NULL, ("rctl_rule_add: NULL proc"));
1045ec125fbbSEdward Tomasz Napierala 
1046ec125fbbSEdward Tomasz Napierala 		rctl_racct_add_rule(p->p_racct, rule);
1047ec125fbbSEdward Tomasz Napierala 		/*
1048ec125fbbSEdward Tomasz Napierala 		 * In case of per-process rule, we don't have anything more
1049ec125fbbSEdward Tomasz Napierala 		 * to do.
1050ec125fbbSEdward Tomasz Napierala 		 */
1051ec125fbbSEdward Tomasz Napierala 		return (0);
1052ec125fbbSEdward Tomasz Napierala 
1053ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_USER:
1054ec125fbbSEdward Tomasz Napierala 		uip = rule->rr_subject.rs_uip;
1055ec125fbbSEdward Tomasz Napierala 		KASSERT(uip != NULL, ("rctl_rule_add: NULL uip"));
1056ec125fbbSEdward Tomasz Napierala 		rctl_racct_add_rule(uip->ui_racct, rule);
1057ec125fbbSEdward Tomasz Napierala 		break;
1058ec125fbbSEdward Tomasz Napierala 
1059ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_LOGINCLASS:
1060415896e3SEdward Tomasz Napierala 		lc = rule->rr_subject.rs_loginclass;
1061ec125fbbSEdward Tomasz Napierala 		KASSERT(lc != NULL, ("rctl_rule_add: NULL loginclass"));
1062ec125fbbSEdward Tomasz Napierala 		rctl_racct_add_rule(lc->lc_racct, rule);
1063ec125fbbSEdward Tomasz Napierala 		break;
1064ec125fbbSEdward Tomasz Napierala 
1065ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_JAIL:
1066a7ad07bfSEdward Tomasz Napierala 		prr = rule->rr_subject.rs_prison_racct;
1067a7ad07bfSEdward Tomasz Napierala 		KASSERT(prr != NULL, ("rctl_rule_add: NULL pr"));
1068a7ad07bfSEdward Tomasz Napierala 		rctl_racct_add_rule(prr->prr_racct, rule);
1069ec125fbbSEdward Tomasz Napierala 		break;
1070ec125fbbSEdward Tomasz Napierala 
1071ec125fbbSEdward Tomasz Napierala 	default:
1072ec125fbbSEdward Tomasz Napierala 		panic("rctl_rule_add: unknown subject type %d",
1073ec125fbbSEdward Tomasz Napierala 		    rule->rr_subject_type);
1074ec125fbbSEdward Tomasz Napierala 	}
1075ec125fbbSEdward Tomasz Napierala 
1076ec125fbbSEdward Tomasz Napierala 	/*
1077ec125fbbSEdward Tomasz Napierala 	 * Now go through all the processes and add the new rule to the ones
1078ec125fbbSEdward Tomasz Napierala 	 * it applies to.
1079ec125fbbSEdward Tomasz Napierala 	 */
1080ec125fbbSEdward Tomasz Napierala 	sx_assert(&allproc_lock, SA_LOCKED);
1081ec125fbbSEdward Tomasz Napierala 	FOREACH_PROC_IN_SYSTEM(p) {
1082ec125fbbSEdward Tomasz Napierala 		cred = p->p_ucred;
1083ec125fbbSEdward Tomasz Napierala 		switch (rule->rr_subject_type) {
1084ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_USER:
1085ec125fbbSEdward Tomasz Napierala 			if (cred->cr_uidinfo == rule->rr_subject.rs_uip ||
1086ec125fbbSEdward Tomasz Napierala 			    cred->cr_ruidinfo == rule->rr_subject.rs_uip)
1087ec125fbbSEdward Tomasz Napierala 				break;
1088ec125fbbSEdward Tomasz Napierala 			continue;
1089ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_LOGINCLASS:
1090415896e3SEdward Tomasz Napierala 			if (cred->cr_loginclass == rule->rr_subject.rs_loginclass)
1091ec125fbbSEdward Tomasz Napierala 				break;
1092ec125fbbSEdward Tomasz Napierala 			continue;
1093ec125fbbSEdward Tomasz Napierala 		case RCTL_SUBJECT_TYPE_JAIL:
1094ec125fbbSEdward Tomasz Napierala 			match = 0;
1095ec125fbbSEdward Tomasz Napierala 			for (pr = cred->cr_prison; pr != NULL; pr = pr->pr_parent) {
1096a7ad07bfSEdward Tomasz Napierala 				if (pr->pr_prison_racct == rule->rr_subject.rs_prison_racct) {
1097ec125fbbSEdward Tomasz Napierala 					match = 1;
1098ec125fbbSEdward Tomasz Napierala 					break;
1099ec125fbbSEdward Tomasz Napierala 				}
1100ec125fbbSEdward Tomasz Napierala 			}
1101ec125fbbSEdward Tomasz Napierala 			if (match)
1102ec125fbbSEdward Tomasz Napierala 				break;
1103ec125fbbSEdward Tomasz Napierala 			continue;
1104ec125fbbSEdward Tomasz Napierala 		default:
1105ec125fbbSEdward Tomasz Napierala 			panic("rctl_rule_add: unknown subject type %d",
1106ec125fbbSEdward Tomasz Napierala 			    rule->rr_subject_type);
1107ec125fbbSEdward Tomasz Napierala 		}
1108ec125fbbSEdward Tomasz Napierala 
1109ec125fbbSEdward Tomasz Napierala 		rctl_racct_add_rule(p->p_racct, rule);
1110ec125fbbSEdward Tomasz Napierala 	}
1111ec125fbbSEdward Tomasz Napierala 
1112ec125fbbSEdward Tomasz Napierala 	return (0);
1113ec125fbbSEdward Tomasz Napierala }
1114ec125fbbSEdward Tomasz Napierala 
1115ec125fbbSEdward Tomasz Napierala static void
1116ec125fbbSEdward Tomasz Napierala rctl_rule_remove_callback(struct racct *racct, void *arg2, void *arg3)
1117ec125fbbSEdward Tomasz Napierala {
1118ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *filter = (struct rctl_rule *)arg2;
1119ec125fbbSEdward Tomasz Napierala 	int found = 0;
1120ec125fbbSEdward Tomasz Napierala 
1121ec125fbbSEdward Tomasz Napierala 	rw_wlock(&rctl_lock);
1122ec125fbbSEdward Tomasz Napierala 	found += rctl_racct_remove_rules(racct, filter);
1123ec125fbbSEdward Tomasz Napierala 	rw_wunlock(&rctl_lock);
1124ec125fbbSEdward Tomasz Napierala 
1125ec125fbbSEdward Tomasz Napierala 	*((int *)arg3) += found;
1126ec125fbbSEdward Tomasz Napierala }
1127ec125fbbSEdward Tomasz Napierala 
1128ec125fbbSEdward Tomasz Napierala /*
1129ec125fbbSEdward Tomasz Napierala  * Remove all rules that match the filter.
1130ec125fbbSEdward Tomasz Napierala  */
1131ec125fbbSEdward Tomasz Napierala int
1132ec125fbbSEdward Tomasz Napierala rctl_rule_remove(struct rctl_rule *filter)
1133ec125fbbSEdward Tomasz Napierala {
1134ec125fbbSEdward Tomasz Napierala 	int found = 0;
1135ec125fbbSEdward Tomasz Napierala 	struct proc *p;
1136ec125fbbSEdward Tomasz Napierala 
1137ec125fbbSEdward Tomasz Napierala 	if (filter->rr_subject_type == RCTL_SUBJECT_TYPE_PROCESS &&
1138ec125fbbSEdward Tomasz Napierala 	    filter->rr_subject.rs_proc != NULL) {
1139ec125fbbSEdward Tomasz Napierala 		p = filter->rr_subject.rs_proc;
1140ec125fbbSEdward Tomasz Napierala 		rw_wlock(&rctl_lock);
1141ec125fbbSEdward Tomasz Napierala 		found = rctl_racct_remove_rules(p->p_racct, filter);
1142ec125fbbSEdward Tomasz Napierala 		rw_wunlock(&rctl_lock);
1143ec125fbbSEdward Tomasz Napierala 		if (found)
1144ec125fbbSEdward Tomasz Napierala 			return (0);
1145ec125fbbSEdward Tomasz Napierala 		return (ESRCH);
1146ec125fbbSEdward Tomasz Napierala 	}
1147ec125fbbSEdward Tomasz Napierala 
1148ec125fbbSEdward Tomasz Napierala 	loginclass_racct_foreach(rctl_rule_remove_callback, filter,
1149ec125fbbSEdward Tomasz Napierala 	    (void *)&found);
1150ec125fbbSEdward Tomasz Napierala 	ui_racct_foreach(rctl_rule_remove_callback, filter,
1151ec125fbbSEdward Tomasz Napierala 	    (void *)&found);
1152ec125fbbSEdward Tomasz Napierala 	prison_racct_foreach(rctl_rule_remove_callback, filter,
1153ec125fbbSEdward Tomasz Napierala 	    (void *)&found);
1154ec125fbbSEdward Tomasz Napierala 
1155ec125fbbSEdward Tomasz Napierala 	sx_assert(&allproc_lock, SA_LOCKED);
1156ec125fbbSEdward Tomasz Napierala 	rw_wlock(&rctl_lock);
1157ec125fbbSEdward Tomasz Napierala 	FOREACH_PROC_IN_SYSTEM(p) {
1158ec125fbbSEdward Tomasz Napierala 		found += rctl_racct_remove_rules(p->p_racct, filter);
1159ec125fbbSEdward Tomasz Napierala 	}
1160ec125fbbSEdward Tomasz Napierala 	rw_wunlock(&rctl_lock);
1161ec125fbbSEdward Tomasz Napierala 
1162ec125fbbSEdward Tomasz Napierala 	if (found)
1163ec125fbbSEdward Tomasz Napierala 		return (0);
1164ec125fbbSEdward Tomasz Napierala 	return (ESRCH);
1165ec125fbbSEdward Tomasz Napierala }
1166ec125fbbSEdward Tomasz Napierala 
1167ec125fbbSEdward Tomasz Napierala /*
1168ec125fbbSEdward Tomasz Napierala  * Appends a rule to the sbuf.
1169ec125fbbSEdward Tomasz Napierala  */
1170ec125fbbSEdward Tomasz Napierala static void
1171ec125fbbSEdward Tomasz Napierala rctl_rule_to_sbuf(struct sbuf *sb, const struct rctl_rule *rule)
1172ec125fbbSEdward Tomasz Napierala {
1173ec125fbbSEdward Tomasz Napierala 	int64_t amount;
1174ec125fbbSEdward Tomasz Napierala 
1175ec125fbbSEdward Tomasz Napierala 	sbuf_printf(sb, "%s:", rctl_subject_type_name(rule->rr_subject_type));
1176ec125fbbSEdward Tomasz Napierala 
1177ec125fbbSEdward Tomasz Napierala 	switch (rule->rr_subject_type) {
1178ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_PROCESS:
1179ec125fbbSEdward Tomasz Napierala 		if (rule->rr_subject.rs_proc == NULL)
1180ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, ":");
1181ec125fbbSEdward Tomasz Napierala 		else
1182ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, "%d:",
1183ec125fbbSEdward Tomasz Napierala 			    rule->rr_subject.rs_proc->p_pid);
1184ec125fbbSEdward Tomasz Napierala 		break;
1185ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_USER:
1186ec125fbbSEdward Tomasz Napierala 		if (rule->rr_subject.rs_uip == NULL)
1187ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, ":");
1188ec125fbbSEdward Tomasz Napierala 		else
1189ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, "%d:",
1190ec125fbbSEdward Tomasz Napierala 			    rule->rr_subject.rs_uip->ui_uid);
1191ec125fbbSEdward Tomasz Napierala 		break;
1192ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_LOGINCLASS:
1193415896e3SEdward Tomasz Napierala 		if (rule->rr_subject.rs_loginclass == NULL)
1194ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, ":");
1195ec125fbbSEdward Tomasz Napierala 		else
1196ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, "%s:",
1197415896e3SEdward Tomasz Napierala 			    rule->rr_subject.rs_loginclass->lc_name);
1198ec125fbbSEdward Tomasz Napierala 		break;
1199ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_JAIL:
1200a7ad07bfSEdward Tomasz Napierala 		if (rule->rr_subject.rs_prison_racct == NULL)
1201ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, ":");
1202ec125fbbSEdward Tomasz Napierala 		else
1203ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, "%s:",
1204a7ad07bfSEdward Tomasz Napierala 			    rule->rr_subject.rs_prison_racct->prr_name);
1205ec125fbbSEdward Tomasz Napierala 		break;
1206ec125fbbSEdward Tomasz Napierala 	default:
1207ec125fbbSEdward Tomasz Napierala 		panic("rctl_rule_to_sbuf: unknown subject type %d",
1208ec125fbbSEdward Tomasz Napierala 		    rule->rr_subject_type);
1209ec125fbbSEdward Tomasz Napierala 	}
1210ec125fbbSEdward Tomasz Napierala 
1211ec125fbbSEdward Tomasz Napierala 	amount = rule->rr_amount;
1212ec125fbbSEdward Tomasz Napierala 	if (amount != RCTL_AMOUNT_UNDEFINED &&
121385a2f1b4SEdward Tomasz Napierala 	    RACCT_IS_IN_MILLIONS(rule->rr_resource))
121485a2f1b4SEdward Tomasz Napierala 		amount /= 1000000;
1215ec125fbbSEdward Tomasz Napierala 
1216ec125fbbSEdward Tomasz Napierala 	sbuf_printf(sb, "%s:%s=%jd",
1217ec125fbbSEdward Tomasz Napierala 	    rctl_resource_name(rule->rr_resource),
1218ec125fbbSEdward Tomasz Napierala 	    rctl_action_name(rule->rr_action),
1219ec125fbbSEdward Tomasz Napierala 	    amount);
1220ec125fbbSEdward Tomasz Napierala 
1221ec125fbbSEdward Tomasz Napierala 	if (rule->rr_per != rule->rr_subject_type)
1222ec125fbbSEdward Tomasz Napierala 		sbuf_printf(sb, "/%s", rctl_subject_type_name(rule->rr_per));
1223ec125fbbSEdward Tomasz Napierala }
1224ec125fbbSEdward Tomasz Napierala 
1225ec125fbbSEdward Tomasz Napierala /*
1226ec125fbbSEdward Tomasz Napierala  * Routine used by RCTL syscalls to read in input string.
1227ec125fbbSEdward Tomasz Napierala  */
1228ec125fbbSEdward Tomasz Napierala static int
1229ec125fbbSEdward Tomasz Napierala rctl_read_inbuf(char **inputstr, const char *inbufp, size_t inbuflen)
1230ec125fbbSEdward Tomasz Napierala {
1231ec125fbbSEdward Tomasz Napierala 	int error;
1232ec125fbbSEdward Tomasz Napierala 	char *str;
1233ec125fbbSEdward Tomasz Napierala 
1234ec125fbbSEdward Tomasz Napierala 	if (inbuflen <= 0)
1235ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
1236786813aaSEdward Tomasz Napierala 	if (inbuflen > RCTL_MAX_INBUFLEN)
1237786813aaSEdward Tomasz Napierala 		return (E2BIG);
1238ec125fbbSEdward Tomasz Napierala 
1239ec125fbbSEdward Tomasz Napierala 	str = malloc(inbuflen + 1, M_RCTL, M_WAITOK);
1240ec125fbbSEdward Tomasz Napierala 	error = copyinstr(inbufp, str, inbuflen, NULL);
1241ec125fbbSEdward Tomasz Napierala 	if (error != 0) {
1242ec125fbbSEdward Tomasz Napierala 		free(str, M_RCTL);
1243ec125fbbSEdward Tomasz Napierala 		return (error);
1244ec125fbbSEdward Tomasz Napierala 	}
1245ec125fbbSEdward Tomasz Napierala 
1246ec125fbbSEdward Tomasz Napierala 	*inputstr = str;
1247ec125fbbSEdward Tomasz Napierala 
1248ec125fbbSEdward Tomasz Napierala 	return (0);
1249ec125fbbSEdward Tomasz Napierala }
1250ec125fbbSEdward Tomasz Napierala 
1251ec125fbbSEdward Tomasz Napierala /*
1252ec125fbbSEdward Tomasz Napierala  * Routine used by RCTL syscalls to write out output string.
1253ec125fbbSEdward Tomasz Napierala  */
1254ec125fbbSEdward Tomasz Napierala static int
1255ec125fbbSEdward Tomasz Napierala rctl_write_outbuf(struct sbuf *outputsbuf, char *outbufp, size_t outbuflen)
1256ec125fbbSEdward Tomasz Napierala {
1257ec125fbbSEdward Tomasz Napierala 	int error;
1258ec125fbbSEdward Tomasz Napierala 
1259ec125fbbSEdward Tomasz Napierala 	if (outputsbuf == NULL)
1260ec125fbbSEdward Tomasz Napierala 		return (0);
1261ec125fbbSEdward Tomasz Napierala 
1262ec125fbbSEdward Tomasz Napierala 	sbuf_finish(outputsbuf);
1263ec125fbbSEdward Tomasz Napierala 	if (outbuflen < sbuf_len(outputsbuf) + 1) {
1264ec125fbbSEdward Tomasz Napierala 		sbuf_delete(outputsbuf);
1265ec125fbbSEdward Tomasz Napierala 		return (ERANGE);
1266ec125fbbSEdward Tomasz Napierala 	}
1267ec125fbbSEdward Tomasz Napierala 	error = copyout(sbuf_data(outputsbuf), outbufp,
1268ec125fbbSEdward Tomasz Napierala 	    sbuf_len(outputsbuf) + 1);
1269ec125fbbSEdward Tomasz Napierala 	sbuf_delete(outputsbuf);
1270ec125fbbSEdward Tomasz Napierala 	return (error);
1271ec125fbbSEdward Tomasz Napierala }
1272ec125fbbSEdward Tomasz Napierala 
1273ec125fbbSEdward Tomasz Napierala static struct sbuf *
1274ec125fbbSEdward Tomasz Napierala rctl_racct_to_sbuf(struct racct *racct, int sloppy)
1275ec125fbbSEdward Tomasz Napierala {
1276ec125fbbSEdward Tomasz Napierala 	int i;
1277ec125fbbSEdward Tomasz Napierala 	int64_t amount;
1278ec125fbbSEdward Tomasz Napierala 	struct sbuf *sb;
1279ec125fbbSEdward Tomasz Napierala 
1280ec125fbbSEdward Tomasz Napierala 	sb = sbuf_new_auto();
1281ec125fbbSEdward Tomasz Napierala 	for (i = 0; i <= RACCT_MAX; i++) {
12824fe84775SEdward Tomasz Napierala 		if (sloppy == 0 && RACCT_IS_SLOPPY(i))
1283ec125fbbSEdward Tomasz Napierala 			continue;
1284ec125fbbSEdward Tomasz Napierala 		amount = racct->r_resources[i];
128585a2f1b4SEdward Tomasz Napierala 		if (RACCT_IS_IN_MILLIONS(i))
1286cff08ec0SEdward Tomasz Napierala 			amount /= 1000000;
1287ec125fbbSEdward Tomasz Napierala 		sbuf_printf(sb, "%s=%jd,", rctl_resource_name(i), amount);
1288ec125fbbSEdward Tomasz Napierala 	}
1289ec125fbbSEdward Tomasz Napierala 	sbuf_setpos(sb, sbuf_len(sb) - 1);
1290ec125fbbSEdward Tomasz Napierala 	return (sb);
1291ec125fbbSEdward Tomasz Napierala }
1292ec125fbbSEdward Tomasz Napierala 
1293ec125fbbSEdward Tomasz Napierala int
12948451d0ddSKip Macy sys_rctl_get_racct(struct thread *td, struct rctl_get_racct_args *uap)
1295ec125fbbSEdward Tomasz Napierala {
1296ec125fbbSEdward Tomasz Napierala 	int error;
1297ec125fbbSEdward Tomasz Napierala 	char *inputstr;
1298ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *filter;
1299ec125fbbSEdward Tomasz Napierala 	struct sbuf *outputsbuf = NULL;
1300ec125fbbSEdward Tomasz Napierala 	struct proc *p;
1301ec125fbbSEdward Tomasz Napierala 	struct uidinfo *uip;
1302ec125fbbSEdward Tomasz Napierala 	struct loginclass *lc;
1303a7ad07bfSEdward Tomasz Napierala 	struct prison_racct *prr;
1304ec125fbbSEdward Tomasz Napierala 
1305415896e3SEdward Tomasz Napierala 	error = priv_check(td, PRIV_RCTL_GET_RACCT);
1306ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1307ec125fbbSEdward Tomasz Napierala 		return (error);
1308ec125fbbSEdward Tomasz Napierala 
1309ec125fbbSEdward Tomasz Napierala 	error = rctl_read_inbuf(&inputstr, uap->inbufp, uap->inbuflen);
1310ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1311ec125fbbSEdward Tomasz Napierala 		return (error);
1312ec125fbbSEdward Tomasz Napierala 
1313ec125fbbSEdward Tomasz Napierala 	sx_slock(&allproc_lock);
1314ec125fbbSEdward Tomasz Napierala 	error = rctl_string_to_rule(inputstr, &filter);
1315ec125fbbSEdward Tomasz Napierala 	free(inputstr, M_RCTL);
1316ec125fbbSEdward Tomasz Napierala 	if (error != 0) {
1317ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1318ec125fbbSEdward Tomasz Napierala 		return (error);
1319ec125fbbSEdward Tomasz Napierala 	}
1320ec125fbbSEdward Tomasz Napierala 
1321ec125fbbSEdward Tomasz Napierala 	switch (filter->rr_subject_type) {
1322ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_PROCESS:
1323ec125fbbSEdward Tomasz Napierala 		p = filter->rr_subject.rs_proc;
1324ec125fbbSEdward Tomasz Napierala 		if (p == NULL) {
1325ec125fbbSEdward Tomasz Napierala 			error = EINVAL;
1326ec125fbbSEdward Tomasz Napierala 			goto out;
1327ec125fbbSEdward Tomasz Napierala 		}
1328ec125fbbSEdward Tomasz Napierala 		outputsbuf = rctl_racct_to_sbuf(p->p_racct, 0);
1329ec125fbbSEdward Tomasz Napierala 		break;
1330ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_USER:
1331ec125fbbSEdward Tomasz Napierala 		uip = filter->rr_subject.rs_uip;
1332ec125fbbSEdward Tomasz Napierala 		if (uip == NULL) {
1333ec125fbbSEdward Tomasz Napierala 			error = EINVAL;
1334ec125fbbSEdward Tomasz Napierala 			goto out;
1335ec125fbbSEdward Tomasz Napierala 		}
1336ec125fbbSEdward Tomasz Napierala 		outputsbuf = rctl_racct_to_sbuf(uip->ui_racct, 1);
1337ec125fbbSEdward Tomasz Napierala 		break;
1338ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_LOGINCLASS:
1339415896e3SEdward Tomasz Napierala 		lc = filter->rr_subject.rs_loginclass;
1340ec125fbbSEdward Tomasz Napierala 		if (lc == NULL) {
1341ec125fbbSEdward Tomasz Napierala 			error = EINVAL;
1342ec125fbbSEdward Tomasz Napierala 			goto out;
1343ec125fbbSEdward Tomasz Napierala 		}
1344ec125fbbSEdward Tomasz Napierala 		outputsbuf = rctl_racct_to_sbuf(lc->lc_racct, 1);
1345ec125fbbSEdward Tomasz Napierala 		break;
1346ec125fbbSEdward Tomasz Napierala 	case RCTL_SUBJECT_TYPE_JAIL:
1347a7ad07bfSEdward Tomasz Napierala 		prr = filter->rr_subject.rs_prison_racct;
1348a7ad07bfSEdward Tomasz Napierala 		if (prr == NULL) {
1349ec125fbbSEdward Tomasz Napierala 			error = EINVAL;
1350ec125fbbSEdward Tomasz Napierala 			goto out;
1351ec125fbbSEdward Tomasz Napierala 		}
1352a7ad07bfSEdward Tomasz Napierala 		outputsbuf = rctl_racct_to_sbuf(prr->prr_racct, 1);
1353ec125fbbSEdward Tomasz Napierala 		break;
1354ec125fbbSEdward Tomasz Napierala 	default:
1355ec125fbbSEdward Tomasz Napierala 		error = EINVAL;
1356ec125fbbSEdward Tomasz Napierala 	}
1357ec125fbbSEdward Tomasz Napierala out:
1358ec125fbbSEdward Tomasz Napierala 	rctl_rule_release(filter);
1359ec125fbbSEdward Tomasz Napierala 	sx_sunlock(&allproc_lock);
1360ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1361ec125fbbSEdward Tomasz Napierala 		return (error);
1362ec125fbbSEdward Tomasz Napierala 
1363ec125fbbSEdward Tomasz Napierala 	error = rctl_write_outbuf(outputsbuf, uap->outbufp, uap->outbuflen);
1364ec125fbbSEdward Tomasz Napierala 
1365ec125fbbSEdward Tomasz Napierala 	return (error);
1366ec125fbbSEdward Tomasz Napierala }
1367ec125fbbSEdward Tomasz Napierala 
1368ec125fbbSEdward Tomasz Napierala static void
1369ec125fbbSEdward Tomasz Napierala rctl_get_rules_callback(struct racct *racct, void *arg2, void *arg3)
1370ec125fbbSEdward Tomasz Napierala {
1371ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *filter = (struct rctl_rule *)arg2;
1372ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
1373ec125fbbSEdward Tomasz Napierala 	struct sbuf *sb = (struct sbuf *)arg3;
1374ec125fbbSEdward Tomasz Napierala 
1375ec125fbbSEdward Tomasz Napierala 	rw_rlock(&rctl_lock);
1376ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &racct->r_rule_links, rrl_next) {
1377ec125fbbSEdward Tomasz Napierala 		if (!rctl_rule_matches(link->rrl_rule, filter))
1378ec125fbbSEdward Tomasz Napierala 			continue;
1379ec125fbbSEdward Tomasz Napierala 		rctl_rule_to_sbuf(sb, link->rrl_rule);
1380ec125fbbSEdward Tomasz Napierala 		sbuf_printf(sb, ",");
1381ec125fbbSEdward Tomasz Napierala 	}
1382ec125fbbSEdward Tomasz Napierala 	rw_runlock(&rctl_lock);
1383ec125fbbSEdward Tomasz Napierala }
1384ec125fbbSEdward Tomasz Napierala 
1385ec125fbbSEdward Tomasz Napierala int
13868451d0ddSKip Macy sys_rctl_get_rules(struct thread *td, struct rctl_get_rules_args *uap)
1387ec125fbbSEdward Tomasz Napierala {
1388ec125fbbSEdward Tomasz Napierala 	int error;
1389ec125fbbSEdward Tomasz Napierala 	size_t bufsize = RCTL_DEFAULT_BUFSIZE;
1390ec125fbbSEdward Tomasz Napierala 	char *inputstr, *buf;
1391ec125fbbSEdward Tomasz Napierala 	struct sbuf *sb;
1392ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *filter;
1393ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
1394ec125fbbSEdward Tomasz Napierala 	struct proc *p;
1395ec125fbbSEdward Tomasz Napierala 
1396ec125fbbSEdward Tomasz Napierala 	error = priv_check(td, PRIV_RCTL_GET_RULES);
1397ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1398ec125fbbSEdward Tomasz Napierala 		return (error);
1399ec125fbbSEdward Tomasz Napierala 
1400ec125fbbSEdward Tomasz Napierala 	error = rctl_read_inbuf(&inputstr, uap->inbufp, uap->inbuflen);
1401ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1402ec125fbbSEdward Tomasz Napierala 		return (error);
1403ec125fbbSEdward Tomasz Napierala 
1404ec125fbbSEdward Tomasz Napierala 	sx_slock(&allproc_lock);
1405ec125fbbSEdward Tomasz Napierala 	error = rctl_string_to_rule(inputstr, &filter);
1406ec125fbbSEdward Tomasz Napierala 	free(inputstr, M_RCTL);
1407ec125fbbSEdward Tomasz Napierala 	if (error != 0) {
1408ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1409ec125fbbSEdward Tomasz Napierala 		return (error);
1410ec125fbbSEdward Tomasz Napierala 	}
1411ec125fbbSEdward Tomasz Napierala 
1412ec125fbbSEdward Tomasz Napierala again:
1413ec125fbbSEdward Tomasz Napierala 	buf = malloc(bufsize, M_RCTL, M_WAITOK);
1414ec125fbbSEdward Tomasz Napierala 	sb = sbuf_new(NULL, buf, bufsize, SBUF_FIXEDLEN);
1415ec125fbbSEdward Tomasz Napierala 	KASSERT(sb != NULL, ("sbuf_new failed"));
1416ec125fbbSEdward Tomasz Napierala 
1417ec125fbbSEdward Tomasz Napierala 	sx_assert(&allproc_lock, SA_LOCKED);
1418ec125fbbSEdward Tomasz Napierala 	FOREACH_PROC_IN_SYSTEM(p) {
1419ec125fbbSEdward Tomasz Napierala 		rw_rlock(&rctl_lock);
1420ec125fbbSEdward Tomasz Napierala 		LIST_FOREACH(link, &p->p_racct->r_rule_links, rrl_next) {
1421ec125fbbSEdward Tomasz Napierala 			/*
1422ec125fbbSEdward Tomasz Napierala 			 * Non-process rules will be added to the buffer later.
1423ec125fbbSEdward Tomasz Napierala 			 * Adding them here would result in duplicated output.
1424ec125fbbSEdward Tomasz Napierala 			 */
1425ec125fbbSEdward Tomasz Napierala 			if (link->rrl_rule->rr_subject_type !=
1426ec125fbbSEdward Tomasz Napierala 			    RCTL_SUBJECT_TYPE_PROCESS)
1427ec125fbbSEdward Tomasz Napierala 				continue;
1428ec125fbbSEdward Tomasz Napierala 			if (!rctl_rule_matches(link->rrl_rule, filter))
1429ec125fbbSEdward Tomasz Napierala 				continue;
1430ec125fbbSEdward Tomasz Napierala 			rctl_rule_to_sbuf(sb, link->rrl_rule);
1431ec125fbbSEdward Tomasz Napierala 			sbuf_printf(sb, ",");
1432ec125fbbSEdward Tomasz Napierala 		}
1433ec125fbbSEdward Tomasz Napierala 		rw_runlock(&rctl_lock);
1434ec125fbbSEdward Tomasz Napierala 	}
1435ec125fbbSEdward Tomasz Napierala 
1436ec125fbbSEdward Tomasz Napierala 	loginclass_racct_foreach(rctl_get_rules_callback, filter, sb);
1437ec125fbbSEdward Tomasz Napierala 	ui_racct_foreach(rctl_get_rules_callback, filter, sb);
1438ec125fbbSEdward Tomasz Napierala 	prison_racct_foreach(rctl_get_rules_callback, filter, sb);
1439ec125fbbSEdward Tomasz Napierala 	if (sbuf_error(sb) == ENOMEM) {
1440ec125fbbSEdward Tomasz Napierala 		sbuf_delete(sb);
1441ec125fbbSEdward Tomasz Napierala 		free(buf, M_RCTL);
1442ec125fbbSEdward Tomasz Napierala 		bufsize *= 4;
1443ec125fbbSEdward Tomasz Napierala 		goto again;
1444ec125fbbSEdward Tomasz Napierala 	}
1445ec125fbbSEdward Tomasz Napierala 
1446ec125fbbSEdward Tomasz Napierala 	/*
1447ec125fbbSEdward Tomasz Napierala 	 * Remove trailing ",".
1448ec125fbbSEdward Tomasz Napierala 	 */
1449ec125fbbSEdward Tomasz Napierala 	if (sbuf_len(sb) > 0)
1450ec125fbbSEdward Tomasz Napierala 		sbuf_setpos(sb, sbuf_len(sb) - 1);
1451ec125fbbSEdward Tomasz Napierala 
1452ec125fbbSEdward Tomasz Napierala 	error = rctl_write_outbuf(sb, uap->outbufp, uap->outbuflen);
1453ec125fbbSEdward Tomasz Napierala 
1454ec125fbbSEdward Tomasz Napierala 	rctl_rule_release(filter);
1455ec125fbbSEdward Tomasz Napierala 	sx_sunlock(&allproc_lock);
1456ec125fbbSEdward Tomasz Napierala 	free(buf, M_RCTL);
1457ec125fbbSEdward Tomasz Napierala 	return (error);
1458ec125fbbSEdward Tomasz Napierala }
1459ec125fbbSEdward Tomasz Napierala 
1460ec125fbbSEdward Tomasz Napierala int
14618451d0ddSKip Macy sys_rctl_get_limits(struct thread *td, struct rctl_get_limits_args *uap)
1462ec125fbbSEdward Tomasz Napierala {
1463ec125fbbSEdward Tomasz Napierala 	int error;
1464ec125fbbSEdward Tomasz Napierala 	size_t bufsize = RCTL_DEFAULT_BUFSIZE;
1465ec125fbbSEdward Tomasz Napierala 	char *inputstr, *buf;
1466ec125fbbSEdward Tomasz Napierala 	struct sbuf *sb;
1467ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *filter;
1468ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
1469ec125fbbSEdward Tomasz Napierala 
1470ec125fbbSEdward Tomasz Napierala 	error = priv_check(td, PRIV_RCTL_GET_LIMITS);
1471ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1472ec125fbbSEdward Tomasz Napierala 		return (error);
1473ec125fbbSEdward Tomasz Napierala 
1474ec125fbbSEdward Tomasz Napierala 	error = rctl_read_inbuf(&inputstr, uap->inbufp, uap->inbuflen);
1475ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1476ec125fbbSEdward Tomasz Napierala 		return (error);
1477ec125fbbSEdward Tomasz Napierala 
1478ec125fbbSEdward Tomasz Napierala 	sx_slock(&allproc_lock);
1479ec125fbbSEdward Tomasz Napierala 	error = rctl_string_to_rule(inputstr, &filter);
1480ec125fbbSEdward Tomasz Napierala 	free(inputstr, M_RCTL);
1481ec125fbbSEdward Tomasz Napierala 	if (error != 0) {
1482ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1483ec125fbbSEdward Tomasz Napierala 		return (error);
1484ec125fbbSEdward Tomasz Napierala 	}
1485ec125fbbSEdward Tomasz Napierala 
1486ec125fbbSEdward Tomasz Napierala 	if (filter->rr_subject_type == RCTL_SUBJECT_TYPE_UNDEFINED) {
1487ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(filter);
1488ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1489ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
1490ec125fbbSEdward Tomasz Napierala 	}
1491ec125fbbSEdward Tomasz Napierala 	if (filter->rr_subject_type != RCTL_SUBJECT_TYPE_PROCESS) {
1492ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(filter);
1493ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1494ec125fbbSEdward Tomasz Napierala 		return (EOPNOTSUPP);
1495ec125fbbSEdward Tomasz Napierala 	}
1496ec125fbbSEdward Tomasz Napierala 	if (filter->rr_subject.rs_proc == NULL) {
1497ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(filter);
1498ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1499ec125fbbSEdward Tomasz Napierala 		return (EINVAL);
1500ec125fbbSEdward Tomasz Napierala 	}
1501ec125fbbSEdward Tomasz Napierala 
1502ec125fbbSEdward Tomasz Napierala again:
1503ec125fbbSEdward Tomasz Napierala 	buf = malloc(bufsize, M_RCTL, M_WAITOK);
1504ec125fbbSEdward Tomasz Napierala 	sb = sbuf_new(NULL, buf, bufsize, SBUF_FIXEDLEN);
1505ec125fbbSEdward Tomasz Napierala 	KASSERT(sb != NULL, ("sbuf_new failed"));
1506ec125fbbSEdward Tomasz Napierala 
1507ec125fbbSEdward Tomasz Napierala 	rw_rlock(&rctl_lock);
1508ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &filter->rr_subject.rs_proc->p_racct->r_rule_links,
1509ec125fbbSEdward Tomasz Napierala 	    rrl_next) {
1510ec125fbbSEdward Tomasz Napierala 		rctl_rule_to_sbuf(sb, link->rrl_rule);
1511ec125fbbSEdward Tomasz Napierala 		sbuf_printf(sb, ",");
1512ec125fbbSEdward Tomasz Napierala 	}
1513ec125fbbSEdward Tomasz Napierala 	rw_runlock(&rctl_lock);
1514ec125fbbSEdward Tomasz Napierala 	if (sbuf_error(sb) == ENOMEM) {
1515ec125fbbSEdward Tomasz Napierala 		sbuf_delete(sb);
1516ec125fbbSEdward Tomasz Napierala 		free(buf, M_RCTL);
1517ec125fbbSEdward Tomasz Napierala 		bufsize *= 4;
1518ec125fbbSEdward Tomasz Napierala 		goto again;
1519ec125fbbSEdward Tomasz Napierala 	}
1520ec125fbbSEdward Tomasz Napierala 
1521ec125fbbSEdward Tomasz Napierala 	/*
1522ec125fbbSEdward Tomasz Napierala 	 * Remove trailing ",".
1523ec125fbbSEdward Tomasz Napierala 	 */
1524ec125fbbSEdward Tomasz Napierala 	if (sbuf_len(sb) > 0)
1525ec125fbbSEdward Tomasz Napierala 		sbuf_setpos(sb, sbuf_len(sb) - 1);
1526ec125fbbSEdward Tomasz Napierala 
1527ec125fbbSEdward Tomasz Napierala 	error = rctl_write_outbuf(sb, uap->outbufp, uap->outbuflen);
1528ec125fbbSEdward Tomasz Napierala 	rctl_rule_release(filter);
1529ec125fbbSEdward Tomasz Napierala 	sx_sunlock(&allproc_lock);
1530ec125fbbSEdward Tomasz Napierala 	free(buf, M_RCTL);
1531ec125fbbSEdward Tomasz Napierala 	return (error);
1532ec125fbbSEdward Tomasz Napierala }
1533ec125fbbSEdward Tomasz Napierala 
1534ec125fbbSEdward Tomasz Napierala int
15358451d0ddSKip Macy sys_rctl_add_rule(struct thread *td, struct rctl_add_rule_args *uap)
1536ec125fbbSEdward Tomasz Napierala {
1537ec125fbbSEdward Tomasz Napierala 	int error;
1538ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
1539ec125fbbSEdward Tomasz Napierala 	char *inputstr;
1540ec125fbbSEdward Tomasz Napierala 
1541ec125fbbSEdward Tomasz Napierala 	error = priv_check(td, PRIV_RCTL_ADD_RULE);
1542ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1543ec125fbbSEdward Tomasz Napierala 		return (error);
1544ec125fbbSEdward Tomasz Napierala 
1545ec125fbbSEdward Tomasz Napierala 	error = rctl_read_inbuf(&inputstr, uap->inbufp, uap->inbuflen);
1546ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1547ec125fbbSEdward Tomasz Napierala 		return (error);
1548ec125fbbSEdward Tomasz Napierala 
1549ec125fbbSEdward Tomasz Napierala 	sx_slock(&allproc_lock);
1550ec125fbbSEdward Tomasz Napierala 	error = rctl_string_to_rule(inputstr, &rule);
1551ec125fbbSEdward Tomasz Napierala 	free(inputstr, M_RCTL);
1552ec125fbbSEdward Tomasz Napierala 	if (error != 0) {
1553ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1554ec125fbbSEdward Tomasz Napierala 		return (error);
1555ec125fbbSEdward Tomasz Napierala 	}
1556ec125fbbSEdward Tomasz Napierala 	/*
1557ec125fbbSEdward Tomasz Napierala 	 * The 'per' part of a rule is optional.
1558ec125fbbSEdward Tomasz Napierala 	 */
1559ec125fbbSEdward Tomasz Napierala 	if (rule->rr_per == RCTL_SUBJECT_TYPE_UNDEFINED &&
1560ec125fbbSEdward Tomasz Napierala 	    rule->rr_subject_type != RCTL_SUBJECT_TYPE_UNDEFINED)
1561ec125fbbSEdward Tomasz Napierala 		rule->rr_per = rule->rr_subject_type;
1562ec125fbbSEdward Tomasz Napierala 
1563ec125fbbSEdward Tomasz Napierala 	if (!rctl_rule_fully_specified(rule)) {
1564ec125fbbSEdward Tomasz Napierala 		error = EINVAL;
1565ec125fbbSEdward Tomasz Napierala 		goto out;
1566ec125fbbSEdward Tomasz Napierala 	}
1567ec125fbbSEdward Tomasz Napierala 
1568ec125fbbSEdward Tomasz Napierala 	error = rctl_rule_add(rule);
1569ec125fbbSEdward Tomasz Napierala 
1570ec125fbbSEdward Tomasz Napierala out:
1571ec125fbbSEdward Tomasz Napierala 	rctl_rule_release(rule);
1572ec125fbbSEdward Tomasz Napierala 	sx_sunlock(&allproc_lock);
1573ec125fbbSEdward Tomasz Napierala 	return (error);
1574ec125fbbSEdward Tomasz Napierala }
1575ec125fbbSEdward Tomasz Napierala 
1576ec125fbbSEdward Tomasz Napierala int
15778451d0ddSKip Macy sys_rctl_remove_rule(struct thread *td, struct rctl_remove_rule_args *uap)
1578ec125fbbSEdward Tomasz Napierala {
1579ec125fbbSEdward Tomasz Napierala 	int error;
1580ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *filter;
1581ec125fbbSEdward Tomasz Napierala 	char *inputstr;
1582ec125fbbSEdward Tomasz Napierala 
1583ec125fbbSEdward Tomasz Napierala 	error = priv_check(td, PRIV_RCTL_REMOVE_RULE);
1584ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1585ec125fbbSEdward Tomasz Napierala 		return (error);
1586ec125fbbSEdward Tomasz Napierala 
1587ec125fbbSEdward Tomasz Napierala 	error = rctl_read_inbuf(&inputstr, uap->inbufp, uap->inbuflen);
1588ec125fbbSEdward Tomasz Napierala 	if (error != 0)
1589ec125fbbSEdward Tomasz Napierala 		return (error);
1590ec125fbbSEdward Tomasz Napierala 
1591ec125fbbSEdward Tomasz Napierala 	sx_slock(&allproc_lock);
1592ec125fbbSEdward Tomasz Napierala 	error = rctl_string_to_rule(inputstr, &filter);
1593ec125fbbSEdward Tomasz Napierala 	free(inputstr, M_RCTL);
1594ec125fbbSEdward Tomasz Napierala 	if (error != 0) {
1595ec125fbbSEdward Tomasz Napierala 		sx_sunlock(&allproc_lock);
1596ec125fbbSEdward Tomasz Napierala 		return (error);
1597ec125fbbSEdward Tomasz Napierala 	}
1598ec125fbbSEdward Tomasz Napierala 
1599ec125fbbSEdward Tomasz Napierala 	error = rctl_rule_remove(filter);
1600ec125fbbSEdward Tomasz Napierala 	rctl_rule_release(filter);
1601ec125fbbSEdward Tomasz Napierala 	sx_sunlock(&allproc_lock);
1602ec125fbbSEdward Tomasz Napierala 
1603ec125fbbSEdward Tomasz Napierala 	return (error);
1604ec125fbbSEdward Tomasz Napierala }
1605ec125fbbSEdward Tomasz Napierala 
1606ec125fbbSEdward Tomasz Napierala /*
1607ec125fbbSEdward Tomasz Napierala  * Update RCTL rule list after credential change.
1608ec125fbbSEdward Tomasz Napierala  */
1609ec125fbbSEdward Tomasz Napierala void
1610ec125fbbSEdward Tomasz Napierala rctl_proc_ucred_changed(struct proc *p, struct ucred *newcred)
1611ec125fbbSEdward Tomasz Napierala {
1612ec125fbbSEdward Tomasz Napierala 	int rulecnt, i;
1613ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link, *newlink;
1614ec125fbbSEdward Tomasz Napierala 	struct uidinfo *newuip;
1615ec125fbbSEdward Tomasz Napierala 	struct loginclass *newlc;
1616a7ad07bfSEdward Tomasz Napierala 	struct prison_racct *newprr;
1617ec125fbbSEdward Tomasz Napierala 	LIST_HEAD(, rctl_rule_link) newrules;
1618ec125fbbSEdward Tomasz Napierala 
1619ec125fbbSEdward Tomasz Napierala 	newuip = newcred->cr_ruidinfo;
1620ec125fbbSEdward Tomasz Napierala 	newlc = newcred->cr_loginclass;
1621a7ad07bfSEdward Tomasz Napierala 	newprr = newcred->cr_prison->pr_prison_racct;
1622ec125fbbSEdward Tomasz Napierala 
1623ec125fbbSEdward Tomasz Napierala 	LIST_INIT(&newrules);
1624ec125fbbSEdward Tomasz Napierala 
1625ec125fbbSEdward Tomasz Napierala again:
1626ec125fbbSEdward Tomasz Napierala 	/*
1627ec125fbbSEdward Tomasz Napierala 	 * First, count the rules that apply to the process with new
1628ec125fbbSEdward Tomasz Napierala 	 * credentials.
1629ec125fbbSEdward Tomasz Napierala 	 */
1630ec125fbbSEdward Tomasz Napierala 	rulecnt = 0;
1631ec125fbbSEdward Tomasz Napierala 	rw_rlock(&rctl_lock);
1632ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &p->p_racct->r_rule_links, rrl_next) {
1633ec125fbbSEdward Tomasz Napierala 		if (link->rrl_rule->rr_subject_type ==
1634ec125fbbSEdward Tomasz Napierala 		    RCTL_SUBJECT_TYPE_PROCESS)
1635ec125fbbSEdward Tomasz Napierala 			rulecnt++;
1636ec125fbbSEdward Tomasz Napierala 	}
1637ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &newuip->ui_racct->r_rule_links, rrl_next)
1638ec125fbbSEdward Tomasz Napierala 		rulecnt++;
1639ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &newlc->lc_racct->r_rule_links, rrl_next)
1640ec125fbbSEdward Tomasz Napierala 		rulecnt++;
1641a7ad07bfSEdward Tomasz Napierala 	LIST_FOREACH(link, &newprr->prr_racct->r_rule_links, rrl_next)
1642ec125fbbSEdward Tomasz Napierala 		rulecnt++;
1643ec125fbbSEdward Tomasz Napierala 	rw_runlock(&rctl_lock);
1644ec125fbbSEdward Tomasz Napierala 
1645ec125fbbSEdward Tomasz Napierala 	/*
1646ec125fbbSEdward Tomasz Napierala 	 * Create temporary list.  We've dropped the rctl_lock in order
1647ec125fbbSEdward Tomasz Napierala 	 * to use M_WAITOK.
1648ec125fbbSEdward Tomasz Napierala 	 */
1649ec125fbbSEdward Tomasz Napierala 	for (i = 0; i < rulecnt; i++) {
1650ec125fbbSEdward Tomasz Napierala 		newlink = uma_zalloc(rctl_rule_link_zone, M_WAITOK);
1651ec125fbbSEdward Tomasz Napierala 		newlink->rrl_rule = NULL;
1652ec125fbbSEdward Tomasz Napierala 		LIST_INSERT_HEAD(&newrules, newlink, rrl_next);
1653ec125fbbSEdward Tomasz Napierala 	}
1654ec125fbbSEdward Tomasz Napierala 
1655ec125fbbSEdward Tomasz Napierala 	newlink = LIST_FIRST(&newrules);
1656ec125fbbSEdward Tomasz Napierala 
1657ec125fbbSEdward Tomasz Napierala 	/*
1658ec125fbbSEdward Tomasz Napierala 	 * Assign rules to the newly allocated list entries.
1659ec125fbbSEdward Tomasz Napierala 	 */
1660ec125fbbSEdward Tomasz Napierala 	rw_wlock(&rctl_lock);
1661ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &p->p_racct->r_rule_links, rrl_next) {
1662ec125fbbSEdward Tomasz Napierala 		if (link->rrl_rule->rr_subject_type ==
1663ec125fbbSEdward Tomasz Napierala 		    RCTL_SUBJECT_TYPE_PROCESS) {
1664ec125fbbSEdward Tomasz Napierala 			if (newlink == NULL)
1665ec125fbbSEdward Tomasz Napierala 				goto goaround;
1666ec125fbbSEdward Tomasz Napierala 			rctl_rule_acquire(link->rrl_rule);
1667ec125fbbSEdward Tomasz Napierala 			newlink->rrl_rule = link->rrl_rule;
1668ec125fbbSEdward Tomasz Napierala 			newlink = LIST_NEXT(newlink, rrl_next);
1669ec125fbbSEdward Tomasz Napierala 			rulecnt--;
1670ec125fbbSEdward Tomasz Napierala 		}
1671ec125fbbSEdward Tomasz Napierala 	}
1672ec125fbbSEdward Tomasz Napierala 
1673ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &newuip->ui_racct->r_rule_links, rrl_next) {
1674ec125fbbSEdward Tomasz Napierala 		if (newlink == NULL)
1675ec125fbbSEdward Tomasz Napierala 			goto goaround;
1676ec125fbbSEdward Tomasz Napierala 		rctl_rule_acquire(link->rrl_rule);
1677ec125fbbSEdward Tomasz Napierala 		newlink->rrl_rule = link->rrl_rule;
1678ec125fbbSEdward Tomasz Napierala 		newlink = LIST_NEXT(newlink, rrl_next);
1679ec125fbbSEdward Tomasz Napierala 		rulecnt--;
1680ec125fbbSEdward Tomasz Napierala 	}
1681ec125fbbSEdward Tomasz Napierala 
1682ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &newlc->lc_racct->r_rule_links, rrl_next) {
1683ec125fbbSEdward Tomasz Napierala 		if (newlink == NULL)
1684ec125fbbSEdward Tomasz Napierala 			goto goaround;
1685ec125fbbSEdward Tomasz Napierala 		rctl_rule_acquire(link->rrl_rule);
1686ec125fbbSEdward Tomasz Napierala 		newlink->rrl_rule = link->rrl_rule;
1687ec125fbbSEdward Tomasz Napierala 		newlink = LIST_NEXT(newlink, rrl_next);
1688ec125fbbSEdward Tomasz Napierala 		rulecnt--;
1689ec125fbbSEdward Tomasz Napierala 	}
1690ec125fbbSEdward Tomasz Napierala 
1691a7ad07bfSEdward Tomasz Napierala 	LIST_FOREACH(link, &newprr->prr_racct->r_rule_links, rrl_next) {
1692ec125fbbSEdward Tomasz Napierala 		if (newlink == NULL)
1693ec125fbbSEdward Tomasz Napierala 			goto goaround;
1694ec125fbbSEdward Tomasz Napierala 		rctl_rule_acquire(link->rrl_rule);
1695ec125fbbSEdward Tomasz Napierala 		newlink->rrl_rule = link->rrl_rule;
1696ec125fbbSEdward Tomasz Napierala 		newlink = LIST_NEXT(newlink, rrl_next);
1697ec125fbbSEdward Tomasz Napierala 		rulecnt--;
1698ec125fbbSEdward Tomasz Napierala 	}
1699ec125fbbSEdward Tomasz Napierala 
1700ec125fbbSEdward Tomasz Napierala 	if (rulecnt == 0) {
1701ec125fbbSEdward Tomasz Napierala 		/*
1702ec125fbbSEdward Tomasz Napierala 		 * Free the old rule list.
1703ec125fbbSEdward Tomasz Napierala 		 */
1704ec125fbbSEdward Tomasz Napierala 		while (!LIST_EMPTY(&p->p_racct->r_rule_links)) {
1705ec125fbbSEdward Tomasz Napierala 			link = LIST_FIRST(&p->p_racct->r_rule_links);
1706ec125fbbSEdward Tomasz Napierala 			LIST_REMOVE(link, rrl_next);
1707ec125fbbSEdward Tomasz Napierala 			rctl_rule_release(link->rrl_rule);
1708ec125fbbSEdward Tomasz Napierala 			uma_zfree(rctl_rule_link_zone, link);
1709ec125fbbSEdward Tomasz Napierala 		}
1710ec125fbbSEdward Tomasz Napierala 
1711ec125fbbSEdward Tomasz Napierala 		/*
1712ec125fbbSEdward Tomasz Napierala 		 * Replace lists and we're done.
1713ec125fbbSEdward Tomasz Napierala 		 *
1714ec125fbbSEdward Tomasz Napierala 		 * XXX: Is there any way to switch list heads instead
1715ec125fbbSEdward Tomasz Napierala 		 *      of iterating here?
1716ec125fbbSEdward Tomasz Napierala 		 */
1717ec125fbbSEdward Tomasz Napierala 		while (!LIST_EMPTY(&newrules)) {
1718ec125fbbSEdward Tomasz Napierala 			newlink = LIST_FIRST(&newrules);
1719ec125fbbSEdward Tomasz Napierala 			LIST_REMOVE(newlink, rrl_next);
1720ec125fbbSEdward Tomasz Napierala 			LIST_INSERT_HEAD(&p->p_racct->r_rule_links,
1721ec125fbbSEdward Tomasz Napierala 			    newlink, rrl_next);
1722ec125fbbSEdward Tomasz Napierala 		}
1723ec125fbbSEdward Tomasz Napierala 
1724ec125fbbSEdward Tomasz Napierala 		rw_wunlock(&rctl_lock);
1725ec125fbbSEdward Tomasz Napierala 
1726ec125fbbSEdward Tomasz Napierala 		return;
1727ec125fbbSEdward Tomasz Napierala 	}
1728ec125fbbSEdward Tomasz Napierala 
1729ec125fbbSEdward Tomasz Napierala goaround:
1730ec125fbbSEdward Tomasz Napierala 	rw_wunlock(&rctl_lock);
1731ec125fbbSEdward Tomasz Napierala 
1732ec125fbbSEdward Tomasz Napierala 	/*
1733ec125fbbSEdward Tomasz Napierala 	 * Rule list changed while we were not holding the rctl_lock.
1734ec125fbbSEdward Tomasz Napierala 	 * Free the new list and try again.
1735ec125fbbSEdward Tomasz Napierala 	 */
1736ec125fbbSEdward Tomasz Napierala 	while (!LIST_EMPTY(&newrules)) {
1737ec125fbbSEdward Tomasz Napierala 		newlink = LIST_FIRST(&newrules);
1738ec125fbbSEdward Tomasz Napierala 		LIST_REMOVE(newlink, rrl_next);
1739ec125fbbSEdward Tomasz Napierala 		if (newlink->rrl_rule != NULL)
1740ec125fbbSEdward Tomasz Napierala 			rctl_rule_release(newlink->rrl_rule);
1741ec125fbbSEdward Tomasz Napierala 		uma_zfree(rctl_rule_link_zone, newlink);
1742ec125fbbSEdward Tomasz Napierala 	}
1743ec125fbbSEdward Tomasz Napierala 
1744ec125fbbSEdward Tomasz Napierala 	goto again;
1745ec125fbbSEdward Tomasz Napierala }
1746ec125fbbSEdward Tomasz Napierala 
1747ec125fbbSEdward Tomasz Napierala /*
1748ec125fbbSEdward Tomasz Napierala  * Assign RCTL rules to the newly created process.
1749ec125fbbSEdward Tomasz Napierala  */
1750ec125fbbSEdward Tomasz Napierala int
1751ec125fbbSEdward Tomasz Napierala rctl_proc_fork(struct proc *parent, struct proc *child)
1752ec125fbbSEdward Tomasz Napierala {
1753ec125fbbSEdward Tomasz Napierala 	int error;
1754ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
1755ec125fbbSEdward Tomasz Napierala 	struct rctl_rule *rule;
1756ec125fbbSEdward Tomasz Napierala 
1757ec125fbbSEdward Tomasz Napierala 	LIST_INIT(&child->p_racct->r_rule_links);
1758ec125fbbSEdward Tomasz Napierala 
17590b18eb6dSEdward Tomasz Napierala 	KASSERT(parent->p_racct != NULL, ("process without racct; p = %p", parent));
1760ec125fbbSEdward Tomasz Napierala 
1761ec125fbbSEdward Tomasz Napierala 	rw_wlock(&rctl_lock);
1762ec125fbbSEdward Tomasz Napierala 
1763ec125fbbSEdward Tomasz Napierala 	/*
1764ec125fbbSEdward Tomasz Napierala 	 * Go through limits applicable to the parent and assign them
1765ec125fbbSEdward Tomasz Napierala 	 * to the child.  Rules with 'process' subject have to be duplicated
1766ec125fbbSEdward Tomasz Napierala 	 * in order to make their rr_subject point to the new process.
1767ec125fbbSEdward Tomasz Napierala 	 */
1768ec125fbbSEdward Tomasz Napierala 	LIST_FOREACH(link, &parent->p_racct->r_rule_links, rrl_next) {
1769ec125fbbSEdward Tomasz Napierala 		if (link->rrl_rule->rr_subject_type ==
1770ec125fbbSEdward Tomasz Napierala 		    RCTL_SUBJECT_TYPE_PROCESS) {
1771ec125fbbSEdward Tomasz Napierala 			rule = rctl_rule_duplicate(link->rrl_rule, M_NOWAIT);
1772ec125fbbSEdward Tomasz Napierala 			if (rule == NULL)
1773ec125fbbSEdward Tomasz Napierala 				goto fail;
1774ec125fbbSEdward Tomasz Napierala 			KASSERT(rule->rr_subject.rs_proc == parent,
1775ec125fbbSEdward Tomasz Napierala 			    ("rule->rr_subject.rs_proc != parent"));
1776ec125fbbSEdward Tomasz Napierala 			rule->rr_subject.rs_proc = child;
1777ec125fbbSEdward Tomasz Napierala 			error = rctl_racct_add_rule_locked(child->p_racct,
1778ec125fbbSEdward Tomasz Napierala 			    rule);
1779ec125fbbSEdward Tomasz Napierala 			rctl_rule_release(rule);
1780ec125fbbSEdward Tomasz Napierala 			if (error != 0)
1781ec125fbbSEdward Tomasz Napierala 				goto fail;
1782ec125fbbSEdward Tomasz Napierala 		} else {
1783ec125fbbSEdward Tomasz Napierala 			error = rctl_racct_add_rule_locked(child->p_racct,
1784ec125fbbSEdward Tomasz Napierala 			    link->rrl_rule);
1785ec125fbbSEdward Tomasz Napierala 			if (error != 0)
1786ec125fbbSEdward Tomasz Napierala 				goto fail;
1787ec125fbbSEdward Tomasz Napierala 		}
1788ec125fbbSEdward Tomasz Napierala 	}
1789ec125fbbSEdward Tomasz Napierala 
1790ec125fbbSEdward Tomasz Napierala 	rw_wunlock(&rctl_lock);
1791ec125fbbSEdward Tomasz Napierala 	return (0);
1792ec125fbbSEdward Tomasz Napierala 
1793ec125fbbSEdward Tomasz Napierala fail:
1794ec125fbbSEdward Tomasz Napierala 	while (!LIST_EMPTY(&child->p_racct->r_rule_links)) {
1795ec125fbbSEdward Tomasz Napierala 		link = LIST_FIRST(&child->p_racct->r_rule_links);
1796ec125fbbSEdward Tomasz Napierala 		LIST_REMOVE(link, rrl_next);
1797ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(link->rrl_rule);
1798ec125fbbSEdward Tomasz Napierala 		uma_zfree(rctl_rule_link_zone, link);
1799ec125fbbSEdward Tomasz Napierala 	}
1800ec125fbbSEdward Tomasz Napierala 	rw_wunlock(&rctl_lock);
1801ec125fbbSEdward Tomasz Napierala 	return (EAGAIN);
1802ec125fbbSEdward Tomasz Napierala }
1803ec125fbbSEdward Tomasz Napierala 
1804ec125fbbSEdward Tomasz Napierala /*
1805ec125fbbSEdward Tomasz Napierala  * Release rules attached to the racct.
1806ec125fbbSEdward Tomasz Napierala  */
1807ec125fbbSEdward Tomasz Napierala void
1808ec125fbbSEdward Tomasz Napierala rctl_racct_release(struct racct *racct)
1809ec125fbbSEdward Tomasz Napierala {
1810ec125fbbSEdward Tomasz Napierala 	struct rctl_rule_link *link;
1811ec125fbbSEdward Tomasz Napierala 
1812ec125fbbSEdward Tomasz Napierala 	rw_wlock(&rctl_lock);
1813ec125fbbSEdward Tomasz Napierala 	while (!LIST_EMPTY(&racct->r_rule_links)) {
1814ec125fbbSEdward Tomasz Napierala 		link = LIST_FIRST(&racct->r_rule_links);
1815ec125fbbSEdward Tomasz Napierala 		LIST_REMOVE(link, rrl_next);
1816ec125fbbSEdward Tomasz Napierala 		rctl_rule_release(link->rrl_rule);
1817ec125fbbSEdward Tomasz Napierala 		uma_zfree(rctl_rule_link_zone, link);
1818ec125fbbSEdward Tomasz Napierala 	}
1819ec125fbbSEdward Tomasz Napierala 	rw_wunlock(&rctl_lock);
1820ec125fbbSEdward Tomasz Napierala }
1821ec125fbbSEdward Tomasz Napierala 
1822ec125fbbSEdward Tomasz Napierala static void
1823ec125fbbSEdward Tomasz Napierala rctl_init(void)
1824ec125fbbSEdward Tomasz Napierala {
1825ec125fbbSEdward Tomasz Napierala 
1826ec125fbbSEdward Tomasz Napierala 	rctl_rule_link_zone = uma_zcreate("rctl_rule_link",
1827ec125fbbSEdward Tomasz Napierala 	    sizeof(struct rctl_rule_link), NULL, NULL, NULL, NULL,
1828ec125fbbSEdward Tomasz Napierala 	    UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
1829ec125fbbSEdward Tomasz Napierala 	rctl_rule_zone = uma_zcreate("rctl_rule", sizeof(struct rctl_rule),
1830ec125fbbSEdward Tomasz Napierala 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
1831ec125fbbSEdward Tomasz Napierala }
1832ec125fbbSEdward Tomasz Napierala 
1833ec125fbbSEdward Tomasz Napierala #else /* !RCTL */
1834ec125fbbSEdward Tomasz Napierala 
1835ec125fbbSEdward Tomasz Napierala int
18368451d0ddSKip Macy sys_rctl_get_racct(struct thread *td, struct rctl_get_racct_args *uap)
1837ec125fbbSEdward Tomasz Napierala {
1838ec125fbbSEdward Tomasz Napierala 
1839ec125fbbSEdward Tomasz Napierala 	return (ENOSYS);
1840ec125fbbSEdward Tomasz Napierala }
1841ec125fbbSEdward Tomasz Napierala 
1842ec125fbbSEdward Tomasz Napierala int
18438451d0ddSKip Macy sys_rctl_get_rules(struct thread *td, struct rctl_get_rules_args *uap)
1844ec125fbbSEdward Tomasz Napierala {
1845ec125fbbSEdward Tomasz Napierala 
1846ec125fbbSEdward Tomasz Napierala 	return (ENOSYS);
1847ec125fbbSEdward Tomasz Napierala }
1848ec125fbbSEdward Tomasz Napierala 
1849ec125fbbSEdward Tomasz Napierala int
18508451d0ddSKip Macy sys_rctl_get_limits(struct thread *td, struct rctl_get_limits_args *uap)
1851ec125fbbSEdward Tomasz Napierala {
1852ec125fbbSEdward Tomasz Napierala 
1853ec125fbbSEdward Tomasz Napierala 	return (ENOSYS);
1854ec125fbbSEdward Tomasz Napierala }
1855ec125fbbSEdward Tomasz Napierala 
1856ec125fbbSEdward Tomasz Napierala int
18578451d0ddSKip Macy sys_rctl_add_rule(struct thread *td, struct rctl_add_rule_args *uap)
1858ec125fbbSEdward Tomasz Napierala {
1859ec125fbbSEdward Tomasz Napierala 
1860ec125fbbSEdward Tomasz Napierala 	return (ENOSYS);
1861ec125fbbSEdward Tomasz Napierala }
1862ec125fbbSEdward Tomasz Napierala 
1863ec125fbbSEdward Tomasz Napierala int
18648451d0ddSKip Macy sys_rctl_remove_rule(struct thread *td, struct rctl_remove_rule_args *uap)
1865ec125fbbSEdward Tomasz Napierala {
1866ec125fbbSEdward Tomasz Napierala 
1867ec125fbbSEdward Tomasz Napierala 	return (ENOSYS);
1868ec125fbbSEdward Tomasz Napierala }
1869ec125fbbSEdward Tomasz Napierala 
1870ec125fbbSEdward Tomasz Napierala #endif /* !RCTL */
1871