1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2024 Benjamin Tissoires
3 */
4
5 #include "bpf_experimental.h"
6 #include <bpf/bpf_helpers.h>
7 #include "bpf_misc.h"
8 #include "../test_kmods/bpf_testmod_kfunc.h"
9
10 char _license[] SEC("license") = "GPL";
11
12 struct elem {
13 struct bpf_wq w;
14 };
15
16 struct {
17 __uint(type, BPF_MAP_TYPE_ARRAY);
18 __uint(max_entries, 2);
19 __type(key, int);
20 __type(value, struct elem);
21 } array SEC(".maps");
22
23 struct {
24 __uint(type, BPF_MAP_TYPE_LRU_HASH);
25 __uint(max_entries, 4);
26 __type(key, int);
27 __type(value, struct elem);
28 } lru SEC(".maps");
29
30 /* callback for non sleepable workqueue */
wq_callback(void * map,int * key,void * value)31 static int wq_callback(void *map, int *key, void *value)
32 {
33 bpf_kfunc_common_test();
34 return 0;
35 }
36
37 /* callback for sleepable workqueue */
wq_cb_sleepable(void * map,int * key,void * value)38 static int wq_cb_sleepable(void *map, int *key, void *value)
39 {
40 bpf_kfunc_call_test_sleepable();
41 return 0;
42 }
43
44 SEC("tc")
45 /* test that bpf_wq_init takes a map as a second argument
46 */
47 __log_level(2)
__flag(BPF_F_TEST_STATE_FREQ)48 __flag(BPF_F_TEST_STATE_FREQ)
49 __failure
50 __msg(": (85) call bpf_wq_init#") /* anchor message */
51 __msg("pointer in R2 isn't map pointer")
52 long test_wq_init_nomap(void *ctx)
53 {
54 struct bpf_wq *wq;
55 struct elem *val;
56 int key = 0;
57
58 val = bpf_map_lookup_elem(&array, &key);
59 if (!val)
60 return -1;
61
62 wq = &val->w;
63 if (bpf_wq_init(wq, &key, 0) != 0)
64 return -3;
65
66 return 0;
67 }
68
69 SEC("tc")
70 /* test that the workqueue is part of the map in bpf_wq_init
71 */
72 __log_level(2)
__flag(BPF_F_TEST_STATE_FREQ)73 __flag(BPF_F_TEST_STATE_FREQ)
74 __failure
75 __msg(": (85) call bpf_wq_init#") /* anchor message */
76 __msg("workqueue pointer in R1 map_uid=0 doesn't match map pointer in R2 map_uid=0")
77 long test_wq_init_wrong_map(void *ctx)
78 {
79 struct bpf_wq *wq;
80 struct elem *val;
81 int key = 0;
82
83 val = bpf_map_lookup_elem(&array, &key);
84 if (!val)
85 return -1;
86
87 wq = &val->w;
88 if (bpf_wq_init(wq, &lru, 0) != 0)
89 return -3;
90
91 return 0;
92 }
93
94 SEC("?tc")
95 __log_level(2)
96 __failure
97 /* check that the first argument of bpf_wq_set_callback()
98 * is a correct bpf_wq pointer.
99 */
100 __msg(": (85) call bpf_wq_set_callback_impl#") /* anchor message */
101 __msg("arg#0 doesn't point to a map value")
test_wrong_wq_pointer(void * ctx)102 long test_wrong_wq_pointer(void *ctx)
103 {
104 int key = 0;
105 struct bpf_wq *wq;
106
107 wq = bpf_map_lookup_elem(&array, &key);
108 if (!wq)
109 return 1;
110
111 if (bpf_wq_init(wq, &array, 0))
112 return 2;
113
114 if (bpf_wq_set_callback((void *)&wq, wq_callback, 0))
115 return 3;
116
117 return -22;
118 }
119
120 SEC("?tc")
121 __log_level(2)
122 __failure
123 /* check that the first argument of bpf_wq_set_callback()
124 * is a correct bpf_wq pointer.
125 */
126 __msg(": (85) call bpf_wq_set_callback_impl#") /* anchor message */
127 __msg("off 1 doesn't point to 'struct bpf_wq' that is at 0")
test_wrong_wq_pointer_offset(void * ctx)128 long test_wrong_wq_pointer_offset(void *ctx)
129 {
130 int key = 0;
131 struct bpf_wq *wq;
132
133 wq = bpf_map_lookup_elem(&array, &key);
134 if (!wq)
135 return 1;
136
137 if (bpf_wq_init(wq, &array, 0))
138 return 2;
139
140 if (bpf_wq_set_callback((void *)wq + 1, wq_cb_sleepable, 0))
141 return 3;
142
143 return -22;
144 }
145