1#include "bpf_experimental.h"
2#include "bpf_misc.h"
3
4struct val_t {
5	long b, c, d;
6};
7
8struct val2_t {
9	long b;
10};
11
12struct val_with_ptr_t {
13	char *p;
14};
15
16struct val_with_rb_root_t {
17	struct bpf_spin_lock lock;
18};
19
20struct val_600b_t {
21	char b[600];
22};
23
24struct elem {
25	long sum;
26	struct val_t __percpu_kptr *pc;
27};
28
29struct {
30	__uint(type, BPF_MAP_TYPE_ARRAY);
31	__uint(max_entries, 1);
32	__type(key, int);
33	__type(value, struct elem);
34} array SEC(".maps");
35
36long ret;
37
38SEC("?fentry/bpf_fentry_test1")
39__failure __msg("store to referenced kptr disallowed")
40int BPF_PROG(test_array_map_1)
41{
42	struct val_t __percpu_kptr *p;
43	struct elem *e;
44	int index = 0;
45
46	e = bpf_map_lookup_elem(&array, &index);
47	if (!e)
48		return 0;
49
50	p = bpf_percpu_obj_new(struct val_t);
51	if (!p)
52		return 0;
53
54	p = bpf_kptr_xchg(&e->pc, p);
55	if (p)
56		bpf_percpu_obj_drop(p);
57
58	e->pc = (struct val_t __percpu_kptr *)ret;
59	return 0;
60}
61
62SEC("?fentry/bpf_fentry_test1")
63__failure __msg("invalid kptr access, R2 type=percpu_ptr_val2_t expected=ptr_val_t")
64int BPF_PROG(test_array_map_2)
65{
66	struct val2_t __percpu_kptr *p2;
67	struct val_t __percpu_kptr *p;
68	struct elem *e;
69	int index = 0;
70
71	e = bpf_map_lookup_elem(&array, &index);
72	if (!e)
73		return 0;
74
75	p2 = bpf_percpu_obj_new(struct val2_t);
76	if (!p2)
77		return 0;
78
79	p = bpf_kptr_xchg(&e->pc, p2);
80	if (p)
81		bpf_percpu_obj_drop(p);
82
83	return 0;
84}
85
86SEC("?fentry.s/bpf_fentry_test1")
87__failure __msg("R1 type=scalar expected=percpu_ptr_, percpu_rcu_ptr_, percpu_trusted_ptr_")
88int BPF_PROG(test_array_map_3)
89{
90	struct val_t __percpu_kptr *p, *p1;
91	struct val_t *v;
92	struct elem *e;
93	int index = 0;
94
95	e = bpf_map_lookup_elem(&array, &index);
96	if (!e)
97		return 0;
98
99	p = bpf_percpu_obj_new(struct val_t);
100	if (!p)
101		return 0;
102
103	p1 = bpf_kptr_xchg(&e->pc, p);
104	if (p1)
105		bpf_percpu_obj_drop(p1);
106
107	v = bpf_this_cpu_ptr(p);
108	ret = v->b;
109	return 0;
110}
111
112SEC("?fentry.s/bpf_fentry_test1")
113__failure __msg("arg#0 expected for bpf_percpu_obj_drop_impl()")
114int BPF_PROG(test_array_map_4)
115{
116	struct val_t __percpu_kptr *p;
117
118	p = bpf_percpu_obj_new(struct val_t);
119	if (!p)
120		return 0;
121
122	bpf_obj_drop(p);
123	return 0;
124}
125
126SEC("?fentry.s/bpf_fentry_test1")
127__failure __msg("arg#0 expected for bpf_obj_drop_impl()")
128int BPF_PROG(test_array_map_5)
129{
130	struct val_t *p;
131
132	p = bpf_obj_new(struct val_t);
133	if (!p)
134		return 0;
135
136	bpf_percpu_obj_drop(p);
137	return 0;
138}
139
140SEC("?fentry.s/bpf_fentry_test1")
141__failure __msg("bpf_percpu_obj_new type ID argument must be of a struct of scalars")
142int BPF_PROG(test_array_map_6)
143{
144	struct val_with_ptr_t __percpu_kptr *p;
145
146	p = bpf_percpu_obj_new(struct val_with_ptr_t);
147	if (!p)
148		return 0;
149
150	bpf_percpu_obj_drop(p);
151	return 0;
152}
153
154SEC("?fentry.s/bpf_fentry_test1")
155__failure __msg("bpf_percpu_obj_new type ID argument must not contain special fields")
156int BPF_PROG(test_array_map_7)
157{
158	struct val_with_rb_root_t __percpu_kptr *p;
159
160	p = bpf_percpu_obj_new(struct val_with_rb_root_t);
161	if (!p)
162		return 0;
163
164	bpf_percpu_obj_drop(p);
165	return 0;
166}
167
168SEC("?fentry.s/bpf_fentry_test1")
169__failure __msg("bpf_percpu_obj_new type size (600) is greater than 512")
170int BPF_PROG(test_array_map_8)
171{
172	struct val_600b_t __percpu_kptr *p;
173
174	p = bpf_percpu_obj_new(struct val_600b_t);
175	if (!p)
176		return 0;
177
178	bpf_percpu_obj_drop(p);
179	return 0;
180}
181
182char _license[] SEC("license") = "GPL";
183