]> git.proxmox.com Git - mirror_ubuntu-eoan-kernel.git/blame - include/linux/bpf-cgroup.h
bpf: rework cgroup storage pointer passing
[mirror_ubuntu-eoan-kernel.git] / include / linux / bpf-cgroup.h
CommitLineData
b2441318 1/* SPDX-License-Identifier: GPL-2.0 */
30070984
DM
2#ifndef _BPF_CGROUP_H
3#define _BPF_CGROUP_H
4
8bad74f9 5#include <linux/bpf.h>
f292b87d 6#include <linux/errno.h>
30070984 7#include <linux/jump_label.h>
aa0ad5b0 8#include <linux/percpu.h>
de9cbbaa 9#include <linux/rbtree.h>
30070984
DM
10#include <uapi/linux/bpf.h>
11
12struct sock;
4fbac77d 13struct sockaddr;
30070984
DM
14struct cgroup;
15struct sk_buff;
de9cbbaa
RG
16struct bpf_map;
17struct bpf_prog;
40304b2a 18struct bpf_sock_ops_kern;
de9cbbaa 19struct bpf_cgroup_storage;
30070984
DM
20
21#ifdef CONFIG_CGROUP_BPF
22
23extern struct static_key_false cgroup_bpf_enabled_key;
24#define cgroup_bpf_enabled static_branch_unlikely(&cgroup_bpf_enabled_key)
25
f294b37e
RG
26DECLARE_PER_CPU(struct bpf_cgroup_storage*,
27 bpf_cgroup_storage[MAX_BPF_CGROUP_STORAGE_TYPE]);
8bad74f9
RG
28
29#define for_each_cgroup_storage_type(stype) \
30 for (stype = 0; stype < MAX_BPF_CGROUP_STORAGE_TYPE; stype++)
aa0ad5b0 31
de9cbbaa
RG
32struct bpf_cgroup_storage_map;
33
34struct bpf_storage_buffer {
35 struct rcu_head rcu;
36 char data[0];
37};
38
39struct bpf_cgroup_storage {
40 struct bpf_storage_buffer *buf;
41 struct bpf_cgroup_storage_map *map;
42 struct bpf_cgroup_storage_key key;
43 struct list_head list;
44 struct rb_node node;
45 struct rcu_head rcu;
46};
47
324bda9e
AS
48struct bpf_prog_list {
49 struct list_head node;
50 struct bpf_prog *prog;
8bad74f9 51 struct bpf_cgroup_storage *storage[MAX_BPF_CGROUP_STORAGE_TYPE];
324bda9e
AS
52};
53
54struct bpf_prog_array;
55
30070984 56struct cgroup_bpf {
324bda9e
AS
57 /* array of effective progs in this cgroup */
58 struct bpf_prog_array __rcu *effective[MAX_BPF_ATTACH_TYPE];
59
60 /* attached progs to this cgroup and attach flags
61 * when flags == 0 or BPF_F_ALLOW_OVERRIDE the progs list will
62 * have either zero or one element
63 * when BPF_F_ALLOW_MULTI the list can have up to BPF_CGROUP_MAX_PROGS
30070984 64 */
324bda9e
AS
65 struct list_head progs[MAX_BPF_ATTACH_TYPE];
66 u32 flags[MAX_BPF_ATTACH_TYPE];
67
68 /* temp storage for effective prog array used by prog_attach/detach */
69 struct bpf_prog_array __rcu *inactive;
30070984
DM
70};
71
72void cgroup_bpf_put(struct cgroup *cgrp);
324bda9e 73int cgroup_bpf_inherit(struct cgroup *cgrp);
30070984 74
324bda9e
AS
75int __cgroup_bpf_attach(struct cgroup *cgrp, struct bpf_prog *prog,
76 enum bpf_attach_type type, u32 flags);
77int __cgroup_bpf_detach(struct cgroup *cgrp, struct bpf_prog *prog,
78 enum bpf_attach_type type, u32 flags);
468e2f64
AS
79int __cgroup_bpf_query(struct cgroup *cgrp, const union bpf_attr *attr,
80 union bpf_attr __user *uattr);
30070984 81
324bda9e
AS
82/* Wrapper for __cgroup_bpf_*() protected by cgroup_mutex */
83int cgroup_bpf_attach(struct cgroup *cgrp, struct bpf_prog *prog,
84 enum bpf_attach_type type, u32 flags);
85int cgroup_bpf_detach(struct cgroup *cgrp, struct bpf_prog *prog,
86 enum bpf_attach_type type, u32 flags);
468e2f64
AS
87int cgroup_bpf_query(struct cgroup *cgrp, const union bpf_attr *attr,
88 union bpf_attr __user *uattr);
30070984 89
b2cd1257
DA
90int __cgroup_bpf_run_filter_skb(struct sock *sk,
91 struct sk_buff *skb,
92 enum bpf_attach_type type);
93
61023658
DA
94int __cgroup_bpf_run_filter_sk(struct sock *sk,
95 enum bpf_attach_type type);
96
4fbac77d
AI
97int __cgroup_bpf_run_filter_sock_addr(struct sock *sk,
98 struct sockaddr *uaddr,
1cedee13
AI
99 enum bpf_attach_type type,
100 void *t_ctx);
4fbac77d 101
40304b2a
LB
102int __cgroup_bpf_run_filter_sock_ops(struct sock *sk,
103 struct bpf_sock_ops_kern *sock_ops,
104 enum bpf_attach_type type);
105
ebc614f6
RG
106int __cgroup_bpf_check_dev_permission(short dev_type, u32 major, u32 minor,
107 short access, enum bpf_attach_type type);
108
8bad74f9
RG
109static inline enum bpf_cgroup_storage_type cgroup_storage_type(
110 struct bpf_map *map)
aa0ad5b0 111{
8bad74f9
RG
112 return BPF_CGROUP_STORAGE_SHARED;
113}
114
115static inline void bpf_cgroup_storage_set(struct bpf_cgroup_storage
116 *storage[MAX_BPF_CGROUP_STORAGE_TYPE])
117{
118 enum bpf_cgroup_storage_type stype;
aa0ad5b0 119
f294b37e
RG
120 for_each_cgroup_storage_type(stype)
121 this_cpu_write(bpf_cgroup_storage[stype], storage[stype]);
aa0ad5b0
RG
122}
123
8bad74f9
RG
124struct bpf_cgroup_storage *bpf_cgroup_storage_alloc(struct bpf_prog *prog,
125 enum bpf_cgroup_storage_type stype);
de9cbbaa
RG
126void bpf_cgroup_storage_free(struct bpf_cgroup_storage *storage);
127void bpf_cgroup_storage_link(struct bpf_cgroup_storage *storage,
128 struct cgroup *cgroup,
129 enum bpf_attach_type type);
130void bpf_cgroup_storage_unlink(struct bpf_cgroup_storage *storage);
131int bpf_cgroup_storage_assign(struct bpf_prog *prog, struct bpf_map *map);
132void bpf_cgroup_storage_release(struct bpf_prog *prog, struct bpf_map *map);
133
b2cd1257
DA
134/* Wrappers for __cgroup_bpf_run_filter_skb() guarded by cgroup_bpf_enabled. */
135#define BPF_CGROUP_RUN_PROG_INET_INGRESS(sk, skb) \
136({ \
137 int __ret = 0; \
138 if (cgroup_bpf_enabled) \
139 __ret = __cgroup_bpf_run_filter_skb(sk, skb, \
140 BPF_CGROUP_INET_INGRESS); \
141 \
142 __ret; \
30070984
DM
143})
144
b2cd1257
DA
145#define BPF_CGROUP_RUN_PROG_INET_EGRESS(sk, skb) \
146({ \
147 int __ret = 0; \
148 if (cgroup_bpf_enabled && sk && sk == skb->sk) { \
149 typeof(sk) __sk = sk_to_full_sk(sk); \
150 if (sk_fullsock(__sk)) \
151 __ret = __cgroup_bpf_run_filter_skb(__sk, skb, \
152 BPF_CGROUP_INET_EGRESS); \
153 } \
154 __ret; \
30070984
DM
155})
156
aac3fc32 157#define BPF_CGROUP_RUN_SK_PROG(sk, type) \
61023658
DA
158({ \
159 int __ret = 0; \
ee07862f 160 if (cgroup_bpf_enabled) { \
aac3fc32 161 __ret = __cgroup_bpf_run_filter_sk(sk, type); \
61023658
DA
162 } \
163 __ret; \
164})
165
aac3fc32
AI
166#define BPF_CGROUP_RUN_PROG_INET_SOCK(sk) \
167 BPF_CGROUP_RUN_SK_PROG(sk, BPF_CGROUP_INET_SOCK_CREATE)
168
169#define BPF_CGROUP_RUN_PROG_INET4_POST_BIND(sk) \
170 BPF_CGROUP_RUN_SK_PROG(sk, BPF_CGROUP_INET4_POST_BIND)
171
172#define BPF_CGROUP_RUN_PROG_INET6_POST_BIND(sk) \
173 BPF_CGROUP_RUN_SK_PROG(sk, BPF_CGROUP_INET6_POST_BIND)
174
4fbac77d
AI
175#define BPF_CGROUP_RUN_SA_PROG(sk, uaddr, type) \
176({ \
177 int __ret = 0; \
178 if (cgroup_bpf_enabled) \
1cedee13
AI
179 __ret = __cgroup_bpf_run_filter_sock_addr(sk, uaddr, type, \
180 NULL); \
4fbac77d
AI
181 __ret; \
182})
183
1cedee13 184#define BPF_CGROUP_RUN_SA_PROG_LOCK(sk, uaddr, type, t_ctx) \
d74bad4e
AI
185({ \
186 int __ret = 0; \
187 if (cgroup_bpf_enabled) { \
188 lock_sock(sk); \
1cedee13
AI
189 __ret = __cgroup_bpf_run_filter_sock_addr(sk, uaddr, type, \
190 t_ctx); \
d74bad4e
AI
191 release_sock(sk); \
192 } \
193 __ret; \
194})
195
4fbac77d
AI
196#define BPF_CGROUP_RUN_PROG_INET4_BIND(sk, uaddr) \
197 BPF_CGROUP_RUN_SA_PROG(sk, uaddr, BPF_CGROUP_INET4_BIND)
198
199#define BPF_CGROUP_RUN_PROG_INET6_BIND(sk, uaddr) \
200 BPF_CGROUP_RUN_SA_PROG(sk, uaddr, BPF_CGROUP_INET6_BIND)
201
d74bad4e
AI
202#define BPF_CGROUP_PRE_CONNECT_ENABLED(sk) (cgroup_bpf_enabled && \
203 sk->sk_prot->pre_connect)
204
205#define BPF_CGROUP_RUN_PROG_INET4_CONNECT(sk, uaddr) \
206 BPF_CGROUP_RUN_SA_PROG(sk, uaddr, BPF_CGROUP_INET4_CONNECT)
207
208#define BPF_CGROUP_RUN_PROG_INET6_CONNECT(sk, uaddr) \
209 BPF_CGROUP_RUN_SA_PROG(sk, uaddr, BPF_CGROUP_INET6_CONNECT)
210
211#define BPF_CGROUP_RUN_PROG_INET4_CONNECT_LOCK(sk, uaddr) \
1cedee13 212 BPF_CGROUP_RUN_SA_PROG_LOCK(sk, uaddr, BPF_CGROUP_INET4_CONNECT, NULL)
d74bad4e
AI
213
214#define BPF_CGROUP_RUN_PROG_INET6_CONNECT_LOCK(sk, uaddr) \
1cedee13
AI
215 BPF_CGROUP_RUN_SA_PROG_LOCK(sk, uaddr, BPF_CGROUP_INET6_CONNECT, NULL)
216
217#define BPF_CGROUP_RUN_PROG_UDP4_SENDMSG_LOCK(sk, uaddr, t_ctx) \
218 BPF_CGROUP_RUN_SA_PROG_LOCK(sk, uaddr, BPF_CGROUP_UDP4_SENDMSG, t_ctx)
219
220#define BPF_CGROUP_RUN_PROG_UDP6_SENDMSG_LOCK(sk, uaddr, t_ctx) \
221 BPF_CGROUP_RUN_SA_PROG_LOCK(sk, uaddr, BPF_CGROUP_UDP6_SENDMSG, t_ctx)
d74bad4e 222
40304b2a
LB
223#define BPF_CGROUP_RUN_PROG_SOCK_OPS(sock_ops) \
224({ \
225 int __ret = 0; \
226 if (cgroup_bpf_enabled && (sock_ops)->sk) { \
227 typeof(sk) __sk = sk_to_full_sk((sock_ops)->sk); \
df39a9f1 228 if (__sk && sk_fullsock(__sk)) \
40304b2a
LB
229 __ret = __cgroup_bpf_run_filter_sock_ops(__sk, \
230 sock_ops, \
231 BPF_CGROUP_SOCK_OPS); \
232 } \
233 __ret; \
234})
ebc614f6
RG
235
236#define BPF_CGROUP_RUN_PROG_DEVICE_CGROUP(type, major, minor, access) \
237({ \
238 int __ret = 0; \
239 if (cgroup_bpf_enabled) \
240 __ret = __cgroup_bpf_check_dev_permission(type, major, minor, \
241 access, \
242 BPF_CGROUP_DEVICE); \
243 \
244 __ret; \
245})
fdb5c453
SY
246int cgroup_bpf_prog_attach(const union bpf_attr *attr,
247 enum bpf_prog_type ptype, struct bpf_prog *prog);
248int cgroup_bpf_prog_detach(const union bpf_attr *attr,
249 enum bpf_prog_type ptype);
250int cgroup_bpf_prog_query(const union bpf_attr *attr,
251 union bpf_attr __user *uattr);
30070984
DM
252#else
253
fdb5c453 254struct bpf_prog;
30070984
DM
255struct cgroup_bpf {};
256static inline void cgroup_bpf_put(struct cgroup *cgrp) {}
324bda9e 257static inline int cgroup_bpf_inherit(struct cgroup *cgrp) { return 0; }
30070984 258
fdb5c453
SY
259static inline int cgroup_bpf_prog_attach(const union bpf_attr *attr,
260 enum bpf_prog_type ptype,
261 struct bpf_prog *prog)
262{
263 return -EINVAL;
264}
265
266static inline int cgroup_bpf_prog_detach(const union bpf_attr *attr,
267 enum bpf_prog_type ptype)
268{
269 return -EINVAL;
270}
271
272static inline int cgroup_bpf_prog_query(const union bpf_attr *attr,
273 union bpf_attr __user *uattr)
274{
275 return -EINVAL;
276}
277
8bad74f9
RG
278static inline void bpf_cgroup_storage_set(
279 struct bpf_cgroup_storage *storage[MAX_BPF_CGROUP_STORAGE_TYPE]) {}
de9cbbaa
RG
280static inline int bpf_cgroup_storage_assign(struct bpf_prog *prog,
281 struct bpf_map *map) { return 0; }
282static inline void bpf_cgroup_storage_release(struct bpf_prog *prog,
283 struct bpf_map *map) {}
284static inline struct bpf_cgroup_storage *bpf_cgroup_storage_alloc(
8bad74f9 285 struct bpf_prog *prog, enum bpf_cgroup_storage_type stype) { return 0; }
de9cbbaa
RG
286static inline void bpf_cgroup_storage_free(
287 struct bpf_cgroup_storage *storage) {}
288
13193b0f 289#define cgroup_bpf_enabled (0)
d74bad4e 290#define BPF_CGROUP_PRE_CONNECT_ENABLED(sk) (0)
30070984
DM
291#define BPF_CGROUP_RUN_PROG_INET_INGRESS(sk,skb) ({ 0; })
292#define BPF_CGROUP_RUN_PROG_INET_EGRESS(sk,skb) ({ 0; })
61023658 293#define BPF_CGROUP_RUN_PROG_INET_SOCK(sk) ({ 0; })
4fbac77d
AI
294#define BPF_CGROUP_RUN_PROG_INET4_BIND(sk, uaddr) ({ 0; })
295#define BPF_CGROUP_RUN_PROG_INET6_BIND(sk, uaddr) ({ 0; })
aac3fc32
AI
296#define BPF_CGROUP_RUN_PROG_INET4_POST_BIND(sk) ({ 0; })
297#define BPF_CGROUP_RUN_PROG_INET6_POST_BIND(sk) ({ 0; })
d74bad4e
AI
298#define BPF_CGROUP_RUN_PROG_INET4_CONNECT(sk, uaddr) ({ 0; })
299#define BPF_CGROUP_RUN_PROG_INET4_CONNECT_LOCK(sk, uaddr) ({ 0; })
300#define BPF_CGROUP_RUN_PROG_INET6_CONNECT(sk, uaddr) ({ 0; })
301#define BPF_CGROUP_RUN_PROG_INET6_CONNECT_LOCK(sk, uaddr) ({ 0; })
1cedee13
AI
302#define BPF_CGROUP_RUN_PROG_UDP4_SENDMSG_LOCK(sk, uaddr, t_ctx) ({ 0; })
303#define BPF_CGROUP_RUN_PROG_UDP6_SENDMSG_LOCK(sk, uaddr, t_ctx) ({ 0; })
40304b2a 304#define BPF_CGROUP_RUN_PROG_SOCK_OPS(sock_ops) ({ 0; })
ebc614f6 305#define BPF_CGROUP_RUN_PROG_DEVICE_CGROUP(type,major,minor,access) ({ 0; })
30070984 306
8bad74f9
RG
307#define for_each_cgroup_storage_type(stype) for (; false; )
308
30070984
DM
309#endif /* CONFIG_CGROUP_BPF */
310
311#endif /* _BPF_CGROUP_H */