1/*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2010 The FreeBSD Foundation
5 *
6 * This software was developed by Edward Tomasz Napierala under sponsorship
7 * from the FreeBSD Foundation.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 * SUCH DAMAGE.
29 */
30
31/*
32 * Resource Limits.
33 */
34
35#ifndef _RCTL_H_
36#define	_RCTL_H_
37
38#include <sys/cdefs.h>
39#include <sys/queue.h>
40#include <sys/types.h>
41#include <sys/_task.h>
42
43struct proc;
44struct uidinfo;
45struct loginclass;
46struct prison_racct;
47struct ucred;
48struct rctl_rule_link;
49
50#ifdef _KERNEL
51
52/*
53 * Rules describe an action to be taken when conditions defined
54 * in the rule are met.  There is no global list of rules; instead,
55 * rules are linked to by the racct structures for all the subjects
56 * they apply to - for example, a rule of type "user" is linked to the
57 * appropriate struct uidinfo, and to all the processes of that user.
58 *
59 * 'rr_refcount' is equal to the number of rctl_rule_link structures
60 * pointing to the rule.
61 *
62 * This structure must never change after being added, via rctl_rule_link
63 * structures, to subjects.  In order to change a rule, add a new rule
64 * and remove the previous one.
65 */
66struct rctl_rule {
67	int		rr_subject_type;
68	union {
69		struct proc		*rs_proc;
70		struct uidinfo		*rs_uip;
71		struct loginclass	*rs_loginclass;
72		struct prison_racct	*rs_prison_racct;
73	} rr_subject;
74	int		rr_per;
75	int		rr_resource;
76	int		rr_action;
77	int64_t		rr_amount;
78	u_int		rr_refcount;
79	struct task	rr_task;
80};
81
82/*
83 * Allowed values for rr_subject_type and rr_per fields.
84 */
85#define	RCTL_SUBJECT_TYPE_UNDEFINED	-1
86#define	RCTL_SUBJECT_TYPE_PROCESS	0x0000
87#define	RCTL_SUBJECT_TYPE_USER		0x0001
88#define	RCTL_SUBJECT_TYPE_LOGINCLASS	0x0003
89#define	RCTL_SUBJECT_TYPE_JAIL		0x0004
90#define	RCTL_SUBJECT_TYPE_MAX		RCTL_SUBJECT_TYPE_JAIL
91
92/*
93 * Allowed values for rr_action field.
94 */
95#define	RCTL_ACTION_UNDEFINED		-1
96#define	RCTL_ACTION_SIGHUP		SIGHUP
97#define	RCTL_ACTION_SIGINT		SIGINT
98#define	RCTL_ACTION_SIGQUIT		SIGQUIT
99#define	RCTL_ACTION_SIGILL		SIGILL
100#define	RCTL_ACTION_SIGTRAP		SIGTRAP
101#define	RCTL_ACTION_SIGABRT		SIGABRT
102#define	RCTL_ACTION_SIGEMT		SIGEMT
103#define	RCTL_ACTION_SIGFPE		SIGFPE
104#define	RCTL_ACTION_SIGKILL		SIGKILL
105#define	RCTL_ACTION_SIGBUS		SIGBUS
106#define	RCTL_ACTION_SIGSEGV		SIGSEGV
107#define	RCTL_ACTION_SIGSYS		SIGSYS
108#define	RCTL_ACTION_SIGPIPE		SIGPIPE
109#define	RCTL_ACTION_SIGALRM		SIGALRM
110#define	RCTL_ACTION_SIGTERM		SIGTERM
111#define	RCTL_ACTION_SIGURG		SIGURG
112#define	RCTL_ACTION_SIGSTOP		SIGSTOP
113#define	RCTL_ACTION_SIGTSTP		SIGTSTP
114#define	RCTL_ACTION_SIGCHLD		SIGCHLD
115#define	RCTL_ACTION_SIGTTIN		SIGTTIN
116#define	RCTL_ACTION_SIGTTOU		SIGTTOU
117#define	RCTL_ACTION_SIGIO		SIGIO
118#define	RCTL_ACTION_SIGXCPU		SIGXCPU
119#define	RCTL_ACTION_SIGXFSZ		SIGXFSZ
120#define	RCTL_ACTION_SIGVTALRM		SIGVTALRM
121#define	RCTL_ACTION_SIGPROF		SIGPROF
122#define	RCTL_ACTION_SIGWINCH		SIGWINCH
123#define	RCTL_ACTION_SIGINFO		SIGINFO
124#define	RCTL_ACTION_SIGUSR1		SIGUSR1
125#define	RCTL_ACTION_SIGUSR2		SIGUSR2
126#define	RCTL_ACTION_SIGTHR		SIGTHR
127#define	RCTL_ACTION_SIGNAL_MAX		RCTL_ACTION_SIGTHR
128#define	RCTL_ACTION_DENY		(RCTL_ACTION_SIGNAL_MAX + 1)
129#define	RCTL_ACTION_LOG			(RCTL_ACTION_SIGNAL_MAX + 2)
130#define	RCTL_ACTION_DEVCTL		(RCTL_ACTION_SIGNAL_MAX + 3)
131#define	RCTL_ACTION_THROTTLE		(RCTL_ACTION_SIGNAL_MAX + 4)
132#define	RCTL_ACTION_MAX			RCTL_ACTION_THROTTLE
133
134#define	RCTL_AMOUNT_UNDEFINED		-1
135
136struct rctl_rule *rctl_rule_alloc(int flags);
137struct rctl_rule *rctl_rule_duplicate(const struct rctl_rule *rule, int flags);
138void	rctl_rule_acquire(struct rctl_rule *rule);
139void	rctl_rule_release(struct rctl_rule *rule);
140int	rctl_rule_add(struct rctl_rule *rule);
141int	rctl_rule_remove(struct rctl_rule *filter);
142int	rctl_enforce(struct proc *p, int resource, uint64_t amount);
143void	rctl_throttle_decay(struct racct *racct, int resource);
144int64_t	rctl_pcpu_available(const struct proc *p);
145uint64_t rctl_get_limit(struct proc *p, int resource);
146uint64_t rctl_get_available(struct proc *p, int resource);
147const char *rctl_resource_name(int resource);
148void	rctl_proc_ucred_changed(struct proc *p, struct ucred *newcred);
149int	rctl_proc_fork(struct proc *parent, struct proc *child);
150void	rctl_racct_release(struct racct *racct);
151#else /* !_KERNEL */
152
153/*
154 * Syscall interface.
155 */
156__BEGIN_DECLS
157int	rctl_get_racct(const char *inbufp, size_t inbuflen, char *outbufp,
158	    size_t outbuflen);
159int	rctl_get_rules(const char *inbufp, size_t inbuflen, char *outbufp,
160	    size_t outbuflen);
161int	rctl_get_limits(const char *inbufp, size_t inbuflen, char *outbufp,
162	    size_t outbuflen);
163int	rctl_add_rule(const char *inbufp, size_t inbuflen, char *outbufp,
164	    size_t outbuflen);
165int	rctl_remove_rule(const char *inbufp, size_t inbuflen, char *outbufp,
166	    size_t outbuflen);
167__END_DECLS
168
169#endif /* !_KERNEL */
170
171#endif /* !_RCTL_H_ */
172