racct.h revision 284665
1/*-
2 * Copyright (c) 2010 The FreeBSD Foundation
3 * All rights reserved.
4 *
5 * This software was developed by Edward Tomasz Napierala under sponsorship
6 * from the FreeBSD Foundation.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 * $FreeBSD: stable/10/sys/sys/racct.h 284665 2015-06-21 06:28:26Z trasz $
30 */
31
32/*
33 * Resource accounting.
34 */
35
36#ifndef _RACCT_H_
37#define	_RACCT_H_
38
39#include <sys/cdefs.h>
40#include <sys/queue.h>
41#include <sys/types.h>
42
43struct proc;
44struct rctl_rule_link;
45struct ucred;
46
47/*
48 * Resources.
49 */
50#define	RACCT_UNDEFINED		-1
51#define	RACCT_CPU		0
52#define	RACCT_DATA		1
53#define	RACCT_STACK		2
54#define	RACCT_CORE		3
55#define	RACCT_RSS		4
56#define	RACCT_MEMLOCK		5
57#define	RACCT_NPROC		6
58#define	RACCT_NOFILE		7
59#define	RACCT_VMEM		8
60#define	RACCT_NPTS		9
61#define	RACCT_SWAP		10
62#define	RACCT_NTHR		11
63#define	RACCT_MSGQQUEUED	12
64#define	RACCT_MSGQSIZE		13
65#define	RACCT_NMSGQ		14
66#define	RACCT_NSEM		15
67#define	RACCT_NSEMOP		16
68#define	RACCT_NSHM		17
69#define	RACCT_SHMSIZE		18
70#define	RACCT_WALLCLOCK		19
71#define	RACCT_PCTCPU		20
72#define	RACCT_MAX		RACCT_PCTCPU
73
74/*
75 * Resource properties.
76 */
77#define	RACCT_IN_MILLIONS	0x01
78#define	RACCT_RECLAIMABLE	0x02
79#define	RACCT_INHERITABLE	0x04
80#define	RACCT_DENIABLE		0x08
81#define	RACCT_SLOPPY		0x10
82#define	RACCT_DECAYING		0x20
83
84extern int racct_types[];
85extern int racct_enable;
86
87#define ASSERT_RACCT_ENABLED()	KASSERT(racct_enable, \
88				    ("%s called with !racct_enable", __func__))
89
90/*
91 * Amount stored in c_resources[] is 10**6 times bigger than what's
92 * visible to the userland.  It gets fixed up when retrieving resource
93 * usage or adding rules.
94 */
95#define	RACCT_IS_IN_MILLIONS(X)	(racct_types[X] & RACCT_IN_MILLIONS)
96
97/*
98 * Resource usage can drop, as opposed to only grow.  When the process
99 * terminates, its resource usage is freed from the respective
100 * per-credential racct containers.
101 */
102#define	RACCT_IS_RECLAIMABLE(X)	(racct_types[X] & RACCT_RECLAIMABLE)
103
104/*
105 * Children inherit resource usage.
106 */
107#define	RACCT_IS_INHERITABLE(X)	(racct_types[X] & RACCT_INHERITABLE)
108
109/*
110 * racct_{add,set}(9) can actually return an error and not update resource
111 * usage counters.  Note that even when resource is not deniable, allocating
112 * resource might cause signals to be sent by RCTL code.
113 */
114#define	RACCT_IS_DENIABLE(X)		(racct_types[X] & RACCT_DENIABLE)
115
116/*
117 * Per-process resource usage information makes no sense, but per-credential
118 * one does.  This kind of resources are usually allocated for process, but
119 * freed using credentials.
120 */
121#define	RACCT_IS_SLOPPY(X)		(racct_types[X] & RACCT_SLOPPY)
122
123/*
124 * When a process terminates, its resource usage is not automatically
125 * subtracted from per-credential racct containers.  Instead, the resource
126 * usage of per-credential racct containers decays in time.
127 * Resource usage can olso drop for such resource.
128 * So far, the only such resource is RACCT_PCTCPU.
129 */
130#define RACCT_IS_DECAYING(X)		(racct_types[X] & RACCT_DECAYING)
131
132/*
133 * Resource usage can drop, as opposed to only grow.
134 */
135#define RACCT_CAN_DROP(X)		(RACCT_IS_RECLAIMABLE(X) | RACCT_IS_DECAYING(X))
136
137/*
138 * The 'racct' structure defines resource consumption for a particular
139 * subject, such as process or jail.
140 *
141 * This structure must be filled with zeroes initially.
142 */
143struct racct {
144	int64_t				r_resources[RACCT_MAX + 1];
145	LIST_HEAD(, rctl_rule_link)	r_rule_links;
146};
147
148int	racct_add(struct proc *p, int resource, uint64_t amount);
149void	racct_add_cred(struct ucred *cred, int resource, uint64_t amount);
150void	racct_add_force(struct proc *p, int resource, uint64_t amount);
151int	racct_set(struct proc *p, int resource, uint64_t amount);
152void	racct_set_force(struct proc *p, int resource, uint64_t amount);
153void	racct_sub(struct proc *p, int resource, uint64_t amount);
154void	racct_sub_cred(struct ucred *cred, int resource, uint64_t amount);
155uint64_t	racct_get_limit(struct proc *p, int resource);
156uint64_t	racct_get_available(struct proc *p, int resource);
157
158void	racct_create(struct racct **racctp);
159void	racct_destroy(struct racct **racctp);
160
161int	racct_proc_fork(struct proc *parent, struct proc *child);
162void	racct_proc_fork_done(struct proc *child);
163void	racct_proc_exit(struct proc *p);
164
165void	racct_proc_ucred_changed(struct proc *p, struct ucred *oldcred,
166	    struct ucred *newcred);
167void	racct_move(struct racct *dest, struct racct *src);
168
169#endif /* !_RACCT_H_ */
170