xref: /OK3568_Linux_fs/kernel/tools/testing/selftests/bpf/progs/bpf_iter.h (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /* SPDX-License-Identifier: GPL-2.0 */
2*4882a593Smuzhiyun /* Copyright (c) 2020 Facebook */
3*4882a593Smuzhiyun /* "undefine" structs in vmlinux.h, because we "override" them below */
4*4882a593Smuzhiyun #define bpf_iter_meta bpf_iter_meta___not_used
5*4882a593Smuzhiyun #define bpf_iter__bpf_map bpf_iter__bpf_map___not_used
6*4882a593Smuzhiyun #define bpf_iter__ipv6_route bpf_iter__ipv6_route___not_used
7*4882a593Smuzhiyun #define bpf_iter__netlink bpf_iter__netlink___not_used
8*4882a593Smuzhiyun #define bpf_iter__task bpf_iter__task___not_used
9*4882a593Smuzhiyun #define bpf_iter__task_file bpf_iter__task_file___not_used
10*4882a593Smuzhiyun #define bpf_iter__tcp bpf_iter__tcp___not_used
11*4882a593Smuzhiyun #define tcp6_sock tcp6_sock___not_used
12*4882a593Smuzhiyun #define bpf_iter__udp bpf_iter__udp___not_used
13*4882a593Smuzhiyun #define udp6_sock udp6_sock___not_used
14*4882a593Smuzhiyun #define bpf_iter__bpf_map_elem bpf_iter__bpf_map_elem___not_used
15*4882a593Smuzhiyun #define bpf_iter__bpf_sk_storage_map bpf_iter__bpf_sk_storage_map___not_used
16*4882a593Smuzhiyun #define bpf_iter__sockmap bpf_iter__sockmap___not_used
17*4882a593Smuzhiyun #define btf_ptr btf_ptr___not_used
18*4882a593Smuzhiyun #define BTF_F_COMPACT BTF_F_COMPACT___not_used
19*4882a593Smuzhiyun #define BTF_F_NONAME BTF_F_NONAME___not_used
20*4882a593Smuzhiyun #define BTF_F_PTR_RAW BTF_F_PTR_RAW___not_used
21*4882a593Smuzhiyun #define BTF_F_ZERO BTF_F_ZERO___not_used
22*4882a593Smuzhiyun #include "vmlinux.h"
23*4882a593Smuzhiyun #undef bpf_iter_meta
24*4882a593Smuzhiyun #undef bpf_iter__bpf_map
25*4882a593Smuzhiyun #undef bpf_iter__ipv6_route
26*4882a593Smuzhiyun #undef bpf_iter__netlink
27*4882a593Smuzhiyun #undef bpf_iter__task
28*4882a593Smuzhiyun #undef bpf_iter__task_file
29*4882a593Smuzhiyun #undef bpf_iter__tcp
30*4882a593Smuzhiyun #undef tcp6_sock
31*4882a593Smuzhiyun #undef bpf_iter__udp
32*4882a593Smuzhiyun #undef udp6_sock
33*4882a593Smuzhiyun #undef bpf_iter__bpf_map_elem
34*4882a593Smuzhiyun #undef bpf_iter__bpf_sk_storage_map
35*4882a593Smuzhiyun #undef bpf_iter__sockmap
36*4882a593Smuzhiyun #undef btf_ptr
37*4882a593Smuzhiyun #undef BTF_F_COMPACT
38*4882a593Smuzhiyun #undef BTF_F_NONAME
39*4882a593Smuzhiyun #undef BTF_F_PTR_RAW
40*4882a593Smuzhiyun #undef BTF_F_ZERO
41*4882a593Smuzhiyun 
42*4882a593Smuzhiyun struct bpf_iter_meta {
43*4882a593Smuzhiyun 	struct seq_file *seq;
44*4882a593Smuzhiyun 	__u64 session_id;
45*4882a593Smuzhiyun 	__u64 seq_num;
46*4882a593Smuzhiyun } __attribute__((preserve_access_index));
47*4882a593Smuzhiyun 
48*4882a593Smuzhiyun struct bpf_iter__ipv6_route {
49*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
50*4882a593Smuzhiyun 	struct fib6_info *rt;
51*4882a593Smuzhiyun } __attribute__((preserve_access_index));
52*4882a593Smuzhiyun 
53*4882a593Smuzhiyun struct bpf_iter__netlink {
54*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
55*4882a593Smuzhiyun 	struct netlink_sock *sk;
56*4882a593Smuzhiyun } __attribute__((preserve_access_index));
57*4882a593Smuzhiyun 
58*4882a593Smuzhiyun struct bpf_iter__task {
59*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
60*4882a593Smuzhiyun 	struct task_struct *task;
61*4882a593Smuzhiyun } __attribute__((preserve_access_index));
62*4882a593Smuzhiyun 
63*4882a593Smuzhiyun struct bpf_iter__task_file {
64*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
65*4882a593Smuzhiyun 	struct task_struct *task;
66*4882a593Smuzhiyun 	__u32 fd;
67*4882a593Smuzhiyun 	struct file *file;
68*4882a593Smuzhiyun } __attribute__((preserve_access_index));
69*4882a593Smuzhiyun 
70*4882a593Smuzhiyun struct bpf_iter__bpf_map {
71*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
72*4882a593Smuzhiyun 	struct bpf_map *map;
73*4882a593Smuzhiyun } __attribute__((preserve_access_index));
74*4882a593Smuzhiyun 
75*4882a593Smuzhiyun struct bpf_iter__tcp {
76*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
77*4882a593Smuzhiyun 	struct sock_common *sk_common;
78*4882a593Smuzhiyun 	uid_t uid;
79*4882a593Smuzhiyun } __attribute__((preserve_access_index));
80*4882a593Smuzhiyun 
81*4882a593Smuzhiyun struct tcp6_sock {
82*4882a593Smuzhiyun 	struct tcp_sock	tcp;
83*4882a593Smuzhiyun 	struct ipv6_pinfo inet6;
84*4882a593Smuzhiyun } __attribute__((preserve_access_index));
85*4882a593Smuzhiyun 
86*4882a593Smuzhiyun struct bpf_iter__udp {
87*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
88*4882a593Smuzhiyun 	struct udp_sock *udp_sk;
89*4882a593Smuzhiyun 	uid_t uid __attribute__((aligned(8)));
90*4882a593Smuzhiyun 	int bucket __attribute__((aligned(8)));
91*4882a593Smuzhiyun } __attribute__((preserve_access_index));
92*4882a593Smuzhiyun 
93*4882a593Smuzhiyun struct udp6_sock {
94*4882a593Smuzhiyun 	struct udp_sock	udp;
95*4882a593Smuzhiyun 	struct ipv6_pinfo inet6;
96*4882a593Smuzhiyun } __attribute__((preserve_access_index));
97*4882a593Smuzhiyun 
98*4882a593Smuzhiyun struct bpf_iter__bpf_map_elem {
99*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
100*4882a593Smuzhiyun 	struct bpf_map *map;
101*4882a593Smuzhiyun 	void *key;
102*4882a593Smuzhiyun 	void *value;
103*4882a593Smuzhiyun };
104*4882a593Smuzhiyun 
105*4882a593Smuzhiyun struct bpf_iter__bpf_sk_storage_map {
106*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
107*4882a593Smuzhiyun 	struct bpf_map *map;
108*4882a593Smuzhiyun 	struct sock *sk;
109*4882a593Smuzhiyun 	void *value;
110*4882a593Smuzhiyun };
111*4882a593Smuzhiyun 
112*4882a593Smuzhiyun struct bpf_iter__sockmap {
113*4882a593Smuzhiyun 	struct bpf_iter_meta *meta;
114*4882a593Smuzhiyun 	struct bpf_map *map;
115*4882a593Smuzhiyun 	void *key;
116*4882a593Smuzhiyun 	struct sock *sk;
117*4882a593Smuzhiyun };
118*4882a593Smuzhiyun 
119*4882a593Smuzhiyun struct btf_ptr {
120*4882a593Smuzhiyun 	void *ptr;
121*4882a593Smuzhiyun 	__u32 type_id;
122*4882a593Smuzhiyun 	__u32 flags;
123*4882a593Smuzhiyun };
124*4882a593Smuzhiyun 
125*4882a593Smuzhiyun enum {
126*4882a593Smuzhiyun 	BTF_F_COMPACT	=	(1ULL << 0),
127*4882a593Smuzhiyun 	BTF_F_NONAME	=	(1ULL << 1),
128*4882a593Smuzhiyun 	BTF_F_PTR_RAW	=	(1ULL << 2),
129*4882a593Smuzhiyun 	BTF_F_ZERO	=	(1ULL << 3),
130*4882a593Smuzhiyun };
131