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 "../bpf_testmod/bpf_testmod_kfunc.h"
  9
 10char _license[] SEC("license") = "GPL";
 11
 12struct elem {
 13	struct bpf_wq w;
 14};
 15
 16struct {
 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
 23struct {
 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 */
 31static 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 */
 38static int wq_cb_sleepable(void *map, int *key, void *value)
 39{
 40	bpf_kfunc_call_test_sleepable();
 41	return 0;
 42}
 43
 44SEC("tc")
 45/* test that bpf_wq_init takes a map as a second argument
 46 */
 47__log_level(2)
 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")
 52long 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
 69SEC("tc")
 70/* test that the workqueue is part of the map in bpf_wq_init
 71 */
 72__log_level(2)
 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")
 77long 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
 94SEC("?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")
102long 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
120SEC("?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")
128long 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}